import FWCore.ParameterSet.Config as cms

process = cms.Process("MCMatch")

process.load("FWCore.MessageService.MessageLogger_cfi")
process.MessageLogger.cerr.FwkReport.reportEvery = 1000

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

process.source = cms.Source("PoolSource",
    fileNames = cms.untracked.vstring(
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M4500/RSGravToZZ_kMpl01_M-4500_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_1.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M4500/RSGravToZZ_kMpl01_M-4500_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_2.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M4500/RSGravToZZ_kMpl01_M-4500_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_3.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M4500/RSGravToZZ_kMpl01_M-4500_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_4.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M4500/RSGravToZZ_kMpl01_M-4500_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_5.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M4500/RSGravToZZ_kMpl01_M-4500_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_6.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M4500/RSGravToZZ_kMpl01_M-4500_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_7.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M4500/RSGravToZZ_kMpl01_M-4500_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_8.root',
    )
)
process.Ztoee = cms.EDFilter("CandViewSelector",
    src = cms.InputTag("prunedGenParticles"),
    cut = cms.string("pdgId()==23 && abs(daughter(0).pdgId())==11"),
    filter = cms.bool(True)
)
process.electronMCMatch = cms.EDProducer("MCTruthDeltaRMatcherNew",
     src = cms.InputTag("slimmedElectrons"),
     matched = cms.InputTag("prunedGenParticles"),
     distMin = cms.double(0.2),
     matchPDGId = cms.vint32(11)  
)
process.veto = cms.EDAnalyzer('EDBRGenStudies',
     electrons = cms.InputTag("slimmedElectrons"),
     associatedGenParticle = cms.InputTag("electronMCMatch"),
     electronIDs = cms.InputTag("cutBasedElectronID-CSA14-PU20bx25-V0-standalone-veto")
)
process.loose  = process.veto.clone(electronIDs = "cutBasedElectronID-CSA14-PU20bx25-V0-standalone-loose") 
process.medium = process.veto.clone(electronIDs = "cutBasedElectronID-CSA14-PU20bx25-V0-standalone-medium") 
process.tight  = process.veto.clone(electronIDs = "cutBasedElectronID-CSA14-PU20bx25-V0-standalone-tight") 

process.TFileService = cms.Service("TFileService",
                                   fileName = cms.string('RSGravToZZ_M4500_ntuple.root')
                                   )

process.p = cms.Path(process.Ztoee + process.electronMCMatch + process.veto + process.loose + process.medium + process.tight)