blender/io_mspgl/__init__.py

   1 bl_info = {
   2         "name": "Msp GL datafiles",
   3         "author": "Mikko Rasa",
   4         "location": "File > Export",
   5         "description": "Export Msp GL data",
   6         "category": "Import-Export" }
   7
   8 if "bpy" in locals():
   9         import imp
  10         for sub in "armature", "export_armature", "export_mesh", "export_object", "export_scene", "mesh", "outfile", "properties", "util":
  11                 if sub in locals():
  12                         imp.reload(locals()[sub])
  13
  14 import bpy
  15 from bpy_extras.io_utils import ExportHelper
  16
  17 class ExportMspGLBase(ExportHelper):
  18         show_progress = bpy.props.BoolProperty(name="Show progress", description="Display progress indicator while exporting", default=True)
  19
  20         def execute(self, context):
  21                 exporter = self.create_exporter()
  22                 self.prepare_exporter(exporter)
  23                 exporter.export(context, self.filepath)
  24                 return {"FINISHED"}
  25
  26         def create_exporter(self):
  27                 raise Exception("create_exporter must be overridden")
  28
  29         def prepare_exporter(self, exporter):
  30                 for k, v in self.as_keywords().items():
  31                         setattr(exporter, k, v)
  32
  33 class ExportMspGLMeshBase(ExportMspGLBase):
  34         use_strips = bpy.props.BoolProperty(name="Use strips", description="Combine faces into triangle strips", default=True)
  35         use_degen_tris = bpy.props.BoolProperty(name="Use degen tris", description="Concatenate triangle strips with degenerate triangles", default=False)
  36         max_strip_len = bpy.props.IntProperty(name="Max strip length", description="Maximum length for a triangle strip", default=1024, min=4, max=16384)
  37         optimize_cache = bpy.props.BoolProperty(name="Optimize cache", description="Optimize element order for vertex cache", default=True)
  38         cache_size = bpy.props.IntProperty(name="Cache size", description="Simulated vertex cache size used in optimization", default=64, min=8, max=1024)
  39         export_lines = bpy.props.BoolProperty(name="Export lines", description="Export edges without faces as lines", default=False)
  40         export_uv = bpy.props.EnumProperty(name="Export UV", description="Export UV coordinates", default="UNIT0",
  41                 items=(("NONE", "None", "No UV coordinates are exported"),
  42                         ("UNIT0", "Unit 0", "UV coordinates for unit 0 are exported"),
  43                         ("ALL", "All", "All UV coordinates are exported")))
  44         tbn_vecs = bpy.props.BoolProperty(name="TBN vectors", description="Compute tangent and binormal vectors for vertices", default=False)
  45         tbn_uvtex = bpy.props.StringProperty(name="TBN UV layer", description="UV layer to use as basis for TBN vectors", default="")
  46         compound = bpy.props.BoolProperty(name="Compound", description="Combine all selected objects into one for exporting", default=False)
  47         smoothing = bpy.props.EnumProperty(name="Smoothing", description="Smoothing method to use", default="MSPGL",
  48                 items=(("NONE", "None", "No smoothing"),
  49                         ("BLENDER", "Blender", "Use Blender's vertex normals"),
  50                         ("MSPGL", "MspGL", "Compute vertex normals internally")))
  51         export_groups = bpy.props.BoolProperty(name="Vertex groups", description="Export vertex groups and weights", default=False)
  52
  53         def draw(self, context):
  54                 col = self.layout.column()
  55                 col.prop(self, "export_lines")
  56                 col.prop(self, "compound")
  57                 col.prop(self, "smoothing")
  58                 col.prop(self, "export_groups")
  59                 self.general_col = col
  60
  61                 self.layout.separator()
  62
  63                 col = self.layout.column()
  64                 col.label("Triangle strips")
  65                 col.prop(self, "use_strips")
  66                 col.prop(self, "use_degen_tris")
  67                 col.prop(self, "max_strip_len")
  68
  69                 self.layout.separator()
  70
  71                 col = self.layout.column()
  72                 col.label("Texturing")
  73                 col.prop(self, "export_uv")
  74                 col.prop(self, "tbn_vecs")
  75                 col.prop(self, "tbn_uvtex")
  76                 self.texturing_col = col
  77
  78                 self.layout.separator()
  79
  80                 col = self.layout.column()
  81                 col.label("Vertex cache")
  82                 col.prop(self, "optimize_cache")
  83                 col.prop(self, "cache_size")
  84
  85 class ExportMspGLMesh(bpy.types.Operator, ExportMspGLMeshBase):
  86         bl_idname = "export_mesh.mspgl_mesh"
  87         bl_label = "Export Msp GL mesh"
  88
  89         filename_ext = ".mesh"
  90
  91         def create_exporter(self):
  92                 from .export_mesh import MeshExporter
  93                 return MeshExporter()
  94
  95 class ExportMspGLObject(bpy.types.Operator, ExportMspGLMeshBase):
  96         bl_idname = "export_mesh.mspgl_object"
  97         bl_label = "Export Msp GL object"
  98
  99         filename_ext = ".object"
 100
 101         external_tech = bpy.props.BoolProperty(name="External technique", description="Use an external technique specified in the object's properties", default=True)
 102
 103         textures = bpy.props.EnumProperty(name="Textures", description="Export textures", default="REF",
 104                 items=(("NONE", "None", "Ignore textures"),
 105                         ("REF", "Referenced", "Reference external data"),
 106                         ("INLINE", "Inline", "Embed textures in the object")))
 107         material_tex = bpy.props.BoolProperty(name="Material texture", description="Generate a texture based on material colors", default=False)
 108         srgb_colors = bpy.props.BoolProperty(name="sRGB colors", description="Export material colors as sRGB instead of linear", default=True)
 109
 110         separate_mesh = bpy.props.BoolProperty(name="Separate mesh", description="Write mesh data into a separate file", default=False)
 111         shared_mesh = bpy.props.BoolProperty(name="Shared mesh", description="Use mesh name for mesh file to enable sharing", default=True)
 112         separate_tech = bpy.props.BoolProperty(name="Separate technique", description="Write technique data into a separate file", default=False)
 113         shared_tech = bpy.props.BoolProperty(name="Shared technique", description="Use material name for technique file to enable sharing", default=True)
 114
 115         export_lods = bpy.props.BoolProperty(name="Export LoDs", description="Export all levels of detail", default=True)
 116
 117         def create_exporter(self):
 118                 from .export_object import ObjectExporter
 119                 return ObjectExporter()
 120
 121         def draw(self, context):
 122                 super().draw(context)
 123
 124                 col = self.general_col
 125                 col.prop(self, "external_tech")
 126                 col.prop(self, "export_lods")
 127
 128                 col = self.texturing_col
 129                 col.prop(self, "textures")
 130                 col.prop(self, "material_tex")
 131                 col.prop(self, "srgb_colors")
 132
 133                 self.layout.separator();
 134
 135                 col = self.layout.column()
 136                 col.label("Files")
 137                 col.prop(self, "separate_mesh")
 138                 if self.separate_mesh:
 139                         col.prop(self, "shared_mesh")
 140                 col.prop(self, "separate_tech")
 141                 if self.separate_tech:
 142                         col.prop(self, "shared_tech")
 143
 144 class ExportMspGLArmature(bpy.types.Operator, ExportMspGLBase):
 145         bl_idname = "export.mspgl_armature"
 146         bl_label = "Export Msp GL armature"
 147
 148         filename_ext = ".arma"
 149
 150         def create_exporter(self):
 151                 from .export_armature import ArmatureExporter
 152                 return ArmatureExporter()
 153
 154 class ExportMspGLScene(bpy.types.Operator, ExportMspGLBase):
 155         bl_idname = "export_scene.mspgl_scene"
 156         bl_label = "Export Msp GL scene"
 157
 158         filename_ext = ".scene"
 159
 160         external_tech = bpy.props.BoolProperty(name="External techniques", description="Use external techniques specified in objects' properties", default=True)
 161         resource_collection = bpy.props.BoolProperty(name="Resource collection", description="Put resources to a single collection file", default=True)
 162
 163         def create_exporter(self):
 164                 from .export_scene import SceneExporter
 165                 return SceneExporter()
 166
 167         def draw(self, context):
 168                 col = self.layout.column()
 169                 col.prop(self, "resource_collection")
 170
 171 def menu_func_export(self, context):
 172         self.layout.operator(ExportMspGLMesh.bl_idname, text="Msp GL mesh")
 173         self.layout.operator(ExportMspGLObject.bl_idname, text="Msp GL object")
 174         self.layout.operator(ExportMspGLArmature.bl_idname, text="Msp GL armature")
 175         self.layout.operator(ExportMspGLScene.bl_idname, text="Msp GL scene")
 176
 177 from .properties import MspGLMeshProperties, MspGLObjectProperties
 178
 179 def register():
 180         bpy.utils.register_module(__name__)
 181
 182         bpy.types.INFO_MT_file_export.append(menu_func_export)
 183
 184         from .properties import register_properties
 185         register_properties();
 186
 187 def unregister():
 188         bpy.utils.unregister_module(__name__)
 189
 190         bpy.types.INFO_MT_file_export.remove(menu_func_export)
 191
 192 if __name__=="__main__":
 193         register()