rtang/ANASEN_analysis
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modified: GainMatchSX3.C

Browse Source
This commit is contained in:
Vignesh Sitaraman 2025-06-10 11:24:01 -04:00
parent fb355a3cc4
commit 050cb425d5
1 changed files with 40 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
GainMatchSX3.C
View File

@ -26,7 +26,7 @@ int padID = 0;
SX3 sx3_contr; SX3 sx3_contr;
TCutG *cut; TCutG *cut;
std::map<std::tuple<int, int, int>, std::vector<std::tuple<double, double, double>>> dataPoints; std::map<std::tuple<int, int, int, int>, std::vector<std::tuple< double, double, double>>> dataPoints;
void GainMatchSX3::Begin(TTree * /*tree*/) void GainMatchSX3::Begin(TTree * /*tree*/)
{ {
@ -85,11 +85,11 @@ Bool_t GainMatchSX3::Process(Long64_t entry)
for (int i = 0; i < sx3.multi; i++) for (int i = 0; i < sx3.multi; i++)
{ {
for (int j = i + 1; j < sx3.multi; j++) // for (int j = i + 1; j < sx3.multi; j++)
{ // {
if (sx3.id[i] == 3) // if (sx3.id[i] == 3)
hsx3Coin->Fill(sx3.index[i], sx3.index[j]); // hsx3Coin->Fill(sx3.index[i], sx3.index[j]);
} // }
if (sx3.e[i] > 100) if (sx3.e[i] > 100)
{ {
ID.push_back(std::pair<int, int>(sx3.id[i], i)); ID.push_back(std::pair<int, int>(sx3.id[i], i));
@ -193,7 +193,7 @@ Bool_t GainMatchSX3::Process(Long64_t entry)
// if (sx3.id[i] < 24 && sx3ChUp < 4 && sx3ChBk < 4 && std::isfinite(sx3EUp) && std::isfinite(sx3EDn) && std::isfinite(sx3EBk)) // if (sx3.id[i] < 24 && sx3ChUp < 4 && sx3ChBk < 4 && std::isfinite(sx3EUp) && std::isfinite(sx3EDn) && std::isfinite(sx3EBk))
{ {
// Accumulate data for gain matching // Accumulate data for gain matching
dataPoints[{sx3.id[i], sx3ChUp, sx3ChBk}].emplace_back(sx3EBk, sx3EUp, sx3EDn); dataPoints[{sx3.id[i], sx3ChBk, sx3ChUp, sx3ChDn}].emplace_back(sx3EBk, sx3EUp, sx3EDn);
} }
} }
} }
@ -210,10 +210,10 @@ void GainMatchSX3::Terminate()
const int MAX_DOWN = 4; const int MAX_DOWN = 4;
const int MAX_BK = 4; const int MAX_BK = 4;
double gainArray[MAX_DET][MAX_UP][MAX_BK] = {{{0}}}; double gainArray[MAX_DET][MAX_BK][MAX_UP][MAX_DOWN] = {{{{0}}}};
bool gainValid[MAX_DET][MAX_UP][MAX_BK] = {{{false}}}; bool gainValid[MAX_DET][MAX_BK][MAX_UP][MAX_DOWN] = {{{{false}}}};
double fbgain[MAX_DET][MAX_UP][MAX_DOWN] = {{{0}}}; double fbgain[MAX_DET][MAX_BK][MAX_UP][MAX_DOWN]= {{{{0}}}};
bool fbgainValid[MAX_DET][MAX_UP][MAX_DOWN] = {{{false}}}; bool fbgainValid[MAX_DET][MAX_BK][MAX_UP][MAX_DOWN] = {{{{false}}}};
// std::map<int, TH2F *> updn2DHistos; // std::map<int, TH2F *> updn2DHistos;
std::map<int, double> upCorrFactor; std::map<int, double> upCorrFactor;
@ -242,7 +242,7 @@ void GainMatchSX3::Terminate()
{ {
// kv.first is a tuple of (id, up, bk) // kv.first is a tuple of (id, up, bk)
// kv.second is a vector of tuples (bkE, upE, dnE) // kv.second is a vector of tuples (bkE, upE, dnE)
auto [id, ud, bk] = kv.first; auto [id,bk,u,d] = kv.first;
const auto &pts = kv.second; const auto &pts = kv.second;
// Check if we have enough points for fitting // Check if we have enough points for fitting
if (pts.size() < 5) if (pts.size() < 5)
@ -263,8 +263,8 @@ void GainMatchSX3::Terminate()
// Fit the graph to a linear function // Fit the graph to a linear function
TF1 f("f", "[0]*x", 0, 16000); TF1 f("f", "[0]*x", 0, 16000);
g.Fit(&f, "QNR"); g.Fit(&f, "QNR");
gainArray[id][ud][bk] = f.GetParameter(0); gainArray[id][bk][u][d] = f.GetParameter(0);
gainValid[id][ud][bk] = true; gainValid[id][bk][u][d] = true;
} }
// Output results // Output results
@ -272,14 +272,18 @@ void GainMatchSX3::Terminate()
{ {
for (int bk = 0; bk < MAX_BK; ++bk) for (int bk = 0; bk < MAX_BK; ++bk)
{ {
for (int ud = 0; ud < MAX_UP; ++ud) for (int u = 0; u < MAX_UP; ++u)
{ {
if (gainValid[id][ud][bk]) for( int d = 0; d < MAX_DOWN; ++d)
{ {
outFile1 << id << " " << bk << " " << ud << " " << gainArray[id][ud][bk] << std::endl; // Check if the gain is valid for this detector, back, up, and down
printf("Gain match Det%d Up+Dn%d Back%d → %.4f \n", id, ud, bk, gainArray[id][ud][bk]); if (gainValid[id][bk][u][d])
{
outFile1 << id << " " << bk << " " << u <<" "<< d << " " << gainArray[id][u][d][bk] << std::endl;
printf("Gain match Det%d Up%dDn%d Back%d → %.4f \n", id, u,d, bk, gainArray[id][u][d][bk]);
} }
} }
}
} }
} }
@ -302,10 +306,10 @@ void GainMatchSX3::Terminate()
// Fill histograms // Fill histograms
for (const auto &kv : dataPoints) for (const auto &kv : dataPoints)
{ {
auto [id, ud, bk] = kv.first; auto [id, u,d, bk] = kv.first;
if (!gainValid[id][ud][bk]) if (!gainValid[id][u][d][bk])
continue; continue;
double gain = gainArray[id][ud][bk]; double gain = gainArray[id][u][d][bk];
// Prepare vectors to hold the points for TGraph // Prepare vectors to hold the points for TGraph
std::vector<double> xVals; std::vector<double> xVals;
@ -338,11 +342,11 @@ void GainMatchSX3::Terminate()
TGraph g2(xVals.size(), xVals.data(), yVals.data()); TGraph g2(xVals.size(), xVals.data(), yVals.data());
TF1 f1("f1", "[0]*x", 0, 16000); TF1 f1("f1", "[0]*x", 0, 16000);
g2.Fit(&f1, "QNR"); g2.Fit(&f1, "QNR");
fbgain[id][ud][bk] = f1.GetParameter(0); fbgain[id][u][d][bk] = f1.GetParameter(0);
fbgainValid[id][ud][bk] = true; fbgainValid[id][u][d][bk] = true;
// Optional: save the graph or the fit result if you want // Optional: save the graph or the fit result if you want
// g2.Write(Form("gFVB_id%d_U%d_B%d", id, ud, bk)); // g2.Write(Form("gFVB_id%d_U%d_B%d", id, ud, bk));
printf("Gain match Det%d Up+Dn%d Back%d → %.4f \n", id, ud, bk, fbgain[id][ud][bk]); printf("Gain match Det%d Up%d Dn%d Back%d → %.4f \n", id, u,d, bk, fbgain[id][u][d][bk]);
} }
} }
// Output results // Output results
@ -350,14 +354,18 @@ void GainMatchSX3::Terminate()
{ {
for (int bk = 0; bk < MAX_BK; ++bk) for (int bk = 0; bk < MAX_BK; ++bk)
{ {
for (int ud = 0; ud < MAX_UP; ++ud) for (int u = 0; u < MAX_UP; ++u)
{ {
if (fbgainValid[id][ud][bk]) for( int d = 0; d < MAX_DOWN; ++d)
{ {
outFile2 << id << " " << bk << " " << ud << " " << fbgain[id][ud][bk] << std::endl; // Check if the gain is valid for this detector, back, up, and down
printf("Gain match Det%d Up+Dn%d Back%d → %.4f \n", id, ud, bk, fbgain[id][ud][bk]); if (fbgainValid[id][u][d][bk])
{
outFile2 << id << " " << bk << " " << u<<" "<<d << " " << fbgain[id][u][d][bk] << std::endl;
printf("Gain match Det%d Up%d Dn%d Back%d → %.4f \n", id, u,d, bk, fbgain[id][u][d][bk]);
} }
} }
}
} }
} }
@ -369,8 +377,8 @@ void GainMatchSX3::Terminate()
for (const auto &kv : dataPoints) for (const auto &kv : dataPoints)
{ {
auto [id, ud, bk] = kv.first; auto [id, u,d, bk] = kv.first;
if (!fbgainValid[id][ud][bk]) if (!fbgainValid[id][u][d][bk])
continue; continue;
// double factor = fbgain[id][ud][bk]; // double factor = fbgain[id][ud][bk];
@ -379,8 +387,8 @@ void GainMatchSX3::Terminate()
double correctedBack, eBk, eUp, eDn; double correctedBack, eBk, eUp, eDn;
std::tie(eBk, eUp, eDn) = pr; std::tie(eBk, eUp, eDn) = pr;
correctedBack = eBk * gainArray[id][ud][bk]; correctedBack = eBk * gainArray[id][u][d][bk];
double eUpCorr = eUp * fbgain[id][ud][bk]; double eUpCorr = eUp * fbgain[id][u][d][bk];
double eDnCorr = eDn; double eDnCorr = eDn;
double eSumCorr = eUpCorr + eDnCorr; double eSumCorr = eUpCorr + eDnCorr;
if (correctedBack == 0 || eSumCorr == 0) if (correctedBack == 0 || eSumCorr == 0)