Skip to content

Commit

Permalink
Merge pull request #155 from ChengWeiShih/main
Browse files Browse the repository at this point in the history 
THe INTTHitMapEvt now has the capability to only plot the hit map of …
  • Loading branch information
@ChengWeiShih
ChengWeiShih authored Dec 7, 2024
2 parents 9513a6b + 77cf48b commit cf4aced
Show file tree
Hide file tree
Showing 13 changed files with 1,070 additions and 17 deletions.
63 changes: 54 additions & 9 deletions general_codes/CWShih/INTTHitMapEvt/INTTHitMapEvt.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,8 @@ INTTHitMapEvt::INTTHitMapEvt(
const int bco_diff_peak_in,
const bool ApplyHitQA_in,
const bool clone_hit_remove_BCO_tag_in,
const bool setADCinZaxis_tag_in
const bool setADCinZaxis_tag_in,
const std::vector<int> adc_conversion_vec_in
):
SubsysReco(name),
process_id(process_id_in),
Expand All @@ -87,7 +88,8 @@ INTTHitMapEvt::INTTHitMapEvt(
bco_diff_peak(bco_diff_peak_in),
ApplyHitQA(ApplyHitQA_in),
clone_hit_remove_BCO_tag(clone_hit_remove_BCO_tag_in),
setADCinZaxis_tag(setADCinZaxis_tag_in)
setADCinZaxis_tag(setADCinZaxis_tag_in),
adc_conversion_vec(adc_conversion_vec_in)

{
std::cout << "INTTHitMapEvt::INTTHitMapEvt(const std::string &name) Calling ctor" << std::endl;
Expand Down Expand Up @@ -124,18 +126,54 @@ INTTHitMapEvt::INTTHitMapEvt(

PrepareHotChannel();

check_event_bcofull_map.clear();
check_event_bcofull_map_map.clear();

}

int INTTHitMapEvt::SetCheckEventBCOFULLsHalfLadder(std::vector<std::tuple<long long, int, int>> input_bcofull_halfladder_vec)
{
check_event_bcofull_map_map.clear();

for (auto &bcofull_halfladder : input_bcofull_halfladder_vec)
{
long long bco_full = std::get<0>(bcofull_halfladder);
int FELIX = std::get<1>(bcofull_halfladder);
int FELIX_ch = std::get<2>(bcofull_halfladder);

if (check_event_bcofull_map_map.find(bco_full) == check_event_bcofull_map_map.end())
{
check_event_bcofull_map_map.insert(
std::make_pair(
bco_full,
std::map<std::string, int>{ { Form("%d_%d",FELIX,FELIX_ch), 1 } }
)
);
}
else {
check_event_bcofull_map_map[bco_full].insert(
std::make_pair(
Form("%d_%d",FELIX,FELIX_ch), 1
)
);
}

}

halfladder_specified = true;

return Fun4AllReturnCodes::EVENT_OK;
}

int INTTHitMapEvt::SetCheckEventBCOFULLs(std::vector<long long> input_bcofull_vec)
{
check_event_bcofull_map.clear();
check_event_bcofull_map_map.clear();
for (auto &bcofull : input_bcofull_vec)
{
check_event_bcofull_map.insert(std::make_pair(bcofull,1));
check_event_bcofull_map_map.insert(std::make_pair(bcofull, std::map<std::string, int>{}));
}

halfladder_specified = false;

return Fun4AllReturnCodes::EVENT_OK;
}

Expand Down Expand Up @@ -198,10 +236,11 @@ int INTTHitMapEvt::process_event(PHCompositeNode *topNode)
InttRawHit* intthit = inttcont->get_hit(0);
intt_bco = intthit->get_bco();

if (check_event_bcofull_map.find(intt_bco) == check_event_bcofull_map.end())
if (check_event_bcofull_map_map.size() != 0 && check_event_bcofull_map_map.find(intt_bco) == check_event_bcofull_map_map.end())
{
return Fun4AllReturnCodes::EVENT_OK;
}


for (unsigned int i = 0; i < inttcont->get_nhits(); i++)
{
Expand Down Expand Up @@ -278,18 +317,24 @@ int INTTHitMapEvt::process_event(PHCompositeNode *topNode)
{
if (evt_inttHits_map.find(Form("%d_%d_%d_%d_%d",server,felix_ch,chip,channel,bco)) == evt_inttHits_map.end()) // note : if it's not found, we just add it
{
evt_inttHits_map[Form("%d_%d_%d_%d_%d",server,felix_ch,chip,channel,bco)] = {server,felix_ch,chip,channel,adc,bco, bco_diff};
evt_inttHits_map[Form("%d_%d_%d_%d_%d",server,felix_ch,chip,channel,bco)] = {server,felix_ch,chip,channel,adc_conversion_vec[adc],bco, bco_diff};
}
}
else // note : if we don't want to remove the clone hits
{
evt_inttHits_map[Form("%d",i)] = {server,felix_ch,chip,channel,adc,bco, bco_diff}; // note : only index i to make the key unique
evt_inttHits_map[Form("%d",i)] = {server,felix_ch,chip,channel,adc_conversion_vec[adc],bco, bco_diff}; // note : only index i to make the key unique
}

} // note : end of INTT raw hit loop

for (int felix_i = 0; felix_i < nFelix; felix_i++){
for (int fch_i = 0; fch_i < nFelix_channel; fch_i++){

if (check_event_bcofull_map_map[intt_bco].size() != 0 && check_event_bcofull_map_map[intt_bco].find(Form("%d_%d",felix_i,fch_i)) == check_event_bcofull_map_map[intt_bco].end())
{
continue;
}

h2_hitmap_map.insert(std::make_pair(Form("bcofull%ld_F%d_Fch%d",intt_bco,felix_i,fch_i), new TH2D(Form("bcofull%ld_F%d_Fch%d",intt_bco,felix_i,fch_i), Form("bcofull%ld_F%d_Fch%d;Chip ID (12 -> 0);Channel ID",intt_bco,felix_i,fch_i),13,0,13,265,-5,260)));
}
}
Expand All @@ -300,7 +345,7 @@ int INTTHitMapEvt::process_event(PHCompositeNode *topNode)

if(h2_hitmap_map.find(Form("bcofull%ld_F%d_Fch%d",intt_bco,hit.hit_server,hit.hit_felix_ch)) == h2_hitmap_map.end())
{
std::cout << "eID: "<< eID_count <<" INTTBcoResolution::PrepareINTT - hitmap not found" << std::endl;
if (!halfladder_specified) {std::cout << "eID: "<< eID_count <<" INTTBcoResolution::PrepareINTT - hitmap not found" << std::endl;}
continue;
}

Expand Down
12 changes: 9 additions & 3 deletions general_codes/CWShih/INTTHitMapEvt/INTTHitMapEvt.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,8 @@ class INTTHitMapEvt : public SubsysReco
const int bco_diff_peak_in = 0,
const bool ApplyHitQA_in = true,
const bool clone_hit_remove_BCO_tag_in = true,
const bool setADCinZaxis_tag_in = false
const bool setADCinZaxis_tag_in = false,
const std::vector<int> adc_conversion_vec_in = {35, 45, 60, 90, 120, 150, 180, 210}
);

~INTTHitMapEvt() override;
Expand Down Expand Up @@ -95,6 +96,8 @@ class INTTHitMapEvt : public SubsysReco
std::string GetOutputFileName() {return output_filename;}

int SetCheckEventBCOFULLs(std::vector<long long> input_bcofull_vec);

int SetCheckEventBCOFULLsHalfLadder(std::vector<std::tuple<long long, int, int>> input_bcofull_halfladder_vec); // note : {BCOFULL, FELIX, FELIX_ch}

private:
struct inttHitstr{
Expand All @@ -119,6 +122,7 @@ class INTTHitMapEvt : public SubsysReco
bool ApplyHitQA;
bool clone_hit_remove_BCO_tag;
bool setADCinZaxis_tag;
std::vector<int> adc_conversion_vec;

std::string output_filename;

Expand All @@ -133,6 +137,10 @@ class INTTHitMapEvt : public SubsysReco

long long eID_count;

// note : -------------------------------- for selecting the event ------------------------------
bool halfladder_specified;
std::map<long long, std::map<std::string, int>> check_event_bcofull_map_map;

// note : -------------------------------- The constant values ------------------------------
const int nFelix = 8;
const int Felix_offset = 3001;
Expand Down Expand Up @@ -165,8 +173,6 @@ class INTTHitMapEvt : public SubsysReco
std::map<std::string,std::string> hot_channel_map;
std::map<std::string, inttHitstr> evt_inttHits_map;

std::map<long long, int> check_event_bcofull_map;

};

#endif // INTTHITMAPEVT_H
12 changes: 9 additions & 3 deletions general_codes/CWShih/INTTHitMapEvt/build/install/include/intthitmapevt/INTTHitMapEvt.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,8 @@ class INTTHitMapEvt : public SubsysReco
const int bco_diff_peak_in = 0,
const bool ApplyHitQA_in = true,
const bool clone_hit_remove_BCO_tag_in = true,
const bool setADCinZaxis_tag_in = false
const bool setADCinZaxis_tag_in = false,
const std::vector<int> adc_conversion_vec_in = {35, 45, 60, 90, 120, 150, 180, 210}
);

