L1TCommon/scripts/optionsL1T.py

0001 #
0002 # L1T Command Line Options:
0003 #
0004 # Append this snippet to you cmsDriver.py config file like this:
0005 #
0006 #   cat L1Trigger/L1TCommon/scripts/optionsL1T.py
0007 #
0008 # to provide support for command-line options such as:
0009 #
0010 #    maxEvents=<n>
0011 #    skip=<n>
0012 #    ntuple=<file>
0013 #    inputFiles="file1.root,file2.root"
0014 #
0015 #    menu (not yet implemented)
0016 #
0017 import FWCore.ParameterSet.Config as cms
0018 import FWCore.ParameterSet.VarParsing as VarParsing
0019 import os
0020 import sys
0021 options = VarParsing.VarParsing ('analysis')
0022 options = VarParsing.VarParsing ('analysis')
0023 options.register ('ntuple', "",  VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.string,  "The output ntuple file name")
0024 options.register ('menu',   "",  VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.string,  "Override the L1 menu with specified XML file")
0025 options.register ('skip',   "",  VarParsing.VarParsing.multiplicity.singleton, VarParsing.VarParsing.varType.int,     "The number of events to skip initially")
0026 options.ntuple = ''
0027 options.menu = ''
0028 options.skip = 0
0029 print(options)
0030
0031 options.parseArguments()
0032
0033 if (not options.menu == ""):
0034     print("L1T INFO:  menu override in command line not yet implemented...")
0035
0036 if (not options.ntuple == ""):
0037     print("L1T INFO:  using command line option ntuple:  ", options.ntuple)
0038     if (hasattr(process,"TFileService")):
0039         process.TFileService.fileName = cms.string(options.ntuple)
0040         #print process.TFileService
0041
0042 if (hasattr(process,"maxEvents")):
0043     print("L1T INFO:  using command line option maxEvents:  ", options.maxEvents)
0044     process.maxEvents.input = options.maxEvents
0045     #print process.maxEvents
0046
0047 if (hasattr(process,"source")):
0048     if options.skip > 0:
0049         print("L1T INFO:  using command line option skip:  ", options.skip)
0050         process.source.skipEvents = cms.untracked.uint32(options.skip)
0051     if (not options.inputFiles == []):
0052         print("L1T INFO:  using command line option inputFiles:  ", options.inputFiles)
0053         process.source.fileNames = cms.untracked.vstring(options.inputFiles)