MVALikelihood/MVALikelihood_Original.cpp at master · abdollah110/MVALikelihood · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#include "Analysis/MVALikelihood/interface/MVALikelihood.h"
#include "TFile.h"
#include "TTree.h"
#include "TH1.h"
#include "TMath.h"
int main(int argc, char * argv[]) {

  float ptRatio;
  float dcaSig;
  float dPhiMuMet;
  float diMuEta;

  // Creating RooT file for storage of histograms
  TFile * file = new TFile("MVALikelihood.root","recreate");
  file->cd("");

  // histograms

  // Z->TauTau sample
  TH1F * diMuPtRatioZTauTauH = new TH1F("diMuPtRatioZTauTauH","",25,0.,1.); // dimuon pt - to scalar sum of muons' pt ratio
  TH1F * dcaSigZTauTauH = new TH1F("dcaSigZTauTauH","",25,-3,2);
  TH1F * dPhiMuMetZTauTauH = new TH1F("dPhiLepMetZTauTauH","",25,0.,TMath::Pi());
  TH1F * diMuEtaZTauTauH = new TH1F("diMuEtaZTauTauH","",30,-6,6);
  TH1F * likelihoodZTauTauH = new TH1F("likelihoodZTauTauH","",20,0,1);

  // Z->MuMu sample
  TH1F * diMuPtRatioZMuMuH = new TH1F("diMuPtRatioZMuMuH","",25,0.,1.); // dimuon pt - to scalar sum of muons' pt ratio
  TH1F * dcaSigZMuMuH = new TH1F("dcaSigZMuMuH","",25,-3,2);
  TH1F * dPhiMuMetZMuMuH = new TH1F("dPhiLepMetZMuMuH","",25,0.,TMath::Pi());
  TH1F * diMuEtaZMuMuH = new TH1F("diMuEtaZMuMuH","",30,-6,6);
  TH1F * likelihoodZMuMuH = new TH1F("likelihoodZMuMuH","",20,0,1);

  // Initialization of the MVA
  StringVector classNames;
  classNames.clear();
  classNames.push_back("ZToTauTau");
  classNames.push_back("ZToMuMu");

  StringVector variableNames;
  variableNames.clear();
  variableNames.push_back("ptRatio");
  variableNames.push_back("dcaSig");
  variableNames.push_back("diLepEta");
  variableNames.push_back("dPhiPos");

  MVALikelihood * mvaLikelihood = new MVALikelihood(classNames,variableNames);


  TFile * fileZTauTau = new TFile("ZToTauTau_Like.root");
  if (fileZTauTau->IsZombie()) {
    std::cout << "File with name ZToTauTau_Like.root does not exist" << std::endl;
    std::cout << "Terminating program" << std::endl;
    exit(-1);
  }
  TTree * tree = (TTree*)fileZTauTau->Get("likeTree");

  tree->SetBranchAddress("DiLepEta",&diMuEta);
  tree->SetBranchAddress("PosMETDPhi",&dPhiMuMet);
  tree->SetBranchAddress("MuonDCASig",&dcaSig);
  tree->SetBranchAddress("DiMuonPtRatio",&ptRatio);

  int nEvt = tree->GetEntries();
  for (int iEvt=0;iEvt<nEvt;++iEvt) {

    tree->GetEntry(iEvt);

    StringDoubleMap variables;
    variables[std::string("ptRatio")] = ptRatio;
    variables[std::string("dcaSig")]  = dcaSig;
    variables[std::string("diLepEta")] = diMuEta;
    variables[std::string("dPhiPos")] = dPhiMuMet;

    float likelihood = float(mvaLikelihood->getLikelihood(variables,"ZToTauTau"));

    diMuPtRatioZTauTauH->Fill(ptRatio);
    dcaSigZTauTauH->Fill(dcaSig);
    dPhiMuMetZTauTauH->Fill(dPhiMuMet);
    diMuEtaZTauTauH->Fill(diMuEta);
    likelihoodZTauTauH->Fill(likelihood);

  }

  TFile * fileZMuMu = new TFile("ZToMuMu_Like.root");
  if (fileZMuMu->IsZombie()) {
    std::cout << "File with name ZToMuMu_Like.root does not exist" << std::endl;
    std::cout << "Terminating program" << std::endl;
    exit(-1);
  }
  tree = (TTree*)fileZMuMu->Get("likeTree");

  tree->SetBranchAddress("DiLepEta",&diMuEta);
  tree->SetBranchAddress("PosMETDPhi",&dPhiMuMet);
  tree->SetBranchAddress("MuonDCASig",&dcaSig);
  tree->SetBranchAddress("DiMuonPtRatio",&ptRatio);

  nEvt = tree->GetEntries();
  for (int iEvt=0;iEvt<nEvt;++iEvt) {

    tree->GetEntry(iEvt);

    StringDoubleMap variables;
    variables[std::string("ptRatio")] = ptRatio;
    variables[std::string("dcaSig")]  = dcaSig;
    variables[std::string("diLepEta")] = diMuEta;
    variables[std::string("dPhiPos")] = dPhiMuMet;

    float likelihood = float(mvaLikelihood->getLikelihood(variables,"ZToTauTau"));

    diMuPtRatioZMuMuH->Fill(ptRatio);
    dcaSigZMuMuH->Fill(dcaSig);
    dPhiMuMetZMuMuH->Fill(dPhiMuMet);
    diMuEtaZMuMuH->Fill(diMuEta);
    likelihoodZMuMuH->Fill(likelihood);

  }

  file->cd("");
  file->Write();
  file->Close();
  delete file;
  delete mvaLikelihood;
  delete fileZMuMu;
  delete fileZTauTau;

}