Create a python package for IfcBlender [#3412173]

2026-09-26 02:07:36 +00:00 · 2011-09-22 09:16:58 +00:00
parent 1b9c9615d1
commit eda8fd0c25
1 changed files with 39 additions and 30 deletions
@@ -0,0 +1,119 @@
+###############################################################################
+#                                                                             #
+# This file is part of IfcOpenShell.                                          #
+#                                                                             #
+# IfcOpenShell is free software: you can redistribute it and/or modify        #
+# it under the terms of the Lesser GNU General Public License as published by #
+# the Free Software Foundation, either version 3.0 of the License, or         #
+# (at your option) any later version.                                         #
+#                                                                             #
+# IfcOpenShell 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                #
+# Lesser GNU General Public License for more details.                         #
+#                                                                             #
+# You should have received a copy of the Lesser GNU General Public License    #
+# along with this program. If not, see <http://www.gnu.org/licenses/>.        #
+#                                                                             #
+###############################################################################
+
+# <pep8 compliant>
+
+###############################################################################
+#                                                                             #
+# Based on the Wavefront OBJ File importer by Campbell Barton                 #
+#                                                                             #
+###############################################################################
+
+bl_info = {
+    "name": "IfcBlender",
+    "description": "Import files in the "\
+        "Industry Foundation Classes (.ifc) file format",
+    "author": "Thomas Krijnen, IfcOpenShell",
+    "blender": (2, 5, 8),
+    "api": 37702,
+    "location": "File > Import",
+    "warning": "",
+    "wiki_url": "http://sourceforge.net/apps/"\
+        "mediawiki/ifcopenshell/index.php",
+    "tracker_url": "http://sourceforge.net/tracker/?group_id=543113",
+    "category": "Import-Export"}
+
+if "bpy" in locals():
+    import imp
+    if "IfcImport" in locals():
+        imp.reload(IfcImport)
+
+import bpy
+import mathutils
+from bpy.props import StringProperty
+from bpy_extras.io_utils import ImportHelper
+
+
+class ImportIFC(bpy.types.Operator, ImportHelper):
+    bl_idname = "import_scene.ifc"
+    bl_label = "Import .ifc file"
+
+    filename_ext = ".ifc"
+    filter_glob = StringProperty(default="*.ifc", options={'HIDDEN'})
+
+    def execute(self, context):
+        from . import IfcImport
+        if not IfcImport.Init(self.filepath):
+            return {'FINISHED'}
+        ids = {}
+        while True:
+            ob = IfcImport.Get()
+            if not ob.type in ['IfcSpace', 'IfcOpeningElement']:
+                f = ob.mesh.faces
+                v = ob.mesh.verts
+                m = ob.matrix
+                t = ob.type[0:21]
+                if ob.mesh.id in ids:
+                    me = ids[ob.mesh.id]
+                else:
+                    verts = [[v[i],v[i+1],v[i+2]] for i in range(0,len(v),3)]
+                    faces = [[f[i],f[i+1],f[i+2]] for i in range(0,len(f),3)]
+                    me = bpy.data.meshes.new('mesh%d' % ob.mesh.id)
+                    me.from_pydata(verts, [], faces)
+                    if t in bpy.data.materials:
+                        mat = bpy.data.materials[t]
+                        mat.use_fake_user = True
+                    else:
+                        mat = bpy.data.materials.new(t)
+                    me.materials.append(mat)
+                    ids[ob.mesh.id] = me
+                bob = bpy.data.objects.new(ob.name, me)
+                bob.matrix_world = mathutils.Matrix(([m[0], m[1], m[2], 0],
+                    [m[3], m[4], m[5], 0],
+                    [m[6], m[7], m[8], 0],
+                    [m[9], m[10], m[11], 1]))
+                bpy.context.scene.objects.link(bob)
+                #me.update()
+                bpy.ops.object.select_all(action='DESELECT')
+                bpy.ops.object.select_name(name=bob.name)
+                bpy.ops.object.mode_set(mode='EDIT')
+                bpy.ops.mesh.normals_make_consistent()
+                bpy.ops.object.mode_set(mode='OBJECT')
+            if not IfcImport.Next():
+                break
+        return {'FINISHED'}
+
+
+def menu_func_import(self, context):
+    self.layout.operator(ImportIFC.bl_idname,
+        text="Industry Foundation Classes (.ifc)")
+
+
+def register():
+    bpy.utils.register_module(__name__)
+    bpy.types.INFO_MT_file_import.append(menu_func_import)
+
+
+def unregister():
+    bpy.utils.unregister_module(__name__)
+    bpy.types.INFO_MT_file_import.remove(menu_func_import)
+
+
+if __name__ == "__main__":
+    register()