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

generator = cms.EDFilter("Pythia8HadronizerFilter",
    maxEventsToPrint = cms.untracked.int32(1),
    pythiaPylistVerbosity = cms.untracked.int32(1),
    pythiaHepMCVerbosity = cms.untracked.bool(True),
    comEnergy = cms.double(13000.0),
    PythiaParameters = cms.PSet(
        pythia8CommonSettingsBlock,
        pythia8CP5SettingsBlock,
        processParameters = cms.vstring( # put below any needed pythia parameter
            '541:m0 = 6.27447',
            '541:tau0 = 0.153'
        ),
        parameterSets = cms.vstring(
            'pythia8CommonSettings',
            'pythia8CP5Settings',
            'processParameters',
        ),
    ),

    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'),
            list_forced_decays = cms.vstring(
                'MyBc+',
                'MyBc-',
            ),
            operates_on_particles = cms.vint32(541, -541),
            convertPythiaCodes = cms.untracked.bool(False),
            user_decay_embedded = cms.vstring(
'Particle   pi+           1.3957039e-01   0.0000000e+00',
'Particle   pi-           1.3957039e-01   0.0000000e+00',
'Particle   J/psi         3.0969000e+00   9.2900006e-05', ## 443
'Particle   psi(2S)       3.6860970e+00   2.9400000e-04', ## 100443
'Particle   K+            4.9367700e-01   0.0000000e+00',
'Particle   K-            4.9367700e-01   0.0000000e+00',
'Particle   rho+          7.7511000e-01   1.4910000e-01',
'Particle   rho-          7.7511000e-01   1.4910000e-01',
'Particle   B_c+          6.2744700e+00   0.0000000e+00',
'Particle   B_c-          6.2744700e+00   0.0000000e+00',
'#',
'#',
'Alias      MyBc+            B_c+',
'Alias      MyBc-            B_c-',
'ChargeConj MyBc-            MyBc+',
'#',
'Alias      Mypsi(2S)  psi(2S)', 
'ChargeConj Mypsi(2S)  Mypsi(2S)', 
'#',
'Alias      MyJ/psi  J/psi', 
'ChargeConj MyJ/psi  MyJ/psi', 
'#',
'#',
'#',
'Decay MyJ/psi', 
'1.000         mu+         mu-      PHOTOS VLL;', 
'Enddecay', 
'#',
'Decay Mypsi(2S)', 
'1.000      MyJ/psi    pi+       pi-       VVPIPI;',
'Enddecay', 
'#',
'Decay MyBc+',
'0.708      Mypsi(2S)     pi+     PHOTOS SVS;',
'0.056      Mypsi(2S)     K+      PHOTOS SVS;',
'0.236      Mypsi(2S)     rho+    PHOTOS SVV_HELAMP 1.0 0.0 1.0 0.0 1.0 0.0;',
'Enddecay',
'CDecay MyBc-',
'#',
'End'   
),
        ),
        parameterSets = cms.vstring('EvtGen130'),
    ),

)

generator.PythiaParameters.processParameters.extend(EvtGenExtraParticles)

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.6, 2.6),
  MinEta          = cms.untracked.vdouble(-2.6, -2.6),
  MaxEta          = cms.untracked.vdouble( 2.6,  2.6)
)

psi2sfilter = cms.EDFilter("PythiaDauVFilter",
  verbose         = cms.untracked.int32(0),
  NumberDaughters = cms.untracked.int32(3),
  MotherID        = cms.untracked.int32(541),
  ParticleID      = cms.untracked.int32(100443),
  DaughterIDs     = cms.untracked.vint32(443, 211, -211),
  MinPt           = cms.untracked.vdouble(6.0, 0.35, 0.35),
  MinEta          = cms.untracked.vdouble(-4.0, -3.0, -3.0),
  MaxEta          = cms.untracked.vdouble( 4.0,  3.0,  3.0)
)

ProductionFilterSequence = cms.Sequence(generator*jpsifilter*psi2sfilter)