diff options
Diffstat (limited to 'fg21sim/galactic/snr.py')
| -rw-r--r-- | fg21sim/galactic/snr.py | 25 |
1 files changed, 20 insertions, 5 deletions
diff --git a/fg21sim/galactic/snr.py b/fg21sim/galactic/snr.py index b7bfadc..ed66649 100644 --- a/fg21sim/galactic/snr.py +++ b/fg21sim/galactic/snr.py @@ -345,6 +345,11 @@ class SuperNovaRemnants: def output(self, hpmap, frequency): """Write the simulated free-free map to disk with proper header keywords and history. + + Returns + ------- + filepath : str + The (absolute) path to the output HEALPix map file. """ if not os.path.exists(self.output_dir): os.mkdir(self.output_dir) @@ -364,6 +369,7 @@ class SuperNovaRemnants: write_fits_healpix(filepath, hpmap, header=header, clobber=self.clobber, checksum=self.checksum) logger.info("Write simulated map to file: {0}".format(filepath)) + return filepath def preprocess(self): """Perform the preparation procedures for the final simulations. @@ -399,7 +405,10 @@ class SuperNovaRemnants: Returns ------- hpmap_f : 1D `~numpy.ndarray` - HEALPix map data in RING ordering + The HEALPix map (RING ordering) at the input frequency. + filepath : str + The (absolute) path to the output HEALPix file if saved, + otherwise ``None``. See Also -------- @@ -415,8 +424,10 @@ class SuperNovaRemnants: hpmap_f[hpidx] += hpval # if self.save: - self.output(hpmap_f, frequency) - return hpmap_f + filepath = self.output(hpmap_f, frequency) + else: + filepath = None + return (hpmap_f, filepath) def simulate(self, frequencies): """Simulate the emission (HEALPix) maps of all Galactic SNRs for @@ -432,12 +443,16 @@ class SuperNovaRemnants: ------- hpmaps : list[1D `~numpy.ndarray`] List of HEALPix maps (in RING ordering) at each frequency. + paths : list[str] + List of (absolute) path to the output HEALPix maps. """ hpmaps = [] + paths = [] for f in np.array(frequencies, ndmin=1): - hpmap_f = self.simulate_frequency(f) + hpmap_f, filepath = self.simulate_frequency(f) hpmaps.append(hpmap_f) - return hpmaps + paths.append(filepath) + return (hpmaps, paths) def postprocess(self): """Perform the post-simulation operations before the end.""" |