import FWCore.ParameterSet.Config as cms

from Configuration.Generator.PythiaUEZ2starSettings_cfi import * 

generator = cms.EDFilter("Pythia6GeneratorFilter",
    pythiaPylistVerbosity = cms.untracked.int32(1),
    filterEfficiency = cms.untracked.double(1.0),
    pythiaHepMCVerbosity = cms.untracked.bool(False),
    maxEventsToPrint = cms.untracked.int32(2),
    comEnergy = cms.double(13000.0),    
    ExternalDecays = cms.PSet(
        EvtGen = cms.untracked.PSet(
             operates_on_particles = cms.vint32( 0 ), # 0 (zero) means default list (hardcoded)
                                                      # you can put here the list of particles (PDG IDs)
                                                      # that you want decayed by EvtGen
         use_default_decay = cms.untracked.bool(False),
             decay_table = cms.FileInPath('GeneratorInterface/ExternalDecays/data/DECAY_NOLONGLIFE.DEC'),
              # decay_table = cms.FileInPath('GeneratorInterface/ExternalDecays/data/DECAY.DEC'),
              particle_property_file = cms.FileInPath('GeneratorInterface/ExternalDecays/data/evt.pdl'),
              user_decay_file = cms.FileInPath('GeneratorInterface/ExternalDecays/data/Dstar_D0_K3pi.dec'),
              list_forced_decays = cms.vstring('MyD*+','MyD*-')
              ),
         parameterSets = cms.vstring('EvtGen')
    ),
    PythiaParameters = cms.PSet(
      #process.pythiaUESettingsBlock,
      pythiaUESettingsBlock,
      ccbarSettings= cms.vstring('MSEL=4     ! ccbar '),
      # This is a vector of ParameterSet names to be read, in this order
      parameterSets = cms.vstring('pythiaUESettings','ccbarSettings')
    )    
)


DstarFilter = cms.EDFilter("PythiaMomDauFilter",
    ParticleID = cms.untracked.int32(413),
    DaughterID = cms.untracked.int32(421),
    ChargeConjugation = cms.untracked.bool(True),
    MinEta = cms.untracked.double(-100.),
    MaxEta = cms.untracked.double(100.),
    DaughterIDs = cms.untracked.vint32(421,211),
    NumberDaughters = cms.untracked.int32(2),
    MomMinPt = cms.untracked.double(3.9),
    NumberDescendants = cms.untracked.int32(4),
    DescendantsIDs = cms.untracked.vint32(-321,211,211,-211)
)

ProductionFilterSequence = cms.Sequence(generator*DstarFilter)