molde 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

molde/utils/format_sequences.py

@@ -1,44 +1,44 @@
-from math import ceil, floor
-
-
-def format_long_sequence(
-    sequence, max_width=30, max_lines=3, item_separator=", ", identation="► "
-):
-    strings = [str(i) for i in sequence]
-
-    initial_lines = []
-    for _ in range(ceil(max_lines / 2)):
-        new_line = _extract_line(strings, max_width, len(item_separator))
-        if not new_line:
-            break
-        initial_lines.append(new_line)
-        strings = strings[len(new_line) :]
-
-    strings.reverse()
-    final_lines = []
-    for _ in range(floor(max_lines / 2)):
-        new_line = _extract_line(strings, max_width, len(item_separator))
-        if not new_line:
-            break
-        final_lines.append(new_line)
-        strings = strings[len(new_line) :]
-    final_lines.reverse()
-
-    if strings and initial_lines:
-        initial_lines[-1] = ["..."]
-    lines = initial_lines + final_lines
-
-    formated_lines = [item_separator.join(line) for line in lines]
-    concatenated_lines = identation + f"\n{identation}".join(formated_lines)
-    return concatenated_lines
-
-
-def _extract_line(strings, max_width, separator_size):
-    line = []
-    current_width = 0
-    for string in strings:
-        if len(string) + current_width + separator_size > max_width:
-            break
-        line.append(string)
-        current_width += len(string) + separator_size
-    return line
+from math import ceil, floor
+
+
+def format_long_sequence(
+    sequence, max_width=30, max_lines=3, item_separator=", ", identation="► "
+):
+    strings = [str(i) for i in sequence]
+
+    initial_lines = []
+    for _ in range(ceil(max_lines / 2)):
+        new_line = _extract_line(strings, max_width, len(item_separator))
+        if not new_line:
+            break
+        initial_lines.append(new_line)
+        strings = strings[len(new_line) :]
+
+    strings.reverse()
+    final_lines = []
+    for _ in range(floor(max_lines / 2)):
+        new_line = _extract_line(strings, max_width, len(item_separator))
+        if not new_line:
+            break
+        final_lines.append(new_line)
+        strings = strings[len(new_line) :]
+    final_lines.reverse()
+
+    if strings and initial_lines:
+        initial_lines[-1] = ["..."]
+    lines = initial_lines + final_lines
+
+    formated_lines = [item_separator.join(line) for line in lines]
+    concatenated_lines = identation + f"\n{identation}".join(formated_lines)
+    return concatenated_lines
+
+
+def _extract_line(strings, max_width, separator_size):
+    line = []
+    current_width = 0
+    for string in strings:
+        if len(string) + current_width + separator_size > max_width:
+            break
+        line.append(string)
+        current_width += len(string) + separator_size
+    return line

molde/utils/poly_data_utils.py

