proj1_rebin.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": null,
   "id": "3c998019",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2022-06-08T21:15:30.234250Z",
     "start_time": "2022-06-08T21:15:30.113414Z"
    },
    "code_folding": [
     7,
     11
    ]
   },
   "outputs": [],
   "source": [
    "# function of method 1\n",
    "def plot_operator(operator, df_all, reconst_branch=reconst_branch[1:], set_log=False):\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",
    "    # prepare bins for hist\n",
    "    nbins = 5\n",
    "    df = df_all.copy()\n",
    "    c1_bin_size = df['c1_w'].sum() / 5\n",
    "    c2_bin_size = df['c2_w'].sum() / 5\n",
    "    c3_bin_size = df['c3_w'].sum() / 5\n",
    "    c1_index = [0, c1_bin_size, c1_bin_size*2, c1_bin_size*3, c1_bin_size*4, c1_bin_size*5]\n",
    "    c2_index = [0, c2_bin_size, c2_bin_size*2, c2_bin_size*3, c2_bin_size*4, c2_bin_size*5]\n",
    "    c3_index = [0, c3_bin_size, c3_bin_size*2, c3_bin_size*3, c3_bin_size*4, c3_bin_size*5]\n",
    "\n",
    "    # prepare canvas\n",
    "    rows = 4\n",
    "    if len(reconst_branch) < 16:\n",
    "        rows = len(reconst_branch)**0.5\n",
    "    cols = int(np.ceil(len(reconst_branch)/4))\n",
    "    can = ROOT.TCanvas('can1', operator+'_can', rows*1000,750*cols)\n",
    "    can.Divide(rows, cols)\n",
    "    ican = 1\n",
    "\n",
    "    hss=[]\n",
    "    \n",
    "    for r in reconst_branch:\n",
    "        can.cd(ican)\n",
    "        setlogx = lambda x: (x[-1] in ['e','t','m']) and (x[1] != 'j')\n",
    "        can.cd(ican).SetLogx(setlogx(r))\n",
    "        can.cd(ican).SetLogy(set_log)\n",
    "        # def bins\n",
    "        if r=='jj_pt':\n",
    "            bins_c1 = [0,55,80,125,180,270,840]\n",
    "            bins_c2 = bins_c1\n",
    "            bins_c3 = bins_c1\n",
    "        elif r=='jj_m':\n",
    "            bins_c1 = [1000,1450,1800,2300,5300]\n",
    "            bins_c2 = bins_c1\n",
    "            bins_c3 = bins_c1\n",
    "        elif r=='jj_dphi_signed':\n",
    "            bins_c1 = [-3.1416, -2.9452,-2.7489,-2.4435, 0.0, 2.4435, 2.7489, 2.9452, 3.1416]\n",
    "            bins_c2 = bins_c1\n",
    "            bins_c3 = bins_c1\n",
    "        else:\n",
    "            df = df.sort_values(r)\n",
    "            df['cum_c1'] = df['c1_w'].cumsum()\n",
    "            df['cum_c2'] = df['c2_w'].cumsum()\n",
    "            df['cum_c3'] = df['c3_w'].cumsum()\n",
    "            bins_c1, bins_c2, bins_c3 = [df[r][0]], [df[r][0]], [df[r][0]]\n",
    "            for i in range(1,5):\n",
    "                bins_c1 += [ df[df['cum_c1']>=c1_index[i]][r][0] ]\n",
    "                bins_c2 += [ df[df['cum_c2']>=c2_index[i]][r][0] ]\n",
    "                bins_c3 += [ df[df['cum_c3']>=c3_index[i]][r][0] ]\n",
    "            bins_c1 += [df[r].iloc[-1]]\n",
    "            bins_c2 += [df[r].iloc[-1]]\n",
    "            bins_c3 += [df[r].iloc[-1]]\n",
    "\n",
    "        # generate hist\n",
    "        h1 = ROOT.TH1F(r+\"_h1\", 'h1;'+r+';numbers', (len(bins_c1)-1), array('d',bins_c1))\n",
    "        h2 = ROOT.TH1F(r+\"_h2\", 'h2;'+r+';numbers', (len(bins_c2)-1), array('d',bins_c2))\n",
    "        h3 = ROOT.TH1F(r+\"_h3\", 'h3;'+r+';numbers', (len(bins_c3)-1), array('d',bins_c3))\n",
    "\n",
    "        for i in range(len(df)):\n",
    "            h1.Fill(df[r][i], df['c1_w'][i])\n",
    "            h2.Fill(df[r][i], df['c2_w'][i])\n",
    "            h3.Fill(df[r][i], df['c3_w'][i])\n",
    "\n",
    "        h3.SetFillColorAlpha(800, 0.7)\n",
    "        h2.SetFillColorAlpha(860+1,0.7)\n",
    "        h1.SetFillColorAlpha(1,0.7)\n",
    "        h1.SetFillStyle(3944)\n",
    "\n",
    "        #h1.SetLineStyle(3)\n",
    "        h3.SetLineColor(800-3)\n",
    "        h2.SetLineColor(860-3)\n",
    "        h1.SetLineColorAlpha(1, 0.6)\n",
    "\n",
    "        h3.SetLineWidth(1)\n",
    "        h2.SetLineWidth(1)\n",
    "        h1.SetLineWidth(1)\n",
    "\n",
    "        h3.SetMarkerStyle(9)\n",
    "        h2.SetMarkerStyle(9)\n",
    "        h1.SetMarkerStyle(9)\n",
    "\n",
    "        hs = ROOT.THStack (r+\"hs\",\" THStack \")\n",
    "        hs.Add (h3)\n",
    "        hs.Add (h2)\n",
    "        hs.Add (h1)\n",
    "        \n",
    "        hs.Draw (\"NOSTACK, L,h\")\n",
    "        if (r[1] != 'j'):\n",
    "            hs.SetMinimum(1600)\n",
    "        hss += [hs,h1,h2,h3]\n",
    "\n",
    "        legend = ROOT.TLegend(0.75,0.75,0.99,0.99)\n",
    "        legend.AddEntry ( h1 ,operator +' = '+str(c1)+' TeV^(-4)')\n",
    "        legend.AddEntry ( h2 ,operator +' = '+str(c2)+' TeV^(-4)')\n",
    "        legend.AddEntry ( h3 ,operator +' = '+str(c3)+' TeV^(-4)')\n",
    "        legend.Draw (\" same \")\n",
    "        hss += [legend]\n",
    "\n",
    "        hs.GetYaxis().SetTitle(\"Number\")\n",
    "        hs.GetXaxis().SetTitle(r)\n",
    "        '''hs.GetYaxis().SetTitleOffset (0.8)\n",
    "        hs.GetYaxis().SetLabelSize(0.02)\n",
    "        '''\n",
    "        can.cd(ican).SetBottomMargin (0.14)\n",
    "        can.cd(ican).SetTopMargin (0)\n",
    "        can.cd(ican).SetRightMargin(0)\n",
    "        can.cd(ican).SetLeftMargin(0.14)\n",
    "        ican += 1\n",
    "\n",
    "        f = ROOT.TFile(\"histogram.root\",\"recreate\")\n",
    "        f.cd()\n",
    "        h1.Write()\n",
    "        h2.Write()\n",
    "        h3.Write()\n",
    "        f.Close()\n",
    "    \n",
    "    can.Draw()\n",
    "    can.Update()\n",
    "    can.Print('test.pdf')\n",
    "    #can.print ('test.png')\n",
    "    can.SaveAs(generate_output_path(operator, add=reconst_branch[0], format='png'))\n",
    "    can.Clear()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 248,
   "id": "3c998019",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2022-06-08T21:15:30.234250Z",
     "start_time": "2022-06-08T21:15:30.113414Z"
    },
    "code_folding": [
     7,
     11
    ]
   },
   "outputs": [],
   "source": [
    "# function of method 1\n",
    "def plot_operator2(operator, df_all, reconst_branch=reconst_branch[1:], long_format=1):\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",
    "    # prepare bins for hist\n",
    "    df = df_all.copy()\n",
    "\n",
    "    # prepare canvas\n",
    "    rows = 4\n",
    "    if len(reconst_branch) < 16:\n",
    "        rows = int(len(reconst_branch)**0.5)\n",
    "    cols = int(np.ceil(len(reconst_branch)/rows))\n",
    "    if not long_format:\n",
    "        rows, cols = cols, rows \n",
    "    can = ROOT.TCanvas('can1', operator+'_can', rows*1000,750*cols)\n",
    "    can.Divide(rows, cols)\n",
    "    ican = 1\n",
    "\n",
    "    hss=[]\n",
    "    bins_arr = []\n",
    "    \n",
    "    for r in reconst_branch:\n",
    "        set_log = r[-1] in ['e','t','m'] and r[1] != 'j'\n",
    "        can.cd(ican)\n",
    "\n",
    "        # upper part\n",
    "        pad1 = ROOT.TPad (\" pad1 \",\" pad1 \" ,0,0,1,1)\n",
    "        pad1.Draw()\n",
    "        pad1.SetLogy ( set_log )\n",
    "        pad1.cd()\n",
    "\n",
    "        # def bins\n",
    "        max_x = sm_df[r].max()\n",
    "        min_x = sm_df[r].min()\n",
    "        is_jj = (r in ['jj_pt', 'jj_m', 'jj_dphi_signed'])\n",
    "\n",
    "        if is_jj:\n",
    "            if r=='jj_pt':\n",
    "                bins = [0,55,80,125,180,270,840]\n",
    "            elif r=='jj_m':\n",
    "                bins = [1000,1450,1800,2300,5300]\n",
    "            elif r=='jj_dphi_signed':\n",
    "                bins = [-3.1416, -2.9452,-2.7489,-2.4435, 0.0, 2.4435, 2.7489, 2.9452, 3.1416]\n",
    "            h_sm = ROOT.TH1F(r+\"_sm\", 'sm;'+r+';numbers', (len(bins)-1), array('d',bins))\n",
    "            h1 = ROOT.TH1F(r+\"_h1\", 'h1;'+r+';numbers', (len(bins)-1), array('d',bins))\n",
    "            h2 = ROOT.TH1F(r+\"_h2\", 'h2;'+r+';numbers', (len(bins)-1), array('d',bins))\n",
    "            max_x = bins[-1]\n",
    "            min_x = bins[0]\n",
    "        else:\n",
    "            if set_log:\n",
    "                bins = np.linspace(1.7, 1.7, 6).cumprod()\n",
    "                bins = np.insert(bins, 0, 0)\n",
    "                bins = [i/bins[-1]*max_x for i in bins]\n",
    "                '''for i in range(8):\n",
    "                    bins = np.append(bins, bins[-1]*1.7)'''\n",
    "                bins = array('d',bins)  #[i for i in bins if i < max_x])\n",
    "                h_sm = ROOT.TH1F(r+\"_sm\", 'sm;'+r+';numbers', len(bins)-1, bins)\n",
    "                h1 = ROOT.TH1F(r+\"_h1\", 'h1;'+r+';numbers', len(bins)-1, bins)\n",
    "                h2 = ROOT.TH1F(r+\"_h2\", 'h2;'+r+';numbers', len(bins)-1, bins)\n",
    "            else:\n",
    "                h_sm = ROOT.TH1F(r+\"_sm\", 'sm;'+r+';numbers', 8, min_x, max_x)\n",
    "                h1 = ROOT.TH1F(r+\"_h1\", 'h1;'+r+';numbers', 8, min_x, max_x)\n",
    "                h2 = ROOT.TH1F(r+\"_h2\", 'h2;'+r+';numbers', 8, min_x, max_x)\n",
    "                bins = np.linspace(min_x,max_x,8+1)\n",
    "\n",
    "        bins_arr += [bins]\n",
    "\n",
    "        for i in range(len(sm_df)):\n",
    "            h_sm.Fill(sm_df[r][i], sm_df['weight'][i])\n",
    "        for i in range(len(df)):\n",
    "            h1.Fill(df[r][i], df['c1_w'][i])\n",
    "            h2.Fill(df[r][i], df['c2_w'][i])\n",
    "\n",
    "        h_sm.SetFillColorAlpha(18,0.9)\n",
    "        h1.SetFillColorAlpha(800,0)\n",
    "        h2.SetFillColorAlpha(800,0)   \n",
    "        h_sm.SetLineColorAlpha(392,0)\n",
    "        h1.SetLineColorAlpha(38, 1)\n",
    "        h2.SetLineColorAlpha(46, 1)\n",
    "        h_sm.SetMarkerStyle(20)\n",
    "        h1.SetMarkerStyle(21)\n",
    "        h2.SetMarkerStyle(22)\n",
    "        h1.SetMarkerSize(3)\n",
    "        h2.SetMarkerSize(3)\n",
    "        h1.SetMarkerColor(38)\n",
    "        h2.SetMarkerColor(46)\n",
    "        h_sm.SetMarkerColorAlpha(800,0)\n",
    "        h2.SetLineWidth(1)\n",
    "        h1.SetLineWidth(1)\n",
    "\n",
    "        hs = ROOT.THStack (r+\"hs\",\" THStack \")\n",
    "        hs.Add (h_sm)\n",
    "        hs.Add (h1)\n",
    "        hs.Add (h2)\n",
    "        \n",
    "        hs.Draw (\"NOSTACK, h\")\n",
    "        hs.SetMinimum(0.1)\n",
    "        if not(r in ['j1_pt','wy_m', 'w_mt']):\n",
    "            hs.SetMinimum(1)\n",
    "        '''n, m = len(bins)-1, 0\n",
    "        while (n >= 1) and (h2.GetBinContent(n) <= 0.5):\n",
    "            n -= 1\n",
    "        max_x = bins[n]\n",
    "        while (m < len(bins)-1) and (h2.GetBinContent(n) <= 0.5):\n",
    "            m += 1\n",
    "        min_x = bins[m]'''\n",
    "        hs.GetXaxis().SetRangeUser(min_x, max_x) \n",
    "        if r=='jj_pt':\n",
    "            hs.SetMinimum(70)\n",
    "        elif r=='jj_m':\n",
    "            hs.SetMinimum(100)\n",
    "        hss += [hs,h1,h2,h_sm]\n",
    "\n",
    "        legend = ROOT.TLegend(0.65,0.75,0.99,0.99)\n",
    "        legend.AddEntry(h_sm, \"Standard Model\", \"f\")\n",
    "        legend.AddEntry ( h1 ,'f_{0} /#Lambda^{3} = {1} TeV^{2}'.format({operator}, c1,{-4},{4}),\"lp\")\n",
    "        legend.AddEntry ( h2 ,'f_{0} /#Lambda^{3} = {1} TeV^{2}'.format({operator}, c2,{-4},{4}),\"lp\")\n",
    "        legend.Draw (\" same \")\n",
    "        legend.SetFillColorAlpha(0, 0.8)\n",
    "        hss += [legend]\n",
    "\n",
    "        hs.GetYaxis().SetTitle(\"Number\")\n",
    "        hs.GetXaxis().SetTitle(unit_dict[r])\n",
    "        pad1.SetBottomMargin (0.25)\n",
    "        pad1.SetTopMargin (0)\n",
    "        pad1.SetRightMargin(0)\n",
    "        pad1.SetLeftMargin(0.13)\n",
    "\n",
    "\n",
    "        # lower part\n",
    "        pad2 = ROOT.TPad (\" pad2 \",\" pad2 \" ,0 ,0 ,1 ,0.25)\n",
    "        pad2.Draw()\n",
    "        pad2.cd()\n",
    "        \n",
    "        h1_low = h1-h_sm\n",
    "        h1_low.Divide(h_sm)\n",
    "        h1_low.Draw()\n",
    "\n",
    "        h2_low = h2-h_sm\n",
    "        h2_low.Divide(h_sm)\n",
    "        h2_low.Draw('same')\n",
    "\n",
    "        h1_low.GetYaxis().SetRangeUser(-0.2,0.2)\n",
    "        h1_low.GetXaxis().SetRangeUser(min_x, max_x) \n",
    "        h1_low.GetYaxis().SetNdivisions (207)\n",
    "        h1_low.SetMarkerSize(2)\n",
    "        h2_low.SetMarkerSize(2)\n",
    "\n",
    "        h1_markers, h2_markers = [], []\n",
    "        for i in range(len(bins)):\n",
    "            if (h1_low.GetBinContent(i) > 0.2):\n",
    "                h1_markers += [ROOT.TMarker(h1_low.GetBinCenter(i), 0.2,21)]\n",
    "            elif (h1_low.GetBinContent(i) < -0.2):\n",
    "                h1_markers += [ROOT.TMarker(h1_low.GetBinCenter(i), -0.2,21)]\n",
    "            if (h2_low.GetBinContent(i) > 0.2):\n",
    "                h2_markers += [ROOT.TMarker(h2_low.GetBinCenter(i), 0.2,22)]\n",
    "            elif (h2_low.GetBinContent(i) < -0.2):\n",
    "                H2_markers += [ROOT.TMarker(h2_low.GetBinCenter(i), -0.2,23)] \n",
    "        for j in h1_markers:\n",
    "            j.SetMarkerSize(2)\n",
    "            j.SetMarkerColor(38)\n",
    "            j.Draw('same')\n",
    "        for j in h2_markers:\n",
    "            j.SetMarkerSize(2)\n",
    "            j.SetMarkerColor(46)\n",
    "            j.Draw('same')\n",
    "        hss += [h1_markers, h2_markers]\n",
    "        #print(markers, len(markers))\n",
    "\n",
    "        '''if set_log and r!='jj_m':\n",
    "            min_x = 0\n",
    "        elif r == 'jj_m':\n",
    "            min_x =1000\n",
    "        elif 'dphi' in r:\n",
    "            min_x = -3.1'''\n",
    "        line = ROOT.TLine (min_x, 0, max_x+1000,0)\n",
    "        line.SetLineColor (1)\n",
    "        line.SetLineWidth (1)\n",
    "        line.Draw (\" same \")\n",
    "        hss += [h1_low,h2_low,line]\n",
    "        \n",
    "        h1_low.GetYaxis().SetTitle (\"ALL/SM-1  \")\n",
    "        h1_low.GetXaxis().SetTitle(unit_dict[r])\n",
    "        h1_low.GetXaxis().SetTitleSize (0.25)\n",
    "        h1_low.GetYaxis().SetTitleSize (0.14)\n",
    "        h1_low.GetXaxis().SetLabelSize(0.15)\n",
    "        h1_low.GetYaxis().SetLabelSize(0.14)\n",
    "        h1_low.GetYaxis().SetTitleOffset (0.27)\n",
    "        h1_low.GetXaxis().SetTitleOffset (0.7)\n",
    "        pad2.SetTopMargin (0)\n",
    "        pad2.SetRightMargin (0)\n",
    "        pad2.SetLeftMargin(0.13)\n",
    "        pad2.SetBottomMargin (0.45)\n",
    "\n",
    "        ican += 1\n",
    "\n",
    "        '''f = ROOT.TFile(\"histogram.root\",\"recreate\")\n",
    "        f.cd()\n",
    "        h1.Write()\n",
    "        h2.Write()\n",
    "        f.Close()'''\n",
    "    \n",
    "    can.Draw()\n",
    "    can.Update()\n",
    "    can.Print('test.pdf')\n",
    "    #can.Print('test.png')\n",
    "    #can.SaveAs(generate_output_path(operator))\n",
    "    can.Clear()\n",
    "\n",
    "    return bins_arr"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 276,
   "id": "c74815d9",
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n"
     ]
    }
   ],
   "source": [
    "#tmp = plot_operator2('T5', dfs_t[3], reconst_branch=l2, long_format=0)\n",
    "#tmp = plot_operator2('M2', dfs_m[2], reconst_branch=l1,long_format=0)\n",
    "#tmp = plot_operator2('M2', dfs_m[2], reconst_branch=['lep_pt','gam_e','wy_m','wy_pt'],long_format=0)\n",
    "# tmp = plot_operator2('T6', dfs_t[4],reconst_branch=['dphi_lepy','dphi_wy', 'dr_lepy','dr_wy'],long_format=0)\n",
    "tmp = plot_operator2('T6', dfs_t[4], reconst_branch=['lep_pt','gam_e','wy_m','dphi_lepy','dr_lepy'],long_format=0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 226,
   "id": "f1705403",
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Warning in <TCanvas::Constructor>: Deleting canvas with same name: can1\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/M0.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/M1.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/M2.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/M3.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/M4.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/M5.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/M7.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/T0.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/T1.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/T2.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/T5.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/T6.pdf has been created\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: jj_dphi_signed_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: gam_e_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: lep_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j0_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: j1_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: w_mt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_m_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: wy_pt_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dphi_wy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_lepy_h2 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_sm (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h1 (Potential memory leak).\n",
      "Warning in <TROOT::Append>: Replacing existing TH1: dr_wy_h2 (Potential memory leak).\n",
      "Info in <TCanvas::Print>: pdf file test.pdf has been created\n",
      "Info in <TCanvas::Print>: pdf file ./output_pdf/rebin_reconstructed/T7.pdf has been created\n"
     ]
    }
   ],
   "source": [
    "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",
    "l3 = ['dphi_lepy','dphi_wy','dr_lepy','dr_wy']\n",
    "\n",
    "for i in range(len(operators_m)):\n",
    "    tmp = plot_operator2(operators_m[i], dfs_m[i], reconst_branch=l2+l1+l3)\n",
    "for i in range(len(operators_t)):\n",
    "    tmp = plot_operator2(operators_t[i], dfs_t[i], reconst_branch=l2+l1+l3)\n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "3c998019",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2022-05-25T20:34:05.049427Z",
     "start_time": "2022-05-25T20:34:05.049401Z"
    },
    "code_folding": [
     7,
     11
    ]
   },
   "outputs": [],
   "source": [
    "def plot_2_branch_rank(operator, branches, bins, df, weight_name = 'weight', c=1,isPDF=True):\n",
    "    '''\n",
    "    ranked by correlation\n",
    "    '''\n",
    "    df = df.copy()\n",
    "    hs = []\n",
    "    # add histogram\n",
    "    for b in range(len(branches)):\n",
    "        for a in range(b):    \n",
    "            \n",
    "            # variable name\n",
    "            var_x = branches[a]\n",
    "            var_y = branches[b]\n",
    "            bins_x = bins[a]\n",
    "            bins_y = bins[b]\n",
    " \n",
    "            h1 = ROOT.TH2F(var_x+var_y, \";\"+var_x+\";\"+var_y, len(bins_x)-1, array('d',bins_x), len(bins_y)-1, array('d',bins_y))\n",
    "\n",
    "            '''if operator[0] == 'T':\n",
    "                for i in range(len(df)):\n",
    "                    h1.Fill(df[var_x][i], df[var_y][i], df['c2_w'][i])\n",
    "            else:\n",
    "                for i in range(len(df)):\n",
    "                    h1.Fill(df[var_x][i], df[var_y][i], df['c1_w'][i])'''\n",
    "            for i in range(len(df)):\n",
    "                h1.Fill(df[var_x][i], df[var_y][i], df[weight_name][i])\n",
    "\n",
    "            h1.Draw()\n",
    "            hs += [h1]\n",
    "\n",
    "    hs = sorted(hs, key=lambda x: x.GetCorrelationFactor(1,2), reverse=True)\n",
    "\n",
    "    #ready to plot\n",
    "    nvars = len(branches)   # number of variables to plot\n",
    "    can = ROOT.TCanvas('can1', '',4000,4000)\n",
    "    nplots = int(np.ceil(((nvars-1) * nvars/2) **0.5))\n",
    "    can.Divide(nplots, nplots)\n",
    "\n",
    "    ican = 1\n",
    "    for h in hs:\n",
    "        can.cd(ican).SetRightMargin(0)\n",
    "        can.cd(ican).SetLeftMargin(0.15)\n",
    "        can.cd(ican).SetTopMargin(0)\n",
    "        can.cd(ican).SetBottomMargin(0.15)\n",
    "        if (ican % nplots == 0) or (ican == len(hs)):\n",
    "            can.cd(ican).SetRightMargin(.15)\n",
    " \n",
    "        can.cd(ican).SetLogz(True)\n",
    "        h.DrawClone('Colz')\n",
    "        t = ROOT.TLatex()\n",
    "        t.SetNDC()\n",
    "        t.SetTextColor( 1 )\n",
    "        t.SetTextSize( 0.05 )\n",
    "        t.SetTextAlign( 12 )\n",
    "        t.DrawLatex( 0.2, 0.95, 'correlation coefficient is %.3f'%h.GetCorrelationFactor(1,2) )\n",
    "        ican += 1\n",
    "\n",
    "    can.Update()\n",
    "    can.Draw()\n",
    "    #can.Print('test.png')\n",
    "\n",
    "    can.SaveAs(generate_output_path(operator, add='2d'))\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "cdb1b848",
   "metadata": {},
   "outputs": [],
   "source": [
    "#o = 'M2'\n",
    "\n",
    "l1 = ['gam_e','lep_pt','wy_m','wy_pt']\n",
    "l2 = ['jj_m','jj_pt','jj_dphi_signed']\n",
    "vs = l2+l1\n",
    "bins = []\n",
    "\n",
    "for o in operators:\n",
    "    df = generate_df(o)\n",
    "    bins += plot_operator2(o, df, reconst_branch=vs)\n",
    "    plot_2_branch_rank(o, vs, bins, df)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "3c998019",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2022-06-08T21:15:30.234250Z",
     "start_time": "2022-06-08T21:15:30.113414Z"
    },
    "code_folding": [
     7,
     11
    ]
   },
   "outputs": [],
   "source": [
    "# function of method 1\n",
    "def plot_operators(operators, dfs, reconst_branch=reconst_branch[1:], long_format=1):\n",
    "    # load values of coupling\n",
    "    c = 1\n",
    "    if operators[0][0] == 'T':\n",
    "        weight_col = 'c2_w'\n",
    "    else:\n",
    "        weight_col = 'c1_w'\n",
    "\n",
    "    # prepare canvas\n",
    "    rows = 4\n",
    "    if len(reconst_branch) < 16:\n",
    "        rows = int(len(reconst_branch)**0.5)\n",
    "    cols = int(np.ceil(len(reconst_branch)/rows))\n",
    "    if not long_format:\n",
    "        rows, cols = cols, rows \n",
    "    can = ROOT.TCanvas('can1', operators[0][0] +'_can', rows*1000,1000*cols)\n",
    "    can.Divide(rows, cols)\n",
    "    ican = 1\n",
    "\n",
    "    hss=[]\n",
    "    colors = [2,3,4,6,7,8,9]\n",
    "    markers = [20,21,22,23,33,34,47]\n",
    "    \n",
    "    for r in reconst_branch:\n",
    "        can.cd(ican)\n",
    "        can.cd(ican).SetLogy (1)\n",
    "\n",
    "        # def bins\n",
    "        max_x = sm_df[r].max()\n",
    "        min_x = sm_df[r].min()\n",
    "        is_jj = (r in ['jj_pt', 'jj_m', 'jj_dphi_signed'])\n",
    "\n",
    "        if is_jj:\n",
    "            if r=='jj_pt':\n",
    "                bins = [0,55,80,125,180,270,840]\n",
    "            elif r=='jj_m':\n",
    "                bins = [1000,1450,1800,2300,5300]\n",
    "            elif r=='jj_dphi_signed':\n",
    "                bins = [-3.1416, -2.9452,-2.7489,-2.4435, 0.0, 2.4435, 2.7489, 2.9452, 3.1416]\n",
    "            h_sm = ROOT.TH1F(r+\"_sm\", 'sm;'+r+';numbers', (len(bins)-1), array('d',bins))\n",
    "        else:\n",
    "            bins = np.linspace(1.7,1.7, 6).cumprod()\n",
    "            bins = np.insert(bins, 0, 0)\n",
    "            bins = [i/bins[-1]*max_x for i in bins]\n",
    "            h_sm = ROOT.TH1F(r+\"_sm\", 'sm;'+r+';numbers', len(bins)-1, array('d',bins))\n",
    "\n",
    "        for i in range(len(sm_df)):\n",
    "            h_sm.Fill(sm_df[r][i], sm_df['weight'][i])\n",
    "\n",
    "        h_sm.SetFillColorAlpha(18,0.9)\n",
    "        h_sm.SetLineColorAlpha(392,0)\n",
    "        h_sm.SetMarkerColorAlpha(800,0)\n",
    "\n",
    "        hs = ROOT.THStack (r+\"hs\",\" THStack \")\n",
    "        hs.Add (h_sm)\n",
    "\n",
    "        hos = []\n",
    "        for j in range(len(operators)):\n",
    "            operator = operators[j]\n",
    "            ho = ROOT.TH1F(r+operator, operator+';'+r+';numbers', len(bins)-1, array('d',bins))\n",
    "\n",
    "            df = dfs[j].copy()\n",
    "            for i in range(len(df)):\n",
    "                ho.Fill(df[r][i], df[weight_col][i])\n",
    "    \n",
    "            ho.SetFillColorAlpha(800,0)\n",
    "            ho.SetLineColorAlpha(colors[j], 1)\n",
    "            ho.SetMarkerColor(colors[j])\n",
    "            ho.SetMarkerStyle(markers[j])\n",
    "            ho.SetMarkerSize(3)\n",
    "            ho.SetLineWidth(1)\n",
    "\n",
    "            hs.Add (ho)\n",
    "            hos += [ho]\n",
    "\n",
    "        legend = ROOT.TLegend(0.65,0.65,0.99,0.99)\n",
    "        legend.AddEntry(h_sm, \"Standard Model\", \"f\")\n",
    "        for i in range(len(hos)):\n",
    "            legend.AddEntry (hos[i],'f_{0} /#Lambda^{3} = {1} TeV^{2}'.format({operators[i]}, c,{-4},{4}),\"lp\")\n",
    "\n",
    "        hs.Draw (\"NOSTACK, h\")\n",
    "        legend.Draw (\" same \")\n",
    "        hs.SetMinimum(0.1)\n",
    "        if not(r in ['j1_pt','wy_m', 'w_mt']):\n",
    "            hs.SetMinimum(1)\n",
    "        if r=='jj_pt':\n",
    "            hs.SetMinimum(50)\n",
    "        elif r=='jj_m':\n",
    "            hs.SetMinimum(100)\n",
    "\n",
    "        hss += [legend, hs, h_sm, hos]\n",
    "\n",
    "        hs.GetYaxis().SetTitle(\"Number\")\n",
    "        hs.GetXaxis().SetTitle(unit_dict[r])\n",
    "        can.cd(ican).SetBottomMargin (0.14)\n",
    "        can.cd(ican).SetTopMargin (0)\n",
    "        can.cd(ican).SetRightMargin(0)\n",
    "        can.cd(ican).SetLeftMargin(0.125)\n",
    "\n",
    "        ican += 1\n",
    "    \n",
    "    can.Draw()\n",
    "    can.Update()\n",
    "    can.Print('test.pdf')\n",
    "    #can.Print('test.png')\n",
    "    can.SaveAs(generate_output_path(operators[0][0] +'all'))\n",
    "    can.Clear()\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "30157b87",
   "metadata": {},
   "outputs": [],
   "source": [
    "plot_operators(operators_t, dfs_t, reconst_branch=['wy_m'], long_format=0)\n",
    "plot_operators(operators_m, dfs_m, reconst_branch=['wy_m'], long_format=0)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 87,
   "id": "6c3c5be8",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[array('d', [0.0, 61.021563263061026, 103.73665754720373, 176.35231783024634, 299.79894031141873, 509.65819852941183, 866.4189375000001]),\n",
       " array('d', [0.0, 141.90015648315537, 241.2302660213641, 410.09145223631896, 697.1554688017421, 1185.1642969629615, 2014.7793048370345]),\n",
       " array('d', [0.0, 184.31483813787406, 313.3352248343859, 532.669882218456, 905.538799771375, 1539.4159596113375, 2617.0071313392737]),\n",
       " array('d', [0.0, 69.56131269922787, 118.25423158868738, 201.03219370076852, 341.75472929130643, 580.9830397952209, 987.6711676518755])]"
      ]
     },
     "execution_count": 87,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "tmp=[array('d', [0.0, 61.02, 103.73, 176.35, 299.79,369.75,509.65, 866.41]),\n",
    " array('d', [0.0, 141.90, 241.23, 410.09, 697.15, 1185.16, 2014.77]),\n",
    " array('d', [0.0, 184.31, 313.33, 532.66, 905.53, 1539.41, 2617.00]),\n",
    " array('d', [0.0, 69.56, 118.25, 201.03, 341.75, 580.98, 987.67])]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 273,
   "id": "d7f756af",
   "metadata": {},
   "outputs": [],
   "source": [
    "def plot_contribution(operators, dfs, branches, y_max=0.1, y_min=-0.1, col_width=0.17):\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.19\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, hist\")\n",
    "\n",
    "            hs.SetMinimum(y_min)\n",
    "            hs.SetMaximum(y_max)\n",
    "            hs.GetYaxis().SetNdivisions (407)\n",
    "            hs.GetXaxis().SetNdivisions (407)\n",
    "            hs.GetYaxis().SetTickLength(0.02)\n",
    "            hs.GetXaxis().SetTickLength(0.02)\n",
    "            hs.GetXaxis().SetTitle(unit_dict[branch]+' ')\n",
    "            hs.GetXaxis().SetTitleOffset (0.85)\n",
    "            hs.GetXaxis().SetTitleSize (0.115)\n",
    "            hs.GetYaxis().SetTitle('Ratio to SM   ')\n",
    "            hs.GetYaxis().SetTitleOffset (1)\n",
    "            hs.GetYaxis().SetTitleSize (0.115)\n",
    "            hs.GetXaxis().SetLabelSize (0.09)\n",
    "            hs.GetYaxis().SetLabelSize (0.11)\n",
    "\n",
    "            h_i.SetLineColorAlpha(38,1)\n",
    "            h_a.SetLineColorAlpha(ROOT.kBlack,.8)\n",
    "            h_q.SetLineColorAlpha(46, 1)\n",
    "            h_i.SetLineStyle(ROOT.kDashed)\n",
    "            h_q.SetLineStyle(ROOT.kDotted)\n",
    "            h_a.SetLineWidth(1)\n",
    "            h_i.SetMarkerSize(0)\n",
    "            h_q.SetMarkerSize(0)\n",
    "            h_a.SetMarkerSize(0)\n",
    "            h_i.SetMarkerColorAlpha(800,0)\n",
    "            h_a.SetMarkerColorAlpha(800,0)\n",
    "            h_q.SetMarkerColorAlpha(800,0)\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.25)\n",
    "            elif j == len(branches)-1:\n",
    "                c=0.1\n",
    "                if operator[0]=='M':\n",
    "                    c=1\n",
    "\n",
    "                if i == len(dfs)-1:\n",
    "                    legend = ROOT.TLegend(0.02,0.45,0.8,0.97)\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",
    "                else:\n",
    "                    legend = ROOT.TLegend(0.02,0.8,0.8,0.9)\n",
    "                legend.SetHeader('f_{0} /#Lambda^{3} = {1} TeV^{2}'.format({operator}, c,{-4},{4}),\"lp\")\n",
    "                legend.Draw (\" same \")\n",
    "                legend.SetFillColorAlpha(0, 0.8)\n",
    "                hss += [legend]\n",
    "            if i == 0:\n",
    "                hs.GetYaxis().SetTitleSize (0.095)\n",
    "                hs.GetYaxis().SetLabelSize (0.085)\n",
    "                hs.GetYaxis().SetTitleOffset (1.2)\n",
    "                hss+=[hs]\n",
    "                pad.SetBottomMargin (0.25)\n",
    "    \n",
    "    can.Draw()\n",
    "    can.Update()\n",
    "    can.Print(branches[0]+'test.pdf') \n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 274,
   "id": "568061fa",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "complete first\n"
     ]
    },
    {
     "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.23)\n",
    "print('complete first')\n",
    "plot_contribution(os,df_arr,bs2,y_max=0.005, y_min=-0.0025, col_width=0.23)"
   ]
  },
  {
   "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
}