import argparse
argParser = argparse.ArgumentParser(
description="Argument parser for samples file")
argParser.add_argument('--overwrite',
action='store_true',
help="Overwrite current entry in db?")
argParser.add_argument('--update',
action='store_true',
help="Update current entry in db?")
return argParser
# Logging
if __name__ == "__main__":
import nanoMET.tools.logger as logger
logger = logger.get_logger("INFO", logFile=None)
import RootTools.core.logger as logger_rt
logger_rt = logger_rt.get_logger("INFO", logFile=None)
options = get_parser().parse_args()
ov = options.overwrite
if options.update:
ov = 'update'
else:
import logging
logger = logging.getLogger(__name__)
ov = False
# Redirector
from nanoMET.tools.user import redirector_global as redirector
# DB
argParser.add_argument('--year', action='store', default=2016)
argParser.add_argument('--tuneEra', action='store', default=False)
argParser.add_argument('--small', action='store_true', help='Small?')
args = argParser.parse_args()
#RootTools
from RootTools.core.standard import *
plot_directory = "/afs/hephy.at/user/d/dspitzbart/www/nanoMET/validation/"
if args.tuneEra: plot_directory += "tune%s/"%args.tuneEra
import nanoMET.tools.logger as logger
import RootTools.core.logger as logger_rt
logger = logger.get_logger( args.logLevel, logFile = None)
logger_rt = logger_rt.get_logger(args.logLevel, logFile = None)
from nanoMET.samples.color import *
data_directory = "/afs/hephy.at/data/dspitzbart03/nanoSamples/"
postProcessing_directory = "2016_v6/Inclusive/"
dirs = {}
dirs['WJetsToLNu'] = ["WJetsToLNu_ext"]
dirs['TTZToQQ'] = ['TTZToQQ']
directories = { key : [ os.path.join( data_directory, postProcessing_directory, dir) for dir in dirs[key]] for key in dirs.keys()}
WJetsToLNu_16 = Sample.fromDirectory(name="WJets", treeName="Events", isData=False, color=color.WJets, texName="W+Jets", directory=directories['WJetsToLNu'])
TTZToQQ_16 = Sample.fromDirectory(name="TTZToQQ", treeName="Events", isData=False, color=color.TTZ, texName="t#bar{t}Z, had", directory=directories['TTZToQQ'])
data_directory = "/afs/hephy.at/data/dspitzbart03/nanoSamples/"
help="Prepare, don't acutally run")
argParser.add_argument('--overwrite', action='store_true', help="Overwrite")
argParser.add_argument('--year',
action='store',
default=None,
help="Which year? Important for json file.")
argParser.add_argument('--era',
action='store',
default="v1",
help="Which era/subdirectory?")
options = argParser.parse_args()
# Logger
import nanoMET.tools.logger as logger
import RootTools.core.logger as logger_rt
logger = logger.get_logger(options.logLevel, logFile=None)
logger_rt = logger_rt.get_logger(options.logLevel, logFile=None)
# from RootTools
from RootTools.core.standard import *
def nonEmptyFile(f, treeName='Events'):
logger.info("Checking file: %s" % f)
rf = ROOT.TFile.Open(f)
if not rf: return False
tree = getattr(rf, treeName)
nonEmpty = True if tree.GetEntries() else False
if not nonEmpty: logger.info("File is empty")
rf.Close()
return nonEmpty
