support_ops.py

# Copyright (C) 2019 h0bB1T
#
# 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 3 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

import bpy, os, re

from bpy.types              import Operator

from . properties           import Properties, StringProperty
from . preview_helper       import PreviewHelper
from . preferences          import PreferencesPanel
from . utils                import export_file, CategoriesCache, ASSET_TYPE_OBJECT, ASSET_TYPE_MATERIAL, PREVIEW_EXT

class RefreshObjectPreviews(Operator):
    bl_idname = "asset_wizard.refresh_object_previews_op"
    bl_label = "Refresh"
    bl_description = "Refresh previews for this category (if e.g. externally modified)"

    def execute(self, context):
        PreviewHelper.setData(
            ASSET_TYPE_OBJECT, 
            (ASSET_TYPE_OBJECT, Properties.get().iobj_categories),
            True
        )
        CategoriesCache.update_cache(ASSET_TYPE_OBJECT)
        return {'FINISHED'}


class ReRenderObjectPreview(Operator):
    bl_idname = "asset_wizard.rerender_object_preview_op"
    bl_label = "ReRender"
    bl_description = "ReRender preview for current selection"

    def execute(self, context):
        Properties.get_render_previews().add_job(
            ASSET_TYPE_OBJECT, 
            Properties.get().iobj_previews
        )
        return {'FINISHED'}        


class RefreshMaterialPreviews(Operator):
    bl_idname = "asset_wizard.refresh_material_previews_op"
    bl_label = "Refresh"
    bl_description = "Refresh previews for this category (if e.g. externally modified)"

    def execute(self, context):
        PreviewHelper.setData(
            ASSET_TYPE_MATERIAL, 
            (ASSET_TYPE_MATERIAL, Properties.get().imat_categories),
            True
        )
        CategoriesCache.update_cache(ASSET_TYPE_MATERIAL)
        return {'FINISHED'}        


class ReRenderMaterialPreview(Operator):
    bl_idname = "asset_wizard.rerender_material_preview_op"
    bl_label = "ReRender"
    bl_description = "ReRender preview for current selection"

    def execute(self, context):
        Properties.get_render_previews().add_job(
            ASSET_TYPE_MATERIAL, 
            Properties.get().imat_previews
        )
        return {'FINISHED'}        


class RemoveAsset(Operator):
    bl_idname = "asset_wizard.remove_asset_op"
    bl_label = "Remove Asset?"
    bl_description = "Remove asset from library"


    asset_type: StringProperty()
    asset: StringProperty()


    def execute(self, context):
        asset = os.path.join(PreferencesPanel.get().root, self.asset)
        preview = os.path.splitext(asset)[0] + PREVIEW_EXT

        failed = False
        try:
            if os.path.exists(asset):
                os.remove(asset)
        except Exception as ex:
            failed = True

        failed = False
        try:
            if os.path.exists(preview):
                os.remove(preview)
        except Exception as ex:
            failed = True

        if self.asset_type == ASSET_TYPE_OBJECT:
            bpy.ops.asset_wizard.refresh_object_previews_op()
        elif self.asset_type == ASSET_TYPE_MATERIAL:
            bpy.ops.asset_wizard.refresh_material_previews_op()

        if failed:
            self.report({'ERROR'}, "Removing the asset failed")
        else:
            self.report({'INFO'}, "Asset removed")

        return {'FINISHED'}


    def invoke(self, context, event):
        return context.window_manager.invoke_confirm(self, event)

class AutoNumberExportName(Operator):
    bl_idname = "asset_wizard.auto_number_export_name_op"
    bl_label = "Next number"
    bl_description = "Increase postfix number of export asset name"


    asset_type: StringProperty()    


    def execute(self, context):

        if self.asset_type == ASSET_TYPE_OBJECT:
            props = Properties.get()
            name = props.eobj_asset_name

            m = re.search(r'\d+$', name)
            if m is not None:
                num = m.group()
                base = name[:-len(num)]
                name = base + str(int(num) + 1)
            else:
                name = name + "1"

            props.eobj_asset_name = name

        return {'FINISHED'}