Skip to content
Snippets Groups Projects
operator.py 7.05 KiB
Newer Older
  • Learn to ignore specific revisions
  • # ##### 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>
    
    import bpy
    from . import presets
    
    # These operators are only defined because it seems impossible to directly edit properties from
    # UI code…
    
    
    # A sorting func for collections (working in-place).
    # XXX Not optimized at all…
    # XXX If some items in the collection do not have the sortkey property, they are just ignored…
    def collection_property_sort(collection, sortkey, start_idx=0):
        while start_idx + 1 < len(collection):
            while not hasattr(collection[start_idx], sortkey):
                start_idx += 1
                if start_idx + 1 >= len(collection):
                    return collection
            min_idx = start_idx
            min_prop = collection[start_idx]
            for i, prop in enumerate(collection[start_idx + 1:]):
                if not hasattr(prop, sortkey):
                    continue
                if getattr(prop, sortkey) < getattr(min_prop, sortkey):
                    min_prop = prop
                    min_idx = i + start_idx + 1
            collection.move(min_idx, start_idx)
            start_idx += 1
        return collection
    
    
    class RenderCopySettingsPrepare(bpy.types.Operator):
        '''
        Prepare internal data for render_copy_settings (gathering all existing render settings,
        and scenes)
        '''
        bl_idname = "scene.render_copy_settings_prepare"
        bl_label = "Render: Copy Settings Prepare"
        bl_option = {'REGISTER'}
    
        @classmethod
        def poll(cls, context):
            return context.scene != None
    
        def execute(self, context):
            cp_sett = context.scene.render_copy_settings
    
            # Get all available render settings, and update accordingly affected_settings…
            props = {}
            for prop in context.scene.render.bl_rna.properties:
                if prop.identifier in {'rna_type'}:
                    continue
                if prop.is_readonly:
                    continue
                props[prop.identifier] = prop.name
            corr = 0
            for i, sett in enumerate(cp_sett.affected_settings):
                if sett.strid not in props:
                    cp_sett.affected_settings.remove(i-corr)
                    corr += 1
                else:
                    del props[sett.strid]
            for strid, name in props.items():
                sett = cp_sett.affected_settings.add()
                sett.name = "{} [{}]".format(name, strid)
                sett.strid = strid
            collection_property_sort(cp_sett.affected_settings, "name")
    
            # Get all available scenes, and update accordingly allowed_scenes…
            regex = None
            if cp_sett.filter_scene:
                try:
                    import re
                    try:
                        regex = re.compile(cp_sett.filter_scene)
                    except Exception as e:
                        self.report('ERROR_INVALID_INPUT', "The filter-scene regex " \
                                    "did not compile:\n    (%s)." % str(e))
                        return {'CANCELLED'}
                except:
                    regex = None
                    self.report('WARNING', "Unable to import the re module. Regex " \
                                "scene filtering will be disabled!")
            scenes = set()
            for scene in bpy.data.scenes:
                if scene == bpy.context.scene:  # Exclude current scene!
                    continue
                if regex:  # If a valid filtering regex, only keep scenes matching it.
                    if regex.match(scene.name):
                        scenes.add(scene.name)
                else:
                    scenes.add(scene.name)
            for i, scene in enumerate(cp_sett.allowed_scenes):
                if scene.name not in scenes:
                    cp_sett.allowed_scenes.remove(i)
                else:
                    scenes.remove(scene.name)
            for scene in scenes:
                sett = cp_sett.allowed_scenes.add()
                sett.name = scene
            collection_property_sort(cp_sett.allowed_scenes, "name")
    
            return {'FINISHED'}
    
    
    from bpy.props import EnumProperty
    
    
    class RenderCopySettingsPreset(bpy.types.Operator):
        '''Apply some presets of render settings to copy to other scenes'''
        bl_idname = "scene.render_copy_settings_preset"
        bl_label = "Render: Copy Settings Preset"
        bl_description = "Apply or clear this preset of render settings"
        # Enable undo…
        bl_option = {'REGISTER', 'UNDO'}
    
        presets = EnumProperty(items=(p.rna_enum for p in presets.presets),
                               default=set(),
                               options={"ENUM_FLAG"})
    
        @staticmethod
        def process_elements(settings, elts):
            setts = []
            val = True
            for sett in settings:
                if sett.strid in elts:
                    setts.append(sett)
                    val = val and sett.copy
            for e in setts:
                e.copy = not val
    
        @classmethod
        def poll(cls, context):
            return context.scene != None
    
        def execute(self, context):
            cp_sett = context.scene.render_copy_settings
            for p in presets.presets:
                if p.rna_enum[0] in self.presets:
                    self.process_elements(cp_sett.affected_settings, p.elements)
            return {'FINISHED'}
    
    
    # Real interesting stuff…
    
    def do_copy(context, affected_settings, allowed_scenes):
        # Stores render settings from current scene.
        p = {sett: getattr(context.scene.render, sett) for sett in affected_settings}
        # put it in all other (valid) scenes’ render settings!
        for scene in bpy.data.scenes:
            # If scene not in allowed scenes, skip.
            if scene.name not in allowed_scenes:
                continue
            # Propagate all affected settings.
            for sett, val in p.items():
                setattr(scene.render, sett, val)
    
    
    class RenderCopySettings(bpy.types.Operator):
        '''Copy render settings from current scene to others'''
        bl_idname = "scene.render_copy_settings"
        bl_label = "Render: Copy Settings"
        # Enable undo…
        bl_option = {'REGISTER', 'UNDO'}
    
        @classmethod
        def poll(cls, context):
            return context.scene != None
    
        def execute(self, context):
            regex = None
            cp_sett = context.scene.render_copy_settings
            affected_settings = set([sett.strid for sett in cp_sett.affected_settings if sett.copy])
            allowed_scenes = set([sce.name for sce in cp_sett.allowed_scenes if sce.allowed])
            do_copy(context, affected_settings=affected_settings, allowed_scenes=allowed_scenes)
            return {'FINISHED'}
    
    
    if __name__ == "__main__":
        bpy.ops.scene.render_copy_settings()