ras-commander 0.1.6__py2.py3-none-any.whl → 0.20.dev0__py2.py3-none-any.whl

ras_commander/execution.py

@@ -1,315 +0,0 @@
-"""
-Execution operations for runningHEC-RAS simulations using subprocess.
-Based on the HEC-Commander project's "Command Line is All You Need" approach, leveraging the -c compute flag to run HEC-RAS and orchestrating changes directly in the RAS input files to achieve automation outcomes. 
-"""
-
-import os
-import subprocess
-import shutil
-from pathlib import Path
-from concurrent.futures import ThreadPoolExecutor, as_completed
-from .file_operations import FileOperations
-from .plan_operations import PlanOperations
-import subprocess
-import os
-import logging
-import time
-from .project_config import ProjectConfig
-import pandas as pd
-
-class RasExecutor:
-
-    @staticmethod
-    def compute_hecras_plan(compute_plan_file):
-        """
-        Compute a HEC-RAS plan using the provided plan file.
-
-        This method executes a HEC-RAS plan by running the HEC-RAS executable
-        with the specified plan file. It uses the plan file's path to determine
-        the corresponding project file path.
-
-        Args:
-            compute_plan_file (str): The full path to the HEC-RAS plan file (.p**)
-
-        Returns:
-            bool: True if the plan computation was successful, False otherwise.
-
-        Raises:
-            subprocess.CalledProcessError: If the HEC-RAS execution fails.
-
-        Note:
-            This method assumes that the project file (.prj) is in the same
-            directory as the plan file and has the same name (different extension).
-        """
-        config = ProjectConfig()
-        config.check_initialized()
-
-        # Derive the project file path from the plan file path
-        compute_project_file = Path(compute_plan_file).with_suffix('.prj')
-
-        cmd = f'"{config.hecras_exe_path}" -c "{compute_project_file}" "{compute_plan_file}"'
-        print(f"Running command: {cmd}")
-        try:
-            subprocess.run(cmd, check=True, shell=True, capture_output=True, text=True)
-            logging.info(f"HEC-RAS execution completed for plan: {Path(compute_plan_file).name}")
-            return True
-        except subprocess.CalledProcessError as e:
-            logging.error(f"Error running plan: {Path(compute_plan_file).name}")
-            logging.error(f"Error message: {e.output}")
-            return False
-        
-        
-    @staticmethod
-    def compute_hecras_plan_from_folder(plan_for_compute, folder_for_compute):
-        """
-        Compute a HEC-RAS plan from a specified folder.
-
-        This function allows running a plan directly from a different folder than the project folder,
-        which is useful for the -test function and parallel runs. It uses the HEC-RAS executable path
-        from the ProjectConfig, but derives other paths from the provided arguments.
-
-        Args:
-            plan_for_compute (str): The full path to the HEC-RAS plan file (.p**) to be computed.
-            folder_for_compute (str): The folder containing the HEC-RAS project files.
-
-        Returns:
-            bool: True if the plan computation was successful, False otherwise.
-
-        Raises:
-            subprocess.CalledProcessError: If the HEC-RAS execution fails.
-
-        Note:
-            This function uses the ProjectConfig only to get the hecras_exe_path.
-            Other paths are derived from plan_for_compute and folder_for_compute.
-        """
-        config = ProjectConfig()
-        config.check_initialized()
-        compute_project_file = FileOperations.find_hecras_project_file(folder_for_compute)
-        cmd = f'"{config.hecras_exe_path}" -c "{compute_project_file}" "{plan_for_compute}"'
-        print(f"Running command: {cmd}")
-        try:
-            subprocess.run(cmd, check=True, shell=True, capture_output=True, text=True)
-            logging.info(f"HEC-RAS execution completed for plan: {Path(plan_for_compute).name}")
-            return True
-        except subprocess.CalledProcessError as e:
-            logging.error(f"Error running plan: {Path(plan_for_compute).name}")
-            logging.error(f"Error message: {e.output}")
-            return False 
-
-
-    @staticmethod
-    def recreate_test_function(project_folder):
-        """
-        Recreate the -test function of HEC-RAS command line.
-        
-        Parameters:
-        project_folder (str): Path to the HEC-RAS project folder
-        
-        Returns:
-        None
-        
-        
-        This function executes all ras plans in a separate folder defined by compute_folder='[Test]', so we need to call the individual functions and use distinct variable names
-        For this function, we are using "compute" in the varable names as these are the values used for the compute operations in a separate copy of the ras project folder
-     
-        """
-        print("Starting the recreate_test_function...")
-
-        # Create the test folder path
-        compute_folder='[Test]'
-        folder_for_compute = Path(project_folder).parent / f"{Path(project_folder).name} {compute_folder}"
-        print(f"Creating the test folder: {folder_for_compute}...")
-
-        # Copy the project folder to the test folder
-        print("Copying project folder to the test folder...")
-        shutil.copytree(project_folder, folder_for_compute, dirs_exist_ok=True)
-        print(f"Test folder created at: {folder_for_compute}")
-
-        # Find the project file
-        print("Finding the project file...")
-        compute_project_file = FileOperations.find_hecras_project_file(folder_for_compute)
-
-        if not compute_project_file:
-            print("Project file not found.")
-            return
-        print(f"Project file found: {compute_project_file}")
-
-        # Parse the project file to get plan entries
-        print("Parsing the project file to get plan entries...")
-        ras_compute_plan_entries = FileOperations.get_plan_entries(compute_project_file)
-        print("Parsed project file successfully.")
-
-        # Enforce recomputing of geometry preprocessor and IB tables
-        print("Enforcing recomputing of geometry preprocessor and IB tables...")
-        for plan_file in ras_compute_plan_entries['full_path']:
-            PlanOperations.update_geompre_flags(plan_file, run_htab_value=-1, use_ib_tables_value=-1)
-        print("Recomputing enforced successfully.")
-
-        # Change max cores to 1
-        print("Changing max cores to 2 for all plan files...")
-        for plan_file in ras_compute_plan_entries['full_path']:
-            PlanOperations.set_num_cores(plan_file, num_cores=2)
-        print("Max cores updated successfully.")
-
-        # Run all plans sequentially
-        print("Running all plans sequentially...")
-        for _, plan in ras_compute_plan_entries.iterrows():
-            plan_for_compute = plan["full_path"]
-            RasExecutor.compute_hecras_plan_from_folder(plan_for_compute, folder_for_compute)
-
-        print("All plans have been executed.")
-        print("recreate_test_function completed.")
-        
-    @staticmethod    
-    def run_plans_parallel(config, max_workers, cores_per_run):
-        """
-        Run HEC-RAS plans in parallel using ThreadPoolExecutor.
-        
-        Parameters:
-        config (ProjectConfig): Configuration object containing project information
-        max_workers (int): Maximum number of parallel runs
-        cores_per_run (int): Number of cores to use per run
-        
-        Returns:
-        dict: Dictionary with plan numbers as keys and execution success as values
-        
-        This function executes all ras plans in separate folders defined by the max_workers and the numbering of the test folders [Test 1], [Test 2], etc. 
-        Each worker operates sequentially on its assigned folder while all workers operate in parallel.
-     
-        Revisions:
-        1. Created a pandas DataFrame to map each folder to a plan for execution.
-        2. Implemented logic to assign worker numbers and compute folders.
-        3. Used find_hecras_project_file function to get the full path of project files.
-        4. Updated plan file paths to use compute folders.
-        5. Revised the parallel execution to use separate threads for each worker.
-        6. Implemented a queue system for each worker to handle plans sequentially.
-        7. Updated the cleanup process to use the new folder structure.
-        """
-        import queue
-        from threading import Thread
-
-        project_folder = Path(config.project_file).parent
-        test_folders = []
-
-        # Create multiple copies of the project folder
-        for i in range(1, max_workers + 1):
-            folder_for_compute = project_folder.parent / f"{project_folder.name} [Test {i}]"
-            shutil.copytree(project_folder, folder_for_compute, dirs_exist_ok=True)
-            test_folders.append(folder_for_compute)
-            print(f"Created test folder: {folder_for_compute}")
-
-        compute_parallel_entries = []
-        for i, (_, plan_row) in enumerate(config.ras_plan_entries.iterrows()):
-            worker_number = i % max_workers
-            compute_folder = test_folders[worker_number]
-            compute_project_file = FileOperations.find_hecras_project_file(compute_folder)
-            compute_plan_file = compute_folder / Path(plan_row['full_path']).name
-            compute_parallel_entries.append({
-                'worker_number': worker_number,
-                'compute_folder': compute_folder,
-                'compute_project_file': compute_project_file,
-                'compute_plan_file': compute_plan_file,
-                'plan_number': plan_row['plan_number']
-            })
-
-        compute_parallel_df = pd.DataFrame(compute_parallel_entries)
-        print("compute_parallel_entries dataframe:")
-        display(compute_parallel_df)
-
-        results = {}
-        worker_queues = [queue.Queue() for _ in range(max_workers)]
-
-        def worker_thread(worker_id):
-            """
-            Execute HEC-RAS plans assigned to a specific worker thread.
-
-            This function continuously processes plans from the worker's queue until it's empty.
-            It sets the number of cores for each plan, computes the plan, and records the result.
-
-            Parameters:
-            worker_id (int): The ID of the worker thread.
-
-            Notes:
-            - Uses PlanOperations.set_num_cores to set the number of cores for each plan.
-            - Uses RasExecutor.compute_hecras_plan to execute each plan.
-            - Records success or failure in the 'results' dictionary.
-            - Prints status messages for completed or failed plans.
-            """
-            while True:
-                try:
-                    row = worker_queues[worker_id].get(block=False)
-                    PlanOperations.set_num_cores(str(row['compute_plan_file']), cores_per_run)
-                    success = RasExecutor.compute_hecras_plan(row['compute_plan_file'])
-                    results[row['plan_number']] = success
-                    print(f"Completed: Plan {row['plan_number']} in worker {worker_id}")
-                except queue.Empty:
-                    break
-                except Exception as e:
-                    results[row['plan_number']] = False
-                    print(f"Failed: Plan {row['plan_number']} in worker {worker_id}. Error: {str(e)}")
-
-        # Distribute plans to worker queues
-        for _, row in compute_parallel_df.iterrows():
-            worker_queues[row['worker_number']].put(row)
-
-        # Start worker threads
-        threads = []
-        for i in range(max_workers):
-            thread = Thread(target=worker_thread, args=(i,))
-            thread.start()
-            threads.append(thread)
-
-        # Wait for all threads to complete
-        for thread in threads:
-            thread.join()
-
-        # Clean up and consolidate results
-        time.sleep(3)  # Allow files to close
-        final_test_folder = project_folder.parent / f"{project_folder.name} [Test]"
-        final_test_folder.mkdir(exist_ok=True)
-        
-        for test_folder in test_folders:
-            for item in test_folder.iterdir():
-                dest_path = final_test_folder / item.name
-                if dest_path.exists():
-                    if dest_path.is_dir():
-                        shutil.rmtree(dest_path)
-                    else:
-                        dest_path.unlink()
-                shutil.move(str(item), final_test_folder)
-            shutil.rmtree(test_folder)
-            print(f"Moved and removed test folder: {test_folder}")
-
-        return results
-    
-    @staticmethod    
-    def run_all_plans_parallel(project_folder, hecras_exe_path):
-        """
-        Run all plans in a project folder in parallel.
-        
-        Parameters:
-        project_folder (str): The path to the project folder.
-        hecras_exe_path (str): The path to the HEC-RAS executable.
-        
-        Returns:
-        dict: A dictionary with plan numbers as keys and execution success status as values.
-        """
-        config = ProjectConfig.init_ras_project(project_folder, hecras_exe_path)
-        
-        if config:
-            print("ras_plan_entries dataframe:")
-            display(config.ras_plan_entries)
-            
-            max_workers = 2  # Number of parallel runs
-            cores_per_run = 2  # Number of cores per run
-            
-            results = RasExecutor.run_plans_parallel(config, max_workers, cores_per_run)
-            
-            print("\nExecution Results:")
-            for plan_number, success in results.items():
-                print(f"Plan {plan_number}: {'Successful' if success else 'Failed'}")
-            
-            return results
-        else:
-            print("Failed to initialize project configuration.")
-            return None