~INTTHitMapEvt() override;
Expand Down Expand Up @@ -95,6 +96,8 @@ class INTTHitMapEvt : public SubsysReco
std::string GetOutputFileName() {return output_filename;}

int SetCheckEventBCOFULLs(std::vector<long long> input_bcofull_vec);

int SetCheckEventBCOFULLsHalfLadder(std::vector<std::tuple<long long, int, int>> input_bcofull_halfladder_vec); // note : {BCOFULL, FELIX, FELIX_ch}

private:
struct inttHitstr{
Expand All @@ -119,6 +122,7 @@ class INTTHitMapEvt : public SubsysReco
bool ApplyHitQA;
bool clone_hit_remove_BCO_tag;
bool setADCinZaxis_tag;
std::vector<int> adc_conversion_vec;

std::string output_filename;

Expand All @@ -133,6 +137,10 @@ class INTTHitMapEvt : public SubsysReco

long long eID_count;

// note : -------------------------------- for selecting the event ------------------------------
bool halfladder_specified;
std::map<long long, std::map<std::string, int>> check_event_bcofull_map_map;

// note : -------------------------------- The constant values ------------------------------
const int nFelix = 8;
const int Felix_offset = 3001;
Expand Down Expand Up @@ -165,8 +173,6 @@ class INTTHitMapEvt : public SubsysReco
std::map<std::string,std::string> hot_channel_map;
std::map<std::string, inttHitstr> evt_inttHits_map;

std::map<long long, int> check_event_bcofull_map;

};

#endif // INTTHITMAPEVT_H
Binary file modified general_codes/CWShih/INTTHitMapEvt/build/install/lib/libINTTHitMapEvt.so.0.0.0
View file
Binary file not shown.
Loading

0 comments on commit cf4aced

Please sign in to comment.