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

generator = cms.EDFilter("Pythia8GeneratorFilter",
                         pythiaPylistVerbosity = cms.untracked.int32(0),
                         pythiaHepMCVerbosity = cms.untracked.bool(False),
                         comEnergy = cms.double(13000.0),
                         ##crossSection = cms.untracked.double(54000000000), # Given by PYTHIA after running
                         ##filterEfficiency = cms.untracked.double(0.004), # Given by PYTHIA after running
                         maxEventsToPrint = cms.untracked.int32(0),

                         ExternalDecays = cms.PSet(
                                EvtGen130 = cms.untracked.PSet(
                                    convertPythiaCodes = cms.untracked.bool(False),
                                    decay_table = cms.string('GeneratorInterface/EvtGenInterface/data/DECAY_2014_NOLONGLIFE.DEC'),
                                    list_forced_decays = cms.vstring('theJpsi'),
                                    operates_on_particles = cms.vint32(443,-443),
                                    particle_property_file = cms.FileInPath('GeneratorInterface/EvtGenInterface/data/evt_2014.pdl'),
                                    user_decay_embedded = cms.vstring(
"""
Alias      theJpsi  J/psi
ChargeConj theJpsi  theJpsi
#
Decay theJpsi
  1.00000  mu+        mu-                               PHOTOS VLL ;
Enddecay
#
End
"""
                                        ),
                                    ),
                                parameterSets = cms.vstring('EvtGen130')
                                ),

                        PythiaParameters = cms.PSet(pythia8CommonSettingsBlock,
                                                    pythia8CP5SettingsBlock,
                                                    ## this is original from 2018
                                                    processParameters = cms.vstring( 'SoftQCD:nonDiffractive = on',
                                                                                     'PTFilter:filter = on', # this turn on the filter
                                                                                     'PTFilter:quarkToFilter = 5', # PDG id of q quark (can be any other)
                                                                                     'PTFilter:scaleToFilter = 1.0' ),
#                                                     ## this is what was there in 2012
#                                                     processParameters = cms.vstring(
#                                                         'Bottomonium:all = on', # Quarkonia, MSEL=61
#                                                         ## (sara: i think it was msel = 1), and is the equivalent in pythia6 of softQCD:nonDiffractive = on
#
#                                                         'ParticleDecays:allowPhotonRadiation = on', # Turn on QED FSR
#                                                         ### from the manual:
#                                                         ### Allow or not photon radiations in decays to a lepton pair.
#                                                         ### Note: The current default is to have radiation switched off,
#                                                         ### in order to avoid double-counting of emissions if you link to
#                                                         ### an external QED-radiation program, as is the norm in many collaborations.
#
#                                                         'ParticleDecays:mixB = off',
#                                                         '443:onMode = off', # Turn off J/psi decays
#                                                         '443:onIfMatch = 13 -13' # just let J/psi -> mu+ mu-
#                                                     ),

                                                    parameterSets = cms.vstring('pythia8CommonSettings',
                                                                                'pythia8CP5Settings',
                                                                                'processParameters',
                                                                                )
                                                    )
                        )




###########
# Filters #
###########
bfilter = cms.EDFilter("PythiaFilter",
                       ParticleID = cms.untracked.int32(5)
                       )

jpsifilter = cms.EDFilter("PythiaFilter",
                          Status = cms.untracked.int32(2),
                          MaxEta = cms.untracked.double(20.0),
                          MinEta = cms.untracked.double(-20.0),
                          MinPt = cms.untracked.double(3.0),
                          ParticleID = cms.untracked.int32(443)
                          )


mumugenfilter = cms.EDFilter("MCParticlePairFilter",
                             Status = cms.untracked.vint32(1, 1),
                             MinPt = cms.untracked.vdouble(0.5, 0.5),
                             MinP = cms.untracked.vdouble(0.,0.),
                             MaxEta = cms.untracked.vdouble(2.5, 2.5),
                             MinEta = cms.untracked.vdouble(-2.5, -2.5),
                             ParticleCharge = cms.untracked.int32(-1),
                             MaxInvMass = cms.untracked.double(4.0),
                             MinInvMass = cms.untracked.double(2.0),
                             ParticleID1 = cms.untracked.vint32(13),
                             ParticleID2 = cms.untracked.vint32(13)
                             )

ProductionFilterSequence = cms.Sequence(generator*bfilter*jpsifilter*mumugenfilter)