import FWCore.ParameterSet.Config as cms

from Configuration.Generator.Pythia8CommonSettings_cfi import *
from Configuration.Generator.MCTunes2017.PythiaCP5Settings_cfi import *


source = cms.Source("EmptySource")
_generator = cms.EDFilter("Pythia8ConcurrentGeneratorFilter",
        comEnergy = cms.double(13000.0),
        crossSection = cms.untracked.double(81870000),
        filterEfficiency = cms.untracked.double(0.000345),
        maxEventsToPrint = cms.untracked.int32(0),
        pythiaHepMCVerbosity = cms.untracked.bool(False),
        pythiaPylistVerbosity = cms.untracked.int32(0),

        PythiaParameters = cms.PSet(
        pythia8CommonSettingsBlock,
        pythia8CP5SettingsBlock,
        processParameters = cms.vstring(
            'HardQCD:all = on',
            'PhaseSpace:pTHatMin = 30.         ! minimum pt hat for hard interactions', 
            'PhaseSpace:pTHatMax = 40.          ! maximum pt hat for hard interactions'),
            parameterSets = cms.vstring('pythia8CommonSettings',
                                        'pythia8CP5Settings',
                                        'processParameters')
        )
)

gj_filter = cms.EDFilter("PythiaFilterGammaGamma",
    PtSeedThr = cms.double(5.0),
    EtaSeedThr = cms.double(2.8),
    PtGammaThr = cms.double(0.0),
    EtaGammaThr = cms.double(2.8),
    PtElThr = cms.double(2.0),
    EtaElThr = cms.double(2.8),
    dRSeedMax = cms.double(0.0),
    dPhiSeedMax = cms.double(0.2),
    dEtaSeedMax = cms.double(0.12),
    dRNarrowCone = cms.double(0.02),
    PtTkThr = cms.double(1.6),
    EtaTkThr = cms.double(2.2),
    dRTkMax = cms.double(0.2),
    EnergyCut = cms.double(1.0),
    PtMinCandidate1 = cms.double(15.),
    PtMinCandidate2 = cms.double(15.),
    EtaMaxCandidate = cms.double(3.0),
    NTkConeMax = cms.int32(2),
    NTkConeSum = cms.int32(4),
    InvMassMin = cms.double(80.0),
    InvMassMax = cms.double(14000.0),
    AcceptPrompts = cms.bool(True),
    PromptPtThreshold = cms.double(15.0)                     
 
)
from GeneratorInterface.Core.ExternalGeneratorFilter import ExternalGeneratorFilter
generator = ExternalGeneratorFilter(_generator) 
ProductionFilterSequence = cms.Sequence(generator*gj_filter)