Merge branch 'master' of ssh://github.com/sjkillen/blenderings

vertex_animation.py

@@ -0,0 +1,255 @@
+# See notes for source repo
+
+# ##### 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>
+
+
+bl_info = {
+    "name": "Vertex Animation",
+    "author": "Joshua Bogart",
+    "version": (1, 0),
+    "blender": (2, 83, 0),
+    "location": "View3D > Sidebar > Vertex Animation",
+    "description": "A tool for storing per frame vertex data for use in a vertex shader.",
+    "warning": "",
+    "doc_url": "",
+    "category": "tool",
+}
+
+
+import bpy
+import bmesh
+import os
+
+
+def get_per_frame_mesh_data(context, data, objects):
+    """Return a list of combined mesh data per frame"""
+    meshes = []
+    for i in frame_range(context.scene):
+        context.scene.frame_set(i)
+        depsgraph = context.evaluated_depsgraph_get()
+        bm = bmesh.new()
+        for ob in objects:
+            eval_object = ob.evaluated_get(depsgraph)
+            me = data.meshes.new_from_object(eval_object)
+            me.transform(ob.matrix_world)
+            bm.from_mesh(me)
+            data.meshes.remove(me)
+        me = data.meshes.new("mesh")
+        bm.to_mesh(me)
+        bm.free()
+        me.calc_normals()
+        meshes.append(me)
+    return meshes
+
+
+def create_export_mesh_object(context, data, me):
+    """Return a mesh object with correct UVs"""
+    while len(me.uv_layers) < 2:
+        me.uv_layers.new()
+    uv_layer = me.uv_layers[1]
+    uv_layer.name = "vertex_anim"
+    for loop in me.loops:
+        uv_layer.data[loop.index].uv = ((loop.vertex_index + 0.5)/len(me.vertices), 0.0)
+    ob = data.objects.new("export_mesh", me)
+    context.scene.collection.objects.link(ob)
+    return ob
+
+
+def get_vertex_data(data, meshes):
+    """Return lists of vertex offsets and normals from a list of mesh data"""
+    original = meshes[0].vertices
+    offsets = []
+    normals = []
+    for me in meshes: #for me in reversed(meshes):
+        for v in me.vertices:
+            offset = v.co - original[v.index].co
+            x, y, z = offset
+            offsets.extend((x, -y, z, 1))
+            x, y, z = v.normal
+            normals.extend(((x + 1) * 0.5, (-y + 1) * 0.5, (z + 1) * 0.5, 1))
+        #if not me.users:
+            #data.meshes.remove(me)
+    for me in meshes:    
+        if not me.users:
+            data.meshes.remove(me)
+    return offsets, normals
+
+
+def frame_range(scene):
+    """Return a range object with with scene's frame start, end, and step"""
+    return range(scene.frame_start, scene.frame_end, scene.frame_step)
+
+
+def bake_vertex_data(context, data, offsets, normals, size):
+    """Stores vertex offsets and normals in separate image textures"""
+    width, height = size
+    
+    blend_path = bpy.data.filepath
+    blend_path = os.path.dirname(bpy.path.abspath(blend_path))
+    subfolder_path = os.path.join(blend_path, "vaexport")
+    if not os.path.exists(subfolder_path):
+        os.makedirs(subfolder_path)
+    openexr_filepath = os.path.join(subfolder_path, "offsets.exr")
+    png_filepath = os.path.join(subfolder_path, "normals.png")
+    
+    openexr_export_scene = bpy.data.scenes.new('openexr export scene')
+    openexr_export_scene.sequencer_colorspace_settings.name = 'Non-Color'
+    openexr_export_scene.render.image_settings.color_depth = '16'
+    openexr_export_scene.render.image_settings.color_mode = 'RGBA'
+    openexr_export_scene.render.image_settings.file_format = 'OPEN_EXR'
+    openexr_export_scene.render.image_settings.exr_codec = 'NONE'
+    	
+    if 'offsets' in bpy.data.images:
+        offset_tex = bpy.data.images['offsets']
+        bpy.data.images.remove(offset_tex)
+    
+    offset_texture = data.images.new(
+        name="offsets",
+        width=width,
+        height=height,
+        alpha=True,
+        float_buffer=True
+    )
+    
+    offset_texture.file_format = 'OPEN_EXR'
+    offset_texture.colorspace_settings.name = 'Non-Color'
+    offset_texture.pixels = offsets
+    offset_texture.save_render(openexr_filepath, scene=openexr_export_scene)
+    bpy.data.scenes.remove(openexr_export_scene)
+    
+    png_export_scene = bpy.data.scenes.new('png export scene')
+    png_export_scene.render.image_settings.color_depth = '8'
+    png_export_scene.render.image_settings.color_mode = 'RGBA'
+    png_export_scene.render.image_settings.file_format = 'PNG'
+    png_export_scene.render.image_settings.compression = 15
+    
+    if 'normals' in bpy.data.images:
+        normals_tex = bpy.data.images['normals']
+        bpy.data.images.remove(normals_tex)
+    
+    normal_texture = data.images.new(
+        name="normals",
+        width=width,
+        height=height,
+        alpha=True
+    )
+    
+    normal_texture.file_format = 'PNG'
+    normal_texture.pixels = normals
+    normal_texture.save_render(png_filepath, scene=png_export_scene)
+    bpy.data.scenes.remove(png_export_scene)
+
+
+class OBJECT_OT_ProcessAnimMeshes(bpy.types.Operator):
+    """Store combined per frame vertex offsets and normals for all
+    selected mesh objects into seperate image textures"""
+    bl_idname = "object.process_anim_meshes"
+    bl_label = "Process Anim Meshes"
+
+    @property
+    def allowed_modifiers(self):
+        return [
+            'ARMATURE', 'CAST', 'CURVE', 'DISPLACE', 'HOOK',
+            'LAPLACIANDEFORM', 'LATTICE', 'MESH_DEFORM',
+            'SHRINKWRAP', 'SIMPLE_DEFORM', 'SMOOTH',
+            'CORRECTIVE_SMOOTH', 'LAPLACIANSMOOTH',
+            'SURFACE_DEFORM', 'WARP', 'WAVE', 'MESH_CACHE'
+        ]
+
+    @classmethod
+    def poll(cls, context):
+        ob = context.active_object
+        return ob and ob.type == 'MESH' and ob.mode == 'OBJECT'
+
+    def execute(self, context):
+        units = context.scene.unit_settings
+        data = bpy.data
+        objects = [ob for ob in context.selected_objects if ob.type == 'MESH']
+        vertex_count = sum([len(ob.data.vertices) for ob in objects])
+        frame_count = len(frame_range(context.scene))
+        for ob in objects:
+            for mod in ob.modifiers:
+                if mod.type not in self.allowed_modifiers:
+                    self.report(
+                        {'ERROR'},
+                        f"Objects with {mod.type.title()} modifiers are not allowed!"
+                    )
+                    return {'CANCELLED'}
+        #if units.system != 'METRIC' or round(units.scale_length, 2) != 0.01:
+        #    self.report(
+        #        {'ERROR'},
+        #        "Scene Unit must be Metric with a Unit Scale of 0.01!"
+        #    )
+        #    return {'CANCELLED'}        
+        if vertex_count > 8192:
+            self.report(
+                {'ERROR'},
+                f"Vertex count of {vertex_count :,}, execedes limit of 8,192!"
+            )
+            return {'CANCELLED'}
+        if frame_count > 8192:
+            self.report(
+                {'ERROR'},
+                f"Frame count of {frame_count :,}, execedes limit of 8,192!"
+            )
+            return {'CANCELLED'}
+        meshes = get_per_frame_mesh_data(context, data, objects)
+        export_mesh_data = meshes[0].copy()
+        create_export_mesh_object(context, data, export_mesh_data)
+        offsets, normals = get_vertex_data(data, meshes)
+        texture_size = vertex_count, frame_count
+        bake_vertex_data(context, data, offsets, normals, texture_size)
+        return {'FINISHED'}
+
+
+class VIEW3D_PT_VertexAnimation(bpy.types.Panel):
+    """Creates a Panel in 3D Viewport"""
+    bl_label = "Vertex Animation"
+    bl_idname = "VIEW3D_PT_vertex_animation"
+    bl_space_type = 'VIEW_3D'
+    bl_region_type = 'UI'
+    bl_category = "Vertex Animation"
+
+    def draw(self, context):
+        layout = self.layout
+        layout.use_property_split = True
+        layout.use_property_decorate = False
+        scene = context.scene
+        col = layout.column(align=True)
+        col.prop(scene, "frame_start", text="Frame Start")
+        col.prop(scene, "frame_end", text="End")
+        col.prop(scene, "frame_step", text="Step")
+        row = layout.row()
+        row.operator("object.process_anim_meshes")
+
+
+def register():
+    bpy.utils.register_class(OBJECT_OT_ProcessAnimMeshes)
+    bpy.utils.register_class(VIEW3D_PT_VertexAnimation)
+
+
+def unregister():
+    bpy.utils.unregister_class(OBJECT_OT_ProcessAnimMeshes)
+    bpy.utils.unregister_class(VIEW3D_PT_VertexAnimation)
+
+
+if __name__ == "__main__":
+    register()

