|
6 | 6 | import matplotlib |
7 | 7 | matplotlib.use("Agg") |
8 | 8 | from pspy import pspy_utils, so_dict, so_spectra, so_mcm |
9 | | -from pspipe_utils import pspipe_list, best_fits, kspace, log |
| 9 | +from pspipe_utils import pspipe_list, best_fits, kspace, log, io |
10 | 10 | import numpy as np |
11 | 11 | import pylab as plt |
12 | 12 | import sys |
|
54 | 54 | for sid, spec in enumerate(spec_list): |
55 | 55 | log.info(f"Read all {spec} sim power spectra") |
56 | 56 |
|
| 57 | + # breaking down the strings in spec |
| 58 | + lspec = spec.split("x") |
| 59 | + sv1, m1, = lspec[0].split("_") |
| 60 | + sv2, m2, = lspec[1].split("_") |
| 61 | + |
57 | 62 | ps_list[spec] = {} |
58 | 63 | for scenario in scenarios: |
59 | 64 | for iii in range(iStart, iStop + 1): |
|
62 | 67 | ps_list[spec]["nofilter", scenario] = [] |
63 | 68 | ps_list[spec]["filter", scenario] = [] |
64 | 69 |
|
65 | | - lb, ps_nofilt = so_spectra.read_ps(spec_dir + f"/{type}_{spec}_nofilter_{scenario}_{iii:05d}.dat", spectra=spectra) |
66 | | - lb, ps_filt = so_spectra.read_ps(spec_dir + f"/{type}_{spec}_filter_{scenario}_{iii:05d}.dat", spectra=spectra) |
67 | | - |
| 70 | + lb = io.load_hdf5(spec_dir + f"{type}_for_kspace_all_s_filter_{iii:05d}", path="/l") |
| 71 | + ps_filt = io.load_hdf5(spec_dir + f"{type}_for_kspace_all_s_filter_{iii:05d}", path=f"/({sv1}, {m1}), ({sv2}, {m2}), so_{scenario}") |
| 72 | + ps_nofilt = io.load_hdf5(spec_dir + f"{type}_for_kspace_all_s_nofilter_{iii:05d}", path=f"/({sv1}, {m1}), ({sv2}, {m2}), so_{scenario}") |
| 73 | + |
68 | 74 | ps_list[spec]["nofilter", scenario] += [ps_nofilt] |
69 | 75 | ps_list[spec]["filter", scenario] += [ps_filt] |
70 | 76 |
|
|
0 commit comments