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


generator = cms.EDFilter("Pythia8GeneratorFilter",
                         pythiaPylistVerbosity = cms.untracked.int32(0),
                         pythiaHepMCVerbosity = cms.untracked.bool(False),
                         comEnergy = cms.double(13000.0),
                         maxEventsToPrint = cms.untracked.int32(0),
                         ExternalDecays = cms.PSet(
        EvtGen130 = cms.untracked.PSet(
            decay_table = cms.string('GeneratorInterface/EvtGenInterface/data/DECAY_2010.DEC'),
            particle_property_file = cms.FileInPath('GeneratorInterface/EvtGenInterface/data/evt.pdl'),
            user_decay_file = cms.vstring('GeneratorInterface/ExternalDecays/data/Ds_tau_mumumu.dec'),
            list_forced_decays = cms.vstring('MyD_s+','MyD_s-'),
            operates_on_particles = cms.vint32()
            ),
        parameterSets = cms.vstring('EvtGen130')
        ),
                         PythiaParameters = cms.PSet(
        pythia8CommonSettingsBlock,
        pythia8CUEP8M1SettingsBlock,
        processParameters = cms.vstring(
            'HardQCD:all = on',
            'PhaseSpace:pTHatMin = 8.',
            ),
        parameterSets = cms.vstring('pythia8CommonSettings',
                                    'pythia8CUEP8M1Settings',
                                    'processParameters',
                                    )
        )
                         )



generator.PythiaParameters.processParameters.extend(EvtGenExtraParticles)

# filter to select events with a Ds
DsFilter = cms.EDFilter("PythiaFilter",
       MaxEta = cms.untracked.double(3),
       MinEta = cms.untracked.double(-3),
       MinPt = cms.untracked.double(7),
       ParticleID = cms.untracked.int32(431)  #D_s
   )




# ask 3 muons in the acceptance

genParticlesForFilter = cms.EDProducer("GenParticleProducer",
                                       saveBarCodes = cms.untracked.bool(True),
                                       src = cms.InputTag("generator"),
                                       abortOnUnknownPDGCode = cms.untracked.bool(False)
)


muonParticlesInAcc = cms.EDFilter("GenParticleSelector",
                                  filter = cms.bool(False),
                                  src = cms.InputTag("genParticlesForFilter"),
                                  cut = cms.string('pt > 1. && abs(pdgId) == 13 && abs(eta) < 2.4'),
                                  stableOnly = cms.bool(True)
                                  )
threeMuonFilter = cms.EDFilter("CandViewCountFilter",
                               src = cms.InputTag("muonParticlesInAcc"),
                               minNumber = cms.uint32(3))


# Production Info
configurationMetadata = cms.untracked.PSet(
    version = cms.untracked.string('$Revision: 1.3 $'),
    annotation = cms.untracked.string('MinBias_Ds_13TeV_pythia8_cff.py nevts:1'),
    name = cms.untracked.string('PyReleaseValidation')
)

ProductionFilterSequence = cms.Sequence(generator * DsFilter * genParticlesForFilter * muonParticlesInAcc * threeMuonFilter)