From ad4a43dce176c3f087f5dc5af188dfc081934027 Mon Sep 17 00:00:00 2001 From: Aaron LI Date: Thu, 19 Oct 2017 13:14:14 +0800 Subject: 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. --- fg21sim/extragalactic/clusters/main.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'fg21sim/extragalactic/clusters') 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 -- cgit v1.2.2