@@ -1,66 +1,66 @@
-from pathlib import Path
-
-from vtkmodules.vtkCommonCore import vtkUnsignedCharArray, vtkUnsignedIntArray
-from vtkmodules.vtkCommonDataModel import vtkPolyData
-from vtkmodules.vtkCommonTransforms import vtkTransform
-from vtkmodules.vtkFiltersGeneral import vtkTransformPolyDataFilter
-from vtkmodules.vtkIOGeometry import vtkOBJReader, vtkSTLReader
-
-
-def set_polydata_colors(data: vtkPolyData, color: tuple) -> vtkUnsignedCharArray:
-    n_cells = data.GetNumberOfCells()
-    cell_colors = vtkUnsignedCharArray()
-    cell_colors.SetName("colors")
-    cell_colors.SetNumberOfComponents(3)
-    cell_colors.SetNumberOfTuples(n_cells)
-    cell_colors.FillComponent(0, color[0])
-    cell_colors.FillComponent(1, color[1])
-    cell_colors.FillComponent(2, color[2])
-    data.GetCellData().SetScalars(cell_colors)
-    return cell_colors
-
-
-def set_polydata_property(
-    data: vtkPolyData, property_data: int, property_name: str
-) -> vtkUnsignedIntArray:
-    n_cells = data.GetNumberOfCells()
-    cell_identifier = vtkUnsignedIntArray()
-    cell_identifier.SetName(property_name)
-    cell_identifier.SetNumberOfTuples(n_cells)
-    cell_identifier.Fill(property_data)
-    data.GetCellData().AddArray(cell_identifier)
-    return cell_identifier
-
-
-def read_obj_file(path: str | Path) -> vtkPolyData:
-    reader = vtkOBJReader()
-    reader.SetFileName(str(path))
-    reader.Update()
-    return reader.GetOutput()
-
-
-def read_stl_file(path: str | Path) -> vtkPolyData:
-    reader = vtkSTLReader()
-    reader.SetFileName(str(path))
-    reader.Update()
-    return reader.GetOutput()
-
-
-def transform_polydata(
-    polydata: vtkPolyData,
-    position=(0, 0, 0),
-    rotation=(0, 0, 0),
-    scale=(1, 1, 1),
-) -> vtkPolyData:
-    transform = vtkTransform()
-    transform.Translate(position)
-    transform.Scale(scale)
-    transform.RotateX(rotation[0])
-    transform.RotateY(rotation[1])
-    transform.RotateZ(rotation[2])
-    transform.Update()
-    transformation = vtkTransformPolyDataFilter()
-    transformation.SetTransform(transform)
-    transformation.SetInputData(polydata)
-    transformation.Update()
-    return transformation.GetOutput()
+from pathlib import Path
+
+from vtkmodules.vtkCommonCore import vtkUnsignedCharArray, vtkUnsignedIntArray
+from vtkmodules.vtkCommonDataModel import vtkPolyData
+from vtkmodules.vtkCommonTransforms import vtkTransform
+from vtkmodules.vtkFiltersGeneral import vtkTransformPolyDataFilter
+from vtkmodules.vtkIOGeometry import vtkOBJReader, vtkSTLReader
+
+
+def set_polydata_colors(data: vtkPolyData, color: tuple) -> vtkUnsignedCharArray:
+    n_cells = data.GetNumberOfCells()
+    cell_colors = vtkUnsignedCharArray()
+    cell_colors.SetName("colors")
+    cell_colors.SetNumberOfComponents(3)
+    cell_colors.SetNumberOfTuples(n_cells)
+    cell_colors.FillComponent(0, color[0])
+    cell_colors.FillComponent(1, color[1])
+    cell_colors.FillComponent(2, color[2])
+    data.GetCellData().SetScalars(cell_colors)
+    return cell_colors
+
+
+def set_polydata_property(
+    data: vtkPolyData, property_data: int, property_name: str
+) -> vtkUnsignedIntArray:
+    n_cells = data.GetNumberOfCells()
+    cell_identifier = vtkUnsignedIntArray()
+    cell_identifier.SetName(property_name)
+    cell_identifier.SetNumberOfTuples(n_cells)
+    cell_identifier.Fill(property_data)
+    data.GetCellData().AddArray(cell_identifier)
+    return cell_identifier
+
+
+def read_obj_file(path: str | Path) -> vtkPolyData:
+    reader = vtkOBJReader()
+    reader.SetFileName(str(path))
+    reader.Update()
+    return reader.GetOutput()
+
+
+def read_stl_file(path: str | Path) -> vtkPolyData:
+    reader = vtkSTLReader()
+    reader.SetFileName(str(path))
+    reader.Update()
+    return reader.GetOutput()
+
+
+def transform_polydata(
+    polydata: vtkPolyData,
+    position=(0, 0, 0),
+    rotation=(0, 0, 0),
+    scale=(1, 1, 1),
+) -> vtkPolyData:
+    transform = vtkTransform()
+    transform.Translate(position)
+    transform.Scale(scale)
+    transform.RotateX(rotation[0])
+    transform.RotateY(rotation[1])
+    transform.RotateZ(rotation[2])
+    transform.Update()
+    transformation = vtkTransformPolyDataFilter()
+    transformation.SetTransform(transform)
+    transformation.SetInputData(polydata)
+    transformation.Update()
+    return transformation.GetOutput()

