FitNtuple/python/FitNtuple_cfg.py

import FWCore.ParameterSet.Config as cms
import os

realData = False
getQscale = False
getAlternating = False

WHICH = int(os.getenv("PROCESS"))

# if WHICH not in (92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111):
#     raise Exception

if WHICH == 0:
    getQscale = True
    sample = "ppMuX_CMSSW382_v6"
elif WHICH in (1, 2, 3, 4):
    getQscale = True
    sample = "InclusiveMu5_Pt30_CMSSW382_v6"
elif WHICH in (5, 6, 7, 8):
    getQscale = True
    sample = "InclusiveMu5_Pt50_CMSSW382_v6"
elif WHICH in (9, 10):
    getQscale = True
    sample = "InclusiveMu5_Pt150_CMSSW382_v6"
elif WHICH == 11:
    getQscale = True
    sample = "InclusiveMu5_Pt250_CMSSW382_v6"
elif WHICH == 12:
    getQscale = True
    sample = "InclusiveMu5_Pt350_CMSSW382_v6"

elif WHICH in (13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26): sample = "DrellYanPythia8_CMSSW382_v6"

elif WHICH == 27: sample = "JPsiToMuMu_CMSSW382_v6"
elif WHICH == 28: sample = "Psi2SToJpsiPiPi_CMSSW382_v6"
elif WHICH == 29: sample = "Psi2SToMuMu_CMSSW382_v6"

elif WHICH in (30, 31, 32, 33, 34, 35, 36, 37, 38, 39):
    getAlternating = True
    sample = "PairGun100_mass50_alternating_CMSSW382_v6"
elif WHICH in (40, 41):
    getAlternating = True
    sample = "PairGun100_alternating_CMSSW382_v6"
elif WHICH in (42, 43):
    getAlternating = True
    sample = "PairGun100_pileup_alternating_CMSSW382_v6"
elif WHICH in (44, 45):
    getAlternating = True
    sample = "PairGun100_pileup5_alternating_CMSSW382_v6"
elif WHICH in (46, 47):
    getAlternating = True
    sample = "PairGun100_dispvert100_alternating_CMSSW382_v6"
elif WHICH in (48, 49):
    getAlternating = True
    sample = "PairPairGun100_pileup_alternating_CMSSW382_v6"

elif WHICH in (50, 51, 52): sample = "NMSSMHiggs_h100_a2_pileup_CMSSW382_v6"
elif WHICH in (53, 54, 55): sample = "EXTRAU1_DMAA_pileup_CMSSW382_v6"
elif WHICH in (56, 57, 58): sample = "EXTRAU1_DMMIXED_pileup_CMSSW382_v6"
elif WHICH in (59, 60, 61): sample = "EXTRAU1_electrons_pileup_CMSSW382_v6"
elif WHICH in (62, 63, 64): sample = "EXTRAU1_pions_pileup_CMSSW382_v6"

elif WHICH == 65: sample = "WZ_CMSSW382_v6"
elif WHICH == 66: sample = "ZZ_CMSSW382_v6"
elif WHICH == 67: sample = "WW_CMSSW382_v6"

elif WHICH in (68, 69, 70, 71, 72, 73, 74, 75):
    getQscale = True
    sample = "inclBB2mu_CMSSW384_v6"
elif WHICH in (76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87):
    getQscale = True
    sample = "inclBB4mu_CMSSW384_v6"

elif WHICH == 88:
    getQscale = True
    sample = "inclBB4mu_Pt50_CMSSW384_v6"
elif WHICH == 89:
    getQscale = True
    sample = "inclBB4mu_Pt150_CMSSW384_v6"
elif WHICH == 90:
    getQscale = True
    sample = "inclBB4mu_Pt250_CMSSW384_v6"
elif WHICH == 91:
    getQscale = True
    sample = "inclBB4mu_Pt350_CMSSW384_v6"

elif WHICH in (92, 93, 94, 95, 96, 97, 98, 99, 100, 101):
    sample = "DataSep17_CMSSW382_v6"
    realData = True
elif WHICH in (102, 103, 104, 105, 106, 107, 108, 109, 110, 111):
    sample = "Data2010BPromptAll_CMSSW384_v6"
    realData = True

elif WHICH == 112: sample = "Squark2mu_200_RECO384p2_PAT387"
elif WHICH == 113: sample = "Squark2mu_250_RECO384p2_PAT387"
elif WHICH == 114: sample = "Squark2mu_300_RECO384p2_PAT387"
elif WHICH == 115: sample = "Squark2mu_350_RECO384p2_PAT387"
elif WHICH == 116: sample = "Squark2mu_400_RECO384p2_PAT387"
elif WHICH == 117: sample = "Squark2mu_450_RECO384p2_PAT387"
elif WHICH == 118: sample = "Squark2mu_500_RECO384p2_PAT387"
elif WHICH == 119: sample = "Squark2mu_550_RECO384p2_PAT387"
elif WHICH == 120: sample = "Squark2mu_600_RECO384p2_PAT387"
elif WHICH == 121: sample = "Squark2mu_650_RECO384p2_PAT387"
elif WHICH == 122: sample = "Squark2mu_700_RECO384p2_PAT387"

elif WHICH == 123: sample = "Squark4mu_200_RECO384p2_PAT387"
elif WHICH == 124: sample = "Squark4mu_250_RECO384p2_PAT387"
elif WHICH == 125: sample = "Squark4mu_300_RECO384p2_PAT387"
elif WHICH == 126: sample = "Squark4mu_350_RECO384p2_PAT387"
elif WHICH == 127: sample = "Squark4mu_400_RECO384p2_PAT387"
elif WHICH == 128: sample = "Squark4mu_450_RECO384p2_PAT387"
elif WHICH == 129: sample = "Squark4mu_500_RECO384p2_PAT387"
elif WHICH == 130: sample = "Squark4mu_550_RECO384p2_PAT387"
elif WHICH == 131: sample = "Squark4mu_600_RECO384p2_PAT387"
elif WHICH == 132: sample = "Squark4mu_650_RECO384p2_PAT387"
elif WHICH == 133: sample = "Squark4mu_700_RECO384p2_PAT387"

elif WHICH == 134: sample = "Squark350_mrs02nl_RECO384p2_PAT387"
elif WHICH == 135: sample = "Squark550_mrs02nl_RECO384p2_PAT387"

