diff options
Diffstat (limited to 'fg21sim/galactic/synchrotron.py')
| -rw-r--r-- | fg21sim/galactic/synchrotron.py | 19 |
1 files changed, 16 insertions, 3 deletions
diff --git a/fg21sim/galactic/synchrotron.py b/fg21sim/galactic/synchrotron.py index f6b1db8..96f4df1 100644 --- a/fg21sim/galactic/synchrotron.py +++ b/fg21sim/galactic/synchrotron.py @@ -66,6 +66,7 @@ class Synchrotron: self.use_float = self.configs.getn("output/use_float") self.checksum = self.configs.getn("output/checksum") self.clobber = self.configs.getn("output/clobber") + self.frequencies = self.configs.frequencies # [MHz] self.freq_unit = au.Unit(self.configs.getn("frequency/unit")) # logger.info("Loaded and setup configurations") @@ -230,10 +231,17 @@ class Synchrotron: filepath = None return (skymap_f, filepath) - def simulate(self, frequencies): + def simulate(self, frequencies=None): """ Simulate the synchrotron map at the specified frequencies. + Parameters + ---------- + frequencies : float, or list[float] + The frequencies where to simulate the foreground map. + Unit: [MHz] + Default: None (i.e., use ``self.frequencies``) + Returns ------- skymaps : list[1D `~numpy.ndarray`] @@ -241,10 +249,15 @@ class Synchrotron: paths : list[str] List of (absolute) path to the output sky maps. """ + if frequencies is not None: + frequencies = np.array(frequencies, ndmin=1) + else: + frequencies = self.frequencies + skymaps = [] paths = [] - for f in np.array(frequencies, ndmin=1): - skymap_f, outfile = self.simulate_frequency(f) + for freq in frequencies: + skymap_f, outfile = self.simulate_frequency(freq) skymaps.append(skymap_f) paths.append(outfile) return (skymaps, paths) |