Skip to content

Commit fbda3f5

Browse files
committed
fix name and read files with h5py
1 parent f0c7d2a commit fbda3f5

2 files changed

Lines changed: 12 additions & 6 deletions

File tree

project/SO/pISO/python/get_spectra_from_maps.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -778,8 +778,8 @@
778778
ps_dict_all['l'] = lb
779779
io.save_hdf5(f"{spec_dir}" + f"{spec_name_all}.h5", ps_dict_all)
780780
else:
781-
spec_name_all = f"{type}{tag}_all_s_cross_filter_{iii:05d}"
782-
spec_name_all_nofilt = f"{type}{tag}_all_s_cross_nofilter_{iii:05d}"
781+
spec_name_all = f"{type}{tag}_all_s_filter_{iii:05d}"
782+
spec_name_all_nofilt = f"{type}{tag}_all_s_nofilter_{iii:05d}"
783783

784784
# each process has separate maps in its mapset
785785
ps_dict_all['l'] = lb

project/SO/pISO/python/kspace/mc_kspace_tf_analysis.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import matplotlib
77
matplotlib.use("Agg")
88
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
1010
import numpy as np
1111
import pylab as plt
1212
import sys
@@ -54,6 +54,11 @@
5454
for sid, spec in enumerate(spec_list):
5555
log.info(f"Read all {spec} sim power spectra")
5656

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+
5762
ps_list[spec] = {}
5863
for scenario in scenarios:
5964
for iii in range(iStart, iStop + 1):
@@ -62,9 +67,10 @@
6267
ps_list[spec]["nofilter", scenario] = []
6368
ps_list[spec]["filter", scenario] = []
6469

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+
6874
ps_list[spec]["nofilter", scenario] += [ps_nofilt]
6975
ps_list[spec]["filter", scenario] += [ps_filt]
7076

0 commit comments

Comments
 (0)