Changeset 18293 for branches/trigger_burst_research

Timestamp:

08/25/15 12:56:28 (9 years ago)

Author:

dneise

Message:

debugging, special treatment of old aux-files (continue), and progressbar

File:

• branches/trigger_burst_research/sketch.py (modified) (7 diffs)

branches/trigger_burst_research/sketch.py

@@ -1 +1 @@
 # coding: utf-8
+import progressbar
 import calendar
 import numpy as np
@@ -32 +33 @@
 
 
-def get_data_runs_from_run_db(night_int, factdb):
+def get_all_data_runs_from_run_db(factdb):
     keys = [
         'fRunID',
@@ -41 +42 @@
 
     sql_query = """SELECT {comma_sep_keys}
-        FROM RunInfo WHERE fRunTypeKey=1 AND fNight={night:d};
+        FROM RunInfo WHERE fRunTypeKey=1 ORDER BY fNight;
         """
     sql_query = sql_query.format(
         comma_sep_keys=', '.join(keys),
-        night=night_int,
         )
     
@@ -57 +57 @@
 
 
+
 def get_trigger_rates(night_int, base_path='/fact/aux/'):
     night_string = str(night_int)
     fits_file = fits.open(
         base_path+'{y}/{m}/{d}/{n}.FTM_CONTROL_TRIGGER_RATES.fits'.format(
-            ni=night_string,
+            n=night_string,
             y=night_string[0:4],
             m=night_string[4:6],
@@ -71 +72 @@
 
 
-def add_ratio_and_more_to_dataframe(data_runs, trigger_rates):
+def add_ratio_and_more_to_dataframe(data_runs):
 
     # create new columns and pre-assign some hopefully good NULL-like-values
@@ -79 +80 @@
     data_runs['fBoardTriggerRateRatioAboveThreshold'] = np.nan
 
-    for dataframe_index in data_runs.index:
+    trigger_rates = None
+    last_night = None
+
+    progress = progressbar.ProgressBar(widgets=[progressbar.Bar('=', '[', ']'), ' ',
+                                            progressbar.Percentage(), ' ',
+                                            progressbar.ETA()])
+
+    for dataframe_index in progress(data_runs.index):
         this_run = data_runs.ix[dataframe_index]
+        if last_night != this_run['fNight']:
+            try:
+                trigger_rates = get_trigger_rates(this_run['fNight'])
+            except (IOError, ValueError):
+                continue
+            last_night = this_run['fNight']
+
         mask = (
             (trigger_rates['Time'] > fjd(this_run['fRunStart'])) *
             (trigger_rates['Time'] < fjd(this_run['fRunStop']))
             )
-        this_board_rates = trigger_rates['BoardRate'][mask]
+        
+        if mask.sum() == 0:
+            continue
+        try:
+            this_board_rates = trigger_rates['BoardRate'][mask]
+        except KeyError:
+            continue
 
         N = this_board_rates.shape[0]
@@ -109 +130 @@
 def main(night_int):
     factdb = create_DB_connection()
-    data_runs = get_data_runs_from_run_db(night_int, factdb)
-    trigger_rates = get_trigger_rates(night_int)
-    add_ratio_and_more_to_dataframe(data_runs, trigger_rates)
+    data_runs = get_all_data_runs_from_run_db(factdb)
+    add_ratio_and_more_to_dataframe(data_runs)
     return data_runs