198 lines
5.1 KiB
Python
198 lines
5.1 KiB
Python
# ##### BEGIN GPL LICENSE BLOCK #####
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
# <pep8 compliant>
|
|
|
|
import sys
|
|
import os
|
|
|
|
|
|
# may split this out into a new file
|
|
def replace_bpy_app_version():
|
|
""" So MD5's are predictable from output which uses blenders versions.
|
|
"""
|
|
|
|
import bpy
|
|
|
|
app = bpy.app
|
|
app_fake = type(bpy)("bpy.app")
|
|
|
|
for attr in dir(app):
|
|
if not attr.startswith("_"):
|
|
setattr(app_fake, attr, getattr(app, attr))
|
|
|
|
app_fake.version = 0, 0, 0
|
|
app_fake.version_string = "0.00 (sub 0)"
|
|
bpy.app = app_fake
|
|
|
|
|
|
def clear_startup_blend():
|
|
import bpy
|
|
|
|
for scene in bpy.data.scenes:
|
|
for obj in scene.objects:
|
|
scene.objects.unlink(obj)
|
|
|
|
|
|
def blend_to_md5():
|
|
import bpy
|
|
scene = bpy.context.scene
|
|
ROUND = 4
|
|
|
|
def matrix2str(matrix):
|
|
return "".join([str(round(axis, ROUND)) for vector in matrix for axis in vector]).encode('ASCII')
|
|
|
|
def coords2str(seq, attr):
|
|
return "".join([str(round(axis, ROUND)) for vertex in seq for axis in getattr(vertex, attr)]).encode('ASCII')
|
|
|
|
import hashlib
|
|
|
|
md5 = hashlib.new("md5")
|
|
md5_update = md5.update
|
|
|
|
for obj in scene.objects:
|
|
md5_update(matrix2str(obj.matrix_world))
|
|
data = obj.data
|
|
|
|
if type(data) == bpy.types.Mesh:
|
|
md5_update(coords2str(data.vertices, "co"))
|
|
elif type(data) == bpy.types.Curve:
|
|
for spline in data.splines:
|
|
md5_update(coords2str(spline.bezier_points, "co"))
|
|
md5_update(coords2str(spline.points, "co"))
|
|
|
|
return md5.hexdigest()
|
|
|
|
|
|
def main():
|
|
argv = sys.argv
|
|
print(" args:", " ".join(argv))
|
|
argv = argv[argv.index("--") + 1:]
|
|
|
|
def arg_extract(arg, optional=True, array=False):
|
|
arg += "="
|
|
if array:
|
|
value = []
|
|
else:
|
|
value = None
|
|
|
|
i = 0
|
|
while i < len(argv):
|
|
if argv[i].startswith(arg):
|
|
item = argv[i][len(arg):]
|
|
del argv[i]
|
|
i -= 1
|
|
|
|
if array:
|
|
value.append(item)
|
|
else:
|
|
value = item
|
|
break
|
|
|
|
i += 1
|
|
|
|
if (not value) and (not optional):
|
|
print(" '%s' not set" % arg)
|
|
sys.exit(1)
|
|
|
|
return value
|
|
|
|
run = arg_extract("--run", optional=False)
|
|
md5 = arg_extract("--md5", optional=False)
|
|
md5_method = arg_extract("--md5_method", optional=False) # 'SCENE' / 'FILE'
|
|
|
|
# only when md5_method is 'FILE'
|
|
md5_source = arg_extract("--md5_source", optional=True, array=True)
|
|
|
|
# save blend file, for testing
|
|
write_blend = arg_extract("--write-blend", optional=True)
|
|
|
|
# ensure files are written anew
|
|
for f in md5_source:
|
|
if os.path.exists(f):
|
|
os.remove(f)
|
|
|
|
import bpy
|
|
|
|
replace_bpy_app_version()
|
|
if not bpy.data.filepath:
|
|
clear_startup_blend()
|
|
|
|
print(" Running: '%s'" % run)
|
|
print(" MD5: '%s'!" % md5)
|
|
|
|
try:
|
|
result = eval(run)
|
|
except:
|
|
import traceback
|
|
traceback.print_exc()
|
|
sys.exit(1)
|
|
|
|
if write_blend is not None:
|
|
print(" Writing Blend: %s" % write_blend)
|
|
bpy.ops.wm.save_mainfile(filepath=write_blend, check_existing=False)
|
|
|
|
print(" Result: '%s'" % str(result))
|
|
if not result:
|
|
print(" Running: %s -> False" % run)
|
|
sys.exit(1)
|
|
|
|
if md5_method == 'SCENE':
|
|
md5_new = blend_to_md5()
|
|
elif md5_method == 'FILE':
|
|
if not md5_source:
|
|
print(" Missing --md5_source argument")
|
|
sys.exit(1)
|
|
|
|
for f in md5_source:
|
|
if not os.path.exists(f):
|
|
print(" Missing --md5_source=%r argument does not point to a file")
|
|
sys.exit(1)
|
|
|
|
import hashlib
|
|
|
|
md5_instance = hashlib.new("md5")
|
|
md5_update = md5_instance.update
|
|
|
|
for f in md5_source:
|
|
filehandle = open(f, "rb")
|
|
md5_update(filehandle.read())
|
|
filehandle.close()
|
|
|
|
md5_new = md5_instance.hexdigest()
|
|
|
|
else:
|
|
print(" Invalid --md5_method=%s argument is not a valid source")
|
|
sys.exit(1)
|
|
|
|
if md5 != md5_new:
|
|
print(" Running: %s\n MD5 Recieved: %s\n MD5 Expected: %s" % (run, md5_new, md5))
|
|
sys.exit(1)
|
|
|
|
print(" Success: %s" % run)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# So a python error exits(1)
|
|
try:
|
|
main()
|
|
except:
|
|
import traceback
|
|
traceback.print_exc()
|
|
sys.exit(1)
|