Added the mean_wavel attribute to ObjectBox

tomasstolker · Jun 28, 2021 · 4aa085b · 4aa085b
1 parent 1dada33
commit 4aa085b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/species/core/box.py b/species/core/box.py
@@ -73,6 +73,7 @@ def create_box(boxtype,
         box = ObjectBox()
         box.name = kwargs['name']
         box.filters = kwargs['filters']
+        box.mean_wavel = kwargs['mean_wavel']
         box.magnitude = kwargs['magnitude']
         box.flux = kwargs['flux']
         box.distance = kwargs['distance']
@@ -317,6 +318,7 @@ def __init__(self):
 
         self.name = None
         self.filters = None
+        self.mean_wavel = None
         self.magnitude = None
         self.flux = None
         self.distance = None

diff --git a/species/data/database.py b/species/data/database.py
@@ -1782,6 +1782,7 @@ def get_object(self,
 
             magnitude = {}
             flux = {}
+            mean_wavel = {}
 
             for observatory in dset.keys():
                 if observatory not in ['distance', 'spectrum']:
@@ -1792,13 +1793,17 @@ def get_object(self,
                             magnitude[name] = dset[name][0:2]
                             flux[name] = dset[name][2:4]
 
+                            filter_trans = read_filter.ReadFilter(name)
+                            mean_wavel[name] = filter_trans.mean_wavelength()
+
             phot_filters = list(magnitude.keys())
 
         else:
 
             magnitude = None
             flux = None
             phot_filters = None
+            mean_wavel = None
 
         if inc_spec and f'objects/{object_name}/spectrum' in h5_file:
             spectrum = {}
@@ -1830,6 +1835,7 @@ def get_object(self,
         return box.create_box('object',
                               name=object_name,
                               filters=phot_filters,
+                              mean_wavel=mean_wavel,
                               magnitude=magnitude,
                               flux=flux,
                               distance=distance,