else: raise Exception

process = cms.Process("FITNTUPLE")
process.load("AnalysisDataFormats.MuJetAnalysis.filelists.%s_cff" % sample)
lenFileNames = len(process.source.fileNames)

if WHICH == 1: process.source.fileNames = process.source.fileNames[lenFileNames*0/4:lenFileNames*1/4]
if WHICH == 2: process.source.fileNames = process.source.fileNames[lenFileNames*1/4:lenFileNames*2/4]
if WHICH == 3: process.source.fileNames = process.source.fileNames[lenFileNames*2/4:lenFileNames*3/4]
if WHICH == 4: process.source.fileNames = process.source.fileNames[lenFileNames*3/4:lenFileNames*4/4]

if WHICH == 5: process.source.fileNames = process.source.fileNames[lenFileNames*0/4:lenFileNames*1/4]
if WHICH == 6: process.source.fileNames = process.source.fileNames[lenFileNames*1/4:lenFileNames*2/4]
if WHICH == 7: process.source.fileNames = process.source.fileNames[lenFileNames*2/4:lenFileNames*3/4]
if WHICH == 8: process.source.fileNames = process.source.fileNames[lenFileNames*3/4:lenFileNames*4/4]

if WHICH == 9: process.source.fileNames = process.source.fileNames[lenFileNames*0/2:lenFileNames*1/2]
if WHICH == 10: process.source.fileNames = process.source.fileNames[lenFileNames*1/2:lenFileNames*2/2]

if WHICH == 13: process.source.fileNames = process.source.fileNames[lenFileNames*0/14:lenFileNames*1/14]
if WHICH == 14: process.source.fileNames = process.source.fileNames[lenFileNames*1/14:lenFileNames*2/14]
if WHICH == 15: process.source.fileNames = process.source.fileNames[lenFileNames*2/14:lenFileNames*3/14]
if WHICH == 16: process.source.fileNames = process.source.fileNames[lenFileNames*3/14:lenFileNames*4/14]
if WHICH == 17: process.source.fileNames = process.source.fileNames[lenFileNames*4/14:lenFileNames*5/14]
if WHICH == 18: process.source.fileNames = process.source.fileNames[lenFileNames*5/14:lenFileNames*6/14]
if WHICH == 19: process.source.fileNames = process.source.fileNames[lenFileNames*6/14:lenFileNames*7/14]
if WHICH == 20: process.source.fileNames = process.source.fileNames[lenFileNames*7/14:lenFileNames*8/14]
if WHICH == 21: process.source.fileNames = process.source.fileNames[lenFileNames*8/14:lenFileNames*9/14]
if WHICH == 22: process.source.fileNames = process.source.fileNames[lenFileNames*9/14:lenFileNames*10/14]
if WHICH == 23: process.source.fileNames = process.source.fileNames[lenFileNames*10/14:lenFileNames*11/14]
if WHICH == 24: process.source.fileNames = process.source.fileNames[lenFileNames*11/14:lenFileNames*12/14]
if WHICH == 25: process.source.fileNames = process.source.fileNames[lenFileNames*12/14:lenFileNames*13/14]
if WHICH == 26: process.source.fileNames = process.source.fileNames[lenFileNames*13/14:lenFileNames*14/14]

if WHICH == 30: process.source.fileNames = process.source.fileNames[lenFileNames*0/10:lenFileNames*1/10]
if WHICH == 31: process.source.fileNames = process.source.fileNames[lenFileNames*1/10:lenFileNames*2/10]
if WHICH == 32: process.source.fileNames = process.source.fileNames[lenFileNames*2/10:lenFileNames*3/10]
if WHICH == 33: process.source.fileNames = process.source.fileNames[lenFileNames*3/10:lenFileNames*4/10]
if WHICH == 34: process.source.fileNames = process.source.fileNames[lenFileNames*4/10:lenFileNames*5/10]
if WHICH == 35: process.source.fileNames = process.source.fileNames[lenFileNames*5/10:lenFileNames*6/10]
if WHICH == 36: process.source.fileNames = process.source.fileNames[lenFileNames*6/10:lenFileNames*7/10]
if WHICH == 37: process.source.fileNames = process.source.fileNames[lenFileNames*7/10:lenFileNames*8/10]
if WHICH == 38: process.source.fileNames = process.source.fileNames[lenFileNames*8/10:lenFileNames*9/10]
if WHICH == 39: process.source.fileNames = process.source.fileNames[lenFileNames*9/10:lenFileNames*10/10]

if WHICH == 40: process.source.fileNames = process.source.fileNames[lenFileNames*0/2:lenFileNames*1/2]
if WHICH == 41: process.source.fileNames = process.source.fileNames[lenFileNames*1/2:lenFileNames*2/2]

if WHICH == 42: process.source.fileNames = process.source.fileNames[lenFileNames*0/2:lenFileNames*1/2]
if WHICH == 43: process.source.fileNames = process.source.fileNames[lenFileNames*1/2:lenFileNames*2/2]

if WHICH == 44: process.source.fileNames = process.source.fileNames[lenFileNames*0/2:lenFileNames*1/2]
if WHICH == 45: process.source.fileNames = process.source.fileNames[lenFileNames*1/2:lenFileNames*2/2]

if WHICH == 46: process.source.fileNames = process.source.fileNames[lenFileNames*0/2:lenFileNames*1/2]
if WHICH == 47: process.source.fileNames = process.source.fileNames[lenFileNames*1/2:lenFileNames*2/2]

if WHICH == 48: process.source.fileNames = process.source.fileNames[lenFileNames*0/2:lenFileNames*1/2]
if WHICH == 49: process.source.fileNames = process.source.fileNames[lenFileNames*1/2:lenFileNames*2/2]

if WHICH == 50: process.source.fileNames = process.source.fileNames[lenFileNames*0/3:lenFileNames*1/3]
if WHICH == 51: process.source.fileNames = process.source.fileNames[lenFileNames*1/3:lenFileNames*2/3]
if WHICH == 52: process.source.fileNames = process.source.fileNames[lenFileNames*2/3:lenFileNames*3/3]

