datasets-2023.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "b9d87283-255e-41b9-b6b6-75295b380188",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Get data file names from stock nano\n",
    "import os\n",
    "import json"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "b90b6992-438a-4a76-85cf-da1962d114e7",
   "metadata": {},
   "outputs": [],
   "source": [
    "datasets = {}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "ef1f8345-19ad-45a7-a57e-1bd727975e6a",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Jet data\n",
    "datasets['JetMET0_Run2023B'] = ['/JetMET0/Run2023B-22Sep2023_v1-v1/NANOAOD']\n",
    "datasets['JetMET0_Run2023C-v1'] = ['/JetMET0/Run2023C-22Sep2023_v1-v1/NANOAOD']\n",
    "datasets['JetMET0_Run2023C-v2'] = ['/JetMET0/Run2023C-22Sep2023_v2-v1/NANOAOD']\n",
    "datasets['JetMET0_Run2023C-v3'] = ['/JetMET0/Run2023C-22Sep2023_v3-v1/NANOAOD']\n",
    "datasets['JetMET0_Run2023C-v4'] = ['/JetMET0/Run2023C-22Sep2023_v4-v1/NANOAOD']\n",
    "datasets['JetMET1_Run2023B'] = ['/JetMET1/Run2023B-22Sep2023_v1-v1/NANOAOD']\n",
    "datasets['JetMET1_Run2023C-v1'] = ['/JetMET1/Run2023C-22Sep2023_v1-v1/NANOAOD']\n",
    "datasets['JetMET1_Run2023C-v2'] = ['/JetMET1/Run2023C-22Sep2023_v2-v1/NANOAOD']\n",
    "datasets['JetMET1_Run2023C-v3'] = ['/JetMET1/Run2023C-22Sep2023_v3-v1/NANOAOD']\n",
    "datasets['JetMET1_Run2023C-v4'] = ['/JetMET1/Run2023C-22Sep2023_v4-v1/NANOAOD']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "355e11a3-79e5-4178-a129-b3e09a25d294",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Muon data\n",
    "datasets['Muon0_Run2023B'] = ['/Muon0/Run2023B-22Sep2023_v1-v1/NANOAOD']\n",
    "datasets['Muon0_Run2023C-v1'] = ['/Muon0/Run2023C-22Sep2023_v1-v1/NANOAOD']\n",
    "datasets['Muon0_Run2023C-v2'] = ['/Muon0/Run2023C-22Sep2023_v2-v1/NANOAOD']\n",
    "datasets['Muon0_Run2023C-v3'] = ['/Muon0/Run2023C-22Sep2023_v3-v1/NANOAOD']\n",
    "datasets['Muon0_Run2023C-v4'] = ['/Muon0/Run2023C-22Sep2023_v4-v1/NANOAOD']\n",
    "datasets['Muon1_Run2023B'] = ['/Muon1/Run2023B-22Sep2023_v1-v1/NANOAOD']\n",
    "datasets['Muon1_Run2023C-v1'] = ['/Muon1/Run2023C-22Sep2023_v1-v1/NANOAOD']\n",
    "datasets['Muon1_Run2023C-v2'] = ['/Muon1/Run2023C-22Sep2023_v2-v1/NANOAOD']\n",
    "datasets['Muon1_Run2023C-v3'] = ['/Muon1/Run2023C-22Sep2023_v3-v1/NANOAOD']\n",
    "datasets['Muon1_Run2023C-v4'] = ['/Muon1/Run2023C-22Sep2023_v4-v1/NANOAOD']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "c38d9b73-8ec0-4d22-a599-299eae9926c7",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Higgs simulation\n",
    "datasets['GluGlu_Hto2B'] = ['/GluGluHto2B_PT-200_M-125_TuneCP5_13p6TeV_powheg-minlo-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['VBF_Hto2B'] = ['/VBFHto2B_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']\n",
    "datasets['WplusH_Hto2B_Wto2Q'] = ['/WplusH_Hto2B_Wto2Q_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['WplusH_Hto2B_WtoLNu'] = ['/WplusH_Hto2B_WtoLNu_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['WminusH_Hto2B_Wto2Q'] = ['/WminusH_Hto2B_Wto2Q_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']\n",
    "datasets['WminusH_Hto2B_WtoLNu'] = ['/WminusH_Hto2B_WtoLNu_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['ZH_Hto2B_Zto2L'] = ['/ZH_Hto2B_Zto2L_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['ZH_Hto2B_Zto2Nu'] = []\n",
    "datasets['ZH_Hto2B_Zto2Q'] = ['/ZH_Hto2B_Zto2Q_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['ggZH_Hto2B_Zto2L'] = ['/ggZH_Hto2B_Zto2L_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']\n",
    "datasets['ggZH_Hto2B_Zto2Nu'] = ['/ggZH_Hto2B_Zto2Nu_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']\n",
    "datasets['ggZH_Hto2B_Zto2Q'] = ['/ggZH_Hto2B_Zto2Q_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']\n",
    "datasets['ttH_Hto2B'] = ['/ttHto2B_M-125_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "a29ededa-0390-4557-8f66-0cc90d165d00",
   "metadata": {},
   "outputs": [],
   "source": [
    "# QCD simulation\n",
    "datasets['QCD_HT100to200'] = ['/QCD-4Jets_HT-100to200_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['QCD_HT200to400'] = ['/QCD-4Jets_HT-200to400_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['QCD_HT400to600'] = ['/QCD-4Jets_HT-400to600_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']\n",
    "datasets['QCD_HT600to800'] = ['/QCD-4Jets_HT-600to800_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['QCD_HT800to1000'] = ['/QCD-4Jets_HT-800to1000_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['QCD_HT1000to1200'] = ['/QCD-4Jets_HT-1000to1200_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['QCD_HT1200to1500'] = ['/QCD-4Jets_HT-1200to1500_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['QCD_HT1500to2000'] = ['/QCD-4Jets_HT-1500to2000_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']\n",
    "datasets['QCD_HT2000toInf'] = ['/QCD-4Jets_HT-2000_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "1b5e4bb4-ce31-4004-b442-4f06aafdbf63",
   "metadata": {},
   "outputs": [],
   "source": [
    "# W+jets simulation\n",
    "datasets['Wto2Q_HT200to400'] = ['/Wto2Q-3Jets_HT-200to400_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['Wto2Q_HT400to600'] = ['/Wto2Q-3Jets_HT-400to600_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['Wto2Q_HT600to800'] = ['/Wto2Q-3Jets_HT-600to800_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['Wto2Q_HT800toInf'] = ['/Wto2Q-3Jets_HT-800_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['WtoLNu']= ['/WtoLNu-4Jets_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "5a5c0143-d59d-4884-a63a-33ea259d793c",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Z+jets simulation\n",
    "datasets['DYJets'] = ['/DYto2L-4Jets_MLL-50_1J_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']\n",
    "datasets['Zto2Nu_HT100to200'] = ['/Zto2Nu-4Jets_HT-100to200_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['Zto2Nu_HT200to400'] = ['/Zto2Nu-4Jets_HT-200to400_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['Zto2Nu_HT400to800'] = ['/Zto2Nu-4Jets_HT-400to800_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['Zto2Nu_HT800to1500'] = ['/Zto2Nu-4Jets_HT-800to1500_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['Zto2Nu_HT1500to2500'] = ['/Zto2Nu-4Jets_HT-1500to2500_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['Zto2Nu_HT2500toInf'] = ['/Zto2Nu-4Jets_HT-2500_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v3/NANOAODSIM']\n",
    "datasets['Zto2Q_HT200to400'] = ['/Zto2Q-4Jets_HT-200to400_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v1/NANOAODSIM']\n",
    "datasets['Zto2Q_HT400to600'] = ['/Zto2Q-4Jets_HT-400to600_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v1/NANOAODSIM']\n",
    "datasets['Zto2Q_HT600to800'] = ['/Zto2Q-4Jets_HT-600to800_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v1/NANOAODSIM']\n",
    "datasets['Zto2Q_HT800toInf'] = ['/Zto2Q-4Jets_HT-800_TuneCP5_13p6TeV_madgraphMLM-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v1/NANOAODSIM']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "57ef9ef5-6e36-49f1-b11e-a96af62e1524",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Diboson\n",
    "datasets['ZZ'] = ['/ZZ_TuneCP5_13p6TeV_pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['WW'] = ['/WW_TuneCP5_13p6TeV_pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['WZ'] = ['/WZ_TuneCP5_13p6TeV_pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "5eecc383-6ef3-40d7-bfef-9920de5c3170",
   "metadata": {},
   "outputs": [],
   "source": [
    "# ttbar\n",
    "datasets['TTto4Q'] = ['/TTto4Q_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['TTto2L2Nu'] = ['/TTto2L2Nu_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['TTtoLNu2Q'] = ['/TTtoLNu2Q_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "eee332d6-4bb1-443a-a142-52b4e99fb94f",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Single top\n",
    "\n",
    "# MISSING TbarWplus_fulllep, TBbar_tchannel, TbarB_tchannel\n",
    "datasets['TWminus_fullhad'] = ['/TWminusto4Q_DS_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['TbarWplus_fullhad'] = ['/TbarWplusto4Q_DS_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v15-v4/NANOAODSIM']\n",
    "datasets['TWminus_fulllep'] = ['/TWminusto2L2Nu_DS_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['TbarWplus_fulllep'] = ['xxx'] \n",
    "datasets['TWminus_semilep'] = ['/TWminustoLNu2Q_DS_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v14-v2/NANOAODSIM']\n",
    "datasets['TbarWplus_semilep'] = ['TbarWplustoLNu2Q_DS_TuneCP5_13p6TeV_powheg-pythia8/Run3Summer23NanoAODv12-130X_mcRun3_2023_realistic_v15-v3/NANOAODSIM']\n",
    "datasets['TBbar_tchannel'] = ['xxx']\n",
    "datasets['TbarB_tchannel'] = ['xxx']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "1e33dcc1-b2fe-4033-aee8-9c5ea294ff72",
   "metadata": {},
   "outputs": [],
   "source": [
    "this_file = 'infiles/2023_datasets.json'\n",
    "with open(this_file, 'w') as json_file:\n",
    "    json.dump(datasets, json_file)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c471c5e7-7759-455b-b9c3-697f38439dcd",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.12.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}