Started improving the structure of printed output, added print_sectio…

…n function in core_util, print multiprocessing status in SpeciesInit, fixed issue with leg_param in plot_spectrum, added verbose parameter to get_median_sample, get_probably_sample, and get_compare_sample

docs/species.util.rst

@@ -20,6 +20,14 @@ species.util.convert\_util module
    :undoc-members:
    :show-inheritance:
 
+species.util.core\_util module
+------------------------------
+
+.. automodule:: species.util.core_util
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
 species.util.data\_util module
 ------------------------------
 

species/core/species_init.py

@@ -8,6 +8,7 @@
 import urllib.request
 
 from configparser import ConfigParser
+from importlib.util import find_spec
 from typeguard import typechecked
 
 import h5py
@@ -124,7 +125,7 @@ def __init__(self) -> None:
                 file_obj.write("\n; Magnitude of Vega for all filters\n")
                 file_obj.write("vega_mag = 0.03\n")
 
-        print("Configuration settings:")
+        print("\nConfiguration settings:")
         print(f"   - Database: {database_file}")
         print(f"   - Data folder: {data_folder}")
         print(f"   - Interpolation method: {interp_method}")
@@ -141,12 +142,17 @@ def __init__(self) -> None:
             os.makedirs(data_folder)
             print(" [DONE]")
 
-        # warnings.warn(
-        #     "Importing the species package had become slow "
-        #     "because of the many classes and functions that "
-        #     "were implicitly imported. The initialization of "
-        #     "the packages has therefore been adjusted. In "
-        #     "the latest version, any functionalities should "
-        #     "be explicitly imported from the modules that "
-        #     "they are part of."
-        # )
+        if find_spec("mpi4py") is None:
+            print("\nMultiprocessing: mpi4py not installed")
+
+        else:
+            from mpi4py import MPI
+
+            # Rank of this process in a communicator
+            mpi_rank = MPI.COMM_WORLD.Get_rank()
+
+            # Number of processes in a communicator
+            mpi_size = MPI.COMM_WORLD.Get_size()
+
+            print("\nMultiprocessing: mpi4py installed")
+            print(f"Process number {mpi_rank+1:d} out of {mpi_size:d}...")

species/data/companion_data/companion_spectra.py

@@ -11,6 +11,8 @@
 
 from typeguard import typechecked
 
+from species.util.core_util import print_section
+
 
 @typechecked
 def companion_spectra(
@@ -48,6 +50,8 @@ def companion_spectra(
         comp_spec = json.load(json_file)
 
     if comp_name in comp_spec:
+        print_section("Get companion spectra")
+
         data_folder = os.path.join(input_path, "companion_data/")
 
         if not os.path.exists(data_folder):