Merge pull request #7 from Lamoot/limit-precision

Option to limit precision of exported data

Author: unelsson

io_scene_dae/__init__.py

@@ -16,7 +16,7 @@
 # ##### END GPL LICENSE BLOCK #####
 
 import bpy
-from bpy.props import StringProperty, BoolProperty, FloatProperty, EnumProperty
+from bpy.props import StringProperty, BoolProperty, FloatProperty, EnumProperty, IntProperty
 
 from bpy_extras.io_utils import ExportHelper
 bl_info = {
@@ -177,6 +177,14 @@ class CE_OT_export_dae(bpy.types.Operator, ExportHelper):
         default=False,
         )    
 
+    use_limit_precision : IntProperty(
+        name="Data Precision",
+        description=("To how many decimals are the exported values limited. \n"
+                     "The lower the number, the smaller the exported file"),
+        min=4, max=16,
+        default=6,
+        )
+
     @property
     def check_extension(self):
         return True
@@ -347,15 +355,41 @@ def draw(self, context):
         col.prop(operator, 'use_anim_skip_noexp')
         col.prop(operator, 'use_anim_optimize')
         col.prop(operator, 'anim_optimize_precision')
+ 
+class DAE_PT_export_extras(bpy.types.Panel):
+    bl_space_type = 'FILE_BROWSER'
+    bl_region_type = 'TOOL_PROPS'
+    bl_label = "Extra"
+    bl_parent_id = "FILE_PT_operator"
+    bl_options = {'DEFAULT_CLOSED'}
 
+    @classmethod
+    def poll(cls, context):
+        sfile = context.space_data
+        operator = sfile.active_operator
+            
+        return operator.bl_idname == "EXPORT_SCENE_OT_dae"
+
+    def draw(self, context):
+        layout = self.layout
+        layout.use_property_split = True
+        layout.use_property_decorate = False
+        
+        sfile = context.space_data
+        operator = sfile.active_operator
+          
+        col = layout.column(align = True)
+        col.prop(operator, 'use_limit_precision')
 
 classes = (
     CE_OT_export_dae,
     DAE_PT_export_include,
     DAE_PT_export_transform,
     DAE_PT_export_geometry,
     DAE_PT_export_armature,
-    DAE_PT_export_animation
+    DAE_PT_export_animation,
+    DAE_PT_export_extras
+
 )
 
 def register():  

io_scene_dae/export_dae.py

@@ -57,16 +57,16 @@ def snap_tup(tup):
     return tup
 
 
-def strmtx(mtx, scale=1.0):
+def strmtx(mtx, scale=1.0, limit=16):
     # When scaling the exported data, we only want to affect distances,
     # so we only multiply location values in the transform matrix.
     mtx_util = Matrix([[0,0,0,0], [0,0,0,0],[0,0,0,0],[0,0,0,scale-1]])
     mtx = (mtx * scale).normalized() - mtx_util    
-    
+
     s = ""
     for x in range(4):
         for y in range(4):
-            s += "{} ".format(mtx[x][y])
+            s += "{} ".format(round((mtx[x][y]), limit))
     s = " {} ".format(s)
     return s
 
