contribution_slice.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "e3202201",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2022-06-08T21:15:24.241281Z",
     "start_time": "2022-06-08T21:15:17.516548Z"
    },
    "code_folding": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Welcome to JupyROOT 6.26/02\n",
      "\n",
      "Applying ATLAS style settings...\n",
      "\n"
     ]
    }
   ],
   "source": [
    "# load style\n",
    "from ROOT import gROOT\n",
    "\n",
    "path = '../../'\n",
    "# path = '/eos/user/y/youyou/'\n",
    "gROOT.SetMacroPath(path+\"atlasstyle/\")\n",
    "gROOT.LoadMacro(\"AtlasLabels.C\")\n",
    "gROOT.LoadMacro(\"AtlasStyle.C\")\n",
    "gROOT.LoadMacro(\"AtlasUtils.C\")\n",
    "from ROOT import SetAtlasStyle, ATLASLabel, myText, myMarkerText\n",
    "SetAtlasStyle()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 225,
   "id": "b23ad328",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2022-06-08T21:15:26.391474Z",
     "start_time": "2022-06-08T21:15:26.288518Z"
    },
    "code_folding": []
   },
   "outputs": [],
   "source": [
    "# load lib\n",
    "import ROOT\n",
    "import os, sys\n",
    "import numpy as np\n",
    "from array import array\n",
    "import pandas as pd"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "08bd1a85",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2022-06-08T21:15:27.588712Z",
     "start_time": "2022-06-08T21:15:27.159789Z"
    },
    "code_folding": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "12"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# load files\n",
    "\n",
    "# standard model tree\n",
    "sm_file = path + 'EFT_MINI_TREES/SM.root'\n",
    "sm_data = ROOT.TFile(sm_file,'READ')\n",
    "sm_tree = sm_data.vbswy_tree\n",
    "\n",
    "# int and quad trees\n",
    "eft_path = path + 'EFT_MINI_TREES/'\n",
    "files = [(eft_path+i) for i in os.listdir(eft_path)]\n",
    "files.sort()\n",
    "files.remove(sm_file)\n",
    "int_files = [i for i in files if 'INT' in i]\n",
    "quad_files = [i for i in files if 'QUAD' in i]\n",
    "\n",
    "# names of all operators we used\n",
    "operators = [i.split('/')[-1].split('_')[0] for i in int_files]\n",
    "operators.index('T7')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "7f395e0e",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2022-06-08T21:15:28.817361Z",
     "start_time": "2022-06-08T21:15:28.700918Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['gam_e',\n",
       " 'gam_eta',\n",
       " 'gam_phi',\n",
       " 'gam_pt',\n",
       " 'j0_e',\n",
       " 'j0_eta',\n",
       " 'j0_phi',\n",
       " 'j0_pt',\n",
       " 'j1_e',\n",
       " 'j1_eta',\n",
       " 'j1_phi',\n",
       " 'j1_pt',\n",
       " 'jj_dphi_signed',\n",
       " 'jj_drap',\n",
       " 'jj_m',\n",
       " 'jj_pt',\n",
       " 'lep_e',\n",
       " 'lep_eta',\n",
       " 'lep_phi',\n",
       " 'lep_pt',\n",
       " 'w_e',\n",
       " 'w_eta',\n",
       " 'w_mt',\n",
       " 'w_phi',\n",
       " 'w_pt',\n",
       " 'wy_m',\n",
       " 'wy_phi',\n",
       " 'wy_pt']"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# list of branches we interested in \n",
    "names = [g.GetName() for g in sm_tree.GetListOfBranches()]\n",
    "truth_branch = [i for i in names if 'truth' in i]\n",
    "truth_branch = list(dict.fromkeys(truth_branch))\n",
    "truth_branch.sort()\n",
    "\n",
    "is_reconst = lambda i: not (('truth' in i) or ('weight' in i) or ('pass_baseline' in i))\n",
    "reconst_branch = [i for i in names if is_reconst(i)]\n",
    "reconst_branch = list(dict.fromkeys(reconst_branch))\n",
    "reconst_branch.sort()\n",
    "\n",
    "reconst_branch[1:]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "0186b190",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'./output_pdf/rebin_reconstructed/T0.pdf'"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# def some helper functions\n",
    "def generate_output_path(operator,add='', format = 'pdf'):\n",
    "    folder = 'rebin_reconstructed/'\n",
    "    folder = './output_'+format+'/'+folder\n",
    "    if not os.path.exists(folder):\n",
    "        os.mkdir(folder)\n",
    "    return (folder+operator+add+'.'+format)\n",
    "\n",
    "generate_output_path('T0')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "09c1f1d9",
   "metadata": {},
   "outputs": [],
   "source": [
    "def pre_selection(tree):\n",
    "    entryList = ROOT.TEntryList(\"entryListName\", \"Title of the entry list\")\n",
    "    for entry in tree:\n",
    "        gam = ROOT.TLorentzVector()\n",
    "        w_boson = ROOT.TLorentzVector()\n",
    "        jet0 = ROOT.TLorentzVector()\n",
    "        jet1 = ROOT.TLorentzVector()\n",
    "        wy = ROOT.TLorentzVector()\n",
    "\n",
    "        gam.SetPtEtaPhiE(entry.gam_pt, entry.gam_eta, entry.gam_phi, entry.gam_e)\n",
    "        w_boson.SetPtEtaPhiE(entry.w_pt, entry.w_eta, entry.w_phi, entry.w_e)\n",
    "        jet0.SetPtEtaPhiE(entry.j0_pt, entry.j0_eta, entry.j0_phi, entry.j0_e)\n",
    "        jet1.SetPtEtaPhiE(entry.j1_pt, entry.j1_eta, entry.j1_phi, entry.j1_e)\n",
    "        wy = gam+w_boson\n",
    "        # i didn't add it as a branch\n",
    "        wy_xi = abs(wy.Rapidity()-0.5*(jet0.Rapidity()+jet1.Rapidity())) / entry.jj_drap\n",
    "\n",
    "        selection = (int(entry.pass_baseline_e) | int(entry.pass_baseline_mu)) and (entry.jj_m > 0.5000) and (entry.jj_drap > 2) and (wy_xi < 0.35)\n",
    "        if selection:\n",
    "            entryList.Enter(tree.GetReadEntry())\n",
    "\n",
    "    tree.SetEntryList(entryList)\n",
    "    \n",
    "    df = pd.DataFrame(columns=reconst_branch+['weight'])\n",
    "    i = 0\n",
    "    tmp = tree.GetEntryNumber(i)\n",
    "    while tmp != -1:\n",
    "        tree.GetEntry(tmp)\n",
    "        list_tmp = []\n",
    "        for r in reconst_branch:\n",
    "            list_tmp += [getattr(tree, r)]\n",
    "        list_tmp += [tree.weight]\n",
    "        df.loc[i] = list_tmp\n",
    "        i += 1\n",
    "        tmp = tree.GetEntryNumber(i)\n",
    "\n",
    "    return df   # tree, entryList\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 156,
   "id": "e766beed",
   "metadata": {},
   "outputs": [],
   "source": [
    "def generate_dr_dphi(df):\n",
    "    df = df.copy()\n",
    "\n",
    "    def mpi_pi(angle):\n",
    "        while (angle >= np.pi):\n",
    "            angle -= 2* np.pi\n",
    "        while (angle < -1* np.pi):\n",
    "             angle += 2* np.pi\n",
    "\n",
    "        return angle\n",
    "\n",
    "    dr_lepy = []\n",
    "    dr_wy = []\n",
    "    dphi_lepy = []\n",
    "    dphi_wy = []\n",
    "    for i in range(len(df)):\n",
    "        gamma = ROOT.TLorentzVector()\n",
    "        lepton = ROOT.TLorentzVector()\n",
    "        w_boson = ROOT.TLorentzVector()\n",
    "\n",
    "        gamma.SetPtEtaPhiE(df['gam_pt'][i],df['gam_eta'][i],df['gam_phi'][i],df['gam_e'][i])\n",
    "        lepton.SetPtEtaPhiE(df['lep_pt'][i],df['lep_eta'][i],df['lep_phi'][i],df['lep_e'][i])\n",
    "        w_boson.SetPtEtaPhiE(df['w_pt'][i],df['w_eta'][i],df['w_phi'][i],df['w_e'][i])\n",
    "\n",
    "        dr_lepy += [lepton.DeltaR(gamma)]\n",
    "        dr_wy += [w_boson.DeltaR(gamma)]\n",
    "\n",
    "        if lepton.Rapidity() > gamma.Rapidity():\n",
    "            lepy_phi_forward = lepton.Phi()\n",
    "            lepy_phi_backward = gamma.Phi()\n",
    "        else:\n",
    "            lepy_phi_forward = gamma.Phi()\n",
    "            lepy_phi_backward = lepton.Phi()\n",
    "        dphi_lepy += [mpi_pi(lepy_phi_forward - lepy_phi_backward)]\n",
    "\n",
    "        if w_boson.Rapidity() > gamma.Rapidity():\n",
    "            wy_phi_forward = w_boson.Phi()\n",
    "            wy_phi_backward = gamma.Phi()\n",
    "        else:\n",
    "            wy_phi_forward = gamma.Phi()\n",
    "            wy_phi_backward = w_boson.Phi()\n",
    "        dphi_wy += [mpi_pi(wy_phi_forward - wy_phi_backward)]\n",
    "\n",
    "    df['dphi_lepy'] = dphi_lepy\n",
    "    df['dphi_wy'] = dphi_wy\n",
    "    df['dr_lepy'] = dr_lepy\n",
    "    df['dr_wy'] = dr_wy\n",
    "    \n",
    "    return df    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 157,
   "id": "5a06eeca",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>dsid</th>\n",
       "      <th>gam_e</th>\n",
       "      <th>gam_eta</th>\n",
       "      <th>gam_phi</th>\n",
       "      <th>gam_pt</th>\n",
       "      <th>j0_e</th>\n",
       "      <th>j0_eta</th>\n",
       "      <th>j0_phi</th>\n",
       "      <th>j0_pt</th>\n",
       "      <th>j1_e</th>\n",
       "      <th>...</th>\n",
       "      <th>w_phi</th>\n",
       "      <th>w_pt</th>\n",
       "      <th>wy_m</th>\n",
       "      <th>wy_phi</th>\n",
       "      <th>wy_pt</th>\n",
       "      <th>weight</th>\n",
       "      <th>dphi_lepy</th>\n",
       "      <th>dphi_wy</th>\n",
       "      <th>dr_lepy</th>\n",
       "      <th>dr_wy</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>363270.0</td>\n",
       "      <td>187.574789</td>\n",
       "      <td>2.042001</td>\n",
       "      <td>-0.503302</td>\n",
       "      <td>47.876445</td>\n",
       "      <td>829.435168</td>\n",
       "      <td>-1.997017</td>\n",
       "      <td>2.031665</td>\n",
       "      <td>221.087766</td>\n",
       "      <td>369.512384</td>\n",
       "      <td>...</td>\n",
       "      <td>-0.263019</td>\n",
       "      <td>107.960786</td>\n",
       "      <td>176.138268</td>\n",
       "      <td>-0.336648</td>\n",
       "      <td>154.881409</td>\n",
       "      <td>0.064699</td>\n",
       "      <td>-0.240210</td>\n",
       "      <td>-0.240283</td>\n",
       "      <td>2.070275</td>\n",
       "      <td>2.070284</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>363270.0</td>\n",
       "      <td>716.515885</td>\n",
       "      <td>-1.210340</td>\n",
       "      <td>2.881138</td>\n",
       "      <td>392.318937</td>\n",
       "      <td>2107.599090</td>\n",
       "      <td>3.068873</td>\n",
       "      <td>0.048355</td>\n",
       "      <td>195.462002</td>\n",
       "      <td>2157.751512</td>\n",
       "      <td>...</td>\n",
       "      <td>-0.059591</td>\n",
       "      <td>51.160642</td>\n",
       "      <td>325.896852</td>\n",
       "      <td>2.851317</td>\n",
       "      <td>342.339108</td>\n",
       "      <td>0.079482</td>\n",
       "      <td>-2.940507</td>\n",
       "      <td>-2.940729</td>\n",
       "      <td>3.138830</td>\n",
       "      <td>3.139038</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>363270.0</td>\n",
       "      <td>76.092459</td>\n",
       "      <td>0.221988</td>\n",
       "      <td>-2.728928</td>\n",
       "      <td>74.255336</td>\n",
       "      <td>409.161278</td>\n",
       "      <td>-2.048432</td>\n",
       "      <td>-0.372453</td>\n",
       "      <td>103.671930</td>\n",
       "      <td>98.808764</td>\n",
       "      <td>...</td>\n",
       "      <td>2.673311</td>\n",
       "      <td>38.194998</td>\n",
       "      <td>46.134245</td>\n",
       "      <td>-3.019382</td>\n",
       "      <td>102.872383</td>\n",
       "      <td>0.058395</td>\n",
       "      <td>0.880951</td>\n",
       "      <td>0.880946</td>\n",
       "      <td>0.894036</td>\n",
       "      <td>0.894031</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>363270.0</td>\n",
       "      <td>28.663095</td>\n",
       "      <td>0.672188</td>\n",
       "      <td>-0.054652</td>\n",
       "      <td>23.217365</td>\n",
       "      <td>1220.408780</td>\n",
       "      <td>-2.114067</td>\n",
       "      <td>2.516062</td>\n",
       "      <td>290.448906</td>\n",
       "      <td>956.535871</td>\n",
       "      <td>...</td>\n",
       "      <td>0.161921</td>\n",
       "      <td>209.252560</td>\n",
       "      <td>48.989156</td>\n",
       "      <td>0.140413</td>\n",
       "      <td>231.981212</td>\n",
       "      <td>0.074781</td>\n",
       "      <td>0.216600</td>\n",
       "      <td>0.216573</td>\n",
       "      <td>0.691692</td>\n",
       "      <td>0.691684</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>363270.0</td>\n",
       "      <td>27.678104</td>\n",
       "      <td>0.501819</td>\n",
       "      <td>2.357176</td>\n",
       "      <td>24.524812</td>\n",
       "      <td>945.942935</td>\n",
       "      <td>3.139888</td>\n",
       "      <td>1.492426</td>\n",
       "      <td>81.739844</td>\n",
       "      <td>336.349470</td>\n",
       "      <td>...</td>\n",
       "      <td>-2.981528</td>\n",
       "      <td>47.384492</td>\n",
       "      <td>73.421696</td>\n",
       "      <td>2.990389</td>\n",
       "      <td>64.877674</td>\n",
       "      <td>0.056811</td>\n",
       "      <td>-0.944477</td>\n",
       "      <td>-0.944481</td>\n",
       "      <td>1.969929</td>\n",
       "      <td>1.969932</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19944</th>\n",
       "      <td>363272.0</td>\n",
       "      <td>73.200317</td>\n",
       "      <td>1.333682</td>\n",
       "      <td>-2.003242</td>\n",
       "      <td>36.072645</td>\n",
       "      <td>192.523441</td>\n",
       "      <td>-1.022055</td>\n",
       "      <td>-0.004206</td>\n",
       "      <td>122.415944</td>\n",
       "      <td>363.627920</td>\n",
       "      <td>...</td>\n",
       "      <td>1.695181</td>\n",
       "      <td>41.136863</td>\n",
       "      <td>75.141904</td>\n",
       "      <td>2.762000</td>\n",
       "      <td>21.771143</td>\n",
       "      <td>0.091927</td>\n",
       "      <td>2.584761</td>\n",
       "      <td>2.584763</td>\n",
       "      <td>2.605220</td>\n",
       "      <td>2.605222</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19945</th>\n",
       "      <td>363272.0</td>\n",
       "      <td>156.622332</td>\n",
       "      <td>-1.909200</td>\n",
       "      <td>1.346155</td>\n",
       "      <td>45.424844</td>\n",
       "      <td>120.358292</td>\n",
       "      <td>-0.219015</td>\n",
       "      <td>-1.085886</td>\n",
       "      <td>117.030790</td>\n",
       "      <td>898.554284</td>\n",
       "      <td>...</td>\n",
       "      <td>2.376111</td>\n",
       "      <td>32.202403</td>\n",
       "      <td>101.735515</td>\n",
       "      <td>1.765037</td>\n",
       "      <td>67.872401</td>\n",
       "      <td>0.086812</td>\n",
       "      <td>1.030489</td>\n",
       "      <td>1.029957</td>\n",
       "      <td>2.318521</td>\n",
       "      <td>2.318285</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19946</th>\n",
       "      <td>363272.0</td>\n",
       "      <td>105.070789</td>\n",
       "      <td>-0.993770</td>\n",
       "      <td>2.689175</td>\n",
       "      <td>68.414883</td>\n",
       "      <td>534.734661</td>\n",
       "      <td>-1.341179</td>\n",
       "      <td>-0.147029</td>\n",
       "      <td>261.314715</td>\n",
       "      <td>491.413472</td>\n",
       "      <td>...</td>\n",
       "      <td>-2.860134</td>\n",
       "      <td>53.229632</td>\n",
       "      <td>81.681235</td>\n",
       "      <td>3.008171</td>\n",
       "      <td>113.677272</td>\n",
       "      <td>0.093809</td>\n",
       "      <td>0.734089</td>\n",
       "      <td>0.733877</td>\n",
       "      <td>1.316264</td>\n",
       "      <td>1.316145</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19947</th>\n",
       "      <td>363272.0</td>\n",
       "      <td>28.963492</td>\n",
       "      <td>0.117419</td>\n",
       "      <td>0.791893</td>\n",
       "      <td>28.764971</td>\n",
       "      <td>2425.672927</td>\n",
       "      <td>-1.740736</td>\n",
       "      <td>-2.701662</td>\n",
       "      <td>825.331489</td>\n",
       "      <td>2164.045073</td>\n",
       "      <td>...</td>\n",
       "      <td>0.158732</td>\n",
       "      <td>70.835859</td>\n",
       "      <td>54.745198</td>\n",
       "      <td>0.337810</td>\n",
       "      <td>95.553110</td>\n",
       "      <td>0.039691</td>\n",
       "      <td>-0.633298</td>\n",
       "      <td>-0.633161</td>\n",
       "      <td>1.182595</td>\n",
       "      <td>1.182522</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19948</th>\n",
       "      <td>363272.0</td>\n",
       "      <td>110.791537</td>\n",
       "      <td>0.236324</td>\n",
       "      <td>-1.484866</td>\n",
       "      <td>107.768125</td>\n",
       "      <td>416.278701</td>\n",
       "      <td>-1.582729</td>\n",
       "      <td>1.976029</td>\n",
       "      <td>163.803453</td>\n",
       "      <td>128.797995</td>\n",
       "      <td>...</td>\n",
       "      <td>-1.309717</td>\n",
       "      <td>79.836623</td>\n",
       "      <td>81.808935</td>\n",
       "      <td>-1.410362</td>\n",
       "      <td>186.901778</td>\n",
       "      <td>0.084516</td>\n",
       "      <td>-0.175378</td>\n",
       "      <td>-0.175149</td>\n",
       "      <td>0.857695</td>\n",
       "      <td>0.857648</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>19949 rows × 34 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "           dsid       gam_e   gam_eta   gam_phi      gam_pt         j0_e  \\\n",
       "0      363270.0  187.574789  2.042001 -0.503302   47.876445   829.435168   \n",
       "1      363270.0  716.515885 -1.210340  2.881138  392.318937  2107.599090   \n",
       "2      363270.0   76.092459  0.221988 -2.728928   74.255336   409.161278   \n",
       "3      363270.0   28.663095  0.672188 -0.054652   23.217365  1220.408780   \n",
       "4      363270.0   27.678104  0.501819  2.357176   24.524812   945.942935   \n",
       "...         ...         ...       ...       ...         ...          ...   \n",
       "19944  363272.0   73.200317  1.333682 -2.003242   36.072645   192.523441   \n",
       "19945  363272.0  156.622332 -1.909200  1.346155   45.424844   120.358292   \n",
       "19946  363272.0  105.070789 -0.993770  2.689175   68.414883   534.734661   \n",
       "19947  363272.0   28.963492  0.117419  0.791893   28.764971  2425.672927   \n",
       "19948  363272.0  110.791537  0.236324 -1.484866  107.768125   416.278701   \n",
       "\n",
       "         j0_eta    j0_phi       j0_pt         j1_e  ...     w_phi        w_pt  \\\n",
       "0     -1.997017  2.031665  221.087766   369.512384  ... -0.263019  107.960786   \n",
       "1      3.068873  0.048355  195.462002  2157.751512  ... -0.059591   51.160642   \n",
       "2     -2.048432 -0.372453  103.671930    98.808764  ...  2.673311   38.194998   \n",
       "3     -2.114067  2.516062  290.448906   956.535871  ...  0.161921  209.252560   \n",
       "4      3.139888  1.492426   81.739844   336.349470  ... -2.981528   47.384492   \n",
       "...         ...       ...         ...          ...  ...       ...         ...   \n",
       "19944 -1.022055 -0.004206  122.415944   363.627920  ...  1.695181   41.136863   \n",
       "19945 -0.219015 -1.085886  117.030790   898.554284  ...  2.376111   32.202403   \n",
       "19946 -1.341179 -0.147029  261.314715   491.413472  ... -2.860134   53.229632   \n",
       "19947 -1.740736 -2.701662  825.331489  2164.045073  ...  0.158732   70.835859   \n",
       "19948 -1.582729  1.976029  163.803453   128.797995  ... -1.309717   79.836623   \n",
       "\n",
       "             wy_m    wy_phi       wy_pt    weight  dphi_lepy   dphi_wy  \\\n",
       "0      176.138268 -0.336648  154.881409  0.064699  -0.240210 -0.240283   \n",
       "1      325.896852  2.851317  342.339108  0.079482  -2.940507 -2.940729   \n",
       "2       46.134245 -3.019382  102.872383  0.058395   0.880951  0.880946   \n",
       "3       48.989156  0.140413  231.981212  0.074781   0.216600  0.216573   \n",
       "4       73.421696  2.990389   64.877674  0.056811  -0.944477 -0.944481   \n",
       "...           ...       ...         ...       ...        ...       ...   \n",
       "19944   75.141904  2.762000   21.771143  0.091927   2.584761  2.584763   \n",
       "19945  101.735515  1.765037   67.872401  0.086812   1.030489  1.029957   \n",
       "19946   81.681235  3.008171  113.677272  0.093809   0.734089  0.733877   \n",
       "19947   54.745198  0.337810   95.553110  0.039691  -0.633298 -0.633161   \n",
       "19948   81.808935 -1.410362  186.901778  0.084516  -0.175378 -0.175149   \n",
       "\n",
       "        dr_lepy     dr_wy  \n",
       "0      2.070275  2.070284  \n",
       "1      3.138830  3.139038  \n",
       "2      0.894036  0.894031  \n",
       "3      0.691692  0.691684  \n",
       "4      1.969929  1.969932  \n",
       "...         ...       ...  \n",
       "19944  2.605220  2.605222  \n",
       "19945  2.318521  2.318285  \n",
       "19946  1.316264  1.316145  \n",
       "19947  1.182595  1.182522  \n",
       "19948  0.857695  0.857648  \n",
       "\n",
       "[19949 rows x 34 columns]"
      ]
     },
     "execution_count": 157,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "sm_df = pre_selection(sm_tree)\n",
    "sm_df = generate_dr_dphi(sm_df)\n",
    "\n",
    "sm_df\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 158,
   "id": "baa4ee7e",
   "metadata": {},
   "outputs": [],
   "source": [
    "def generate_df(operator):\n",
    "    i = operators.index(operator)\n",
    "    int_data = ROOT.TFile(int_files[i],'READ')\n",
    "    int_tree = int_data.vbswy_tree\n",
    "    quad_data = ROOT.TFile(quad_files[i], 'READ')\n",
    "    quad_tree = quad_data.vbswy_tree\n",
    "\n",
    "    # load values of coupling\n",
    "    c1, c2, c3 = 1,10,100\n",
    "    if operator[0] == 'T':\n",
    "        c1, c2, c3 = 0.1,1,10\n",
    "\n",
    "    # apply pre selection\n",
    "    int_df = pre_selection(int_tree)\n",
    "    quad_df = pre_selection(quad_tree)\n",
    "    \n",
    "    int_data.Close()\n",
    "    quad_data.Close()\n",
    "\n",
    "    # add drap and dphi\n",
    "    int_df = generate_dr_dphi(int_df)\n",
    "    quad_df = generate_dr_dphi(quad_df)\n",
    "\n",
    "    sm_df['c1_w'] = sm_df['weight']\n",
    "    sm_df['c2_w'] = sm_df['weight']\n",
    "    sm_df['c3_w'] = sm_df['weight']\n",
    "    int_df['c1_w'] = int_df['weight']*c1/2.4\n",
    "    int_df['c2_w'] = int_df['weight']*c2/2.4\n",
    "    int_df['c3_w'] = int_df['weight']*c3/2.4\n",
    "    quad_df['c1_w'] = quad_df['weight']*(c1/2.4)**2\n",
    "    quad_df['c2_w'] = quad_df['weight']*(c2/2.4)**2\n",
    "    quad_df['c3_w'] = quad_df['weight']*(c3/2.4)**2\n",
    "\n",
    "    df = pd.concat([sm_df, int_df, quad_df], keys=[\"s\", \"i\", \"q\"])\n",
    "    return df\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 159,
   "id": "006a18ad",
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "l1 = ['gam_e','lep_pt','j0_pt','j1_pt','w_mt','wy_m','wy_pt']\n",
    "l2 = ['jj_m','jj_pt','jj_dphi_signed']\n",
    "branches = l2+l1\n",
    "\n",
    "operators_t = [i for i in operators if i[0]=='T']\n",
    "operators_m = [i for i in operators if i[0]=='M']\n",
    "dfs_t = [generate_df(i) for i in operators_t]\n",
    "dfs_m = [generate_df(i) for i in operators_m]\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 163,
   "id": "8819236a",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Lepton p_{T} [GeV/c] '"
      ]
     },
     "execution_count": 163,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "unit_dict= {'gam_e':    \"E_{#gamma} [GeV]\",\n",
    "        'lep_pt':       \"Lepton p_{T} [GeV/c] \",\n",
    "        'j0_pt':        \"Leading jet p_{T}  [GeV/c] \",\n",
    "        'j1_pt':        \"Second Leading jet p_{T} [GeV/c] \",\n",
    "        'jj_m':         \"m_{jj} [GeV/c^{2}]\",\n",
    "        'jj_pt':        \"di-jet p_{T}  [GeV/c] \",\n",
    "        'jj_dphi_signed': \"#Delta#phi_{jj}  [rad] \",\n",
    "        'w_mt':         \"m_{T} (W) [GeV/c^{2}]\",\n",
    "        'wy_m':         \"m_{W#gamma} [GeV/c^{2}]\",\n",
    "        'wy_pt':        \"W#gamma p_{T}  [GeV/c]\",\n",
    "        'dphi_lepy':    'signed #Delta#phi_{Lepton #gamma}  [rad] ',\n",
    "        'dphi_wy':      'signed #Delta#phi_{W#gamma}  [rad]',\n",
    "        'dr_lepy':      '#Delta R_{Lepton #gamma}  [rad]',\n",
    "        'dr_wy':        '#Delta R_{W#gamma}  [rad]'\n",
    "        }\n",
    "\n",
    "unit_dict['lep_pt']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 227,
   "id": "d7f756af",
   "metadata": {},
   "outputs": [],
   "source": [
    "def plot_contribution(operators, dfs, branches, y_max=0.1, y_min=-0.1, col_width=0.2):\n",
    "\n",
    "    can = ROOT.TCanvas('can1', 'can1', 3000, 2500)\n",
    "    can.cd()\n",
    "    pads = []\n",
    "    hss = []\n",
    "    bins = np.linspace(1.5, 1.5, 7).cumprod()\n",
    "    bins = np.insert(bins, 0, 0)\n",
    "\n",
    "    #def width\n",
    "    row_width = 0.192\n",
    "    left_space = 0.999-row_width*len(dfs)\n",
    "    bottom_space = 1-col_width*len(branches)\n",
    "\n",
    "    for i in range(len(dfs)):   #5\n",
    "        operator = operators[i]\n",
    "        df = dfs[i].copy()\n",
    "        df_i = df.loc['i']\n",
    "        df_q = df.loc['q']\n",
    "\n",
    "        for j in range(len(branches)):  #4\n",
    "            branch = branches[j]\n",
    "            max_x = sm_df[branch].max()\n",
    "            if max_x > 10:\n",
    "                bins = [i/bins[-1]*max_x for i in bins]\n",
    "            else:\n",
    "                bins = np.linspace(sm_df[branch].min(), max_x, 7)\n",
    "            bins = array('d',bins)\n",
    "            can.cd()\n",
    "\n",
    "            if j!=0 and i!=0:\n",
    "                pad = ROOT.TPad (\"pad\"+str(i)+str(j),\"pad\"+str(i)+str(j) ,j*col_width+bottom_space, i*row_width+left_space, (j+1)*col_width+bottom_space, (i+1)*(row_width)+left_space)\n",
    "            else:\n",
    "                if i == 0 and j ==0:\n",
    "                    pad = ROOT.TPad (\"pad\"+str(i)+str(j),\"pad\"+str(i)+str(j) ,j*col_width, i*row_width, (j+1)*col_width+bottom_space, (i+1)*(row_width)+left_space)\n",
    "                elif i==0:\n",
    "                    pad = ROOT.TPad (\"pad\"+str(i)+str(j),\"pad\"+str(i)+str(j) ,j*col_width+bottom_space, i*row_width, (j+1)*col_width+bottom_space, (i+1)*(row_width)+left_space)\n",
    "                else:\n",
    "                    pad = ROOT.TPad (\"pad\"+str(i)+str(j),\"pad\"+str(i)+str(j) ,j*col_width, i*row_width+left_space, (j+1)*col_width+bottom_space, (i+1)*(row_width)+left_space)\n",
    "            pads += [pad]\n",
    "            pad.Draw()\n",
    "            pad.cd()\n",
    "\n",
    "            h_i = ROOT.TH1F(\"i\"+str(i)+str(j), 'hi;'+branch+';numbers    .', len(bins)-1, bins)\n",
    "            h_q = ROOT.TH1F(\"q\"+str(i)+str(j), 'hq;'+branch+';numbers    .', len(bins)-1, bins)\n",
    "            h_a = ROOT.TH1F(\"a\"+str(i)+str(j), 'ha;'+branch+';numbers    .', len(bins)-1, bins)\n",
    "            h_s = ROOT.TH1F(\"s\"+str(i)+str(j), 'hs;'+branch+';numbers    .', len(bins)-1, bins)\n",
    "\n",
    "            for a in range(len(sm_df)):\n",
    "                h_s.Fill(sm_df[branch][a], sm_df['weight'][a])\n",
    "            for a in range(len(df_i)):\n",
    "                h_i.Fill(df_i[branch][a], df_i['c1_w'][a])\n",
    "                h_a.Fill(df_i[branch][a], df_i['c1_w'][a])\n",
    "            for a in range(len(df_q)):\n",
    "                h_q.Fill(df_q[branch][a], df_q['c1_w'][a])\n",
    "                h_a.Fill(df_q[branch][a], df_q['c1_w'][a])\n",
    "\n",
    "            h_i.Divide(h_s)\n",
    "            h_q.Divide(h_s)\n",
    "            h_a.Divide(h_s)\n",
    "            hs = ROOT.THStack (branch+operator+\"hs\",\" THStack \")\n",
    "            hs.Add (h_i)\n",
    "            hs.Add (h_q)\n",
    "            hs.Add (h_a)\n",
    "            hs.Draw (\"NOSTACK, h\")\n",
    "\n",
    "            hs.SetMinimum(y_min)\n",
    "            hs.SetMaximum(y_max)\n",
    "            hs.GetYaxis().SetNdivisions (407)\n",
    "            hs.GetYaxis().SetNdivisions (407)\n",
    "            hs.GetXaxis().SetTitle(unit_dict[branch]+'   ')\n",
    "            hs.GetXaxis().SetTitleOffset (0.9)\n",
    "            hs.GetXaxis().SetTitleSize (0.07)\n",
    "            hs.GetYaxis().SetTitle('Ratio to SM   ')\n",
    "            hs.GetYaxis().SetTitleOffset (1.2)\n",
    "            hs.GetYaxis().SetTitleSize (0.06)\n",
    "\n",
    "            h_i.SetLineColorAlpha(38, 1)\n",
    "            h_a.SetLineColorAlpha(46, 0.7)\n",
    "            h_q.SetLineColorAlpha(ROOT.kOrange, 1)\n",
    "            h_i.SetMarkerStyle(20)\n",
    "            h_q.SetMarkerStyle(21)\n",
    "            h_a.SetMarkerStyle(22)\n",
    "            h_i.SetMarkerColor(38)\n",
    "            h_a.SetMarkerColor(46)\n",
    "            h_q.SetMarkerColor(ROOT.kOrange)\n",
    "            h_i.SetMarkerSize(3)\n",
    "            h_a.SetMarkerSize(3)\n",
    "            h_q.SetMarkerSize(3)\n",
    "\n",
    "            h1_markers, h2_markers, h3_markers = [], [], []\n",
    "            for a in range(len(bins)):\n",
    "                if (h_q.GetBinContent(a) >= y_max):\n",
    "                    h1_markers += [ROOT.TMarker(h_q.GetBinCenter(a), y_max*0.9,21)]\n",
    "                if (h_a.GetBinContent(a) >= y_max):\n",
    "                    h2_markers += [ROOT.TMarker(h_a.GetBinCenter(a), y_max*0.9,22)]\n",
    "                if (h_i.GetBinContent(a) <= y_min):\n",
    "                    h3_markers += [ROOT.TMarker(h_a.GetBinCenter(a), y_min*0.9,22)]\n",
    "            for b in h1_markers:\n",
    "                b.SetMarkerSize(2)\n",
    "                b.SetMarkerColor(ROOT.kOrange)\n",
    "                b.Draw('same')\n",
    "            for b in h2_markers:\n",
    "                b.SetMarkerSize(2)\n",
    "                b.SetMarkerColor(46)\n",
    "                b.Draw('same')\n",
    "            for b in h3_markers:\n",
    "                b.SetMarkerSize(2)\n",
    "                b.SetMarkerColor(38)\n",
    "                b.Draw('same')\n",
    "            hss += [h1_markers, h2_markers, h3_markers]\n",
    "\n",
    "            t = ROOT.TLatex()\n",
    "            t.SetNDC()\n",
    "            t.SetTextColor( 1 )\n",
    "            t.SetTextSize( 0.13 )\n",
    "            t.SetTextAlign( 12 )\n",
    "            if operator == 'M2':\n",
    "                t.SetTextSize( 0.11 )\n",
    "            t.DrawLatex( 0.85, 0.85, '#scale[1]{'+operator+'}')\n",
    "\n",
    "            hss += [h_i, h_q, h_a, hs]\n",
    "            pad.RedrawAxis('g')\n",
    "            pad.SetBottomMargin (0)\n",
    "            pad.SetTopMargin (0)\n",
    "            pad.SetRightMargin(0)\n",
    "            pad.SetLeftMargin(0)\n",
    "            if j == 0:\n",
    "                pad.SetLeftMargin(0.18)\n",
    "                legend = ROOT.TLegend(0.25,0.65,0.6,0.97)\n",
    "                c=0.1\n",
    "                if operator[0]=='M':\n",
    "                    c=1\n",
    "                legend.SetHeader('f_{0} /#Lambda^{3} = {1} TeV^{2}'.format({operator}, c,{-4},{4}),\"lp\")\n",
    "                legend.AddEntry ( h_i ,'O_{0}'.format({'Interference term'}),\"lp\")\n",
    "                legend.AddEntry ( h_q ,'O_{0}'.format({'Quadratic term'}),\"lp\")\n",
    "                legend.AddEntry ( h_a ,'O_{0}'.format({'Total aQGC contribution'}),\"lp\")\n",
    "                legend.Draw (\" same \")\n",
    "                legend.SetFillColorAlpha(0, 0.8)\n",
    "                hss += [legend]\n",
    "            if i == 0:\n",
    "                hs.GetYaxis().SetTitleSize (0.05)\n",
    "                hs.GetYaxis().SetTitleOffset (1.35)\n",
    "                hss+=[hs]\n",
    "                pad.SetBottomMargin (0.16)\n",
    "    \n",
    "    can.Draw()\n",
    "    can.Update()\n",
    "    can.Print(branches[0]+'test.pdf') \n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 228,
   "id": "568061fa",
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Warning in <TROOT::Append>: Replacing existing TH1: i00 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q00 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a00 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i01 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q01 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a01 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i02 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q02 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a02 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i03 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q03 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a03 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i10 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q10 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a10 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i11 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q11 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a11 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i12 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q12 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a12 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i13 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q13 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a13 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i20 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q20 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a20 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i21 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q21 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a21 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i22 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q22 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a22 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i23 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q23 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a23 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i30 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q30 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a30 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i31 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q31 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a31 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i32 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q32 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a32 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i33 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q33 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a33 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i40 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q40 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a40 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i41 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q41 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a41 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i42 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q42 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a42 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i43 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q43 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a43 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file lep_pttest.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i00 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q00 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a00 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i01 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q01 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a01 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i02 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q02 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a02 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i03 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q03 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a03 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i10 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q10 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a10 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i11 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q11 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a11 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i12 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q12 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a12 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i13 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q13 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a13 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i20 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q20 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a20 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i21 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q21 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a21 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i22 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q22 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a22 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i23 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q23 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a23 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i30 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q30 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a30 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i31 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q31 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a31 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i32 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q32 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a32 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i33 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q33 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a33 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i40 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q40 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a40 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i41 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q41 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a41 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i42 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q42 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a42 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: i43 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: q43 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: a43 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file jj_mtest.pdf has been created\n"
     ]
    }
   ],
   "source": [
    "os = ['M2', 'T6', 'T1', 'T5', 'T0']\n",
    "df_arr = [dfs_m[2], dfs_t[4], dfs_t[1], dfs_t[3], dfs_t[0]]\n",
    "bs1 = ['lep_pt','gam_e','wy_m', 'jj_pt']\n",
    "bs2 =['jj_m','jj_dphi_signed','dphi_lepy','dr_lepy']\n",
    "\n",
    "plot_contribution(os,df_arr,bs1,y_max=0.5, y_min=-0.15, col_width=0.24)\n",
    "plot_contribution(os,df_arr,bs2,y_max=0.005, y_min=-0.005, col_width=0.24)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d6aae97e",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "interpreter": {
   "hash": "65398ec9edf812bfa2a1cd19e30d7cdb06b1eced593dd679351ef973282302ad"
  },
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.12"
  },
  "varInspector": {
   "cols": {
    "lenName": 16,
    "lenType": 16,
    "lenVar": 40
   },
   "kernels_config": {
    "python": {
     "delete_cmd_postfix": "",
     "delete_cmd_prefix": "del ",
     "library": "var_list.py",
     "varRefreshCmd": "print(var_dic_list())"
    },
    "r": {
     "delete_cmd_postfix": ") ",
     "delete_cmd_prefix": "rm(",
     "library": "var_list.r",
     "varRefreshCmd": "cat(var_dic_list()) "
    }
   },
   "oldHeight": 399.852,
   "position": {
    "height": "40px",
    "left": "1058px",
    "right": "20px",
    "top": "120px",
    "width": "347px"
   },
   "types_to_exclude": [
    "module",
    "function",
    "builtin_function_or_method",
    "instance",
    "_Feature"
   ],
   "varInspector_section_display": "none",
   "window_display": false
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}