toughanimator 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

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/tough_classes.py

@@ -107,11 +107,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 +304,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 +331,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 +355,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 +385,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 +426,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 +459,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 +560,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 +616,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 +702,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 +789,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 +1217,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 +1779,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 +2570,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 +2650,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 +2658,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.dist-info → toughanimator-0.1.12.dist-info}/METADATA

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

toughanimator-0.1.12.dist-info/RECORD

@@ -0,0 +1,7 @@
+toughanimator/__init__.py,sha256=m1f3fJ1SNGLNKxHlS0pLNHwaHlN6UUTh5GpS_8hC8kw,30
+toughanimator/run.py,sha256=20BqrpQyNxdGPlcKFYoiQ8nFGamjDhUYhdmWlQOnIok,1038
+toughanimator/tough_classes.py,sha256=kCXuw1BhCOoLcSi1gVUexpOn6Ynf5OJ1ooU-kSksQXc,124388
+toughanimator-0.1.12.dist-info/METADATA,sha256=h8shCedbwa0imNa6JQt1x6yC4TpdC9P2pxze3LFrcP4,4571
+toughanimator-0.1.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+toughanimator-0.1.12.dist-info/top_level.txt,sha256=pAT-UflvbjT4lTmBdzHqApZGWbywkSM3Y_qsHyLi4pU,14
+toughanimator-0.1.12.dist-info/RECORD,,

toughanimator-0.1.10.dist-info/RECORD

@@ -1,7 +0,0 @@
-toughanimator/__init__.py,sha256=m1f3fJ1SNGLNKxHlS0pLNHwaHlN6UUTh5GpS_8hC8kw,30
-toughanimator/run.py,sha256=niOqZDCWYHBzLdZfRyglHRb8mGfduu9mGgs3Jh2-Ohc,921
-toughanimator/tough_classes.py,sha256=TJP8B1gGCaSFsrjELd01A0aqSmChEXGxQM-WZyHsu94,123281
-toughanimator-0.1.10.dist-info/METADATA,sha256=8sXZSX-4aEuWNG-HlOCgCJX9LwnF50grsRk_j7GStsk,4571
-toughanimator-0.1.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-toughanimator-0.1.10.dist-info/top_level.txt,sha256=pAT-UflvbjT4lTmBdzHqApZGWbywkSM3Y_qsHyLi4pU,14
-toughanimator-0.1.10.dist-info/RECORD,,