import FWCore.ParameterSet.Config as cms

externalLHEProducer = cms.EDProducer("ExternalLHEProducer",
    args = cms.vstring('/cvmfs/cms.cern.ch/phys_generator/gridpacks/2017/13TeV/powheg/V2/TT_hvq/TT_hdampDOWN_NNPDF31_NNLO_dilepton.tgz'),
    nEvents = cms.untracked.uint32(5000),
    numberOfParameters = cms.uint32(1),
    outputFile = cms.string('cmsgrid_final.lhe'),
    scriptName = cms.FileInPath('GeneratorInterface/LHEInterface/data/run_generic_tarball_cvmfs.sh')
)

#Link to datacards:
#https://github.com/jfernan2/genproductions/blob/d0a0f5ae6b1f828b319f5419c75fb7e6682393fa/bin/Powheg/production/2017/13TeV/TT_hvq/TT_hdamp_NNPDF31_NNLO_dilepton.input


import FWCore.ParameterSet.Config as cms
from Configuration.Generator.Pythia8CommonSettings_cfi import *
from Configuration.Generator.MCTunes2017.PythiaCP5Settings_cfi import *
from Configuration.Generator.Pythia8PowhegEmissionVetoSettings_cfi import *
from Configuration.Generator.PSweightsPythia.PythiaPSweightsSettings_cfi import *
generator = cms.EDFilter("Pythia8HadronizerFilter",
maxEventsToPrint = cms.untracked.int32(1),
pythiaPylistVerbosity = cms.untracked.int32(1),
filterEfficiency = cms.untracked.double(1.0),
pythiaHepMCVerbosity = cms.untracked.bool(False),
comEnergy = cms.double(13000.),
PythiaParameters = cms.PSet(
pythia8CommonSettingsBlock,
pythia8CP5SettingsBlock,
pythia8PowhegEmissionVetoSettingsBlock,
pythia8PSweightsSettingsBlock,
processParameters = cms.vstring(
        'POWHEG:nFinal = 2', ## Number of final state particles
        ## (BEFORE THE DECAYS) in the LHE
        ## other than emitted extra parton
        'TimeShower:mMaxGamma = 1.0',#cutting off lepton-pair production
        ##in the electromagnetic shower
        ##to not overlap with ttZ/gamma* samples
),
parameterSets = cms.vstring('pythia8CommonSettings',
'pythia8CP5Settings',
'pythia8PowhegEmissionVetoSettings',
'pythia8PSweightsSettings',
'processParameters'
)
)
)


from PhysicsTools.HepMCCandAlgos.genParticles_cfi import genParticles
from RecoJets.Configuration.GenJetParticles_cff import genParticlesForJetsNoNu
from RecoJets.Configuration.RecoGenJets_cff import ak4GenJetsNoNu

genParticlesForFilter = genParticles.clone( src = cms.InputTag("generator", "unsmeared") )

genParticlesForJetsNoNuForFilter = genParticlesForJetsNoNu.clone( src = cms.InputTag("genParticlesForFilter") )

ak4GenJetsNoNuForFilter = ak4GenJetsNoNu.clone( src = cms.InputTag("genParticlesForJetsNoNuForFilter") )



genHTFilter = cms.EDFilter("GenHTFilter",
   src = cms.InputTag("ak4GenJetsNoNuForFilter"), #GenJet collection as input
   jetPtCut = cms.double(30.0), #GenJet pT cut for HT
   jetEtaCut = cms.double(2.4), #GenJet eta cut for HT
   genHTcut = cms.double(500.0) #genHT cut
)

mcNjetsFilter = cms.EDFilter("NJetsMC",
   GenTag = cms.untracked.InputTag("ak4GenJetsNoNuForFilter"), #GenJet collection as input
   Njets = cms.int32(7), #GenJet multiplicity
   MinPt = cms.double(30.0), #GenJet pT cut
)


ProductionFilterSequence = cms.Sequence(generator*genParticlesForFilter*genParticlesForJetsNoNuForFilter*ak4GenJetsNoNuForFilter*genHTFilter*mcNjetsFilter)