| 1 |
kukartse |
1.1 |
#include <iostream>
|
| 2 |
|
|
//#include <fstream>
|
| 3 |
|
|
//#include <sstream>
|
| 4 |
|
|
//using namespace std;
|
| 5 |
|
|
|
| 6 |
|
|
//#include "TROOT.h"
|
| 7 |
|
|
//#include "TApplication.h"
|
| 8 |
|
|
|
| 9 |
kukartse |
1.2 |
#include "TChain.h"
|
| 10 |
|
|
#include "TFile.h"
|
| 11 |
|
|
#include "TTree.h"
|
| 12 |
|
|
#include "LJetsTopoVars.hh"
|
| 13 |
|
|
|
| 14 |
kukartse |
1.1 |
int main(int argc, char **argv){
|
| 15 |
kukartse |
1.2 |
std::cout << "\nTest application for the LJetsTopoVars class\n" << std::endl;
|
| 16 |
|
|
|
| 17 |
|
|
TFile * out_file = new TFile("./sampleTopoVars.root", "RECREATE");
|
| 18 |
|
|
|
| 19 |
|
|
// Define some mock jets, lepton and MET
|
| 20 |
|
|
std::vector<TLorentzVector> jets;
|
| 21 |
|
|
TLorentzVector jet;
|
| 22 |
|
|
jet.SetPtEtaPhiM(80, 0.1, -2, 0);
|
| 23 |
|
|
jets.push_back(jet);
|
| 24 |
|
|
jet.SetPtEtaPhiM(70, -0.2, 1, 0);
|
| 25 |
|
|
jets.push_back(jet);
|
| 26 |
|
|
jet.SetPtEtaPhiM(60, 0.03, 2, 0);
|
| 27 |
|
|
jets.push_back(jet);
|
| 28 |
|
|
jet.SetPtEtaPhiM(50, -0.04, -3, 0);
|
| 29 |
|
|
jets.push_back(jet);
|
| 30 |
|
|
|
| 31 |
|
|
TLorentzVector lepton;
|
| 32 |
|
|
lepton.SetPtEtaPhiM(50, 0.7, 1.5, 0.105);
|
| 33 |
|
|
|
| 34 |
|
|
TLorentzVector met;
|
| 35 |
|
|
met.SetPtEtaPhiM(20, 0.0, 1.0, 0);
|
| 36 |
|
|
|
| 37 |
|
|
// Instantiate the topovars calculator
|
| 38 |
|
|
LJetsTopoVars vars(jets, lepton, met, true);
|
| 39 |
|
|
|
| 40 |
|
|
// Can reuse the same instance
|
| 41 |
|
|
// vars.setEvent(jets, lepton, met, true);
|
| 42 |
|
|
|
| 43 |
|
|
// compute variables of interest
|
| 44 |
|
|
double _aplanarity = vars.aplanarity();
|
| 45 |
|
|
|
| 46 |
|
|
// create new TTree and fill the varibales in it
|
| 47 |
|
|
TTree * oTree = new TTree("topovars", "topovars");
|
| 48 |
|
|
oTree->Branch("aplanarity", &_aplanarity, "aplanarity/D");
|
| 49 |
|
|
oTree->Fill();
|
| 50 |
|
|
|
| 51 |
|
|
// Write tree to a file and file to disk
|
| 52 |
|
|
out_file->cd();
|
| 53 |
|
|
oTree->Write();
|
| 54 |
|
|
out_file->Write();
|
| 55 |
|
|
delete out_file;
|
| 56 |
kukartse |
1.1 |
|
| 57 |
|
|
return 0;
|
| 58 |
|
|
}
|
