import FWCore.ParameterSet.Config as cms

externalLHEProducer = cms.EDProducer("ExternalLHEProducer",
    args = cms.vstring('/cvmfs/cms.cern.ch/phys_generator/gridpacks/UL/13TeV/madgraph/V5_2.6.1/GJets_1j_Gpt_Update/GJets_1j_Gpt-70To230_5f_NLO_slc7_amd64_gcc700_CMSSW_10_6_0_tarball.tar.xz'),
    nEvents = cms.untracked.uint32(5000),
    numberOfParameters = cms.uint32(1),
    generateConcurrently = cms.untracked.bool(True),
	outputFile = cms.string('cmsgrid_final.lhe'),
    scriptName = cms.FileInPath('GeneratorInterface/LHEInterface/data/run_generic_tarball_cvmfs.sh')
)
import FWCore.ParameterSet.Config as cms

from Configuration.Generator.Pythia8CommonSettings_cfi import *
from Configuration.Generator.MCTunes2017.PythiaCP5Settings_cfi import *
from Configuration.Generator.Pythia8aMCatNLOSettings_cfi import *
from Configuration.Generator.PSweightsPythia.PythiaPSweightsSettings_cfi import *

generator = cms.EDFilter("Pythia8ConcurrentHadronizerFilter",
    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,
        pythia8aMCatNLOSettingsBlock,
        pythia8PSweightsSettingsBlock,
        processParameters = cms.vstring(
            'TimeShower:nPartonsInBorn = 1', #number of coloured particles (before resonance decays) in born matrix element
        ),
        parameterSets = cms.vstring('pythia8CommonSettings',
                                    'pythia8CP5Settings',
                                    'pythia8aMCatNLOSettings',
                                    'pythia8PSweightsSettings',
                                    'processParameters',
                                    )
    )
)

LHEPtFilter = cms.EDFilter("LHEPtFilter",
                           selectedPdgIds = cms.vint32(22),
                           ptMax = cms.double(200.0),
                           ptMin = cms.double(100.0),
                           src = cms.InputTag("externalLHEProducer")
)

tmpGenParticles = cms.EDProducer("GenParticleProducer",
        saveBarCodes = cms.untracked.bool(True),
        src = cms.InputTag("generator", "unsmeared"),
        abortOnUnknownPDGCode = cms.untracked.bool(False)
        )

tmpGenParticlesForJetsNoNu = cms.EDProducer("InputGenJetsParticleSelector",
        src = cms.InputTag("tmpGenParticles"),
        ignoreParticleIDs = cms.vuint32(
            1000022,
         1000012, 1000014, 1000016,
         2000012, 2000014, 2000016,
         1000039, 5100039,
         4000012, 4000014, 4000016,
         9900012, 9900014, 9900016,
         39,12,14,16),
        partonicFinalState = cms.bool(False),
        excludeResonances = cms.bool(False),
        excludeFromResonancePids = cms.vuint32(12, 13, 14, 16),
        tausAsJets = cms.bool(False)
)

from RecoJets.JetProducers.ak4GenJets_cfi import ak4GenJets
tmpak4GenJetsNoNu = ak4GenJets.clone( src = "tmpGenParticlesForJetsNoNu" )

ajjGenJetFilterPhotonInBarrelMjj300 = cms.EDFilter("AJJGenJetFilter",
                                                   GenJetCollection = cms.InputTag('tmpak4GenJetsNoNu'),
                                                   genParticles = cms.InputTag("tmpGenParticles"),
                                                   
                                                   #following cuts are applied to select jets
                                                   minPt = cms.double( 40 ),
                                                   minEta = cms.double( -4.5 ),
                                                   maxEta = cms.double( 4.5 ),
                                                   deltaRJetLep = cms.double( 0.3 ),

                                                   #following cuts are applied on the first two leading jets
                                                   minDeltaEta = cms.double( 3.0 ),
                                                   maxDeltaEta = cms.double( 999.0 ),
                                                   MinInvMass = cms.double( 300 ),
                                                   
                                                   #the cut on the photon eta
                                                   maxPhotonEta = cms.double( 1.48 ),
                                                   minPhotonPt  = cms.double( 50 ),
                                                   maxPhotonPt  = cms.double( 10000 )
)

ProductionFilterSequence = cms.Sequence(generator*LHEPtFilter*tmpGenParticles*tmpGenParticlesForJetsNoNu*tmpak4GenJetsNoNu*ajjGenJetFilterPhotonInBarrelMjj300)


# Link to generator fragment:
# ./genFragments/Hadronizer/13TeV/Hadronizer_TuneCP5_13TeV_Gpt675toInf_aMCatNLO_1p_LHE_pythia8_cff.py