Python AssociationMaps Examples

Example #1

File: run_split.py Project: attikis/cmssw

# Include the latest pixel templates, which are not in DB. 
# These are to be used for pixel splitting.
process.load('RecoLocalTracker.SiPixelRecHits.PixelCPETemplateReco_cfi')
process.templates.LoadTemplatesFromDB = False

# This is the default speed. Recommended.
process.StripCPEfromTrackAngleESProducer.TemplateRecoSpeed = 0;

############################## inserted new stuff

process.newrechits = cms.Sequence(process.mySiPixelRecHits*process.mySiStripRecHits)

######## track to vertex assoc ##################3
from CommonTools.RecoUtils.pf_pu_assomap_cfi import AssociationMaps
process.Vertex2TracksDefault = AssociationMaps.clone(
    AssociationType = cms.InputTag("VertexToTracks"),
    MaxNumberOfAssociations = cms.int32(1)
)

# The commands included in splitter_tracking_setup_cff.py instruct 
# the tracking machinery to use the clusters and rechits generated after 
# cluster splitting (instead of the default clusters and rechits)
#process.load('RecoLocalTracker.SubCollectionProducers.splitter_tracking_setup2_cff')
from RecoLocalTracker.SubCollectionProducers.splitter_tracking_setup_cff import customizeTracking
customizeTracking('splitClusters', 'splitClusters', 'mySiPixelRecHits', 'mySiStripRecHits')

process.fullreco = cms.Sequence(process.globalreco*process.highlevelreco)
process.Globalreco = cms.Sequence(process.globalreco)
process.Highlevelreco = cms.Sequence(process.highlevelreco)

process.options = cms.untracked.PSet(

Example #2

# These are to be used for pixel splitting.
process.load('RecoLocalTracker.SiPixelRecHits.PixelCPETemplateReco_cfi')
process.templates.LoadTemplatesFromDB = False

# This is the default speed. Recommended.
process.StripCPEfromTrackAngleESProducer.TemplateRecoSpeed = 0

############################## inserted new stuff

process.newrechits = cms.Sequence(process.mySiPixelRecHits *
                                  process.mySiStripRecHits)

######## track to vertex assoc ##################3
from CommonTools.RecoUtils.pf_pu_assomap_cfi import AssociationMaps
process.Vertex2TracksDefault = AssociationMaps.clone(
    AssociationType=cms.InputTag("VertexToTracks"),
    MaxNumberOfAssociations=cms.int32(1))

tgrIndex = process.globalreco.index(process.trackingGlobalReco)
tgrIndexFromReco = process.reconstruction_fromRECO.index(
    process.trackingGlobalReco)
process.globalreco.remove(process.trackingGlobalReco)
process.reconstruction_fromRECO.remove(process.trackingGlobalReco)
del process.trackingGlobalReco
del process.ckftracks
del process.ckftracks_wodEdX
del process.ckftracks_plus_pixelless
del process.ckftracks_woBH
del process.iterTracking
del process.InitialStep
del process.LowPtTripletStep

Example #3

File: pfcand_nopu_witham_cfg.py Project: UVa-IHEP/cmg-cmssw

process.load("Configuration.EventContent.EventContent_cff")
process.load("Configuration.StandardSequences.MagneticField_cff")

process.selectedPrimaryVertexQuality = cms.EDFilter(
    "VertexSelector",
    src=cms.InputTag('offlinePrimaryVertices'),
    cut=cms.string(
        "isValid & ndof >= 4 & chi2 > 0 & tracksSize > 0 & abs(z) < 24 & abs(position.Rho) < 2."
    ),
    filter=cms.bool(True),
)

### PFCandidate AssociationMap-specific includes
from CommonTools.RecoUtils.pf_pu_assomap_cfi import AssociationMaps

process.assMap = AssociationMaps.clone(
    VertexCollection=cms.InputTag('selectedPrimaryVertexQuality'), )

### PFCandidate AssociationMap-specific includes
from CommonTools.RecoUtils.pfcand_assomap_cfi import PFCandAssoMap

process.PFCand2VertexAM = PFCandAssoMap.clone(
    VertexCollection=cms.InputTag('selectedPrimaryVertexQuality'), )

### PFCandidateCollection-specific includes
from CommonTools.RecoUtils.pfcand_nopu_witham_cfi import FirstVertexPFCandidates

process.PFCand = FirstVertexPFCandidates.clone(
    VertexPFCandAssociationMap=cms.InputTag('PFCand2VertexAM'),
    VertexCollection=cms.InputTag('selectedPrimaryVertexQuality'),
)

Example #4

File: pfcand_nopu_witham_cfg.py Project: Andrej-CMS/cmssw

process.load('Configuration.Geometry.GeometryPilot2_cff')
process.load("Configuration.StandardSequences.RawToDigi_cff")
process.load("Configuration.EventContent.EventContent_cff")
process.load("Configuration.StandardSequences.MagneticField_cff")	
	
process.selectedPrimaryVertexQuality = cms.EDFilter("VertexSelector",
   	src = cms.InputTag('offlinePrimaryVertices'),
	cut = cms.string("isValid & ndof >= 4 & chi2 > 0 & tracksSize > 0 & abs(z) < 24 & abs(position.Rho) < 2."),
	filter = cms.bool(True),
)
		
### PFCandidate AssociationMap-specific includes
from CommonTools.RecoUtils.pf_pu_assomap_cfi import AssociationMaps
		
process.assMap = AssociationMaps.clone(
          VertexCollection = cms.InputTag('selectedPrimaryVertexQuality'),
)
		
### PFCandidate AssociationMap-specific includes
from CommonTools.RecoUtils.pfcand_assomap_cfi import PFCandAssoMap
		
process.PFCand2VertexAM = PFCandAssoMap.clone(
          VertexCollection = cms.InputTag('selectedPrimaryVertexQuality'),
)
		
### PFCandidateCollection-specific includes
from CommonTools.RecoUtils.pfcand_nopu_witham_cfi import FirstVertexPFCandidates
		
process.PFCand = FirstVertexPFCandidates.clone(
          VertexPFCandAssociationMap = cms.InputTag('PFCand2VertexAM'),
          VertexCollection = cms.InputTag('selectedPrimaryVertexQuality'),