if WHICH == 53: process.source.fileNames = process.source.fileNames[lenFileNames*0/3:lenFileNames*1/3]
if WHICH == 54: process.source.fileNames = process.source.fileNames[lenFileNames*1/3:lenFileNames*2/3]
if WHICH == 55: process.source.fileNames = process.source.fileNames[lenFileNames*2/3:lenFileNames*3/3]

if WHICH == 56: process.source.fileNames = process.source.fileNames[lenFileNames*0/3:lenFileNames*1/3]
if WHICH == 57: process.source.fileNames = process.source.fileNames[lenFileNames*1/3:lenFileNames*2/3]
if WHICH == 58: process.source.fileNames = process.source.fileNames[lenFileNames*2/3:lenFileNames*3/3]

if WHICH == 59: process.source.fileNames = process.source.fileNames[lenFileNames*0/3:lenFileNames*1/3]
if WHICH == 60: process.source.fileNames = process.source.fileNames[lenFileNames*1/3:lenFileNames*2/3]
if WHICH == 61: process.source.fileNames = process.source.fileNames[lenFileNames*2/3:lenFileNames*3/3]

if WHICH == 62: process.source.fileNames = process.source.fileNames[lenFileNames*0/3:lenFileNames*1/3]
if WHICH == 63: process.source.fileNames = process.source.fileNames[lenFileNames*1/3:lenFileNames*2/3]
if WHICH == 64: process.source.fileNames = process.source.fileNames[lenFileNames*2/3:lenFileNames*3/3]

if WHICH == 68: process.source.fileNames = process.source.fileNames[lenFileNames*0/8:lenFileNames*1/8]
if WHICH == 69: process.source.fileNames = process.source.fileNames[lenFileNames*1/8:lenFileNames*2/8]
if WHICH == 70: process.source.fileNames = process.source.fileNames[lenFileNames*2/8:lenFileNames*3/8]
if WHICH == 71: process.source.fileNames = process.source.fileNames[lenFileNames*3/8:lenFileNames*4/8]
if WHICH == 72: process.source.fileNames = process.source.fileNames[lenFileNames*4/8:lenFileNames*5/8]
if WHICH == 73: process.source.fileNames = process.source.fileNames[lenFileNames*5/8:lenFileNames*6/8]
if WHICH == 74: process.source.fileNames = process.source.fileNames[lenFileNames*6/8:lenFileNames*7/8]
if WHICH == 75: process.source.fileNames = process.source.fileNames[lenFileNames*7/8:lenFileNames*8/8]

if WHICH == 76: process.source.fileNames = process.source.fileNames[lenFileNames*0/12:lenFileNames*1/12]
if WHICH == 77: process.source.fileNames = process.source.fileNames[lenFileNames*1/12:lenFileNames*2/12]
if WHICH == 78: process.source.fileNames = process.source.fileNames[lenFileNames*2/12:lenFileNames*3/12]
if WHICH == 79: process.source.fileNames = process.source.fileNames[lenFileNames*3/12:lenFileNames*4/12]
if WHICH == 80: process.source.fileNames = process.source.fileNames[lenFileNames*4/12:lenFileNames*5/12]
if WHICH == 81: process.source.fileNames = process.source.fileNames[lenFileNames*5/12:lenFileNames*6/12]
if WHICH == 82: process.source.fileNames = process.source.fileNames[lenFileNames*6/12:lenFileNames*7/12]
if WHICH == 83: process.source.fileNames = process.source.fileNames[lenFileNames*7/12:lenFileNames*8/12]
if WHICH == 84: process.source.fileNames = process.source.fileNames[lenFileNames*8/12:lenFileNames*9/12]
if WHICH == 85: process.source.fileNames = process.source.fileNames[lenFileNames*9/12:lenFileNames*10/12]
if WHICH == 86: process.source.fileNames = process.source.fileNames[lenFileNames*10/12:lenFileNames*11/12]
if WHICH == 87: process.source.fileNames = process.source.fileNames[lenFileNames*11/12:lenFileNames*12/12]

if WHICH == 92: process.source.fileNames = process.source.fileNames[lenFileNames*0/10:lenFileNames*1/10]
if WHICH == 93: process.source.fileNames = process.source.fileNames[lenFileNames*1/10:lenFileNames*2/10]
if WHICH == 94: process.source.fileNames = process.source.fileNames[lenFileNames*2/10:lenFileNames*3/10]
if WHICH == 95: process.source.fileNames = process.source.fileNames[lenFileNames*3/10:lenFileNames*4/10]
if WHICH == 96: process.source.fileNames = process.source.fileNames[lenFileNames*4/10:lenFileNames*5/10]
if WHICH == 97: process.source.fileNames = process.source.fileNames[lenFileNames*5/10:lenFileNames*6/10]
if WHICH == 98: process.source.fileNames = process.source.fileNames[lenFileNames*6/10:lenFileNames*7/10]
if WHICH == 99: process.source.fileNames = process.source.fileNames[lenFileNames*7/10:lenFileNames*8/10]
if WHICH == 100: process.source.fileNames = process.source.fileNames[lenFileNames*8/10:lenFileNames*9/10]
if WHICH == 101: process.source.fileNames = process.source.fileNames[lenFileNames*9/10:lenFileNames*10/10]

if WHICH == 102: process.source.fileNames = process.source.fileNames[lenFileNames*0/10:lenFileNames*1/10]
if WHICH == 103: process.source.fileNames = process.source.fileNames[lenFileNames*1/10:lenFileNames*2/10]
if WHICH == 104: process.source.fileNames = process.source.fileNames[lenFileNames*2/10:lenFileNames*3/10]
if WHICH == 105: process.source.fileNames = process.source.fileNames[lenFileNames*3/10:lenFileNames*4/10]
if WHICH == 106: process.source.fileNames = process.source.fileNames[lenFileNames*4/10:lenFileNames*5/10]
if WHICH == 107: process.source.fileNames = process.source.fileNames[lenFileNames*5/10:lenFileNames*6/10]
if WHICH == 108: process.source.fileNames = process.source.fileNames[lenFileNames*6/10:lenFileNames*7/10]
if WHICH == 109: process.source.fileNames = process.source.fileNames[lenFileNames*7/10:lenFileNames*8/10]
if WHICH == 110: process.source.fileNames = process.source.fileNames[lenFileNames*8/10:lenFileNames*9/10]
if WHICH == 111: process.source.fileNames = process.source.fileNames[lenFileNames*9/10:lenFileNames*10/10]

