scratch_lg.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import weights as weights\n",
    "import pickle\n",
    "import zlib\n",
    "import numpy as np"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "# import a bunch of correction histograms\n",
    "test = weights.extractor()\n",
    "correctionDescriptions = open(\"newCorrectionFiles.txt\").readlines()\n",
    "test.add_weight_sets(correctionDescriptions)\n",
    "test.finalize()\n",
    "test_eval = test.make_evaluator()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "before pickle: [-2.5   -2.    -1.566 -1.444 -0.8    0.     0.8    1.444  1.566  2.\n",
      "  2.5  ] (5, 10)\n",
      "[ 20.  35.  50.  90. 150. 500.] (5, 10)\n",
      "(array([0, 0, 1, 4, 4]), array([1, 1, 5, 9, 9]))\n",
      "[0.95460796 0.95460796 0.969697   1.0199004  1.0199004 ]\n"
     ]
    }
   ],
   "source": [
    "#make some fake pt / eta pairs\n",
    "pts = np.array([-25,25,40,1000,2000])\n",
    "etas = np.array([-1.8,-1.8,0.,2.2,3.0])\n",
    "print 'before pickle:', test_eval['phoScaleFactor_MediumId_POG'](etas,pts)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#let's pickle and zip it\n",
    "trypkl = pickle.dumps(test_eval)\n",
    "print len(trypkl)\n",
    "trypkl = zlib.compress(trypkl)\n",
    "print len(trypkl)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#then undo the whole thing\n",
    "unpickle = pickle.loads(zlib.decompress(trypkl))\n",
    "\n",
    "print 'after pickle :', unpickle['phoScaleFactor_MediumId_POG'](etas,pts)\n",
    "print '             :', unpickle['muScaleFactor_LooseId'](etas,pts)\n",
    "print '             :', unpickle['Z2Jets_kNLO_nominal'](pts)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": false
   },
   "outputs": [],
   "source": [
    "from striped.job import SinglePointStripedSession as Session, IPythonDisplay\n",
    "\n",
    "from histbook import Hist, beside, groupby, below\n",
    "from histbook import bin as hbin\n",
    "import re\n",
    "import uproot\n",
    "import numpy\n",
    "\n",
    "\n",
    "\n",
    "corrDict = {}\n",
    "\n",
    "\n",
    "\n",
    "correctionFiles = open(\"correctionFiles.txt\").readlines()\n",
    "\n",
    "\n",
    "# Load histograms into numpy arrays\n",
    "for f in correctionFiles:\n",
    "  \n",
    "    dumFile = uproot.open(\"correction_files/\"+f.strip())\n",
    "    for i, key in enumerate(dumFile.keys()):\n",
    "        TH1D = \"<class 'uproot.rootio.TH1D'>\"\n",
    "        TH2D = \"<class 'uproot.rootio.TH2D'>\"\n",
    "        TH1F = \"<class 'uproot.rootio.TH1F'>\"\n",
    "        TH2F = \"<class 'uproot.rootio.TH2F'>\"\n",
    "        RootDir = \"<class 'uproot.rootio.ROOTDirectory'>\"\n",
    "        histType = str(type(dumFile[key[:-2]]))\n",
    "        if histType == TH1D or histType == TH2D or histType==TH1F or histType==TH2F:\n",
    "            if  not (\"bound method\" in str(dumFile[key[:-2]].numpy)):\n",
    "                corrDict[f[:-6]+\"_\"+dumFile.keys()[i][:-2]] = dumFile[key[:-2]].numpy\n",
    "            else:\n",
    "                corrDict[f[:-6]+\"_\"+dumFile.keys()[i][:-2]] = dumFile[key[:-2]].numpy()\n",
    "                \n",
    "                \n",
    "        elif histType == RootDir: #means there are subdirectories wihin main directory\n",
    "               for j, key2 in enumerate(dumFile[key[:-2]].keys()):\n",
    "                    histType2 = str(type(dumFile[key[:-2]][key2[:-2]]))\n",
    "                    if histType2 == TH1D or histType2 == TH2D or histType2==TH1F or histTyp2==TH2F:\n",
    "                        if not (\"bound method\" in str(dumFile[key[:-2]][key2[:-2]].numpy)):\n",
    "                            corrDict[f[:-6]+\"_\"+key[:-2]+\"_\"+key2[:-2]] = dumFile[key[:-2]][key2[:-2]].numpy\n",
    "                        else:\n",
    "                            corrDict[f[:-6]+\"_\"+key[:-2]+\"_\"+key2[:-2]] = dumFile[key[:-2]][key2[:-2]].numpy()\n",
    "                            \n",
    "                            \n",
    "        else:\n",
    "        \n",
    "            tempArrX= dumFile[key[:-2]]._fEX\n",
    "            tempArrY= dumFile[key[:-2]]._fEY\n",
    "            corrDict[key[:-2]] = [tempArrX, tempArrY]\n",
    "          \n",
    "    \n",
    "\n",
    "\n",
    "\n",
    "def valueAt(dictionary ,key,x,y = None):\n",
    "    if y ==None:\n",
    "        if not isinstance(dictionary[key][1][0], numpy.float64):\n",
    "            print \"dictionary[\" +key+\"] is  2D\"\n",
    "            return 'NaN'\n",
    "        xBinMin = dictionary[key][1][0]\n",
    "        xBinMax = dictionary[key][1][len(dictionary[key][1])-1]\n",
    "        if x<xBinMin or x> xBinMax:\n",
    "            print \"point out  of range\" + \" (\"+str(xBinMin) + \",\"+ str(xBinMax) +\")\"\n",
    "            return 'NaN'\n",
    "        i = 0\n",
    "\n",
    "        xBin = dictionary[key][1][0]\n",
    "\n",
    "\n",
    "        while(x>xBin):\n",
    "            i = i +1\n",
    "            xBin = dictionary[key][1][i]\n",
    "\n",
    "        i = i-1\n",
    "\n",
    "        return dictionary[key][0][i]\n",
    "    else:\n",
    "        if isinstance(dictionary[key][1][0], numpy.float64):\n",
    "            print \"dictionary[\" +key+\"] is not 2D\"\n",
    "            return 'NaN'\n",
    "        xBinMin = dictionary[key][1][0][0]\n",
    "        xBinMax = dictionary[key][1][0][len(dictionary[key][1][0])-1]\n",
    "        yBinMin = dictionary[key][1][1][0]\n",
    "        yBinMax = dictionary[key][1][1][len(dictionary[key][1][1])-1]\n",
    "        if x<xBinMin or x> xBinMax or y < yBinMin or y > yBinMax:\n",
    "            print \"point out  of range x:(\"+str(xBinMin)+\",\"+str(xBinMax)+\") y:(\"+str(yBinMin)+\",\"+str(yBinMax) +\")\"\n",
    "            return 'NaN'\n",
    "        i = 0\n",
    "        j = 0\n",
    "        xBin = dictionary[key][1][0][0]\n",
    "        yBin = dictionary[key][1][1][0]\n",
    "\n",
    "        while(x>xBin):\n",
    "            i = i +1\n",
    "            xBin = dictionary[key][1][0][i]\n",
    "        while(y>yBin):\n",
    "            j = j +1\n",
    "            yBin = dictionary[key][1][1][j]\n",
    "        i = i-1\n",
    "        j = j-1\n",
    "        return dictionary[key][0][i][j]\n",
    "\n",
    "    \n",
    "def csvToVec(csvFilePath):\n",
    "    p = csvFilePath\n",
    "    f = open(p).readlines()\n",
    "    \n",
    "    retVec = []\n",
    "    for line in f:\n",
    "        line = re.sub(',', '', line)\n",
    "        vec = line.split()\n",
    "        retVec.append(vec)\n",
    "    retVec.pop(0)\n",
    "    for i, e in enumerate(retVec):\n",
    "        retVec[i][0]= int(e[0])\n",
    "        retVec[i][3]=int(e[3])\n",
    "        retVec[i][4]=float(e[4])\n",
    "        retVec[i][5]=float(e[5])\n",
    "        retVec[i][6]=float(e[6])\n",
    "        retVec[i][7]=float(e[7])\n",
    "        retVec[i][8]=float(e[8])\n",
    "        retVec[i][9]=float(e[9])\n",
    "        retVec[i][10]=eval(e[10])\n",
    "        \n",
    "    return retVec\n",
    "\n",
    "    \n",
    "csvVec = csvToVec(\"correction_files/CSVv2_Moriond17_B_H.csv\")\n",
    "\n",
    "\n",
    "job_server = (\"ifdb01.fnal.gov\", 8765)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "bg_datasets = \"\"\"\n",
    "DYJetsToLL_M-50_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToNuNu_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "WJetsToLNu_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "ZJetsToNuNu_HT-100To200_13TeV-madgraph\n",
    "ZJetsToNuNu_HT-200To400_13TeV-madgraph\n",
    "ZJetsToNuNu_HT-400To600_13TeV-madgraph\n",
    "ZJetsToNuNu_HT-600To800_13TeV-madgraph\n",
    "ZJetsToNuNu_HT-800To1200_13TeV-madgraph\n",
    "ZJetsToNuNu_HT-1200To2500_13TeV-madgraph\n",
    "ZJetsToNuNu_HT-2500ToInf_13TeV-madgraph\n",
    "ZJetsToNuNu_HT-600ToInf_13TeV-madgraph\n",
    "DYJetsToLL_M-50_HT-70to100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "DYJetsToLL_M-50_HT-100to200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "DYJetsToLL_M-50_HT-200to400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "DYJetsToLL_M-50_HT-400to600_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "DYJetsToLL_M-50_HT-600toInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "DYJetsToLL_M-50_HT-1200to2500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "DYJetsToLL_M-50_HT-2500toInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "DYJetsToLL_M-50_HT-600to800_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "DYJetsToLL_M-50_HT-800to1200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_HT-70To100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_HT-100To200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_HT-200To400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_HT-400To600_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_HT-600To800_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_HT-800To1200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_HT-1200To2500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_HT-2500ToInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_HT-600ToInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "WJetsToLNu_Pt-100To250_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "WJetsToLNu_Pt-250To400_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "WJetsToLNu_Pt-400To600_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "WJetsToLNu_Pt-600ToInf_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToLL_Pt-50To100_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToLL_Pt-100To250_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToLL_Pt-250To400_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToLL_Pt-400To650_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToLL_Pt-650ToInf_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToNuNu_PtZ-100To250_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToNuNu_PtZ-250To400_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToNuNu_PtZ-400To650_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "DYJetsToNuNu_PtZ-650ToInf_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\n",
    "GJets_HT-40To100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "GJets_HT-100To200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "GJets_HT-200To400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "GJets_HT-400To600_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "GJets_HT-600ToInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "QCD_HT50to100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "QCD_HT100to200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "QCD_HT200to300_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "QCD_HT300to500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "QCD_HT500to700_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "QCD_HT700to1000_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "QCD_HT1000to1500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "QCD_HT1500to2000_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "QCD_HT2000toInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ST_t-channel_antitop_4f_leptonDecays_13TeV-powheg-pythia8\n",
    "ST_t-channel_top_4f_leptonDecays_13TeV-powheg-pythia8\n",
    "ST_tW_top_5f_inclusiveDecays_13TeV-powheg-pythia8_TuneCUETP8M1\n",
    "ST_tW_antitop_5f_inclusiveDecays_13TeV-powheg-pythia8_TuneCUETP8M1\n",
    "ST_t-channel_antitop_4f_inclusiveDecays_13TeV-powhegV2-madspin-pythia8_TuneCUETP8M1\n",
    "ST_t-channel_top_4f_inclusiveDecays_13TeV-powhegV2-madspin-pythia8_TuneCUETP8M1\n",
    "ST_s-channel_4f_leptonDecays_13TeV-amcatnlo-pythia8_TuneCUETP8M1\n",
    "ST_t-channel_5f_leptonDecays_13TeV-amcatnlo-pythia8_TuneCUETP8M1\n",
    "ST_t-channel_top_4f_leptonDecays_13TeV-powheg-pythia8_TuneCUETP8M1\n",
    "TTJets_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "TTJets_DiLept_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "TTJets_SingleLeptFromT_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "TTJets_SingleLeptFromTbar_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "TTJets_SingleLeptFromT_TuneCUETP8M2T4_13TeV-amcatnloFXFX-pythia8\n",
    "TTJets_SingleLeptFromTbar_TuneCUETP8M2T4_13TeV-amcatnloFXFX-pythia8\n",
    "TT_TuneEE5C_13TeV-powheg-herwigpp\n",
    "TTJets_TuneCUETP8M2T4_13TeV-amcatnloFXFX-pythia8\n",
    "TT_TuneCUETP8M2T4_13TeV-powheg-pythia8\n",
    "TT_TuneCUETP8M2T4_13TeV-powheg-isrdown-pythia8\n",
    "TT_TuneCUETP8M2T4_13TeV-powheg-isrup-pythia8\n",
    "TT_TuneCUETP8M2T4down_13TeV-powheg-pythia8\n",
    "TT_TuneCUETP8M2T4up_13TeV-powheg-pythia8\n",
    "TTTo2L2Nu_TuneCUETP8M2_ttHtranche3_13TeV-powheg-pythia8\n",
    "tZq_ll_4f_13TeV-amcatnlo-pythia8\n",
    "tZq_nunu_4f_13TeV-amcatnlo-pythia8_TuneCUETP8M1\n",
    "TTGJets_TuneCUETP8M1_13TeV-amcatnloFXFX-madspin-pythia8\n",
    "TGJets_TuneCUETP8M1_13TeV_amcatnlo_madspin_pythia8\n",
    "EWKZ2Jets_ZToNuNu_13TeV-madgraph-pythia8\n",
    "EWKZ2Jets_ZToLL_M-50_13TeV-madgraph-pythia8\n",
    "EWKWPlus2Jets_WToLNu_M-50_13TeV-madgraph-pythia8\n",
    "EWKWMinus2Jets_WToLNu_M-50_13TeV-madgraph-pythia8\n",
    "WW_TuneCUETP8M1_13TeV-pythia8\n",
    "WZ_TuneCUETP8M1_13TeV-pythia8\n",
    "ZZ_TuneCUETP8M1_13TeV-pythia8\n",
    "WWTo2L2Nu_13TeV-powheg\n",
    "WWTo4Q_13TeV-powheg\n",
    "WWToLNuQQ_13TeV-powheg\n",
    "WZTo1L1Nu2Q_13TeV_amcatnloFXFX_madspin_pythia8\n",
    "WZTo1L3Nu_13TeV_amcatnloFXFX_madspin_pythia8\n",
    "WZTo2L2Q_13TeV_amcatnloFXFX_madspin_pythia8\n",
    "WZTo3LNu_TuneCUETP8M1_13TeV-powheg-pythia8\n",
    "ZZTo2L2Nu_13TeV_powheg_pythia8\n",
    "ZZTo2L2Q_13TeV_amcatnloFXFX_madspin_pythia8\n",
    "ZZTo4L_13TeV_powheg_pythia8\n",
    "GluGluToContinToZZTo2e2nu_13TeV_MCFM701_pythia8\n",
    "GluGluToContinToZZTo2mu2nu_13TeV_MCFM701_pythia8\n",
    "GluGluToContinToZZTo2e2mu_13TeV_MCFM701_pythia8\n",
    "GluGluToContinToZZTo2e2tau_13TeV_MCFM701_pythia8\n",
    "GluGluToContinToZZTo2mu2tau_13TeV_MCFM701_pythia8\n",
    "GluGluToContinToZZTo4e_13TeV_MCFM701_pythia8\n",
    "GluGluToContinToZZTo4mu_13TeV_MCFM701_pythia8\n",
    "GluGluToContinToZZTo4tau_13TeV_MCFM701_pythia8\n",
    "ZH_HToBB_ZToNuNu_M125_13TeV_powheg_pythia8\n",
    "ZH_HToBB_ZToNuNu_M125_13TeV_powheg_pythia8\n",
    "ZH_HToBB_ZToLL_M125_13TeV_powheg_pythia8\n",
    "ZH_HToBB_ZToQQ_M125_13TeV_powheg_pythia8\n",
    "ggZH_HToBB_ZToNuNu_M125_13TeV_powheg_pythia8\n",
    "ggZH_HToBB_ZToLL_M125_13TeV_powheg_pythia8\n",
    "WminusH_HToBB_WToLNu_M125_13TeV_powheg_pythia8\n",
    "WplusH_HToBB_WToLNu_M125_13TeV_powheg_pythia8\n",
    "WplusH_HToBB_WToQQ_M125_13TeV_powheg_pythia8\n",
    "WminusH_HToBB_WToQQ_M125_13TeV_powheg_pythia8\n",
    "WWTo4Q_4f_13TeV_amcatnloFXFX_madspin_pythia8\n",
    "WJetsToQQ_HT180_13TeV-madgraphMLM-pythia8\n",
    "ZZTo4Q_13TeV_amcatnloFXFX_madspin_pythia8\n",
    "DYJetsToQQ_HT180_13TeV-madgraphMLM-pythia8\n",
    "ttHTobb_M125_13TeV_powheg_pythia8\n",
    "ttHTobb_M125_TuneCUETP8M2_ttHtranche3_13TeV-powheg-pythia8\n",
    "GluGluHToBB_M125_13TeV_powheg_pythia8\n",
    "VBFHToBB_M125_13TeV_amcatnlo_pythia8\n",
    "VBFHToBB_M-125_13TeV_powheg_pythia8_weightfix\n",
    "\"\"\".split()\n",
    "signal_samples = \"\"\"\n",
    "TTbarDMJets_pseudoscalar_Mchi-1_Mphi-100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "thq\n",
    "thw\n",
    "GluGlu_HToInvisible_M125_13TeV_powheg_pythia8\n",
    "VBF_HToInvisible_M125_13TeV_powheg_pythia8\n",
    "VBF_HToInvisible_M110_13TeV_powheg_pythia8\n",
    "VBF_HToInvisible_M150_13TeV_powheg_pythia8\n",
    "VBF_HToInvisible_M200_13TeV_powheg_pythia8\n",
    "VBF_HToInvisible_M300_13TeV_powheg_pythia8\n",
    "VBF_HToInvisible_M500_13TeV_powheg_pythia8\n",
    "VBF_HToInvisible_M600_13TeV_powheg_pythia8\n",
    "VBF_HToInvisible_M800_13TeV_powheg_pythia8\n",
    "VBF_HToInvisible_M1000_13TeV_powheg_pythia8\n",
    "GluGlu_HToInvisible_M110_13TeV_powheg_pythia8\n",
    "GluGlu_HToInvisible_M150_13TeV_powheg_pythia8\n",
    "GluGlu_HToInvisible_M200_13TeV_powheg_pythia8\n",
    "GluGlu_HToInvisible_M300_13TeV_powheg_pythia8\n",
    "GluGlu_HToInvisible_M500_13TeV_powheg_pythia8\n",
    "GluGlu_HToInvisible_M600_13TeV_powheg_pythia8\n",
    "GluGlu_HToInvisible_M800_13TeV_powheg_pythia8\n",
    "GluGlu_HToInvisible_M1000_13TeV_powheg_pythia8\n",
    "ZprimeToTTJet_M-500_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTTJet_M-750_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTTJet_M-1000_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTTJet_M-1250_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTTJet_M-1500_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTTJet_M-2000_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTTJet_M-2500_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTTJet_M-3000_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTTJet_M-3500_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTTJet_M-4000_TuneCUETP8M1_13TeV-amcatnlo-pythia8\n",
    "ZprimeToTT_M-1000_W-100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-1000_W-10_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-1000_W-300_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-1250_W-125_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-1250_W-12p5_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-1500_W-150_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-1500_W-15_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-2000_W-200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-2000_W-20_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-2000_W-600_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-2500_W-250_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-2500_W-25_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-3000_W-300_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-3000_W-30_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-3500_W-350_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-3500_W-35_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-4000_W-1200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-4000_W-400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-4000_W-40_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-4500_W-450_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-4500_W-45_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-5000_W-1500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-5000_W-500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-5000_W-50_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-500_W-50_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-500_W-5_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-750_W-75_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "ZprimeToTT_M-750_W-7p5_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\n",
    "thq\n",
    "thw\n",
    "ZprimeToWW_narrow_M-800_13TeV-madgraph\n",
    "ZprimeToWW_narrow_M-1000_13TeV-madgraph\n",
    "ZprimeToWW_narrow_M-1200_13TeV-madgraph\n",
    "ZprimeToWW_narrow_M-1400_13TeV-madgraph\n",
    "ZprimeToWW_narrow_M-1600_13TeV-madgraph\n",
    "ZprimeToWW_narrow_M-1800_13TeV-madgraph\n",
    "ZprimeToWW_narrow_M-2000_13TeV-madgraph\n",
    "ZprimeToWW_narrow_M-2500_13TeV-madgraph\n",
    "ST_tch_DM-scalar_LO-100_1-13_TeV\n",
    "ST_tch_DM-scalar_LO-300_1-13_TeV\n",
    "ST_tch_DM-scalar_LO-500_1-13_TeV\n",
    "ST_tch_DM-scalar_LO-1000_1-13_TeV\n",
    "TT_DM-scalar_LO-300_1-13_TeV\n",
    "TT_DM-scalar_LO-500_1-13_TeV\n",
    "TT_DM-scalar_LO-1000_1-13_TeV\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-1000_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-15_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-20_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-50_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-50_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-50_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-95_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-100_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-100_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-200_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-200_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-200_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-295_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-300_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-300_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-500_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-500_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-500_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-995_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-1000_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-1000_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-1000_MChi-1000_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-1995_MChi-1000_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-2000_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-2000_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-1000_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-300_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10_MChi-1000_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-15_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-20_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-50_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-50_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-50_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-95_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-100_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-100_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-200_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-200_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-200_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-295_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-300_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-300_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-500_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-500_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-500_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-995_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-1000_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-1000_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-1000_MChi-1000_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-1995_MChi-1000_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-2000_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-2000_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-1_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-10_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-50_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-150_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-500_13TeV-madgraph\n",
    "MonoHbb_ZpBaryonic_MZp-10000_MChi-1000_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2750_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2750_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2750_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-3000_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-3000_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-3000_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-3500_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-3500_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-3500_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-4000_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-4000_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-4000_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-300_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-800_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-700_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-600_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-500_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-400_13TeV-madgraph\n",
    "ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-300_13TeV-madgraph\n",
    "TTbarDMJets_scalar_Mchi-10_Mphi-100:(TTbarDMJets_scalar_Mchi-10_Mphi-100,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-10_Mphi-10:(TTbarDMJets_scalar_Mchi-10_Mphi-10,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-10_Mphi-15:(TTbarDMJets_scalar_Mchi-10_Mphi-15,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-10_Mphi-50:(TTbarDMJets_scalar_Mchi-10_Mphi-50,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-1_Mphi-10000:(TTbarDMJets_scalar_Mchi-1_Mphi-10000,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-1_Mphi-100:(TTbarDMJets_scalar_Mchi-1_Mphi-100,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-1_Mphi-10:(TTbarDMJets_scalar_Mchi-1_Mphi-10,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-1_Mphi-200:(TTbarDMJets_scalar_Mchi-1_Mphi-200,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-1_Mphi-20:(TTbarDMJets_scalar_Mchi-1_Mphi-20,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-1_Mphi-300:(TTbarDMJets_scalar_Mchi-1_Mphi-300,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-1_Mphi-500:(TTbarDMJets_scalar_Mchi-1_Mphi-500,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-1_Mphi-50:(TTbarDMJets_scalar_Mchi-1_Mphi-50,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-50_Mphi-10:(TTbarDMJets_scalar_Mchi-50_Mphi-10,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-50_Mphi-200:(TTbarDMJets_scalar_Mchi-50_Mphi-200,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-50_Mphi-300:(TTbarDMJets_scalar_Mchi-50_Mphi-300,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-50_Mphi-50:(TTbarDMJets_scalar_Mchi-50_Mphi-50,MC,1),\n",
    "TTbarDMJets_scalar_Mchi-50_Mphi-95:(TTbarDMJets_scalar_Mchi-50_Mphi-95,MC,1),\n",
    "BBbarDM_LO_MZprime-2500_Mhs-50_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-70_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-70_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-70_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-90_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-90_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-90_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-70_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-50_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-70_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-90_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-70_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-90_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-1000_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-50_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-90_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-70_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-50_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-90_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-50_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-90_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-70_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-50_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-500_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-2000_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-90_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-90_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-90_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-70_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-50_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-90_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-50_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-50_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-90_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-50_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-90_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-50_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-90_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-70_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-70_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-90_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-70_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-50_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-70_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-70_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-50_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-300_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-50_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-50_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-50_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-50_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-70_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-90_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-70_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-300_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-100_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-90_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-2000_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-70_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-3000_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-50_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-90_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-70_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-90_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-70_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-500_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-90_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-70_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-90_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-50_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-50_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-1500_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-90_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-70_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-70_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-50_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-70_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-90_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-50_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-100_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-2500_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-70_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-90_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-90_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-50_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-50_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-50_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-1500_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-90_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-70_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-70_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-70_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-70_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-90_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-90_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-50_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-50_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-70_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-50_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-50_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-2500_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-90_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2500_Mhs-50_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-1000_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-70_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-70_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-50_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-500_Mhs-50_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1500_Mhs-70_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-3000_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-50_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-1000_Mhs-90_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-2000_Mhs-90_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-90_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-90_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-90_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-90_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-90_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-90_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-90_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-70_Mchi-100_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-70_Mchi-150_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-70_Mchi-200_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-70_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-70_Mchi-300_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-70_Mchi-400_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-3000_Mhs-70_Mchi-50_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-495_Mhs-50_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-495_Mhs-70_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "BBbarDM_LO_MZprime-495_Mhs-90_Mchi-250_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-50_Mhs-150_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "DiJetsDM_LO_MZprime-50_Mhs-50_Mchi-10_gSM-0p25_gDM-1p0_th_0p01_13TeV-madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_300_Mchi_1_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_300_Mchi_200_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_300_Mchi_500_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_300_Mchi_1000_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1000_Mchi_1000_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1000_Mchi_1_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1000_Mchi_200_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1000_Mchi_490_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1500_Mchi_1000_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1500_Mchi_1_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1500_Mchi_200_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1500_Mchi_500_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1750_Mchi_1000_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1750_Mchi_1_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1750_Mchi_200_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_1750_Mchi_500_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_2000_Mchi_1_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_2000_Mchi_200_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_2000_Mchi_500_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_2000_Mchi_990_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_2250_Mchi_1_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_2500_Mchi_1000_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_2500_Mchi_1_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_2500_Mchi_200_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_2500_Mchi_500_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_500_Mchi_1000_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_500_Mchi_1_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_500_Mchi_500_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_750_Mchi_1000_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_750_Mchi_1_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_750_Mchi_200_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "Vector_MonoTop_Leptonic_NLO_Mphi_750_Mchi_500_gSM_0p25_gDM_1p0_13TeV_madgraph\n",
    "\"\"\".split()\n",
    "    \n",
    "\n",
    "\n",
    "bg_datasets = [ds.strip() for ds in bg_datasets if ds.strip()]\n",
    "signal_samples = [ds.strip() for ds in signal_samples if ds.strip()]\n",
    "def category(ds):\n",
    "        \n",
    "         \n",
    "      \n",
    "        if \"DYJetsToLL_M-50_TuneCUETP8M1\" in ds:  return \"ZJets_nlo\"\n",
    "        if \"DYJetsToNuNu_TuneCUETP8M1\" in ds:  return \"ZtoNuNu_nlo\"\n",
    "        if \"WJetsToLNu_TuneCUETP8M1\" in ds:  return \"WJets_nlo\"\n",
    "        if \"ZJetsToNuNu_HT-100To200\" in ds:  return \"ZtoNuNu_ht100to200\"\n",
    "        if \"ZJetsToNuNu_HT-200To400\" in ds:  return \"ZtoNuNu_ht200to400\"\n",
    "        if \"ZJetsToNuNu_HT-400To600\" in ds:  return \"ZtoNuNu_ht400to600\"\n",
    "        if \"ZJetsToNuNu_HT-600To800\" in ds:  return \"ZtoNuNu_ht600to800\"\n",
    "        if \"ZJetsToNuNu_HT-800To1200\" in ds:  return \"ZtoNuNu_ht800to1200\"\n",
    "        if \"ZJetsToNuNu_HT-1200To2500\" in ds:  return \"ZtoNuNu_ht1200to2500\"\n",
    "        if \" ZJetsToNuNu_HT-2500ToInf\" in ds:  return \"ZtoNuNu_ht2500to inf\"\n",
    "        if \" ZJetsToNuNu_HT-600ToInf\" in ds:  return \"ZtoNuNu_ht600to inf\"     \n",
    "        if \"DYJetsToLL_M-50_HT-70to100\" in ds:  return \"ZJets_ht50to100\"\n",
    "        if \"DYJetsToLL_M-50_HT-100to200\" in ds:  return \"ZJets_ht100to200\"\n",
    "        if \"DYJetsToLL_M-50_HT-200to400\" in ds:  return \"ZJets_ht200to400\"\n",
    "        if \"DYJetsToLL_M-50_HT-400to600\" in ds:  return \"ZJets_ht400to600\"\n",
    "        if \"DYJetsToLL_M-50_HT-600toInf\" in ds:  return \"ZJets_ht600to inf\"\n",
    "        if \"DYJetsToLL_M-50_HT-1200to2500\" in ds:  return \"ZJets_ht1200to2500\"\n",
    "        if \"DYJetsToLL_M-50_HT-2500toInf\" in ds:  return \"ZJets_ht2500to inf\"\n",
    "        if \"DYJetsToLL_M-50_HT-600to800\" in ds:  return \"ZJets_ht600to800\"\n",
    "        if \"DYJetsToLL_M-50_HT-800to1200\" in ds:  return \"ZJets_ht800to1200\"\n",
    "        if \"WJetsToLNu_HT-70To100\" in ds:  return \"WJets_ht70to100\"\n",
    "        if \"WJetsToLNu_HT-100To200\" in ds:  return \"WJets_ht100to200\"\n",
    "        if \"WJetsToLNu_HT-200To400\" in ds:  return \"WJets_ht200to400\"\n",
    "        if \"WJetsToLNu_HT-400To600\" in ds:  return \"WJets_ht400to600\"\n",
    "        if \"WJetsToLNu_HT-600To800\" in ds:  return \"WJets_ht600to800\"\n",
    "        if \"WJetsToLNu_HT-800To1200\" in ds:  return \"WJets_ht800to1200\"\n",
    "        if \"WJetsToLNu_HT-1200To2500\" in ds:  return \"WJets_ht1200to2500\"\n",
    "        if \"WJetsToLNu_HT-2500ToInf\" in ds:  return \"WJets_ht2500to inf\"\n",
    "        if \"WJetsToLNu_HT-600ToInf\" in ds:  return \"WJets_ht600to inf\"\n",
    "        if \"WJetsToLNu_Pt-100To250\" in ds:  return \"WJets_pt100to250\"\n",
    "        if \"WJetsToLNu_Pt-250To400\" in ds:  return \"WJets_pt250to400\"\n",
    "        if \"WJetsToLNu_Pt-400To600\" in ds:  return \"WJets_pt400to600\"\n",
    "        if \"WJetsToLNu_Pt-600ToInf\" in ds:  return \"WJets_pt600to inf\"\n",
    "        if \"DYJetsToLL_Pt-50To100\" in ds:  return \"ZJets_pt50to100\"\n",
    "        if \"DYJetsToLL_Pt-100To250\" in ds:  return \"ZJets_pt100to250\"\n",
    "        if \"DYJetsToLL_Pt-250To400\" in ds:  return \"ZJets_pt250to400\"\n",
    "        if \"DYJetsToLL_Pt-400To650\" in ds:  return \"ZJets_pt400to650\"\n",
    "        if \"DYJetsToLL_Pt-650ToInf\" in ds:  return \"ZJets_pt650to inf\"\n",
    "        if \"DYJetsToNuNu_PtZ-100To250\" in ds:  return \"ZtoNuNu_pt100to250\"\n",
    "        if \"DYJetsToNuNu_PtZ-250To400\" in ds:  return \"ZtoNuNu_pt250to400\"\n",
    "        if \"DYJetsToNuNu_PtZ-400To650\" in ds:  return \"ZtoNuNu_pt400to650\"\n",
    "        if \"DYJetsToNuNu_PtZ-650ToInf\" in ds:  return \"ZtoNuNu_pt650to inf\"\n",
    "        if \"GJets_HT-40To100\" in ds:  return \"GJets_ht40to100\"\n",
    "        if \"GJets_HT-100To200\" in ds:  return \"GJets_ht100to200\"\n",
    "        if \"GJets_HT-200To400\" in ds:  return \"GJets_ht200to400\"\n",
    "        if \"GJets_HT-400To600\" in ds:  return \"GJets_ht400to600\"\n",
    "        if \"GJets_HT-600ToInf\" in ds:  return \"GJets_ht600to inf\"\n",
    "        if \"QCD_HT50to100\" in ds:  return \"QCD_ht50to100\"\n",
    "        if \"QCD_HT100to200\" in ds:  return \"QCD_ht100to200\"\n",
    "        if \"QCD_HT200to300\" in ds:  return \"QCD_ht200to300\"\n",
    "        if \"QCD_HT300to500\" in ds:  return \"QCD_ht300to500\"\n",
    "        if \"QCD_HT500to700\" in ds:  return \"QCD_ht500to700\"\n",
    "        if \"QCD_HT700to1000\" in ds: return \"QCD_ht700to1000\"\n",
    "        if \"QCD_HT1000to1500\" in ds: return \"QCD_ht1000to1500\"\n",
    "        if \"QCD_HT1500to2000\" in ds: return \"QCD_ht1500to2000\"\n",
    "        if \"QCD_HT2000toInf\" in ds:  return \"QCD_ht2000to inf\"\n",
    "        if \"WW_TuneCUETP8M1\" in ds:  return \"Diboson_ww\"\n",
    "        if \"WZ_TuneCUETP8M1\" in ds:  return \"Diboson_wz\"\n",
    "        if \"ZZ_TuneCUETP8M1\" in ds:  return \"Diboson_zz\"\n",
    "        if \"WWTo2L2Nu\" in ds:  return \"WWTo2L2Nu\"\n",
    "        if \"WWTo4Q\" in ds:  return \"WWTo4Q\"\n",
    "        if \"WWToLNuQQ\" in ds:  return \"WWToLNuQQ\"\n",
    "        if \"WZTo1L1Nu2Q\"  in ds:  return \"WZTo1L1Nu2Q\"\n",
    "        if \"WZTo1L3Nu\"  in ds:  return \"WZTo1L3Nu\"\n",
    "        if \"WZTo2L2Q\" in ds:  return \"WZTo2L2Q\"\n",
    "        if \"WZTo3LNu\" in ds:  return \"WZTo3LNu\"\n",
    "        if \"ZZTo2L2Nu\" in ds:  return \"ZZTo2L2Nu\"\n",
    "        if \"ZZTo2L2Q\"  in ds:  return \"ZZTo2L2Q\"\n",
    "        if \"ZZTo4L\" in ds:  return \"ZZTo4L\"\n",
    "        if  \"ZH_HToBB_ZToNuNu\" in ds:  return \"ZnunuH\"\n",
    "        if \"ZH_HToBB_ZToNuNu\" in ds:  return \"ZnunuH\"\n",
    "        if  \"ZH_HToBB_ZToLL\" in ds:  return \"ZllH\"\n",
    "        if\"ZH_HToBB_ZToQQ\" in ds:  return \"ZqqH\"\n",
    "        if\"ggZH_HToBB_ZToNuNu\" in ds:  return \"ggZnunuH\"\n",
    "        if\"ggZH_HToBB_ZToLL\" in ds:  return \"ggZllH\"\n",
    "        if\"WminusH_HToBB_WToLNu\" in ds:  return \"WmH\"\n",
    "        if \"WplusH_HToBB_WToLNu\" in ds:  return \"WpH\"\n",
    "        if\"WplusH_HToBB_WToQQ\" in ds:  return \"WpH_qq\"\n",
    "        if\"WminusH_HToBB_WToQQ\" in ds:  return \"WmH_qq\"\n",
    "        if \"WWTo4Q_4f\" in ds:  return \"WW_4q\"\n",
    "        if \"WJetsToQQ_HT180\" in ds:  return \"WJets_qq\"\n",
    "        if\"ZZTo4Q\" in ds:  return \"ZZ_4q\"\n",
    "        if \"DYJetsToQQ_HT180\" in ds:  return \"ZJets_qq\"\n",
    "        if \"ttHTobb\" in ds:  return \"ttH\"\n",
    "        if \"ttHTobb\" in ds:  return \"ttHtranche3\"\n",
    "        if \"GluGluHToBB\" in ds:  return \"ggH\"\n",
    "        if \"VBFHToBB_M125_13TeV_amcatnlo\" in ds:  return \"VBFH\"\n",
    "        if \"VBFHToBB_M-125_13TeV_powheg\" in ds:  return \"VBFH_powheg\"\n",
    "        if \"GluGluToCont inToZZTo2e2nu\" in ds:  return \"ggZZTo2e2nu\"\n",
    "        if \"GluGluToCont inToZZTo2mu2nu\" in ds:  return \"ggZZTo2mu2tau\"\n",
    "        if \"GluGluToContinToZZTo2e2mu\" in ds:  return \"ggZZTo2e2mu\"\n",
    "        if \"GluGluToContinToZZTo2e2tau\" in ds:  return \"ggZZTo2e2tau\"\n",
    "        if \"GluGluToContinToZZTo2mu2tau\" in ds:  return \"ggZZTo2mu2tau\"\n",
    "        if \"GluGluToContinToZZTo4e\" in ds:  return \"ggZZTo4e\"\n",
    "        if \"GluGluToContinToZZTo4mu\" in ds:  return \"ggZZTo4mu\"\n",
    "        if \"GluGluToContinToZZTo4tau\" in ds:  return \"ggZZTo4tau\"\n",
    "        if \"ST_tW_top_5f\" in ds:  return \"SingleTop_tW\"\n",
    "        if \"ST_tW_antitop_5f\" in ds:  return \"SingleTop_tbarW\"\n",
    "        if \"ST_t-channel_antitop_4f_leptonDecays\" in ds: return \"SingleTop_tTbar_lep\"\n",
    "        if \"ST_t-channel_top_4f_leptonDecays\" in ds and  not (\"TuneCUETP8M1\" in ds): return \"SingleTop_tT_lep\"\n",
    "        if \"ST_tW_top_5f\" in ds: return \"SingleTop_tW\"\n",
    "        if \"ST_tW_antitop_5f\" in ds: return  \"SingleTop_tbar\"\n",
    "        if \"ST_t-channel_antitop_4f\"  in ds: return \"SingleTop_tTbar\"\n",
    "        if \"ST_t-channel_top_4f_inclusiveDecays\" in ds: return \"SingleTop_tT\"\n",
    "        if \"ST_s-channel_4f_leptonDecays\" in ds: return \"SingleTop_s_lep\"\n",
    "        if \"ST_t-channel_5f_leptonDecays\" in ds: return \"SingleTop_t_lep\"\n",
    "        if \"ST_t-channel_top_4f_leptonDecays\" in ds: return \"SingleTop_tchannel\"\n",
    "        if \"TTJets_TuneCUETP8M1\"    in ds: return  \"TTbar_MLM\"\n",
    "        if \"TTJets_DiLept_TuneCUETP8M1\"   in ds: return  \"TTbar_2L\"\n",
    "        if \"TTJets_SingleLeptFromT_TuneCUETP8M1\"  in ds: return \"TTbar_1LT\"\n",
    "        if \"TTJets_SingleLeptFromTbar_TuneCUETP8M1\" in ds: return \"TTbar_1LTbar\"\n",
    "        if \"TTJets_SingleLeptFromT_TuneCUETP8M2T4\" in ds: return \"TTbar_FXFX_1LT\"\n",
    "        if \"TTJets_SingleLeptFromTbar_TuneCUETP8M2T4\" in ds: return \"TTbar_FXFX_1LTbar\"\n",
    "        if \"TT_TuneEE5C_13TeV-powheg-herwigpp\" in ds: return \"TTbar_Herwig\"\n",
    "        if \"TTJets_TuneCUETP8M2T4_13TeV-amcatnloFXFX-pythia8\"   in ds: return     \"TTbar_FXFX\"\n",
    "        if \"TT_TuneCUETP8M2T4_13TeV-powheg-pythia8\"  in ds: return  \"TTbar_Powheg\"\n",
    "        if \"TT_TuneCUETP8M2T4_13TeV-powheg-isrdown\" in ds: return  \"TTbar_PowhegISRDown\"\n",
    "        if \"TT_TuneCUETP8M2T4_13TeV-powheg-isrup\"   in ds: return  \"TTbar_PowhegISRUp\"\n",
    "        if \"TT_TuneCUETP8M2T4down\"  in ds: return   \"TTbar_PowhegTuneDown\"\n",
    "        if \"TT_TuneCUETP8M2T4up\"    in ds: return   \"TTbar_PowhegTuneUp\"\n",
    "        if \"TTTo2L2Nu_TuneCUETP8M2_ttHtranche3\" in ds: return \"TTTo2L2Nu\"\n",
    "        if \"EWKZ2Jets_ZToNuNu\" in ds:  return \"ZtoNuNu_EWK\"\n",
    "        if \"EWKZ2Jets_ZToLL\" in ds:  return \"ZJets_EWK\"\n",
    "        if \"EWKWPlus2Jets_WToLNu\" in ds:  return \"WJets_EWKWPlus\"\n",
    "        if \"EWKWMinus2Jets_WToLNu\" in ds:  return \"WJets_EWKWMinus\" \n",
    "        if \"tZq_ll_4f\" in ds:  return \"SingleTop_tZll\"\n",
    "        if \"tZq_nunu_4f\" in ds:  return \"SingleTop_tZnunu\"\n",
    "        if \"TTGJets_TuneCUETP8M1\" in ds:  return \"TTbar_GJets\"\n",
    "        if \"TGJets_TuneCUETP8M1\" in ds:  return \"SingleTop_tG\"\n",
    "\n",
    "        \n",
    "def luminosity(ds):\n",
    "        if \"thq\" in ds: return 0.7927\n",
    "        if \"thw\" in ds: return 0.147\n",
    "        if \"DYJetsToLL_M-50_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 6025.2\n",
    "        if \"DYJetsToNuNu_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 11433.\n",
    "        if \"WJetsToLNu_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 61527.\n",
    "        if \"ZJetsToNuNu_HT-100To200_13TeV-madgraph\" in ds: return 280.5\n",
    "        if \"ZJetsToNuNu_HT-200To400_13TeV-madgraph\" in ds: return 77.7\n",
    "        if \"ZJetsToNuNu_HT-400To600_13TeV-madgraph\" in ds: return 10.71\n",
    "        if \"ZJetsToNuNu_HT-600To800_13TeV-madgraph\" in ds: return 2.562\n",
    "        if \"ZJetsToNuNu_HT-800To1200_13TeV-madgraph\" in ds: return 1.183\n",
    "        if \"ZJetsToNuNu_HT-1200To2500_13TeV-madgraph\" in ds: return 0.286\n",
    "        if \"ZJetsToNuNu_HT-2500ToInf_13TeV-madgraph\" in ds: return 0.006945\n",
    "        if \"ZJetsToNuNu_HT-600ToInf_13TeV-madgraph\" in ds: return 4.098\n",
    "        if \"DYJetsToLL_M-50_HT-70to100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 175.3\n",
    "        if \"DYJetsToLL_M-50_HT-100to200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 148.\n",
    "        if \"DYJetsToLL_M-50_HT-200to400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 40.94\n",
    "        if \"DYJetsToLL_M-50_HT-400to600_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 5.497\n",
    "        if \"DYJetsToLL_M-50_HT-600toInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 2.193\n",
    "        if \"DYJetsToLL_M-50_HT-1200to2500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 0.1514\n",
    "        if \"DYJetsToLL_M-50_HT-2500toInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 0.003565\n",
    "        if \"DYJetsToLL_M-50_HT-600to800_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 1.367\n",
    "        if \"DYJetsToLL_M-50_HT-800to1200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 0.6304\n",
    "        if \"WJetsToLNu_HT-70To100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 1319\n",
    "        if \"WJetsToLNu_HT-100To200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 1343\n",
    "        if \"WJetsToLNu_HT-200To400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 359.6\n",
    "        if \"WJetsToLNu_HT-400To600_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 48.85\n",
    "        if \"WJetsToLNu_HT-600To800_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 12.05\n",
    "        if \"WJetsToLNu_HT-800To1200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 5.501\n",
    "        if \"WJetsToLNu_HT-1200To2500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 1.329\n",
    "        if \"WJetsToLNu_HT-2500ToInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 0.03216\n",
    "        if \"WJetsToLNu_HT-600ToInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 18.91\n",
    "        if \"WJetsToLNu_Pt-100To250_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 677.82\n",
    "        if \"WJetsToLNu_Pt-250To400_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 24.083\n",
    "        if \"WJetsToLNu_Pt-400To600_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 3.0563\n",
    "        if \"WJetsToLNu_Pt-600ToInf_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 0.4602\n",
    "        if \"DYJetsToLL_Pt-50To100_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 374.6800\n",
    "        if \"DYJetsToLL_Pt-100To250_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 86.5200\n",
    "        if \"DYJetsToLL_Pt-250To400_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 3.3247\n",
    "        if \"DYJetsToLL_Pt-400To650_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 0.4491\n",
    "        if \"DYJetsToLL_Pt-650ToInf_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 0.0422\n",
    "        if \"DYJetsToNuNu_PtZ-100To250_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 3*54.8229\n",
    "        if \"DYJetsToNuNu_PtZ-250To400_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 3*2.0705\n",
    "        if \"DYJetsToNuNu_PtZ-400To650_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 3*0.2779\n",
    "        if \"DYJetsToNuNu_PtZ-650ToInf_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8\" in ds: return 3*0.0261                             \n",
    "        if \"GJets_HT-40To100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 23080.0\n",
    "        if \"GJets_HT-100To200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 9235\n",
    "        if \"GJets_HT-200To400_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 2298\n",
    "        if \"GJets_HT-400To600_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 277.6\n",
    "        if \"GJets_HT-600ToInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 93.47  \n",
    "        if \"QCD_HT50to100_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 278700000\n",
    "        if \"QCD_HT100to200_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 27990000\n",
    "        if \"QCD_HT200to300_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 1735000\n",
    "        if \"QCD_HT300to500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 366800\n",
    "        if \"QCD_HT500to700_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 29370\n",
    "        if \"QCD_HT700to1000_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 6524\n",
    "        if \"QCD_HT1000to1500_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 1064\n",
    "        if \"QCD_HT1500to2000_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 121.5\n",
    "        if \"QCD_HT2000toInf_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 25.42  \n",
    "        if \"ST_t-channel_antitop_4f_leptonDecays_13TeV-powheg-pythia8\" in ds: return 26.22\n",
    "        if \"ST_t-channel_top_4f_leptonDecays_13TeV-powheg-pythia8\" in ds: return 44.07\n",
    "        if \"ST_tW_top_5f_inclusiveDecays_13TeV-powheg-pythia8_TuneCUETP8M1\" in ds: return 35.85\n",
    "        if \"ST_tW_antitop_5f_inclusiveDecays_13TeV-powheg-pythia8_TuneCUETP8M1\" in ds: return 35.85\n",
    "        if \"ST_t-channel_antitop_4f_inclusiveDecays_13TeV-powhegV2-madspin-pythia8_TuneCUETP8M1\" in ds: return 80.95\n",
    "        if \"ST_t-channel_top_4f_inclusiveDecays_13TeV-powhegV2-madspin-pythia8_TuneCUETP8M1\" in ds: return 136.02\n",
    "        if \"ST_s-channel_4f_leptonDecays_13TeV-amcatnlo-pythia8_TuneCUETP8M1\" in ds: return 10.11\n",
    "        if \"ST_t-channel_5f_leptonDecays_13TeV-amcatnlo-pythia8_TuneCUETP8M1\" in ds: return 216.99\n",
    "        if \"ST_t-channel_top_4f_leptonDecays_13TeV-powheg-pythia8_TuneCUETP8M1\" in ds: return 44.3\n",
    "        if \"tZq_ll_4f_13TeV-amcatnlo-pythia8\" in ds: return 0.0758\n",
    "        if \"tZq_nunu_4f_13TeV-amcatnlo-pythia8_TuneCUETP8M1\" in ds: return 0.1379\n",
    "        if \"TTGJets_TuneCUETP8M1_13TeV-amcatnloFXFX-madspin-pythia8\" in ds: return 3.786\n",
    "        if \"TGJets_TuneCUETP8M1_13TeV_amcatnlo_madspin_pythia8\" in ds: return 2.967       \n",
    "        if \"EWKZ2Jets_ZToNuNu_13TeV-madgraph-pythia8\" in ds: return 10.04\n",
    "        if \"EWKZ2Jets_ZToLL_M-50_13TeV-madgraph-pythia8\" in ds: return 3.99\n",
    "        if \"EWKWPlus2Jets_WToLNu_M-50_13TeV-madgraph-pythia8\" in ds: return 25.81\n",
    "        if \"EWKWMinus2Jets_WToLNu_M-50_13TeV-madgraph-pythia8\" in ds: return 20.35       \n",
    "        if \"WW_TuneCUETP8M1_13TeV-pythia8\" in ds: return 118.7\n",
    "        if \"WZ_TuneCUETP8M1_13TeV-pythia8\" in ds: return 47.13\n",
    "        if \"ZZ_TuneCUETP8M1_13TeV-pythia8\" in ds: return 16.523      \n",
    "        if \"ZH_HToBB_ZToNuNu_M125_13TeV_powheg_pythia8\" in ds: return 0.08912\n",
    "        if \"ZH_HToBB_ZToNuNu_M125_13TeV_powheg_pythia8\" in ds: return 0.884*0.582*0.20\n",
    "        if \"ZH_HToBB_ZToLL_M125_13TeV_powheg_pythia8\" in ds: return 0.04865\n",
    "        if \"ZH_HToBB_ZToQQ_M125_13TeV_powheg_pythia8\" in ds: return 0.607*0.582\n",
    "        if \"ggZH_HToBB_ZToNuNu_M125_13TeV_powheg_pythia8\" in ds: return 0.014366\n",
    "        if \"ggZH_HToBB_ZToLL_M125_13TeV_powheg_pythia8\" in ds: return 0.007842\n",
    "        if \"WminusH_HToBB_WToLNu_M125_13TeV_powheg_pythia8\" in ds: return 0.100\n",
    "        if \"WplusH_HToBB_WToLNu_M125_13TeV_powheg_pythia8\" in ds: return 0.159\n",
    "        if \"WplusH_HToBB_WToQQ_M125_13TeV_powheg_pythia8\" in ds: return 0.84*(1-3*0.108)*0.582\n",
    "        if \"WminusH_HToBB_WToQQ_M125_13TeV_powheg_pythia8\" in ds: return 0.533*(1-3*0.108)*0.582\n",
    "        if \"WWTo4Q_4f_13TeV_amcatnloFXFX_madspin_pythia8\" in ds: return 51.723\n",
    "        if \"WJetsToQQ_HT180_13TeV-madgraphMLM-pythia8\" in ds: return 2788\n",
    "        if \"ZZTo4Q_13TeV_amcatnloFXFX_madspin_pythia8\" in ds: return 6.842\n",
    "        if \"DYJetsToQQ_HT180_13TeV-madgraphMLM-pythia8\" in ds: return 1187\n",
    "        if \"ttHTobb_M125_13TeV_powheg_pythia8\" in ds: return 0.506*0.5824\n",
    "        if \"ttHTobb_M125_TuneCUETP8M2_ttHtranche3_13TeV-powheg-pythia8\" in ds: return 0.506*0.5824\n",
    "        if \"GluGluHToBB_M125_13TeV_powheg_pythia8\" in ds: return 48.48*0.5824\n",
    "        if \"VBFHToBB_M125_13TeV_amcatnlo_pythia8\" in ds: return 3.782*0.5824\n",
    "        if \"VBFHToBB_M-125_13TeV_powheg_pythia8_weightfix\" in ds: return 3.782*0.5824     \n",
    "        if \"TTJets_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 831.76\n",
    "        if \"TTJets_DiLept_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 831.76*(1-0.68)*(1-0.68)\n",
    "        if \"TTJets_SingleLeptFromT_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 831.76*0.68*(1-0.68)\n",
    "        if \"TTJets_SingleLeptFromTbar_TuneCUETP8M1_13TeV-madgraphMLM-pythia8\" in ds: return 831.76*0.68*(1-0.68)\n",
    "        if \"TTJets_SingleLeptFromT_TuneCUETP8M2T4_13TeV-amcatnloFXFX-pythia8\" in ds: return 831.76*0.68*(1-0.68)\n",
    "        if \"TTJets_SingleLeptFromTbar_TuneCUETP8M2T4_13TeV-amcatnloFXFX-pythia8\" in ds: return 831.76*0.68*(1-0.68)\n",
    "        if \"TT_TuneEE5C_13TeV-powheg-herwigpp\" in ds: return 831.76\n",
    "        if \"TTJets_TuneCUETP8M2T4_13TeV-amcatnloFXFX-pythia8\" in ds: return 831.76\n",
    "        if \"TT_TuneCUETP8M2T4_13TeV-powheg-pythia8\" in ds: return 831.76\n",
    "        if \"TT_TuneCUETP8M2T4_13TeV-powheg-isrdown-pythia8\" in ds: return 831.76\n",
    "        if \"TT_TuneCUETP8M2T4_13TeV-powheg-isrup-pythia8\" in ds: return 831.76\n",
    "        if \"TT_TuneCUETP8M2T4down_13TeV-powheg-pythia8\" in ds: return 831.76\n",
    "        if \"TT_TuneCUETP8M2T4up_13TeV-powheg-pythia8\" in ds: return 831.76\n",
    "        if \"TTTo2L2Nu_TuneCUETP8M2_ttHtranche3_13TeV-powheg-pythia8\" in ds: return 88.288           \n",
    "        if \"WWTo2L2Nu_13TeV-powheg\" in ds: return (118.7-3.974)*0.1086*0.1086*9\n",
    "        if \"WWTo4Q_13TeV-powheg\" in ds: return 51.723\n",
    "        if \"WWToLNuQQ_13TeV-powheg\" in ds: return 49.997\n",
    "        if \"WZTo1L1Nu2Q_13TeV_amcatnloFXFX_madspin_pythia8\" in ds: return 10.71\n",
    "        if \"WZTo1L3Nu_13TeV_amcatnloFXFX_madspin_pythia8\" in ds: return 3.033\n",
    "        if \"WZTo2L2Q_13TeV_amcatnloFXFX_madspin_pythia8\" in ds: return 5.595\n",
    "        if \"WZTo3LNu_TuneCUETP8M1_13TeV-powheg-pythia8\" in ds: return 4.430\n",
    "        if \"ZZTo2L2Nu_13TeV_powheg_pythia8\" in ds: return 0.5644\n",
    "        if \"ZZTo2L2Q_13TeV_amcatnloFXFX_madspin_pythia8\" in ds: return 3.22\n",
    "        if \"ZZTo4L_13TeV_powheg_pythia8\" in ds: return 1.212\n",
    "        if \"GluGluToContinToZZTo2e2nu_13TeV_MCFM701_pythia8\" in ds: return 0.003956\n",
    "        if \"GluGluToContinToZZTo2mu2nu_13TeV_MCFM701_pythia8\" in ds: return 0.003956\n",
    "        if \"GluGluToContinToZZTo2e2mu_13TeV_MCFM701_pythia8\" in ds: return 0.0073462\n",
    "        if \"GluGluToContinToZZTo2e2tau_13TeV_MCFM701_pythia8\" in ds: return 0.0073462\n",
    "        if \"GluGluToContinToZZTo2mu2tau_13TeV_MCFM701_pythia8\" in ds: return 0.0073462\n",
    "        if \"GluGluToContinToZZTo4e_13TeV_MCFM701_pythia8\" in ds: return 0.0036478\n",
    "        if \"GluGluToContinToZZTo4mu_13TeV_MCFM701_pythia8\" in ds: return 0.0036478\n",
    "        if \"GluGluToContinToZZTo4tau_13TeV_MCFM701_pythia8\" in ds: return 0.0036478       \n",
    "        if \"GluGlu_HToInvisible_M125_13TeV_powheg_pythia8\" in ds: return 48.6\n",
    "        if \"VBF_HToInvisible_M125_13TeV_powheg_pythia8\" in ds: return 3.78\n",
    "        if \"VBF_HToInvisible_M110_13TeV_powheg_pythia8\" in ds: return 4.34\n",
    "        if \"VBF_HToInvisible_M150_13TeV_powheg_pythia8\" in ds: return 3.239\n",
    "        if \"VBF_HToInvisible_M200_13TeV_powheg_pythia8\" in ds: return 2.282\n",
    "        if \"VBF_HToInvisible_M300_13TeV_powheg_pythia8\" in ds: return 1.256\n",
    "        if \"VBF_HToInvisible_M500_13TeV_powheg_pythia8\" in ds: return 0.4872\n",
    "        if \"VBF_HToInvisible_M600_13TeV_powheg_pythia8\" in ds: return 0.3274\n",
    "        if \"VBF_HToInvisible_M800_13TeV_powheg_pythia8\" in ds: return 0.1622\n",
    "        if \"VBF_HToInvisible_M1000_13TeV_powheg_pythia8\" in ds: return 0.08732\n",
    "        if \"GluGlu_HToInvisible_M110_13TeV_powheg_pythia8\" in ds: return 57.90\n",
    "        if \"GluGlu_HToInvisible_M150_13TeV_powheg_pythia8\" in ds: return 31.29\n",
    "        if \"GluGlu_HToInvisible_M200_13TeV_powheg_pythia8\" in ds: return 16.94\n",
    "        if \"GluGlu_HToInvisible_M300_13TeV_powheg_pythia8\" in ds: return 6.590\n",
    "        if \"GluGlu_HToInvisible_M500_13TeV_powheg_pythia8\" in ds: return 1.709\n",
    "        if \"GluGlu_HToInvisible_M600_13TeV_powheg_pythia8\" in ds: return 1.001\n",
    "        if \"GluGlu_HToInvisible_M800_13TeV_powheg_pythia8\" in ds: return 0.4015\n",
    "        if \"GluGlu_HToInvisible_M1000_13TeV_powheg_pythia8\" in ds: return 0.1845\n",
    "        if \"thq\" in ds: return 0.7927\n",
    "        if \"thw\" in ds: return 0.147\n",
    "        if \"ST_tch_DM-scalar_LO-100_1-13_TeV\" in ds: return 0.293*0.68\n",
    "        if \"ST_tch_DM-scalar_LO-300_1-13_TeV\" in ds: return 0.03202*0.68\n",
    "        if \"ST_tch_DM-scalar_LO-500_1-13_TeV\" in ds: return 0.004996*0.68\n",
    "        if \"ST_tch_DM-scalar_LO-1000_1-13_TeV\" in ds: return 0.0003009*0.68\n",
    "        if \"TT_DM-scalar_LO-300_1-13_TeV\" in ds: return 0.03045\n",
    "        if \"TT_DM-scalar_LO-500_1-13_TeV\" in ds: return 0.004947\n",
    "        if \"TT_DM-scalar_LO-1000_1-13_TeV\" in ds: return 0.000736\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-1_13TeV-madgraph\" in ds: return 1.4971719048020\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-10_13TeV-madgraph\" in ds: return 0.0068827725015\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-50_13TeV-madgraph\" in ds: return 0.0001182107976\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-150_13TeV-madgraph\" in ds: return 0.0000102513973\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-500_13TeV-madgraph\" in ds: return 0.0000000026006\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-1000_13TeV-madgraph\" in ds: return 0.0000000000160\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-15_MChi-10_13TeV-madgraph\" in ds: return 0.0241243264082\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-20_MChi-1_13TeV-madgraph\" in ds: return 1.5708303614837\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-50_MChi-1_13TeV-madgraph\" in ds: return 1.8787525219888\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-50_MChi-10_13TeV-madgraph\" in ds: return 1.8686472610676\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-50_MChi-50_13TeV-madgraph\" in ds: return 0.0041445173799\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-95_MChi-50_13TeV-madgraph\" in ds: return 0.1070881411920\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-100_MChi-1_13TeV-madgraph\" in ds: return 1.8350014490170\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-100_MChi-10_13TeV-madgraph\" in ds: return 1.8311374853540\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-200_MChi-1_13TeV-madgraph\" in ds: return 1.4723542212437\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-200_MChi-50_13TeV-madgraph\" in ds: return 1.2073355514877\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-200_MChi-150_13TeV-madgraph\" in ds: return 0.0018402642784\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-295_MChi-150_13TeV-madgraph\" in ds: return 0.0772562835636\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-300_MChi-1_13TeV-madgraph\" in ds: return 1.3081597528204\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-300_MChi-50_13TeV-madgraph\" in ds: return 1.1234969709678\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-500_MChi-1_13TeV-madgraph\" in ds: return 0.6310480800692\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-500_MChi-150_13TeV-madgraph\" in ds: return 0.3922022706199\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-500_MChi-500_13TeV-madgraph\" in ds: return 0.0000118690907\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-995_MChi-500_13TeV-madgraph\" in ds: return 0.0000000466117\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-1000_MChi-1_13TeV-madgraph\" in ds: return 0.1164211478301\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-1000_MChi-150_13TeV-madgraph\" in ds: return 0.1130981455808\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-1000_MChi-1000_13TeV-madgraph\" in ds: return 0.0000003182419\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-1995_MChi-1000_13TeV-madgraph\" in ds: return 0.0004210309298\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-2000_MChi-1_13TeV-madgraph\" in ds: return 0.0080403926209\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-2000_MChi-500_13TeV-madgraph\" in ds: return 0.0073116588302\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-1_13TeV-madgraph\" in ds: return 0.0000000068756\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-10_13TeV-madgraph\" in ds: return 0.0000000068844\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-50_13TeV-madgraph\" in ds: return 0.0000000068310\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-150_13TeV-madgraph\" in ds: return 0.0000000066844\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-500_13TeV-madgraph\" in ds: return 0.0000000055185\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-1000_13TeV-madgraph\" in ds: return 0.0000000038070\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-600_13TeV-madgraph\" in ds: return 0.010101*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-500_13TeV-madgraph\" in ds: return 0.0367*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-400_13TeV-madgraph\" in ds: return 0.091124*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-300_13TeV-madgraph\" in ds: return 0.27765*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-400_13TeV-madgraph\" in ds: return 0.063916*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-300_13TeV-madgraph\" in ds: return 0.45127*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-800_13TeV-madgraph\" in ds: return 0.0007806*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-700_13TeV-madgraph\" in ds: return 0.0008699*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-600_13TeV-madgraph\" in ds: return 0.0009678*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-500_13TeV-madgraph\" in ds: return 0.0010859*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-400_13TeV-madgraph\" in ds: return 0.0013246*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-300_13TeV-madgraph\" in ds: return 0.0025458*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-800_13TeV-madgraph\" in ds: return 0.0020981*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-700_13TeV-madgraph\" in ds: return 0.0024651*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-600_13TeV-madgraph\" in ds: return 0.0028691*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-500_13TeV-madgraph\" in ds: return 0.003341*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-400_13TeV-madgraph\" in ds: return 0.0041934*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-300_13TeV-madgraph\" in ds: return 0.0082317*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-800_13TeV-madgraph\" in ds: return 0.0036537*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-700_13TeV-madgraph\" in ds: return 0.0045823*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-600_13TeV-madgraph\" in ds: return 0.0056149*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-500_13TeV-madgraph\" in ds: return 0.0068221*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-400_13TeV-madgraph\" in ds: return 0.0088466*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-300_13TeV-madgraph\" in ds: return 0.017786*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-800_13TeV-madgraph\" in ds: return 0.0055221*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-700_13TeV-madgraph\" in ds: return 0.0079659*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-600_13TeV-madgraph\" in ds: return 0.010822*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-500_13TeV-madgraph\" in ds: return 0.014225*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-400_13TeV-madgraph\" in ds: return 0.019609*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-300_13TeV-madgraph\" in ds: return 0.041208*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-800_13TeV-madgraph\" in ds: return 0.005655*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-700_13TeV-madgraph\" in ds: return 0.010028*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-600_13TeV-madgraph\" in ds: return 0.015763*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-500_13TeV-madgraph\" in ds: return 0.022935*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-400_13TeV-madgraph\" in ds: return 0.034039*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-300_13TeV-madgraph\" in ds: return 0.075451*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-800_13TeV-madgraph\" in ds: return 0.0027483*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-700_13TeV-madgraph\" in ds: return 0.0085776*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-600_13TeV-madgraph\" in ds: return 0.01897*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-500_13TeV-madgraph\" in ds: return 0.03431*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-400_13TeV-madgraph\" in ds: return 0.058824*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-300_13TeV-madgraph\" in ds: return 0.14383*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-1_13TeV-madgraph\" in ds: return 2.594752001*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-10_13TeV-madgraph\" in ds: return 0.0119285485*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-50_13TeV-madgraph\" in ds: return 0.00020487139962*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-150_13TeV-madgraph\" in ds: return 3.14857697501e-06*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-500_13TeV-madgraph\" in ds: return 4.5071850285e-09*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10_MChi-1000_13TeV-madgraph\" in ds: return 2.7728742863e-11*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-15_MChi-10_13TeV-madgraph\" in ds: return 0.041809924451*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-20_MChi-1_13TeV-madgraph\" in ds: return 2.72240963862*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-50_MChi-1_13TeV-madgraph\" in ds: return 3.25607022875*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-50_MChi-10_13TeV-madgraph\" in ds: return 3.23855677828*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-50_MChi-50_13TeV-madgraph\" in ds: return 0.00718287240877*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-95_MChi-50_13TeV-madgraph\" in ds: return 0.185594698773*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-100_MChi-1_13TeV-madgraph\" in ds: return 3.18024514561*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-100_MChi-10_13TeV-madgraph\" in ds: return 3.17354850148*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-200_MChi-1_13TeV-madgraph\" in ds: return 2.5517404181*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-200_MChi-50_13TeV-madgraph\" in ds: return 2.09243596445*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-200_MChi-150_13TeV-madgraph\" in ds: return 0.00318936616712*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-295_MChi-150_13TeV-madgraph\" in ds: return 0.133893039105*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-300_MChi-1_13TeV-madgraph\" in ds: return 2.26717461494*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-300_MChi-50_13TeV-madgraph\" in ds: return 1.94713513166*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-500_MChi-1_13TeV-madgraph\" in ds: return 1.09367084934*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-500_MChi-150_13TeV-madgraph\" in ds: return 0.679726638856*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-500_MChi-500_13TeV-madgraph\" in ds: return 2.05703477766e-05*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-995_MChi-500_13TeV-madgraph\" in ds: return 0.0111215826866*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-1000_MChi-1_13TeV-madgraph\" in ds: return 0.201769753605*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-1000_MChi-150_13TeV-madgraph\" in ds: return 0.19601065092*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-1000_MChi-1000_13TeV-madgraph\" in ds: return 5.51545710471e-07*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-1995_MChi-1000_13TeV-madgraph\" in ds: return 0.000729689653045*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-2000_MChi-1_13TeV-madgraph\" in ds: return 0.0139348225665*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-2000_MChi-500_13TeV-madgraph\" in ds: return 0.012671852392*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-1_13TeV-madgraph\" in ds: return 1.19161364309e-08*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-10_13TeV-madgraph\" in ds: return 1.19314052769e-08*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-50_13TeV-madgraph\" in ds: return 1.18388857948e-08*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-150_13TeV-madgraph\" in ds: return 1.15848042141e-08*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-500_13TeV-madgraph\" in ds: return 9.56417074962e-09*0.578\n",
    "        if \"MonoHbb_ZpBaryonic_MZp-10000_MChi-1000_13TeV-madgraph\" in ds: return 6.59793071766e-09*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2750_MA0-300_13TeV-madgraph\" in ds: return 0.0015511*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2750_MA0-500_13TeV-madgraph\" in ds: return 0.00067388*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2750_MA0-800_13TeV-madgraph\" in ds: return 0.00050351*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-3000_MA0-300_13TeV-madgraph\" in ds: return 0.00092077*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-3000_MA0-500_13TeV-madgraph\" in ds: return 0.0004022*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-3000_MA0-800_13TeV-madgraph\" in ds: return 0.00030894*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-3500_MA0-300_13TeV-madgraph\" in ds: return 0.00033718*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-3500_MA0-500_13TeV-madgraph\" in ds: return 0.00014885*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-3500_MA0-800_13TeV-madgraph\" in ds: return 0.00011855*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-4000_MA0-300_13TeV-madgraph\" in ds: return 0.00012964*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-4000_MA0-500_13TeV-madgraph\" in ds: return 5.7149e-05*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-4000_MA0-800_13TeV-madgraph\" in ds: return 4.6312e-05*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-600_13TeV-madgraph\" in ds: return 0.010350*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-500_13TeV-madgraph\" in ds: return 0.037616*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-400_13TeV-madgraph\" in ds: return 0.093376*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-800_MA0-300_13TeV-madgraph\" in ds: return 0.2848*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-400_13TeV-madgraph\" in ds: return 0.065307*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-300_13TeV-madgraph\" in ds: return 0.46223*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-800_13TeV-madgraph\" in ds: return 0.00082449*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-700_13TeV-madgraph\" in ds: return 0.00091718*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-600_13TeV-madgraph\" in ds: return 0.0010187*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-500_13TeV-madgraph\" in ds: return 0.0011446*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-400_13TeV-madgraph\" in ds: return 0.0013922*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2500_MA0-300_13TeV-madgraph\" in ds: return 0.0026787*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-800_13TeV-madgraph\" in ds: return 0.0021931*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-700_13TeV-madgraph\" in ds: return 0.0025816*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-600_13TeV-madgraph\" in ds: return 0.0029991*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-500_13TeV-madgraph\" in ds: return 0.0034998*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-400_13TeV-madgraph\" in ds: return 0.0043899*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-2000_MA0-300_13TeV-madgraph\" in ds: return 0.0086008*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-800_13TeV-madgraph\" in ds: return 0.0038043*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-700_13TeV-madgraph\" in ds: return 0.0047758*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-600_13TeV-madgraph\" in ds: return 0.0058468*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-500_13TeV-madgraph\" in ds: return 0.0071071*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-400_13TeV-madgraph\" in ds: return 0.0092082*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1700_MA0-300_13TeV-madgraph\" in ds: return 0.018545*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-800_13TeV-madgraph\" in ds: return 0.0057179*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-700_13TeV-madgraph\" in ds: return 0.0082681*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-600_13TeV-madgraph\" in ds: return 0.011224*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-500_13TeV-madgraph\" in ds: return 0.014723*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-400_13TeV-madgraph\" in ds: return 0.020306*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1400_MA0-300_13TeV-madgraph\" in ds: return 0.042687*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-800_13TeV-madgraph\" in ds: return 0.0058353*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-700_13TeV-madgraph\" in ds: return 0.010364*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-600_13TeV-madgraph\" in ds: return 0.016254*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-500_13TeV-madgraph\" in ds: return 0.023706*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-400_13TeV-madgraph\" in ds: return 0.03524*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1200_MA0-300_13TeV-madgraph\" in ds: return 0.078164*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-800_13TeV-madgraph\" in ds: return 0.00283922*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-700_13TeV-madgraph\" in ds: return 0.00881814*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-600_13TeV-madgraph\" in ds: return 0.019542*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-500_13TeV-madgraph\" in ds: return 0.035347*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-400_13TeV-madgraph\" in ds: return 0.060531*0.578\n",
    "        if \"ZprimeToA0hToA0chichihbb_2HDM_MZp-1000_MA0-300_13TeV-madgraph\" in ds: return 0.14771*0.578\n",
    "\n",
    "        return 1.0\n",
    "\n",
    "\n",
    "datasets = [(ds, category(ds)) for ds in bg_datasets + signal_samples]\n",
    "luminosities = [(ds,luminosity(ds)) for ds in  bg_datasets]\n",
    "\n",
    "\n",
    "\n",
    "#for ds, cat in datasets: print ds, cat\n",
    "#for ds, lum in luminosities: print ds, lum\n",
    "    \n",
    "test_dataset =[[\"ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-300_13TeV-madgraph\",\"a\"],[category(\"ZprimeToA0hToA0chichihbb_2HDM_MZp-600_MA0-300_13TeV-madgraph\"),\"b\"]]\n",
    "\n",
    "\n",
    "\n",
    "#test_dataset = [ds.strip() for ds in test_dataset if ds.strip]\n",
    "session = Session(job_server)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#__worker_class__\n",
    "\n",
    "import numpy as np\n",
    "\n",
    "def p4_pt(p4):\n",
    "    # p4 is [n,4] numpy array\n",
    "    # returns [n] numpy array\n",
    "    return np.sqrt(p4[:,0]**2 + p4[:,1]**2)\n",
    "\n",
    "def p4_eta(p4):\n",
    "    p42 = p4**2\n",
    "    p3 = np.sqrt(np.sum(p42[:,:3], axis=-1))\n",
    "    return p4[:,2]/p3\n",
    "\n",
    "def valueAt(dictionary ,key,x,y = None):\n",
    "    if y ==None:     # if 1D hist\n",
    "        if not isinstance(dictionary[key][1][0], np.float64):\n",
    "            print \"dictionary[\" +key+\"] is  2D\"\n",
    "            return \"dictionary[\" +key+\"] is  2D\"\n",
    "        xBinMin = dictionary[key][1][0]\n",
    "        xBinMax = dictionary[key][1][len(dictionary[key][1])-2]\n",
    "        if x<xBinMin or x> xBinMax:\n",
    "            print \"point out  of range\" + \" (\"+str(xBinMin) + \",\"+ str(xBinMax) +\")\"\n",
    "            return 1\n",
    "        i = 0\n",
    "\n",
    "        xBin = dictionary[key][1][0]\n",
    "\n",
    "\n",
    "        while(x>xBin and i < len(dictionary[key][1])-2):\n",
    "            i = i +1\n",
    "            xBin = dictionary[key][1][i]\n",
    "\n",
    "        i = i-1\n",
    "        \n",
    "        return dictionary[key][0][i] \n",
    "    else:  # if 2d hist\n",
    "    \n",
    "        if isinstance(dictionary[key][1][0], np.float64):\n",
    "           # print \"dictionary[\" +key+\"] is not 2D\"\n",
    "            return \"dictionary[\" +key+\"] is not 2D\"\n",
    "        xBinMin = dictionary[key][1][0][0]\n",
    "        xBinMax = dictionary[key][1][0][len(dictionary[key][1][0])-1]\n",
    "        yBinMin = dictionary[key][1][1][0]\n",
    "        yBinMax = dictionary[key][1][1][len(dictionary[key][1][1])-1]\n",
    "        #if x<xBinMin or x> xBinMax or y < yBinMin or y > yBinMax:\n",
    "            #print \"point out  of range x:(\"+str(xBinMin)+\",\"+str(xBinMax)+\") y:(\"+str(yBinMin)+\",\"+str(yBinMax) +\")\"\n",
    "            #return x,xBinMin,xBinMax,\"aaa\",y,yBinMin,yBinMax#\"point out  of range x:(\"+str(xBinMin)+\",\"+str(xBinMax)+\") y:(\"+str(yBinMin)+\",\"+str(yBinMax) +\")\"\n",
    "            \n",
    "            #return 1\n",
    "        i = 0\n",
    "        j = 0\n",
    "        xBin = dictionary[key][1][0][0]\n",
    "        yBin = dictionary[key][1][1][0]\n",
    "        \n",
    "        while(x>xBin and i< len(dictionary[key][1][0])-2):\n",
    "            i = i +1\n",
    "            xBin = dictionary[key][1][0][i]\n",
    "        while(y>yBin and j < len(dictionary[key][1][1])-2):\n",
    "            j = j +1\n",
    "            yBin = dictionary[key][1][1][j]\n",
    "        i = i-1\n",
    "        j = j-1\n",
    "        if x <=xBinMin:\n",
    "            i = 0\n",
    "        if x >=xBinMax:\n",
    "            i = len(dictionary[key][1][0])-2\n",
    "        if y < yBinMin:\n",
    "            j = 0\n",
    "        if y >= yBinMax:\n",
    "            j = len(dictionary[key][1][1])-2\n",
    "        return dictionary[key][0][j][i]\n",
    "\n",
    "def makeSfSpecificVec(csvVec,OP, mType,sysType, flavor):\n",
    "    OPVec = []\n",
    "    for i,e in enumerate(csvVec):\n",
    "        if e[0] == OP and e[1] ==mType and e[2]==sysType and e[3] == flavor:\n",
    "            e.pop(0)\n",
    "            e.pop(0)\n",
    "            e.pop(0)\n",
    "            e.pop(0)\n",
    "            OPVec.append(e)\n",
    "    \n",
    "    return OPVec\n",
    "\n",
    "def calcSFBtag(sfCalcVec, eta, pt):\n",
    "    ret = 0\n",
    "    if pt < sfCalcVec[0][2]:\n",
    "        pt = sfCalcVec[0][2]+0.0001\n",
    "    if pt > sfCalcVec[len(sfCalcVec)-1][3]:\n",
    "        pt = sfCalcVec[len(sfCalcVec)-1][3] -0.0001\n",
    "    for e in sfCalcVec:\n",
    "        if pt > e[2] and pt < e[3]:\n",
    "            x = pt\n",
    "            ret = eval(e[6])\n",
    "            return ret\n",
    "        \n",
    "\n",
    "\n",
    "def binPt(pt):\n",
    "    vbtagpt = [20.0,50.0,80.0,120.0,200.0,300.0,400.0,500.0,700.0,1000.0]\n",
    "    if pt < vbtagpt[0]:\n",
    "        return 0\n",
    "    elif pt > vbtagpt[len(vbtagpt)-1]:\n",
    "        return len(vbtagpt)-2\n",
    "    else:\n",
    "        i = 0\n",
    "        while pt > vbtagpt[i]:\n",
    "            i = i +1\n",
    "    return i - 1\n",
    "\n",
    "\n",
    "def binEta(pt):\n",
    "    vbtageta = [0.0,0.5,1.5,2.5]\n",
    "    if pt < vbtageta[0]:\n",
    "        return 0\n",
    "    elif pt > vbtageta[len(vbtageta)-1]:\n",
    "        return len(vbtageta)-2\n",
    "    else:\n",
    "        i = 0\n",
    "        while pt > vbtageta[i]:\n",
    "            i = i +1\n",
    "    return i - 1\n",
    "\n",
    "def btagEff(eta,pt):\n",
    "\n",
    "    beff = [[0.791,0.815,0.825,0.835,0.821,0.799,0.784,0.767,0.760],\n",
    "          [0.794,0.816,0.829,0.836,0.823,0.804,0.798,0.792,0.789],\n",
    "          [0.739,0.767,0.780,0.789,0.776,0.771,0.779,0.787,0.806]]\n",
    "    return beff[binEta(eta)][binPt(pt)]\n",
    "\n",
    "class Worker(object):\n",
    "    '''\n",
    "    Columns = [\"Weight\", \"NMuons\", \"HT\", \"MHT\", \"NElectrons\", \"NJets\", \"BTags\",\n",
    "            \"isoElectronTracks\", \"isoMuonTracks\", \"isoPionTracks\",\n",
    "            \"DeltaPhi1\", \"DeltaPhi2\", \"DeltaPhi3\", \"DeltaPhi4\",\n",
    "            \"JetID\", \"PFCaloMETRatio\",\"Trigger.pass\",\n",
    "            \"NVtx\", \"eeBadScFilter\", \"HBHENoiseFilter\",\n",
    "            \"HBHEIsoNoiseFilter\", \"globalTightHalo2016Filter\",\n",
    "            \"EcalDeadCellTriggerPrimitiveFilter\", \"BadChargedCandidateFilter\",\n",
    "            \"BadPFMuonFilter\",\n",
    "             \"Jet.p4\"]\n",
    "    '''\n",
    "    Columns = [\"weight\",\"muons.pt_\",\"muons.eta_\",\"muons.phi_\",\"muons_size\",\"muons.phIso\",\"muons.chIso\",\n",
    "               \"muons.nhIso\",\"muons.puIso\", \"npvTrue\",\"muons.loose\",\"muons.tight\", \"electrons.loose\", \n",
    "               \"electrons.veto\",\"electrons.tight\",\"electrons.pt_\", \"electrons.eta_\", \"electrons_size\",\n",
    "               \"ak4GenJets.eta_\",\"ak4GenJets.pt_\",\"npv\",\"genParticles.packedPt\", \n",
    "               \"genParticles.packedPhi\",\"genParticles.packedEta\", \"genParticles.parent_\",\n",
    "               \"genParticles.finalState\",\"genParticles.pdgid\",\"genParticles.statusFlags\",\"genParticles.finalState\",\n",
    "               \"photons.csafeVeto\", \"photons.loose\", \"photons.pt_\", \"photons.eta_\",\"photons.tight\" ]\n",
    "    def run(self, events, job):\n",
    "        \n",
    "        category = job[\"category\"]\n",
    "        lumi = job[\"lumi\"]\n",
    "        xsec_weight = lumi*events.weight/9000\n",
    "        corrDict = job[\"corrDict\"]\n",
    "        csvVec = job[\"csvVec\"]\n",
    "        \n",
    "        \n",
    "        sfTable=makeSfSpecificVec(csvVec,0,'mujets','central',0)\n",
    "        job.message(str(sfTable))\n",
    "        # **********************************sf_tt **************************************************\n",
    "        events.sf_tt = np.zeros((events.count,), np.float64)\n",
    "        for i, e in enumerate(events):\n",
    "            gen = events.genParticles\n",
    "            id = gen.pdgid\n",
    "            pt = gen.packedPt\n",
    "            pt = pt[((abs(id) == 6))]\n",
    "            id = id[abs(id)==6]\n",
    "                     #to do: check that top quark is first copy\n",
    "            pt_t = pt[id == 6]\n",
    "            pt_tbar = pt[id == -6]\n",
    "           # events.sf_tt[i] = np.sqrt(np.exp(0.0615-0.0005*np.min(400.0,pt_t))*np.exp(0.0615-0.0005*np.min(400.0,pt_tbar)))\n",
    "        # **********************************sf_pu **************************************************\n",
    "        sf_pu = [valueAt(corrDict,\"puWeights_80x_37ifb_puWeights\",i) for i in events.npvTrue]\n",
    "        \n",
    "        #  *********************************sf_btag ************************************************  \n",
    "        jets_filter = events.filter(events.ak4GenJets.count > 0)\n",
    "        events = jets_filter(events)                # filter out events without jets\n",
    "        sfTemp = []\n",
    "        effAll = []\n",
    "        sfEffAll = []\n",
    "        for i, e in enumerate(events):\n",
    "            jets = e.ak4GenJets\n",
    "           \n",
    "            for j in jets:\n",
    "                sfBtag = calcSFBtag(sfTable,j.eta_ ,j.pt_)\n",
    "                effBtag = btagEff(abs(j.eta_),j.pt_)\n",
    "                sfEff = effBtag*sfBtag\n",
    "                effAll.append(effBtag)\n",
    "                sfEffAll.append(sfEff)\n",
    "        events.ak4GenJets.sfEff = np.array(sfEffAll)\n",
    "        events.ak4GenJets.eff = np.array(effAll)\n",
    "        \n",
    "        events.btagW = np.zeros((events.count,), np.float64)   # assume weight = 0 if the event has no jets\n",
    "        #           calculate data_prob/MC_prob\n",
    "        for ie, e in enumerate(events):\n",
    "            event_jets = e.ak4GenJets                      # this holds all the jets from the event\n",
    "            jet_eff = event_jets.eff  \n",
    "            jet_sfEff = event_jets.sfEff# numpy vector with jet probabilities for this event\n",
    "            effProd = np.prod(1.0-jet_eff)\n",
    "            sfEffProd = np.prod(1.0-jet_sfEff)\n",
    "            data_eff = effProd*jet_eff/(1.0-jet_eff)\n",
    "            MC_eff = sfEffProd*jet_sfEff/(1.0-jet_sfEff)\n",
    "            data = np.sum(data_eff)\n",
    "            MC = np.sum(MC_eff)\n",
    "            btagWeight = data/MC\n",
    "            events.btagW[ie] = data/MC\n",
    "        \n",
    "        \n",
    "        \n",
    "        \n",
    "        \n",
    "        #  *********************************sf_lepID sf_lepIso sf_ ************************************************  \n",
    "        sf_lepID = np.array([])\n",
    "        sf_lepIso = np.array([])\n",
    "        sf_lepTrack = np.array([])\n",
    "        for ie,e in enumerate(events):\n",
    "            sfLepID = 1\n",
    "            sfLepIso = 1\n",
    "            sfLepTrack = 1\n",
    "            ePt = e.electrons.pt_\n",
    "            eEta = e.electrons.eta_\n",
    "            eT = e.electrons.tight\n",
    "            eType = [11 for x in e.electrons.pt_]\n",
    "            \n",
    "            electrons = zip(ePt,eEta,eT,eType)\n",
    "        \n",
    "            mPt = e.muons.pt_\n",
    "            mEta = e.muons.eta_\n",
    "            mT = e.muons.tight\n",
    "            npv = e.npv\n",
    "            mType = [13 for x in e.muons.pt_]\n",
    "            \n",
    "            muons = zip(mPt,mEta,mT,mType)\n",
    "            if not ePt.size == 0 and  mPt.size == 0:\n",
    "                leptons = []\n",
    "            elif  ePt.size== 0:\n",
    "                leptons = muons\n",
    "            elif  mPt.size == 0:\n",
    "                leptons = electrons\n",
    "            else:\n",
    "                leptons = (electrons+muons)\n",
    "            leptons = sorted(leptons)[::-1]\n",
    "            \n",
    "            for i in range(0, int(np.minimum(2,len(leptons)))):\n",
    "                if leptons[i][3] ==11:\n",
    "                    if abs(leptons[i][1]) <2.5 and leptons[i][0] > 10:\n",
    "                        if leptons[i][2]:\n",
    "                     \n",
    "                            sfLepID = sfLepID *valueAt(corrDict,\"leptonic_scalefactors_80x_egpog_37ifb_scalefactors_Tight_Electron\",leptons[i][1], leptons[i][0])\n",
    "                            \n",
    "                        else:\n",
    "                            \n",
    "                            sfLepID = sfLepID *valueAt(corrDict,\"scaleFactor_electron_summer16_scaleFactor_electron_vetoid_RooCMSShape_pu_0_100\",leptons[i][1], leptons[i][0])\n",
    "                        sfLepTrack = sfLepTrack*valueAt(corrDict,\"leptonic_scalefactors_80x_egpog_37ifb_scalefactors_Reco_Electron\", leptons[i][1], leptons[i][0])\n",
    "\n",
    "                if leptons[i][3]==13:\n",
    "                    if abs(leptons[i][1]) <2.4 and leptons[i][0] > 10:\n",
    "                        if leptons[i][2]:\n",
    "                            \n",
    "                            sfLepID = sfLepID*valueAt(corrDict,\"leptonic_muon_scalefactors_37ifb_scalefactors_TightId_Muon\",abs(leptons[i][1]), leptons[i][0])\n",
    "                            sfLepISO = sfLepIso*valueAt(corrDict,\"leptonic_muon_scalefactors_37ifb_scalefactors_Iso_MuonTightId\",abs(leptons[i][1]),leptons[i][0])\n",
    "                        else:\n",
    "                            \n",
    "                            sfLepID=sfLepID*valueAt(corrDict,\"leptonic_muon_scalefactors_37ifb_scalefactors_MuonLooseId_Muon\",abs(leptons[i][1]), leptons[i][0])\n",
    "                            sfLepISO = sfLepIso*valueAt(corrDict,\"leptonic_muon_scalefactors_37ifb_scalefactors_Iso_MuonTightId\",abs(leptons[i][1]),leptons[i][0])\n",
    "                        sfLepTrack = sfLepTrack*valueAt(corrDict,\"leptonic_Tracking_EfficienciesAndSF_BCDEFGH_ratio_eff_eta3_dr030e030_corr\",npv)\n",
    "       \n",
    "        sf_lepID = np.append(sf_lepID,sfLepID) \n",
    "        sf_lepIso = np.append(sf_lepIso,sfLepIso)\n",
    "        sf_lepTrack = np.append(sf_lepTrack,sfLepTrack)\n",
    "#*******************************************sf_ewkV sf_qcdV sf_qcdV2j **************************************************\n",
    "        events.sf_ewkV = np.zeros((events.count,), np.float64)\n",
    "        events.sf_qcdV = np.zeros((events.count,), np.float64)\n",
    "        hZLO = np.zeros((events.count,), np.float64)\n",
    "        kfactorFile = \"kfactors_\"\n",
    "        for i, e in enumerate(events):\n",
    "            target = 22\n",
    "            gen = e.genParticles\n",
    "            \n",
    "            pt = gen.packedPt\n",
    "            pt = pt*(gen.finalState)*(gen.pdgid==target)\n",
    "            pt[::-1].sort()\n",
    "            events.sf_ewkV[i] = valueAt(corrDict,str(kfactorFile)+\"EWKcorr_Z\",pt[0])\n",
    "            sf_qcdV = valueAt(corrDict, str(kfactorFile)+\"ZJets_012j_NLO_nominal\", pt[0])\n",
    "            hZLO = valueAt(corrDict, str(kfactorFile)+\"ZJets_LO_inv_pt\", pt[0])\n",
    "\n",
    "        events.sf_ewkV = events.sf_ewkV/events.sf_qcdV\n",
    "        events.sf_qcdV = events.sf_qcdV/hZLO\n",
    "        \n",
    "#******************************************sf_metTrig *********************************************\n",
    "     \n",
    "        events.sf_metTrig = np.zeros((events.count,), np.float64)\n",
    "        for i, e in enumerate(events):\n",
    "            mu = e.muons\n",
    "            pt = mu.pt_\n",
    "            tempIso = mu.nhIso+mu.phIso-0.5*mu.puIso\n",
    "            zeroArr = tempIso*0\n",
    "            tempIso = np.maximum(tempIso, zeroArr)\n",
    "            combIso = mu.chIso + tempIso\n",
    "            pt = pt*(mu.loose)*(mu.pt_>10)*(abs(mu.eta_)<2.4)*(0.5*mu.pt_>combIso)\n",
    "            px = pt*np.cos(mu.phi_)\n",
    "            py = pt*np.sin(mu.phi_)\n",
    "            pxTot = np.sum(px)\n",
    "            pyTot = np.sum(py)\n",
    "            ptTot = np.sqrt(pxTot*pxTot +pyTot*pyTot)\n",
    "            \n",
    "            events.sf_metTrig[i] = valueAt(corrDict, \"metTriggerEfficiency_recoil_monojet_TH1F_hden_monojet_recoil_clone_passed\",ptTot)\n",
    " #*******************************sf_pho*********************************************************************************\n",
    "        events.sf_pho = np.zeros((events.count,), np.float64)\n",
    "        events.sf_phoTrig = np.zeros((events.count,), np.float64)\n",
    "        for i, e in enumerate(events):\n",
    "            pho = e.photons\n",
    "            if pho.pt_.size > 0:\n",
    "               \n",
    "                pt = pho.pt_\n",
    "                eta = pho.eta_\n",
    "                tight = pho.tight\n",
    "                ptMax = pt\n",
    "                ptMax[::-1].sort()\n",
    "                ptMax = ptMax[0]\n",
    "                pt = pt*(pho.loose)*(pho.csafeVeto)*(pt == ptMax)\n",
    "                eta = eta*(pho.loose)*(pho.csafeVeto)*(pt == ptMax)\n",
    "                tight = tight*(pho.loose)*(pho.csafeVeto)*(pt == ptMax)\n",
    "                ptFinal = pt[pt>0]\n",
    "                etaFinal = eta[pt>0]\n",
    "                tight = tight[pt>0]\n",
    "                events.sf_pho[i] = valueAt(corrDict, \"scalefactors_80x_medium_photon_37ifb_EGamma_SF2D\",etaFinal,ptFinal)\n",
    "                pho = pho[(pho.pt_==ptFinal)]\n",
    "                if tight:\n",
    "                    events.sf_phoTrig[i] = valueAt(corrDict, \"photonTriggerEfficiency_photon_TH1F_hden_photonpt_clone_passed\", ptFinal) \n",
    "   \n",
    "        '''\n",
    "        \n",
    "        basic_cuts = events.filter((events.NJets >= 2) * (events.HT > 300) * (events.MHT>300)\n",
    "            * (events.NMuons == 0) * (events.NElectrons == 0) * (events.isoElectronTracks==0)\n",
    "            * (events.isoMuonTracks==0) * (events.isoPionTracks==0) \n",
    "            * (events.DeltaPhi1 > 0.5) * (events.DeltaPhi2 > 0.5) \n",
    "            * (events.DeltaPhi3 > 0.3) * (events.DeltaPhi4 > 0.3)\n",
    "            * (events.JetID==1) * (events.PFCaloMETRatio < 5) * (events.NVtx > 0)\n",
    "            * (events.eeBadScFilter==1)\n",
    "            * (events.HBHENoiseFilter==1)\n",
    "            * (events.HBHEIsoNoiseFilter==1)\n",
    "            * (events.globalTightHalo2016Filter==1)def valueAt(dictionary ,key,x,y = None):\n",
    "    \n",
    "\n",
    "            * (events.EcalDeadCellTriggerPrimitiveFilter==1)\n",
    "            * (events.BadChargedCandidateFilter==1)\n",
    "            * (events.BadPFMuonFilter==1))\n",
    "      \n",
    "        job.fill(category=category, basic_cut_acceptance=[basic_cuts.ratio])\n",
    "        \n",
    "        events = basic_cuts(events)\n",
    "        '''\n",
    "        '''\n",
    "      # HT5/HT filter\n",
    "        jets = events.Jet\n",
    "        jets_pt = p4_pt(jets.p4)\n",
    "        jets_eta = p4_eta(jets.p4)\n",
    "        jet_filter_out = (jets_pt <= 30) + (jets_eta >= 5)\n",
    "        pt_good = jets_pt.copy()\n",
    "        pt_good[jet_filter_out] = 0.0\n",
    "        jets.pt_good = pt_good\n",
    "        ht5 = np.empty_like(events.HT)\n",
    "        trigger_filter = events.filter()\n",
    "        for ie, e in enumerate(events):\n",
    "            \n",
    "            #if basic_cuts.Mask[ie]:\n",
    "            \n",
    "                # ht5 calculation\n",
    "                ejets = e.Jet\n",
    "                ht5[ie] = sum([j.pt_good for j in ejets])\n",
    "\n",
    "                # good trigger\n",
    "                # can not write \"e.Trigger[i].pass\" because \"pass\" is reserved word in Python \n",
    "                trigger_filter.Mask[ie] = np.any([e.Trigger[i].dot(\"pass\") for i in [53, 54, 55, 57, 58, 59]])\n",
    "            \n",
    "        ht5ht_filter = events.filter((ht5/np.maximum(events.HT, 1.0) <= 2.0))\n",
    "        \n",
    "        final_filter = trigger_filter * ht5ht_filter\n",
    "        \n",
    "        final_events = final_filter(events)\n",
    "        '''\n",
    "        job.fill(\n",
    "            category=category,\n",
    "            NJets = 0,#final_events.NJets,\n",
    "            #BTags = 0 #final_events.BTags,\n",
    "          #  HT = 0# final_events.HT,\n",
    "           # MHT = 0#final_events.MHT,\n",
    "            weight = events.weight\n",
    "            #final_events.Weight\n",
    "            \n",
    "        )\n",
    "        \n",
    "        "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "basic_cut_narrow = Hist(hbin(\"basic_cut_acceptance\", 50, 0, 0.1), groupby(\"category\", keeporder=True))\n",
    "basic_cut_wide = Hist(hbin(\"basic_cut_acceptance\", 50, 0, 1.0), groupby(\"category\", keeporder=True))\n",
    "\n",
    "NJets = Hist(hbin(\"NJets\", 20, 0, 20), groupby(\"category\", keeporder=True), weight=\"weight\")\n",
    "BTags = Hist(hbin(\"BTags\", 10, 0, 10), groupby(\"category\", keeporder=True), weight=\"weight\")\n",
    "HT = Hist(hbin(\"HT\", 50, 300, 1800), groupby(\"category\", keeporder=True), weight=\"weight\")\n",
    "MHT = Hist(hbin(\"MHT\", 50, 300, 1300), groupby(\"category\", keeporder=True), weight=\"weight\")\n",
    "\n",
    "display = IPythonDisplay(\n",
    "    below(\n",
    "        beside(\n",
    "            basic_cut_narrow.overlay(\"category\").step(\"basic_cut_acceptance\", yscale={\"type\":\"log\"}),\n",
    "            basic_cut_wide.overlay(\"category\").step(\"basic_cut_acceptance\", yscale={\"type\":\"log\"})\n",
    "        ),\n",
    "        beside(\n",
    "            NJets.overlay(\"category\").step(\"NJets\"),\n",
    "            BTags.overlay(\"category\").step(\"BTags\")\n",
    "        ),\n",
    "        beside(\n",
    "            HT.overlay(\"category\").step(\"HT\"),\n",
    "            MHT.overlay(\"category\").step(\"MHT\")\n",
    "        )\n",
    "    )\n",
    ")\n",
    "\n",
    "class Callback:\n",
    "    \n",
    "    def __init__(self, display):\n",
    "        self.Display = display\n",
    "        \n",
    "    def on_histogram_update(self, nevents):\n",
    "        self.Display.update()\n",
    "    def on_stream_update(self,nEvents, data):\n",
    "        for x in data[\"a\"]:\n",
    "            print x\n",
    "    def on_message(self,workerId, nEvents, message):\n",
    "        print message\n",
    "    def on_exception(self,workerId, info):\n",
    "        print info\n",
    "        "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "display.init()\n",
    "callback = Callback(display)\n",
    "\n",
    "import time\n",
    "nevents_total = 0\n",
    "t1 = time.time()\n",
    "for dataset, category in test_dataset:\n",
    "    lumi = luminosity(test_dataset)\n",
    "    job = session.createJob(dataset, \n",
    "            fraction= 0.2,                   # look at 20% events from each dataset\n",
    "            user_callback=callback, \n",
    "            user_params = {\"category\":category, \"lumi\":lumi, \"corrDict\":corrDict,\"csvVec\":csvVec},\n",
    "            histograms = [basic_cut_narrow, basic_cut_wide, NJets, BTags, HT, MHT]\n",
    "    )\n",
    "   \n",
    "    job.run()\n",
    "    runtime = job.TFinish - job.TStart\n",
    "    nevents = job.EventsProcessed\n",
    "    nevents_total += nevents\n",
    "   # print \"%-70s %7.3f M events, %7.3f M events/sec\" % (dataset[:70], float(nevents)/1000000, nevents/runtime/1000000)\n",
    "    #print \"%s: %.6f million events/second\" % (dataset, nevents/runtime/1000000)\n",
    "   # display.update()\n",
    "\n",
    "t2 = time.time()\n",
    "print \"Total events processed: %d in %.1f seconds -> %.6f million events/second\" %(nevents_total, t2-t1, nevents_total/(t2-t1)/1000000)\n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 2",
   "language": "python",
   "name": "python2"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.10"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}