From 6391211e6fa149aaa99379b33a8afdbc1fcb5d8d Mon Sep 17 00:00:00 2001
From: keskitalo <reijo.keskitalo@gmail.com>
Date: Fri, 5 Jan 2024 14:07:10 -0800
Subject: [PATCH] Add --sample_rate and --fsample to toast_sim_satellite.py
---
src/toast/instrument.py | 2 +-
workflows/toast_sim_satellite.py | 24 +++++++++++++++++++++++-
2 files changed, 24 insertions(+), 2 deletions(-)
diff --git a/src/toast/instrument.py b/src/toast/instrument.py
index e507f9f0f..ac8311238 100644
--- a/src/toast/instrument.py
+++ b/src/toast/instrument.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2019-2023 by the parties listed in the AUTHORS file.
+# Copyright (c) 2019-2024 by the parties listed in the AUTHORS file.
# All rights reserved. Use of this source code is governed by
# a BSD-style license that can be found in the LICENSE file.
diff --git a/workflows/toast_sim_satellite.py b/workflows/toast_sim_satellite.py
index 424206aaa..59119754e 100644
--- a/workflows/toast_sim_satellite.py
+++ b/workflows/toast_sim_satellite.py
@@ -66,6 +66,20 @@ def parse_config(operators, templates, comm):
help="The output directory",
)
+ parser.add_argument(
+ "--sample_rate",
+ required=False,
+ type=float,
+ help="Override focalplane sampling rate [Hz]",
+ )
+
+ parser.add_argument(
+ "--thinfp",
+ required=False,
+ type=int,
+ help="Only sample the provided focalplane pixels",
+ )
+
# Build a config dictionary starting from the operator defaults, overriding with any
# config files specified with the '--config' commandline option, followed by any
# individually specified parameter overrides.
@@ -96,7 +110,15 @@ def load_instrument_and_schedule(args, comm):
timer = toast.timing.Timer()
timer.start()
- focalplane = toast.instrument.Focalplane()
+ if args.sample_rate is not None:
+ sample_rate = args.sample_rate * u.Hz
+ else:
+ sample_rate = None
+ focalplane = toast.instrument.Focalplane(
+ sample_rate=sample_rate,
+ thinfp=args.thinfp,
+ )
+
with toast.io.H5File(args.focalplane, "r", comm=comm, force_serial=True) as f:
focalplane.load_hdf5(f.handle, comm=comm)
log.info_rank("Loaded focalplane in", comm=comm, timer=timer)