aboutsummaryrefslogtreecommitdiffstats
path: root/fg21sim/configs
diff options
context:
space:
mode:
authorAaron LI <aly@aaronly.me>2018-01-02 13:34:34 +0800
committerAaron LI <aly@aaronly.me>2018-01-02 13:34:34 +0800
commit4a49490eb32398255dee3ee252e87aef626fb5d7 (patch)
tree6d4f28c146a678f099bd6a6ac217e18111e5a5a4 /fg21sim/configs
parent91b880f0cf0cc33b8e591d731cd5c20e96b13c5d (diff)
downloadfg21sim-4a49490eb32398255dee3ee252e87aef626fb5d7.tar.bz2
configs/checkers: add _check_file_existence() and some cleanups
Diffstat (limited to 'fg21sim/configs')
-rw-r--r--fg21sim/configs/checkers.py63
1 files changed, 49 insertions, 14 deletions
diff --git a/fg21sim/configs/checkers.py b/fg21sim/configs/checkers.py
index 2d23706..78d5b20 100644
--- a/fg21sim/configs/checkers.py
+++ b/fg21sim/configs/checkers.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2016-2017 Weitian LI <weitian@aaronly.me>
+# Copyright (c) 2016-2018 Weitian LI <weitian@aaronly.me>
# MIT license
"""
@@ -18,7 +18,9 @@ from ..errors import ConfigError
def _check_missing(configs, keys):
- """Check whether the required config is provided by the user."""
+ """
+ Check whether the required config is provided by the user.
+ """
results = {}
if isinstance(keys, str):
keys = [keys, ]
@@ -29,7 +31,9 @@ def _check_missing(configs, keys):
def _check_existence(configs, keys):
- """Check whether the file/directory corresponding to the config exists."""
+ """
+ Check whether the file/directory corresponding to the config exists.
+ """
if isinstance(keys, str):
keys = [keys, ]
results = {}
@@ -44,14 +48,37 @@ def _check_existence(configs, keys):
return results
+def _check_file_existence(configs, key, ext=None):
+ """
+ Check the existence of the file corresponding to the given config key.
+
+ If the ``ext`` is specified, then the original file extension is
+ first replaced with the given one, then to check its existence.
+ """
+ res = _check_missing(configs, key)
+ if res == {}:
+ filepath = configs.get_path(key)
+ if ext:
+ filepath = os.path.splitext(filepath)[0] + ext
+ key += "#%s" % ext
+ if not os.path.exists(filepath):
+ res[key] = 'File not exist: "%s"' % filepath
+ return res
+
+
def _is_power2(n):
- """Check a number whether a power of 2"""
- # Credit: https://stackoverflow.com/a/29489026/4856091
+ """
+ Check a number whether a power of 2
+
+ Credit: https://stackoverflow.com/a/29489026/4856091
+ """
return (n and not (n & (n-1)))
def check_foregrounds(configs):
- """Check the "[foregrounds]" section of the configurations."""
+ """
+ Check the "[foregrounds]" section of the configurations.
+ """
results = {}
# Check enabled foreground components
fg = configs.foregrounds
@@ -61,7 +88,9 @@ def check_foregrounds(configs):
def check_sky(configs):
- """Check the "[sky]" section of the configurations."""
+ """
+ Check the "[sky]" section of the configurations.
+ """
results = {}
skytype = configs.getn("sky/type")
if skytype == "patch":
@@ -80,7 +109,9 @@ def check_sky(configs):
def check_frequency(configs):
- """Check the "[frequency]" section of the configurations."""
+ """
+ Check the "[frequency]" section of the configurations.
+ """
results = {}
if configs.getn("frequency/type") == "custom":
results.update(_check_missing(configs, "frequency/frequencies"))
@@ -94,7 +125,9 @@ def check_frequency(configs):
def check_galactic_synchrotron(configs):
- """Check the "[galactic][synchrotron]" section of the configurations."""
+ """
+ Check the "[galactic][synchrotron]" section of the configurations.
+ """
comp = "galactic/synchrotron"
comp_enabled = configs.foregrounds[0]
if comp not in comp_enabled:
@@ -136,7 +169,9 @@ def check_galactic_freefree(configs):
def check_galactic_snr(configs):
- """Check the "[galactic][snr]" section of the configurations."""
+ """
+ Check the "[galactic][snr]" section of the configurations.
+ """
comp = "galactic/snr"
comp_enabled = configs.foregrounds[0]
results = {}
@@ -168,11 +203,11 @@ def check_extragalactic_clusters(configs):
results.update(_check_missing(configs, comp+"/catalog_outfile"))
# dumped halos data required when enabled to use it
if configs.get(comp+"/use_dump_halos_data"):
- results.update(_check_existence(configs,
- comp+"/halos_catalog_outfile"))
+ results.update(_check_file_existence(
+ configs, comp+"/halos_catalog_outfile", ext=".pkl"))
else:
- results.update(_check_missing(configs,
- comp+"/halos_catalog_outfile"))
+ results.update(_check_missing(
+ configs, comp+"/halos_catalog_outfile"))
results.update(_check_missing(configs, comp+"/output_dir"))
return results