import FWCore.ParameterSet.Config as cms

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

externalLHEProducer = cms.EDProducer("ExternalLHEProducer",
    args = cms.vstring("/cvmfs/cms.cern.ch/phys_generator/gridpacks/UL/13TeV/madgraph/V5_2.6.5/sus_sms/SMS-SqSq_v2/SMS-SqSq_mSq-1800_slc7_amd64_gcc700_CMSSW_10_6_19_tarball.tar.xz"),
    nEvents = cms.untracked.uint32(5000),
    numberOfParameters = cms.uint32(1),
    outputFile = cms.string('cmsgrid_final.lhe'),
    scriptName = cms.FileInPath('GeneratorInterface/LHEInterface/data/run_generic_tarball_cvmfs.sh'),
    generateConcurrently = cms.untracked.bool(True)
)

import math

baseSLHATable="""
BLOCK MASS
   2000001     1800
   2000002     1800
   2000003     1800
   2000004     1800
   2000005     1.00000000E+05
   2000006     1.00000000E+05
   2000011     1.00000000E+05
   2000013     1.00000000E+05
   2000015     1.00000000E+05
   1000001     1800
   1000002     1800
   1000003     1800
   1000004     1800
   1000005     1.00000000E+05
   1000006     1.00000000E+05
   1000011     1.00000000E+05
   1000012     1.00000000E+05
   1000013     1.00000000E+05
   1000014     1.00000000E+05
   1000015     1.00000000E+05
   1000016     1.00000000E+05
   1000021     1.00000000E+05
   1000022     1.00000000E+00
   1000023     800
   1000024     800
   1000025     1.00000000E+05
   1000035     1.00000000E+05
   1000037     1.00000000E+05
#
DECAY   2000001     1.00000000E+00
     5.00000000E-01    2          1    1000023
     5.00000000E-01    2          2    -1000024
DECAY   2000002     1.00000000E+00
     5.00000000E-01    2          2    1000023
     5.00000000E-01    2          1    1000024
DECAY   2000003     1.00000000E+00
     5.00000000E-01    2          3    1000023
     5.00000000E-01    2          4    -1000024
DECAY   2000004     1.00000000E+00
     5.00000000E-01    2          4    1000023
     5.00000000E-01    2          3    1000024
DECAY   2000005     0.00000000E+00
DECAY   2000006     0.00000000E+00
DECAY   2000011     0.00000000E+00
DECAY   2000013     0.00000000E+00
DECAY   2000015     0.00000000E+00
DECAY   1000001     1.00000000E+00
     5.00000000E-01    2          1    1000023
     5.00000000E-01    2          2    -1000024
DECAY   1000002     1.00000000E+00
     5.00000000E-01    2          2    1000023
     5.00000000E-01    2          1    1000024
DECAY   1000003     1.00000000E+00
     5.00000000E-01    2          3    1000023
     5.00000000E-01    2          4    -1000024
DECAY   1000004     1.00000000E+00
     5.00000000E-01    2          4    1000023
     5.00000000E-01    2          3    1000024
DECAY   1000005     0.00000000E+00
DECAY   1000006     0.00000000E+00
DECAY   1000011     0.00000000E+00
DECAY   1000012     0.00000000E+00
DECAY   1000013     0.00000000E+00
DECAY   1000014     0.00000000E+00
DECAY   1000015     0.00000000E+00
DECAY   1000016     0.00000000E+00
DECAY   1000021     0.00000000E+00
DECAY   1000023     1.00000000E-01
     1.00000000E+00    2         22    1000022
DECAY   1000024     1.00000000E-01
     0.0000000    3     1000022        -1      2
     1.0000000    2     1000022        24
DECAY   1000022     0.00000000E+00
"""


generator = cms.EDFilter("Pythia8ConcurrentHadronizerFilter",
  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.),
  PythiaParameters = cms.PSet(
    pythia8CommonSettingsBlock,
    pythia8CP5SettingsBlock,
    JetMatchingParameters = cms.vstring(
      'JetMatching:setMad = off',
      'JetMatching:scheme = 1',
      'JetMatching:merge = on',
      'JetMatching:jetAlgorithm = 2',
      'JetMatching:etaJetMax = 5.',
      'JetMatching:coneRadius = 1.',
      'JetMatching:slowJetPower = 1',
      'JetMatching:qCut = 74', #this is the actual merging scale
      'JetMatching:nQmatch = 5', #4 corresponds to 4-flavour scheme (no matching of b-quarks), 5 for 5-flavour scheme
      'JetMatching:nJetMax = 2', #number of partons in born matrix element for highest multiplicity
      'JetMatching:doShowerKt = off', #off for MLM matching, turn on for shower-kT matching
      '6:m0 = 172.5',
      '24:mMin = 7',
      'Check:abortIfVeto = on',
    ), 
    parameterSets = cms.vstring('pythia8CommonSettings',
      'pythia8CP5Settings',
      'JetMatchingParameters'
    )
  ),
  SLHATableForPythia8 = cms.string(baseSLHATable),
)