Newer
Older
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
#include "Shower2.h"
#include "THcShower.h"
#include "THcHallCSpectrometer.h"
#include "THaEvData.h"
#include "THaDetMap.h"
#include "THcDetectorMap.h"
#include "THcGlobals.h"
#include "THaCutList.h"
#include "THcParmList.h"
#include "VarDef.h"
#include "VarType.h"
#include "THaTrack.h"
#include "TClonesArray.h"
#include "THaTrackProj.h"
#include "TMath.h"
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <numeric>
#include "nlohmann/json.hpp"
using namespace std;
namespace hcana {
THaAnalysisObject::EStatus Shower2::Init(const TDatime &date) {
auto res = THcShower::Init(date);
LoadJsonCalibration(_run_number);
// hard coding cal_*_cal_const to 0.001.
// Not sure what their purpose is...
for (UInt_t i=0; i<fNTotBlocks; i++) {
fPosGain[i] = 0.001 * _pos_gain_cor[i];
fNegGain[i] = 0.001 * _neg_gain_cor[i];
}
if(fHasArray) {
auto gains = fArray->GetGains();
for (Int_t i=0; i<fArray->GetNelem(); i++) {
gains[i] = 0.001 * _arr_gain_cor[i];
}
}
return fStatus = res;
}
void Shower2::LoadJsonCalibration(int rn, std::string fname) {
// file checks assumed already complete.
using nlohmann::json;
json j_database;
{
std::ifstream json_input_file(fname);
json_input_file >> j_database;
}
std::cout << " runs : ";
std::vector<int> runs;
for (json::iterator it = j_database.begin(); it != j_database.end(); ++it) {
std::cout << it.key() << ", ";
runs.push_back(std::stoi(it.key()));
}
std::cout << "\n";
auto closest = [](std::vector<int> const &vec, int value) {
auto it = std::lower_bound(vec.begin(), vec.end(), value);
if (*it > value) {
it = it - 1;
}
if (it == vec.end()) {
return -1;
}
return *it;
};
// auto best_run = std::upper_bound(runs.begin(), runs.end(), rn);
// std::cout << "using run upper " << rn << " :::: " << *best_run << "\n";
// auto best_run = std::lower_bound(runs.begin(), runs.end(), rn);
// std::cout << "using run lower " << rn << " :::: " << *(best_run) << "\n";
int best_run = closest(runs, rn);
std::cout << " closest " << best_run << std::endl;
json j = j_database[std::to_string(best_run)];
// std::cout << j_database[std::to_string(best_run)] << "\n";;
// std::cout << j.dump(-1) << "\n";
if (j.find("histogram") != j.end()) {
}
if (j.find("cal") != j.end()) {
json j_cal = j["cal"];
if (j_cal.find("neg_gain_cor") != j_cal.end()) {
// std::cout << " neg_gain_cor!!!\n";
_neg_gain_cor = j["cal"]["neg_gain_cor"].get<std::vector<double>>();
}
if (j_cal.find("pos_gain_cor") != j_cal.end()) {
_pos_gain_cor = j["cal"]["pos_gain_cor"].get<std::vector<double>>();
}
if (j_cal.find("arr_gain_cor") != j_cal.end()) {
_arr_gain_cor = j["cal"]["arr_gain_cor"].get<std::vector<double>>();
}
}
}
} // namespace hcana