X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=blender%2Fio_mspgl%2F__init__.py;h=79575f7df23b2c80761e24c09f499f68b7f45ee6;hb=27a3dcc73777bcc53f93eaf66e90370b061d3988;hp=b5892b1c67abde272a210d384f13ebf83fafaa63;hpb=be2d4eb0b735500a50d20ee2d9979c9f17d79e3a;p=libs%2Fgl.git diff --git a/blender/io_mspgl/__init__.py b/blender/io_mspgl/__init__.py index b5892b1c..79575f7d 100644 --- a/blender/io_mspgl/__init__.py +++ b/blender/io_mspgl/__init__.py @@ -8,16 +8,15 @@ 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", "scene", "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) +import bpy_extras +class ExportHelper(bpy_extras.io_utils.ExportHelper): def set_extension(self, ext): ext_changed = (ext!=self.filename_ext) if ext_changed: @@ -26,91 +25,43 @@ class ExportMspGLBase(ExportHelper): self.filename_ext = ext return ext_changed - def execute(self, context): - exporter = self.create_exporter() - self.prepare_exporter(exporter) - exporter.export_to_file(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): - export_all: bpy.props.BoolProperty(name="Export all selected", description="Export all selected objects (use generated filenames)", default=False) - 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() - - col = self.layout.column() - if len(context.selected_objects)>1: - col.label(text="Object selection") - col.prop(self, "export_all") - col.label(text="Triangle strips") - col.prop(self, "use_strips") - col.prop(self, "use_degen_tris") - -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" +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" - 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) - export_lods: bpy.props.BoolProperty(name="Export LoDs", description="Export all levels of detail", default=True) - use_textures: bpy.props.BoolProperty(name="Use textures", description="Use textures in the exported object", default=True) + @classmethod + def poll(cls, context): + return len(context.selected_objects)>0 - 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 + 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_object import ObjectExporter - return ObjectExporter() + 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): - super().draw(context) - - self.general_col.prop(self, "export_lods") - self.general_col.prop(self, "use_textures") - 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): +class ExportMspGLAnimation(bpy.types.Operator, ExportHelper): bl_idname = "export.mspgl_animation" bl_label = "Export Msp GL animation" bl_description = "Export one or more animations in Msp GL format" @@ -126,9 +77,14 @@ class ExportMspGLAnimation(bpy.types.Operator, ExportMspGLBase): super_result = super().check(context) return ext_changed or super_result - def create_exporter(self): + def execute(self, context): from .export_animation import AnimationExporter - return AnimationExporter() + exporter = AnimationExporter() + exporter.export_all = self.export_all + exporter.collection = self.collection + exporter.looping_threshold = self.looping_threshold + exporter.export_to_file(context, self.filepath) + return {'FINISHED'} def draw(self, context): col = self.layout.column() @@ -137,41 +93,45 @@ class ExportMspGLAnimation(bpy.types.Operator, ExportMspGLBase): col.prop(self, "collection") col.prop(self, "looping_threshold") -class ExportMspGLScene(bpy.types.Operator, ExportMspGLBase): +class ExportMspGLScene(bpy.types.Operator, ExportHelper): 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" - selected_only: bpy.props.BoolProperty(name="Selected objects only", description="Only export the selected objects") - visible_collections: bpy.props.BoolProperty(name="Visible collections 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) + selected_only: bpy.props.BoolProperty(name="Selected objects only", description="Only export the selected objects", default=False) + visible_only: bpy.props.BoolProperty(name="Visible only", description="Only export objects in visible collections", default=True) + collection: bpy.props.BoolProperty(name="As a collection", description="Export the scene and all resources as a single collection file", default=False) skip_existing: bpy.props.BoolProperty(name="Skip existing files", description="Skip resources that already exist as files", default=True) - def create_exporter(self): + def invoke(self, context, event): + self.filepath = context.scene.name+".scene" + return super().invoke(context, event) + + def check(self, context): + ext_changed = self.set_extension(".mdc" if self.collection else ".scene") + super_result = super().check(context) + return ext_changed or super_result + + def execute(self, context): from .export_scene import SceneExporter - return SceneExporter() + exporter = SceneExporter() + exporter.selected_only = self.selected_only + exporter.visible_only = self.visible_only + exporter.collection = self.collection + exporter.skip_existing = self.skip_existing + exporter.export_to_file(context, self.filepath) + return {'FINISHED'} def draw(self, context): col = self.layout.column() col.prop(self, "selected_only") - col.prop(self, "visible_collections") - col.prop(self, "resource_collection") - if self.resource_collection: + col.prop(self, "visible_only") + col.prop(self, "collection") + if self.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" @@ -197,14 +157,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: