toughanimator 0.1.10__tar.gz → 0.1.11__tar.gz

{toughanimator-0.1.10 → toughanimator-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: toughanimator
-Version: 0.1.10
+Version: 0.1.11
 Summary: A tool for visualizing TOUGH simulation outputs.
 Home-page: https://github.com/scarletref/toughanimator
 Author: scarletref

{toughanimator-0.1.10 → toughanimator-0.1.11}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='toughanimator',  # Package name on PyPI
-    version='0.1.10',
+    version='0.1.11',
     description='A tool for visualizing TOUGH simulation outputs.',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',

{toughanimator-0.1.10 → toughanimator-0.1.11}/toughanimator/run.py

@@ -6,11 +6,12 @@ import matplotlib.pyplot as plt
 dir_name = "unresolved" 
 #dir_name = "test_cases"
 #case_name = "PetraSim_2D_Conceptual"
-case_name = "2D_Utransport_MINC"
+#case_name = "cp950_problem"
+case_name = "3D five spot MINC"
 test_case_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), dir_name) 
 
 case_dir = os.path.join(test_case_dir, case_name)
-
+case_dir = r"D:\Projects\tough-processing\uwc\tough_case\uwc_polygonal_20251231\PetraSim"
 #case_dir = r"D:\Projects\202511\ta-post\0 Base_res_k"
 #case_dir = r"D:\Projects\202504\polygonal\poly_test"
 #case_dir = r"D:\Projects\202507\tough系列output\tough output format\TR_MINC_exe"
@@ -18,10 +19,10 @@ case_dir = os.path.join(test_case_dir, case_name)
 #case_dir = r"D:\Projects\202511\case_TRv4\uwc"
 #case_dir = r"D:\Projects\202501\toughanimator\test_cases\2DCCS 100yrs_RC"
 reader = ta.vis_reader(case_dir)
-#reader.write_eleme_conne()
-#reader.write_geometry()
+reader.write_eleme_conne()
+reader.write_geometry()
 #reader.write_incon()
 #reader.write_result()
-reader.write_all()
+#reader.write_all()
 
 

{toughanimator-0.1.10 → toughanimator-0.1.11}/toughanimator/tough_classes.py

@@ -9,6 +9,7 @@ import pandas as pd
 from vtkmodules.all import *
 import pathlib
 import json
+import chardet
 
 from enum import Enum
 
@@ -107,11 +108,39 @@ class vis_reader:
         if os.path.isdir(case_dir):
             config_path = os.path.join(case_dir, "config.json")
             if os.path.exists(config_path):
-                with open(config_path, "r", encoding="utf-8") as config_file:
+                with open(config_path,  "r", encoding="latin-1") as config_file:
                     config = json.load(config_file)
+            
             else:
-                print(f"Config file:({config_path}) not found. Please create it.")
-                sys.exit(1)
+                
+                # find if INFILE or flow.inp under case_dir
+                input_files = []
+                output_files = []
+                corners_file = ""
+                for file_name in os.listdir(case_dir):
+                    if file_name.upper() == "INFILE" or file_name.lower() == "flow.inp":
+                        input_files.append(file_name)
+                    if file_name.lower().endswith(".tec"):
+                        output_files.append(file_name)
+                   
+                    # if the file name contains "corners"
+                    if "corners" in file_name.lower():
+                        corners_file = file_name
+
+                # atomatically create a config.json file
+                config = {
+                    "case_name": case_dir.split(os.sep)[-1],
+                    "input_files": input_files,
+                    "output_files": output_files,
+                    "corners_file": corners_file
+                }
+  
+                # save the file
+                with open(config_path, "w", encoding="latin-1") as config_file:
+                    json.dump(config, config_file, indent=4)
+
+                #print(f"Config file:({config_path}) not found. Please create it.")
+                #sys.exit(1)
         else:
             print(f"Case directory:({case_dir}) not found. Please check it.")
             sys.exit(1)
@@ -276,7 +305,8 @@ class vis_reader:
         has_elem = False
         for input_file_path in self.setting.input_file_paths:
             line_counter = 0
-            with open(input_file_path, encoding="utf-8") as f:
+            with open(input_file_path, "r", encoding="latin-1") as f:
+
                 reading_elem = False
 
                 for line in f:
@@ -302,7 +332,7 @@ class vis_reader:
     def __check_num_of_minc(self):
         #self.minc_buffer = io.StringIO()
         minc_num = 0
-        with open(self.minc_file, encoding="utf-8") as f:
+        with open(self.minc_file, "r", encoding="latin-1") as f:
             reading_minc = False
             for line in f:
                 if line.startswith('ELEME-') or line.startswith('ELEME'):
@@ -326,7 +356,7 @@ class vis_reader:
         has_conne = False
         for input_file_path in self.setting.input_file_paths:
             line_counter = 0
-            with open(input_file_path, encoding="utf-8") as f:
+            with open(input_file_path,  "r", encoding="latin-1") as f:
                 
                 reading_conne = False
                 for line in f:
@@ -356,7 +386,7 @@ class vis_reader:
         
         for input_file_path in self.setting.input_file_paths:
             line_counter = 0