@@ -776,9 +776,12 @@ def export_mesh(self, node, armature=None, skeyindex=-1, skel_source=None,
         self.writel(S_GEOM, 3, "<source id=\"{}-positions\">".format(meshid))
         float_values = ""
         sf = self.config["scale_factor"]
+        lim = self.config["use_limit_precision"]
         for v in vertices:
             float_values += " {} {} {}".format(
-                v.vertex.x * sf, v.vertex.y * sf, v.vertex.z * sf)
+                round(v.vertex.x * sf, lim),
+                round(v.vertex.y * sf, lim),
+                round(v.vertex.z * sf, lim))
         self.writel(
             S_GEOM, 4, "<float_array id=\"{}-positions-array\" "
             "count=\"{}\">{}</float_array>".format(
@@ -799,7 +802,9 @@ def export_mesh(self, node, armature=None, skeyindex=-1, skel_source=None,
         float_values = ""
         for v in vertices:
             float_values += " {} {} {}".format(
-                v.normal.x, v.normal.y, v.normal.z)
+                round(v.normal.x, lim),
+                round(v.normal.y, lim),
+                round(v.normal.z, lim))
         self.writel(
             S_GEOM, 4, "<float_array id=\"{}-normals-array\" "
             "count=\"{}\">{}</float_array>".format(
@@ -821,7 +826,9 @@ def export_mesh(self, node, armature=None, skeyindex=-1, skel_source=None,
             float_values = ""
             for v in vertices:
                 float_values += " {} {} {}".format(
-                    v.tangent.x, v.tangent.y, v.tangent.z)
+                    round(v.tangent.x, lim),
+                    round(v.tangent.y, lim),
+                    round(v.tangent.z, lim))
             self.writel(
                 S_GEOM, 4, "<float_array id=\"{}-tangents-array\" "
                 "count=\"{}\">{}</float_array>".format(
@@ -842,7 +849,9 @@ def export_mesh(self, node, armature=None, skeyindex=-1, skel_source=None,
             float_values = ""
             for v in vertices:
                 float_values += " {} {} {}".format(
-                    v.bitangent.x, v.bitangent.y, v.bitangent.z)
+                    round(v.bitangent.x, lim),
+                    round(v.bitangent.y, lim),
+                    round(v.bitangent.z, lim))
             self.writel(
                 S_GEOM, 4, "<float_array id=\"{}-bitangents-array\" "
                 "count=\"{}\">{}</float_array>".format(
@@ -865,7 +874,9 @@ def export_mesh(self, node, armature=None, skeyindex=-1, skel_source=None,
             float_values = ""
             for v in vertices:
                 try:
-                    float_values += " {} {}".format(v.uv[uvi].x, v.uv[uvi].y)
+                    float_values += " {} {}".format(
+                        round(v.uv[uvi].x, lim),
+                        round(v.uv[uvi].y, lim))
                 except:
                     # TODO: Review, understand better the multi-uv-layer API
                     float_values += " 0 0 "
@@ -891,7 +902,9 @@ def export_mesh(self, node, armature=None, skeyindex=-1, skel_source=None,
             float_values = ""
             for v in vertices:
                 float_values += " {} {} {}".format(
-                    v.color.x, v.color.y, v.color.z)
+                    round(v.color.x, lim),
+                    round(v.color.y, lim),
+                    round(v.color.z, lim))
             self.writel(
                 S_GEOM, 4, "<float_array id=\"{}-colors-array\" "
                 "count=\"{}\">{}</float_array>".format(
@@ -998,10 +1011,11 @@ def export_mesh(self, node, armature=None, skeyindex=-1, skel_source=None,
                     skel_source))
             else:
                 self.writel(S_SKIN, 2, "<skin source=\"#{}\">".format(meshid))
-
+            
+            lim = self.config["use_limit_precision"]
             self.writel(
                 S_SKIN, 3, "<bind_shape_matrix>{}</bind_shape_matrix>".format(
-                    strmtx(node.matrix_world)))
+                    strmtx(node.matrix_world, 1, lim)))
             # Joint Names
             self.writel(S_SKIN, 3, "<source id=\"{}-joints\">".format(contid))
             name_values = ""
@@ -1026,8 +1040,9 @@ def export_mesh(self, node, armature=None, skeyindex=-1, skel_source=None,
                 contid))
             pose_values = ""
             sf = self.config["scale_factor"]
+            lim = self.config["use_limit_precision"]
             for v in si["bone_bind_poses"]:
-                pose_values += " {}".format(strmtx(v, sf))
+                pose_values += " {}".format(strmtx(v, sf, lim))
 
             self.writel(
                 S_SKIN, 4, "<float_array id=\"{}-bind_poses-array\" "
@@ -1051,7 +1066,7 @@ def export_mesh(self, node, armature=None, skeyindex=-1, skel_source=None,
             for v in vertices:
                 skin_weights_total += len(v.weights)
                 for w in v.weights:
-                    skin_weights += " {}".format(w)
+                    skin_weights += " {}".format(round(w, lim))
 
             self.writel(
                 S_SKIN, 4, "<float_array id=\"{}-skin_weights-array\" "
@@ -1239,10 +1254,11 @@ def export_armature_bone(self, bone, il, si):
             si["skeleton_nodes"].append(boneid)
 
         sf = self.config["scale_factor"]
+        lim = self.config["use_limit_precision"]
         if (is_ctrl_bone is False):
             self.writel(
                 S_NODES, il, "<matrix sid=\"transform\">{}</matrix>".format(
-                    strmtx(xform, sf)))
+                    strmtx(xform, sf, lim)))
 
         for c in bone.children:
             self.export_armature_bone(c, il, si)
@@ -1584,9 +1600,10 @@ def export_node(self, node, il):
         il += 1
 
         sf = self.config["scale_factor"]
+        lim = self.config["use_limit_precision"]
         self.writel(
             S_NODES, il, "<matrix sid=\"transform\">{}</matrix>".format(
-                strmtx(node.matrix_local, sf)))
+                strmtx(node.matrix_local, sf, lim)))
         if (node.type == "MESH"):
             self.export_mesh_node(node, il)
         elif (node.type == "CURVE"):
@@ -1682,10 +1699,11 @@ def export_animation_transform_channel(self, target, keys, matrices=True):
         source_interps = ""
 
         sf = self.config["scale_factor"]
+        lim = self.config["use_limit_precision"]
         for k in keys:
             source_frames += " {}".format(k[0])
             if (matrices):
-                source_transforms += " {}".format(strmtx(k[1], sf))
+                source_transforms += " {}".format(strmtx(k[1], sf, lim))
             else:
                 source_transforms += " {}".format(k[1])
 
@@ -2019,8 +2037,9 @@ def sort_by_frame(item):
             markers.append(m)        
         markers.sort(key=sort_by_frame)
 
+        lim = self.config["use_limit_precision"]
         for m in markers:
-            tkf = round(m.frame * 1/scene.render.fps, 6)
+            tkf = round(m.frame * 1/scene.render.fps, lim)
             textkeys_output.write(m.name + " " + str(tkf) + '\n' )        
         textkeys_output.close()