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
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"
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"
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:
--- /dev/null
+import os
+
+class DataExporter:
+ def __init__(self):
+ self.show_progress = True
+ self.collection = False
+ self.shared_resources = True
+
+ def export_to_file(self, context, out_fn):
+ from .util import Progress
+ progress = Progress(self.show_progress and context)
+
+ objects = context.selected_objects
+
+ resources = {}
+ material_atlases = {}
+
+ dummy_res = self.export_resources(context, objects, resources, material_atlases, progress)
+
+ path, base = os.path.split(out_fn)
+ base, ext = os.path.splitext(base)
+
+ refs = dummy_res.collect_references()
+ if not self.shared_resources:
+ numbers = {}
+ for r in refs:
+ res_ext = os.path.splitext(r.name)[1]
+ n = numbers.get(res_ext, 0)
+ if n>0:
+ r.name = "{}_{}{}".format(base, n, res_ext)
+ else:
+ r.name = base+res_ext
+ numbers[res_ext] = n+1
+
+ if self.collection:
+ dummy_res.write_collection(out_fn, exclude_self=True)
+ else:
+ for r in refs:
+ r.write_to_file(os.path.join(path, r.name))
+
+ def export_resources(self, context, objects, resources, material_atlases, progress):
+ if material_atlases is None:
+ material_atlases = {}
+
+ object_exporter = None
+ camera_exporter = None
+ armature_exporter = None
+
+ from .datafile import Resource
+ dummy_res = Resource("dummy", "dummy")
+
+ for i, obj in enumerate(objects):
+ progress.push_task_slice(obj.name, i, len(objects))
+ res = None
+ if obj.type=='MESH':
+ if not object_exporter:
+ from .export_object import ObjectExporter
+ object_exporter = ObjectExporter()
+ object_exporter.export_object_resources(context, obj, resources, material_atlases, progress)
+ res = object_exporter.export_object(obj, resources, progress)
+ elif obj.type=='CAMERA':
+ if not camera_exporter:
+ from .export_camera import CameraExporter
+ camera_exporter = CameraExporter()
+ res = camera_exporter.export_camera(obj)
+ elif obj.type=='ARMATURE':
+ if not armature_exporter:
+ from .export_armature import ArmatureExporter
+ armature_exporter = ArmatureExporter()
+ res = armature_exporter.export_armature(context, obj)
+
+ if res:
+ resources[res.name] = res
+ dummy_res.create_reference_statement("ref", res)
+
+ progress.pop_task()
+
+ return dummy_res
class ArmatureExporter:
- def export_to_file(self, context, out_fn):
- obj = context.active_object
-
- statements = self.export_armature(context, obj)
-
- with open(out_fn, "w") as out_file:
- for s in statements:
- s.write_to_file(out_file)
-
def export_armature(self, obj):
if obj.type!="ARMATURE":
- raise Exception("Object is not an armature")
+ raise ValueError("Object is not an armature")
from .armature import Armature
armature = Armature(obj.data)
import mathutils
class CameraExporter:
- def export_to_file(self, context, out_fn):
- obj = context.active_object
-
- resource = self.export_camera(obj)
-
- resource.write_to_file(out_fn)
-
def export_camera(self, obj):
if obj.type!='CAMERA':
- raise Exception("Object is not a camera")
+ raise ValueError("Object is not a camera")
from .datafile import Resource, Statement
resource = Resource(obj.name+".camera", "camera")
import mathutils
class MeshExporter:
- def __init__(self):
- self.show_progress = True
- self.export_all = False
-
- def export_to_file(self, context, out_fn):
- if self.export_all:
- objs = [o for o in context.selected_objects if o.type=="MESH"]
- else:
- objs = [context.active_object]
-
- from .util import Progress
-
- path, base = os.path.split(out_fn)
- base, ext = os.path.splitext(base)
-
- progress = Progress(self.show_progress and context)
- for i, obj in enumerate(objs):
- if self.export_all:
- out_fn = os.path.join(path, obj.data.name+ext)
-
- progress.push_task_slice(obj.data.name, i, len(objs))
- resource = self.export_mesh(context, obj, progress)
-
- resource.write_to_file(out_fn)
- progress.pop_task()
-
def export_mesh(self, context, mesh_or_obj, progress):
from .mesh import Mesh, create_mesh_from_object
import mathutils
class ObjectExporter:
- def __init__(self):
- self.show_progress = True
- self.export_all = False
- self.collection = False
- self.shared_resources = True
-
def compute_bounding_sphere(self, obj):
p1 = max(((v.co, v.co.length) for v in obj.data.vertices), key=lambda x:x[1])[0]
p2 = max(((v.co, (v.co-p1).length) for v in obj.data.vertices), key=lambda x:x[1])[0]
material_atlas_export = MaterialAtlasExporter()
return material_atlas_export
- def export_to_file(self, context, out_fn):
- if self.export_all:
- objs = [o for o in context.selected_objects if o.type=="MESH"]
- else:
- objs = [context.active_object]
-
- from .util import Progress
- progress = Progress(self.show_progress and context)
-
- path, base = os.path.split(out_fn)
- base, ext = os.path.splitext(base)
-
- resources = {}
- for i, obj in enumerate(objs):
- if self.export_all:
- out_fn = os.path.join(path, obj.name+ext)
-
- progress.push_task_slice(obj.name, i, len(objs))
- self.export_object_resources(context, obj, resources, None, progress)
-
- obj_res = self.export_object(context, obj, resources, progress)
- refs = obj_res.collect_references()
- if not self.shared_resources:
- numbers = {}
- for r in refs:
- res_ext = os.path.splitext(r.name)[1]
- n = numbers.get(res_ext, 0)
- if n>0:
- r.name = "{}_{}{}".format(base, n, res_ext)
- else:
- r.name = base+res_ext
- numbers[res_ext] = n+1
-
- if self.collection:
- obj_res.write_collection(out_fn)
- else:
- for r in refs:
- r.write_to_file(os.path.join(path, r.name))
- obj_res.write_to_file(out_fn)
-
- progress.pop_task()
-
def export_object_resources(self, context, obj, resources, material_atlases, progress):
if material_atlases is None:
material_atlases = {}
progress.pop_task()
- def export_object(self, context, obj, resources, progress):
- if resources is None:
- resources = {}
- self.export_object_resources(context, obj, resources, None, progress)
+ def export_object(self, obj, resources, progress):
+ if obj.type!='MESH':
+ raise ValueError("Object is not a mesh")
lods = self.collect_object_lods(obj)
if self.visible_collections:
collections = [c.collection for c in context.view_layer.layer_collection.children if not (c.hide_viewport or c.collection.hide_viewport)]
objs = [o for o in objs if any((o.name in c.all_objects) for c in collections)]
- objs = [o for o in objs if o.type=="MESH" and not o.lod_for_parent]
+ objs = [o for o in objs if o.type=="MESH" and not o.lod_for_parent and o.data.vertices]
objs = [o for o in objs if (not o.compound or o.parent not in objs)]
objs.sort(key=lambda x:x.name)
path, base = os.path.split(out_fn)
base, ext = os.path.splitext(base)
- from .export_object import ObjectExporter
- object_export = ObjectExporter()
-
object_prototypes = {}
unique_objects = []
export_names = {}
from .util import Progress
progress = Progress(self.show_progress and context)
+ from .export import DataExporter
+ data_exporter = DataExporter()
+
resources = {}
- self.export_scene_resources(context, unique_objects, resources, progress)
+ data_exporter.export_resources(context, unique_objects, resources, None, progress)
for n, r in resources.items():
if r.name in export_names:
r.name = export_names[r.name]
scene_res.write_to_file(out_fn)
- def export_scene_resources(self, context, objs, resources, progress):
- from .export_object import ObjectExporter
- object_export = ObjectExporter()
- object_export.single_file = False
-
- material_atlases = {}
-
- for i, o in enumerate(objs):
- progress.push_task_slice(o.name, i, len(objs))
- object_export.export_object_resources(context, o, resources, material_atlases, progress)
- obj_name = o.name+".object"
- resources[obj_name] = object_export.export_object(context, o, resources, progress)
- progress.pop_task()
-
def export_scene(self, context, objs, resources, prototypes, progress):
from .datafile import Resource, Statement, Token
scene_res = Resource("scene.scene", "scene")
projects / libs / gl.git / commitdiff