]> git.tdb.fi Git - libs/gl.git/blobdiff - blender/io_mspgl/__init__.py
Rename visible_collections to visible_only
[libs/gl.git] / blender / io_mspgl / __init__.py
index 724d625be566ee853f7a3f44836c32f34a885e10..13375e2426795d4995e1505987210dd3c4f75476 100644 (file)
@@ -1,5 +1,6 @@
 bl_info = {
        "name": "Msp GL datafiles",
+       "blender": (2, 80, 0),
        "author": "Mikko Rasa",
        "location": "File > Export",
        "description": "Export Msp GL data",
@@ -7,20 +8,29 @@ bl_info = {
 
 if "bpy" in locals():
        import imp
-       for sub in "armature", "export_armature", "export_mesh", "export_object", "export_scene", "mesh", "outfile", "properties", "util":
+       for sub in "animation", "armature", "datafile", "export", "export_animation", "export_armature", "export_camera", "export_material", "export_mesh", "export_object", "export_scene", "export_texture", "material", "mesh", "properties", "util":
                if sub in locals():
                        imp.reload(locals()[sub])
 
+import os
 import bpy
 from bpy_extras.io_utils import ExportHelper
 
 class ExportMspGLBase(ExportHelper):
-       show_progress = bpy.props.BoolProperty(name="Show progress", description="Display progress indicator while exporting", default=True)
+       show_progress: bpy.props.BoolProperty(name="Show progress", description="Display progress indicator while exporting", default=True)
+
+       def set_extension(self, ext):
+               ext_changed = (ext!=self.filename_ext)
+               if ext_changed:
+                       if self.filepath.endswith(self.filename_ext):
+                               self.filepath = self.filepath[:-len(self.filename_ext)]
+                       self.filename_ext = ext
+               return ext_changed
 
        def execute(self, context):
                exporter = self.create_exporter()
                self.prepare_exporter(exporter)
-               exporter.export(context, self.filepath)
+               exporter.export_to_file(context, self.filepath)
                return {"FINISHED"}
 
        def create_exporter(self):
@@ -30,83 +40,80 @@ class ExportMspGLBase(ExportHelper):
                for k, v in self.as_keywords().items():
                        setattr(exporter, k, v)
 
-class ExportMspGLMeshBase(ExportMspGLBase):
-       use_strips = bpy.props.BoolProperty(name="Use strips", description="Combine faces into triangle strips", default=True)
-       use_degen_tris = bpy.props.BoolProperty(name="Use degen tris", description="Concatenate triangle strips with degenerate triangles", default=False)
-
-       def draw(self, context):
-               self.general_col = self.layout.column()
+class ExportMspGLData(bpy.types.Operator):
+       bl_idname = "export.mspgl_data"
+       bl_label = "Export Msp GL data"
+       bl_description = "Export object data in Msp GL format"
 
-               col = self.layout.column()
-               col.label("Triangle strips")
-               col.prop(self, "use_strips")
-               col.prop(self, "use_degen_tris")
+       filepath: bpy.props.StringProperty(name="File path", description="File path for exporting the data", subtype='FILE_PATH')
+       collection: bpy.props.BoolProperty(name="As a collection", description="Export all data as a single collection file", default=False)
+       shared_resources: bpy.props.BoolProperty(name="Shared resources", description="Use global names for resource files to enable sharing", default=True)
 
-class ExportMspGLMesh(bpy.types.Operator, ExportMspGLMeshBase):
-       bl_idname = "export_mesh.mspgl_mesh"
-       bl_label = "Export Msp GL mesh"
+       @classmethod
+       def poll(cls, context):
+               return len(context.selected_objects)>0
 
-       filename_ext = ".mesh"
+       def invoke(self, context, event):
+               blend_filepath = context.blend_data.filepath
+               if blend_filepath:
+                       self.filepath = os.path.splitext(blend_filepath)[0]+".mdc"
+               else:
+                       self.filepath = "data.mdc"
+               context.window_manager.fileselect_add(self)
+               return {'RUNNING_MODAL'}
 
-       def create_exporter(self):
-               from .export_mesh import MeshExporter
-               return MeshExporter()
+       def execute(self, context):
+               from .export import DataExporter
+               exporter = DataExporter()
+               exporter.collection = self.collection
+               exporter.shared_resources = self.shared_resources
+               exporter.export_to_file(context, self.filepath)
+               return {'FINISHED'}
 
-class ExportMspGLObject(bpy.types.Operator, ExportMspGLMeshBase):
-       bl_idname = "export_mesh.mspgl_object"
-       bl_label = "Export Msp GL object"
+       def draw(self, context):
+               col = self.layout.column()
+               col.label(text="Files")
+               col.prop(self, "collection")
+               col.prop(self, "shared_resources")
 
-       filename_ext = ".object"
+class ExportMspGLAnimation(bpy.types.Operator, ExportMspGLBase):
+       bl_idname = "export.mspgl_animation"
+       bl_label = "Export Msp GL animation"
+       bl_description = "Export one or more animations in Msp GL format"
 
-       textures = bpy.props.EnumProperty(name="Textures", description="Export textures", default="REF",
-               items=(("NONE", "None", "Ignore textures"),
-                       ("REF", "Referenced", "Reference external data"),
-                       ("INLINE", "Inline", "Embed textures in the object")))
+       filename_ext = ".anim"
 
-       separate_mesh = bpy.props.BoolProperty(name="Separate mesh", description="Write mesh data into a separate file", default=False)
-       separate_tech = bpy.props.BoolProperty(name="Separate technique", description="Write technique data into a separate file", default=False)
-       shared_resources = bpy.props.BoolProperty(name="Shared resources", description="Use global names for resource files to enable sharing", default=True)
+       export_all: bpy.props.BoolProperty(name="Export all animations", description="Export all animations present on the selected objects' NLA tracks")
+       collection: bpy.props.BoolProperty(name="As a collection", description="Export the animations as a single collection file", default=True)
+       looping_threshold: bpy.props.FloatProperty(name="Looping threshold", description="Tolerance for curve beginning and end values for looping", min=0.0, soft_max=1.0, precision=4, default=0.001)
 
-       export_lods = bpy.props.BoolProperty(name="Export LoDs", description="Export all levels of detail", default=True)
+       def check(self, context):
+               ext_changed = self.set_extension(".mdc" if self.export_all and self.collection else ".anim")
+               super_result = super().check(context)
+               return ext_changed or super_result
 
        def create_exporter(self):
-               from .export_object import ObjectExporter
-               return ObjectExporter()
+               from .export_animation import AnimationExporter
+               return AnimationExporter()
 
        def draw(self, context):
-               super().draw(context)
-
-               col = self.general_col
-               col.prop(self, "export_lods")
-               col.prop(self, "textures")
-               col.separator()
-
-               self.layout.separator()
-
                col = self.layout.column()
-               col.label("Files")
-               col.prop(self, "separate_mesh")
-               col.prop(self, "separate_tech")
-               if self.separate_mesh or self.separate_tech:
-                       col.prop(self, "shared_resources")
-
-class ExportMspGLArmature(bpy.types.Operator, ExportMspGLBase):
-       bl_idname = "export.mspgl_armature"
-       bl_label = "Export Msp GL armature"
-
-       filename_ext = ".arma"
-
-       def create_exporter(self):
-               from .export_armature import ArmatureExporter
-               return ArmatureExporter()
+               col.prop(self, "export_all")
+               if self.export_all:
+                       col.prop(self, "collection")
+               col.prop(self, "looping_threshold")
 
 class ExportMspGLScene(bpy.types.Operator, ExportMspGLBase):
        bl_idname = "export_scene.mspgl_scene"
        bl_label = "Export Msp GL scene"
+       bl_description = "Export the active scene in Msp GL format"
 
        filename_ext = ".scene"
 
-       resource_collection = bpy.props.BoolProperty(name="Resource collection", description="Put resources to a single collection file", default=True)
+       selected_only: bpy.props.BoolProperty(name="Selected objects only", description="Only export the selected objects")
+       visible_only: bpy.props.BoolProperty(name="Visible only", description="Only export objects in visible collections", default=True)
+       resource_collection: bpy.props.BoolProperty(name="Resource collection", description="Put resources to a single collection file", default=True)
+       skip_existing: bpy.props.BoolProperty(name="Skip existing files", description="Skip resources that already exist as files", default=True)
 
        def create_exporter(self):
                from .export_scene import SceneExporter
@@ -114,28 +121,57 @@ class ExportMspGLScene(bpy.types.Operator, ExportMspGLBase):
 
        def draw(self, context):
                col = self.layout.column()
+               col.prop(self, "selected_only")
+               col.prop(self, "visible_only")
                col.prop(self, "resource_collection")
+               if self.resource_collection:
+                       col.prop(self, "skip_existing")
+
+class AddUniform(bpy.types.Operator):
+       bl_idname = "material.add_uniform"
+       bl_label = "Add Uniform"
+       bl_description = "Add a new uniform value to the material"
+
+       def execute(self, context):
+               mat = context.active_object.active_material
+               mat.uniforms.add()
+               mat.active_uniform_index = len(mat.uniforms)-1
+
+               return {"FINISHED"}
+
+class RemoveUniform(bpy.types.Operator):
+       bl_idname = "material.remove_uniform"
+       bl_label = "Remove Uniform"
+       bl_description = "Remove the selected uniform from the material"
+
+       def execute(self, context):
+               mat = context.active_object.active_material
+               mat.uniforms.remove(mat.active_uniform_index)
+               mat.active_uniform_index = min(mat.active_uniform_index, len(mat.uniforms)-1)
+
+               return {"FINISHED"}
 
 def menu_func_export(self, context):
-       self.layout.operator(ExportMspGLMesh.bl_idname, text="Msp GL mesh")
-       self.layout.operator(ExportMspGLObject.bl_idname, text="Msp GL object")
-       self.layout.operator(ExportMspGLArmature.bl_idname, text="Msp GL armature")
+       self.layout.operator(ExportMspGLData.bl_idname, text="Msp GL data")
+       self.layout.operator(ExportMspGLAnimation.bl_idname, text="Msp GL animation")
        self.layout.operator(ExportMspGLScene.bl_idname, text="Msp GL scene")
 
-from .properties import MspGLMeshProperties, MspGLObjectProperties
+classes = [ExportMspGLData, ExportMspGLAnimation, ExportMspGLScene, AddUniform, RemoveUniform]
 
 def register():
-       bpy.utils.register_module(__name__)
+       for c in classes:
+               bpy.utils.register_class(c)
 
-       bpy.types.INFO_MT_file_export.append(menu_func_export)
+       bpy.types.TOPBAR_MT_file_export.append(menu_func_export)
 
        from .properties import register_properties
        register_properties()
 
 def unregister():
-       bpy.utils.unregister_module(__name__)
+       for c in classes:
+               bpy.utils.unregister_class(c)
 
-       bpy.types.INFO_MT_file_export.remove(menu_func_export)
+       bpy.types.TOPBAR_MT_file_export.remove(menu_func_export)
 
-if __name__=="__main__":
-       register()
+       from .properties import unregister_properties
+       unregister_properties()