#!/usr/bin/env python2

import os, sys, os.path, commands


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

input_name = "2011_comet_cosmics_reco.root"
if not os.path.exists(input_name): 
    print "MISSING INPUT"
    sys.exit(0)


dumpValidationFile = '2011_comet_cosmics_reco.dump'
# Get a dump of the file
status = os.system("ValidateReconGlobal_2GeVMuon.exe  -O filesuffix=_2011cosmic -O no_interactive_window "
                   + " -O tpc_min_frac=70.0 -O fgd_min_frac=50.0 -O p0d_min_frac=25.0 -O tecal_min_frac=50.0 -O dsecal_min_frac=12.0 -O smrd_min_frac=70.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; too many multi-track events."
    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 TPC mu-pull is sensible.
if eventDump.find('Good, the mean TPC Mu-pull is in range') == -1:
    print "!!!!!! Problem with TPC Mu-pull PID."
    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 670Validate2011CosmicRecon validation cleanly."
    print "" 
    print "FAIL"