create_images.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import training as training\n",
    "import reader as read\n",
    "import procesador as process\n",
    "import images as im\n",
    "import random as random"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['Dyjets_json.json',\n",
       " 'json_readme.txt',\n",
       " 'TTjets_json.json',\n",
       " 'Wjets_json.json',\n",
       " 'ZZjets_json.json']"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "from os import listdir\n",
    "listdir('json_files')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import json\n",
    "\n",
    "cwd = os.getcwd()\n",
    "json_path = cwd + '/json_files'\n",
    "\n",
    "with open(json_path + '/Dyjets_json.json') as json_file:\n",
    "    DYjets = json.load(json_file)\n",
    "\n",
    "with open(json_path + '/TTjets_json.json') as json_file:\n",
    "    TTjets = json.load(json_file)\n",
    "\n",
    "with open(json_path + '/Wjets_json.json') as json_file:\n",
    "    Wjets = json.load(json_file)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['67242185', '67242186', '67242187', '67242188', '67242190']"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "list(DYjets['Event'].keys())[:5]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "DYjets Before Filter histograms created\n",
      "TTjets Before Filter histograms created\n",
      "Wjets Before Filter histograms created\n"
     ]
    }
   ],
   "source": [
    "im.generateAllhistograms(DYjets, DirName = 'DYjets Before Filter', origen = 'DY')\n",
    "im.generateAllhistograms(TTjets, DirName = 'TTjets Before Filter', origen = 'TTjets')\n",
    "im.generateAllhistograms(Wjets, DirName = 'Wjets Before Filter', origen = 'Wjets')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Filter the jets, muons and electrons\n",
    "\n",
    "process.leptonFilter(DYjets, 'muon', minPt = 20, maxPt = False, minEta = False, maxEta = False, tightIDFilter = True, tightIsoFilter = True)\n",
    "process.leptonFilter(DYjets, 'electron', minPt = 20, maxPt = False, minEta = False, maxEta = False, tightIDFilter = True, tightIsoFilter = True)\n",
    "process.jetFilter(DYjets, minpt =30, maxpt = False, maxAbsEta = 2.4, jetNumber=False, tightIDFilter = True)\n",
    "\n",
    "process.leptonFilter(TTjets, 'muon', minPt = 20, maxPt = False, minEta = False, maxEta = False, tightIDFilter = True, tightIsoFilter = True)\n",
    "process.leptonFilter(TTjets, 'electron', minPt = 20, maxPt = False, minEta = False, maxEta = False, tightIDFilter = True, tightIsoFilter = True)\n",
    "process.jetFilter(TTjets, minpt =30, maxpt = False, maxAbsEta = 2.4, jetNumber=False, tightIDFilter = True)\n",
    "\n",
    "process.leptonFilter(Wjets, 'muon', minPt = 20, maxPt = False, minEta = False, maxEta = False, tightIDFilter = True, tightIsoFilter = True)\n",
    "process.leptonFilter(Wjets, 'electron', minPt = 20, maxPt = False, minEta = False, maxEta = False, tightIDFilter = True, tightIsoFilter = True)\n",
    "process.jetFilter(Wjets, minpt =30, maxpt = False, maxAbsEta = 2.4, jetNumber=False, tightIDFilter = True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "DY After Lepton and Jet Filter histograms created\n",
      "TTjets After Lepton and Jet Filter histograms created\n",
      "Wjets After Lepton and Jet Filter histograms created\n"
     ]
    }
   ],
   "source": [
    "im.generateAllhistograms(DYjets, DirName = 'DY After Lepton and Jet Filter', origen = 'DY')\n",
    "im.generateAllhistograms(TTjets, DirName = 'TTjets After Lepton and Jet Filter', origen = 'TTjets')\n",
    "im.generateAllhistograms(Wjets, DirName = 'Wjets After Lepton and Jet Filter', origen = 'Wjets')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "process.leptonNumberFilter(DYjets, electronNumber = False, muonNumber = False, leptonNumber = 1)\n",
    "process.leptonNumberFilter(TTjets, electronNumber = False, muonNumber = False, leptonNumber = 1)\n",
    "process.leptonNumberFilter(Wjets, electronNumber = False, muonNumber = False, leptonNumber = 1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "DY After Event Filter histograms created\n",
      "TTjets After Event Filter histograms created\n",
      "Wjets After Event Filter histograms created\n"
     ]
    }
   ],
   "source": [
    "im.generateAllhistograms(DYjets, DirName = 'DY After Event Filter', origen = 'DY')\n",
    "im.generateAllhistograms(TTjets, DirName = 'TTjets After Event Filter', origen = 'TTjets')\n",
    "im.generateAllhistograms(Wjets, DirName = 'Wjets After Event Filter', origen = 'Wjets')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "DY_Class = training.selectClass_KnownClass(DYjets, ClassName = 'DYjets', ClassNumber = '0')\n",
    "TT_Class = training.selectClass_KnownClass(TTjets, ClassName = 'TTjets', ClassNumber = '1')\n",
    "W_Class = training.selectClass_KnownClass(Wjets, ClassName = 'Wjets', ClassNumber = '2')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "0 out of 1370 completed\n",
      "13 out of 1370 completed\n",
      "27 out of 1370 completed\n",
      "41 out of 1370 completed\n",
      "54 out of 1370 completed\n",
      "68 out of 1370 completed\n",
      "82 out of 1370 completed\n",
      "95 out of 1370 completed\n",
      "109 out of 1370 completed\n",
      "123 out of 1370 completed\n",
      "137 out of 1370 completed\n",
      "150 out of 1370 completed\n",
      "164 out of 1370 completed\n",
      "178 out of 1370 completed\n",
      "191 out of 1370 completed\n",
      "205 out of 1370 completed\n",
      "219 out of 1370 completed\n",
      "232 out of 1370 completed\n",
      "246 out of 1370 completed\n",
      "260 out of 1370 completed\n",
      "274 out of 1370 completed\n",
      "287 out of 1370 completed\n",
      "301 out of 1370 completed\n",
      "315 out of 1370 completed\n",
      "328 out of 1370 completed\n",
      "342 out of 1370 completed\n",
      "356 out of 1370 completed\n",
      "369 out of 1370 completed\n",
      "383 out of 1370 completed\n",
      "397 out of 1370 completed\n",
      "411 out of 1370 completed\n",
      "424 out of 1370 completed\n",
      "438 out of 1370 completed\n",
      "452 out of 1370 completed\n",
      "465 out of 1370 completed\n",
      "479 out of 1370 completed\n",
      "493 out of 1370 completed\n",
      "506 out of 1370 completed\n",
      "520 out of 1370 completed\n",
      "534 out of 1370 completed\n",
      "548 out of 1370 completed\n",
      "561 out of 1370 completed\n",
      "575 out of 1370 completed\n",
      "589 out of 1370 completed\n",
      "602 out of 1370 completed\n",
      "616 out of 1370 completed\n",
      "630 out of 1370 completed\n",
      "643 out of 1370 completed\n",
      "657 out of 1370 completed\n",
      "671 out of 1370 completed\n",
      "685 out of 1370 completed\n",
      "698 out of 1370 completed\n",
      "712 out of 1370 completed\n",
      "726 out of 1370 completed\n",
      "739 out of 1370 completed\n",
      "753 out of 1370 completed\n",
      "767 out of 1370 completed\n",
      "780 out of 1370 completed\n",
      "794 out of 1370 completed\n",
      "808 out of 1370 completed\n",
      "822 out of 1370 completed\n",
      "835 out of 1370 completed\n",
      "849 out of 1370 completed\n",
      "863 out of 1370 completed\n",
      "876 out of 1370 completed\n",
      "890 out of 1370 completed\n",
      "904 out of 1370 completed\n",
      "917 out of 1370 completed\n",
      "931 out of 1370 completed\n",
      "945 out of 1370 completed\n",
      "959 out of 1370 completed\n",
      "972 out of 1370 completed\n",
      "986 out of 1370 completed\n",
      "1000 out of 1370 completed\n",
      "1013 out of 1370 completed\n",
      "1027 out of 1370 completed\n",
      "1041 out of 1370 completed\n",
      "1054 out of 1370 completed\n",
      "1068 out of 1370 completed\n",
      "1082 out of 1370 completed\n",
      "1096 out of 1370 completed\n",
      "1109 out of 1370 completed\n",
      "1123 out of 1370 completed\n",
      "1137 out of 1370 completed\n",
      "1150 out of 1370 completed\n",
      "1164 out of 1370 completed\n",
      "1178 out of 1370 completed\n",
      "1191 out of 1370 completed\n",
      "1205 out of 1370 completed\n",
      "1219 out of 1370 completed\n",
      "1233 out of 1370 completed\n",
      "1246 out of 1370 completed\n",
      "1260 out of 1370 completed\n",
      "1274 out of 1370 completed\n",
      "1287 out of 1370 completed\n",
      "1301 out of 1370 completed\n",
      "1315 out of 1370 completed\n",
      "1328 out of 1370 completed\n",
      "1342 out of 1370 completed\n",
      "1356 out of 1370 completed\n",
      "All DYjets events have been created\n",
      "0 out of 1239 completed\n",
      "12 out of 1239 completed\n",
      "24 out of 1239 completed\n",
      "37 out of 1239 completed\n",
      "49 out of 1239 completed\n",
      "61 out of 1239 completed\n",
      "74 out of 1239 completed\n",
      "86 out of 1239 completed\n",
      "99 out of 1239 completed\n",
      "111 out of 1239 completed\n",
      "123 out of 1239 completed\n",
      "136 out of 1239 completed\n",
      "148 out of 1239 completed\n",
      "161 out of 1239 completed\n",
      "173 out of 1239 completed\n",
      "185 out of 1239 completed\n",
      "198 out of 1239 completed\n",
      "210 out of 1239 completed\n",
      "223 out of 1239 completed\n",
      "235 out of 1239 completed\n",
      "247 out of 1239 completed\n",
      "260 out of 1239 completed\n",
      "272 out of 1239 completed\n",
      "284 out of 1239 completed\n",
      "297 out of 1239 completed\n",
      "309 out of 1239 completed\n",
      "322 out of 1239 completed\n",
      "334 out of 1239 completed\n",
      "346 out of 1239 completed\n",
      "359 out of 1239 completed\n",
      "371 out of 1239 completed\n",
      "384 out of 1239 completed\n",
      "396 out of 1239 completed\n",
      "408 out of 1239 completed\n",
      "421 out of 1239 completed\n",
      "433 out of 1239 completed\n",
      "446 out of 1239 completed\n",
      "458 out of 1239 completed\n",
      "470 out of 1239 completed\n",
      "483 out of 1239 completed\n",
      "495 out of 1239 completed\n",
      "507 out of 1239 completed\n",
      "520 out of 1239 completed\n",
      "532 out of 1239 completed\n",
      "545 out of 1239 completed\n",
      "557 out of 1239 completed\n",
      "569 out of 1239 completed\n",
      "582 out of 1239 completed\n",
      "594 out of 1239 completed\n",
      "607 out of 1239 completed\n",
      "619 out of 1239 completed\n",
      "631 out of 1239 completed\n",
      "644 out of 1239 completed\n",
      "656 out of 1239 completed\n",
      "669 out of 1239 completed\n",
      "681 out of 1239 completed\n",
      "693 out of 1239 completed\n",
      "706 out of 1239 completed\n",
      "718 out of 1239 completed\n",
      "731 out of 1239 completed\n",
      "743 out of 1239 completed\n",
      "755 out of 1239 completed\n",
      "768 out of 1239 completed\n",
      "780 out of 1239 completed\n",
      "792 out of 1239 completed\n",
      "805 out of 1239 completed\n",
      "817 out of 1239 completed\n",
      "830 out of 1239 completed\n",
      "842 out of 1239 completed\n",
      "854 out of 1239 completed\n",
      "867 out of 1239 completed\n",
      "879 out of 1239 completed\n",
      "892 out of 1239 completed\n",
      "904 out of 1239 completed\n",
      "916 out of 1239 completed\n",
      "929 out of 1239 completed\n",
      "941 out of 1239 completed\n",
      "954 out of 1239 completed\n",
      "966 out of 1239 completed\n",
      "978 out of 1239 completed\n",
      "991 out of 1239 completed\n",
      "1003 out of 1239 completed\n",
      "1015 out of 1239 completed\n",
      "1028 out of 1239 completed\n",
      "1040 out of 1239 completed\n",
      "1053 out of 1239 completed\n",
      "1065 out of 1239 completed\n",
      "1077 out of 1239 completed\n",
      "1090 out of 1239 completed\n",
      "1102 out of 1239 completed\n",
      "1115 out of 1239 completed\n",
      "1127 out of 1239 completed\n",
      "1139 out of 1239 completed\n",
      "1152 out of 1239 completed\n",
      "1164 out of 1239 completed\n",
      "1177 out of 1239 completed\n",
      "1189 out of 1239 completed\n",
      "1201 out of 1239 completed\n",
      "1214 out of 1239 completed\n",
      "1226 out of 1239 completed\n",
      "All TTjets events have been created\n",
      "0 out of 1553 completed\n",
      "15 out of 1553 completed\n",
      "31 out of 1553 completed\n",
      "46 out of 1553 completed\n",
      "62 out of 1553 completed\n",
      "77 out of 1553 completed\n",
      "93 out of 1553 completed\n",
      "108 out of 1553 completed\n",
      "124 out of 1553 completed\n",
      "139 out of 1553 completed\n",
      "155 out of 1553 completed\n",
      "170 out of 1553 completed\n",
      "186 out of 1553 completed\n",
      "201 out of 1553 completed\n",
      "217 out of 1553 completed\n",
      "232 out of 1553 completed\n",
      "248 out of 1553 completed\n",
      "264 out of 1553 completed\n",
      "279 out of 1553 completed\n",
      "295 out of 1553 completed\n",
      "310 out of 1553 completed\n",
      "326 out of 1553 completed\n",
      "341 out of 1553 completed\n",
      "357 out of 1553 completed\n",
      "372 out of 1553 completed\n",
      "388 out of 1553 completed\n",
      "403 out of 1553 completed\n",
      "419 out of 1553 completed\n",
      "434 out of 1553 completed\n",
      "450 out of 1553 completed\n",
      "465 out of 1553 completed\n",
      "481 out of 1553 completed\n",
      "496 out of 1553 completed\n",
      "512 out of 1553 completed\n",
      "528 out of 1553 completed\n",
      "543 out of 1553 completed\n",
      "559 out of 1553 completed\n",
      "574 out of 1553 completed\n",
      "590 out of 1553 completed\n",
      "605 out of 1553 completed\n",
      "621 out of 1553 completed\n",
      "636 out of 1553 completed\n",
      "652 out of 1553 completed\n",
      "667 out of 1553 completed\n",
      "683 out of 1553 completed\n",
      "698 out of 1553 completed\n",
      "714 out of 1553 completed\n",
      "729 out of 1553 completed\n",
      "745 out of 1553 completed\n",
      "760 out of 1553 completed\n",
      "776 out of 1553 completed\n",
      "792 out of 1553 completed\n",
      "807 out of 1553 completed\n",
      "823 out of 1553 completed\n",
      "838 out of 1553 completed\n",
      "854 out of 1553 completed\n",
      "869 out of 1553 completed\n",
      "885 out of 1553 completed\n",
      "900 out of 1553 completed\n",
      "916 out of 1553 completed\n",
      "931 out of 1553 completed\n",
      "947 out of 1553 completed\n",
      "962 out of 1553 completed\n",
      "978 out of 1553 completed\n",
      "993 out of 1553 completed\n",
      "1009 out of 1553 completed\n",
      "1024 out of 1553 completed\n",
      "1040 out of 1553 completed\n",
      "1056 out of 1553 completed\n",
      "1071 out of 1553 completed\n",
      "1087 out of 1553 completed\n",
      "1102 out of 1553 completed\n",
      "1118 out of 1553 completed\n",
      "1133 out of 1553 completed\n",
      "1149 out of 1553 completed\n",
      "1164 out of 1553 completed\n",
      "1180 out of 1553 completed\n",
      "1195 out of 1553 completed\n",
      "1211 out of 1553 completed\n",
      "1226 out of 1553 completed\n",
      "1242 out of 1553 completed\n",
      "1257 out of 1553 completed\n",
      "1273 out of 1553 completed\n",
      "1288 out of 1553 completed\n",
      "1304 out of 1553 completed\n",
      "1320 out of 1553 completed\n",
      "1335 out of 1553 completed\n",
      "1351 out of 1553 completed\n",
      "1366 out of 1553 completed\n",
      "1382 out of 1553 completed\n",
      "1397 out of 1553 completed\n",
      "1413 out of 1553 completed\n",
      "1428 out of 1553 completed\n",
      "1444 out of 1553 completed\n",
      "1459 out of 1553 completed\n",
      "1475 out of 1553 completed\n",
      "1490 out of 1553 completed\n",
      "1506 out of 1553 completed\n",
      "1521 out of 1553 completed\n",
      "1537 out of 1553 completed\n",
      "All Wjets events have been created\n"
     ]
    }
   ],
   "source": [
    "#%% Image Creation\n",
    "\n",
    "# Draw the events\n",
    "\n",
    "import images as im\n",
    "\n",
    "im.generateAllEvents(DYjets, 'DYjets', 'json')\n",
    "\n",
    "im.generateAllEvents(TTjets, 'TTjets', 'json')\n",
    "\n",
    "im.generateAllEvents(Wjets, 'Wjets', 'json')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1370 1553 1239\n"
     ]
    }
   ],
   "source": [
    "print(len(listdir('CreatedImages/DYjets Images')),\n",
    "len(listdir('CreatedImages/Wjets Images')),\n",
    "len(listdir('CreatedImages/TTjets Images')))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "CMS",
   "language": "python",
   "name": "cms"
  },
  "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.7.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}