Validation/RecoParticleFlow/test/pfanalysis_ntuple.py - master - githubmirror/cmssw

Validation/RecoParticleFlow/test/pfanalysis_ntuple.py
53 строки2 KB
Joosep Pata
MLPF 2025/2026 paper checkpoint
02 мар 2026, 13:20
02 мар 2026, 13:202fde14b
import FWCore.ParameterSet.Config as cms
from Configuration.StandardSequences.Eras import eras

process = cms.Process("Demo")
process.load('Configuration.StandardSequences.Services_cff')
process.load('SimGeneral.HepPDTESSource.pythiapdt_cfi')
process.load('Configuration.StandardSequences.GeometrySimDB_cff')
process.load("Configuration.StandardSequences.GeometryRecoDB_cff")
process.load('Configuration.StandardSequences.MagneticField_cff')
process.load('Configuration.EventContent.EventContent_cff')
process.load("FWCore.MessageService.MessageLogger_cfi")
process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
process.load('Configuration.StandardSequences.SimIdeal_cff')
#process.hcalTopologyIdeal.MergePosition = False

#Add track sim to reco associations
process.load("SimTracker.TrackerHitAssociation.tpClusterProducer_cfi")
process.load("SimTracker.TrackAssociatorProducers.quickTrackAssociatorByHits_cfi")
process.load("SimTracker.TrackAssociatorProducers.trackAssociatorByHits_cfi")
process.load("SimTracker.TrackAssociation.trackingParticleRecoTrackAsssociation_cfi")
  
process.trackingParticleGsfTrackAssociation = process.trackingParticleRecoTrackAsssociation.clone(label_tr="electronGsfTracks")

from Configuration.AlCa.GlobalTag import GlobalTag
process.GlobalTag = GlobalTag(process.GlobalTag, "auto:phase1_2022_realistic")
from FastSimulation.Event.ParticleFilter_cfi import *

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

process.source = cms.Source("PoolSource",
    fileNames = cms.untracked.vstring(
        'file://step3_phase1_new.root'
    ),
    duplicateCheckMode = cms.untracked.string("noDuplicateCheck")
)

process.pfana = cms.EDAnalyzer('PFAnalysis')

process.TFileService = cms.Service("TFileService",
    fileName = cms.string("pfntuple.root")
)
process.p = cms.Path(
  process.tpClusterProducer*
  process.quickTrackAssociatorByHits*
  process.trackingParticleRecoTrackAsssociation*
  process.trackingParticleGsfTrackAssociation*
  process.pfana)


#Enable LogTrace in PFAnalysisNtuplizer
## scram b -j8 USER_CXXFLAGS="-DEDM_ML_DEBUG"
#process.MessageLogger.cerr.threshold = "DEBUG"
#process.MessageLogger.debugModules = ["pfana"]