|
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
109
110
111
|
#include "FastSimulation/CTPPSFastGeometry/interface/CTPPSToFDetector.h"
#include <algorithm>
#include <cmath>
CTPPSToFDetector::CTPPSToFDetector(
int ncellx, int ncelly, std::vector<double>& cellw, double cellh, double pitchx, double pitchy, double pos, int res)
: nCellX_(ncellx),
nCellY_(ncelly),
cellW_(cellw),
cellH_(cellh),
pitchX_(pitchx),
pitchY_(pitchy),
fToFResolution_(res),
detPosition_(pos) {
// the vertical positions starts from the negative(bottom) to the positive(top) corner
// vector index points to the row number from below
cellRow_.push_back(std::pair<double, double>(-cellH_ * 0.5, cellH_ * 0.5));
cellColumn_.reserve(nCellX_); // vector index points to the column number
for (int i = 0; i < nCellX_; i++) {
double x1 = 0., x2 = 0.;
if (i == 0) {
detW_ = pitchX_;
x1 = -(detPosition_ + detW_);
} else
x1 = -detPosition_ + detW_; //detPosition_ - shift the limit of a column depending on the detector position
x2 = x1 - cellW_.at(i);
detW_ += (x2 - x1) - pitchX_;
cellColumn_.push_back(std::pair<double, double>(x1, x2));
}
//diamond geometry
detH_ = nCellY_ * cellH_;
detW_ = -detW_ - 2 * pitchX_;
};
CTPPSToFDetector::CTPPSToFDetector(
int ncellx, int ncelly, double cellwq, double cellh, double pitchx, double pitchy, double pos, int res)
: nCellX_(ncellx),
nCellY_(ncelly),
cellWq_(cellwq),
cellH_(cellh),
pitchX_(pitchx),
pitchY_(pitchy),
fToFResolution_(res),
detPosition_(pos) {
//
detW_ = nCellX_ * cellWq_ + (nCellX_ - 1) * pitchX_;
detH_ = nCellY_ * cellH_ + (nCellY_ - 1) * pitchY_;
// the vertical positions starts from the negative(bottom) to the positive(top) corner
cellRow_.reserve(nCellY_); // vector index points to the row number from below
for (int i = 0; i < nCellY_; i++) {
double y1 = cellH_ * (i - nCellY_ * 0.5) + pitchY_ * (i - (nCellY_ - 1) * 0.5);
double y2 = y1 + cellH_;
cellRow_.push_back(std::pair<double, double>(y1, y2));
}
cellColumn_.reserve(nCellX_); // vector index points to the column number
for (int i = 0; i < nCellX_; i++) {
double x1 = -(cellWq_ * i + pitchX_ * i);
x1 -= detPosition_; // shift the limit of a column depending on the detector position
double x2 = x1 - cellWq_;
cellColumn_.push_back(std::pair<double, double>(x1, x2));
}
};
void CTPPSToFDetector::AddHit(double x, double y, double tof) {
int cellid = findCellId(x, y);
if (cellid == 0)
return;
if (theToFInfo.find(cellid) == theToFInfo.end())
theToFInfo[cellid]; // add empty cell
std::vector<double>* tofs = &(theToFInfo.find(cellid)->second);
int ntof = tofs->size();
int i = 0;
double oneOverRes = 1.0 / fToFResolution_;
for (; i < ntof; i++) {
if (fabs(tofs->at(i) - tof) * oneOverRes < 3) {
tofs->at(i) = (tofs->at(i) + tof) / 2.;
nADC_.at(cellid).at(i)++;
return;
}
}
tofs->push_back(tof); // no other ToF inside resolution found
nHits_++;
nADC_[cellid].push_back(1);
}
int CTPPSToFDetector::findCellId(double x, double y) {
auto it = std::find_if(
cellRow_.begin(), cellRow_.end(), [y](const auto& cell) { return y >= cell.first && y <= cell.second; });
if (it == cellRow_.end())
return 0;
unsigned int y_idx = std::distance(cellRow_.begin(), it) + 1;
it = std::find_if(
cellColumn_.begin(), cellColumn_.end(), [x](const auto& cell) { return x <= cell.first && x > cell.second; });
if (it == cellColumn_.end())
return 0;
unsigned int x_idx = std::distance(cellColumn_.begin(), it) + 1;
return 100 * y_idx + x_idx;
}
bool CTPPSToFDetector::get_CellCenter(int cell_id, double& x, double& y) {
if (cell_id == 0)
return false;
//if(!isValidCellId(cell_id)) return 0;
unsigned int y_idx = int(cell_id * 0.01);
unsigned int x_idx = cell_id - y_idx * 100;
x = (cellColumn_.at(x_idx - 1).first + cellColumn_.at(x_idx - 1).second) / 2.0;
y = (cellRow_.at(y_idx - 1).first + cellRow_.at(y_idx - 1).second) / 2.0;
return true;
}
|