HiggsAnalysisTools/src/RedIsolationOptimTree.cc at master · VecbosApp/HiggsAnalysisTools · 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
#include "HiggsAnalysisTools/include/RedIsolationOptimTree.hh"

// C++
#include <iostream>
#include <iomanip>
#include <vector>
#include <string>
#include <cmath>

// Root
#include "TFile.h"
#include "TTree.h"

RedIsolationOptimTree::RedIsolationOptimTree(const char * filename) {

  myFile = new TFile(filename,"RECREATE");
  myTree = new TTree("T1","eleID tree");

  myTree->Branch("H_trackerIsol", &myH_trackerIsol, "H_trackerIsol/F");
  myTree->Branch("H_hcalIsol",    &myH_hcalIsol,    "H_hcalIsol/F");
  myTree->Branch("H_ecalIsol",    &myH_ecalIsol,    "H_ecalIsol/F");
  myTree->Branch("L_trackerIsol", &myL_trackerIsol, "L_trackerIsol/F");
  myTree->Branch("L_hcalIsol",    &myL_hcalIsol,    "L_hcalIsol/F");
  myTree->Branch("L_ecalIsol",    &myL_ecalIsol,    "L_ecalIsol/F");
}

RedIsolationOptimTree::~RedIsolationOptimTree()
{
  delete myFile;
}

void RedIsolationOptimTree::store()
{
  myTree->Fill();
}


void RedIsolationOptimTree::save()
{
  myFile->cd();
  myTree->Write();
  myFile->Close();
}

void RedIsolationOptimTree::fillAll(float fhtracker, float fhhcal, float fhecal, float fltracker, float flhcal, float flecal)
{
  myH_trackerIsol = fhtracker;
  myH_hcalIsol    = fhhcal;
  myH_ecalIsol    = fhecal;
  myL_trackerIsol = fltracker;
  myL_hcalIsol    = flhcal;
  myL_ecalIsol    = flecal;
}