ras_commander/file_operations.py

@@ -1,173 +0,0 @@
-"""
-File operations for HEC-RAS project files.
-"""
-import re
-from pathlib import Path
-import pandas as pd
-
-class FileOperations:
-    """
-    A class for HEC-RAS file operations.
-    
-    
-    Revision Notes: All functions from class ProjectManager should be moved here
-    
-    
-    """
-    @staticmethod
-    def find_hecras_project_file(folder_path):
-        """
-        Find the appropriate HEC-RAS project file (.prj) in the given folder.
-        
-        Parameters:
-        folder_path (str or Path): Path to the folder containing HEC-RAS files.
-        
-        Returns:
-        Path: The full path of the selected .prj file or None if no suitable file is found.
-        """
-        print(f"running find_hecras_project_file with folder_path: {folder_path}")
-        folder_path = Path(folder_path)
-        print("Searching for .prj files...")
-        prj_files = list(folder_path.glob("*.prj"))
-        # print(f"Found {len(prj_files)} .prj files")
-        # print("Searching for .rasmap files...")
-        rasmap_files = list(folder_path.glob("*.rasmap"))
-        #print(f"Found {len(rasmap_files)} .rasmap files")
-        if len(prj_files) == 1:
-            project_file = prj_files[0]
-            # print(f"Only one .prj file found. Selecting: {project_file}")
-            # print(f"Full path: {project_file.resolve()}")
-            return project_file.resolve()
-        if len(prj_files) > 1:
-            print("Multiple .prj files found.")
-            if len(rasmap_files) == 1:
-                base_filename = rasmap_files[0].stem
-                project_file = folder_path / f"{base_filename}.prj"
-                # print(f"Found single .rasmap file. Using its base name: {base_filename}")
-                # print(f"Full path: {project_file.resolve()}")
-                return project_file.resolve()
-            print("Multiple .prj files and no single .rasmap file. Searching for 'Proj Title=' in .prj files...")
-            for prj_file in prj_files:
-                # print(f"Checking file: {prj_file.name}")
-                with open(prj_file, 'r') as file:
-                    if "Proj Title=" in file.read():
-                        # print(f"Found 'Proj Title=' in file: {prj_file.name}")
-                        # print(f"Full path: {prj_file.resolve()}")
-                        return prj_file.resolve()
-        print("No suitable .prj file found after all checks.")
-        return project_file
-
-    @staticmethod
-    def get_project_name(project_path):
-        """
-        Extract the project name from the given project path.
-        
-        Parameters:
-        project_path (Path): Path object representing the project file path
-        
-        Returns:
-        str: The project name derived from the file name without extension
-        """
-        project_name = project_path.stem
-        return project_name
-
-    @staticmethod
-    def get_plan_entries(project_file):
-        """
-        Parse HEC-RAS project file and create dataframe for plan entries.
-        
-        Parameters:
-        project_file (str): Full path to HEC-RAS project file (.prj)
-        
-        Returns:
-        pandas DataFrame: DataFrame containing plan entries
-        """
-        project_path = Path(project_file)
-        project_name = FileOperations.get_project_name(project_path)
-        project_dir = project_path.parent
-
-        with open(project_file, 'r') as f:
-            content = f.read()
-
-        plan_entries = re.findall(r'Plan File=(.*?)(?:\n|$)', content)
-
-        ras_plan_entries = pd.DataFrame({
-            'plan_number': [re.findall(r'\d+', entry)[0] for entry in plan_entries],
-            'file_name': [f"{project_name}.{entry.strip().zfill(2)}" for entry in plan_entries],
-            'full_path': [str(project_dir / f"{project_name}.{entry.strip().zfill(2)}") for entry in plan_entries],
-            'results_path': [str(project_dir / f"{project_name}.{entry.strip().zfill(2)}.hdf") for entry in plan_entries]
-        })
-
-        return ras_plan_entries
-
-    @staticmethod
-    def get_flow_entries(project_file):
-        """
-        Parse HEC-RAS project file and create dataframe for flow entries.
-        
-        Parameters:
-        project_file (str): Full path to HEC-RAS project file (.prj)
-        
-        Returns:
-        pandas DataFrame: DataFrame containing flow entries
-        """
-        project_path = Path(project_file)
-        project_name = FileOperations.get_project_name(project_path)
-        project_dir = project_path.parent
-        with open(project_file, 'r') as f:
-            content = f.read()
-        flow_entries = re.findall(r'Flow File=(.*?)(?:\n|$)', content)
-        ras_flow_entries = pd.DataFrame({
-            'flow_number': [re.findall(r'\d+', entry)[0] for entry in flow_entries],
-            'file_name': [f"{project_name}.{entry.strip().zfill(2)}" for entry in flow_entries],
-            'full_path': [str(project_dir / f"{project_name}.{entry.strip().zfill(2)}") for entry in flow_entries]
-        })
-        return ras_flow_entries
-
-    @staticmethod
-    def get_unsteady_entries(project_file):
-        """
-        Parse HEC-RAS project file and create dataframe for unsteady entries.
-        
-        Parameters:
-        project_file (str): Full path to HEC-RAS project file (.prj)
-        
-        Returns:
-        pandas DataFrame: DataFrame containing unsteady entries
-        """
-        project_path = Path(project_file)
-        project_name = FileOperations.get_project_name(project_path)
-        project_dir = project_path.parent
-        with open(project_file, 'r') as f:
-            content = f.read()
-        unsteady_entries = re.findall(r'Unsteady File=(.*?)(?:\n|$)', content)
-        ras_unsteady_entries = pd.DataFrame({
-            'unsteady_number': [re.findall(r'\d+', entry)[0] for entry in unsteady_entries],
-            'file_name': [f"{project_name}.{entry.strip().zfill(2)}" for entry in unsteady_entries],
-            'full_path': [str(project_dir / f"{project_name}.{entry.strip().zfill(2)}") for entry in unsteady_entries]
-        })
-        return ras_unsteady_entries
-
-    @staticmethod
-    def get_geom_entries(project_file):
-        """
-        Parse HEC-RAS project file and create dataframe for geometry entries.
-        
-        Parameters:
-        project_file (str): Full path to HEC-RAS project file (.prj)
-        
-        Returns:
-        pandas DataFrame: DataFrame containing geometry entries
-        """
-        project_path = Path(project_file)
-        project_name = FileOperations.get_project_name(project_path)
-        project_dir = project_path.parent
-        with open(project_file, 'r') as f:
-            content = f.read()
-        geom_entries = re.findall(r'Geom File=(.*?)(?:\n|$)', content)
-        ras_geom_entries = pd.DataFrame({
-            'geom_number': [re.findall(r'\d+', entry)[0] for entry in geom_entries],
-            'file_name': [f"{project_name}.{entry.strip().zfill(2)}" for entry in geom_entries],
-            'full_path': [str(project_dir / f"{project_name}.{entry.strip().zfill(2)}") for entry in geom_entries]
-        })
-        return ras_geom_entries

