]> git.tdb.fi Git - libs/gl.git/blobdiff - blender/io_mspgl/__init__.py
Minor tweaks to the Blender exporter
[libs/gl.git] / blender / io_mspgl / __init__.py
index 2ac3bd19f48d6f58731dbccb958218d267a66eca..1e085baf9cf4559d9f85885c78174ef5310b2627 100644 (file)
@@ -7,14 +7,28 @@ bl_info = {
 
 if "bpy" in locals():
        import imp
-       for sub in "export_mesh", "mesh", "outfile", "util":
+       for sub in "armature", "export_armature", "export_mesh", "export_object", "mesh", "outfile", "properties", "util":
                if sub in locals():
                        imp.reload(locals()[sub])
 
 import bpy
 from bpy_extras.io_utils import ExportHelper
 
-class ExportMspGLMeshBase(ExportHelper):
+class ExportMspGLBase(ExportHelper):
+       def execute(self, context):
+               exporter = self.create_exporter()
+               self.prepare_exporter(exporter)
+               exporter.export(context, self.filepath)
+               return {"FINISHED"}
+
+       def create_exporter(self):
+               raise Exception("create_exporter must be overridden")
+
+       def prepare_exporter(self, exporter):
+               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)
        max_strip_len = bpy.props.IntProperty(name="Max strip length", description="Maximum length for a triangle strip", default=1024, min=4, max=16384)
@@ -32,39 +46,39 @@ class ExportMspGLMeshBase(ExportHelper):
                items=(("NONE", "None", "No smoothing"),
                        ("BLENDER", "Blender", "Use Blender's vertex normals"),
                        ("MSPGL", "MspGL", "Compute vertex normals internally")))
-
-       def execute(self, context):
-               from .export_mesh import MeshExporter
-               exporter = MeshExporter()
-               self.prepare_exporter(exporter)
-               exporter.export(context, self.filepath)
-               return {"FINISHED"}
-
-       def prepare_exporter(self, exporter):
-               for k, v in self.as_keywords().items():
-                       setattr(exporter, k, v)
+       export_groups = bpy.props.BoolProperty(name="Vertex groups", description="Export vertex groups and weights", default=False)
 
        def draw(self, context):
                col = self.layout.column()
                col.prop(self, "export_lines")
                col.prop(self, "compound")
                col.prop(self, "smoothing")
+               col.prop(self, "export_groups")
+               self.general_col = col
+
                self.layout.separator()
+
                col = self.layout.column()
                col.label("Triangle strips")
                col.prop(self, "use_strips")
                col.prop(self, "use_degen_tris")
                col.prop(self, "max_strip_len")
+
                self.layout.separator()
+
+               col = self.layout.column()
+               col.label("Texturing")
+               col.prop(self, "export_uv")
+               col.prop(self, "tbn_vecs")
+               col.prop(self, "tbn_uvtex")
+               self.texturing_col = col
+
+               self.layout.separator()
+
                col = self.layout.column()
                col.label("Vertex cache")
                col.prop(self, "optimize_cache")
                col.prop(self, "cache_size")
-               self.layout.separator()
-               col = self.layout.column()
-               col.label("TBN vectors")
-               col.prop(self, "tbn_vecs")
-               col.prop(self, "tbn_uvtex")
 
 class ExportMspGLMesh(bpy.types.Operator, ExportMspGLMeshBase):
        bl_idname = "export_mesh.mspgl_mesh"
@@ -72,39 +86,78 @@ class ExportMspGLMesh(bpy.types.Operator, ExportMspGLMeshBase):
 
        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"
 
        filename_ext = ".object"
 
+       external_tech = bpy.props.BoolProperty(name="External technique", description="Use an external technique specified in the object", default=True)
+
        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")))
        material_tex = bpy.props.BoolProperty(name="Material texture", description="Generate a texture based on material colors", default=False)
+       srgb_colors = bpy.props.BoolProperty(name="sRGB colors", description="Export material colors as sRGB instead of linear", default=True)
 
-       def prepare_exporter(self, exporter):
-               super().prepare_exporter(exporter)
-               exporter.object = True
+       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_tech = bpy.props.BoolProperty(name="Shared technique", description="Use material name for technique file to enable sharing", default=True)
+
+       def create_exporter(self):
+               from .export_object import ObjectExporter
+               return ObjectExporter()
 
        def draw(self, context):
                super().draw(context)
-               self.layout.separator()
-               col = self.layout.column()
-               col.label("Texturing")
+
+               col = self.general_col
+               col.prop(self, "external_tech")
+
+               col = self.texturing_col
                col.prop(self, "textures")
                col.prop(self, "material_tex")
+               col.prop(self, "srgb_colors")
+
+               self.layout.separator();
+
+               col = self.layout.column()
+               col.label("Files")
+               col.prop(self, "separate_mesh")
+               col.prop(self, "separate_tech")
+               if self.separate_tech:
+                       col.prop(self, "shared_tech")
+
+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()
 
 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")
+
+from .properties import MspGLProperties
 
 def register():
        bpy.utils.register_module(__name__)
 
        bpy.types.INFO_MT_file_export.append(menu_func_export)
 
+       from .properties import register_properties
+       register_properties();
+
 def unregister():
        bpy.utils.unregister_module(__name__)