implement clean up.

[meshio/pymeshio.git] / blender26-meshio / import_pmx.py

diff --git a/blender26-meshio/import_pmx.py b/blender26-meshio/import_pmx.py
index d04044d..3641050 100644 (file)
--- a/blender26-meshio/import_pmx.py
+++ b/blender26-meshio/import_pmx.py
@@ -163,20 +163,43 @@ def _execute(filepath):
             for pos in (v.position for v in model.vertices)]
 
     # マテリアル毎にメッシュを作成する
+    def get_object_name(index, name):
+        """
+        object名を作る。最大21バイト
+        """
+        len_list=[len(name[:i].encode('utf-8')) for i in range(1, len(name)+1, 1)]
+        letter_count=0
+        for str_len in len_list:
+            if str_len<18: # 21-3
+                letter_count+=1
+            else:
+                break
+        name="{0:02}:{1}".format(index, name[:letter_count])
+        print("%s(%d)" % (name, letter_count))
+        return name
     for i, m in enumerate(model.materials):
-        print(m.name)
-        # material作成
+        ####################
+        # material
+        ####################
         material=__create_a_material(m, m.name, textures_and_images)
+
+        ####################
+        # mesh object
+        ####################
         # object名はutf-8で21byteまで
-        mesh, mesh_object=bl.mesh.create("object:{0:02}".format(i))
+        mesh, mesh_object=bl.mesh.create(get_object_name(i, m.name))
         bl.mesh.addMaterial(mesh, material)
         # activate object
         bl.object.deselectAll()
         bl.object.activate(mesh_object)
         bl.object.makeParent(root_object, mesh_object)
+
+        ####################
         # vertices & faces
+        ####################
         indices=[next(index_generator)
                     for _ in range(m.vertex_count)]
+        used_indices=set(indices)
         bl.mesh.addGeometry(mesh, vertices,
                 [(indices[i], indices[i+1], indices[i+2])
                     for i in range(0, len(indices), 3)])
@@ -203,6 +226,9 @@ def _execute(filepath):
                     ],
                     image)
 
+        ####################
+        # armature
+        ####################
         if armature_object:
             # armature modifirer
             bl.modifier.addArmature(mesh_object, armature_object)
@@ -221,5 +247,33 @@ def _execute(filepath):
                 else:
                     raise Exception("unknown deform: %s" % v.deform)
 
+        ####################
+        # shape keys
+        ####################
+        # set shape_key pin
+        bl.object.pinShape(mesh_object, True)
+        # create base key
+        baseShapeBlock=bl.object.addShapeKey(mesh_object, bl.BASE_SHAPE_NAME)
+        mesh.update()
+        for m in model.morphs:
+            new_shape_key=bl.object.addShapeKey(mesh_object, m.name)
+            for o in m.offsets:
+                if isinstance(o, pmx.VertexMorphOffset):
+                    bl.shapekey.assign(new_shape_key, 
+                            o.vertex_index, 
+                            mesh.vertices[o.vertex_index].co+
+                            bl.createVector(*convert_coord(o.position_offset)))
+                else:
+                    raise Exception("unknown morph type: %s" % o)
+        # select base shape
+        bl.object.setActivateShapeKey(mesh_object, 0)
+
+        #############################
+        # clean up not used vertices
+        # in the material.
+        #############################
+        bl.mesh.vertsDelete(mesh, [i for i in range(len(mesh.vertices))
+            if i not in used_indices])
+
     return {'FINISHED'}