Make "frequencies" optional in the simulate() method

XXX: foregrounds.py needs update!

Signed-off-by: Aaron LI <aly@aaronly.me>

1 files changed, 16 insertions, 3 deletions

diff --git a/fg21sim/galactic/freefree.py b/fg21sim/galactic/freefree.py
index 9c9424f..a1fcf6d 100644
--- a/fg21sim/galactic/freefree.py
+++ b/fg21sim/galactic/freefree.py
@@ -86,6 +86,7 @@ class FreeFree:
         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 set up configurations")
@@ -292,10 +293,17 @@ class FreeFree:
             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`]
@@ -303,10 +311,15 @@ class FreeFree:
         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)