ATLMCPROD-7955

common/MadGraph/MadGraphControl_loopZp2Hy.py

+from MadGraphControl.MadGraphUtils import *
+
+mode=0
+
+# extract dataset short name from filename, should be of the form MC15.999999.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_750_NW.py
+shortname=runArgs.jobConfig[0].split('/')[-1].split('.')[2].split('_')
+# decode dataset short name, should be of the form MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_750_NW (split by '_')
+Y1decay=shortname[2]
+Hdecay=shortname[3]
+resonanceMass=int(shortname[4])
+resonanceWidthPercent=shortname[5]
+
+print 'Y1decay: %s' % Y1decay
+print 'Hdecay: %s' % Hdecay
+print 'resonanceMass: %d' % resonanceMass
+print 'resonanceWidthPercent: %s' % resonanceWidthPercent
+
+#is_qq = runArgs.jobConfig[0].find('qq')
+#is_gg = runArgs.jobConfig[0].find('gg')
+#is_pp = runArgs.jobConfig[0].find('pp')
+
+resonanceWidth = 0
+
+if resonanceWidthPercent == "NW":
+    resonanceWidth = 0.0042  #GeV
+else:
+    resonanceWidthPercent = float(resonanceWidthPercent[1:-1]) * 0.01 #Converting string to percent of mass
+    resonanceWidth = resonanceWidthPercent * resonanceMass
+
+if resonanceWidth == 0:
+    raise RuntimeError("Resonance width not correctly interpreted from JO name")
+
+
+#---------------------------------------------------------------------------------------------------
+# Setting X0 mass and width for param_card.dat
+#---------------------------------------------------------------------------------------------------
+Y1Mass  = {'9906663': '%e # MZp'%resonanceMass}               #Mass
+Y1Decay = {'9906663':'DECAY 9906663 %e # WZp'%resonanceWidth} #Width
+
+
+#---------------------------------------------------------------------------------------------------
+# Generating pp/qq/gg -> Y1 -> Hy  in Zprime model
+#---------------------------------------------------------------------------------------------------
+fcard = open('proc_card_mg5.dat','w')
+if ("bb" in Hdecay):
+    fcard.write("""
+    import model ZpHgamma_UFO
+    define q = u c d s b u~ c~ d~ s~ b~
+    define p = g u c d s b u~ c~ d~ s~ b~
+    define j = g u c d s b u~ c~ d~ s~ b~
+    generate p p > Zp > h a, h > b b~
+    output -f""")
+elif ("tt" in Hdecay):
+    fcard.write("""
+    import model ZpHgamma_UFO
+    define l+ = e+ mu+ ta+
+    define l- = e- mu- ta-
+    define q = u c d s b u~ c~ d~ s~ b~
+    define p = g u c d s b u~ c~ d~ s~ b~
+    define j = g u c d s b u~ c~ d~ s~ b~
+    generate p p > Zp > h a, h > t t~
+    output -f""")
+elif ("ll" in Hdecay):
+    fcard.write("""
+    import model ZpHgamma_UFO
+    define l+ = e+ mu+ ta+
+    define l- = e- mu- ta-
+    define q = u c d s b u~ c~ d~ s~ b~
+    define p = g u c d s b u~ c~ d~ s~ b~
+    define j = g u c d s b u~ c~ d~ s~ b~
+    generate p p > Zp > h a, z > l+ l-
+    output -f""")
+elif ("ww" in Hdecay):
+    fcard.write("""
+    import model ZpHgamma_UFO
+    define l+ = e+ mu+ ta+
+    define l- = e- mu- ta-
+    define q = u c d s b u~ c~ d~ s~ b~
+    define p = g u c d s b u~ c~ d~ s~ b~
+    define j = g u c d s b u~ c~ d~ s~ b~
+    generate p p > Zp > h a, h > w+ w-
+    output -f""")
+elif ("zz" in Hdecay):
+    fcard.write("""
+    import model ZpHgamma_UFO
+    define l+ = e+ mu+ ta+
+    define l- = e- mu- ta-
+    define q = u c d s b u~ c~ d~ s~ b~
+    define p = g u c d s b u~ c~ d~ s~ b~
+    define j = g u c d s b u~ c~ d~ s~ b~
+    generate p p > Zp > h a, h > z z
+    output -f""")
+else:
+    raise RuntimeError("Not a valid decay: need to be bb/tt/ww/zz/ll.")
+fcard.close()
+
+
+beamEnergy=-999
+if hasattr(runArgs,'ecmEnergy'):
+    beamEnergy = runArgs.ecmEnergy / 2.
+else: 
+    raise RuntimeError("No center of mass energy found.")
+
+process_dir = new_process()
+
+
+#---------------------------------------------------------------------------------------------------
+# Setting some parameters for run_card.dat
+#---------------------------------------------------------------------------------------------------
+#Fetch default NLO run_card.dat and set parameters
+extras = {'lhe_version':'3.0', 
+          'cut_decays' :'F', 
+          'pdlabel'    : "'lhapdf'",
+          'lhaid'      : '247000',
+          'sys_pdf'    : 'NNDF23_lo_as_0130_qed',
+          'parton_shower' :'PYTHIA8',
+          'reweight_scale':'.false.',
+          'bwcutoff'   : '1000'}
+
+#---------------------------------------------------------------------------------------------------
+# Setting the number of generated events to 'safefactor' times maxEvents,
+# to avoid crashing due to not having enough events
+# Also putting protection to avoid from crashing when maxEvents=-1
+#---------------------------------------------------------------------------------------------------
+safefactor = 1.1
+nevents    = 5000*safefactor
+if runArgs.maxEvents > 0:
+    nevents=runArgs.maxEvents*safefactor
+
+
+
+#---------------------------------------------------------------------------------------------------
+# Using the helper function from MadGraphControl for setting up the param_card
+# Build a new param_card.dat from an existing one
+#---------------------------------------------------------------------------------------------------
+
+build_param_card(param_card_old=process_dir+'/Cards/param_card.dat',param_card_new='param_card_new.dat',masses=Y1Mass,decays=Y1Decay)
+
+#---------------------------------------------------------------------------------------------------
+# Using the helper function from MadGraphControl for setting up the run_card
+# Build a new run_card.dat from an existing one
+# Using the values given in "extras" above for the selected parameters when setting up the run_card
+# If not set in "extras", default values are used
+#---------------------------------------------------------------------------------------------------
+extras={}
+extras["scale"] = resonanceMass
+extras["dsqrt_q2fact1"] = resonanceMass
+extras["dsqrt_q2fact2"] = resonanceMass
+extras["pdlabel"]='lhapdf'
+extras["lhaid"]=247000
+build_run_card(run_card_old='MadGraph_run_card_HC_LO.dat',
+               run_card_new='run_card.dat',
+               xqcut=0,
+               nevts=nevents,
+               rand_seed=runArgs.randomSeed,
+               beamEnergy=beamEnergy,
+               scalefact=1.0,
+               alpsfact=1.0,
+               extras=extras)
+
+print_cards()
+
+
+runName='run_01'
+
+generate(run_card_loc='run_card.dat',param_card_loc='param_card_new.dat',mode=mode,proc_dir=process_dir,run_name=runName)
+outputDS=arrange_output(run_name=runName,proc_dir=process_dir,outputDS=runName+'._00001.events.tar.gz',lhe_version=3)
+
+#### Shower
+
+include("MC15JobOptions/Pythia8_A14_NNPDF23LO_EvtGen_Common.py")
+include("MC15JobOptions/Pythia8_MadGraph.py")
+
+
+evgenConfig.generators = ["MadGraph", "Pythia8", "EvtGen"]
+evgenConfig.description = 'Zprime to H gamma via fermion loop'
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]
+evgenConfig.contact = ['Shu Li <Shu.Li@cern.ch>']
+
+runArgs.inputGeneratorFile=outputDS

share/DSID450xxx/MC15.450583.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_700_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 700 GeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450584.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_800_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 800 GeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450585.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_900_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 900 GeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450586.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_1000_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 1 TeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450587.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_1500_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 1.5 TeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450588.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_2000_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 2 TeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450589.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_3000_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 3 TeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450590.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_4000_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 4 TeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450591.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_5000_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 5 TeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450592.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_6000_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 6 TeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]

share/DSID450xxx/MC15.450593.MadGraphPythia8EvtGen_A14NNPDF23LO_loopZp2Hy_bby_7000_NW.py

+include('MC15JobOptions/MadGraphControl_loopZp2Hy.py')
+
+evgenConfig.description = "Loop induced Zprime decay model for Spin-1 qq->Zprime->Higgs+gamma 7 TeV narrow width resonance in b-bbar decay model"
+evgenConfig.keywords = ["exotic", "Higgs", "photon", "LO", "bbbar", "spin1"]