Upload files to "/"

simbelmyne.py

@@ -0,0 +1,104 @@
+def main_simbelmyne(parsed_args):
+    from args_main import parse_arguments_main
+    from low_level import print_starting_module, print_message, print_ending_module
+    from os.path import isfile
+    import subprocess
+    
+    print_starting_module("simbelmyne", verbose=parsed_args.verbose)
+    main_dict = parse_arguments_main(parsed_args)
+    param_file=parsed_args.paramfile
+    if param_file is None:
+        param_file = main_dict["paramdir"]+"parameters_"+main_dict["simname"]+".sbmy"
+    log_file = main_dict["logdir"]+main_dict["mode"]+".log"
+
+    if parsed_args.execution == "local":
+        print_message("Running Simbelyne in local mode.", 1, "simbelmyne", verbose=parsed_args.verbose)
+
+        
+        command_args = ["simbelmyne", param_file, log_file]
+
+        if parsed_args.verbose < 2:
+            from io import BytesIO
+            from low_level import stdout_redirector, stderr_redirector
+
+            f = BytesIO()
+            g = BytesIO()
+            with stdout_redirector(f):
+                with stderr_redirector(g):
+                    subprocess.run(command_args)
+                g.close()
+            f.close()
+        else:
+            subprocess.run(command_args)
+
+        print_message("Simbelyne finished.", 1, "simbelmyne", verbose=parsed_args.verbose)
+    
+    elif parsed_args.execution == "slurm":
+        from slurm_submission import create_slurm_script, parse_arguments_slurm
+        from args_main import parse_arguments_main
+        print_message("Running monofonic in slurm mode.", 1, "simbelmyne", verbose=parsed_args.verbose)
+        slurm_dict=parse_arguments_slurm(parsed_args)
+        main_dict=parse_arguments_main(parsed_args)
+        slurm_script = slurm_dict["scripts"]+"simbelmyne.sh"
+
+        if not isfile(slurm_script):
+            print_message(f"SLURM script {slurm_script} does not exist. Creating it.", 2, "simbelmyne", verbose=parsed_args.verbose)
+            create_slurm_script(
+                slurm_template=slurm_dict["simbelmyne_template"],
+                slurm_script=slurm_script,
+                job="simbelmyne",
+                job_config_file=param_file,
+                job_log=log_file,
+            )
+            print_message(f"SLURM script written to {slurm_script}.", 3, "simbelmyne", verbose=parsed_args.verbose)
+        else:
+            print_message(f"SLURM script {slurm_script} exists.", 2, "simbelmyne", verbose=parsed_args.verbose)
+        
+        print_message(f"Submitting monofonic job to SLURM.", 1, "simbelmyne", verbose=parsed_args.verbose)
+
+        command_args = ["sbatch", slurm_script]
+
+        if parsed_args.verbose < 2:
+            from io import BytesIO
+            from low_level import stdout_redirector, stderr_redirector
+
+            f = BytesIO()
+            g = BytesIO()
+            with stdout_redirector(f):
+                with stderr_redirector(g):
+                    subprocess.run(command_args)
+                g.close()
+            f.close()
+        else:
+            subprocess.run(command_args)
+
+        print_message("Monofonic job submitted.", 2, "simbelmyne", verbose=parsed_args.verbose)
+    
+    else:
+        raise ValueError(f"Execution mode {parsed_args.execution} not recognized.")
+    
+    print_ending_module("simbelmyne", verbose=parsed_args.verbose)
+
+
+if __name__ == "__main__":
+    from argparse import ArgumentParser
+    from args_main import register_arguments_main
+    from timestepping import register_arguments_timestepping, main_timestepping
+    from parameters_card import register_arguments_card, main_parameter_card
+    from cosmo_params import register_arguments_cosmo
+    from parameters_monofonic import register_arguments_monofonic
+    from slurm_submission import register_arguments_slurm
+
+    parser = ArgumentParser(description="Run Simbelmyne.")
+    # TODO: reduce the volume of arguments
+    register_arguments_main(parser)
+    register_arguments_timestepping(parser)
+    register_arguments_monofonic(parser)
+    register_arguments_slurm(parser)
+    register_arguments_card(parser)
+    register_arguments_cosmo(parser)
+    parsed_args = parser.parse_args()
+
+    main_parameter_card(parsed_args)
+    main_timestepping(parsed_args)
+    main_simbelmyne(parsed_args)