Black

Author: aothms

ifcgraph.py

@@ -6,19 +6,19 @@
 
 
 class IfcEntity:
-        def __init__(self, data):
-                self.ifctype = data['ifc_type']
-                self.id = data['id']
-                
-        def __str__(self):
-                return "#" + str(self.id) + "_" + self.ifctype
-        __repr__ = __str__
+    def __init__(self, data):
+        self.ifctype = data["ifc_type"]
+        self.id = data["id"]
 
+    def __str__(self):
+        return "#" + str(self.id) + "_" + self.ifctype
 
+    __repr__ = __str__
 
-def get_cmap(n, name='hsv'):
-    '''Returns a function that maps each index in 0, 1, ..., n-1 to a distinct 
-    RGB color; the keyword argument name must be a standard mpl colormap name.'''
+
+def get_cmap(n, name="hsv"):
+    """Returns a function that maps each index in 0, 1, ..., n-1 to a distinct
+    RGB color; the keyword argument name must be a standard mpl colormap name."""
     return plt.cm.get_cmap(name, n)
 
 
@@ -32,14 +32,14 @@ def get_cmap(n, name='hsv'):
 types_color = {}
 
 for e in ents.values():
-        for r in e['attributes'][1]:
-                G.add_edge(e['id'], r)
-                color_map.append('red')
-                ifctypes.add(e['ifc_type'])
-                ifctypes.add(ents[r]['ifc_type'])
-                
+    for r in e["attributes"][1]:
+        G.add_edge(e["id"], r)
+        color_map.append("red")
+        ifctypes.add(e["ifc_type"])
+        ifctypes.add(ents[r]["ifc_type"])
+
 
-cmap = get_cmap(len(ifctypes),'rainbow')
+cmap = get_cmap(len(ifctypes), "rainbow")
 
 colors = [cmap(i) for i in range(len(ifctypes))]
 
@@ -49,14 +49,14 @@ def get_cmap(n, name='hsv'):
 
 i = 0
 for eid in G.nodes():
-        t = ents[eid]['ifc_type']
-        # print(t)
-        if t in type_color_mapping.keys():
-                cols.append(type_color_mapping[t])
-        else:
-                type_color_mapping[t] = colors[i]
-                i = i + 1
-                cols.append(type_color_mapping[t])
+    t = ents[eid]["ifc_type"]
+    # print(t)
+    if t in type_color_mapping.keys():
+        cols.append(type_color_mapping[t])
+    else:
+        type_color_mapping[t] = colors[i]
+        i = i + 1
+        cols.append(type_color_mapping[t])
 
 
 # k controls the distance between the nodes and varies between 0 and 1
@@ -70,27 +70,23 @@ def get_cmap(n, name='hsv'):
 
 c_map = [cmap(i) for i in range(len(nodes))]
 
-nc = nx.draw_networkx_nodes(G, pos,nodelist=nodes, node_color=cols,node_size=200)
+nc = nx.draw_networkx_nodes(G, pos, nodelist=nodes, node_color=cols, node_size=200)
 
 # edges
-elarge = [(u, v) for (u, v, d) in G.edges(data=True) ]
-nx.draw_networkx_edges(G, pos,edgelist=elarge, width=1)
+elarge = [(u, v) for (u, v, d) in G.edges(data=True)]
+nx.draw_networkx_edges(G, pos, edgelist=elarge, width=1)
 
 nx.draw_networkx_labels(G, pos, font_size=8, font_family="sans-serif")
 
 
-red_patch = mpatches.Patch(color='red', label='The red data')
-blue_patch = mpatches.Patch(color='blue', label='The blue data')
+red_patch = mpatches.Patch(color="red", label="The red data")
+blue_patch = mpatches.Patch(color="blue", label="The blue data")
 
 patches = []
-for k,v in type_color_mapping.items():
-        patches.append(mpatches.Patch(color=v, label=k))
+for k, v in type_color_mapping.items():
+    patches.append(mpatches.Patch(color=v, label=k))
 
 
-plt.legend(handles=patches,fontsize=8)
+plt.legend(handles=patches, fontsize=8)
 plt.axis("off")
 plt.show()
-
-
-
-

main.py

