aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAaron LI <aaronly.me@outlook.com>2016-11-24 10:46:18 +0800
committerAaron LI <aaronly.me@outlook.com>2016-11-24 10:46:18 +0800
commit2eb0f559b424fa4a2bbf96d2a563435d0c78732b (patch)
treee3bda82966a067f3cfc33465296852457a0cabdc
parentfa1d1e5ac7a101068016296c00baa55a420f7d40 (diff)
downloadfg21sim-2eb0f559b424fa4a2bbf96d2a563435d0c78732b.tar.bz2
products.py: Add method "get_root_dir()" and use it
-rw-r--r--fg21sim/products.py41
1 files changed, 32 insertions, 9 deletions
diff --git a/fg21sim/products.py b/fg21sim/products.py
index 6b83c8b..992b2af 100644
--- a/fg21sim/products.py
+++ b/fg21sim/products.py
@@ -29,6 +29,8 @@ class Products:
manifest : dict
The manifest of the simulation products.
See the below "Manifest Format" section for more details.
+ manifestfile : str
+ The absolute path to the loaded manifest file.
Manifest Format
---------------
@@ -164,12 +166,12 @@ class Products:
# Initialize the manifest array for this component
self.manifest[comp_id] = [{} for i in range(len(frequencies))]
#
- curdir = os.path.dirname(self.manifestfile)
+ root_dir = self.get_root_dir()
self.manifest[comp_id][freq_id] = {
"frequency": frequencies[freq_id],
"healpix": {
# Relative path to the HEALPix map file from this manifest
- "path": os.path.relpath(filepath, curdir),
+ "path": os.path.relpath(filepath, root_dir),
# File size in bytes
"size": os.path.getsize(filepath),
"md5": md5(filepath),
@@ -230,9 +232,9 @@ class Products:
hash : str
The MD5 checksum value of the on-disk product.
"""
- curdir = os.path.dirname(self.manifestfile)
+ root_dir = self.get_root_dir()
metadata = self.get_product(comp_id, freq_id)
- filepath = os.path.join(curdir, metadata["healpix"]["path"])
+ filepath = os.path.join(root_dir, metadata["healpix"]["path"])
hash_true = metadata["healpix"]["md5"]
hash_ondisk = md5(filepath)
if hash_ondisk == hash_true:
@@ -241,6 +243,27 @@ class Products:
match = False
return (match, hash_ondisk)
+ def get_root_dir(self):
+ """
+ Get the root directory where the products locate, which is also
+ the directory where the manifest file locates.
+
+ Returns
+ -------
+ root_dir : str
+ The absolute path of the products root directory.
+
+ Raises
+ ------
+ ManifestError :
+ The manifest currently not loaded, thus unable to determine
+ the products root directory.
+ """
+ try:
+ return os.path.dirname(self.manifestfile)
+ except AttributeError:
+ raise ManifestError("Manifest currently not loaded!")
+
def get_product(self, comp_id, freq_id):
return self.manifest[comp_id][freq_id]
@@ -273,9 +296,9 @@ class Products:
"""
if ptype not in ["healpix", "hpx"]:
raise ValueError("Invalid ptype: {0}".format(ptype))
- curdir = os.path.dirname(self.manifestfile)
+ root_dir = self.get_root_dir()
metadata = self.get_product(comp_id, freq_id)
- abspath = os.path.join(curdir, metadata[ptype]["path"])
+ abspath = os.path.join(root_dir, metadata[ptype]["path"])
return abspath
def convert_hpx(self, comp_id, freq_id, clobber=False):
@@ -291,9 +314,9 @@ class Products:
from astropy.io import fits
from .utils.healpix import healpix2hpx
#
- curdir = os.path.dirname(self.manifestfile)
+ root_dir = self.get_root_dir()
metadata = self.get_product(comp_id, freq_id)
- infile = os.path.join(curdir, metadata["healpix"]["path"])
+ infile = os.path.join(root_dir, metadata["healpix"]["path"])
outfile = os.path.splitext(infile)[0] + "_hpx.fits"
if os.path.exists(outfile):
if clobber:
@@ -311,7 +334,7 @@ class Products:
size = os.path.getsize(outfile)
md5sum = md5(outfile)
metadata["hpx"] = {
- "path": os.path.relpath(outfile, curdir),
+ "path": os.path.relpath(outfile, root_dir),
"size": size,
"md5": md5sum,
}