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 db2c3d7..3641050 100644 (file)
--- a/blender26-meshio/import_pmx.py
+++ b/blender26-meshio/import_pmx.py
@@ -178,9 +178,14 @@ def _execute(filepath):
         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(get_object_name(i, m.name))
         bl.mesh.addMaterial(mesh, material)
@@ -188,9 +193,13 @@ def _execute(filepath):
         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)])
@@ -217,6 +226,9 @@ def _execute(filepath):
                     ],
                     image)
 
+        ####################
+        # armature
+        ####################
         if armature_object:
             # armature modifirer
             bl.modifier.addArmature(mesh_object, armature_object)
@@ -235,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'}