]> git.tdb.fi Git - libs/gl.git/blob - blender/io_mspgl/__init__.py
Raise an exception if the TBN UV layer is not found during export
[libs/gl.git] / blender / io_mspgl / __init__.py
1 bl_info = {
2         "name": "Msp GL datafiles",
3         "blender": (2, 80, 0),
4         "author": "Mikko Rasa",
5         "location": "File > Export",
6         "description": "Export Msp GL data",
7         "category": "Import-Export" }
8
9 if "bpy" in locals():
10         import imp
11         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":
12                 if sub in locals():
13                         imp.reload(locals()[sub])
14
15 import bpy
16 from bpy_extras.io_utils import ExportHelper
17
18 class ExportMspGLBase(ExportHelper):
19         show_progress = bpy.props.BoolProperty(name="Show progress", description="Display progress indicator while exporting", default=True)
20
21         def execute(self, context):
22                 exporter = self.create_exporter()
23                 self.prepare_exporter(exporter)
24                 exporter.export_to_file(context, self.filepath)
25                 return {"FINISHED"}
26
27         def create_exporter(self):
28                 raise Exception("create_exporter must be overridden")
29
30         def prepare_exporter(self, exporter):
31                 for k, v in self.as_keywords().items():
32                         setattr(exporter, k, v)
33
34 class ExportMspGLMeshBase(ExportMspGLBase):
35         use_strips = bpy.props.BoolProperty(name="Use strips", description="Combine faces into triangle strips", default=True)
36         use_degen_tris = bpy.props.BoolProperty(name="Use degen tris", description="Concatenate triangle strips with degenerate triangles", default=False)
37
38         def draw(self, context):
39                 self.general_col = self.layout.column()
40
41                 col = self.layout.column()
42                 col.label(text="Triangle strips")
43                 col.prop(self, "use_strips")
44                 col.prop(self, "use_degen_tris")
45
46 class ExportMspGLMesh(bpy.types.Operator, ExportMspGLMeshBase):
47         bl_idname = "export_mesh.mspgl_mesh"
48         bl_label = "Export Msp GL mesh"
49
50         filename_ext = ".mesh"
51
52         def create_exporter(self):
53                 from .export_mesh import MeshExporter
54                 return MeshExporter()
55
56 class ExportMspGLObject(bpy.types.Operator, ExportMspGLMeshBase):
57         bl_idname = "export_mesh.mspgl_object"
58         bl_label = "Export Msp GL object"
59
60         filename_ext = ".object"
61
62         single_file = bpy.props.BoolProperty(name="Single file", description="Write all data into a single file", default=True)
63         shared_resources = bpy.props.BoolProperty(name="Shared resources", description="Use global names for resource files to enable sharing", default=True)
64
65         export_lods = bpy.props.BoolProperty(name="Export LoDs", description="Export all levels of detail", default=True)
66         use_textures = bpy.props.BoolProperty(name="Use textures", description="Use textures in the exported object", default=True)
67
68         def create_exporter(self):
69                 from .export_object import ObjectExporter
70                 return ObjectExporter()
71
72         def draw(self, context):
73                 super().draw(context)
74
75                 self.general_col.prop(self, "export_lods")
76                 self.general_col.prop(self, "use_textures")
77
78                 col = self.layout.column()
79                 col.label(text="Files")
80                 col.prop(self, "single_file")
81                 if not self.single_file:
82                         col.prop(self, "shared_resources")
83
84 class ExportMspGLArmature(bpy.types.Operator, ExportMspGLBase):
85         bl_idname = "export.mspgl_armature"
86         bl_label = "Export Msp GL armature"
87
88         filename_ext = ".arma"
89
90         def create_exporter(self):
91                 from .export_armature import ArmatureExporter
92                 return ArmatureExporter()
93
94 class ExportMspGLAnimation(bpy.types.Operator, ExportMspGLBase):
95         bl_idname = "export.mspgl_animation"
96         bl_label = "Export Msp GL animation"
97
98         filename_ext = ".anim"
99
100         export_all = bpy.props.BoolProperty(name="Export all animations", description="Export all animations present on the selected objects' NLA tracks")
101         collection = bpy.props.BoolProperty(name="As a collection", description="Export the animations as a single collection file", default=True)
102         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)
103
104         def check(self, context):
105                 result = False
106
107                 ext = ".mdc" if self.export_all and self.collection else ".anim"
108                 ext_changed = (ext!=self.filename_ext)
109                 if ext_changed:
110                         if self.filepath.endswith(self.filename_ext):
111                                 self.filepath = self.filepath[:-len(self.filename_ext)]
112                         self.filename_ext = ext
113
114                 super_result = super().check(context)
115
116                 return ext_changed or super_result
117
118         def create_exporter(self):
119                 from .export_animation import AnimationExporter
120                 return AnimationExporter()
121
122         def draw(self, context):
123                 col = self.layout.column()
124                 col.prop(self, "export_all")
125                 if self.export_all:
126                         col.prop(self, "collection")
127                 col.prop(self, "looping_threshold")
128
129 class ExportMspGLScene(bpy.types.Operator, ExportMspGLBase):
130         bl_idname = "export_scene.mspgl_scene"
131         bl_label = "Export Msp GL scene"
132
133         filename_ext = ".scene"
134
135         selected_only = bpy.props.BoolProperty(name="Selected objects only", description="Only export the selected objects")
136         visible_collections = bpy.props.BoolProperty(name="Visible collections only", description="Only export objects in visible collections", default=True)
137         resource_collection = bpy.props.BoolProperty(name="Resource collection", description="Put resources to a single collection file", default=True)
138         skip_existing = bpy.props.BoolProperty(name="Skip existing files", description="Skip resources that already exist as files", default=True)
139
140         def create_exporter(self):
141                 from .export_scene import SceneExporter
142                 return SceneExporter()
143
144         def draw(self, context):
145                 col = self.layout.column()
146                 col.prop(self, "selected_only")
147                 col.prop(self, "visible_collections")
148                 col.prop(self, "resource_collection")
149                 if self.resource_collection:
150                         col.prop(self, "skip_existing")
151
152 class ExportMspGLCamera(bpy.types.Operator, ExportMspGLBase):
153         bl_idname = "export.mspgl_camera"
154         bl_label = "Export Msp GL camera"
155
156         filename_ext = ".camera"
157
158         def create_exporter(self):
159                 from .export_camera import CameraExporter
160                 return CameraExporter()
161
162 def menu_func_export(self, context):
163         self.layout.operator(ExportMspGLMesh.bl_idname, text="Msp GL mesh")
164         self.layout.operator(ExportMspGLObject.bl_idname, text="Msp GL object")
165         self.layout.operator(ExportMspGLArmature.bl_idname, text="Msp GL armature")
166         self.layout.operator(ExportMspGLAnimation.bl_idname, text="Msp GL animation")
167         self.layout.operator(ExportMspGLScene.bl_idname, text="Msp GL scene")
168         self.layout.operator(ExportMspGLCamera.bl_idname, text="Msp GL camera")
169
170 classes = [ExportMspGLMesh, ExportMspGLObject, ExportMspGLArmature, ExportMspGLAnimation, ExportMspGLScene, ExportMspGLCamera]
171
172 def register():
173         for c in classes:
174                 bpy.utils.register_class(c)
175
176         bpy.types.TOPBAR_MT_file_export.append(menu_func_export)
177
178         from .properties import register_properties
179         register_properties()
180
181 def unregister():
182         for c in classes:
183                 bpy.utils.unregister_class(c)
184
185         bpy.types.TOPBAR_MT_file_export.remove(menu_func_export)
186
187         from .properties import unregister_properties
188         unregister_properties()