Specially import grids even though they are 2D as grids are pretty important.

This commit is contained in:
Dion Moult
2020-02-19 16:20:02 +11:00
parent cfe18d41d6
commit 6c47ff73fd
+26 -4
View File
@@ -126,6 +126,8 @@ class IfcImporter():
self.settings = ifcopenshell.geom.settings() self.settings = ifcopenshell.geom.settings()
if self.ifc_import_settings.should_import_curves: if self.ifc_import_settings.should_import_curves:
self.settings.set(self.settings.INCLUDE_CURVES, True) self.settings.set(self.settings.INCLUDE_CURVES, True)
self.settings_2d = ifcopenshell.geom.settings()
self.settings_2d.set(self.settings_2d.INCLUDE_CURVES, True)
self.project = None self.project = None
self.spatial_structure_elements = {} self.spatial_structure_elements = {}
self.elements = {} self.elements = {}
@@ -152,6 +154,7 @@ class IfcImporter():
self.purge_diff() self.purge_diff()
self.patch_ifc() self.patch_ifc()
self.create_type_products() self.create_type_products()
self.create_grids()
# TODO: Deprecate after bug #682 is fixed and the new importer is stable # TODO: Deprecate after bug #682 is fixed and the new importer is stable
if self.ifc_import_settings.should_use_legacy or self.diff: if self.ifc_import_settings.should_use_legacy or self.diff:
self.create_products_legacy() self.create_products_legacy()
@@ -234,6 +237,21 @@ class IfcImporter():
if element.ObjectPlacement.RelativePlacement.RefDirection: if element.ObjectPlacement.RelativePlacement.RefDirection:
element.ObjectPlacement.RelativePlacement.RefDirection.DirectionRatios = (1., 0., 0.) element.ObjectPlacement.RelativePlacement.RefDirection.DirectionRatios = (1., 0., 0.)
def create_grids(self):
grids = self.file.by_type('IfcGrid')
for grid in grids:
collection = bpy.data.collections.new(self.get_name(grid))
self.project['blender'].children.link(collection)
self.create_grid_axes(grid.UAxes, collection)
self.create_grid_axes(grid.VAxes, collection)
def create_grid_axes(self, axes, grid):
for axis in axes:
shape = ifcopenshell.geom.create_shape(self.settings_2d, axis.AxisCurve)
mesh = self.create_mesh(axis, shape)
obj = bpy.data.objects.new(f'IfcGrisAxis/{axis.AxisTag}', mesh)
grid.objects.link(obj)
def create_type_products(self): def create_type_products(self):
type_products = self.file.by_type('IfcTypeProduct') type_products = self.file.by_type('IfcTypeProduct')
for type_product in type_products: for type_product in type_products:
@@ -579,10 +597,14 @@ class IfcImporter():
def create_mesh(self, element, shape): def create_mesh(self, element, shape):
try: try:
mesh = bpy.data.meshes.new(shape.geometry.id) if hasattr(shape, 'geometry'):
f = shape.geometry.faces geometry = shape.geometry
e = shape.geometry.edges else:
v = shape.geometry.verts geometry = shape
mesh = bpy.data.meshes.new(geometry.id)
f = geometry.faces
e = geometry.edges
v = geometry.verts
vertices = [[v[i], v[i + 1], v[i + 2]] vertices = [[v[i], v[i + 1], v[i + 2]]
for i in range(0, len(v), 3)] for i in range(0, len(v), 3)]
faces = [[f[i], f[i + 1], f[i + 2]] faces = [[f[i], f[i + 1], f[i + 2]]