process.maxEvents = cms.untracked.PSet(input = cms.untracked.int32(-1))

process.MessageLogger = cms.Service("MessageLogger", destinations = cms.untracked.vstring("cout"), cout = cms.untracked.PSet(threshold = cms.untracked.string("ERROR")))
process.load("Configuration/StandardSequences/FrontierConditions_GlobalTag_cff")
process.GlobalTag.globaltag = "MC_38Y_V12::All"

process.NoScraping = cms.EDFilter("FilterOutScraping",
                                  applyfilter = cms.untracked.bool(True),
                                  debugOn = cms.untracked.bool(False),
                                  numtrack = cms.untracked.uint32(10),
                                  thresh = cms.untracked.double(0.25)
                                  )

process.load("RecoVertex.PrimaryVertexProducer.OfflinePrimaryVertices_cfi")
process.OneGoodVertexFilter = cms.EDFilter("VertexSelector",
                                           src = cms.InputTag("offlinePrimaryVertices"),
                                           cut = cms.string("!isFake && tracksSize > 3 && abs(z) <= 24"),
                                           filter = cms.bool(True),
                                           )

process.load("AnalysisAlgos.MuJetProducer.MuJetProducer_cff")
process.MuJetProducer.muons = cms.InputTag("cleanPatMuonsTriggerMatch")

process.load("AnalysisAlgos.MuJetPlusFakeMuon.MuJetPlusFakeMuon_cfi")
process.MuJetPlusTracks = process.MuJetPlusFakeMuon.clone(numberOfFakes = cms.uint32(2))

process.FakeMuons = cms.EDProducer(
    "FakeMuons",
    tracks = cms.InputTag("generalTracks"),
    muons = cms.InputTag("cleanPatMuons"),
    minPt = cms.double(5.),
    maxAbsEta = cms.double(2.4), 
    )
process.DisplacedMuJet = process.MuJetProducer.clone(
    groupingMode = cms.string("GroupByVertexProb"))
process.DisplacedMuJetFake = process.MuJetProducer.clone(
    muons = cms.InputTag("FakeMuons"),
    groupingMode = cms.string("GroupByVertexProb"),
    minSegmentMatches = cms.int32(-1))

process.FitNtuple = cms.EDAnalyzer("FitNtuple",
                                   getQscale = cms.bool(getQscale),
                                   getAlternating = cms.bool(getAlternating),
                                   fillGenLevel = cms.bool(not realData),
                                   muJets = cms.InputTag("MuJetProducer"),
                                   muJetOrphans = cms.InputTag("MuJetProducer", "Orphans"),
                                   muJetPlusTracks = cms.InputTag("MuJetPlusTracks"),
                                   )

# special mu-jets with a mass cut of 9 GeV/c^2
process.MuJetProducer9 = process.MuJetProducer.clone(maxMass = cms.double(9.))
process.MuJetPlusTracks9 = process.MuJetPlusTracks.clone(muJets = cms.InputTag("MuJetProducer9"))
process.FitNtuple9 = cms.EDAnalyzer("FitNtuple",
                                     getQscale = cms.bool(getQscale),
                                     getAlternating = cms.bool(getAlternating),
                                     fillGenLevel = cms.bool(not realData),
                                     muJets = cms.InputTag("MuJetProducer9"),
                                     muJetOrphans = cms.InputTag("MuJetProducer9", "Orphans"),
                                     muJetPlusTracks = cms.InputTag("MuJetPlusTracks9"),
                                     )

# special mu-jets with a mass cut of 15 GeV/c^2
process.MuJetProducer15 = process.MuJetProducer.clone(maxMass = cms.double(15.))
process.MuJetPlusTracks15 = process.MuJetPlusTracks.clone(muJets = cms.InputTag("MuJetProducer15"))
process.FitNtuple15 = cms.EDAnalyzer("FitNtuple",
                                     getQscale = cms.bool(getQscale),
                                     getAlternating = cms.bool(getAlternating),
                                     fillGenLevel = cms.bool(not realData),
                                     muJets = cms.InputTag("MuJetProducer15"),
                                     muJetOrphans = cms.InputTag("MuJetProducer15", "Orphans"),
                                     muJetPlusTracks = cms.InputTag("MuJetPlusTracks15"),
                                     )

if getAlternating:
    process.Path = cms.Path(process.MuJetProducer * process.MuJetPlusTracks * process.FakeMuons * process.DisplacedMuJet * process.DisplacedMuJetFake * process.FitNtuple * process.MuJetProducer9 * process.MuJetPlusTracks9 * process.FitNtuple9 * process.MuJetProducer15 * process.MuJetPlusTracks15 * process.FitNtuple15)
else:
    process.Path = cms.Path(process.NoScraping * process.OneGoodVertexFilter * process.MuJetProducer * process.MuJetPlusTracks * process.FakeMuons * process.DisplacedMuJet * process.DisplacedMuJetFake * process.FitNtuple * process.MuJetProducer9 * process.MuJetPlusTracks9 * process.FitNtuple9 * process.MuJetProducer15 * process.MuJetPlusTracks15 * process.FitNtuple15)

