import FWCore.ParameterSet.Config as cms

# Z2 star tune with pT-ordered showers
#from Configuration.Generator.PythiaUEZ2starSettings_cfi import *
#from GeneratorInterface.ExternalDecays.TauolaSettings_cff import *

generator = cms.EDFilter("Pythia8GeneratorFilter",
    maxEventsToPrint = cms.untracked.int32(10),
    pythiaPylistVerbosity = cms.untracked.int32(1),
    filterEfficiency = cms.untracked.double(1.0),
    pythiaHepMCVerbosity = cms.untracked.bool(True),
    comEnergy = cms.double(8000.),
        PythiaParameters = cms.PSet(
        py8HDecaySettings = cms.vstring(  
                         '25:m0 = 150.0',   # Mass of The Higgs 
                         '23:mMin = 50.0',
                         'HiggsSM:all = off',      # turn OFF all H production mode
                         'HiggsSM:ffbar2HW = on',  # Turn ON WH production mode
                         'HiggsSM:ffbar2HZ = on',  # Turn ON ZH production mode
                         '25:onMode = off', # turn OFF all H decays
                         '25:onIfMatch = 23 22', #turn ON H->Zgamma
                          'Tune:pp = 5',
                         # '23:onMode = off',
                         # '23:onIfAny = 11 13 15', #turn ON Z->ee, mumu, tautau
                         # '24:onMode = off',
                         # '24:onIfAny = 11 13 15', #turn ON W->enu, munu, taunu, 
                        #                 'ParticleDecays:sophisticatedTau = 1',        # interface to Tauola for the tau decay  
                                          'Tune:pp = 5',   # UE tune
                          'PDF:pSet = 7'            
                          
                         ),
        parameterSets = cms.vstring('py8HDecaySettings')
    )
)

# if you need some filter modules define and configure them here
genParticlesForFilter = cms.EDProducer("GenParticleProducer",
    saveBarCodes = cms.untracked.bool(True),
    src = cms.InputTag("generator"),
    abortOnUnknownPDGCode = cms.untracked.bool(False)
)

genSelectorTwoLepFromHiggs = cms.EDFilter("GenParticleSelector",
  filter = cms.bool(True),
  src = cms.InputTag('genParticlesForFilter'),
  cut = cms.string('(abs(pdgId()) == 11 || abs(pdgId()) == 13 || abs(pdgId()) == 15) && mother().pdgId() == 23 && mother().mother().pdgId() == 25'),
)

selectedTwoLepFromHiggsCandFilter = cms.EDFilter("CandViewCountFilter",
   src = cms.InputTag('genSelectorTwoLepFromHiggs'),
   filter = cms.bool(True),
   minNumber = cms.uint32(2)
)
           
ProductionFilterSequence = cms.Sequence(generator * (genParticlesForFilter + genSelectorTwoLepFromHiggs + selectedTwoLepFromHiggsCandFilter))