Files
IfcOpenShell/src/ifcopenshell-python/ifcopenshell/draw.py
T

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

742 lines
29 KiB
Python
Raw Normal View History

# IfcOpenShell - IFC toolkit and geometry engine
# Copyright (C) 2021 Thomas Krijnen <thomas@aecgeeks.com>
#
# This file is part of IfcOpenShell.
#
# IfcOpenShell is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 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
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with IfcOpenShell. If not, see <http://www.gnu.org/licenses/>.
"""2D drawing generation and serialisation"""
2021-08-17 14:55:32 +02:00
import itertools
2021-08-17 14:55:32 +02:00
import json
2025-12-19 18:53:04 +05:00
import math
import operator
import warnings
2025-12-19 18:53:04 +05:00
from collections.abc import Callable, Sequence
from dataclasses import dataclass, field, fields
from functools import reduce
from xml.dom.minidom import parseString
import numpy
import shapely
2021-08-17 14:55:32 +02:00
import ifcopenshell
import ifcopenshell.geom
import ifcopenshell.util.element
import ifcopenshell.util.selector
2021-08-17 14:55:32 +02:00
W = ifcopenshell.ifcopenshell_wrapper
WHITE = numpy.array((1.0, 1.0, 1.0))
DO_NOTHING = lambda *args: None
2026-04-14 19:42:20 +05:00
ARRANGE_POLYGON_SETTINGS = W.arrange_polygon_settings() if hasattr(W, "arrange_polygon_settings") else None
2021-08-17 14:55:32 +02:00
@dataclass
class draw_settings:
width: float = 297.0
height: float = 420.0
scale: float = 1.0 / 100.0
2021-08-17 15:58:15 +02:00
auto_elevation: bool = False
auto_section: bool = False
auto_floorplan: bool = True
space_names: bool = False
space_areas: bool = False
door_arcs: bool = False
2021-11-05 11:03:44 +01:00
subtract_before_hlr: bool = False
cache: bool = False
css: str = ""
2021-08-17 15:58:15 +02:00
storey_heights: str = "none"
storey_filter: str = ""
2021-08-17 14:55:32 +02:00
include_entities: str = ""
exclude_entities: str = "IfcOpeningElement"
drawing_guid: str = field(
default="",
metadata={
"doc": "Use a drawing with the provided GlobalId. Setting takes priority over 'drawing_object_type'."
},
)
drawing_object_type: str = field(
default="", metadata={"doc": 'Use IfcAnnotations with provided ObjectType for drawings (e.g. "DRAWING").'}
)
2021-11-20 14:05:11 +01:00
profile_threshold: int = -1
cells: bool = True
2022-04-24 14:54:49 +02:00
merge_cells: bool = False
2022-11-12 15:49:29 +01:00
include_projection: bool = True
2024-07-10 15:30:11 +05:00
hlr_poly: bool = False
2023-06-29 13:10:17 +02:00
prefilter: bool = True
2023-08-10 15:44:21 +08:00
include_curves: bool = False
2023-08-25 12:43:32 +02:00
unify_inputs: bool = True
arrange_spaces: bool = False
arrange_zones: bool = False
zone_filter: str = ""
zone_label: str = ""
2025-06-01 09:21:59 +02:00
mirror_y: bool = False
class type_unsafe_value:
def __init__(self, value):
self.value = value
def __eq__(self, other):
try:
v = type(other)(self.value)
except:
return False
return v == other
2024-07-02 11:40:38 +05:00
def main(
settings: draw_settings,
files: list[ifcopenshell.file],
iterators: Sequence[ifcopenshell.geom.iterator] = (),
merge_projection: bool = True,
progress_function: Callable = DO_NOTHING,
logger=None,
2024-07-02 11:40:38 +05:00
):
if logger is None and ifcopenshell.logger is not None:
logger = ifcopenshell.logger.Root()
2021-11-05 11:03:44 +01:00
def by_guid(g):
for f in files:
try:
return f.by_guid(g)
except:
pass
2021-11-05 11:03:44 +01:00
geom_settings = ifcopenshell.geom.settings(
# when not doing booleans, proper solids from shells isn't a requirement
REORIENT_SHELLS=settings.subtract_before_hlr,
# SVG serialiazation depends on element hierarchy now to look up the parent
ELEMENT_HIERARCHY=True,
2021-11-05 11:03:44 +01:00
)
2024-07-02 14:53:19 +05:00
# this is required for serialization
dimensionality = W.CURVES_SURFACES_AND_SOLIDS if settings.include_curves else W.SURFACES_AND_SOLIDS
geom_settings.set("dimensionality", dimensionality)
2024-07-02 14:53:19 +05:00
geom_settings.set("iterator-output", ifcopenshell.ifcopenshell_wrapper.NATIVE)
geom_settings.set("apply-default-materials", True)
if not iterators:
iterator_kwargs = {}
if settings.include_entities:
iterator_kwargs["include"] = settings.include_entities.split(",")
elif settings.exclude_entities:
iterator_kwargs["exclude"] = settings.exclude_entities.split(",")
def yield_parents(el):
yield el
if par := ifcopenshell.util.element.get_parent(el):
yield from yield_parents(par)
def has_selected_parent(el):
return any(settings.storey_filter in (x.Name or x.GlobalId) for x in yield_parents(el))
def create_iter(f):
if settings.storey_filter and iterator_kwargs.get("include"):
iterator_kwargs["include"] = list(
filter(has_selected_parent, sum((f.by_type(x) for x in iterator_kwargs["include"]), []))
)
return ifcopenshell.geom.iterator(geom_settings, f, logger=logger, **iterator_kwargs)
# We have to keep the iterator in memory because otherwise
# the styles are cleared up.
iterators = list(
map(
create_iter,
files,
)
2021-08-17 14:55:32 +02:00
)
2022-01-10 15:42:24 +11:00
2021-11-05 11:03:44 +01:00
if settings.cache:
2024-08-26 17:33:17 +05:00
serializer_settings = ifcopenshell.geom.serializer_settings()
cache = ifcopenshell.geom.serializers.hdf5("cache.h5", geom_settings, serializer_settings)
2021-11-05 11:03:44 +01:00
for it in iterators:
it.set_cache(cache)
2021-08-17 14:55:32 +02:00
# Initialize serializer
buffer = ifcopenshell.geom.serializers.buffer()
2024-07-02 14:53:19 +05:00
serialiser_settings = ifcopenshell.geom.serializer_settings()
sr = ifcopenshell.geom.serializers.svg(buffer, geom_settings, serialiser_settings)
2021-08-17 14:55:32 +02:00
2021-08-17 15:58:15 +02:00
sr.setFile(files[0])
if settings.auto_floorplan:
sr.setSectionHeightsFromStoreys()
2021-08-17 14:55:32 +02:00
# setElevationRefGuid and setElevationRef are also mutually exclusive in C-code.
# Note that guid or object type are not checked anywhere to be valid,
# it's up to user to keep them valid for the provided projects.
if settings.drawing_guid or settings.drawing_object_type:
if settings.drawing_guid:
if not by_guid(settings.drawing_guid):
raise ValueError(f"Unable to find guid {settings.drawing_guid!r}")
sr.setElevationRefGuid(settings.drawing_guid)
elif settings.drawing_object_type:
sr.setElevationRef(settings.drawing_object_type)
sr.setWithoutStoreys(True)
2021-08-17 14:55:32 +02:00
# required for svgfill
sr.setPolygonal(True)
sr.setUseNamespace(True)
2022-11-12 15:49:29 +01:00
sr.setAlwaysProject(settings.include_projection)
2021-08-17 14:55:32 +02:00
2021-11-20 14:05:11 +01:00
sr.setProfileThreshold(settings.profile_threshold)
2021-08-17 14:55:32 +02:00
sr.setBoundingRectangle(settings.width, settings.height)
sr.setScale(settings.scale)
sr.setAutoElevation(settings.auto_elevation)
sr.setAutoSection(settings.auto_section)
2021-08-17 15:58:15 +02:00
sr.setPrintSpaceNames(settings.space_names)
sr.setPrintSpaceAreas(settings.space_areas)
sr.setDrawDoorArcs(settings.door_arcs)
sr.setNoCSS(not not settings.css)
2021-11-05 11:03:44 +01:00
if settings.subtract_before_hlr:
sr.setSubtractionSettings(W.ALWAYS)
2024-07-10 15:30:11 +05:00
sr.setUseHlrPoly(settings.hlr_poly)
2023-06-29 13:10:17 +02:00
sr.setUsePrefiltering(settings.prefilter)
2023-08-25 12:43:32 +02:00
sr.setUnifyInputs(settings.unify_inputs)
2025-06-01 09:21:59 +02:00
sr.setMirrorY(settings.mirror_y)
2023-06-29 13:10:17 +02:00
2021-08-17 15:58:15 +02:00
try:
2022-01-10 15:42:24 +11:00
sh = ["none", "full", "left"].index(settings.storey_heights)
2021-08-17 15:58:15 +02:00
sr.setDrawStoreyHeights(sh)
except:
raise ValueError("storey_heights should be one of {'none', 'full', 'left'}")
2021-08-17 14:55:32 +02:00
"""
# It is also possible to add drawing planes manually
import bpy
obj = bpy.context.active_object
2021-08-17 14:55:32 +02:00
sr.addDrawing(
obj.matrix_world.transposed()[3][0:3], # location
obj.matrix_world.transposed()[2][0:3], # z axis (view direction)
obj.matrix_world.transposed()[0][0:3], # x axis
2021-08-17 14:55:32 +02:00
"Test", # drawing name
True # include projection
)
"""
# Initialize tree
tree = ifcopenshell.geom.tree()
# This instructs the tree to explode BReps into faces and return
# the style of the face when running tree.select_ray()
tree.enable_face_styles(True)
# Loop over iterators for geometric content
for i, it in enumerate(iterators):
2024-07-02 11:40:38 +05:00
for elem in it:
2021-08-17 14:55:32 +02:00
sr.write(elem)
if elem.type != "IfcSpace":
tree.add_element(elem)
progress_function("file", i, "progress", it.progress())
progress_function("hidden line rendering")
sr.finalize()
# Obtain SVG output from serializer buffer
svg_data_1 = buffer.get_value()
if not merge_projection:
return svg_data_1
2022-01-10 15:42:24 +11:00
2021-11-20 14:05:11 +01:00
if not settings.cells:
return svg_data_1.encode("ascii", "xmlcharrefreplace")
2024-07-02 14:49:14 +05:00
2024-12-04 14:17:39 +05:00
def yield_groups(n, tag="g"):
if n.nodeType == n.ELEMENT_NODE and n.tagName == tag:
2021-08-17 14:55:32 +02:00
yield n
for c in n.childNodes:
yield from yield_groups(c, tag=tag)
2021-08-17 14:55:32 +02:00
2022-04-24 14:54:49 +02:00
dom1 = parseString(svg_data_1)
svg1 = dom1.childNodes[0]
2021-08-17 14:55:32 +02:00
# From file 1 we take the groups to be substituted
groups1 = [g for g in yield_groups(svg1) if g.getAttribute("class") == "projection"]
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
# Parse SVG into vector of line segments
#
# The second argument 'projection' tells the parser to only include <g> groups
# that have the classname 'projection'. The IfcOpenShell SVG serializer puts
# the hidden line rendering output into this group. So the sections are not
# included here as they already form closed loops.
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
ls_groups = W.svg_to_line_segments(svg_data_1, "projection")
for i, (ls, g1) in enumerate(zip(ls_groups, groups1)):
progress_function("creating cells", i)
2024-07-02 14:49:14 +05:00
2021-08-17 14:55:32 +02:00
projection, g1 = g1, g1.parentNode
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
svgfill_context = W.context(W.FILTERED_CARTESIAN_QUOTIENT, 1.0e-3)
2023-04-21 16:29:04 +02:00
# remove duplicates (without tolerance)
ls = list(map(tuple, set(map(frozenset, ls))))
2022-04-24 14:54:49 +02:00
svgfill_context.add(ls)
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
if settings.merge_cells:
# To be refined:
# - Find cells on original line segments
# - Associate cells with IFC entities for merging
# - Merge cells by discarding edges
2024-07-02 14:49:14 +05:00
# - Associate cells with IFC entities for styling
2022-04-24 14:54:49 +02:00
num_passes = 1
else:
num_passes = 0
2024-07-02 14:49:14 +05:00
for iteration in range(num_passes + 1):
2022-04-24 14:54:49 +02:00
# initialize empty group, note that in the current approach only one
# group is stored
ps = W.svg_groups_of_polygons()
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
if iteration != 0 or svgfill_context.build():
svgfill_context.write(ps)
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
"""
# Debugging tool to plot line segments and cells
from matplotlib import pyplot as plt
arr = numpy.array(ls).reshape((-1, 2, 2))
for x in arr:
plt.plot(x.T[0], x.T[1])
for x in ps[0]:
plt.fill(numpy.array(x.boundary).T[0], numpy.array(x.boundary).T[1])
"""
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
if iteration != num_passes:
pairs = svgfill_context.get_face_pairs()
2024-07-02 14:49:14 +05:00
semantics = [None] * (max(pairs) + 1)
2022-04-24 14:54:49 +02:00
# For every edge print the two neighbouring faces
# for x in range(0, len(pairs), 2):
# print(x // 2, *pairs[x:x+2])
# Reserialize cells into an SVG string
svg_data_2 = W.polygons_to_svg(ps, True)
# We parse both SVG files to create on document with the combination of sections from
# the output directly from the serializer and the cells found from the hidden line
# rendering
dom2 = parseString(svg_data_2)
svg2 = dom2.childNodes[0]
# file 2 only has the groups we are interested in.
# in fact in the approach, it's only a single group
2024-07-02 14:49:14 +05:00
g2 = next(yield_groups(svg2))
2022-04-24 14:54:49 +02:00
# These are attributes on the original group that we can use to reconstruct
# a 4x4 matrix of the projection used in the SVG generation process
nm = g1.getAttribute("ifc:name")
m4 = numpy.array(json.loads(g1.getAttribute("ifc:plane")))
m3 = numpy.array(json.loads(g1.getAttribute("ifc:matrix3")))
m44 = numpy.eye(4)
m44[0][0:2] = m3[0][0:2]
m44[1][0:2] = m3[1][0:2]
m44[0][3] = m3[0][2]
m44[1][3] = m3[1][2]
m44 = numpy.linalg.inv(m44)
def project(xy, z=0.0):
xyzw = m44 @ numpy.array(xy + [z, 1.0])
xyzw[1] *= -1.0
return (m4 @ xyzw)[0:3]
def pythonize(arr):
return tuple(map(float, arr))
# Loop over the cell paths
for pi, p in enumerate(g2.getElementsByTagName("path")):
progress_function("group", i, "pass", iteration, "path", pi)
d = p.getAttribute("d")
# point inside is an attribute that comes from line_segments_to_polygons()
# it is an arbitrary point guaranteed to be inside the polygon and outside
# of any potential inner bounds. We can use this to construct a ray to find
# the face of the IFC element that the cell belongs to.
assert p.hasAttribute("ifc:pointInside")
xy = list(map(float, p.getAttribute("ifc:pointInside").split(",")))
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
a, b = project(xy, 0.0), project(xy, -100.0)
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
inside_elements = tree.select(pythonize(a))
2024-07-02 14:49:14 +05:00
if inside_elements:
2022-04-24 14:54:49 +02:00
elements = None
if iteration != num_passes:
semantics[pi] = (inside_elements[0], -1)
else:
elements = tree.select_ray(pythonize(a), pythonize(b - a))
if elements:
# Put the IFC element entity type on the path for CSS-based styling
p.setAttribute("class", elements[0].instance.is_a())
# Obtain style (IfcOpenShell IfcGeom::Material)
style = tree.styles()[elements[0].style_index]
# This is just a demonstration. We compose a factor of using:
# - ray intersection distance
# - dot product ray . face normal
# - style transparency
# the factor determines how much white will be interpolated
# into the style diffuse color.
def clr(c):
if isinstance(c, ifcopenshell.ifcopenshell_wrapper.colour):
return c.r(), c.g(), c.b()
else:
return c
clr = numpy.array(clr(style.diffuse) if style else (0.6, 0.6, 0.6))
2022-04-24 14:54:49 +02:00
factor = (math.log(elements[0].distance + 2.0) / 7.0) * (1.0 - 0.5 * abs(elements[0].dot_product))
if style and style.has_transparency:
2022-04-24 14:54:49 +02:00
factor *= 1.0 - style.transparency
clr = WHITE * (1.0 - factor) + clr * factor
svg_fill = "rgb(%s)" % ", ".join(str(f * 255.0) for f in clr[0:3])
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
if iteration != num_passes:
semantics[pi] = elements[0]
else:
svg_fill = "none"
p.setAttribute("style", "fill: " + svg_fill)
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
if iteration != num_passes:
to_remove = []
for he_idx in range(0, len(pairs), 2):
# @todo instead of ray_distance, better do (x.point - y.point).dot(x.normal)
# to see if they're coplanar, because ray-distance will be different in case
# of element surfaces non-orthogonal to the view direction
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
def format(x):
2024-07-02 14:49:14 +05:00
if x is None:
return None
2022-04-24 14:54:49 +02:00
elif isinstance(x, tuple):
# found to be inside element using tree.select() no face or style info
return x
2024-07-02 14:49:14 +05:00
else:
return (x.instance.is_a(), x.ray_distance, tuple(x.position))
pp = pairs[he_idx : he_idx + 2]
2022-04-24 14:54:49 +02:00
if pp == (-1, -1):
continue
data = list(map(format, map(semantics.__getitem__, pp)))
2024-07-02 14:49:14 +05:00
if None not in data and data[0][0] == data[1][0] and abs(data[0][1] - data[1][1]) < 1.0e-5:
2022-04-24 14:54:49 +02:00
to_remove.append(he_idx // 2)
# Print edge index and semantic data
# print(he_idx // 2, *data)
2024-07-02 14:49:14 +05:00
2022-04-24 14:54:49 +02:00
svgfill_context.merge(to_remove)
2021-08-17 14:55:32 +02:00
# Swap the XML nodes from the files
# Remove the original hidden line node we still have in the serializer output
g1.removeChild(projection)
g2.setAttribute("class", "projection")
# Find the children of the projection node parent
children = [x for x in g1.childNodes if x.nodeType == x.ELEMENT_NODE]
if children:
# Insert the new semantically enriched cell-based projection node
# *before* the node with sections from the serializer. SVG derives
# draw order from node order in the DOM so sections are draw over
# the projections.
g1.insertBefore(g2, children[0])
else:
# This generally shouldn't happen
g1.appendChild(g2)
if settings.arrange_spaces or settings.arrange_zones:
if settings.storey_filter:
# delete storey groups not selected by filter
# sometimes happens in case of elements protruding multiple stories
# are assigned wrongly in the decomposition tree
for rg in (g for g in yield_groups(svg1) if g.parentNode.tagName == "svg"):
storey_guid = ifcopenshell.guid.compress(rg.attributes["id"].value.split("-", 1)[1].replace("-", ""))
storey = by_guid(storey_guid)
if settings.storey_filter not in (storey.Name or storey.GlobalId):
rg.parentNode.removeChild(rg)
root_groups = [g for g in yield_groups(svg1) if g.parentNode.tagName == "svg"]
zone_groups = []
for grp in root_groups:
path_objects = [p for p in yield_groups(grp, "path") if "IfcSpace" in p.parentNode.getAttribute("class")]
2024-12-04 14:17:39 +05:00
polies = [p.getAttribute("d") for p in path_objects]
def break_at_second(char, s, offset=1):
i = s.find(char, offset)
if i == -1:
return s
else:
2025-06-01 09:23:10 +02:00
warnings.warn("Polygons with holes are not supported")
return s[0:i]
def svg_to_coordlist(str):
return [[*map(float, s[1:].split(","))] for s in break_at_second("M", str).split(" ")[:-1]]
polies = [svg_to_coordlist(d) for d in polies]
2024-12-04 14:17:39 +05:00
def create_poly(b):
p = ifcopenshell.ifcopenshell_wrapper.polygon_2()
p.boundary = b
return p
2024-12-04 14:17:39 +05:00
def min_bound_extent_p2(p):
arr = numpy.array(p.boundary)
return (arr.max(axis=0) - arr.min(axis=0)).min() > 0.5
def min_bound_extent_li(p):
arr = numpy.array(p)
return (arr.max(axis=0) - arr.min(axis=0)).min() > 0.5
def polygon_to_svg_path(doc, polygon: shapely.Polygon) -> str:
def ring_to_path(coords):
coords = list(coords)[:-1]
return "M " + " L ".join(f"{x},{y}" for x, y in coords) + " Z"
rings = [polygon.exterior, *polygon.interiors]
d = " ".join(ring_to_path(r.coords) for r in rings)
path = doc.createElement("path")
path.setAttribute("d", d)
path.setAttribute("fill-rule", "evenodd")
return path
included = list(map(min_bound_extent_li, polies))
polies = [p for p, incl in zip(polies, included) if incl]
path_objects = [p for p, incl in zip(path_objects, included) if incl]
section_polies = list(map(shapely.Polygon, polies))
polies = [create_poly(p) for p in polies]
def has_relevant_zone(i):
inst = by_guid(path_objects[i].parentNode.attributes["ifc:guid"].value)
groups = [rel.RelatingGroup for rel in inst.HasAssignments if rel.is_a("IfcRelAssignsToGroup")]
if settings.zone_filter:
query, val = settings.zone_filter.rsplit("=", 1)
# cast to type encountered in pset prop
val = type_unsafe_value(val)
groups = [g for g in groups if ifcopenshell.util.selector.get_element_value(g, query) == val]
return len(groups) > 0
if settings.arrange_zones:
path_objects, section_polies, polies = zip(
*(tup for i, tup in enumerate(zip(path_objects, section_polies, polies)) if has_relevant_zone(i))
)
arranged = W.arrange_polygons(
*filter(None, (ARRANGE_POLYGON_SETTINGS,)),
polies, # ty: ignore[too-many-positional-arguments]
*((logger,) if logger is not None else ()),
)
svg_data_3 = W.polygons_to_svg(arranged, False)
dom3 = parseString(svg_data_3)
svg3 = dom3.childNodes[0]
g3 = next(yield_groups(svg3))
arranged_polies = []
for p in g3.getElementsByTagName("path"):
# arranged path
sp = shapely.Polygon(svg_to_coordlist(p.getAttribute("d")))
arranged_polies.append(sp)
# Mapping of arrange_poly_idx (stored as list) -> instance
# Obtained by point containment of sectionpoly in arranged
mapping = [None] * len(arranged_polies)
for i, p in enumerate(section_polies):
g = by_guid(path_objects[i].parentNode.attributes["ifc:guid"].value)
contains = [ap.contains(p.representative_point()) for ap in arranged_polies]
# arrangement, so this can generally be assumed
if sum(contains) == 1:
idx = contains.index(True)
# mapping[idx].append(g)
if mapping[idx] is None:
mapping[idx] = g
else:
warnings.warn(f"Not applying {g}; overlapping with {mapping[idx]}")
else:
warnings.warn(f"Point {p.representative_point()} not covered by arrangement")
if settings.arrange_zones:
# @todo test whether output with only arrange_spaces is unaltered for bimlegal to function
zone_assignment = [None] * len(arranged_polies)
for i, m in enumerate(mapping):
if m is not None:
groups = [rel.RelatingGroup for rel in m.HasAssignments if rel.is_a("IfcRelAssignsToGroup")]
if settings.zone_filter:
query, val = settings.zone_filter.rsplit("=", 1)
# cast to type encountered in pset prop
val = type_unsafe_value(val)
groups = [
g for g in groups if ifcopenshell.util.selector.get_element_value(g, query) == val
]
zone_assignment[i] = frozenset(groups)
all_zones = list(set(itertools.chain.from_iterable(a or () for a in zone_assignment)))
all_zone_memberships = [
{i for i, zns in enumerate(zone_assignment) if zns is not None and zn in zns} for zn in all_zones
]
def union_find(sets):
merged = True
while merged:
merged = False
result = []
while sets:
first, *rest = sets
first = set(first)
# merge any overlapping sets into `first`
changed = True
while changed:
changed = False
new_rest = []
for r in rest:
if first & r: # overlap?
first |= r
changed = True
merged = True
else:
new_rest.append(r)
rest = new_rest
result.append(first)
sets = rest
sets[:] = result
return sets
to_merge = union_find(all_zone_memberships)
zone_groups.append(dom1.createElement("g"))
for idx_set in to_merge:
zone = next(iter(zone_assignment[max(idx_set)]))
poly = shapely.unary_union([arranged_polies[i] for i in idx_set])
poly = poly.geoms if poly.geom_type == "MultiPolygon" else [poly]
g = dom1.createElement("g")
def classes():
yield zone.is_a()
for propname, propval in reduce(
operator.or_, ifcopenshell.util.element.get_psets(zone).values()
).items():
if propval is True:
yield propname
g.setAttribute("class", " ".join(classes()))
for p in poly:
g.appendChild(polygon_to_svg_path(dom1, p))
if (lbl := settings.zone_label) and (
val := ifcopenshell.util.selector.get_element_value(zone, lbl)
):
pt = p.representative_point()
text_el = dom1.createElement("text")
text_el.setAttribute("x", str(pt.x))
text_el.setAttribute("y", str(pt.y))
text_el.appendChild(dom1.createTextNode(str(val)))
g.appendChild(text_el)
zone_groups[-1].appendChild(g)
else:
zone_groups.append(g3)
for rg, zg in zip(root_groups, zone_groups):
if not settings.arrange_zones:
for p in rg.getElementsByTagName("path"):
p.setAttribute("style", "fill: none; stroke: black; stroke-width: 0.05")
rg.appendChild(zg)
if settings.css:
svg = dom1.documentElement
style = dom1.createElement("style")
style.setAttribute("type", "text/css")
style.appendChild(
dom1.createTextNode(open(settings.css[1:]).read() if settings.css.startswith("@") else settings.css)
)
svg.insertBefore(style, svg.firstChild)
data = dom1.toprettyxml()
2021-08-17 14:55:32 +02:00
data = data.encode("ascii", "xmlcharrefreplace")
return data
if __name__ == "__main__":
2025-12-19 18:53:04 +05:00
import argparse
2021-11-20 14:05:53 +01:00
import time
2022-01-10 15:42:24 +11:00
2021-11-20 14:05:53 +01:00
times = []
2022-01-10 15:42:24 +11:00
2021-11-20 14:05:53 +01:00
def measure(task, fn):
t0 = time.time()
r = fn()
dt = time.time() - t0
times.append((task, dt))
return r
2022-01-10 15:42:24 +11:00
2021-11-20 14:05:53 +01:00
def print_progress(*args):
2022-01-10 15:42:24 +11:00
print("\r", *args, " " * 10, end="", flush=True)
2021-08-17 14:55:32 +02:00
parser = argparse.ArgumentParser()
parser.add_argument(
"files",
type=str,
nargs="+",
help=(
"List of files for script to use. "
"Last file is considered an output file (.svg), all other files are existing IFC files."
),
)
2021-08-17 14:55:32 +02:00
for field in fields(draw_settings):
name = field.name.replace("_", "-")
description = field.metadata.get("doc") or ""
description += " " if description else ""
description += f"Default: {repr(field.default)}."
2021-08-17 14:55:32 +02:00
if field.type == bool:
parser.add_argument(
f"--{name}",
help=description,
dest=field.name,
action="store_true",
2021-08-17 14:55:32 +02:00
)
parser.add_argument(f"--no-{name}", dest=field.name, action="store_false")
parser.set_defaults(**{field.name: field.default})
else:
parser.add_argument(f"--{name}", help=description, dest=field.name, type=field.type, default=field.default)
2021-08-17 14:55:32 +02:00
args = vars(parser.parse_args())
if len(args["files"]) < 2:
parser.error("At least 2 files are required: one or more input files and one output file.")
2021-08-17 14:55:32 +02:00
files = args.pop("files")
output = files.pop()
settings = draw_settings(**args)
2021-11-20 14:05:53 +01:00
files = measure("open files", lambda: list(map(ifcopenshell.open, files)))
result = measure("processing", lambda: main(settings, files, progress_function=print_progress))
open(output, "wb").write(result)
2022-01-10 15:42:24 +11:00
2021-11-20 14:05:53 +01:00
print("\r Done!", " " * 20)
2022-01-10 15:42:24 +11:00
2021-11-20 14:05:53 +01:00
for t, dt in times:
print(f"{t}: {dt}")