-            with open(input_file_path, encoding="utf-8") as f:
+            with open(input_file_path, "r", encoding="latin-1") as f:
                 reading_rocks = False
                 for line in f:
                     if line.startswith('ROCKS-'):
@@ -397,7 +427,7 @@ class vis_reader:
         
         for input_file_path in self.setting.input_file_paths:
             line_counter = 0
-            with open(input_file_path, encoding="utf-8") as f:
+            with open(input_file_path,  "r", encoding="latin-1") as f:
                 
                 reading_incon = False
                 for line in f:
@@ -430,7 +460,7 @@ class vis_reader:
             reading_pram = False
             for input_file_path in self.setting.input_file_paths:
                 line_counter = 0
-                with open(input_file_path, encoding="utf-8") as f:
+                with open(input_file_path, "r", encoding="latin-1") as f:
                     reading_pram = False
                     for line in f:
                         if line.startswith('PARAM-'):
@@ -531,7 +561,7 @@ class vis_reader:
             buffer.flush()
             buffer.close()
 
-        with open(self.current_out_file, encoding="utf-8") as f:
+        with open(self.current_out_file, "r", encoding="latin-1") as f:
             for line in f:
                 line_number = line_number + 1
                 values = line.strip().split(',')
@@ -587,7 +617,7 @@ class vis_reader:
         start_index = -1
         self.time_steps_list = []
         
-        with open(self.current_out_file, encoding="utf-8") as f:
+        with open(self.current_out_file, "r", encoding="latin-1") as f:
             for line in f:
                 line_number = line_number + 1
                 values = line.strip().split(',')
@@ -673,8 +703,8 @@ class vis_reader:
         value_type = ValueType.Unknown
         start_index = -1
         self.time_steps_list = []
-        
-        with open(self.current_out_file, encoding="utf-8") as f:
+
+        with open(self.current_out_file, "r", encoding="latin-1") as f:
             for line in f:
                 line_number = line_number + 1
                 values = line.strip().split(',')
@@ -760,7 +790,7 @@ class vis_reader:
         scalar_headers = []
         self.time_steps_list = []
 
-        with open(self.current_out_file, encoding="utf-8") as f:
+        with open(self.current_out_file, "r", encoding="latin-1") as f:
             for line in f:
                 if line.strip().lower().startswith('Variables'.lower()):
                     headers_value = line.strip().split('=')[1]
@@ -1188,12 +1218,10 @@ class vis_reader:
             index = self.sequence_dist[i]
             if 'ELEM' in dataframe.columns:
                 index = dataframe['ELEM'].tolist().index(elemID)
-                #elem_string = dataframe['ELEM'].iloc[index]
-                #target_row = dataframe.iloc[index]
-                #print(f'    Processing ELEM {elem_string} at index {index}')
+            else:
+                index = int(index * minc_ratio)
             for header in headers:
-                minc_index = int(index * minc_ratio)
-                value = float(self.__parse_float(dataframe[header].iloc[minc_index]))
+                value = float(self.__parse_float(dataframe[header].iloc[index]))
                 vtr.GetCellData().GetArray(header).InsertNextValue(value)
 
 
@@ -1752,7 +1780,7 @@ class vis_reader:
             corners_buffer = io.StringIO()
             csv_headers = []
             line_number = -1
-            with open(self.setting.corners_file, encoding="utf-8") as f:
+            with open(self.setting.corners_file, "r", encoding="latin-1") as f:
                 for line in f:
                     line_number = line_number + 1
                     values = line.strip().split(',')
@@ -2543,7 +2571,7 @@ class vis_reader:
         elif extension == '.csv':
             self.setting.out_format_type = OutType.CSV
             line_number = 0
-            with open(out_file_path, encoding="utf-8") as f:
+            with open(out_file_path, "r", encoding="latin-1") as f:
                 for line in f:
                     if line_number == 0:
                         first_col = line.split(',')[0].strip().lower()
@@ -2623,7 +2651,7 @@ class vis_charting:
 
         variable_list_path = os.path.join(case_dir, "tough_vis", "variable_list.json")
         if os.path.isfile(variable_list_path):
-            with open(variable_list_path, "r", encoding="utf-8") as f:
+            with open(variable_list_path,  "r", encoding="latin-1") as f:
                 self.variable_list = json.load(f)
         else:
             print(f'Case variable_list.json({variable_list_path}) not found.')
@@ -2631,7 +2659,7 @@ class vis_charting:
 
         timestep_list_path = os.path.join(case_dir, "tough_vis", "timestep_list.json")
         if os.path.isfile(timestep_list_path):
-            with open(timestep_list_path, "r", encoding="utf-8") as f:
+            with open(timestep_list_path,  "r", encoding="latin-1") as f:
                 self.time_steps_list = json.load(f)
         else:
             print(f'Case timestep_list.json({timestep_list_path}) not found.')

{toughanimator-0.1.10 → toughanimator-0.1.11}/toughanimator.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: toughanimator
-Version: 0.1.10
+Version: 0.1.11
 Summary: A tool for visualizing TOUGH simulation outputs.
 Home-page: https://github.com/scarletref/toughanimator
 Author: scarletref