diff options
| author | Aaron LI <aly@aaronly.me> | 2017-10-19 13:14:14 +0800 |
|---|---|---|
| committer | Aaron LI <aly@aaronly.me> | 2017-10-19 13:14:14 +0800 |
| commit | ad4a43dce176c3f087f5dc5af188dfc081934027 (patch) | |
| tree | 6137d40bf12053d304079d2d7e3ee5ee04793775 /fg21sim/extragalactic/clusters | |
| parent | eac586b473c31c2ca607a3cf9c27109b87fc8043 (diff) | |
| download | fg21sim-ad4a43dce176c3f087f5dc5af188dfc081934027.tar.bz2 | |
clusters/halo: Support use the previously dumped halos raw data
Add option "use_dump_halos_data" to control whether to just use the previously
dumped halos data, therefore, the radio emissions at additional frequencies
can be simply calculated.
Diffstat (limited to 'fg21sim/extragalactic/clusters')
| -rw-r--r-- | fg21sim/extragalactic/clusters/main.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/fg21sim/extragalactic/clusters/main.py b/fg21sim/extragalactic/clusters/main.py index 06ee9f6..e838094 100644 --- a/fg21sim/extragalactic/clusters/main.py +++ b/fg21sim/extragalactic/clusters/main.py @@ -81,6 +81,8 @@ class GalaxyClusters: self.halos_data_dumpfile = os.path.splitext( self.halos_catalog_outfile)[0] + ".pkl" self.dump_halos_data = self.configs.getn(comp+"/dump_halos_data") + self.use_dump_halos_data = self.configs.getn( + comp+"/use_dump_halos_data") self.prefix = self.configs.getn(comp+"/prefix") self.output_dir = self.configs.get_path(comp+"/output_dir") self.merger_mass_min = self.configs.getn(comp+"/merger_mass_min") @@ -409,7 +411,15 @@ class GalaxyClusters: self._process_catalog() self._simulate_mergers() - self._simulate_halos() + if self.use_dump_halos_data: + logger.info("Use existing dumped halos raw data: %s" % + self.halos_data_dumpfile) + self.halos = pickle_load(self.halos_data_dumpfile) + logger.info("Loaded data of %d halos" % len(self.halos)) + else: + self._simulate_halos() + + self._calc_halos_emission() self._draw_halos() self._preprocessed = True |