import FWCore.ParameterSet.Config as cms

process = cms.Process("MCMatch")

process.load("FWCore.MessageService.MessageLogger_cfi")
process.options = cms.untracked.PSet( wantSummary = cms.untracked.bool(True) )
process.MessageLogger.cerr.FwkReport.reportEvery = 10000

# 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/M1000/RSGravToZZ_kMpl01_M-1000_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_1.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M1000/RSGravToZZ_kMpl01_M-1000_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_2.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M1000/RSGravToZZ_kMpl01_M-1000_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_3.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M1000/RSGravToZZ_kMpl01_M-1000_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_4.root',
       'root://cmsxrootd.fnal.gov//store/user/jruizvar/RSGravToZZ/M1000/RSGravToZZ_kMpl01_M-1000_Tune4C_13TeV-pythia8_MINIAODSIM_PU20bx25_5.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_M1000_ntuple.root')
                                   )

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

#process.load("SimGeneral.HepPDTESSource.pythiapdt_cfi")
#process.printTree = cms.EDAnalyzer("ParticleListDrawer",
#  maxEventsToPrint = cms.untracked.int32(5),
#  printVertex = cms.untracked.bool(False),
#  src = cms.InputTag("prunedGenParticles")
#)
#process.p=cms.Path(process.Ztoee+process.printTree)