EditTSReader

cgre-aachen · Oct 16, 2024 · 01bce2e · 01bce2e
1 parent 38ea999
commit 01bce2e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/gemgis/raster.py b/gemgis/raster.py
@@ -1915,7 +1915,10 @@ def read_ts(path: Union[str, Path]) -> Tuple[list, list]:
             if line_type == "PROPERTIES":
                 columns += values
 
-            elif line_type == "TFACE":
+                # Deleting duplicate column names
+                columns = list(dict.fromkeys(columns))
+
+            elif line_type == "TFACE" or line_type == "END":
                 # Creating array for faces
                 faces = np.array(faces, dtype=np.int32)
 

diff --git a/gemgis/visualization.py b/gemgis/visualization.py
@@ -1193,7 +1193,7 @@ def create_polydata_from_ts(
         for i in range(len(data[0])):
             # Creating faces for PyVista PolyData
             faces = np.hstack(
-                np.pad(data[1][i], ((0, 0), (1, 0)), "constant", constant_values=3)
+                np.pad(data[1][i]-1, ((0, 0), (1, 0)), "constant", constant_values=3)
             )
 
             # Creating vertices for PyVista Polydata