Implemented airglow correction routines

ladsantos · May 4, 2018 · 2b131fe · 2b131fe
1 parent 3e1c512
commit 2b131fe
Show file tree

Hide file tree

Showing 3 changed files with 466 additions and 55 deletions.
diff --git a/sunburn/analysis.py b/sunburn/analysis.py
@@ -149,12 +149,14 @@ class LightCurve(object):
 
         line_list (``dict``, optional): Spectral line list.
     """
-    def __init__(self, visit, transit=None, line_list=None):
+    def __init__(self, visit, transit=None, line_list=None,
+                 transit_search_expansion=(0.0 * u.d, 0.0 * u.d)):
         self.visit = visit
         self.transit = transit
         self.line_list = line_list
 
         # Instantiating useful global variables
+        self._expand = transit_search_expansion
         self.integrated_flux = None
         self.time = []
         self.t_span = []
@@ -196,7 +198,8 @@ def __init__(self, visit, transit=None, line_list=None):
         # Figure out the range of julian dates spanned by the visit
         jd_start = np.array(jd_start)
         jd_end = np.array(jd_end)
-        self.jd_range = (np.min(jd_start), np.max(jd_end))
+        self.jd_range = (np.min(jd_start) - self._expand[0],
+                         np.max(jd_end) + self._expand[1])
 
         # Transform lists into numpy arrays
         self.time = np.array(self.time)
@@ -578,3 +581,12 @@ def plot(self, norm=1, figure_sizes=(9.0, 6.5), axes_font_size=18,
         else:
             plt.ylabel(r'Normalized integrated flux')
         plt.xlabel('Time (h)')
+
+
+# Light curve object for a specific line
+class LineLightCurve(object):
+    """
+
+    """
+    def __init__(self, spectral_lin):
+        pass