UnROOT_benchmark.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[32m\u001b[1m      Status\u001b[22m\u001b[39m `/home/jiling/UnROOT_RDataFrame_MiniBenchmark/Project.toml`\n",
      " \u001b[90m [68837c9b] \u001b[39mFHist v0.6.1\n",
      " \u001b[90m [7073ff75] \u001b[39mIJulia v1.23.2\n",
      " \u001b[90m [3a55db76] \u001b[39mLVCyl v0.1.0 `https://github.com/JuliaHEP/LVCyl.jl#master`\n",
      " \u001b[90m [f517fe37] \u001b[39mPolyester v0.4.2\n",
      " \u001b[90m [3cd96dde] \u001b[39mUnROOT v0.5.1\n"
     ]
    }
   ],
   "source": [
    "using Pkg\n",
    "Pkg.status()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "┌ Info: Precompiling UnROOT [3cd96dde-e98d-4713-81e9-a4a1b0235ce9]\n",
      "└ @ Base loading.jl:1423\n",
      "┌ Info: Precompiling FHist [68837c9b-b678-4cd5-9925-8a54edc8f695]\n",
      "└ @ Base loading.jl:1423\n",
      "┌ Info: Precompiling Polyester [f517fe37-dbe3-4b94-8317-1923a5111588]\n",
      "└ @ Base loading.jl:1423\n",
      "┌ Info: Precompiling LVCyl [3a55db76-103f-4f27-8e51-63f2a02e4e27]\n",
      "└ @ Base loading.jl:1423\n"
     ]
    }
   ],
   "source": [
    "using UnROOT, FHist, Polyester, LVCyl"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "\"./Run2012BC_DoubleMuParked_Muons.root\""
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# http://opendata.web.cern.ch/record/12341\n",
    "rootfname = \"./Run2012BC_DoubleMuParked_Muons.root\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "2.0903061451390386"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "filesize(rootfname) / 1024^3 # in GB"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "const r = ROOTFile(rootfname);"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "const mytree = LazyTree(r, \"Events\");"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div class=\"data-frame\"><p>3 rows × 6 columns (omitted printing of 1 columns)</p><table class=\"data-frame\"><thead><tr><th></th><th>nMuon</th><th>Muon_pt</th><th>Muon_eta</th><th>Muon_phi</th><th>Muon_mass</th></tr><tr><th></th><th title=\"UInt32\">UInt32</th><th title=\"SubArray{Float32, 1, Vector{Float32}, Tuple{UnitRange{Int64}}, true}\">SubArra…</th><th title=\"SubArray{Float32, 1, Vector{Float32}, Tuple{UnitRange{Int64}}, true}\">SubArra…</th><th title=\"SubArray{Float32, 1, Vector{Float32}, Tuple{UnitRange{Int64}}, true}\">SubArra…</th><th title=\"SubArray{Float32, 1, Vector{Float32}, Tuple{UnitRange{Int64}}, true}\">SubArra…</th></tr></thead><tbody><tr><th>1</th><td>2</td><td>[10.7637, 15.7365]</td><td>[1.06683, -0.563787]</td><td>[-0.0342727, 2.54262]</td><td>[0.105658, 0.105658]</td></tr><tr><th>2</th><td>2</td><td>[10.5385, 16.3271]</td><td>[-0.42778, 0.349225]</td><td>[-0.274792, 2.53978]</td><td>[0.105658, 0.105658]</td></tr><tr><th>3</th><td>1</td><td>[3.27533]</td><td>[2.21086]</td><td>[-1.22341]</td><td>[0.105658]</td></tr></tbody></table></div>"
      ],
      "text/latex": [
       "\\begin{tabular}{r|cccccc}\n",
       "\t& nMuon & Muon\\_pt & Muon\\_eta & Muon\\_phi & Muon\\_mass & \\\\\n",
       "\t\\hline\n",
       "\t& UInt32 & SubArra… & SubArra… & SubArra… & SubArra… & \\\\\n",
       "\t\\hline\n",
       "\t1 & 2 & [10.7637, 15.7365] & [1.06683, -0.563787] & [-0.0342727, 2.54262] & [0.105658, 0.105658] & $\\dots$ \\\\\n",
       "\t2 & 2 & [10.5385, 16.3271] & [-0.42778, 0.349225] & [-0.274792, 2.53978] & [0.105658, 0.105658] & $\\dots$ \\\\\n",
       "\t3 & 1 & [3.27533] & [2.21086] & [-1.22341] & [0.105658] & $\\dots$ \\\\\n",
       "\\end{tabular}\n"
      ],
      "text/plain": [
       "\u001b[1m Row \u001b[0m│\u001b[1m nMuon  \u001b[0m\u001b[1m Muon_pt          \u001b[0m\u001b[1m Muon_eta         \u001b[0m\u001b[1m Muon_phi         \u001b[0m\u001b[1m Muon_mass\u001b[0m ⋯\n",
       "\u001b[1m     \u001b[0m│\u001b[90m UInt32 \u001b[0m\u001b[90m SubArray{Float3  \u001b[0m\u001b[90m SubArray{Float3  \u001b[0m\u001b[90m SubArray{Float3  \u001b[0m\u001b[90m SubArray{\u001b[0m ⋯\n",
       "─────┼──────────────────────────────────────────────────────────────────────────\n",
       " 1   │ 2       Float32[10.76369  Float32[1.066827  Float32[-0.03427  Float32[0 ⋯\n",
       " 2   │ 2       Float32[10.53849  Float32[-0.42778  Float32[-0.27479  Float32[0 ⋯\n",
       " 3   │ 1       Float32[3.275326  Float32[2.210855  Float32[-1.22341  Float32[0 ⋯\n",
       "\u001b[36m                                                               2 columns omitted\u001b[0m\n"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "mytree[begin:3]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "reco_zz_to_4l (generic function with 1 method)"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "const z_mass = 91.2\n",
    "const LV32 = LorentzVectorCyl{Float32}\n",
    "function reco_zz_to_4l(pts, etas, phis, masses, charges)\n",
    "    idx = [Int[], Int[]]\n",
    "\n",
    "    # Find first lepton pair with invariant mass closest to Z mass\n",
    "    best_mass = -Inf\n",
    "    best_i1 = best_i2 = 1\n",
    "    for i1 in eachindex(pts), i2 in (i1 + 1):lastindex(pts)\n",
    "        charges[i1] == charges[i2] && continue\n",
    "        p1 = LV32(pts[i1], etas[i1], phis[i1], masses[i1])\n",
    "        p2 = LV32(pts[i2], etas[i2], phis[i2], masses[i2])\n",
    "        this_mass = (p1+p2).mass\n",
    "        if (abs(z_mass - this_mass) < abs(z_mass - best_mass))\n",
    "            best_mass = this_mass\n",
    "            best_i1 = i1\n",
    "            best_i2 = i2\n",
    "        end\n",
    "    end\n",
    "    push!(idx[1], best_i1)\n",
    "    push!(idx[1], best_i2)\n",
    "\n",
    "    #Reconstruct second Z from remaining lepton pair\n",
    "    for i in 1:4\n",
    "        if (i != best_i1 && i != best_i2)\n",
    "            push!(idx[2], i)\n",
    "        end\n",
    "    end\n",
    "\n",
    "    # Return indices of the pairs building two Z bosons\n",
    "    return idx\n",
    "end\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "filter_z_dr (generic function with 1 method)"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "function filter_z_dr(idx, etas, phis)\n",
    "    for pair in idx\n",
    "        i1, i2 = pair\n",
    "        dr = sqrt((etas[i1]-etas[i2])^2 + (phis[i1]-phis[i2])^2)\n",
    "        dr < 0.02 && return false;\n",
    "    end\n",
    "    return true\n",
    "end"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "filter_z_candidates (generic function with 1 method)"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "function filter_z_candidates(Z_mass)\n",
    "    40 < Z_mass[1] < 120 || return false\n",
    "    12 < Z_mass[2] < 120 || return false\n",
    "    return true\n",
    "end"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "compute_z_masses_4l (generic function with 1 method)"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "function compute_z_masses_4l(idx, pts, etas, phis, masses)\n",
    "    z_masses = zeros(2)\n",
    "    for (i, pair) in enumerate(idx)\n",
    "        i1, i2 = pair\n",
    "        p1 = LV32(pts[i1], etas[i1], phis[i1], masses[i1])\n",
    "        p2 = LV32(pts[i2], etas[i2], phis[i2], masses[i2])\n",
    "        z_masses[i] = (p1+p2).mass\n",
    "    end\n",
    "   return abs(z_masses[1] - z_mass) < abs(z_masses[2] - z_mass) ?\n",
    "      z_masses : reverse!(z_masses);\n",
    "end"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "compute_higgs_mass_4l (generic function with 1 method)"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "function compute_higgs_mass_4l(idx, pts, etas, phis, masses)\n",
    "    ((i1, i2), (i3, i4)) = idx\n",
    "    p1 = LV32(pts[i1], etas[i1], phis[i1], masses[i1]);\n",
    "    p2 = LV32(pts[i2], etas[i2], phis[i2], masses[i2]);\n",
    "    p3 = LV32(pts[i3], etas[i3], phis[i3], masses[i3]);\n",
    "    p4 = LV32(pts[i4], etas[i4], phis[i4], masses[i4]);\n",
    "    return (p1 + p2 + p3 + p4).mass\n",
    "end"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      " 23.127858 seconds (18.04 M allocations: 24.054 GiB, 19.15% gc time)\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div style=\"display: flex;\">\n",
       "    <div style=\"float:left; margin:5px\"><svg width=\"250\" height=\"200\" version=\"1.1\" xmlns=\"http://www.w3.org/2000/svg\">\n",
       "    <polyline points=\"12.5,180.0,12.5,19.999999999999996,18.92857142857143,19.999999999999996,18.92857142857143,130.56237879767292,25.357142857142858,130.56237879767292,25.357142857142858,130.58536235006824,31.78571428571429,130.58536235006824,31.78571428571429,131.13696760755585,38.214285714285715,131.13696760755585,38.214285714285715,134.6189757954464,44.642857142857146,134.6189757954464,44.642857142857146,139.20419449831215,51.07142857142858,139.20419449831215,51.07142857142858,141.36464842347195,57.50000000000001,141.36464842347195,57.50000000000001,144.08819938231701,63.92857142857143,144.08819938231701,63.92857142857143,148.58148387560152,70.35714285714286,148.58148387560152,70.35714285714286,149.46635064282123,76.78571428571429,149.46635064282123,76.78571428571429,152.71852330675858,83.21428571428572,152.71852330675858,83.21428571428572,153.86770092652446,89.64285714285715,153.86770092652446,89.64285714285715,155.69489334195217,96.07142857142858,155.69489334195217,96.07142857142858,158.02772391007684,102.50000000000001,158.02772391007684,102.50000000000001,159.4641959347842,108.92857142857143,159.4641959347842,108.92857142857143,160.211161387632,115.35714285714286,160.211161387632,115.35714285714286,162.00387847446672,121.78571428571429,162.00387847446672,121.78571428571429,163.15305609423257,128.21428571428572,163.15305609423257,128.21428571428572,164.40565969977735,134.64285714285717,164.40565969977735,134.64285714285717,164.12985707103354,141.07142857142858,164.12985707103354,141.07142857142858,165.65826330532215,147.5,165.65826330532215,147.5,166.79594914889034,153.92857142857144,166.79594914889034,153.92857142857144,167.50843927314517,160.35714285714286,167.50843927314517,160.35714285714286,168.89894419306185,166.7857142857143,168.89894419306185,166.7857142857143,169.3011563599799,173.21428571428572,169.3011563599799,173.21428571428572,170.20900667959492,179.64285714285717,170.20900667959492,179.64285714285717,170.64569417510594,186.07142857142858,170.64569417510594,186.07142857142858,171.43862673274438,192.50000000000003,171.43862673274438,192.50000000000003,172.51885369532428,198.92857142857144,172.51885369532428,198.92857142857144,173.19686849098613,205.35714285714286,173.19686849098613,205.35714285714286,173.82891618185735,211.7857142857143,173.82891618185735,211.7857142857143,173.77145730086906,218.21428571428572,173.77145730086906,218.21428571428572,174.24262012497306,224.64285714285717,174.24262012497306,224.64285714285717,174.88615959204193,231.07142857142858,174.88615959204193,231.07142857142858,118.87524240465417,237.50000000000003,118.87524240465417,237.5,180.0,12.5,180.0\" stroke=\"black\" fill=\"#ffffff00\" stroke-width=\"1\"/>\n",
       "    <polyline points=\"12.5,180.0,237.5,180.0\" stroke=\"black\" stroke-width=\"1\"/>\n",
       "    <text x=\"12.5\" y=\"190.0\" dominant-baseline=\"middle\" text-anchor=\"start\" fill=\"black\">70.0</text>\n",
       "    <text x=\"237.5\" y=\"190.0\" dominant-baseline=\"middle\" text-anchor=\"end\" fill=\"black\">180.0</text>\n",
       "</svg>\n",
       "</div>\n",
       "    <div style=\"float:left; margin:5px; max-width: 50%; display:flex; justify-content:center; align-items:center;\">\n",
       "        <ul>\n",
       "            <li>edges: 70.0:3.142857142857143:180.0</li>\n",
       "            <li>bin counts: [13923.0, 4302.0, 4300.0, 4252.0, 3949.0, 3550.0, 3362.0, 3125.0, 2734.0, 2657.0  …  852.0, 814.0, 745.0, 651.0, 592.0, 537.0, 542.0, 501.0, 445.0, 5319.0]</li>\n",
       "            <li>maximum count: 13923.0</li>\n",
       "            <li>total count: 80487.0</li>\n",
       "        </ul>\n",
       "    </div>\n",
       "</div>\n",
       "\n"
      ],
      "text/plain": [
       "                  \u001b[90m┌                              ┐\u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 70.0\u001b[90m, \u001b[0m 73.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█████████████████████\u001b[39m\u001b[38;5;2m \u001b[39m\u001b[0m 13923.0\u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 73.0\u001b[90m, \u001b[0m 76.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██████\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 4302.0                \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 76.0\u001b[90m, \u001b[0m 79.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██████\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 4300.0                \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 79.0\u001b[90m, \u001b[0m 82.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██████\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 4252.0                \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 83.0\u001b[90m, \u001b[0m 86.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█████\u001b[39m\u001b[38;5;2m▉\u001b[39m\u001b[0m 3949.0                 \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 86.0\u001b[90m, \u001b[0m 89.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█████\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 3550.0                 \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 89.0\u001b[90m, \u001b[0m 92.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█████\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 3362.0                 \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 92.0\u001b[90m, \u001b[0m 95.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m████\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 3125.0                  \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 95.0\u001b[90m, \u001b[0m 98.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m████\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 2734.0                  \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 98.0\u001b[90m, \u001b[0m101.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m████\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 2657.0                  \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m101.0\u001b[90m, \u001b[0m104.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m███\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 2374.0                   \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m105.0\u001b[90m, \u001b[0m108.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m███\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 2274.0                   \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m108.0\u001b[90m, \u001b[0m111.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m███\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 2115.0                   \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m111.0\u001b[90m, \u001b[0m114.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 1912.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m114.0\u001b[90m, \u001b[0m117.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 1787.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m117.0\u001b[90m, \u001b[0m120.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 1722.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m120.0\u001b[90m, \u001b[0m123.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 1566.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m123.0\u001b[90m, \u001b[0m126.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 1466.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m127.0\u001b[90m, \u001b[0m130.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 1357.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m130.0\u001b[90m, \u001b[0m133.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 1381.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m133.0\u001b[90m, \u001b[0m136.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 1248.0                     \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m136.0\u001b[90m, \u001b[0m139.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 1149.0                     \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m139.0\u001b[90m, \u001b[0m142.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 1087.0                     \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m142.0\u001b[90m, \u001b[0m145.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 966.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m145.0\u001b[90m, \u001b[0m148.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 931.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m149.0\u001b[90m, \u001b[0m152.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 852.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m152.0\u001b[90m, \u001b[0m155.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 814.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m155.0\u001b[90m, \u001b[0m158.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 745.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m158.0\u001b[90m, \u001b[0m161.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▉\u001b[39m\u001b[0m 651.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m161.0\u001b[90m, \u001b[0m164.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 592.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m164.0\u001b[90m, \u001b[0m167.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 537.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m167.0\u001b[90m, \u001b[0m170.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 542.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m171.0\u001b[90m, \u001b[0m174.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 501.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m174.0\u001b[90m, \u001b[0m177.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 445.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m177.0\u001b[90m, \u001b[0m180.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m████████\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 5319.0              \u001b[90m \u001b[39m \n",
       "                  \u001b[90m└                              ┘\u001b[39m \n",
       "edges: 70.0:3.142857142857143:180.0\n",
       "bin counts: [13923.0, 4302.0, 4300.0, 4252.0, 3949.0, 3550.0, 3362.0, 3125.0, 2734.0, 2657.0  …  852.0, 814.0, 745.0, 651.0, 592.0, 537.0, 542.0, 501.0, 445.0, 5319.0]\n",
       "total count: 80487.0"
      ]
     },
     "execution_count": 13,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "let H = Hist1D(Float64; bins = range(70, 180; length=36), overflow=true)\n",
    "    #@time @batch for evt in mytree\n",
    "    @time for evt in mytree\n",
    "        evt.nMuon != 4 && continue\n",
    "        \n",
    "        pts, etas  = evt.Muon_pt, evt.Muon_eta\n",
    "        \n",
    "        (all(pts .> 5) && all(abs.(etas) .< 2.4)) || continue\n",
    "        \n",
    "        phis, masses, charges = evt.Muon_phi, evt.Muon_mass, evt.Muon_charge\n",
    "        \n",
    "        sum(charges) != 0 && continue\n",
    "        \n",
    "        Z_idx = reco_zz_to_4l(pts, etas, phis, masses, charges)\n",
    "        \n",
    "        filter_z_dr(Z_idx, etas, phis) || continue\n",
    "\n",
    "        Z_mass = compute_z_masses_4l(Z_idx, pts, etas, phis, masses)\n",
    "\n",
    "        filter_z_candidates(Z_mass) || continue\n",
    "        h_mass = compute_higgs_mass_4l(Z_idx, pts, etas, phis, masses)\n",
    "        push!(H, h_mass)\n",
    "    end\n",
    "    H\n",
    "end"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "4"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "Threads.nthreads()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "  6.702241 seconds (21.68 M allocations: 25.134 GiB, 5.03% gc time, 0.35% compilation time)\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div style=\"display: flex;\">\n",
       "    <div style=\"float:left; margin:5px\"><svg width=\"250\" height=\"200\" version=\"1.1\" xmlns=\"http://www.w3.org/2000/svg\">\n",
       "    <polyline points=\"12.5,180.0,12.5,19.999999999999996,18.92857142857143,19.999999999999996,18.92857142857143,130.56237879767292,25.357142857142858,130.56237879767292,25.357142857142858,130.58536235006824,31.78571428571429,130.58536235006824,31.78571428571429,131.13696760755585,38.214285714285715,131.13696760755585,38.214285714285715,134.6189757954464,44.642857142857146,134.6189757954464,44.642857142857146,139.20419449831215,51.07142857142858,139.20419449831215,51.07142857142858,141.36464842347195,57.50000000000001,141.36464842347195,57.50000000000001,144.08819938231701,63.92857142857143,144.08819938231701,63.92857142857143,148.58148387560152,70.35714285714286,148.58148387560152,70.35714285714286,149.46635064282123,76.78571428571429,149.46635064282123,76.78571428571429,152.71852330675858,83.21428571428572,152.71852330675858,83.21428571428572,153.86770092652446,89.64285714285715,153.86770092652446,89.64285714285715,155.69489334195217,96.07142857142858,155.69489334195217,96.07142857142858,158.02772391007684,102.50000000000001,158.02772391007684,102.50000000000001,159.4641959347842,108.92857142857143,159.4641959347842,108.92857142857143,160.211161387632,115.35714285714286,160.211161387632,115.35714285714286,162.00387847446672,121.78571428571429,162.00387847446672,121.78571428571429,163.15305609423257,128.21428571428572,163.15305609423257,128.21428571428572,164.40565969977735,134.64285714285717,164.40565969977735,134.64285714285717,164.12985707103354,141.07142857142858,164.12985707103354,141.07142857142858,165.65826330532215,147.5,165.65826330532215,147.5,166.79594914889034,153.92857142857144,166.79594914889034,153.92857142857144,167.50843927314517,160.35714285714286,167.50843927314517,160.35714285714286,168.89894419306185,166.7857142857143,168.89894419306185,166.7857142857143,169.3011563599799,173.21428571428572,169.3011563599799,173.21428571428572,170.20900667959492,179.64285714285717,170.20900667959492,179.64285714285717,170.64569417510594,186.07142857142858,170.64569417510594,186.07142857142858,171.43862673274438,192.50000000000003,171.43862673274438,192.50000000000003,172.51885369532428,198.92857142857144,172.51885369532428,198.92857142857144,173.19686849098613,205.35714285714286,173.19686849098613,205.35714285714286,173.82891618185735,211.7857142857143,173.82891618185735,211.7857142857143,173.77145730086906,218.21428571428572,173.77145730086906,218.21428571428572,174.24262012497306,224.64285714285717,174.24262012497306,224.64285714285717,174.88615959204193,231.07142857142858,174.88615959204193,231.07142857142858,118.87524240465417,237.50000000000003,118.87524240465417,237.5,180.0,12.5,180.0\" stroke=\"black\" fill=\"#ffffff00\" stroke-width=\"1\"/>\n",
       "    <polyline points=\"12.5,180.0,237.5,180.0\" stroke=\"black\" stroke-width=\"1\"/>\n",
       "    <text x=\"12.5\" y=\"190.0\" dominant-baseline=\"middle\" text-anchor=\"start\" fill=\"black\">70.0</text>\n",
       "    <text x=\"237.5\" y=\"190.0\" dominant-baseline=\"middle\" text-anchor=\"end\" fill=\"black\">180.0</text>\n",
       "</svg>\n",
       "</div>\n",
       "    <div style=\"float:left; margin:5px; max-width: 50%; display:flex; justify-content:center; align-items:center;\">\n",
       "        <ul>\n",
       "            <li>edges: 70.0:3.142857142857143:180.0</li>\n",
       "            <li>bin counts: [13923.0, 4302.0, 4300.0, 4252.0, 3949.0, 3550.0, 3362.0, 3125.0, 2734.0, 2657.0  …  852.0, 814.0, 745.0, 651.0, 592.0, 537.0, 542.0, 501.0, 445.0, 5319.0]</li>\n",
       "            <li>maximum count: 13923.0</li>\n",
       "            <li>total count: 80487.0</li>\n",
       "        </ul>\n",
       "    </div>\n",
       "</div>\n",
       "\n"
      ],
      "text/plain": [
       "                  \u001b[90m┌                              ┐\u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 70.0\u001b[90m, \u001b[0m 73.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█████████████████████\u001b[39m\u001b[38;5;2m \u001b[39m\u001b[0m 13923.0\u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 73.0\u001b[90m, \u001b[0m 76.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██████\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 4302.0                \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 76.0\u001b[90m, \u001b[0m 79.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██████\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 4300.0                \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 79.0\u001b[90m, \u001b[0m 82.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██████\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 4252.0                \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 83.0\u001b[90m, \u001b[0m 86.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█████\u001b[39m\u001b[38;5;2m▉\u001b[39m\u001b[0m 3949.0                 \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 86.0\u001b[90m, \u001b[0m 89.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█████\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 3550.0                 \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 89.0\u001b[90m, \u001b[0m 92.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█████\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 3362.0                 \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 92.0\u001b[90m, \u001b[0m 95.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m████\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 3125.0                  \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 95.0\u001b[90m, \u001b[0m 98.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m████\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 2734.0                  \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m 98.0\u001b[90m, \u001b[0m101.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m████\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 2657.0                  \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m101.0\u001b[90m, \u001b[0m104.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m███\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 2374.0                   \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m105.0\u001b[90m, \u001b[0m108.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m███\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 2274.0                   \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m108.0\u001b[90m, \u001b[0m111.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m███\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 2115.0                   \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m111.0\u001b[90m, \u001b[0m114.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 1912.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m114.0\u001b[90m, \u001b[0m117.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 1787.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m117.0\u001b[90m, \u001b[0m120.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 1722.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m120.0\u001b[90m, \u001b[0m123.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 1566.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m123.0\u001b[90m, \u001b[0m126.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 1466.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m127.0\u001b[90m, \u001b[0m130.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 1357.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m130.0\u001b[90m, \u001b[0m133.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m██\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 1381.0                    \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m133.0\u001b[90m, \u001b[0m136.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 1248.0                     \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m136.0\u001b[90m, \u001b[0m139.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 1149.0                     \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m139.0\u001b[90m, \u001b[0m142.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 1087.0                     \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m142.0\u001b[90m, \u001b[0m145.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▌\u001b[39m\u001b[0m 966.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m145.0\u001b[90m, \u001b[0m148.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 931.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m149.0\u001b[90m, \u001b[0m152.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▍\u001b[39m\u001b[0m 852.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m152.0\u001b[90m, \u001b[0m155.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 814.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m155.0\u001b[90m, \u001b[0m158.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m█\u001b[39m\u001b[38;5;2m▎\u001b[39m\u001b[0m 745.0                      \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m158.0\u001b[90m, \u001b[0m161.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▉\u001b[39m\u001b[0m 651.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m161.0\u001b[90m, \u001b[0m164.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 592.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m164.0\u001b[90m, \u001b[0m167.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 537.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m167.0\u001b[90m, \u001b[0m170.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 542.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m171.0\u001b[90m, \u001b[0m174.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▊\u001b[39m\u001b[0m 501.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m174.0\u001b[90m, \u001b[0m177.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m▋\u001b[39m\u001b[0m 445.0                       \u001b[90m \u001b[39m \n",
       "   \u001b[0m\u001b[90m[\u001b[0m177.0\u001b[90m, \u001b[0m180.0\u001b[90m)\u001b[0m \u001b[90m┤\u001b[39m\u001b[38;5;2m████████\u001b[39m\u001b[38;5;2m▏\u001b[39m\u001b[0m 5319.0              \u001b[90m \u001b[39m \n",
       "                  \u001b[90m└                              ┘\u001b[39m \n",
       "edges: 70.0:3.142857142857143:180.0\n",
       "bin counts: [13923.0, 4302.0, 4300.0, 4252.0, 3949.0, 3550.0, 3362.0, 3125.0, 2734.0, 2657.0  …  852.0, 814.0, 745.0, 651.0, 592.0, 537.0, 542.0, 501.0, 445.0, 5319.0]\n",
       "total count: 80487.0"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "let H = Hist1D(Float64; bins = range(70, 180; length=36), overflow=true)\n",
    "    @time @batch for evt in mytree\n",
    "        evt.nMuon != 4 && continue\n",
    "        \n",
    "        pts, etas  = evt.Muon_pt, evt.Muon_eta\n",
    "        \n",
    "        (all(pts .> 5) && all(abs.(etas) .< 2.4)) || continue\n",
    "        \n",
    "        phis, masses, charges = evt.Muon_phi, evt.Muon_mass, evt.Muon_charge\n",
    "        \n",
    "        sum(charges) != 0 && continue\n",
    "        \n",
    "        Z_idx = reco_zz_to_4l(pts, etas, phis, masses, charges)\n",
    "        \n",
    "        filter_z_dr(Z_idx, etas, phis) || continue\n",
    "\n",
    "        Z_mass = compute_z_masses_4l(Z_idx, pts, etas, phis, masses)\n",
    "\n",
    "        filter_z_candidates(Z_mass) || continue\n",
    "\n",
    "        h_mass = compute_higgs_mass_4l(Z_idx, pts, etas, phis, masses)\n",
    "        push!(H, h_mass)\n",
    "    end\n",
    "    H\n",
    "end"
   ]
  }
 ],
 "metadata": {
  "@webio": {
   "lastCommId": null,
   "lastKernelId": null
  },
  "kernelspec": {
   "display_name": "Julia 1.7.0-beta4",
   "language": "julia",
   "name": "julia-1.7"
  },
  "language_info": {
   "file_extension": ".jl",
   "mimetype": "application/julia",
   "name": "julia",
   "version": "1.7.0"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}