korman/korman/operators/op_world.py

#    This file is part of Korman.
#
#    Korman 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.
#
#    Korman 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 Korman.  If not, see <http://www.gnu.org/licenses/>.

import bpy
from bpy.props import *
from pathlib import Path

class AgeOperator:
    @classmethod
    def poll(cls, context):
        return context.scene.render.engine == "PLASMA_GAME"


class GameAddOperator(AgeOperator, bpy.types.Operator):
    bl_idname = "world.plasma_game_add"
    bl_label = "Add Plasma Game"

    filepath = StringProperty(subtype="DIR_PATH")
    directory = BoolProperty(default=True, options={"HIDDEN"})
    game_index = IntProperty(default=-1, options={"HIDDEN"})

    def execute(self, context):
        w = context.world
        if w:
            # First, verify this is a valid Uru directory...
            path = Path(self.filepath)

            # Blendsucks likes to tack filenames onto our doggone directories...
            if not path.is_dir():
                path = path.parent
            if not ((path / "UruExplorer.exe").is_file() or (path / "plClient.exe").is_file()):
                self.report({"ERROR"}, "The selected directory is not a copy of URU.")
                return {"CANCELLED"}

            # New game?
            games = w.plasma_games
            new_game = self.game_index == -1
            if new_game:
                games.active_game_index = len(games.games)
                game = games.games.add()
            else:
                game = games.games[self.game_index]

            # Setup game...
            game.path = str(path)
            if (path / "cypython22.dll").is_file():
                game.version = "pvPots"
            else:
                game.version = "pvMoul"
            game.name = path.name

            return {"FINISHED"}
        else:
            return {"CANCELLED"}


    def invoke(self, context, event):
        context.window_manager.fileselect_add(self)
        return {"RUNNING_MODAL"}


class GameRemoveOperator(AgeOperator, bpy.types.Operator):
    bl_idname = "world.plasma_game_remove"
    bl_label = "Remove Plasma Game"

    def execute(self, context):
        w = context.world
        if w:
            games = w.plasma_games
            if games.active_game_index >= len(games.games):
                return {"CANCELLED"}
            games.games.remove(games.active_game_index)
            return {"FINISHED"}
        else:
            return {"CANCELLED"}


class PageAddOperator(AgeOperator, bpy.types.Operator):
    bl_idname = "world.plasma_page_add"
    bl_label = "Add Page"
    bl_description = "Adds a new Plasma Registry Page"

    def execute(self, context):
        w = context.world
        if w:
            age = w.plasma_age
            page = age.pages.add()

            # Find the first non-zero ID and suggest that.
            suffixes = {p.seq_suffix for p in age.pages}
            if suffixes:
                test = set(range(min(suffixes), max(suffixes)))
                missing = test - suffixes
                try:
                    suffix = missing.pop()
                except KeyError:
                    suffix = max(suffixes) + 1
                page.make_default_name(suffix)
            else:
                # Page 0 is a magic "catch-all" page. The user *may* define it
                # if he wants. If he doesn't, we'll defer it until export time
                page.make_default_name(1)

            # Finally, select the new page
            age.active_page_index = len(age.pages) - 1
            return {"FINISHED"}
        else:
            return {"CANCELLED"}


class PageRemoveOperator(AgeOperator, bpy.types.Operator):
    bl_idname = "world.plasma_page_remove"
    bl_label = "Remove Page"
    bl_description = "Removes the selected Plasma Registry Page"

    def execute(self, context):
        w = context.world
        if w:
            age = w.plasma_age
            if age.active_page_index >= len(age.pages):
                return {"CANCELLED"}
            page = age.pages[age.active_page_index]

            # Need to reassign objects in this page to the default page
            defpg = ""
            for i in age.pages:
                if i.seq_suffix == 0:
                    defpg = i.name
                    break
            for o in bpy.data.objects:
                if o.plasma_object.page == page.name:
                    o.plasma_object.page = defpg
            age.pages.remove(age.active_page_index)
            return {"FINISHED"}
        else:
            return {"CANCELLED"}