1 files changed, 146 insertions, 0 deletions
diff --git a/io_mesh_stl/__init__.py b/io_mesh_stl/__init__.py
new file mode 100644
index 00000000..a96f1a75
--- /dev/null
+++ b/io_mesh_stl/__init__.py
@@ -0,0 +1,146 @@
+# ##### BEGIN GPL LICENSE BLOCK #####
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of the GNU General Public License
+#  as published by the Free Software Foundation; either version 2
+#  of the License, or (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, write to the Free Software Foundation,
+#  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+#
+# ##### END GPL LICENSE BLOCK #####
+
+bl_addon_info = {
+    "name": "STL format",
+    "author": "Guillaume Bouchard (Guillaum)",
+    "version": (1,),
+    "blender": (2, 5, 3),
+    "api": 31667,
+    "location": "File > Import/Export > Stl",
+    "description": "Import/Export STL files",
+    "warning": "",
+    "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.5/Py/"\
+        "Scripts/File I-O/STL",
+    "tracker_url": "https://projects.blender.org/tracker/index.php?"
+        "func=detail&aid=22837&group_id=153&atid=469",
+    "category": "Import/Export"}
+
+# @todo write the wiki page
+
+"""
+Import/Export STL files (binary or ascii)
+
+- Import automatically remove the doubles.
+- Export can export with/without modifiers applied
+
+Issues:
+
+Import:
+    - Does not handle the normal of the triangles
+    - Does not handle endien
+"""
+
+import itertools
+import os
+
+import bpy
+from bpy.props import *
+from io_utils import ExportHelper, ImportHelper
+
+
+try:
+    init_data
+
+    reload(stl_utils)
+    reload(blender_utils)
+except:
+    from io_mesh_stl import stl_utils
+    from io_mesh_stl import blender_utils
+
+init_data = True
+
+
+class StlImporter(bpy.types.Operator, ImportHelper):
+    '''
+    Load STL triangle mesh data
+    '''
+    bl_idname = "import_mesh.stl"
+    bl_label = "Import STL"
+    
+    filename_ext = ".stl"
+
+    files = CollectionProperty(name="File Path",
+                          description="File path used for importing "
+                                      "the STL file",
+                          type=bpy.types.OperatorFileListElement)
+
+    directory = StringProperty()
+
+    def execute(self, context):
+        paths = (os.path.join(self.directory, name.name) for name in self.files)
+
+        for path in paths:
+            objName = bpy.path.display_name(path.split("\\")[-1].split("/")[-1])
+            tris, pts = stl_utils.read_stl(path)
+
+            blender_utils.create_and_link_mesh(objName, tris, pts)
+
+        return {'FINISHED'}
+
+
+class StlExporter(bpy.types.Operator, ExportHelper):
+    '''
+    Save STL triangle mesh data from the active object
+    '''
+    bl_idname = "export_mesh.stl"
+    bl_label = "Export STL"
+
+    filename_ext = ".stl"
+
+    ascii = BoolProperty(name="Ascii",
+                         description="Save the file in ASCII file format",
+                         default=False)
+    apply_modifiers = BoolProperty(name="Apply Modifiers",
+                                   description="Apply the modifiers "
+                                               "before saving",
+                                   default=True)
+
+    def execute(self, context):
+        faces = itertools.chain.from_iterable(
+            blender_utils.faces_from_mesh(ob, self.apply_modifiers)
+            for ob in context.selected_objects)
+
+        stl_utils.write_stl(self.filepath, faces, self.ascii)
+
+        return {'FINISHED'}
+
+
+def menu_import(self, context):
+    self.layout.operator(StlImporter.bl_idname,
+                         text="Stl (.stl)").filepath = "*.stl"
+
+
+def menu_export(self, context):
+    default_path = os.path.splitext(bpy.data.filepath)[0] + ".stl"
+    self.layout.operator(StlExporter.bl_idname,
+                         text="Stl (.stl)").filepath = default_path
+
+
+def register():
+    bpy.types.INFO_MT_file_import.append(menu_import)
+    bpy.types.INFO_MT_file_export.append(menu_export)
+
+
+def unregister():
+    bpy.types.INFO_MT_file_import.remove(menu_import)
+    bpy.types.INFO_MT_file_export.remove(menu_export)
+
+
+if __name__ == "__main__":
+    register()