HLT/pseudotest.cxx at master · emilbols/HLT · 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
void printcor(Double_t x){
    TLatex l; //l.SetTextAlign(12); l.SetTextSize(tsize);
    l.SetNDC();
    l.SetTextFont(72);
    l.SetTextColor(kBlack);
    l.SetTextSize(0.035);
    std::string str = std::to_string(x);
    l.DrawLatex(0.15,0.91,("correlation factor = " + str).c_str());
}

void pseudotest(){

  //TFile *ntfile = new TFile("ntuple_qcd_15_30_Fall17_111.root","READ");
	//TFile *ntfile = new TFile("nomatching.root","READ");
	//TTree *data = (TTree*)ntfile->Get("deepntuplizer/tree");
        TFile *ntfile = new TFile("ntuple.root","READ");
	TTree *data = (TTree*)ntfile->Get("Ntupler/tree");


	float eff[10] = {};
	float pass[10] = {};
	float fail[10] = {};
	float ueff[10] = {};
	float TagVarCSV_trackJetPt[50] = {};
	int TagVarCSV_FirstTrack[50] = {};
	float TagVarCSVTrk_trackPtRel[1000] = {};
	float jetNTracks, TagVarCSV_jetNSecondaryVertices, TagVarCSV_trackSip3dValAboveCharm,TagVarCSV_trackSip2dValAboveCharm,TagVarCSV_trackSip3dSigAboveCharm,TagVarCSV_trackSip2dSigAboveCharm;

	UInt_t OnNStoredVertices,NStoredVertices;

	float TagVarCSV_jetNSelectedTracks[50] = {};
	float trackSumJetDeltaR, vertexCategory, trackSumJetETRatio, jetNTracksEtaRel, OnjetNTracksEtaRel;

	UInt_t n_jetNSelectedTracks_;
	int nJet;

	float trackEtaRel[100] = {};

	float TagVarCSV_flightDistance2dVal[10] = {};
	float TagVarCSV_flightDistance3dVal[10] = {};
	float TagVarCSV_flightDistance2dSig[10] = {};
	float TagVarCSV_flightDistance3dSig[10] = {};

	data->SetBranchAddress("TagVarCSV_flightDistance2dVal",&TagVarCSV_flightDistance2dVal);
	data->SetBranchAddress("TagVarCSV_flightDistance3dVal",&TagVarCSV_flightDistance3dVal);
	data->SetBranchAddress("TagVarCSV_flightDistance2dSig",&TagVarCSV_flightDistance2dSig);
	data->SetBranchAddress("TagVarCSV_flightDistance3dSig",&TagVarCSV_flightDistance3dSig);
	data->SetBranchAddress("TagVarCSV_vertexCategory",&vertexCategory);
/*
	data->SetBranchAddress("TagVarCSV_trackJetPt",&TagVarCSV_trackJetPt);
	data->SetBranchAddress("TagVarCSV_jetNTracks",&TagVarCSV_jetNSelectedTracks);
	data->SetBranchAddress("TagVarCSV_trackPtRel",&TagVarCSVTrk_trackPtRel);
	data->SetBranchAddress("Jet_nFirstTrkTagVarCSV",&TagVarCSV_FirstTrack);
	data->SetBranchAddress("TagVarCSV_vertexMass",&TagVarCSV_vertexMass);
	data->SetBranchAddress("TagVarCSV_vertexNTracks",&TagVarCSV_vertexNTracks);
	data->SetBranchAddress("TagVarCSV_vertexJetDeltaR",&TagVarCSV_vertexJetDeltaR);
	data->SetBranchAddress("TagVarCSV_vertexEnergyRatio",&TagVarCSV_vertexEnergyRatio);
	data->SetBranchAddress("TagVarCSV_flightDistance2dVal",&TagVarCSV_flightDistance2dVal);
	data->SetBranchAddress("TagVarCSV_flightDistance3dVal",&TagVarCSV_flightDistance3dVal);
	data->SetBranchAddress("TagVarCSV_flightDistance2dSig",&TagVarCSV_flightDistance2dSig);
	data->SetBranchAddress("TagVarCSV_flightDistance3dSig",&TagVarCSV_flightDistance3dSig);
	data->SetBranchAddress("DeepCSVProbb",&DeepCSVProbb);
	data->SetBranchAddress("DeepCSVProbbb",&DeepCSVProbbb);
	data->SetBranchAddress("n_TagVarCSV_jetNSelectedTracks",&n_jetNSelectedTracks_);
	data->SetBranchAddress("DeepCSVProbc",&DeepCSVProbc);
	data->SetBranchAddress("DeepCSVProbudsg",&DeepCSVProbudsg);
	data->SetBranchAddress("CSVProbb",&CSVProbb);
	data->SetBranchAddress("trackJetPt",&trackJetPt);
	data->SetBranchAddress("jetNTracks",&jetNTracks);
	data->SetBranchAddress("n_StoredVertices",&NStoredVertices);
	data->SetBranchAddress("TagVarCSV_jetNSecondaryVertices",&TagVarCSV_jetNSecondaryVertices);
	data->SetBranchAddress("TagVarCSV_jetNSelectedTracks",&TagVarCSV_jetNSelectedTracks);
	data->SetBranchAddress("TagVarCSV_trackSip3dValAboveCharm",&TagVarCSV_trackSip3dValAboveCharm);
	data->SetBranchAddress("TagVarCSV_trackSip2dValAboveCharm",&TagVarCSV_trackSip2dValAboveCharm);
	data->SetBranchAddress("TagVarCSV_trackSip3dSigAboveCharm",&TagVarCSV_trackSip3dSigAboveCharm);
	data->SetBranchAddress("TagVarCSV_trackSip2dSigAboveCharm",&TagVarCSV_trackSip2dSigAboveCharm);
	data->SetBranchAddress("TagVarCSV_vertexCategory",&vertexCategory);
	data->SetBranchAddress("TagVarCSV_trackSumJetEtRatio",&trackSumJetETRatio);
	data->SetBranchAddress("TagVarCSV_jetNTracksEtaRel",&jetNTracksEtaRel);
	data->SetBranchAddress("TagVarCSV_trackSumJetDeltaR",&trackSumJetDeltaR);
	data->SetBranchAddress("TagVarCSV_trackEtaRel",&trackEtaRel);
	data->SetBranchAddress("TagVarCSVTrk_trackDeltaR",&TagVarCSVTrk_trackDeltaR);
	data->SetBranchAddress("TagVarCSVTrk_trackPtRatio",&TagVarCSVTrk_trackPtRatio);
	data->SetBranchAddress("TagVarCSVTrk_trackSip3dSig",&TagVarCSVTrk_trackSip3dSig);
	data->SetBranchAddress("TagVarCSVTrk_trackSip2dSig",&TagVarCSVTrk_trackSip2dSig);
	data->SetBranchAddress("TagVarCSVTrk_trackDecayLenVal",&TagVarCSVTrk_trackDecayLenVal);
	data->SetBranchAddress("TagVarCSVTrk_trackJetDistVal",&TagVarCSVTrk_trackJetDistVal);

*/


	Int_t nentries = data->GetEntries();
	int trackcount = 0;
	float alltracks = 0;
	/*
	for(int n = 0; n<nentries; n++){
	  data->GetEntry(n);
	  if(vertexCategory == 1){
	    cout << n << endl;
	  }
	}
	*/
	TagVarCSV_flightDistance2dVal[0] = 0;
	data->GetEntry(143);
	cout << vertexCategory << endl;
	cout << TagVarCSV_flightDistance2dVal[0] << endl;

}