molde/utils/tree_info.py

@@ -1,52 +1,52 @@
-from collections import namedtuple
-
-TreeItem = namedtuple("TreeItem", ["name", "data", "unity"])
-SEPARATOR = TreeItem("", "", "")
-
-
-class TreeInfo:
-    def __init__(self, name: str) -> None:
-        self.name = name
-        self.items = list()
-
-    def add_separator(self):
-        self.items.append(TreeItem("", "", ""))
-        return TreeItem("", "", "")
-
-    def add_item(self, name, data, unity="") -> TreeItem:
-        item = TreeItem(name, data, unity)
-        self.items.append(item)
-        return item
-
-    def items_without_extra_separators(self):
-        extra_separators = 0
-        for item in reversed(self.items):
-            if item != SEPARATOR:
-                break
-            extra_separators += 1
-        return self.items[: len(self.items) - extra_separators]
-
-    def __str__(self):
-        title = self.name.upper()
-        longest_name = max(0, *[len(item.name) for item in self.items])
-        longest_data = max(0, *[len(str(item.data)) for item in self.items])
-        spaces = longest_name + longest_data + 3
-
-        text = f"{title} \n"
-        pruned_items = self.items_without_extra_separators()
-
-        for item in pruned_items[:-1]:
-            if item == SEPARATOR:
-                text += "│\n"
-                continue
-            spacer = " " * (spaces - len(item.name) - len(str(item.data)))
-            unity = f"[{item.unity}]" if item.unity else ""
-            text += f"├─ {item.name}: {spacer}{item.data} {unity}\n"
-
-        # last item
-        item = pruned_items[-1]
-        spacer = " " * (spaces - len(item.name) - len(str(item.data)))
-        unity = f"[{item.unity}]" if item.unity else ""
-        text += f"╰─ {item.name}: {spacer}{item.data} {unity}\n"
-        text += "\n"
-        return text
+from collections import namedtuple
+
+TreeItem = namedtuple("TreeItem", ["name", "data", "unity"])
+SEPARATOR = TreeItem("", "", "")
+
+
+class TreeInfo:
+    def __init__(self, name: str) -> None:
+        self.name = name
+        self.items = list()
+
+    def add_separator(self):
+        self.items.append(TreeItem("", "", ""))
+        return TreeItem("", "", "")
+
+    def add_item(self, name, data, unity="") -> TreeItem:
+        item = TreeItem(name, data, unity)
+        self.items.append(item)
+        return item
+
+    def items_without_extra_separators(self):
+        extra_separators = 0
+        for item in reversed(self.items):
+            if item != SEPARATOR:
+                break
+            extra_separators += 1
+        return self.items[: len(self.items) - extra_separators]
+
+    def __str__(self):
+        title = self.name.upper()
+        longest_name = max(0, *[len(item.name) for item in self.items])
+        longest_data = max(0, *[len(str(item.data)) for item in self.items])
+        spaces = longest_name + longest_data + 3
+
+        text = f"{title} \n"
+        pruned_items = self.items_without_extra_separators()
+
+        for item in pruned_items[:-1]:
+            if item == SEPARATOR:
+                text += "│\n"
+                continue
+            spacer = " " * (spaces - len(item.name) - len(str(item.data)))
+            unity = f"[{item.unity}]" if item.unity else ""
+            text += f"├─ {item.name}: {spacer}{item.data} {unity}\n"
+
+        # last item
+        item = pruned_items[-1]
+        spacer = " " * (spaces - len(item.name) - len(str(item.data)))
+        unity = f"[{item.unity}]" if item.unity else ""
+        text += f"╰─ {item.name}: {spacer}{item.data} {unity}\n"
+        text += "\n"
+        return text