Merge pull request #143 from falconstryker/devel

Added copy fixed file to -split

Author: rurata

bin/MarsFiles.py

@@ -63,6 +63,7 @@
 import shutil       # For OS-friendly file operations
 import functools    # For function decorators
 import traceback    # For printing stack traces
+import shutil       # For copy/pasting fixed file after -split
 
 # Load amesCAP modules
 from amescap.Ncdf_wrapper import (Ncdf, Fort)
@@ -986,12 +987,23 @@ def split_files(file_list, split_dim):
                 )
             exit()
 
-    if split_dim in ('time', 'areo'):
-        time_dim = (np.squeeze(fNcdf.variables['time'][:]))[indices]
-        print(f"time_dim = {time_dim}")
-
     fpath = os.path.dirname(input_file_name)
     fname = os.path.basename(input_file_name)
+    
+    if split_dim in ('time', 'areo'):
+        time_dim = (np.squeeze(fNcdf.variables['time'][:]))[indices]
+        print(f"time_dim = {time_dim}\n")
+        
+        try:
+            org_fixed_file = (os.path.normpath(os.path.join(fpath, f"{original_date}.fixed.nc")))
+            new_fixed_file = (os.path.normpath(os.path.join(fpath, f"{int(time_dim[0]):05d}.fixed.nc")))
+            shutil.copyfile(org_fixed_file, new_fixed_file)
+            print(f"File {original_date}.fixed.nc copied to {int(time_dim[0]):05d}.fixed.nc.\n")
+        except FileNotFoundError:
+            print(f"{Red}No compatible fixed file for {fname} (e.g., {original_date}.fixed.nc) was found in {fpath}{Nclr}\n")
+        except Exception as e:
+            print(f"{Red}An error occurred: {e}\n")
+            
     if split_dim == 'time':
         if len(np.atleast_1d(bounds)) < 2:
             base_name = (f"{int(time_dim):05d}{fname[5:-3]}_nearest_sol"