import FWCore.ParameterSet.Config as cms
from Configuration.Generator.Pythia8CommonSettings_cfi import *
from Configuration.Generator.MCTunes2017.PythiaCP5Settings_cfi import *
from GeneratorInterface.EvtGenInterface.EvtGenSetting_cff import *

generator = cms.EDFilter("Pythia8GeneratorFilter",
                         pythiaPylistVerbosity = cms.untracked.int32(0),
                         pythiaHepMCVerbosity = cms.untracked.bool(False),
                         comEnergy = cms.double(13000.0),
                         maxEventsToPrint = cms.untracked.int32(0),
                         ExternalDecays = cms.PSet(
                         EvtGen130 = cms.untracked.PSet(
                             decay_table = cms.string('GeneratorInterface/EvtGenInterface/data/DECAY_2014_NOLONGLIFE.DEC'),
                             particle_property_file = cms.FileInPath('GeneratorInterface/EvtGenInterface/data/evt_2014.pdl'),
                             user_decay_file = cms.vstring('GeneratorInterface/ExternalDecays/data/Onia_mumu.dec'),
                             list_forced_decays = cms.vstring('MyUpsilon','MyUpsilon(2S)','MyUpsilon(3S)'),
                             operates_on_particles = cms.vint32(553,100553,200553),
                             convertPythiaCodes = cms.untracked.bool(False)
                             ),
                        parameterSets = cms.vstring('EvtGen130')
                        ),
                        PythiaParameters = cms.PSet(
                            pythia8CommonSettingsBlock,
                            pythia8CP5SettingsBlock,
                            processParameters = cms.vstring(
                            'Bottomonium:all = on',                     # Quarkonia, MSEL=62, allow feed-down
                            'PhaseSpace:pTHatMin = 5.0'                 # ckin(3), be aware of this
                            ),
                        parameterSets = cms.vstring('pythia8CommonSettings',
                            'pythia8CP5Settings',
                            'processParameters',
                            )
        ),

)

generator.PythiaParameters.processParameters.extend(EvtGenExtraParticles)

mumugenfilter = cms.EDFilter("MCParticlePairFilter",
    Status = cms.untracked.vint32(1, 1),
    MaxInvMass = cms.untracked.double(11.4),
    MinInvMass = cms.untracked.double(8.6),
    ParticleCharge = cms.untracked.int32(-1),
    ParticleID1 = cms.untracked.vint32(13),
    ParticleID2 = cms.untracked.vint32(13)
)


oniafilter = cms.EDFilter("MCMultiParticleFilter",
     Status = cms.vint32(2, 2, 2),
     ParticleID = cms.vint32(553,100553,200553),
     PtMin = cms.vdouble(7,7,7),
     NumRequired = cms.int32(1),
     EtaMax = cms.vdouble(1.4, 1.4, 1.4),
     EtaMin = cms.vdouble(-1.4, -1.4, -1.4),
     AcceptMore = cms.bool(True)
)

ProductionFilterSequence = cms.Sequence(generator*oniafilter*mumugenfilter)