vertex_animation_notes.md

@@ -0,0 +1,55 @@
+- The alembic format serves as a good way to bake mesh data with lots of modifiers
+- This script will convert mesh animations to shapekeys. Digusting, but it just might work
+- https://gist.github.com/fire/495fb6a35168500df53c002695a1f5fe
+Code reproduced below for archival purposes
+
+- Enable the .mdd addon in blender to export as a mesh cache
+- It might work as a direct export to .mdd, but I did alembic first, then rexported to mdd
+- Create new object with same mesh and empty modifier stack and add a mesh cache modifier to load exported file.
+
+
+- This method is better
+- https://github.com/yanorax/Godot-VertexAnimation-Demo
+- But it would need to be extended to support a mesh cache modifier. (Should be possible given following code
+
+```
+
+# MIT LICENSE
+# Authored by iFire#6518 and alexfreyre#1663
+# This code ONLY apply to a mesh and simulations with ONLY the same vertex number
+
+import bpy
+
+#Converts a MeshCache or Cloth modifiers to ShapeKeys
+frame = bpy.context.scene.frame_start
+for frame in range(bpy.context.scene.frame_end + 1):
+    bpy.context.scene.frame_current = frame
+
+    #for alembic files converted to MDD and loaded as MeshCache
+    bpy.ops.object.modifier_apply_as_shapekey(keep_modifier=True, modifier="MeshCache")
+
+    #for cloth simulations inside blender using a Cloth modifier
+    #bpy.ops.object.modifier_apply_as_shapekey(keep_modifier=True, modifier="Cloth")
+
+# loop through shapekeys and add as keyframe per frame
+# https://blender.stackexchange.com/q/149045/87258
+frame = bpy.context.scene.frame_start
+for frame in range(bpy.context.scene.frame_end + 1):
+    bpy.context.scene.frame_current = frame
+
+    for shapekey in bpy.data.shape_keys:
+        for i, keyblock in enumerate(shapekey.key_blocks):
+            if keyblock.name != "Basis":
+                curr = i - 1
+                if curr != frame:
+                    keyblock.value = 0
+                    keyblock.keyframe_insert("value", frame=frame)
+                else:
+                    keyblock.value = 1
+                    keyblock.keyframe_insert("value", frame=frame)
+
+# bpy.ops.object.modifier_remove(modifier="MeshCache")
+# bpy.ops.object.modifier_remove(modifier="Cloth")
+```
+
+