process.TFileService = cms.Service("TFileService", fileName = cms.string("/uscmst1b_scratch/lpc1/lpctau/aysen/pivarski/FITNTUPLE17/FitNtuple_%s_%d.root" % (sample, WHICH)))
Revision:	1.18
Committed:	Sun Jan 23 00:48:32 2011 UTC (14 years, 3 months ago) by pivarski
Content type:	text/x-python
Branch:	MAIN
CVS Tags:	YP_2012-12-04_2011Data, V_2012_01_29_01, JP-2011-02-03
Changes since 1.17:	+1 -1 lines
Log Message:	just muons
#	User	Rev	Content
1	pivarski	1.1	import FWCore.ParameterSet.Config as cms
2			import os
3
4			realData = False
5			getQscale = False
6			getAlternating = False
7
8			WHICH = int(os.getenv("PROCESS"))
9
10	pivarski	1.17	# if WHICH not in (92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111):
11			# raise Exception
12
13	pivarski	1.1	if WHICH == 0:
14			getQscale = True
15			sample = "ppMuX_CMSSW382_v6"
16			elif WHICH in (1, 2, 3, 4):
17			getQscale = True
18			sample = "InclusiveMu5_Pt30_CMSSW382_v6"
19			elif WHICH in (5, 6, 7, 8):
20			getQscale = True
21			sample = "InclusiveMu5_Pt50_CMSSW382_v6"
22			elif WHICH in (9, 10):
23			getQscale = True
24			sample = "InclusiveMu5_Pt150_CMSSW382_v6"
25			elif WHICH == 11:
26			getQscale = True
27			sample = "InclusiveMu5_Pt250_CMSSW382_v6"
28			elif WHICH == 12:
29			getQscale = True
30			sample = "InclusiveMu5_Pt350_CMSSW382_v6"
31
32	pivarski	1.4	elif WHICH in (13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26): sample = "DrellYanPythia8_CMSSW382_v6"
33	pivarski	1.1
34			elif WHICH == 27: sample = "JPsiToMuMu_CMSSW382_v6"
35			elif WHICH == 28: sample = "Psi2SToJpsiPiPi_CMSSW382_v6"
36			elif WHICH == 29: sample = "Psi2SToMuMu_CMSSW382_v6"
37
38			elif WHICH in (30, 31, 32, 33, 34, 35, 36, 37, 38, 39):
39			getAlternating = True
40			sample = "PairGun100_mass50_alternating_CMSSW382_v6"
41			elif WHICH in (40, 41):
42			getAlternating = True
43			sample = "PairGun100_alternating_CMSSW382_v6"
44			elif WHICH in (42, 43):
45			getAlternating = True
46			sample = "PairGun100_pileup_alternating_CMSSW382_v6"
47			elif WHICH in (44, 45):
48			getAlternating = True
49			sample = "PairGun100_pileup5_alternating_CMSSW382_v6"
50			elif WHICH in (46, 47):
51			getAlternating = True
52			sample = "PairGun100_dispvert100_alternating_CMSSW382_v6"
53			elif WHICH in (48, 49):
54			getAlternating = True
55			sample = "PairPairGun100_pileup_alternating_CMSSW382_v6"
56
57			elif WHICH in (50, 51, 52): sample = "NMSSMHiggs_h100_a2_pileup_CMSSW382_v6"
58			elif WHICH in (53, 54, 55): sample = "EXTRAU1_DMAA_pileup_CMSSW382_v6"
59			elif WHICH in (56, 57, 58): sample = "EXTRAU1_DMMIXED_pileup_CMSSW382_v6"
60			elif WHICH in (59, 60, 61): sample = "EXTRAU1_electrons_pileup_CMSSW382_v6"
61			elif WHICH in (62, 63, 64): sample = "EXTRAU1_pions_pileup_CMSSW382_v6"
62
63			elif WHICH == 65: sample = "WZ_CMSSW382_v6"
64			elif WHICH == 66: sample = "ZZ_CMSSW382_v6"
65			elif WHICH == 67: sample = "WW_CMSSW382_v6"
66
67	pivarski	1.15	elif WHICH in (68, 69, 70, 71, 72, 73, 74, 75):
68			getQscale = True
69			sample = "inclBB2mu_CMSSW384_v6"
70			elif WHICH in (76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87):
71			getQscale = True
72			sample = "inclBB4mu_CMSSW384_v6"
73
74			elif WHICH == 88:
75			getQscale = True
76			sample = "inclBB4mu_Pt50_CMSSW384_v6"
77			elif WHICH == 89:
78			getQscale = True
79			sample = "inclBB4mu_Pt150_CMSSW384_v6"
80			elif WHICH == 90:
81			getQscale = True
82			sample = "inclBB4mu_Pt250_CMSSW384_v6"
83			elif WHICH == 91:
84			getQscale = True
85			sample = "inclBB4mu_Pt350_CMSSW384_v6"
86	pivarski	1.1
87			elif WHICH in (92, 93, 94, 95, 96, 97, 98, 99, 100, 101):
88			sample = "DataSep17_CMSSW382_v6"
89			realData = True
90			elif WHICH in (102, 103, 104, 105, 106, 107, 108, 109, 110, 111):
91			sample = "Data2010BPromptAll_CMSSW384_v6"
92			realData = True
93
94	pivarski	1.15	elif WHICH == 112: sample = "Squark2mu_200_RECO384p2_PAT387"
95			elif WHICH == 113: sample = "Squark2mu_250_RECO384p2_PAT387"
96			elif WHICH == 114: sample = "Squark2mu_300_RECO384p2_PAT387"
97			elif WHICH == 115: sample = "Squark2mu_350_RECO384p2_PAT387"
98			elif WHICH == 116: sample = "Squark2mu_400_RECO384p2_PAT387"
99			elif WHICH == 117: sample = "Squark2mu_450_RECO384p2_PAT387"
100			elif WHICH == 118: sample = "Squark2mu_500_RECO384p2_PAT387"
101			elif WHICH == 119: sample = "Squark2mu_550_RECO384p2_PAT387"
102			elif WHICH == 120: sample = "Squark2mu_600_RECO384p2_PAT387"
103			elif WHICH == 121: sample = "Squark2mu_650_RECO384p2_PAT387"
104			elif WHICH == 122: sample = "Squark2mu_700_RECO384p2_PAT387"
105
106			elif WHICH == 123: sample = "Squark4mu_200_RECO384p2_PAT387"
107			elif WHICH == 124: sample = "Squark4mu_250_RECO384p2_PAT387"
108			elif WHICH == 125: sample = "Squark4mu_300_RECO384p2_PAT387"
109			elif WHICH == 126: sample = "Squark4mu_350_RECO384p2_PAT387"
110			elif WHICH == 127: sample = "Squark4mu_400_RECO384p2_PAT387"
111			elif WHICH == 128: sample = "Squark4mu_450_RECO384p2_PAT387"
112			elif WHICH == 129: sample = "Squark4mu_500_RECO384p2_PAT387"
113			elif WHICH == 130: sample = "Squark4mu_550_RECO384p2_PAT387"
114			elif WHICH == 131: sample = "Squark4mu_600_RECO384p2_PAT387"
115			elif WHICH == 132: sample = "Squark4mu_650_RECO384p2_PAT387"
116			elif WHICH == 133: sample = "Squark4mu_700_RECO384p2_PAT387"
117
118	pivarski	1.16	elif WHICH == 134: sample = "Squark350_mrs02nl_RECO384p2_PAT387"
119			elif WHICH == 135: sample = "Squark550_mrs02nl_RECO384p2_PAT387"
120	pivarski	1.15
121	pivarski	1.1	else: raise Exception
122
123			process = cms.Process("FITNTUPLE")
124			process.load("AnalysisDataFormats.MuJetAnalysis.filelists.%s_cff" % sample)
125			lenFileNames = len(process.source.fileNames)
126
127			if WHICH == 1: process.source.fileNames = process.source.fileNames[lenFileNames0/4:lenFileNames1/4]
128			if WHICH == 2: process.source.fileNames = process.source.fileNames[lenFileNames1/4:lenFileNames2/4]
129			if WHICH == 3: process.source.fileNames = process.source.fileNames[lenFileNames2/4:lenFileNames3/4]
130			if WHICH == 4: process.source.fileNames = process.source.fileNames[lenFileNames3/4:lenFileNames4/4]
131
132			if WHICH == 5: process.source.fileNames = process.source.fileNames[lenFileNames0/4:lenFileNames1/4]
133			if WHICH == 6: process.source.fileNames = process.source.fileNames[lenFileNames1/4:lenFileNames2/4]
134			if WHICH == 7: process.source.fileNames = process.source.fileNames[lenFileNames2/4:lenFileNames3/4]
135			if WHICH == 8: process.source.fileNames = process.source.fileNames[lenFileNames3/4:lenFileNames4/4]
136
137			if WHICH == 9: process.source.fileNames = process.source.fileNames[lenFileNames0/2:lenFileNames1/2]
138			if WHICH == 10: process.source.fileNames = process.source.fileNames[lenFileNames1/2:lenFileNames2/2]
139
140	pivarski	1.4	if WHICH == 13: process.source.fileNames = process.source.fileNames[lenFileNames0/14:lenFileNames1/14]
141			if WHICH == 14: process.source.fileNames = process.source.fileNames[lenFileNames1/14:lenFileNames2/14]
142			if WHICH == 15: process.source.fileNames = process.source.fileNames[lenFileNames2/14:lenFileNames3/14]
143			if WHICH == 16: process.source.fileNames = process.source.fileNames[lenFileNames3/14:lenFileNames4/14]
144			if WHICH == 17: process.source.fileNames = process.source.fileNames[lenFileNames4/14:lenFileNames5/14]
145			if WHICH == 18: process.source.fileNames = process.source.fileNames[lenFileNames5/14:lenFileNames6/14]
146			if WHICH == 19: process.source.fileNames = process.source.fileNames[lenFileNames6/14:lenFileNames7/14]
147			if WHICH == 20: process.source.fileNames = process.source.fileNames[lenFileNames7/14:lenFileNames8/14]
148			if WHICH == 21: process.source.fileNames = process.source.fileNames[lenFileNames8/14:lenFileNames9/14]
149			if WHICH == 22: process.source.fileNames = process.source.fileNames[lenFileNames9/14:lenFileNames10/14]
150			if WHICH == 23: process.source.fileNames = process.source.fileNames[lenFileNames10/14:lenFileNames11/14]
151			if WHICH == 24: process.source.fileNames = process.source.fileNames[lenFileNames11/14:lenFileNames12/14]
152			if WHICH == 25: process.source.fileNames = process.source.fileNames[lenFileNames12/14:lenFileNames13/14]
153			if WHICH == 26: process.source.fileNames = process.source.fileNames[lenFileNames13/14:lenFileNames14/14]
154	pivarski	1.1
155			if WHICH == 30: process.source.fileNames = process.source.fileNames[lenFileNames0/10:lenFileNames1/10]
156			if WHICH == 31: process.source.fileNames = process.source.fileNames[lenFileNames1/10:lenFileNames2/10]
157			if WHICH == 32: process.source.fileNames = process.source.fileNames[lenFileNames2/10:lenFileNames3/10]
158			if WHICH == 33: process.source.fileNames = process.source.fileNames[lenFileNames3/10:lenFileNames4/10]
159			if WHICH == 34: process.source.fileNames = process.source.fileNames[lenFileNames4/10:lenFileNames5/10]
160			if WHICH == 35: process.source.fileNames = process.source.fileNames[lenFileNames5/10:lenFileNames6/10]
161			if WHICH == 36: process.source.fileNames = process.source.fileNames[lenFileNames6/10:lenFileNames7/10]
162			if WHICH == 37: process.source.fileNames = process.source.fileNames[lenFileNames7/10:lenFileNames8/10]
163			if WHICH == 38: process.source.fileNames = process.source.fileNames[lenFileNames8/10:lenFileNames9/10]
164			if WHICH == 39: process.source.fileNames = process.source.fileNames[lenFileNames9/10:lenFileNames10/10]
165
166			if WHICH == 40: process.source.fileNames = process.source.fileNames[lenFileNames0/2:lenFileNames1/2]
167			if WHICH == 41: process.source.fileNames = process.source.fileNames[lenFileNames1/2:lenFileNames2/2]
168
169			if WHICH == 42: process.source.fileNames = process.source.fileNames[lenFileNames0/2:lenFileNames1/2]
170			if WHICH == 43: process.source.fileNames = process.source.fileNames[lenFileNames1/2:lenFileNames2/2]
171
172			if WHICH == 44: process.source.fileNames = process.source.fileNames[lenFileNames0/2:lenFileNames1/2]
173			if WHICH == 45: process.source.fileNames = process.source.fileNames[lenFileNames1/2:lenFileNames2/2]
174
175			if WHICH == 46: process.source.fileNames = process.source.fileNames[lenFileNames0/2:lenFileNames1/2]
176			if WHICH == 47: process.source.fileNames = process.source.fileNames[lenFileNames1/2:lenFileNames2/2]
177
178			if WHICH == 48: process.source.fileNames = process.source.fileNames[lenFileNames0/2:lenFileNames1/2]
179			if WHICH == 49: process.source.fileNames = process.source.fileNames[lenFileNames1/2:lenFileNames2/2]
180
181			if WHICH == 50: process.source.fileNames = process.source.fileNames[lenFileNames0/3:lenFileNames1/3]
182			if WHICH == 51: process.source.fileNames = process.source.fileNames[lenFileNames1/3:lenFileNames2/3]
183			if WHICH == 52: process.source.fileNames = process.source.fileNames[lenFileNames2/3:lenFileNames3/3]
184
185			if WHICH == 53: process.source.fileNames = process.source.fileNames[lenFileNames0/3:lenFileNames1/3]
186			if WHICH == 54: process.source.fileNames = process.source.fileNames[lenFileNames1/3:lenFileNames2/3]
187			if WHICH == 55: process.source.fileNames = process.source.fileNames[lenFileNames2/3:lenFileNames3/3]
188
189			if WHICH == 56: process.source.fileNames = process.source.fileNames[lenFileNames0/3:lenFileNames1/3]
190			if WHICH == 57: process.source.fileNames = process.source.fileNames[lenFileNames1/3:lenFileNames2/3]
191			if WHICH == 58: process.source.fileNames = process.source.fileNames[lenFileNames2/3:lenFileNames3/3]
192
193			if WHICH == 59: process.source.fileNames = process.source.fileNames[lenFileNames0/3:lenFileNames1/3]
194			if WHICH == 60: process.source.fileNames = process.source.fileNames[lenFileNames1/3:lenFileNames2/3]
195			if WHICH == 61: process.source.fileNames = process.source.fileNames[lenFileNames2/3:lenFileNames3/3]
196
197			if WHICH == 62: process.source.fileNames = process.source.fileNames[lenFileNames0/3:lenFileNames1/3]
198			if WHICH == 63: process.source.fileNames = process.source.fileNames[lenFileNames1/3:lenFileNames2/3]
199			if WHICH == 64: process.source.fileNames = process.source.fileNames[lenFileNames2/3:lenFileNames3/3]
200
201			if WHICH == 68: process.source.fileNames = process.source.fileNames[lenFileNames0/8:lenFileNames1/8]
202			if WHICH == 69: process.source.fileNames = process.source.fileNames[lenFileNames1/8:lenFileNames2/8]
203			if WHICH == 70: process.source.fileNames = process.source.fileNames[lenFileNames2/8:lenFileNames3/8]
204			if WHICH == 71: process.source.fileNames = process.source.fileNames[lenFileNames3/8:lenFileNames4/8]
205			if WHICH == 72: process.source.fileNames = process.source.fileNames[lenFileNames4/8:lenFileNames5/8]
206			if WHICH == 73: process.source.fileNames = process.source.fileNames[lenFileNames5/8:lenFileNames6/8]
207			if WHICH == 74: process.source.fileNames = process.source.fileNames[lenFileNames6/8:lenFileNames7/8]
208			if WHICH == 75: process.source.fileNames = process.source.fileNames[lenFileNames7/8:lenFileNames8/8]
209
210			if WHICH == 76: process.source.fileNames = process.source.fileNames[lenFileNames0/12:lenFileNames1/12]
211			if WHICH == 77: process.source.fileNames = process.source.fileNames[lenFileNames1/12:lenFileNames2/12]
212			if WHICH == 78: process.source.fileNames = process.source.fileNames[lenFileNames2/12:lenFileNames3/12]
213			if WHICH == 79: process.source.fileNames = process.source.fileNames[lenFileNames3/12:lenFileNames4/12]
214			if WHICH == 80: process.source.fileNames = process.source.fileNames[lenFileNames4/12:lenFileNames5/12]
215			if WHICH == 81: process.source.fileNames = process.source.fileNames[lenFileNames5/12:lenFileNames6/12]
216			if WHICH == 82: process.source.fileNames = process.source.fileNames[lenFileNames6/12:lenFileNames7/12]
217			if WHICH == 83: process.source.fileNames = process.source.fileNames[lenFileNames7/12:lenFileNames8/12]
218			if WHICH == 84: process.source.fileNames = process.source.fileNames[lenFileNames8/12:lenFileNames9/12]
219			if WHICH == 85: process.source.fileNames = process.source.fileNames[lenFileNames9/12:lenFileNames10/12]
220			if WHICH == 86: process.source.fileNames = process.source.fileNames[lenFileNames10/12:lenFileNames11/12]
221			if WHICH == 87: process.source.fileNames = process.source.fileNames[lenFileNames11/12:lenFileNames12/12]
222
223			if WHICH == 92: process.source.fileNames = process.source.fileNames[lenFileNames0/10:lenFileNames1/10]
224			if WHICH == 93: process.source.fileNames = process.source.fileNames[lenFileNames1/10:lenFileNames2/10]
225			if WHICH == 94: process.source.fileNames = process.source.fileNames[lenFileNames2/10:lenFileNames3/10]
226			if WHICH == 95: process.source.fileNames = process.source.fileNames[lenFileNames3/10:lenFileNames4/10]
227			if WHICH == 96: process.source.fileNames = process.source.fileNames[lenFileNames4/10:lenFileNames5/10]
228			if WHICH == 97: process.source.fileNames = process.source.fileNames[lenFileNames5/10:lenFileNames6/10]
229			if WHICH == 98: process.source.fileNames = process.source.fileNames[lenFileNames6/10:lenFileNames7/10]
230			if WHICH == 99: process.source.fileNames = process.source.fileNames[lenFileNames7/10:lenFileNames8/10]
231			if WHICH == 100: process.source.fileNames = process.source.fileNames[lenFileNames8/10:lenFileNames9/10]
232			if WHICH == 101: process.source.fileNames = process.source.fileNames[lenFileNames9/10:lenFileNames10/10]
233
234			if WHICH == 102: process.source.fileNames = process.source.fileNames[lenFileNames0/10:lenFileNames1/10]
235			if WHICH == 103: process.source.fileNames = process.source.fileNames[lenFileNames1/10:lenFileNames2/10]
236			if WHICH == 104: process.source.fileNames = process.source.fileNames[lenFileNames2/10:lenFileNames3/10]
237			if WHICH == 105: process.source.fileNames = process.source.fileNames[lenFileNames3/10:lenFileNames4/10]
238			if WHICH == 106: process.source.fileNames = process.source.fileNames[lenFileNames4/10:lenFileNames5/10]
239			if WHICH == 107: process.source.fileNames = process.source.fileNames[lenFileNames5/10:lenFileNames6/10]
240			if WHICH == 108: process.source.fileNames = process.source.fileNames[lenFileNames6/10:lenFileNames7/10]
241			if WHICH == 109: process.source.fileNames = process.source.fileNames[lenFileNames7/10:lenFileNames8/10]
242			if WHICH == 110: process.source.fileNames = process.source.fileNames[lenFileNames8/10:lenFileNames9/10]
243			if WHICH == 111: process.source.fileNames = process.source.fileNames[lenFileNames9/10:lenFileNames10/10]
244
245			process.maxEvents = cms.untracked.PSet(input = cms.untracked.int32(-1))
246
247	pivarski	1.2	process.MessageLogger = cms.Service("MessageLogger", destinations = cms.untracked.vstring("cout"), cout = cms.untracked.PSet(threshold = cms.untracked.string("ERROR")))
248	pivarski	1.1	process.load("Configuration/StandardSequences/FrontierConditions_GlobalTag_cff")
249			process.GlobalTag.globaltag = "MC_38Y_V12::All"
250
251	pivarski	1.2	process.NoScraping = cms.EDFilter("FilterOutScraping",
252			applyfilter = cms.untracked.bool(True),
253			debugOn = cms.untracked.bool(False),
254			numtrack = cms.untracked.uint32(10),
255			thresh = cms.untracked.double(0.25)
256			)
257	pivarski	1.1
258	pivarski	1.2	process.load("RecoVertex.PrimaryVertexProducer.OfflinePrimaryVertices_cfi")
259			process.OneGoodVertexFilter = cms.EDFilter("VertexSelector",
260			src = cms.InputTag("offlinePrimaryVertices"),
261			cut = cms.string("!isFake && tracksSize > 3 && abs(z) <= 24"),
262			filter = cms.bool(True),
263			)
264	pivarski	1.1
265			process.load("AnalysisAlgos.MuJetProducer.MuJetProducer_cff")
266			process.MuJetProducer.muons = cms.InputTag("cleanPatMuonsTriggerMatch")
267
268	pivarski	1.8	process.load("AnalysisAlgos.MuJetPlusFakeMuon.MuJetPlusFakeMuon_cfi")
269			process.MuJetPlusTracks = process.MuJetPlusFakeMuon.clone(numberOfFakes = cms.uint32(2))
270
271	pivarski	1.12	process.FakeMuons = cms.EDProducer(
272			"FakeMuons",
273			tracks = cms.InputTag("generalTracks"),
274			muons = cms.InputTag("cleanPatMuons"),
275			minPt = cms.double(5.),
276			maxAbsEta = cms.double(2.4),
277			)
278			process.DisplacedMuJet = process.MuJetProducer.clone(
279			groupingMode = cms.string("GroupByVertexProb"))
280			process.DisplacedMuJetFake = process.MuJetProducer.clone(
281			muons = cms.InputTag("FakeMuons"),
282			groupingMode = cms.string("GroupByVertexProb"),
283			minSegmentMatches = cms.int32(-1))
284
285	pivarski	1.1	process.FitNtuple = cms.EDAnalyzer("FitNtuple",
286			getQscale = cms.bool(getQscale),
287			getAlternating = cms.bool(getAlternating),
288	pivarski	1.8	fillGenLevel = cms.bool(not realData),
289	pivarski	1.13	muJets = cms.InputTag("MuJetProducer"),
290			muJetOrphans = cms.InputTag("MuJetProducer", "Orphans"),
291			muJetPlusTracks = cms.InputTag("MuJetPlusTracks"),
292	pivarski	1.1	)
293
294	pivarski	1.13	# special mu-jets with a mass cut of 9 GeV/c^2
295			process.MuJetProducer9 = process.MuJetProducer.clone(maxMass = cms.double(9.))
296			process.MuJetPlusTracks9 = process.MuJetPlusTracks.clone(muJets = cms.InputTag("MuJetProducer9"))
297			process.FitNtuple9 = cms.EDAnalyzer("FitNtuple",
298			getQscale = cms.bool(getQscale),
299			getAlternating = cms.bool(getAlternating),
300			fillGenLevel = cms.bool(not realData),
301			muJets = cms.InputTag("MuJetProducer9"),
302			muJetOrphans = cms.InputTag("MuJetProducer9", "Orphans"),
303			muJetPlusTracks = cms.InputTag("MuJetPlusTracks9"),
304			)
305
306			# special mu-jets with a mass cut of 15 GeV/c^2
307			process.MuJetProducer15 = process.MuJetProducer.clone(maxMass = cms.double(15.))
308			process.MuJetPlusTracks15 = process.MuJetPlusTracks.clone(muJets = cms.InputTag("MuJetProducer15"))
309			process.FitNtuple15 = cms.EDAnalyzer("FitNtuple",
310			getQscale = cms.bool(getQscale),
311			getAlternating = cms.bool(getAlternating),
312			fillGenLevel = cms.bool(not realData),
313			muJets = cms.InputTag("MuJetProducer15"),
314			muJetOrphans = cms.InputTag("MuJetProducer15", "Orphans"),
315			muJetPlusTracks = cms.InputTag("MuJetPlusTracks15"),
316			)
317
318	pivarski	1.4	if getAlternating:
319	pivarski	1.13	process.Path = cms.Path(process.MuJetProducer * process.MuJetPlusTracks * process.FakeMuons * process.DisplacedMuJet * process.DisplacedMuJetFake * process.FitNtuple * process.MuJetProducer9 * process.MuJetPlusTracks9 * process.FitNtuple9 * process.MuJetProducer15 * process.MuJetPlusTracks15 * process.FitNtuple15)
320	pivarski	1.4	else:
321	pivarski	1.13	process.Path = cms.Path(process.NoScraping * process.OneGoodVertexFilter * process.MuJetProducer * process.MuJetPlusTracks * process.FakeMuons * process.DisplacedMuJet * process.DisplacedMuJetFake * process.FitNtuple * process.MuJetProducer9 * process.MuJetPlusTracks9 * process.FitNtuple9 * process.MuJetProducer15 * process.MuJetPlusTracks15 * process.FitNtuple15)
322	pivarski	1.4
323	pivarski	1.18	process.TFileService = cms.Service("TFileService", fileName = cms.string("/uscmst1b_scratch/lpc1/lpctau/aysen/pivarski/FITNTUPLE17/FitNtuple_%s_%d.root" % (sample, WHICH)))