@@ -28,9 +28,11 @@ def __init__(self, filecontent, exception):
 
     def asdict(self, with_message=True):
         return {
-            "type": "unexpected_token"
-            if isinstance(self.exception, UnexpectedToken)
-            else "unexpected_character",
+            "type": (
+                "unexpected_token"
+                if isinstance(self.exception, UnexpectedToken)
+                else "unexpected_character"
+            ),
             "lineno": self.exception.line,
             "column": self.exception.column,
             "found_type": self.exception.token.type.lower(),
@@ -244,16 +246,19 @@ def enumeration(self, s):
 
 @dataclass
 class entity_instance:
-    id : int
-    type : str
+    id: int
+    type: str
     attributes: tuple
     lines: tuple
+
     def __getitem__(self, k):
         # compatibility with dict
         return getattr(self, k)
+
     def __repr__(self):
         return f'#{self.id}={self.type}({",".join(map(str, self.attributes))})'
 
+
 def create_step_entity(entity_tree):
     entity = {}
     t = T(visit_tokens=True).transform(entity_tree)
@@ -317,7 +322,14 @@ def make_header_ent(ast):
         return ents
 
 
-def parse(*, filename=None, filecontent=None, with_progress=False, with_tree=True, with_header=False):
+def parse(
+    *,
+    filename=None,
+    filecontent=None,
+    with_progress=False,
+    with_tree=True,
+    with_header=False,
+):
     if filename:
         assert not filecontent
         filecontent = builtins.open(filename, encoding=None).read()
@@ -384,13 +396,14 @@ class file:
     """
     A somewhat compatible interface (but very limited) to ifcopenshell.file
     """
+
     def __init__(self, parse_outcomes):
         self.header_, self.data_ = parse_outcomes
-        
+
     @property
     def schema_identifier(self) -> str:
-        return self.header_['FILE_SCHEMA'][0][0]
-        
+        return self.header_["FILE_SCHEMA"][0][0]
+
     @property
     def schema(self) -> str:
         """General IFC schema version: IFC2X3, IFC4, IFC4X3."""
@@ -403,7 +416,10 @@ def schema(self) -> str:
                 ((p, match.group(p)) for p in prefixes),
             )
         )
-        return "".join("".join(map(str, t)) if t[1] else "" for t in zip(prefixes, version_tuple[0:2]))
+        return "".join(
+            "".join(map(str, t)) if t[1] else ""
+            for t in zip(prefixes, version_tuple[0:2])
+        )
 
     @property
     def schema_version(self) -> tuple[int, int, int, int]:
@@ -417,14 +433,14 @@ def schema_version(self) -> tuple[int, int, int, int]:
             number = re.search(prefix + r"(\d)", schema)
             version.append(int(number.group(1)) if number else 0)
         return tuple(version)
-    
+
     @property
     def header(self):
         return types.SimpleNamespace(**{k.lower(): v for k, v in self.header_.items()})
 
     def __getitem__(self, key: numbers.Integral) -> entity_instance:
         return self.by_id(key)
-        
+
     def by_id(self, id: int) -> entity_instance:
         """Return an IFC entity instance filtered by IFC ID.
 
@@ -447,11 +463,18 @@ def by_type(self, type: str) -> list[entity_instance]:
         :rtype: list[entity_instance]
         """
         type_lc = type.lower()
-        return list(filter(lambda ent: ent.type.lower() == type_lc, itertools.chain.from_iterable(self.data_.values())))
+        return list(
+            filter(
+                lambda ent: ent.type.lower() == type_lc,
+                itertools.chain.from_iterable(self.data_.values()),
+            )
+        )
+
 
 def open(fn) -> file:
     return file(parse(filename=fn, with_tree=True, with_header=True))
 
+
 if __name__ == "__main__":
     args = [x for x in sys.argv[1:] if not x.startswith("-")]
     flags = [x for x in sys.argv[1:] if x.startswith("-")]

test_parser.py

@@ -4,18 +4,21 @@
 from main import parse, ValidationError
 from contextlib import nullcontext
 
+
 def create_context(fn):
     if "fail_" in fn:
         return pytest.raises(ValidationError)
     else:
         return nullcontext()
 
+
 @pytest.mark.parametrize("file", glob.glob("fixtures/*.ifc"))
 def test_file_with_tree(file):
-    with create_context(file): 
+    with create_context(file):
         parse(filename=file, with_tree=True)
 
+
 @pytest.mark.parametrize("file", glob.glob("fixtures/*.ifc"))
 def test_file_without_tree(file):
-    with create_context(file): 
+    with create_context(file):
         parse(filename=file, with_tree=False)