QM2023_macros/drawTruthPatternXY.C at master · bkimelman/QM2023_macros · 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
void drawTruthPatternXY(bool pos=true){

	gROOT->LoadMacro("sPhenixStyle.C");
	SetsPhenixStyle();

	TFile *f = new TFile("inputFiles/truthSectors.root","READ");

	TTree *tree = (TTree*)f->Get("truthTree");

	unsigned int id = 0;
	TVector3 *vec = nullptr;

	tree->SetBranchAddress("truthID",&id);
	tree->SetBranchAddress("truthPos",&vec);

	vector<double> X1, X2, X3, Y1, Y2, Y3;

	for(int i=0; i<tree->GetEntries(); i++){
		tree->GetEntry(i);

		if( (pos && id >= 180000) || (!pos && id < 180000 ) ) continue;

		int sec = id/10000;
		int row = (id - (sec*10000))/100;

		if(row<16){
			X1.push_back(vec->X());
			Y1.push_back(vec->Y());
		}else if(row>23){
			X3.push_back(vec->X());
			Y3.push_back(vec->Y());
		}else{
			X2.push_back(vec->X());
			Y2.push_back(vec->Y());
		}
	}

	TGraph *g1 = new TGraph(X1.size(),&X1[0],&Y1[0]);
	g1->SetMarkerStyle(20);
	g1->SetMarkerSize(0.5);
	g1->SetMarkerColor(kBlue);

	TGraph *g2 = new TGraph(X2.size(),&X2[0],&Y2[0]);
	g2->SetMarkerStyle(20);
	g2->SetMarkerSize(0.5);
	g2->SetMarkerColor(kRed);

	TGraph *g3 = new TGraph(X3.size(),&X3[0],&Y3[0]);
	g3->SetMarkerStyle(20);
	g3->SetMarkerSize(0.5);
	g3->SetMarkerColor(kGreen+2);
	g3->GetXaxis()->SetTitle("X [cm]");
	//g3->GetXaxis()->SetTitleOffset(1.2f);
	g3->GetYaxis()->SetTitle("Y [cm]");
	//g3->GetYaxis()->SetTitleOffset(1.2f);
	g3->SetTitle("");


	TLatex sPHENIX;
    sPHENIX.SetTextFont(42);
    sPHENIX.SetTextAlign(12);
    sPHENIX.SetTextSize(0.035);


	TLatex stripe;
	stripe.SetTextSize(42);
	stripe.SetTextAlign(12);
	stripe.SetTextSize(0.035);

	TLatex date;
	date.SetTextSize(35);
	date.SetTextAlign(12);
	date.SetTextSize(0.035);

	TCanvas *c1 = new TCanvas("c1","",1000,1000);
	//c1->SetLeftMargin(3.4);
	//c1->SetBottomMargin(3.4);
	c1->SetFixedAspectRatio();
	g3->Draw("AP");
	g2->Draw("PSAME");
	g1->Draw("PSAME");
    sPHENIX.DrawLatexNDC(0.175,0.92,"#bf{#it{sPHENIX}} Simulation");
    stripe.DrawLatexNDC(0.175,0.875,"Al stripes on TPC CM");
    date.DrawLatexNDC(0.82,0.97,"#it{08/30/2023}");


	c1->SaveAs("TruthPositions.svg");
	c1->SaveAs("TruthPositions.png");

}