diff options
Diffstat (limited to 'object_animrenderbake.py')
| -rw-r--r-- | object_animrenderbake.py | 201 |
1 files changed, 201 insertions, 0 deletions
diff --git a/object_animrenderbake.py b/object_animrenderbake.py new file mode 100644 index 00000000..0c354408 --- /dev/null +++ b/object_animrenderbake.py @@ -0,0 +1,201 @@ +# ##### 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 ##### + +bl_info = { + "name": "Animated Render Baker", + "author": "Janne Karhu (jahka)", + "version": (1, 0), + "blender": (2, 5, 8), + "api": 35622, + "location": "Properties > Render > Bake Panel", + "description": "Renderbakes a series of frames", + "category": "Object", + 'wiki_url': 'http://wiki.blender.org/index.php/Extensions:2.5/Py/' \ + 'Scripts/Object/Animated_Render_Baker', + 'tracker_url': 'https://projects.blender.org/tracker/index.php?'\ + 'func=detail&aid=24836'} + +import bpy +from bpy.props import * + +class OBJECT_OT_animrenderbake(bpy.types.Operator): + bl_label = "Animated Render Bake" + bl_description= "Bake animated image textures of selected objects" + bl_idname = "object.anim_bake_image" + bl_register = True + + def framefile(self, orig, frame): + ''' + Set frame number to file name image.png -> image0013.png + ''' + dot = orig.rfind(".") + return orig[:dot] + ('%04d' % frame) + orig[dot:] + + def invoke(self, context, event): + import bpy + import shutil + + scene = context.scene + + start = scene.animrenderbake_start + end = scene.animrenderbake_end + + # Check for errors before starting + if start >= end: + self.report({'ERROR'}, "Start frame must be smaller than end frame") + return {'CANCELLED'} + + selected = context.selected_objects + + # Only single object baking for now + if scene.render.use_bake_selected_to_active: + if len(selected) > 2: + self.report({'ERROR'}, "Select only two objects for animated baking") + return {'CANCELLED'} + elif len(selected) > 1: + self.report({'ERROR'}, "Select only one object for animated baking") + return {'CANCELLED'} + + if context.active_object.type != 'MESH': + self.report({'ERROR'}, "The baked object must be a mesh object") + return {'CANCELLED'} + + img = None + + #find the image that's used for rendering + for uvtex in context.active_object.data.uv_textures: + if uvtex.active_render == True: + for uvdata in uvtex.data: + if uvdata.image != None: + img = uvdata.image + break + + if img is None: + self.report({'ERROR'}, "No valid image found to bake to") + return {'CANCELLED'} + + if img.is_dirty: + self.report({'ERROR'}, "Save the image that's used for baking before use") + return {'CANCELLED'} + + # make sure we have an absolute path so that copying works for sure + absp = bpy.path.abspath(img.filepath) + + print("Animated baking for frames " + str(start) + " - " + str(end)) + + for cfra in range(start, end+1): + print("Baking frame " + str(cfra)) + + # update scene to new frame and bake to template image + scene.frame_set(cfra) + ret = bpy.ops.object.bake_image() + if 'CANCELLED' in ret: + return {'CANCELLED'} + + #currently the api doesn't allow img.save_as(), so just save the template image as usual for every frame and copy to a file with frame specific filename + img.save() + shutil.copyfile(absp, self.framefile(absp, cfra)) + + print("Saved " + self.framefile(absp, cfra)) + print("Baking done!") + + return{'FINISHED'} + +# modified copy of original bake panel draw function +def draw_animrenderbake(self, context): + layout = self.layout + + rd = context.scene.render + + row = layout.row() + row.operator("object.bake_image", icon='RENDER_STILL') + + #----------- beginning of modifications ---------------- + row.operator("object.anim_bake_image", text="Animated Bake", icon="RENDER_ANIMATION") + row = layout.row(align=True) + row.prop(context.scene, "animrenderbake_start") + row.prop(context.scene, "animrenderbake_end") + #-------------- end of modifications --------------------- + + layout.prop(rd, "bake_type") + + multires_bake = False + if rd.bake_type in ['NORMALS', 'DISPLACEMENT']: + layout.prop(rd, 'use_bake_multires') + multires_bake = rd.use_bake_multires + + if not multires_bake: + if rd.bake_type == 'NORMALS': + layout.prop(rd, "bake_normal_space") + elif rd.bake_type in {'DISPLACEMENT', 'AO'}: + layout.prop(rd, "use_bake_normalize") + + # col.prop(rd, "bake_aa_mode") + # col.prop(rd, "use_bake_antialiasing") + + layout.separator() + + split = layout.split() + + col = split.column() + col.prop(rd, "use_bake_clear") + col.prop(rd, "bake_margin") + col.prop(rd, "bake_quad_split", text="Split") + + col = split.column() + col.prop(rd, "use_bake_selected_to_active") + sub = col.column() + sub.active = rd.use_bake_selected_to_active + sub.prop(rd, "bake_distance") + sub.prop(rd, "bake_bias") + else: + if rd.bake_type == 'DISPLACEMENT': + layout.prop(rd, "use_bake_lores_mesh") + + layout.prop(rd, "use_bake_clear") + layout.prop(rd, "bake_margin") + +def register(): + bpy.utils.register_module(__name__) + + bpy.types.Scene.animrenderbake_start = IntProperty( + name="Start", + description="Start frame of the animated bake", + default=1) + + bpy.types.Scene.animrenderbake_end = IntProperty( + name="End", + description="End frame of the animated bake", + default=250) + + # replace original panel draw function with modified one + panel = bpy.types.RENDER_PT_bake + panel.old_draw = panel.draw + panel.draw = draw_animrenderbake + +def unregister(): + bpy.utils.unregister_module(__name__) + + # restore original panel draw function + bpy.types.RENDER_PT_bake.draw = bpy.types.RENDER_PT_bake.old_draw + del bpy.types.RENDER_PT_bake.old_draw + del bpy.types.Scene.animrenderbake_start + del bpy.types.Scene.animrenderbake_end + +if __name__ == "__main__": + register() |