import FWCore.ParameterSet.Config as cms

from Configuration.Generator.Pythia8CommonSettings_cfi import *
from Configuration.Generator.MCTunes2017.PythiaCP5Settings_cfi import * 
from Configuration.Generator.PSweightsPythia.PythiaPSweightsSettings_cfi import *


generator = cms.EDFilter("Pythia8GeneratorFilter",
        maxEventsToPrint = cms.untracked.int32(1),
        pythiaPylistVerbosity = cms.untracked.int32(1),
        filterEfficiency = cms.untracked.double(1.0),
        pythiaHepMCVerbosity = cms.untracked.bool(False),
        comEnergy = cms.double(13000.0),
    	RandomizedParameters = cms.VPSet(),
    	UserCustomization    = cms.VPSet(cms.PSet(
             pluginName = cms.string("SuepDecay"),
             idDark = cms.int32(999999), # pdgId of the dark meson
             idMediator = cms.int32(25), # pdgId of the mediator
             temperature = cms.double(0.35) # Temperature of the thermal distribution
           )
        ),
)

 
points = [
{'processParameters':['HiggsSM:all = off','HiggsSM:ffbar2HZ = on','25:m0 = 125.000','23:onMode = off','23:onIfAny = 11 13 15','23:mMin = 0.1','999999:all = GeneralResonance void 0 0 0 1.400 0.001 0.0 0.0 0.0','999998:all = GeneralResonance void 1 0 0 0.7 0.001 0.0 0.0 0.0','999999:oneChannel = 1 1.0 101 999998 999998','999998:oneChannel = 1 0.15 101 11 -11','999998:addChannel = 1 0.15 101 13 -13','999998:addChannel = 1 0.7 101 211 -211'],'T':0.35,'name':'SUEP_mS125.000_mPhi1.400_T0.350_modehadronic','filterEfficiency':1.0,'acceptance':0.16}
]  # These are the parameters for all generated points
mcm_eff = 1.000000 # This is the overall mcm efficiency, averaged across all points


for point in points:
   basePythiaParameters = cms.PSet(
            pythia8CommonSettingsBlock,
            pythia8CP5SettingsBlock,
            pythia8PSweightsSettingsBlock,
            processParameters = cms.vstring(point['processParameters']),
            parameterSets = cms.vstring('pythia8CommonSettings',
                                        'pythia8CP5Settings',
                                        'pythia8PSweightsSettings',
                                        'processParameters',
           )
   )
   baseUserCustomization = cms.VPSet(
           cms.PSet(
             pluginName = cms.string("SuepDecay"),
             idDark = cms.int32(999999), # pdgId of the dark meson
             idMediator = cms.int32(25), # pdgId of the mediator
             temperature = cms.double(0.35) # Temperature of the thermal distribution
           )
   )

   generator.RandomizedParameters.append(
        cms.PSet(
            ConfigWeight = cms.double(1./(point['filterEfficiency']*point['acceptance'])), # I.e. bias events so the number of events in SR is independent of point
            ConfigDescription = cms.string(point['name']),
            PythiaParameters = basePythiaParameters,
            UserCustomization = baseUserCustomization,
        ),
    )