import sys

import cdb

def get_beamline(run_number):

    bl = cdb.Beamline()

    beamline_all = bl.get_beamline_for_run(run_number)[run_number]

    return beamline_all

def get_tof_triggers_sum(tof_key, run_list):

    trig_list = []

    for run_number in run_list:

        try:

            trig_list.append(get_beamline(run_number)["scalars"][tof_key])

        except:

            sys.excepthook(*sys.exc_info())

            print "Failed to find key", tof_key, "for run number", run_number

            print get_beamline(run_list[0])["scalars"].keys()

    return sum(trig_list)

def get_time_sum(run_list):

    times = [] # time of each run in seconds

    for run_number in run_list: 

        start = get_beamline(run_number)["start_time"]     

        end = get_beamline(run_number)["end_time"]

        if start == None or end == None:

            print "Failed to get times for run number", run_number, "- Start was", start, "end was", end

            continue

        delta = end - start

        times.append(delta.total_seconds())

    total_seconds = sum(times)

    minutes = int(total_seconds)/60

    hours = minutes/60

    minutes = minutes % 60

    return hours, minutes

def parse_one_setting(run_list):

    tof1_triggers = get_tof_triggers_sum("ToF1 Triggers", run_list)

    tof2_triggers = get_tof_triggers_sum("ToF2 Triggers", run_list)

    lmc1234 = get_tof_triggers_sum("LMC-1234 Count", run_list)

    hours, minutes = get_time_sum(run_list)

    return {"runs":run_list, "tof1_triggers":tof1_triggers, "tof2_triggers":tof2_triggers, "lmc1234":lmc1234, "time":[hours, minutes]}

# +100pc 155 13374/25592 0.523

# +ao    118 11575/20535 0.564

STRAIGHT_TRACKS_240_LIH_RUNS = [7727, 7733, 7737, 7738, 7741, 7775, 7776, 7790, 7794, 7795, 7796, 7805, 7808, 7809, 7813, 7814, 7816, 7817, 7818, 7819, 7844, 7845, 7847, 7848, 7849, 7851, 7852, 7853, 7854, 7855, 7856, 7858, 7859, 7860]

STRAIGHT_TRACKS_170_LIH_RUNS = [7764, 7766, 7767, 7694, 7777, 7782, 7783, 7785, 7786, 7787, 7799, 7800, 7806, 7822, 7823, 7824, 7825, 7826, 7827, 7831, 7832, 7833, 7861, 7863, 7864, 7865, 7866]

if __name__ == "__main__":

    #print parse_one_setting()

    print parse_one_setting(STRAIGHT_TRACKS_170_LIH_RUNS)