Suspension thermal noise sub-budgets

Adds suspension thermal noise sub-budgets. This addresses 1/3 of #51 (closed). Suspension thermal noise can now be broken up into horizontal and vertical noise of each of the suspension stages. With one exception, which does not change the total noise, the calculation is the same as the current calculation but the contributions of each stage and direction are tracked separately.

In the current calculation the horizontal and vertical spring constants of each stage are broken up into an upper and lower joint and are stored separately. The total suspension thermal noise is then calculated by adding the contributions of each of these joints using the appropriate temperature of each joint. Instead of adding the contributions of each joint all at once, this MR makes it possible to track each stage with separate Noise classes and add them all to a sub-budget.

The noise for each stage is the sum of the noise from that stage's lower joint and the noise from the upper joint of the stage below. In the case of the top mass, the contribution from the upper joint is also included. In the current calculation, for the upper stages, the imaginary part of the horizontal spring constant is equally distributed between the upper and lower joints while it is only attributed to the upper joint for the test mass. The one change this MR makes to the calculation itself is to distribute this equally between the top and bottom joints as is done for the upper stages.

The imaginary part of the vertical spring constant is only attributed to the upper joint of each stage. The blade springs are the only contribution to the vertical spring constant for the upper stages while the lower stage also includes the continuum fibers or ribbons. It's unclear how to break this up into an upper and lower stage as is done in figure 14 of https://arxiv.org/abs/2001.11173, so this MR does not have vertical noise of the test mass. (The noise is included, it's just attributed entirely to the PUM.)

The sub-budgets are only implemented for the Cosmic Explorer interferometers since these are the only ones that have any other sub-budgets. The CE1 suspension thermal noise budget is shown below as an example.

gwinc/__main__.py

@@ -6,6 +6,7 @@ import argparse
 import numpy as np
 
 from . import IFOS, load_budget, plot_budget, logger
+from . import io
 
 logger.setLevel(os.getenv('LOG_LEVEL', 'WARNING').upper())
 formatter = logging.Formatter('%(message)s')
@@ -53,7 +54,6 @@ See the inspiral_range package documentation for details.
 IFO = 'aLIGO'
 FREQ = '5:3000:6000'
 FOM = 'range:m1=1.4,m2=1.4'
-DATA_SAVE_FORMATS = ['.hdf5', '.h5']
 
 parser = argparse.ArgumentParser(
     prog='gwinc',
@@ -116,7 +116,7 @@ def main():
     ##########
     # initial arg processing
 
-    if os.path.splitext(os.path.basename(args.IFO))[1] in DATA_SAVE_FORMATS:
+    if os.path.splitext(os.path.basename(args.IFO))[1] in io.DATA_SAVE_FORMATS:
         if args.freq:
             parser.exit(2, "Frequency specification not allowed when loading traces from file.\n")
         if args.ifo:
@@ -178,7 +178,7 @@ def main():
     if args.save:
         args.plot = False
         for path in args.save:
-            if os.path.splitext(path)[1] in DATA_SAVE_FORMATS:
+            if os.path.splitext(path)[1] in io.DATA_SAVE_FORMATS:
                 out_data_files.add(path)
         out_plot_files = set(args.save) - out_data_files
 
@@ -303,12 +303,11 @@ In [.]: plt.savefig("foo.pdf")
 
     # save noise trace to HDF5 file
     if out_data_files:
-        from .io import save_hdf5
         for path in out_data_files:
             logger.info("saving budget trace: {}".format(path))
-            save_hdf5(
-                path=path,
+            io.save_hdf5(
                 trace=trace,
+                path=path,
                 ifo=ifo,
                 plot_style=plot_style,
             )