import FWCore.ParameterSet.Config as cms
from Configuration.Generator.PythiaUEZ2Settings_cfi import *

generator = cms.EDFilter("Pythia6HadronizerFilter",
    maxEventsToPrint = cms.untracked.int32(2),
    pythiaHepMCVerbosity = cms.untracked.bool(False),
    pythiaPylistVerbosity = cms.untracked.int32(1),
    comEnergy = cms.double(8000.),
    PythiaParameters = cms.PSet(
        pythiaUESettingsBlock,
        processParameters = cms.vstring(
            'MSTP(51) = 10042',
            'MSTP(52) = 2',
            'MSTP(61)=0             ! Hadronization of the initial protons initial-state QCD and QED radiation OFF',
            'MDME(997,2) = 0        ! PHASE SPACE',
            'KFDP(997,1) = 211      ! pi+',
            'KFDP(997,2) = 531      ! Bs',
            'KFDP(997,3) = 0        ! nada',
            'KFDP(997,4) = 0        ! nada',
            'KFDP(997,5) = 0        ! nada',
	    'PMAS(143,1) = 6.286', 
            'PMAS(143,4) = 0.138',
            'MDME(953,1) = 0',
            'MDME(954,1) = 0',
            'MDME(955,1) = 0',
            'MDME(956,1) = 0',
            'MDME(957,1) = 0',
            'MDME(958,1) = 0',
            'MDME(959,1) = 0',
            'MDME(960,1) = 0',
            'MDME(961,1) = 0',
            'MDME(962,1) = 0',
            'MDME(963,1) = 0',
            'MDME(964,1) = 0',
            'MDME(965,1) = 0',
            'MDME(966,1) = 0',
            'MDME(967,1) = 0',
            'MDME(968,1) = 0',
            'MDME(969,1) = 0',
            'MDME(970,1) = 0',
            'MDME(971,1) = 0',
            'MDME(972,1) = 0',
            'MDME(973,1) = 0',
            'MDME(974,1) = 0',
            'MDME(975,1) = 0',
            'MDME(976,1) = 0',
            'MDME(977,1) = 0',
            'MDME(978,1) = 0',
            'MDME(979,1) = 0',
            'MDME(980,1) = 0',
            'MDME(981,1) = 0',
            'MDME(982,1) = 1    ! J/psi phi',
            'MDME(983,1) = 0',
            'MDME(984,1) = 0',
            'MDME(985,1) = 0',
            'MDME(986,1) = 0',
            'MDME(987,1) = 0',
            'MDME(988,1) = 0',
            'MDME(989,1) = 0',
            'MDME(990,1) = 0',
            'MDME(991,1) = 0',
            'MDME(858,1) = 0         ! J/psi->e+e-',
            'MDME(859,1) = 1         ! J/psi->mumu',
            'MDME(860,1) = 0         ! J/psi->quarks',
            'MDME(656,1) = 1         ! phi->K+K-',
            'MDME(657,1) = 0',
            'MDME(658,1) = 0',
            'MDME(659,1) = 0',
            'MDME(660,1) = 0',
            'MDME(661,1) = 0',
            'MDME(662,1) = 0',
            'MDME(663,1) = 0',
            'MDME(664,1) = 0',
            'MDME(665,1) = 0',
            'MDME(666,1) = 0',
            'MDME(997,1) = 2         ! on for Bc- only, antiparticle',
            'MDME(998,1) = 3',
            'MDME(999,1) = 3',
            'MDME(1000,1) = 3',
            'MDME(1001,1) = 3',
            'MDME(1002,1) = 3',
            'MDME(1003,1) = 3',
            'MDME(1004,1) = 3',
            'MDME(1005,1) = 3',
            'MDME(1006,1) = 3',
            'MDME(1007,1) = 3',
            'MDME(1008,1) = 3',
            'MDME(1009,1) = 3',
            'MDME(1010,1) = 3',
            'MDME(1011,1) = 3',
            'MDME(1012,1) = 3',
            'MDME(1013,1) = 3',
            'MDME(1014,1) = 3',
            'MDME(1015,1) = 3',
            'MDME(1016,1) = 3',
            'MDME(1017,1) = 3',
            'MDME(1018,1) = 3',
            'MDME(1019,1) = 3',
            'MDME(1020,1) = 3',
            'MDME(1021,1) = 3',
            'MDME(1022,1) = 3',
            'MDME(1023,1) = 3',
            'MDME(1024,1) = 3',
            'MDME(1025,1) = 3',
            'MDME(1026,1) = 3',
            'MDME(1027,1) = 3'),
        parameterSets = cms.vstring('pythiaUESettings','processParameters')
    )
)

bcfilter = cms.EDFilter("PythiaFilter",
    ParticleID = cms.untracked.int32(541),
    MinPt = cms.untracked.double(9.)
)

mumugenfilter = cms.EDFilter("MCParticlePairFilter",
    Status = cms.untracked.vint32(1, 1),
    MaxEta = cms.untracked.vdouble(2.4, 2.4),
    MinEta = cms.untracked.vdouble(-2.4, -2.4),
    ParticleCharge = cms.untracked.int32(-1),
    ParticleID1 = cms.untracked.vint32(13),
    ParticleID2 = cms.untracked.vint32(13)
)

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