]> git.tdb.fi Git - libs/gl.git/blobdiff - blender/io_mspgl/__init__.py
Redesign object exporting
[libs/gl.git] / blender / io_mspgl / __init__.py
index 473750a53441ec3abb591d374e25b80c16f830e3..8de91d25a7ddb91b795c9c53daed4387df9a41db 100644 (file)
@@ -8,10 +8,11 @@ bl_info = {
 
 if "bpy" in locals():
        import imp
-       for sub in "animation", "armature", "datafile", "export_animation", "export_armature", "export_camera", "export_material", "export_mesh", "export_object", "export_scene", "export_texture", "material", "mesh", "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
 
@@ -39,66 +40,42 @@ class ExportMspGLBase(ExportHelper):
                for k, v in self.as_keywords().items():
                        setattr(exporter, k, v)
 
-class ExportMspGLMeshBase(ExportMspGLBase):
-       export_all: bpy.props.BoolProperty(name="Export all selected", description="Export all selected objects (use generated filenames)", default=False)
+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"
 
-       def draw(self, context):
-               self.general_col = self.layout.column()
-
-               col = self.layout.column()
-               if len(context.selected_objects)>1:
-                       col.label(text="Object selection")
-                       col.prop(self, "export_all")
-
-class ExportMspGLMesh(bpy.types.Operator, ExportMspGLMeshBase):
-       bl_idname = "export_mesh.mspgl_mesh"
-       bl_label = "Export Msp GL mesh"
-       bl_description = "Export one or more meshes in Msp GL format"
-
-       filename_ext = ".mesh"
-
-       def create_exporter(self):
-               from .export_mesh import MeshExporter
-               return MeshExporter()
-
-class ExportMspGLObject(bpy.types.Operator, ExportMspGLMeshBase):
-       bl_idname = "export_mesh.mspgl_object"
-       bl_label = "Export Msp GL object"
-       bl_description = "Export one or more objects in Msp GL format"
-
-       filename_ext = ".object"
-
-       collection: bpy.props.BoolProperty(name="As a collection", description="Write all data into a single collection file", default=False)
+       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)
 
-       def check(self, context):
-               ext_changed = self.set_extension(".mdc" if self.collection else ".object")
-               super_result = super().check(context)
-               return ext_changed or super_result
+       @classmethod
+       def poll(cls, context):
+               return len(context.selected_objects)>0
 
-       def create_exporter(self):
-               from .export_object import ObjectExporter
-               return ObjectExporter()
+       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 draw(self, context):
-               super().draw(context)
+       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'}
 
+       def draw(self, context):
                col = self.layout.column()
                col.label(text="Files")
                col.prop(self, "collection")
                col.prop(self, "shared_resources")
 
-class ExportMspGLArmature(bpy.types.Operator, ExportMspGLBase):
-       bl_idname = "export.mspgl_armature"
-       bl_label = "Export Msp GL armature"
-       bl_description = "Export an armature in Msp GL format"
-
-       filename_ext = ".arma"
-
-       def create_exporter(self):
-               from .export_armature import ArmatureExporter
-               return ArmatureExporter()
-
 class ExportMspGLAnimation(bpy.types.Operator, ExportMspGLBase):
        bl_idname = "export.mspgl_animation"
        bl_label = "Export Msp GL animation"
@@ -150,17 +127,6 @@ class ExportMspGLScene(bpy.types.Operator, ExportMspGLBase):
                if self.resource_collection:
                        col.prop(self, "skip_existing")
 
-class ExportMspGLCamera(bpy.types.Operator, ExportMspGLBase):
-       bl_idname = "export.mspgl_camera"
-       bl_label = "Export Msp GL camera"
-       bl_description = "Export a camera in Msp GL format"
-
-       filename_ext = ".camera"
-
-       def create_exporter(self):
-               from .export_camera import CameraExporter
-               return CameraExporter()
-
 class AddUniform(bpy.types.Operator):
        bl_idname = "material.add_uniform"
        bl_label = "Add Uniform"
@@ -186,14 +152,11 @@ class RemoveUniform(bpy.types.Operator):
                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")
-       self.layout.operator(ExportMspGLCamera.bl_idname, text="Msp GL camera")
 
-classes = [ExportMspGLMesh, ExportMspGLObject, ExportMspGLArmature, ExportMspGLAnimation, ExportMspGLScene, ExportMspGLCamera, AddUniform, RemoveUniform]
+classes = [ExportMspGLData, ExportMspGLAnimation, ExportMspGLScene, AddUniform, RemoveUniform]
 
 def register():
        for c in classes: