Skip to content

Commit fe6ea40

Browse files
committed
same detector cuts for sims as for data
1 parent 4095579 commit fe6ea40

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

sotodlib/preprocess/preprocess_util.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -539,6 +539,16 @@ def multilayer_load_and_preprocess_sim(obs_id, configs_init, configs_proc,
539539

540540
if check_cfg_match(aman_cfgs_ref, meta_proc.preprocess['pcfg_ref'],
541541
logger=logger):
542+
pipe_proc = Pipeline(configs_proc["process_pipe"], logger=logger)
543+
544+
logger.info("Restricting detectors on all proc pipeline processes")
545+
keep_all = np.ones(meta_proc.dets.count, dtype=bool)
546+
for process in pipe_proc[:]:
547+
keep = process.select(meta_proc, in_place=False)
548+
if isinstance(keep, np.ndarray):
549+
keep_all &= keep
550+
meta_proc.restrict("dets", meta_proc.dets.vals[keep_all])
551+
meta_init.restrict('dets', meta_proc.dets.vals)
542552
aman = context_init.get_obs(meta_proc, no_signal=True)
543553
aman = hwp_angle_model.apply_hwp_angle_model(aman)
544554
aman.move("signal", None)
@@ -567,7 +577,6 @@ def multilayer_load_and_preprocess_sim(obs_id, configs_init, configs_proc,
567577
T_signal=t2ptemplate_aman.dsT
568578
)
569579

570-
pipe_proc = Pipeline(configs_proc["process_pipe"], logger=logger)
571580
logger.info("Running dependent pipeline")
572581
proc_aman = context_proc.get_meta(obs_id, meta=aman)
573582
aman.preprocess.merge(proc_aman.preprocess)

0 commit comments

Comments
 (0)