changed all __main__ scripts to console_main() functions to be used as entry points

analysis/power_spectrum.py

@@ -263,7 +263,9 @@ def get_ylims_and_yticks(ylims):
     return ylims, yticks
 
 
-if __name__ == "__main__":
+
+
+def console_main():
     from argparse import ArgumentParser
     parser = ArgumentParser(description='Plot power spectra of fields')
 
@@ -405,3 +407,6 @@ if __name__ == "__main__":
         fig.savefig(args.directory+'power_spectrum.png')
 
 
+if __name__ == "__main__":
+    console_main()
+    

analysis/slices.py

@@ -149,8 +149,7 @@ def plot_imshow_diff(data_list,
     return fig, axes
 
 
-
+def console_main():
-if __name__ == "__main__":
     from argparse import ArgumentParser
     parser = ArgumentParser(description='Comparisons of fields slices.')
 
@@ -223,3 +222,8 @@ if __name__ == "__main__":
         fig.savefig(args.output,bbox_inches='tight')
     else:
         fig.savefig(args.directory+'slices.jpg',bbox_inches='tight')
+
+
+
+if __name__ == "__main__":
+    console_main()

main.py

@@ -220,7 +220,7 @@ def check_consistency(card_dict, mode):
             raise ValueError(f"ModulePMCOLA is not 1: ModulePMCOLA={card_dict["ModulePMCOLA"]}")
 
 
-if __name__ == "__main__":
+def console_main():
     from argparse import ArgumentParser
     from args_main import register_arguments_main
     from timestepping import register_arguments_timestepping, main_timestepping
@@ -239,3 +239,8 @@ if __name__ == "__main__":
     register_arguments_cosmo(parser)
     parsed_args = parser.parse_args()
     main(parsed_args)
+
+
+if __name__ == "__main__":
+    console_main()
+    

scripts/convert_snapshot_to_density.py

@@ -36,8 +36,7 @@ def convert_snapshot_to_density(snapshot_path, output_path, N=None, corner=(0.0,
     print("Done.")
 
 
-
+def console_main():
-if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Convert snapshot to density.")
     parser.add_argument(
         "-S",
@@ -77,3 +76,7 @@ if __name__ == "__main__":
         N=args.N,
         corner=args.corner,
     )
+
+
+if __name__ == "__main__":
+    console_main()

scripts/field_to_field.py

@@ -115,8 +115,7 @@ def field_to_field(
     print("Done.")
 
 
-
+def console_main():
-if __name__ == "__main__":
     import argparse
     parser = argparse.ArgumentParser(description="Convert a field from one size to another.")
     parser.add_argument("-i","--input_file", type=str, help="Input field file")
@@ -129,3 +128,7 @@ if __name__ == "__main__":
     args = parser.parse_args()
 
     field_to_field(args.input_file, args.output_file, args.output_size, args.output_L, args.output_dpm, args.output_corner, args.boundary_conditions)
+
+
+if __name__ == "__main__":
+    console_main()

scripts/gather_tiles.py

@@ -99,8 +99,7 @@ def gather_tiles(folder, tile_base, L, Np, N_TILES, buffer):
 
 
 
-
+def console_main():
-if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Gather density from tiles.")
     parser.add_argument("-d","--folder", type=str, default="./", help="Folder containing the tiles")
     parser.add_argument("--tile_base", type=str, default="sCOLA_tile", help="Base name of the tiles")
@@ -122,3 +121,8 @@ if __name__ == "__main__":
     dpm = L/Np_tile
 
     gather_tiles(folder, tile_base, L, Np, N_TILES, buffer)
+
+
+
+if __name__ == "__main__":
+    console_main()

scripts/scola_submit.py

@@ -462,9 +462,7 @@ def scola_submit(directory,
 
 
 
-
+def console_main():
-
-if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Submit slurm jobs for sCOLA tiles.")
 
     parser.add_argument("-d", "--directory", type=str, default="./", help="Main directory where the output will be saved (if other dir and filenames are not specified).")
@@ -496,4 +494,9 @@ if __name__ == "__main__":
 
 
 
+if __name__ == "__main__":
+    console_main()
+
+
+