met_check.cxx

Go to the documentation of this file.

#include "core/test.hpp"

#include <cstdlib>
#include <iostream>
#include <string>
#include <vector>
#include <memory>

#include <unistd.h>
#include <getopt.h>

#include "TError.h"
#include "TColor.h"

#include "core/baby.hpp"
#include "core/process.hpp"
#include "core/named_func.hpp"
#include "core/plot_maker.hpp"
#include "core/plot_opt.hpp"
#include "core/palette.hpp"
#include "core/table.hpp"
#include "core/hist1d.hpp"
#include "core/event_scan.hpp"
#include "core/utilities.hpp"

using namespace std;
using namespace PlotOptTypes;

namespace{
  bool single_thread = false;
}

int main(int argc, char *argv[]){
  string bfolder("");
  string hostname = execute("echo $HOSTNAME");
  if(Contains(hostname, "cms") || Contains(hostname, "compute-"))
    bfolder = "/net/cms2"; // In laptops, you can't create a /net folder
  gErrorIgnoreLevel = 6000;
  GetOptions(argc, argv);

  double lumi = 20;

  string trig_skim_mc = "/cms2r0/babymaker/babies/2016_08_10/mc/merged_higmc_higloose/";
  string trig_skim_signal = "/cms2r0/babymaker/babies/2016_08_10/TChiHH/merged_higmc_higloose/";

  Palette colors("txt/colors.txt", "default");
  auto tchi127 = Process::MakeShared<Baby_full>("m_{#chi} = 127", Process::Type::signal, kAzure,
    {trig_skim_signal+"*TChiHH_mGluino-127*.root"});
  auto tchi300 = Process::MakeShared<Baby_full>("m_{#chi} = 300", Process::Type::signal, kGreen+2,
    {trig_skim_signal+"*TChiHH_mGluino-300*.root"});
  auto tchi400 = Process::MakeShared<Baby_full>("m_{#chi} = 400", Process::Type::signal, kRed,
    {trig_skim_signal+"*TChiHH_mGluino-400*.root"});
  auto tchi500 = Process::MakeShared<Baby_full>("m_{#chi} = 500", Process::Type::signal, kViolet-6,
    {trig_skim_signal+"*TChiHH_mGluino-500*.root"});
  auto tchi600 = Process::MakeShared<Baby_full>("m_{#chi} = 600", Process::Type::signal, kYellow,
    {trig_skim_signal+"*TChiHH_mGluino-600*.root"});
  auto tchi700 = Process::MakeShared<Baby_full>("m_{#chi} = 700", Process::Type::signal, kMagenta+1,
    {trig_skim_signal+"*TChiHH_mGluino-700*.root"});
  auto tchi850 = Process::MakeShared<Baby_full>("m_{#chi} = 850", Process::Type::signal, kCyan+1,
    {trig_skim_signal+"*TChiHH_mGluino-850*.root"});
  auto tchi1000 = Process::MakeShared<Baby_full>("m_{#chi} = 1000", Process::Type::signal, kOrange-3,
    {trig_skim_signal+"*TChiHH_mGluino-1000*.root"});

  vector<shared_ptr<Process> > full_trig_skim = {tchi127, tchi300, tchi400, tchi500, tchi600, tchi700, tchi850, tchi1000};

  PlotOpt log_lumi("txt/plot_styles.txt", "CMSPaper");
  log_lumi.Title(TitleType::preliminary)
    .Bottom(BottomType::off)
    .YAxis(YAxisType::log)
    .Stack(StackType::data_norm);
  PlotOpt lin_lumi = log_lumi().YAxis(YAxisType::linear);
  PlotOpt log_shapes = log_lumi().Stack(StackType::shapes)
    .ShowBackgroundError(false);
  PlotOpt lin_shapes = log_shapes().YAxis(YAxisType::linear);
  PlotOpt log_lumi_info = log_lumi().Title(TitleType::info);
  PlotOpt lin_lumi_info = lin_lumi().Title(TitleType::info);
  PlotOpt log_shapes_info = log_shapes().Title(TitleType::info);
  PlotOpt lin_shapes_info = lin_shapes().Title(TitleType::info);
  vector<PlotOpt> all_plot_types = {log_lumi_info, log_shapes_info};

  PlotMaker pm;

  vector<TString> nbcuts;
  nbcuts.push_back(" nbt >= 2 && nbl <= 3");
  nbcuts.push_back(" nbt >= 2 && nbl >= 4");
  TString metskim("njets>=4&&njets<=5&&met>100&&nvleps==0");
  TString trkskim("njets>=4&&njets<=5&&met>250&&nvleps==0");
  TString skim("njets>=4&&njets<=5&&met>250&&nvleps==0&&ntks==0");
  TString A("&&");
  TString DeltaR("hig_drmax < 2.2");
  TString AverageM("hig_am > 100 && hig_am < 140");
  TString DeltaM("hig_dm < 40");
  TString LDP("!low_dphi");

  pm.Push<Hist1D>(Axis(40,100,1700,"met", "E_{T}^{miss} [GeV]", {150., 250.}),
                  metskim, full_trig_skim, all_plot_types);

  pm.Push<Table>("TChiMass_cutflow", vector<TableRow>{
      TableRow("$MET > 100$, $\\text{2M b-tags}$, $\\text{4 or 5 jets}$, $0\\ell$", "1"),
      TableRow("$\\Delta\\phi_{\\text{min}}$", LDP),
      TableRow("$\\Delta m < 40$", DeltaM+A+LDP,1,0),
      TableRow("$\\left< m \\right> \\in (100,140)$", AverageM+A+LDP),
      TableRow("$\\Delta R_{\\text{max}} < 2.2$", DeltaR+A+LDP)
      },full_trig_skim, 0);

  if(single_thread) pm.multithreaded_ = false;
  pm.min_print_ = true;
  pm.MakePlots(lumi);
}

void GetOptions(int argc, char *argv[]){
  while(true){
    static struct option long_options[] = {
      {"single_thread", no_argument, 0, 's'},
      {0, 0, 0, 0}
    };

    char opt = -1;
    int option_index;
    opt = getopt_long(argc, argv, "s", long_options, &option_index);

    if( opt == -1) break;

    string optname;
    switch(opt){
    case 's':
      single_thread = true;
      break;
    case 0:
      optname = long_options[option_index].name;
      if(false){
      }else{
        printf("Bad option! Found option name %s\n", optname.c_str());
      }
      break;
    default:
      printf("Bad option! getopt_long returned character code 0%o\n", opt);
      break;
    }
  }
}