#Before Filter: total cross section = 5.413e+08 +- 5.386e+06 pb
#Filter efficiency (taking into account weights)= (2) / (10000) = 2.000e-04 +- 1.414e-04
#Filter efficiency (event-level)= (2) / (10000) = 2.000e-04 +- 1.414e-04    [TO BE USED IN MCM]

#After filter: final cross section = 1.083e+05 +- 7.655e+04 pb
#After filter: final fraction of events with negative weights = 0.000e+00 +- 0.000e+00
#After filter: final equivalent lumi for 1M events (1/fb) = 9.238e-03 +- 6.532e-03

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",
                         comEnergy = cms.double(13000.0),
                         pythiaHepMCVerbosity = cms.untracked.bool(False),
                         maxEventsToPrint = cms.untracked.int32(0),
                         pythiaPylistVerbosity = 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_embedded= cms.vstring(
'#',
'Particle   pi+         1.3957061e-01   0.0000000e+00',
'Particle   pi-         1.3957061e-01   0.0000000e+00',
'Particle   K_S0        4.9761100e-01   0.0000000e+00',
'Particle   K*+         8.9176000e-01   5.0300000e-02',
'Particle   K*-         8.9176000e-01   5.0300000e-02',
'Particle   K*0         8.9555000e-01   4.7300000e-02',
'Particle   anti-K*0    8.9555000e-01   4.7300000e-02',
'Particle   rho0        7.7526000e-01   1.4910000e-01',
'Particle   phi         1.0194610e+00   4.2490000e-03',
'Particle   B-          5.2793200e+00   0.0000000e+00',
'Particle   B+          5.2793200e+00   0.0000000e+00',
'Particle   B0          5.2796600e+00   0.0000000e+00',
'Particle   anti-B0     5.2796600e+00   0.0000000e+00',
'Particle   B_s0        5.3669200e+00   0.0000000e+00',
'Particle   anti-B_s0   5.3669200e+00   0.0000000e+00',
'Particle   J/psi       3.0969000e+00   9.2600000e-05',
'Particle   psi(2S)     3.6861000e+00   2.9400000e-04',
'#',
'Alias      MyBs    B_s0',
'Alias      Myanti-Bs  anti-B_s0',
'ChargeConj Myanti-Bs   MyBs',
'Alias       MyPsi  psi(2S)',
'ChargeConj  MyPsi  MyPsi',
'Alias       MyJpsi  J/psi',
'ChargeConj  MyJpsi  MyJpsi',
'Decay MyBs',
'1.000     MyPsi K_S0           SVS;',
'Enddecay',
'Decay Myanti-Bs',
'1.000     MyPsi K_S0           SVS;',
'Enddecay',
'Decay MyPsi',
'1.000     J/psi pi+ pi-               VVPIPI;',
'Enddecay',
'Decay MyJpsi',
'1.000  mu+       mu-                     PHOTOS VLL;',
'Enddecay',
'End'
), 
            list_forced_decays = cms.vstring('MyBs','Myanti-Bs'),
            operates_on_particles = cms.vint32(),
            convertPythiaCodes = cms.untracked.bool(False)
            ),
        parameterSets = cms.vstring('EvtGen130')
        ),
        PythiaParameters = cms.PSet(
        pythia8CommonSettingsBlock,
        pythia8CP5SettingsBlock,
        processParameters = cms.vstring(
            "SoftQCD:nonDiffractive = on",
            'PTFilter:filter = on', # this turn on the filter
            'PTFilter:quarkToFilter = 5', # PDG id of q quark
            'PTFilter:scaleToFilter = 1.0'),
        parameterSets = cms.vstring(
            'pythia8CommonSettings',
            'pythia8CP5Settings',
            'processParameters',
        )
    )
)

from GeneratorInterface.Core.ExternalGeneratorFilter import ExternalGeneratorFilter
generator = ExternalGeneratorFilter(_generator)

lbfilter = cms.EDFilter("PythiaFilter", ParticleID = cms.untracked.int32(531))

decayfilter = cms.EDFilter("PythiaDauVFilter",
   verbose         = cms.untracked.int32(0),
   NumberDaughters = cms.untracked.int32(2),
   MotherID        = cms.untracked.int32(0),
   ParticleID      = cms.untracked.int32(531),
   DaughterIDs     = cms.untracked.vint32(100443   , 310   ),
   MinPt           = cms.untracked.vdouble(5.      , 0.3   ),
   MinEta          = cms.untracked.vdouble(-99999.  , -99999.   ),
   MaxEta          = cms.untracked.vdouble( 99999.  , 99999.    )
)

psifilter = cms.EDFilter("PythiaDauVFilter",
        verbose         = cms.untracked.int32(0),
        NumberDaughters = cms.untracked.int32(3),
        MotherID        = cms.untracked.int32(531),
        ParticleID      = cms.untracked.int32(100443),
        DaughterIDs     = cms.untracked.vint32(443, 211, -211),
        MinPt           = cms.untracked.vdouble(5., 0.3, 0.3),
        MinEta          = cms.untracked.vdouble(-99999., -99999., -99999.),
        MaxEta          = cms.untracked.vdouble(99999., 99999., 99999.)
        )

jpsifilter = cms.EDFilter("PythiaDauVFilter",
        verbose         = cms.untracked.int32(0),
        NumberDaughters = cms.untracked.int32(2),
        MotherID        = cms.untracked.int32(100443),
        ParticleID      = cms.untracked.int32(443),
        DaughterIDs     = cms.untracked.vint32(13, -13),
        MinPt           = cms.untracked.vdouble(2.8, 2.8),
        MinEta          = cms.untracked.vdouble(-2.5, -2.5),
        MaxEta          = cms.untracked.vdouble(2.5, 2.5)
        )

ProductionFilterSequence = cms.Sequence(generator*lbfilter*decayfilter*psifilter*jpsifilter)