ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/CmsHi/JetAnalysis/test/analyzeAlgo.py
Revision: 1.1
Committed: Fri Jun 4 10:22:56 2010 UTC (14 years, 11 months ago) by yilmaz
Content type: text/x-python
Branch: MAIN
Log Message:
Algorithm analyzer

File Contents

# User Rev Content
1 yilmaz 1.1 import FWCore.ParameterSet.VarParsing as VarParsing
2    
3     ivars = VarParsing.VarParsing('standard')
4    
5     ivars.files = 'rfio:/castor/cern.ch/user/y/yilmaz/pat/CMSSW_3_6_0/UnquenchedDijet80to120_runs1to500.root'
6     ivars.output = 'algo_80to120_4.root'
7     ivars.maxEvents = 200
8    
9     ivars.parseArguments()
10    
11     import FWCore.ParameterSet.Config as cms
12    
13     process = cms.Process('ALGO')
14    
15     process.source = cms.Source("PoolSource",
16     fileNames = cms.untracked.vstring(ivars.files),
17     skipEvents = cms.untracked.uint32(200)
18     )
19    
20     process.maxEvents = cms.untracked.PSet(
21     input = cms.untracked.int32(ivars.maxEvents)
22     )
23    
24     process.load('Configuration.StandardSequences.Services_cff')
25     process.load('Configuration.StandardSequences.Geometry_cff')
26     process.load('RecoJets.Configuration.GenJetParticles_cff')
27     process.load('RecoHI.HiJetAlgos.HiGenJets_cff')
28     process.load('RecoHI.HiJetAlgos.HiRecoJets_cff')
29     process.load('PhysicsTools.PatAlgos.patHeavyIonSequences_cff')
30     from PhysicsTools.PatAlgos.tools.heavyIonTools import *
31     configureHeavyIons(process)
32    
33     process.RandomNumberGeneratorService.icPu5CaloJets = process.RandomNumberGeneratorService.generator.clone()
34     process.RandomNumberGeneratorService.icPu5CaloJets0 = process.RandomNumberGeneratorService.generator.clone()
35     process.RandomNumberGeneratorService.icPu5CaloJets1 = process.RandomNumberGeneratorService.generator.clone()
36     process.RandomNumberGeneratorService.icPu5CaloJets2 = process.RandomNumberGeneratorService.generator.clone()
37    
38     process.TFileService = cms.Service('TFileService',
39     fileName = cms.string('treefile4.root')
40     )
41    
42     from RecoJets.JetProducers.CaloJetParameters_cfi import *
43     from RecoJets.JetProducers.AnomalousCellParameters_cfi import *
44    
45     process.icPu5CaloJets = cms.EDProducer('JetAlgorithmAnalyzer',
46     CaloJetParameters,
47     AnomalousCellParameters,
48     jetAlgorithm = cms.string("IterativeCone"),
49     rParam = cms.double(0.5),
50     avoidNegative = cms.bool(False)
51     )
52    
53    
54     process.icPu5CaloJets.doPUOffsetCorr = True
55     process.icPu5CaloJets.doPVCorrection = False
56     process.icPu5CaloJets.jetPtMin = 10
57     process.icPu5CaloJets.centrality = cms.untracked.int32(0)
58    
59     #process.icPu5CaloJets.reRunAlgo = cms.untracked.bool(True)
60    
61     process.icPu5CaloJets0 = process.icPu5CaloJets.clone()
62     process.icPu5CaloJets1 = process.icPu5CaloJets.clone(centrality = cms.untracked.int32(15))
63     process.icPu5CaloJets2 = process.icPu5CaloJets.clone(centrality = cms.untracked.int32(30))
64    
65     process.algoAna = cms.Sequence(process.icPu5CaloJets0 + process.icPu5CaloJets1 + process.icPu5CaloJets2)
66    
67     process.icPu5corr = process.patJetCorrFactors.clone(jetSource = cms.InputTag("icPu5CaloJets"),
68     corrLevels = cms.PSet(L2Relative = cms.string("L2Relative_IC5Calo"),
69     L3Absolute = cms.string("L3Absolute_IC5Calo")))
70     process.icPu5clean = process.heavyIonCleanedGenJets.clone(src = cms.untracked.InputTag('iterativeCone5HiGenJets'))
71     process.icPu5match = process.patJetGenJetMatch.clone(src = cms.InputTag("icPu5CaloJets"),
72     matched = cms.InputTag("heavyIonCleanedGenJets"))
73     process.icPu5parton = process.patJetPartonMatch.clone(src = cms.InputTag("icPu5CaloJets"))
74     process.icPu5patJets = process.patJets.clone(jetSource = cms.InputTag("icPu5CaloJets"),
75     genJetMatch = cms.InputTag("icPu5match"),
76     genPartonMatch= cms.InputTag("icPu5parton"),
77     jetCorrFactorsSource = cms.VInputTag(cms.InputTag("icPu5corr")))
78     process.icPu5patSequence = cms.Sequence(process.icPu5corr*process.icPu5match*process.icPu5parton*process.icPu5patJets)
79    
80     process.load("HeavyIonsAnalysis.Configuration.analysisEventContent_cff")
81    
82     process.output = cms.OutputModule("PoolOutputModule",
83     process.jetTrkSkimContent,
84     fileName = cms.untracked.string(ivars.output)
85     )
86    
87     process.output.outputCommands.append("keep *_*_*_ALGO")
88    
89    
90     process.load("CondCore.DBCommon.CondDBCommon_cfi")
91     process.CondDBCommon.connect = "sqlite_file:/home/yilmaz/cms361v2/src/RecoHI/HiCentralityAlgos/data/CentralityTables.db"
92    
93     process.PoolDBESSource = cms.ESSource("PoolDBESSource",
94     process.CondDBCommon,
95     toGet = cms.VPSet(cms.PSet(record = cms.string('HeavyIonRcd'),
96     tag = cms.string('HFhits40_MC_Hydjet2760GeV_MC_3XY_V24_v0')
97     )
98     )
99     )
100    
101    
102     process.runAllJets = cms.Path(process.hiGenParticlesForJets *
103     process.hiRecoGenJets +
104     process.iterativeConePu5CaloJets *
105     process.makeHeavyIonJets +
106     process.algoAna # *
107     # process.icPu5patSequence
108     )
109    
110     process.out_step = cms.EndPath(process.output)
111