Skip to content

Commit

Permalink
WIP
Browse files Browse the repository at this point in the history
  • Loading branch information
yongbinfeng committed Sep 16, 2024
1 parent 2524ff2 commit fcf6375
Show file tree
Hide file tree
Showing 4 changed files with 83 additions and 2 deletions.
2 changes: 2 additions & 0 deletions setup.sh
Original file line number Diff line number Diff line change
@@ -1 +1,3 @@
source /cvmfs/sft.cern.ch/lcg/views/LCG_106/x86_64-el9-gcc13-dbg/setup.sh
export CC=gcc
export CXX=g++
1 change: 0 additions & 1 deletion sim/exampleB4b.cc
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,6 @@ int main(int argc, char **argv)
CaloTree *histo = new CaloTree(macro, argc, argv);

G4UIExecutive *ui = nullptr;
;
if (!batchJob)
{
ui = new G4UIExecutive(argc, argv);
Expand Down
22 changes: 21 additions & 1 deletion sim/include/CaloTree.h
Original file line number Diff line number Diff line change
Expand Up @@ -133,11 +133,31 @@ class CaloTree

int m_nhits3dSS;
vector<int> m_id3dSS; // channel ID xxxyyyzzz
vector<int> m_tkey3dSS; // T-slice
vector<int> m_zkey3dSS; // Z-slice
vector<int> m_type3dSS;
vector<int> m_area3dSS;
vector<int> m_ix3dSS;
vector<int> m_iy3dSS;
vector<int> m_ixx3dSS;
vector<int> m_iyy3dSS;
vector<int> m_zslice3dSS;
vector<int> m_tslice3dSS;
vector<float> m_ph3dSS; // number of photons
float m_sum3dSS;

int m_nhits3dCC;
vector<int> m_id3dCC; // chnanel ID xxxyyyttt
vector<int> m_id3dCC; // chnanel ID xxxyyyttt
vector<int> m_tkey3dCC;
vector<int> m_zkey3dCC;
vector<int> m_type3dCC;
vector<int> m_area3dCC;
vector<int> m_ix3dCC;
vector<int> m_iy3dCC;
vector<int> m_ixx3dCC;
vector<int> m_iyy3dCC;
vector<int> m_zslice3dCC;
vector<int> m_tslice3dCC;
vector<float> m_ph3dCC; // number of photons
float m_sum3dCC;
};
Expand Down
60 changes: 60 additions & 0 deletions sim/src/CaloTree.cc
Original file line number Diff line number Diff line change
Expand Up @@ -157,11 +157,31 @@ CaloTree::CaloTree(string macFileName, int argc, char **argv)

tree->Branch("nhits3dSS", &m_nhits3dSS);
tree->Branch("id3dSS", &m_id3dSS);
tree->Branch("tkey3dSS", &m_tkey3dSS);
tree->Branch("zkey3dSS", &m_zkey3dSS);
tree->Branch("type3dSS", &m_type3dSS);
tree->Branch("area3dSS", &m_area3dSS);
tree->Branch("ix3dSS", &m_ix3dSS);
tree->Branch("iy3dSS", &m_iy3dSS);
tree->Branch("ixx3dSS", &m_ixx3dSS);
tree->Branch("iyy3dSS", &m_iyy3dSS);
tree->Branch("zslice3dSS", &m_zslice3dSS);
tree->Branch("tslice3dSS", &m_tslice3dSS);
tree->Branch("ph3dSS", &m_ph3dSS);
tree->Branch("sum3dSS", &m_sum3dSS);

tree->Branch("nhits3dCC", &m_nhits3dCC);
tree->Branch("id3dCC", &m_id3dCC);
tree->Branch("tkey3dCC", &m_tkey3dCC);
tree->Branch("zkey3dCC", &m_zkey3dCC);
tree->Branch("type3dCC", &m_type3dCC);
tree->Branch("area3dCC", &m_area3dCC);
tree->Branch("ix3dCC", &m_ix3dCC);
tree->Branch("iy3dCC", &m_iy3dCC);
tree->Branch("ixx3dCC", &m_ixx3dCC);
tree->Branch("iyy3dCC", &m_iyy3dCC);
tree->Branch("zslice3dCC", &m_zslice3dCC);
tree->Branch("tslice3dCC", &m_tslice3dCC);
tree->Branch("ph3dCC", &m_ph3dCC);
tree->Branch("sum3dCC", &m_sum3dCC);
}
Expand Down Expand Up @@ -235,6 +255,16 @@ void CaloTree::EndEvent()
ky = ky + id.iyy() + 1;
} // 3mm SiPM
m_id3dCC.push_back(id.ix() * 10000000 + ky * 1000 + id.tslice());
m_tkey3dCC.push_back(id.getTkey());
m_zkey3dCC.push_back(id.getZkey());
m_type3dCC.push_back(id.type());
m_area3dCC.push_back(id.area());
m_ix3dCC.push_back(id.ix());
m_iy3dCC.push_back(id.iy());
m_ixx3dCC.push_back(id.ixx());
m_iyy3dCC.push_back(id.iyy());
m_zslice3dCC.push_back(id.zslice());
m_tslice3dCC.push_back(id.tslice());
m_ph3dCC.push_back(round(ncer));
}
m_nhits3dCC = m_ph3dCC.size();
Expand All @@ -257,6 +287,16 @@ void CaloTree::EndEvent()
ky = ky + id.iyy() + 1;
} // 3mm SiPM
m_id3dSS.push_back(id.ix() * 10000000 + ky * 1000 + id.tslice());
m_tkey3dSS.push_back(id.getTkey());
m_zkey3dSS.push_back(id.getZkey());
m_type3dSS.push_back(id.type());
m_area3dSS.push_back(id.area());
m_ix3dSS.push_back(id.ix());
m_iy3dSS.push_back(id.iy());
m_ixx3dSS.push_back(id.ixx());
m_iyy3dSS.push_back(id.iyy());
m_zslice3dSS.push_back(id.zslice());
m_tslice3dSS.push_back(id.tslice());
m_ph3dSS.push_back(edepbirk);
}
m_nhits3dSS = m_ph3dSS.size();
Expand Down Expand Up @@ -314,12 +354,32 @@ void CaloTree::clearCaloTree()

m_nhits3dSS = 0;
m_id3dSS.clear();
m_tkey3dSS.clear();
m_zkey3dSS.clear();
m_type3dSS.clear();
m_area3dSS.clear();
m_ix3dSS.clear();
m_iy3dSS.clear();
m_ixx3dSS.clear();
m_iyy3dSS.clear();
m_zslice3dSS.clear();
m_tslice3dSS.clear();
m_ph3dSS.clear();
m_sum3dSS = 0.0;

m_nhits3dCC = 0;
// m_ky3dCC.clear(); // this used for debugging.
m_id3dCC.clear();
m_tkey3dCC.clear();
m_zkey3dCC.clear();
m_type3dCC.clear();
m_area3dCC.clear();
m_ix3dCC.clear();
m_iy3dCC.clear();
m_ixx3dCC.clear();
m_iyy3dCC.clear();
m_zslice3dCC.clear();
m_tslice3dCC.clear();
m_ph3dCC.clear();
m_sum3dCC = 0.0;
}
Expand Down

0 comments on commit fcf6375

Please sign in to comment.