|
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
|
//
// Macro to read in geomdet positions and draw them
//
int plotDets( )
{
TString cut("");
return plotDets( cut );
}
int plotDets( TString cut )
{
int showFlag = 3;
float xmin = 0;
float xmax = 0;
return plotDets( cut, showFlag, xmin, xmax );
}
int plotDets( TString cut, int showFlag, float xmin, float xmax )
{
TCanvas* c1 = new TCanvas("c1","Misalignment validation",10,10,800,420);
c1->Divide(2,1);
int resolution = 1;
TFile* alignedFile = new TFile("aligned.root");
TTree* tmpTree = (TTree*)alignedFile->Get("theTree");
TTree* alignedTree = (TTree*)tmpTree->CopyTree(cut);
TFile* misalignedFile = new TFile("misaligned.root");
tmpTree = (TTree*)misalignedFile->Get("theTree");
TTree* misalignedTree = (TTree*)tmpTree->CopyTree(cut);
// X-Y projection
if ( fabs(xmax)<1e-12 ) xmax = TMath::Nint(alignedTree->GetMaximum("x")+1);
if ( fabs(xmin)<1e-12 ) xmin = TMath::Nint(alignedTree->GetMinimum("x")-1);
int nx = TMath::Nint(xmax-xmin)*resolution;
TH2F* hRange1 = new TH2F("hRange1","X-Y Projection",nx,xmin,xmax,nx,xmin,xmax);
TH2F* hAlignedXY = new TH2F("hAlignedXY","X-Y Projection - Aligned",nx,xmin,xmax,nx,xmin,xmax);
TH2F* hMisalignedXY = new TH2F("hMisalignedXY","X-Y Projection - Misaligned",
nx,xmin,xmax,nx,xmin,xmax);
hAlignedXY->SetMarkerColor(4);
hAlignedXY->SetMarkerStyle(5);
hMisalignedXY->SetMarkerColor(2);
hMisalignedXY->SetMarkerStyle(5);
alignedTree->Project("hAlignedXY","y:x");
misalignedTree->Project("hMisalignedXY","y:x");
c1->cd(1);
hRange1->Draw();
hRange1->SetXTitle("X");
hRange1->SetYTitle("Y");
if ( showFlag&1 ) hAlignedXY->Draw("same");
if ( showFlag&(1<<1) ) hMisalignedXY->Draw("same");
// R-Z projection
xmin = 0;
float zmax = TMath::Nint(alignedTree->GetMaximum("z")+1);
float zmin = TMath::Nint(alignedTree->GetMinimum("z")-1);
int nx = TMath::Nint(xmax-xmin)*resolution;
int nz = TMath::Nint(zmax-zmin)*resolution;
TH2F* hRange2 = new TH2F("hRange2","R-Z Projection",nz,zmin,zmax,nx,xmin,xmax);
TH2F* hAlignedRZ = new TH2F("hAlignedRZ","X-Y Projection - Aligned",nz,zmin,zmax,nx,xmin,xmax);
TH2F* hMisalignedRZ = new TH2F("hMisalignedRZ","X-Y Projection - Misaligned",
nz,zmin,zmax,nx,xmin,xmax);
hAlignedRZ->SetMarkerColor(4);
hMisalignedRZ->SetMarkerColor(2);
alignedTree->Project("hAlignedRZ","sqrt(x^2+y^2):z");
misalignedTree->Project("hMisalignedRZ","sqrt(x^2+y^2):z");
std::cout << hAlignedRZ->GetEntries() << " aligned detectors selected" << std::endl;
std::cout << hMisalignedRZ->GetEntries() << " misaligned detectors selected" << std::endl;
c1->cd(2);
hRange2->Draw();
hRange2->SetYTitle("R");
hRange2->SetXTitle("Z");
if ( showFlag&1 ) hAlignedRZ->Draw("same");
if ( showFlag&(1<<1) ) hMisalignedRZ->Draw("same");
return 0;
}
|