ras_commander/geometry_operations.py

@@ -1,184 +0,0 @@
-"""
-Operations for handling geometry files in HEC-RAS projects.
-"""
-from pathlib import Path
-from .plan_operations import PlanOperations
-from .file_operations import FileOperations
-from .project_config import ProjectConfig
-import re
-
-class GeometryOperations:
-    """
-    A class for operations on HEC-RAS geometry files.
-    """
-    @staticmethod
-    def clear_geometry_preprocessor_files(plan_file):
-        """
-        Clear HEC-RAS geometry preprocessor files for a given plan file.
-        
-        Parameters:
-        plan_file (str): Full path to the HEC-RAS plan file (.p*)
-        
-        Returns:
-        None
-        """
-        config = ProjectConfig()
-        config.check_initialized()
-        
-        plan_path = Path(plan_file)
-        geom_preprocessor_file = plan_path.with_suffix('.c' + plan_path.suffix[2:])
-        if geom_preprocessor_file.exists():
-            print(f"Deleting geometry preprocessor file: {geom_preprocessor_file}")
-            geom_preprocessor_file.unlink()
-            print("File deletion completed successfully.")
-        else:
-            print(f"No geometry preprocessor file found for: {plan_file}")
-
-    @staticmethod
-    def clear_geometry_preprocessor_files_for_all_plans():
-        """
-        Clear HEC-RAS geometry preprocessor files for all plan files in the project directory.
-        
-        Returns:
-        None
-        """
-        config = ProjectConfig()
-        config.check_initialized()
-        
-        for plan_file in config.project_folder.glob("*.p[0-9][0-9]"):
-            GeometryOperations.clear_geometry_preprocessor_files(plan_file)
-
-    @staticmethod
-    def copy_geometry_files(dst_folder, template_geom):
-        """
-        Copy geometry files from a template to the next available geometry number
-        and update the destination folder accordingly.
-
-        Parameters:
-        dst_folder (str): Destination folder path
-        template_geom (str): Template geometry number (e.g., 'g01')
-
-        Returns:
-        str: New geometry number (e.g., 'g03')
-        """
-        config = ProjectConfig()
-        config.check_initialized()
-
-        src_folder = config.project_folder
-        dst_folder = Path(dst_folder)
-        dst_folder.mkdir(parents=True, exist_ok=True)
-
-        # Determine the next available geometry number
-        existing_numbers = []
-        geom_file_pattern = re.compile(r'^Geom File=g(\d+)', re.IGNORECASE)
-
-        for plan_file in src_folder.glob("*.p[0-9][0-9]"):
-            with open(plan_file, 'r') as file:
-                lines = file.readlines()
-                for line in lines:
-                    match = geom_file_pattern.match(line.strip())
-                    if match:
-                        existing_numbers.append(int(match.group(1)))
-
-        if existing_numbers:
-            existing_numbers.sort()
-            next_number = 1
-            for num in existing_numbers:
-                if num == next_number:
-                    next_number += 1
-                else:
-                    break
-        else:
-            next_number = 1
-
-        next_geom_number = f"{next_number:02d}"
-
-        # Copy the template geometry file to the new geometry file
-        template_geom_filename = f"{config.project_name}.g{template_geom}"
-        src_g_file = src_folder / template_geom_filename
-        if src_g_file.exists():
-            new_geom_filename = f"{config.project_name}.g{next_geom_number}"
-            dst_g_file = dst_folder / new_geom_filename
-            dst_g_file.write_bytes(src_g_file.read_bytes())
-            print(f"Copied {src_g_file} to {dst_g_file}")
-        else:
-            raise FileNotFoundError(f"Template geometry file '{src_g_file}' does not exist.")
-
-        # Copy the corresponding .hdf file
-        src_hdf_file = src_folder / f"{template_geom_filename}.hdf"
-        if src_hdf_file.exists():
-            new_hdf_filename = f"{new_geom_filename}.hdf"
-            dst_hdf_file = dst_folder / new_hdf_filename
-            dst_hdf_file.write_bytes(src_hdf_file.read_bytes())
-            print(f"Copied {src_hdf_file} to {dst_hdf_file}")
-        else:
-            raise FileNotFoundError(f"Template geometry .hdf file '{src_hdf_file}' does not exist.")
-        config = ProjectConfig()
-        return f"g{next_geom_number}"
-
-    @staticmethod
-    def rename_geometry_files(old_number, new_number):
-        """
-        Rename geometry files in the project folder.
-        
-        Parameters:
-        old_number (str): Old geometry number (e.g., 'g01')
-        new_number (str): New geometry number (e.g., 'g02')
-        
-        Returns:
-        None
-        """
-        config = ProjectConfig()
-        config.check_initialized()
-        
-        folder = config.project_folder
-        old_g_file = next(folder.glob(f"*.{old_number}"), None)
-        if old_g_file:
-            new_g_file = folder / old_g_file.name.replace(old_number, new_number)
-            old_g_file.rename(new_g_file)
-            print(f"Renamed {old_g_file} to {new_g_file}")
-        else:
-            print(f"No .{old_number} file found in {folder}")
-        
-        old_hdf_file = next(folder.glob(f"*.{old_number}.hdf"), None)
-        if old_hdf_file:
-            new_hdf_file = folder / old_hdf_file.name.replace(old_number, new_number)
-            old_hdf_file.rename(new_hdf_file)
-            print(f"Renamed {old_hdf_file} to {new_hdf_file}")
-        else:
-            print(f"No .{old_number}.hdf file found in {folder}")
-
-    @staticmethod
-    def update_geometry_reference_in_plan(plan_file, new_geometry_number):
-        """
-        Update the geometry reference in a plan file.
-        
-        Parameters:
-        plan_file (str): Full path to the plan file
-        new_geometry_number (str): New geometry number (e.g., 'g02')
-        
-        Returns:
-        None
-        """
-        config = ProjectConfig()
-        config.check_initialized()
-        
-        geometry_full_path = next(config.project_folder.glob(f"*.g{new_geometry_number}"), None)
-        if not geometry_full_path:
-            raise FileNotFoundError(f"Geometry file '*.g{new_geometry_number}' not found in the project folder.")
-        
-        plan_path = Path(plan_file)
-        with open(plan_path, 'r') as file:
-            lines = file.readlines()
-        updated = False
-        for index, line in enumerate(lines):
-            if line.startswith("Geom File="):
-                lines[index] = f"Geom File=g{new_geometry_number}\n"
-                updated = True
-                break
-        if updated:
-            with open(plan_path, 'w') as file:
-                file.writelines(lines)
-            print(f"Updated geometry reference in {plan_file} to {new_geometry_number}")
-        else:
-            print(f"No geometry reference found in {plan_file}")