#!/usr/bin/env python2
# Validate the reconstruction of muons from FGD1 to ECAL.

import os, sys, os.path, commands


try:
    pkgpath = sys.argv[1]
except IndexError:
    pkgpath = ".."

input_name = "sideways-muons-1GeV-FGD1_reco.root"
if not os.path.exists(input_name): 
    print "MISSING INPUT"
    sys.exit(0)


dumpValidationFile = 'sideways-muons-1GeV-FGD1_reco.dump'
# Get a dump of the file
status = os.system("ValidateReconGlobal_2GeVMuon.exe  -O filesuffix=_fgd1ecal -O no_interactive_window -O TrShVal_mincutoff=0.9 -O TrShVal_maxcutoff=1.1 "
                   + " -O ntrack_cutoff=0.25 -O fgd_min_frac=80.0 -O tecal_min_frac=80.0  "
                   + input_name + 
          " > " + dumpValidationFile)

eventDump = commands.getoutput('cat ' + dumpValidationFile)

# Look for strings that should or shouldn't be in the file.
valid = True

# check that ratio of 1:2 tracker tracks and the 0:1 tracker tracks is small
if eventDump.find('Good, the ratio is less') == -1:
    print "!!!!!! Problem with number of reconstructed tracker tracks."
    valid = False

# check that track quality is sensible.
if eventDump.find('Good, the mean is') == -1:
    print "!!!!!! Problem with quality of global PIDs."
    valid = False

# check that ECAL TrShVal is sensible.
if eventDump.find('Good, the mean Tracker ECAL TrShVal') == -1:
    print "!!!!!! Problem with ECAL TrShVal of global PIDs."
    valid = False

# check that fraction of used tracks is reasonable.
if eventDump.find('Bad: too small a fraction of tracks have') != -1:
    print "!!!!!! Problem with fractional contribution from sub-detector to global track."
    valid = False

if not valid: 
    print eventDump
    print "Failed to finish 680ValidateForwardElectronRecon validation cleanly."
    print "" 
    print "FAIL"