TopRefTuple/python/tuple.py

import operator
from FWCore.ParameterSet import Config as cms

class Tuple(object) :

    def __init__(self,process,options) :
        self.process = process
        self.options = options
        self.empty = process.empty = cms.Sequence()
                
    def attr(self, item) : return getattr(self.process, item)

    def tree(self) :
        self.process.topRef = cms.EDAnalyzer( "MakeTree", outputCommands = cms.untracked.vstring(
            'drop *',
            'keep *_tuple*_*_*',) )
        return self.process.topRef
        
    def events(self) :
        self.process.tupleEvents = cms.EDProducer("Tuple_Event")
        return self.empty + self.process.tupleEvents
        
    def gen(self) :
        if self.options.isData : return self.empty
        self.process.tupleGen = cms.EDProducer("Tuple_GenParticle",
                                               InputTag = cms.InputTag('genParticles'),
                                               JetCollections = cms.VInputTag("ak5GenJetsNoNu"),
                                               Prefix = cms.string('gen'),
                                               Suffix = cms.string(''),
                                               GenStatus1PtCut = cms.double(10.0),
                                               GenJetPtCut = cms.double(10.0),
                                               )
        self.process.tuplePileup = cms.EDProducer("Tuple_PileupSummary",
                                                  InputTag = cms.InputTag('addPileupInfo'),
                                                  Prefix = cms.string('pileup'),
                                                  Suffix = cms.string('')
                                                  )
        return self.empty + self.process.tupleGen + self.process.tuplePileup

    def triggers(self) :
        if not self.options.isData : return self.empty
        return self.empty

    def electron(self) :
        self.process.tupleElectron = cms.EDProducer("Tuple_Electron",
                                                    electronTag = cms.InputTag('selectedPatElectrons'+self.options.postfix),
                                                    vertexTag = cms.InputTag('goodOfflinePrimaryVertices'),
                                                    prefix = cms.string('el'),
                                                    electronIDs = cms.vstring('mvaTrigV0')
                                                    )
        return self.empty + self.process.tupleElectron


    def muon(self) :
        self.process.tupleMuon = cms.EDProducer("Tuple_Muon",
                                                muonTag = cms.InputTag('selectedPatMuons'+self.options.postfix),
                                                prefix = cms.string('mu') )
        return self.empty + self.process.tupleMuon
        
    def path(self) :
        return cms.Path( self.events() *
                         self.gen() *
                         self.triggers() * # FIXME
                         self.electron() *
                         #muon
                         #jet
                         #met
                         #vertex
                         #triggers
                         self.tree() )
Revision:	1.5
Committed:	Fri Nov 9 18:03:13 2012 UTC (12 years, 5 months ago) by bbetchar
Content type:	text/x-python
Branch:	MAIN
Changes since 1.4:	+8 -1 lines
Log Message:	Tuple_Muon reforms
#	User	Rev	Content
1	bbetchar	1.1	import operator
2			from FWCore.ParameterSet import Config as cms
3
4			class Tuple(object) :
5
6			def __init__(self,process,options) :
7			self.process = process
8			self.options = options
9			self.empty = process.empty = cms.Sequence()
10
11			def attr(self, item) : return getattr(self.process, item)
12
13			def tree(self) :
14	bbetchar	1.2	self.process.topRef = cms.EDAnalyzer( "MakeTree", outputCommands = cms.untracked.vstring(
15	bbetchar	1.1	'drop *',
16			'keep _tuple__',) )
17	bbetchar	1.2	return self.process.topRef
18	bbetchar	1.1
19			def events(self) :
20			self.process.tupleEvents = cms.EDProducer("Tuple_Event")
21			return self.empty + self.process.tupleEvents
22
23	bbetchar	1.2	def gen(self) :
24			if self.options.isData : return self.empty
25			self.process.tupleGen = cms.EDProducer("Tuple_GenParticle",
26			InputTag = cms.InputTag('genParticles'),
27			JetCollections = cms.VInputTag("ak5GenJetsNoNu"),
28			Prefix = cms.string('gen'),
29			Suffix = cms.string(''),
30			GenStatus1PtCut = cms.double(10.0),
31			GenJetPtCut = cms.double(10.0),
32			)
33	bbetchar	1.4	self.process.tuplePileup = cms.EDProducer("Tuple_PileupSummary",
34			InputTag = cms.InputTag('addPileupInfo'),
35			Prefix = cms.string('pileup'),
36			Suffix = cms.string('')
37			)
38			return self.empty + self.process.tupleGen + self.process.tuplePileup
39
40			def triggers(self) :
41			if not self.options.isData : return self.empty
42			return self.empty
43	bbetchar	1.2
44	bbetchar	1.3	def electron(self) :
45			self.process.tupleElectron = cms.EDProducer("Tuple_Electron",
46			electronTag = cms.InputTag('selectedPatElectrons'+self.options.postfix),
47			vertexTag = cms.InputTag('goodOfflinePrimaryVertices'),
48			prefix = cms.string('el'),
49			electronIDs = cms.vstring('mvaTrigV0')
50			)
51			return self.empty + self.process.tupleElectron
52
53	bbetchar	1.5
54			def muon(self) :
55			self.process.tupleMuon = cms.EDProducer("Tuple_Muon",
56			muonTag = cms.InputTag('selectedPatMuons'+self.options.postfix),
57			prefix = cms.string('mu') )
58			return self.empty + self.process.tupleMuon
59
60	bbetchar	1.1	def path(self) :
61			return cms.Path( self.events() *
62	bbetchar	1.2	self.gen() *
63	bbetchar	1.5	self.triggers() * # FIXME
64	bbetchar	1.3	self.electron() *
65	bbetchar	1.4	#muon
66			#jet
67			#met
68			#vertex
69			#triggers
70	bbetchar	1.1	self.tree() )