Projektpraktikum/notebooks/thesis_electrons.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import uproot\n",
    "import numpy as np\n",
    "import sys\n",
    "import os\n",
    "import matplotlib\n",
    "import matplotlib.pyplot as plt\n",
    "import mplhep\n",
    "from mpl_toolkits import mplot3d\n",
    "import itertools\n",
    "import awkward as ak\n",
    "from scipy.optimize import curve_fit\n",
    "from utils.components import unique_name_ext_re\n",
    "mplhep.style.use([\"LHCbTex2\"])\n",
    "plt.rcParams[\"savefig.dpi\"] = 600\n",
    "%matplotlib inline"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "50501"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "file = uproot.open(\n",
    "    \"/work/cetin/Projektpraktikum/tracking_losses_ntuple_B_rad_length_endVelo2endUT.root:PrDebugTrackingLosses.PrDebugTrackingTool/Tuple;1\"\n",
    ")\n",
    "\n",
    "# selektiere nur elektronen von B->K*ee\n",
    "allcolumns = file.arrays()\n",
    "electrons = allcolumns[(allcolumns.isElectron) & (allcolumns.fromB)]\n",
    "\n",
    "ak.num(electrons, axis=0)\n",
    "# ak.count(found, axis=None)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "50501 * {\n",
      "    all_endvtx_types_length: int32,\n",
      "    all_endvtx_types: var * float32,\n",
      "    all_endvtx_x_length: int32,\n",
      "    all_endvtx_x: var * float32,\n",
      "    all_endvtx_y_length: int32,\n",
      "    all_endvtx_y: var * float32,\n",
      "    all_endvtx_z_length: int32,\n",
      "    all_endvtx_z: var * float32,\n",
      "    brem_photons_pe_length: int32,\n",
      "    brem_photons_pe: var * float32,\n",
      "    brem_photons_px_length: int32,\n",
      "    brem_photons_px: var * float32,\n",
      "    brem_photons_py_length: int32,\n",
      "    brem_photons_py: var * float32,\n",
      "    brem_photons_pz_length: int32,\n",
      "    brem_photons_pz: var * float32,\n",
      "    brem_vtx_x_length: int32,\n",
      "    brem_vtx_x: var * float32,\n",
      "    brem_vtx_y_length: int32,\n",
      "    brem_vtx_y: var * float32,\n",
      "    brem_vtx_z_length: int32,\n",
      "    brem_vtx_z: var * float32,\n",
      "    endvtx_type: int32,\n",
      "    endvtx_x: float64,\n",
      "    endvtx_y: float64,\n",
      "    endvtx_z: float64,\n",
      "    energy: float64,\n",
      "    eta: float64,\n",
      "    event_count: int32,\n",
      "    fromB: bool,\n",
      "    fromD: bool,\n",
      "    fromDecay: bool,\n",
      "    fromHadInt: bool,\n",
      "    fromPV: bool,\n",
      "    fromPairProd: bool,\n",
      "    fromSignal: bool,\n",
      "    fromStrange: bool,\n",
      "    ideal_state_770_qop: float64,\n",
      "    ideal_state_770_tx: float64,\n",
      "    ideal_state_770_ty: float64,\n",
      "    ideal_state_770_x: float64,\n",
      "    ideal_state_770_y: float64,\n",
      "    ideal_state_770_z: float64,\n",
      "    ideal_state_9410_qop: float64,\n",
      "    ideal_state_9410_tx: float64,\n",
      "    ideal_state_9410_ty: float64,\n",
      "    ideal_state_9410_x: float64,\n",
      "    ideal_state_9410_y: float64,\n",
      "    ideal_state_9410_z: float64,\n",
      "    isElectron: bool,\n",
      "    isKaon: bool,\n",
      "    isMuon: bool,\n",
      "    isPion: bool,\n",
      "    isProton: bool,\n",
      "    lost: bool,\n",
      "    lost_in_track_fit: bool,\n",
      "    match_fraction: float32,\n",
      "    mc_chi2: float32,\n",
      "    mc_dSlope: float32,\n",
      "    mc_dSlopeY: float32,\n",
      "    mc_distX: float32,\n",
      "    mc_distY: float32,\n",
      "    mc_teta2: float32,\n",
      "    mc_zMag: float32,\n",
      "    mcp_idx: int32,\n",
      "    mother_id: int32,\n",
      "    mother_key: int32,\n",
      "    originvtx_type: int32,\n",
      "    originvtx_x: float64,\n",
      "    originvtx_y: float64,\n",
      "    originvtx_z: float64,\n",
      "    p: float64,\n",
      "    p_end_scifi: float64,\n",
      "    p_end_ut: float64,\n",
      "    p_end_velo: float64,\n",
      "    phi: float64,\n",
      "    pid: int32,\n",
      "    pt: float64,\n",
      "    px: float64,\n",
      "    py: float64,\n",
      "    pz: float64,\n",
      "    quality: int32,\n",
      "    rad_length_frac: float64,\n",
      "    scifi_hit_pos_x_length: int32,\n",
      "    scifi_hit_pos_x: var * float32,\n",
      "    scifi_hit_pos_y_length: int32,\n",
      "    scifi_hit_pos_y: var * float32,\n",
      "    scifi_hit_pos_z_length: int32,\n",
      "    scifi_hit_pos_z: var * float32,\n",
      "    track_p: float64,\n",
      "    track_pt: float64,\n",
      "    tx: float64,\n",
      "    ty: float64,\n",
      "    ut_hit_pos_x_length: int32,\n",
      "    ut_hit_pos_x: var * float32,\n",
      "    ut_hit_pos_y_length: int32,\n",
      "    ut_hit_pos_y: var * float32,\n",
      "    ut_hit_pos_z_length: int32,\n",
      "    ut_hit_pos_z: var * float32,\n",
      "    velo_hit_pos_x_length: int32,\n",
      "    velo_hit_pos_x: var * float32,\n",
      "    velo_hit_pos_y_length: int32,\n",
      "    velo_hit_pos_y: var * float32,\n",
      "    velo_hit_pos_z_length: int32,\n",
      "    velo_hit_pos_z: var * float32,\n",
      "    velo_track_idx: int32,\n",
      "    velo_track_tx: float64,\n",
      "    velo_track_ty: float64,\n",
      "    velo_track_x: float64,\n",
      "    velo_track_y: float64,\n",
      "    velo_track_z: float64\n",
      "}\n"
     ]
    }
   ],
   "source": [
    "electrons.type.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "cut_prop: bool = electrons.p_end_velo > 3e3\n",
    "found = electrons[~electrons.lost & cut_prop]\n",
    "lost = electrons[electrons.lost & cut_prop]\n",
    "\n",
    "eloss_found = (found[\"p\"] - found[\"p_end_scifi\"]) / found[\"p\"]\n",
    "eloss_lost = (lost[\"p\"] - lost[\"p_end_scifi\"]) / lost[\"p\"]\n",
    "\n",
    "eloss = (electrons[\"p\"] - electrons[\"p_end_scifi\"]) / electrons[\"p\"]\n",
    "eloss_magnet_found = ak.to_numpy(\n",
    "    (found[\"p_end_velo\"] - found[\"p_end_scifi\"]) / found[\"p_end_velo\"]\n",
    ")\n",
    "eloss_magnet_lost = ak.to_numpy(\n",
    "    (lost[\"p_end_velo\"] - lost[\"p_end_scifi\"]) / lost[\"p_end_velo\"]\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [],
   "source": [
    "# eloss_upstream_found = (found[\"p\"] - found[\"p_end_ut\"]) / found[\"p\"]\n",
    "# eloss_upstream_lost = (lost[\"p\"] - lost[\"p_end_ut\"]) / lost[\"p\"]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAABMIAAAOWCAYAAAANzz7PAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjguMCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy81sbWrAAAACXBIWXMAAA9hAAAPYQGoP6dpAABpjklEQVR4nO39z29jaZof+D6KSnsKsK2iooACBGnRQbV3BtxNRfZmNlMZlP8BixHLXoXYy4HQFjP9B3QW2d0C7l0YJqM3tXMkZXs1gG+TkTmzmAHGGaK7AQN3YetULUKXQAMZYsluuMZ2i3eRJovUT0qieEidzwcQIEqH73kOySOJXz3ve5b6/X4/AAAAAOCRe5J2AQAAAAAwC4IwAAAAADJBEAYAAABAJgjCAAAAAMgEQRgAAAAAmSAIAwAAACATBGEAAAAAZIIgDAAAAIBMEIQBAAAAkAmCMAAAAAAyQRAGAAAAQCYIwgAAAADIBEEYAAAAAJkgCAMAAAAgEwRhAAAAAGSCIAwAAACATBCEAQAAAJAJgjAAAAAAMkEQBgAAAEAmCMIAAAAAyARBGAAAAACZIAgDAAAAIBMEYQAAAABkgiAMAAAAgEwQhAEAAACQCYIwAAAAADJBEAYAAABAJgjCAAAAAMiET9IugNv7O3/n78Svf/3r+MEPfhA/+clP0i4HAAAAYGJ/9Vd/FX/zN38TP/zhD+Ov//qvZ7rvpX6/35/pHrm3H/zgB3F2dpZ2GQAAAAB39uTJk/ibv/mbme5TR9gCGgRhT548idXV1bTLubVut7vQ9Uc4hnnhGOaDY5gPjmE+OIb54Bjmx2M4DscwHxzDfHAM8+ExHcMPfvCDme9bELaAfvKTn8Tx8XGsrq7Ghw8f0i7n1tbX1xe6/gjHMC8cw3xwDPPBMcwHxzAfHMP8eAzH4Rjmg2OYD45hPjymY0hjuSeL5QMAAACQCY86CPvVr34Vr169ih/84Afxe7/3e3F6epp2SQAAAACk5FEHYZVKJZrNZvT7/Tg8PIyXL1+mXRIAAAAAKXnUa4S12+1YWlqKiIh+vx+tVivligAAAABIy6PuCHvx4kX0+/3h7WKxmGI1AAAAAKTpUXeE1ev1iIh49+5dFAqFePPmTcoVAQAAAJCWRx2ERfwmDAMAAAAg2x59EAYPYXd3N05PT2N5eTntUu7MMTAtj+F5cAxMy2N4HhwD0/QYngvHwLQ8hufBMfAYLPVHF9FiIayvr8fx8XGsra3Fhw8f0i7n1ha9fpgm5wP8hvMBfsP5AOOcE/Abj+F8SPMYHvVi+QO//OUv0y4BAAAAgJQ9+iDsX/7LfxkbGxvxF3/xF2mXAgAAAECKHn0Q9u2330a/34+vvvoq7VIAAAAASNFcL5b/Z3/2Z3e+b6/Xi6Ojo2g0GrG0tBStViv+6I/+aIrVAQAAALBI5joI29nZiaWlpXuNMbgWQKfTiV/+8pfxW7/1W1OoDAAAAIBFsxBTI+9yYct+vx/9fj+WlpaGYVqj0Zh2aQAAAAAsiIUIwiJ+E2xN+hERsbS0NPa1ZrOZ8lEAAAAAkJa5nhoZEfHs2bOoVCqRz+dvdb+jo6Oo1WqxubkZb968ieXl5QeqkNva3d2N09NTzwmE8wFGOR/gN5wPMM45Ab/hfLifpf5d5h3OyJMnTyJJknut6/Xbv/3b8fz58/gX/+JfTK+wlK2vr8fx8XGsra3Fhw8f0i4HAAAAYGJp5hpzPTUyl8vde3H7f/7P/3l89dVX8ad/+qfTKQoAAACAhTTXQVi1Wr33GMViMfL5fOzt7cUvf/nL+xcFAAAAwEKa6yDs9evXUxmnUChEv9+PWq02lfEAAAAAWDxzHYRN27fffpt2CQAAAACk5NEHYb/61a+i3W5HRESn00m5GgAAAADS8knaBTykv/iLv4hSqRS/+tWvIiIin8+nXNF0dbvdWF9fv3ab3d3d2N3dnVFFAAAAQJbt7+/H/v7+tdt0u90ZVXPRXAdhf//v//073zdJkoiI6Pf7ERGxtLQUxWJxKnXNi7Ozszg+Pr52m9PT0xlVAwAAAGTd6enpjVlFmuY6CDs6OoqlpaVhmHUXo/efxlUo58mTJ09idXX12m2Wl5dnVA0AAACQdcvLy7G2tnbtNt1uN87OzmZU0bil/n1Spgf25MmTWFpauvP9B4eWy+Wi2WzGixcvplVaqtbX1+P4+DjW1tbiw4cPaZcDAAAAMLE0c4257giLGA+znj59OvH9crlc5PP5+PTTT+Of/JN/8lDlAQAAALAg5j4Iq9Vq8Yd/+IdplwEAAADAgnuSdgHXyefzQjAAAAAApmKug7BKpZJ2CQAAAAA8EnMdhL1+/TrtEgAAAAB4JOY6CAMAAACAaZn7xfJ5vH72zddTG+vzn342tbEAAACAx+lRd4T96le/ilevXsUPfvCD+L3f+704PT1NuyQAAAAAUvKog7BKpRLNZjP6/X4cHh7Gy5cv0y4JAAAAgJQ86qmR7XY7lpaWIiKi3+9Hq9VKuSIAAAAA0vKoO8JevHgR/X5/eLtYLKZYDQAAAABpetQdYfV6PSIi3r17F4VCId68eZNyRQAAAACk5VEHYRG/CcMAAAAAyLZHPTUSAAAAAAYEYQAAAABkwqMPwn71q1/FF198kXYZAAAAAKTs0QdhP/rRj6LVasXv/d7vpV0KAAAAACl69EFYRMTz58/j8PAw/uk//adplwIAAABAShbqqpG//OUvo9frxcePHye+T6fTiWazGf1+P+r1evzRH/3RA1YIAAAAwLya+yDsX/2rfxX1ej3a7fa9x+r1evcvCAAAAICFNNdB2B//8R/H559/HhER/X7/zuMsLS3F0tJSFAqFaZUGAAAAwIKZ2zXC/t2/+3dRqVQi4vIQbGlpaaJxlpaWot/vx7Nnz+LNmzdTrREAAACAxTG3HWH1ej0ivr/q487OTmxsbEQ+n4+IiJOTk3j58mWsrKzEV199den9e71e/Pmf/3n82Z/9WXQ6nfid3/mdWZUOAAAAwBya2yCs3W7HxsZG/If/8B8u/f7r16/jz/7sz2JjYyN+67d+69Jt/vE//scREfH555/Hv/k3/+ahSgUAAABgAczt1MgkSYZdYZf5/PPPo9/vR61Wu3acSqUSf/7nfx5/+qd/Ou0SAQAAyKiDg4NIkiTtMubWNC54Bw9hboOwiIjnz59f+b1nz57Fixcvol6vx+np6ZXb5fP5+N3f/d3Y29u7djsAAIBpaDQasbKyMrxo11UfGxsbdwoLOp1ObGxs3Dj+yspKNBqNC/ev1WoT19fr9SaqKUmSqNVqsbW1NVbbxsZGbG5uRrlcjoODg7HxarValEqla4/zpsaHh9DpdKJcLg+PY2VlZXgcpVIpDg4O4uDgIF6/fn3j89dut4f3f+zBUJIk0Wg0YmtrK5aWlmJrayvtksZk6bngekv9+1yO8QE9f/48Dg4Orpz2GPH9C/kf/aN/FH/wB38Q/+yf/bMrt/uDP/iDaDQa8fnnn8cf/dEfPUC1s7W+vh7Hx8extrYWHz58SLucO/vZN19PbazPf/rZ1MYCAIBp6HQ6USqVLnQN7ezsXDv7ZVK9Xi9ev34dBwcHY18vFArRbDaHayzftr69vb2oVqsT1dBut6NarY4FC/l8fvjx8ePHSJIkOp3O2Pcjvg9OisVitFqtsfGazWZ89dVX0ev1IpfLxcnJyUS13Nf5xzOfz0ehUBjWOnoMA9VqNfb29q4cc2VlZRj+5fP5ODo6mn7hKUuSZPg6Oh+czlPckIXnYpGkmWvMbUdYPp+/9L8Xo4rFYvzu7/5u1Ov1+Mu//Msrt3v//n1ERDSbzanWmLZutxvr6+vXfuzv76ddJgAAZFKhUIhKpXLh69MIwSIicrlcvHnz5sLXq9XqjSHYVfXlcrmJQ7ByuRxbW1vDEGx7ezuOjo7i6OgoWq1W1Ov1aDabcXh4GP1+P6rVauRyuUiS5NIpheVyOarVajQajYk70aZp0O0V8f1zdHR0FM1mc3gMR0dHUSwWx+5zU5jy9OnTB6t3XuTz+Tg8PIyTk5NrQ8G0ZeG5mBf7+/s3ZhXdbje1+uY
      "text/plain": [
       "<Figure size 1200x900 with 1 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "nbins = 50\n",
    "plt.hist(\n",
    "    ak.to_numpy(eloss_magnet_lost),\n",
    "    bins=nbins,\n",
    "    density=True,\n",
    "    alpha=0.5,\n",
    "    histtype=\"bar\",\n",
    "    color=\"#F05342\",\n",
    "    label=\"lost\",\n",
    "    range=[0.001, 1],\n",
    ")\n",
    "# #2A9D8F another teal color\n",
    "plt.hist(\n",
    "    ak.to_numpy(eloss_magnet_found),\n",
    "    bins=nbins,\n",
    "    density=True,\n",
    "    alpha=0.5,\n",
    "    histtype=\"bar\",\n",
    "    color=\"#107E7D\",\n",
    "    label=\"found\",\n",
    "    range=[0.001, 1],\n",
    ")\n",
    "\n",
    "plt.xlabel(r\"$E_\\gamma/E_{VELO}$\")\n",
    "plt.ylabel(\"a.u.\")\n",
    "# plt.title(r'$B\\rightarrow K^\\ast ee$, $p>5$GeV, photons w/ brem_vtx_z$<9500$mm')\n",
    "plt.legend(title=\"LHCb Simulation\")\n",
    "plt.show()\n",
    "# plt.savefig(\n",
    "#     \"/work/cetin/Projektpraktikum/thesis/emitted_energy_endVelo2endT.pdf\",\n",
    "#     format=\"PDF\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [],
   "source": [
    "# magnet kick position\n",
    "input_tree = uproot.open(\n",
    "    {\n",
    "        \"/work/cetin/Projektpraktikum/param_data_B_default.root\": \"PrParameterisationData_2ece6184.PrDebugTrackingTool/Tuple;1\"\n",
    "    }\n",
    ")\n",
    "array = input_tree.arrays()\n",
    "array = array[array.isElectron & (array.fromB)]\n",
    "array[\"dSlope_fringe\"] = array[\"tx_ref\"] - array[\"tx\"]\n",
    "array[\"z_mag_x_fringe\"] = (\n",
    "    array[\"x\"]\n",
    "    - array[\"x_ref\"]\n",
    "    - array[\"tx\"] * array[\"z\"]\n",
    "    + array[\"tx_ref\"] * array[\"z_ref\"]\n",
    ") / array[\"dSlope_fringe\"]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(exptext: Custom Text(0.05, 0.95, 'LHCb'),\n",
       " expsuffix: Custom Text(0.05, 0.955, 'Simulation'))"
      ]
     },
     "execution_count": 28,
     "metadata": {},
     "output_type": "execute_result"
    },
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAABMIAAAOWCAYAAAANzz7PAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjguMCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy81sbWrAAAACXBIWXMAAA9hAAAPYQGoP6dpAACfvElEQVR4nOz9T3Ab6b3mez4p143omY5DgNS5J0JBxriYKPXietFigvL+iEDZi9mcEkDW2i4CklcO2iJEbe9ESUC5OF6dEkCVY2ZWRwSl2roFULWbxSkiqVr4LqYKSdaMGIzoHoEQHN2nYzqKOQtewAT/gkASmRC+nwiEAfLN9/2BoBzBp37vm4bruq4AAAAAAACAd9wVvwsAAAAAAAAA+oEgDAAAAAAAAEOBIAwAAAAAAABDgSAMAAAAAAAAQ4EgDAAAAAAAAEOBIAwAAAAAAABDgSAMAAAAAAAAQ4EgDAAAAAAAAEOBIAwAAAAAAABDgSAMAAAAAAAAQ4EgDAAAAAAAAEOBIAwAAAAAAABDgSAMAAAAAAAAQ4EgDAAAAAAAAEOBIAwAAAAAAABDgSAMAAAAAAAAQ4EgDAAAAAAAAEOBIAwAAAAAAABDgSAMAAAAAAAAQ4EgDAAAAAAAAEOBIAwAAAAAAABDgSAMAAAAAAAAQ4EgDAAAAAAAAEOBIAwAAAAAAABDgSAMAAAAAAAAQ4EgDAAAAAAAAEOBIAwAAAAAAABDgSAMAAAAAAAAQ+E9vwsA/v2///f67//9v+snP/mJ/uEf/sHvcgAAAAAAwCX6z//5P+vHH3/Uv/t3/07/9b/+176ubbiu6/Z1ReCIn/zkJ9rf3/e7DAAAAAAA0EdXrlzRjz/+2Nc16QiD75pB2JUrV3Tt2rULXbu7u9v1tb1i7f6v7ff6rM3arM3a7+L6rM3arP3uru33+qzN2qzN2udd/5Of/OQSqjvbwAVh29vbsm1btVpN9Xpdb968kSRdvXpV4XBYpmlqenpaIyMjPleKTv3DP/yDdnZ2dO3aNb1+/fpC105MTHR9ba9Yu/9r+70+a7M2a7P2u7g+a7M2a7+7a/u9PmuzNmuz9nnX+3E80kAEYc+fP1epVNLq6qrq9XpH15imqVgspmQyqVu3bl1ugQAAAAAAAAi8QAdhT548UTableM4kqTDx5kZhnHiNc0x1WpVjuOoUCjINE3dv39fv/71ry+/aAAAAAAAAATSFb8LOMnLly91/fp1pdPpE0Ow5uuTHieNcRxHqVRK/+E//Ad9++23fXsfAAAAAAAACI7AdYQtLS0pl8sdC7VisZgsy9LVq1dlmqbC4bAkaWxsTJJUq9UkSfV6XY7j6M2bN3IcR+VyubWd8vvvv5dlWcpkMvr000/79p4AAAAAAADgv8AEYY1GQzMzM6pUKpIOgq94PK5YLKapqame5n779q2ePn2qQqEg27aVzWZVLpdVLpc5VD9Adnd3NTExceaYhYUFLSws9KkiAAAAAABwEcvLy1peXj5zzO7ubp+qOS4QQdirV680MzOjvb09pVIpZTIZTU5OejZ/KBRSKpVSKpWS4zhKp9NaX1/X9PS0KpWK/u7v/s6ztdC9/f197ezsnDmm0Wj0qRoAAAAAAHBRjUbj3L/t/eR7ELa5ualoNCrLsrSxseFpAHYS0zRVKpVULpeVTCYVjUa1sbFBZ1gAXLlyRdeuXTtzDJ8TAAAAAADBNTIyovHx8TPH7O7uan9/v08VtfM1CNva2tLMzIwSiYRWV1f7unYsFtPW1pai0ahmZ2f15z//ua/r47hr167p9evXF7pmYWFBjUZj6AIyP9+33z/zYX7vfhnWn/mwru2nYf6ZD/N798uw/syHdW0/+f2+h/Uz9/vn7pdh/ZkP69p+Out9d3Kk0cTEhG9dY4Z79FT6Pvrggw8Ui8X0+PFjv0pQvV7X9PS07t69q9/97ne+1THMmv8AxsfHLxyE+WlQ68Zg4vcN/cLvGvqJ3zf0C79r6Cd+39Avg/y75mftvnWE3b9/X6Zp+hqCSVI4HNbGxoZM09TMzIxu3Ljhaz0AAAAAAAC4HFf8Wvjq1asqFot+Ld8mHA5rfX1djuP4XQoAAAAAAAAuiW8dYffu3fNr6RNNTU1pamrK7zIAAAAAAABwSXzrCOu3u3fv+l0CAAAAAAAAfDQ0QVihUND29rbfZQAAAAAAAMAnQxGEra+vy3VdPXv2zO9SAAAAAAAA4BNfzgi7c+eOtra2Ln2dWq2mer3eOgT/8ePH+t3vfnfp6wIAAAAAACB4fAnCwuGwSqWSDMO41HVc12177TiOXr16pRs3blzquhgOCwsLajQaGhkZ8bsUDAF+39Av/K6hn/h9Q7/wu4Z+4vcN/cLvWncM92ha1Aebm5uKRqMdB2Gu6/YcmjXnyGQy+vTTT3uaC96amJjQzs6OxsfH9fr1a7/LAQAAAAAAl8jPHMCXM8KmpqZkmqakg4DqvIdhGB2NO+3R5LquisWiH28ZAAAAAAAAPvPtsPyZmRlJUrlc1v7+/qmPbDYr13VlWZZKpZL29vbOHH/0sbGxocnJSaXTae3v7+u7777z6y0DAAAAAADAR76cESZJs7Ozsm1bt27dOnXM1taW7t+/r3Q6rS+++KKrdSzL0osXL/TBBx/IMAz98z//c7clAwAAAAAAYID52hE2Ojp65phsNivTNLsOwZpM09T8/Lzy+by+/vrrnuYCAAAAAADAYPItCJOkFy9enPn9jY0NpdNpT9aanp6W67rK5/OezAcAAAAAAIDB4msQdh7btmVZlidzjY2NSZJKpZIn8wEAAAAAAGCwBDoIC4fD2tra8mSuZgBWr9c9mQ8AAAAAAACDJdBB2PT0tB4/ftzzPG/fvlWhUJB0EK4BAAAAAABg+AQ6CEulUqpUKvr888+7nqPRaCgWi0mSDMOQaZpelQcAAAAAAIABEuggLJFIaHJyUouLi/r444+1vb19oeufP3+uyclJ2bbd+trc3JzHVQIAAAAAAGAQGK7run4XcRbHcfTBBx/IMAxJkmVZSqfTMk1Tpmnq/ffflyRtb2+rXq/LcRyVSiWtrq6qXq/LdV0ZhiHXdTU6Oqo3b974+G5wkomJCe3s7Gh8fFyvX7/2uxwAAAAAAHCJ/MwB3uvral0wTVMbGxuanp6WdHAnyXQ6fe51zXyvGaAZhqFisXh5hQIAAAAAACDQAr01ssmyLH3//feampqS67odPQzDaHWCua6r1dVV3bp1y++3AgAAAAAAAJ8MRBAmHXSGVSoVra6uyrKsY99vBl9NzQAskUhob29Pt2/f7me5AAAAAAAACJjAb408KpFIKJFIaGtrS+VyWZVKRY7jqFarSZLGxsZkmqbi8bhisZhCoZDPFQMAAAAAACAIAn9YPt59HJYPP9XrdYXDYb/LAC7Vz/70R0/m+cuvfuvJPAAAABhufuYAA7M1EgC8Ytu2MpmMRkdHNT8/73c5AAAAAIA+GbitkZ1YWlrSnTt39NOf/tTvUgBckG3bevr0qWzbbm17bnZtmaapsbExWZbV2v58VDKZlOM4qlQqra/V63WVy2U9ffpU5XJZ9Xq9j+8IAAAAABAUA9cR9vz5c929e1c3b97U8+fPTxwTjUY1NTWljz/+WD/88EOfK8SwKxQKrZs3XOQRjUY9Wb+btUdHR0+dL5lMdjVnuVzuuOZ6va5MJtP6OeRyOZXLZTmO0wqt6vW6bNtWuVxWLpdTPB5vjc9kMioUCkomk1pbWzs2//z8vB4+fKi1tTVCMAAAAAAYYgMThP3hD3/Q1atXlUwmVSgUZNt264D8oxKJhBzH0XfffSfTNPXVV1/1uVoMs1Qqpb29PVUqFSUSiTPHxmIxlUql1ngvuK6rarWqfD5/5tlXpmkqn8+rWq1qb2/v1HHFYlF7e3sqlUondmAdtri4qEqlor29vXPHNjW3KOZyubavJxIJFYvF1nzN91UqlZRKpVrvzbZt5XI5pdPpVgh2NOxqzuO6rkzT7KguAAAAAMC7J/BB2Nu3b3X9+nVlMpnWH8OdnO8fDodVqVR048YNJRIJffnll32oFjgQDodlWZaKxaJSqdS
      "text/plain": [
       "<Figure size 1200x900 with 1 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "fig = plt.figure()\n",
    "plt.hist(array[\"z_mag_x_fringe\"],\n",
    "         bins=50,\n",
    "         range=[5150, 5300],\n",
    "         color=\"#2A9D8F\",\n",
    "         density=True)\n",
    "plt.xlabel(r\"z$_{Mag}$ [mm]\")\n",
    "plt.ylabel(\"Number of Tracks (normalised)\")\n",
    "mplhep.lhcb.text(\"Simulation\")\n",
    "mplhep."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "50501"
      ]
     },
     "execution_count": 29,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ak.num(array, axis=0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "tuner",
   "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.10.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}