import FWCore.ParameterSet.Config as cms

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


generator = cms.EDFilter("Pythia8GeneratorFilter",
        comEnergy = cms.double(13000.0),
        crossSection = cms.untracked.double(4.343),
        filterEfficiency = cms.untracked.double(1),
        maxEventsToPrint = cms.untracked.int32(1),
        pythiaHepMCVerbosity = cms.untracked.bool(False),
        pythiaPylistVerbosity = cms.untracked.int32(1),
        PythiaParameters = cms.PSet(
               pythia8CommonSettingsBlock,
               pythia8CP5SettingsBlock,
               pythia8PSweightsSettingsBlock,
               processParameters = cms.vstring(
                   'PhaseSpace:pTHatMin = 30.',
                   'WeakSingleBoson:ffbar2ffbar(s:gmZ) = on',
                   'Diffraction:doHard = on',
                   'Diffraction:sampleType = 4',
                   'SigmaDiffractive:PomFlux = 5',
                   'PDF:PomSet = 4',
                   '23:onMode = off',  #turn off all Z decays
                   '23:onIfAny = 13' ),  # turn ON Z->mumu
               parameterSets = cms.vstring('pythia8CommonSettings',
                                           'pythia8CP5Settings',
                                           'pythia8PSweightsSettings',
                                           'processParameters')
        )
                         
)
ProductionFilterSequence = cms.Sequence(generator)