# Found 148 output events for 6000 input events.
# Filter efficiency = 0.024667
# Timing = 1.468258 sec/event
# Event size = 711.6 kB/event


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),
#                          maxEventsToPrint = cms.untracked.int32(1),
#                          pythiaPylistVerbosity = cms.untracked.int32(12),
                         ExternalDecays = cms.PSet(
        EvtGen130 = cms.untracked.PSet(
##            decay_table = cms.string('GeneratorInterface/EvtGenInterface/data/DECAY_2010_NOLONGLIFE.DEC'),
            decay_table = cms.string('GeneratorInterface/EvtGenInterface/data/DECAY_2014_NOLONGLIFE.DEC'),
#            particle_property_file = cms.FileInPath('GeneratorInterface/EvtGenInterface/data/evt.pdl'),
            particle_property_file = cms.FileInPath('GeneratorInterface/EvtGenInterface/data/evt_2014.pdl'),
            #user_decay_file = cms.vstring('GeneratorInterface/ExternalDecays/data/Bs_Jpsiphi.dec'),
            user_decay_embedded= cms.vstring(
'#',
'# Particles updated from PDG2021',
'Particle   pi+           1.3957061e-01   0.0000000e+00', ## id 211
'Particle   pi-           1.3957061e-01   0.0000000e+00',
'Particle   mu+           1.0565837e-01   0.0000000e+00', ## id 13
'Particle   mu-           1.0565837e-01   0.0000000e+00',
'Particle   K+            4.9367700e-01   0.0000000e+00', ## id 321
'Particle   K-            4.9367700e-01   0.0000000e+00',
'#',
'Particle   K_S0          4.9761100e-01   0.0000000e+00', ## id 310
'Particle   K*+           8.9167000e-01   5.1400000e-02', ## id 323
'Particle   K*-           8.9167000e-01   5.1400000e-02',
'#',
'Particle   rho0          7.7526000e-01   1.4740000e-01', ## id 113
'Particle   K_10          1.2720000e+00   8.9999999e-02', ## id 10313
'Particle   anti-K_10     1.2720000e+00   8.9999999e-02',
'#',
'Particle   B0            5.2796500e+00   0.0000000e+00', ## id 511
'Particle   anti-B0       5.2796500e+00   0.0000000e+00',
'#',
'Particle   J/psi         3.0969000e+00   9.2900006e-05', ## id 443
'Particle   psi(2S)       3.6860970e+00   2.9400000e-04', ## id 100443
'#',
'#',
'#',
'Alias       MyB        B0',
'Alias       Myanti-B   anti-B0',
'ChargeConj  Myanti-B   MyB',
'#',
'Alias       Mypsi      J/psi',
'ChargeConj  Mypsi      Mypsi',
'#',
'Alias       MyK*+        K*+',
'Alias       MyK*-        K*-',
'ChargeConj  MyK*-        MyK*+',
'#',
'Alias       MyK_10       K_10',
'Alias       Myanti-K_10  anti-K_10',
'ChargeConj  Myanti-K_10  MyK_10',
'#',
'Alias       Myrho0     rho0',
'ChargeConj  Myrho0     Myrho0',
'#',
'Decay Mypsi',
'1.000       mu+    mu-        PHOTOS VLL;',
'Enddecay',
'#',
'Decay Myrho0',
'1.000       pi+    pi-        VSS;',
'Enddecay',
'#',
'Decay MyK*+',
'1.000       K_S0   pi+        VSS;',
'Enddecay',
'CDecay MyK*-',
'#',
'Decay MyK_10',
'0.60       Myrho0 K_S0       VVS_PWAVE 1.0 0.0 0.0 0.0 0.0 0.0;',
'0.40       MyK*+  pi-        VVS_PWAVE 1.0 0.0 0.0 0.0 0.0 0.0;',
'Enddecay',
'CDecay Myanti-K_10',
'#',
'Decay MyB',
'0.50       Mypsi  MyK*+    pi-              PHSP;',
'0.25       Mypsi  Myrho0   K_S0             PHSP;',
'0.25       Mypsi  MyK_10                    SVV_HELAMP 0.5 0.0 1.0 0.0 0.5 0.0;',
'Enddecay',
'CDecay Myanti-B',
'End'
), 
            list_forced_decays = cms.vstring('MyB','Myanti-B'),
            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",
            "511:m0=5.27965",       ## changing also B0 mass in pythia
            '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',
        )
    )
)
_generator.PythiaParameters.processParameters.extend(EvtGenExtraParticles)
from GeneratorInterface.Core.ExternalGeneratorFilter import ExternalGeneratorFilter
generator = ExternalGeneratorFilter(_generator)

###########
# Filters #
###########

#
bfilter = cms.EDFilter("PythiaFilter",
        MaxEta = cms.untracked.double(9999.),
        MinEta = cms.untracked.double(-9999.),
        ParticleID = cms.untracked.int32(511)
)

# psifilter = cms.EDFilter("PythiaFilter",
#         MotherID        = cms.untracked.int32(5122),
#         ParticleID      = cms.untracked.int32(443)
# #         MinPt           = cms.untracked.double(4.95),
# #         MinEta          = cms.untracked.double(-3.0),
# #         MaxEta          = cms.untracked.double( 3.0)
# )
psifilter = cms.EDFilter("PythiaDauVFilter",
        verbose         = cms.untracked.int32(0),
        NumberDaughters = cms.untracked.int32(2),
        MotherID        = cms.untracked.int32(511),
        ParticleID      = cms.untracked.int32(443),
        DaughterIDs     = cms.untracked.vint32(13, -13),
        MinPt           = cms.untracked.vdouble(3., 3.),
        MinEta          = cms.untracked.vdouble(-2.5, -2.5),
        MaxEta          = cms.untracked.vdouble(2.5, 2.5)
)
# decayfilter = cms.EDFilter("PythiaDauVFilter",
# 	    verbose         = cms.untracked.int32(0),
# 	    NumberDaughters = cms.untracked.int32(4),
# 	    MotherID        = cms.untracked.int32(0),
# 	    ParticleID      = cms.untracked.int32(511),
#         DaughterIDs     = cms.untracked.vint32(443, 310, 211, -211),
# 	    MinPt           = cms.untracked.vdouble(5, 0.5, 0.1, 0.1),
# 	    MinEta          = cms.untracked.vdouble(-99999, -99999, -3, -3),
# 	    MaxEta          = cms.untracked.vdouble( 99999,  99999,  3,  3)
# )


# ProductionFilterSequence = cms.Sequence(generator*lbfilter*psifilter)
ProductionFilterSequence = cms.Sequence(generator*bfilter*psifilter)