# script for pass 1 calibration processing

import os,sys
import hdmon_root_utils
from ROOT import TFile

if __name__ == "__main__":
    if len(sys.argv)<2:
        print "Need to specify input file on the command line !"
        sys.exit(1)
    
    # tried to open file
    input_file = TFile(sys.argv[1])
    if input_file is None or input_file.IsZombie():
        print "Could not open file: %s"%(sys.argv[1])
        sys.exit(1)

    # load library functions
    hdmon_root_utils.load_calibration_library()

    # RF calibration and monitoring
    if "HALLD_HOME" not in os.environ:
        print "HALLD_HOME not set!"
    else:
        #hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/HLDetectorTiming/FitScripts/ExtractTDCADCTiming.C(%d)"%(int(os.environ["RUN"]))])
        # TAGM timewalks
        hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/TAGH_timewalk/scripts/gaussian_fits.C(\"%s\",true)"%(os.environ['RUN_OUTPUT_FILENAME'])])
        hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/TAGH_timewalk/scripts/gaussian_fits.C(\"%s\")"%("gaussian-fits-csv")])
        # BCAL calibrations, loop over modules 
        for module in xrange(48):
            hdmon_root_utils.run_calib_script(input_file, 
                                              [".x $HALLD_HOME/src/plugins/Calibration/BCAL_attenlength_gainratio/scripts/plot_results.C(\"%s\",%d)"%(os.environ['RUN_OUTPUT_FILENAME'],module+1)],
                                              "pass3_BCAL_gainratios_module%d.png"%(module+1))

    # cleanup
    input_file.Close()