# 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:
        # refine global timing
        hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/HLDetectorTiming/FitScripts/ExtractTrackBasedTiming.C(%d)"%(int(os.environ["RUN"]))])
        # BCAL timewalks + time offsets
        hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/BCAL_TDC_Timing/FitScripts/ExtractTimeWalk.C(\"%s\")"%(os.environ["RUN_OUTPUT_FILENAME"])])
        hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/BCAL_TDC_Timing/FitScripts/ExtractTimeOffsetsAndCEff.C(%d,\"%s\")"%(int(os.environ["RUN"]),os.environ["RUN_OUTPUT_FILENAME"])])
        # SC timewalks
        hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/st_tw_corr_auto/macros/st_tw_fits.C(\"%s\")"%(os.environ["RUN_OUTPUT_FILENAME"])])
        # PSC timewalks - NEED TO UPDATE
        hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/PSC_TW/tw_corr.C(\"%s\")"%(os.environ["RUN_OUTPUT_FILENAME"])])
        hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/PS_E_calib/PSEcorr.C(\"%s\")"%(os.environ["RUN_OUTPUT_FILENAME"])])
        #hdmon_root_utils.run_root_commands([".x $HALLD_HOME/src/plugins/Calibration/st_tw_corr_auto/macros/st_tw_resols.C"]) - run at the end
        #hdmon_root_utils.run_calib_script(input_file, 
        #                                  [".x $HALLD_HOME/src/plugins/monitoring/RF_online/calib_scripts/RFMacro_ROCTITimes.C"],
        #                                  "pass0_RF_ROCTITimes.png")

    # cleanup
    input_file.Close()