12 #include "GaudiKernel/IEventProcessor.h"
32 while (std::getline(
ss,
line,
'\n'))
60 return StatusCode::SUCCESS;
74 IEventProcessor* appMgr=
nullptr;
75 ATH_CHECK(service(
"ApplicationMgr",appMgr));
77 ATH_MSG_ERROR(
"Failed to retrieve ApplicationMgr as IEventProcessor");
78 return StatusCode::FAILURE;
80 return appMgr->stopRun();
86 return StatusCode::SUCCESS;
101 ATH_MSG_INFO(
"Cannot read more events from file, returning");
102 return StatusCode::SUCCESS;
111 if (hit.getBarcodePt() == 0)
continue;
115 int lyr = hit.getPhysLayer();
116 int eta = hit.getEtaModule();
117 int phi = hit.getPhiModule();
119 if (hit.isPixel() && hit.isBarrel()) {
123 else if (hit.isPixel() && !hit.isBarrel()) {
126 eta = hit.getEtaModule();
127 lyr = hit.getPhysLayer();
133 else if (!hit.isPixel() && hit.isBarrel()) {
137 else if (!hit.isPixel() && !hit.isBarrel()) {
159 m_modules.insert(std::pair<FPGATrackSimModuleId, Module>(
mod.moduleId(),
mod));
168 return StatusCode::SUCCESS;
196 m_pmap <<
"! silicon endCap physDisk physLayer ['stereo' stripSide <strip only>] 'plane1' logiLayer1 'plane2' logiLayer2\n";
200 for (
int lyr = 0; lyr <=
m_pbmax; lyr++) {
203 m_pmap <<
"pixel 0 -1 " << lyr <<
" plane1 " << p1 <<
" plane2 " << p2 <<
"\n";
205 for (
int lyr = 0; lyr <=
m_pemax[0]; lyr++) {
208 m_pmap <<
"pixel 1 " << lyr <<
" " << lyr <<
" plane1 " << p1 <<
" plane2 " << p2 <<
"\n";
210 for (
int lyr = 0; lyr <=
m_pemax[1]; lyr++) {
213 m_pmap <<
"pixel 2 " << lyr <<
" " << lyr <<
" plane1 " << p1 <<
" plane2 " << p2 <<
"\n";
215 for (
int lyr = 0; lyr <=
m_sbmax; lyr++) {
218 m_pmap <<
"SCT 0 -1 " << lyr <<
" stereo " << lyr % 2 <<
" plane1 " << p1 <<
" plane2 " << p2 <<
"\n";
220 for (
int lyr = 0; lyr <=
m_semax[0]; lyr++) {
223 m_pmap <<
"SCT 1 " << lyr/2 <<
" " << lyr <<
" stereo " << lyr % 2 <<
" plane1 " << p1 <<
" plane2 " << p2 <<
"\n";
225 for (
int lyr = 0; lyr <=
m_semax[1]; lyr++) {
228 m_pmap <<
"SCT 2 " << lyr/2 <<
" " << lyr <<
" stereo " << lyr % 2 <<
" plane1 " << p1 <<
" plane2 " << p2 <<
"\n";
235 m_rmap <<
"towers 1 phi 16\n\n0\n";
249 return StatusCode::SUCCESS;
257 std::set<int> key_etamods;
259 float total_hits = 0;
261 total_hits +=
etamod.second;
263 ATH_MSG_INFO(
"Found " << total_hits <<
" hits in the key layer, applying global trim factor of " <<
m_globalTrim <<
"%");
268 key_etamods.insert(
etamod.first);
270 ATH_MSG_INFO(
"Eta module " <<
etamod.first <<
" only contains " <<
etamod.second <<
" out of " << total_hits <<
" hits, excluding from slices.");
276 std::vector<std::vector<int>> key_modules_for_slices;
279 std::vector<int> key_etas;
280 std::vector<int> key_etas2;
281 key_etas.insert(key_etas.end(), key_etamods.begin(), key_etamods.end());
283 for (
unsigned i = 0;
i < key_etas.size();
i++)
285 if (
i >= (key_modules_for_slices.size() * etasPerSlice)) key_modules_for_slices.push_back(std::vector<int>());
286 key_modules_for_slices.back().push_back(key_etas[
i]);
289 std::map<int, int> keymod2slice;
290 for (
unsigned s = 0;
s < key_modules_for_slices.size();
s++)
291 for (
unsigned e = 0;
e < key_modules_for_slices[
s].size();
e++)
292 keymod2slice[key_modules_for_slices[
s][
e]] =
s;
305 std::stringstream eta_slices;
306 for (
unsigned s = 0;
s < key_modules_for_slices.size();
s++){
307 for (
unsigned e = 0;
e < key_modules_for_slices[
s].size();
e++){
308 eta_slices << key_modules_for_slices[
s][
e] <<
" ";
319 std::vector<std::vector<int>> key_modules_for_slices2;
322 for (
unsigned i = 0;
i < key_etas2.size();
i++)
324 if (
i >= (key_modules_for_slices2.size() * etasPerSlice2)) key_modules_for_slices2.push_back(std::vector<int>());
325 key_modules_for_slices2.back().push_back(key_etas2[
i]);
328 std::map<int, int> keymod2slice2;
329 for (
unsigned s = 0;
s < key_modules_for_slices2.size();
s++)
330 for (
unsigned e = 0;
e < key_modules_for_slices2[
s].size();
e++)
331 keymod2slice2[key_modules_for_slices2[
s][
e]] =
s;
343 for (
auto&
m: pair.second)
346 if (keymod2slice[
m->eta] ==
s1)
key1 =
true;
349 if (keymod2slice2[
m->eta] ==
s2)
key2 =
true;
357 if (newSlice + 1 > new_nSlice) new_nSlice = newSlice + 1;
363 ATH_MSG_INFO(
"These slices were further divided based on the key layer 2: '" <<
key2 <<
"'. Now nSlices = " <<
m_nSlices.value());
374 if (
isOnKeyLayer(1,hit.getDetType(),hit.getDetectorZone(), hit.getPhysLayer()))
376 if (keymod2slice.count(hit.getEtaModule()) > 0) {
377 int s = keymod2slice[hit.getEtaModule()];
386 std::stringstream
trim;
387 std::stringstream gtrim;
388 trim << std::fixed << std::setprecision(3) <<
m_trim;
389 gtrim << std::fixed << std::setprecision(3) <<
m_globalTrim;
390 std::string str_trim =
trim.str();
391 std::string str_gtrim = gtrim.str();
392 int dot = str_trim.find_last_of(
".");
393 str_trim.replace(
dot,1,
"p");
394 str_gtrim.replace(str_gtrim.find_last_of(
"."), 1,
"p");
406 if (
m->numTracks.empty()) {
415 std::vector<std::vector<int>> slicedTracks (
m_nSlices.value());
418 slicedTracks[
s].push_back(trk);
432 for (
auto trk : slicedTracks[
s]) {
452 ATH_MSG_INFO(
"Trimmed off " << trimmed <<
" modules that were hit by less than " <<
m_trim <<
"% of tracks");
470 return StatusCode::SUCCESS;
475 std::stringstream
trim;
476 std::stringstream gtrim;
477 trim << std::fixed << std::setprecision(3) <<
m_trim;
478 gtrim << std::fixed << std::setprecision(3) <<
m_globalTrim;
479 std::string str_trim =
trim.str();
480 std::string str_gtrim = gtrim.str();
481 int dot = str_trim.find_last_of(
".");
482 str_trim.replace(
dot,1,
"p");
483 str_gtrim.replace(str_gtrim.find_last_of(
"."), 1,
"p");
485 std::string slicingType =
"";
486 if (
m_key2) slicingType =
"2D";
489 ATH_MSG_INFO(
"Creating eta patterns file: " << etapat_path);
494 for (
auto&
m: pair.second)
508 std::stringstream track_etapatts;
509 unsigned planesDone = 0;
513 if (
m->plane ==
static_cast<int>(
p))
515 track_etapatts << std::to_string(static_cast<int>(
m->det)) <<
"\t" <<
std::to_string(
static_cast<int>(
m->bec)) <<
"\t" <<
std::to_string(
m->eta) <<
"\t\t";
522 m_etapat << track_etapatts.str() <<
"\n";
526 return StatusCode::SUCCESS;
533 for (
const auto& hit: allHits)
537 int lyr = hit.getPhysLayer();
564 m_radfile << std::setprecision(3) << std::fixed <<
avg <<
" ";
586 m_radfile << std::setprecision(3) << std::fixed <<
avg <<
" ";
595 return StatusCode::SUCCESS;
602 for (
const auto& hit: allHits)
606 int lyr = hit.getPhysLayer();
617 m_z[
slice][plane].push_back(hit.getZ());
631 float median = (minZ + maxZ)/2;
646 bool doneInitial =
false;
663 float median = (minZ + maxZ)/2;
674 return StatusCode::SUCCESS;
690 return StatusCode::SUCCESS;
701 std::vector<TH2F*> h_slicemap;
702 char *
hname =
new char[20];
706 sprintf(
hname,
"rz_slice%u",
i);
710 h_slicemap.push_back(
h);
713 for (
const auto& hit: allHits)
717 h_slicemap[
s]->Fill(hit.getZ(),hit.getR());
721 h_slicemap[
i]->Write();
728 int det =
static_cast<int>(t_det);
729 int bec =
static_cast<int>(t_bec);
744 for (
auto& plane : planes) {
745 for (
auto&
layer : plane) {
755 std::stringstream rmap_line;
756 std::set<int>
etas, phis;
758 for(
int lyr = 0; lyr <=
max; lyr++)
762 for (
const auto& hit:
hits)
764 if(
static_cast<int>(hit.getPhysLayer()) == lyr && hit.getDetectorZone() ==
bec)
766 etas.insert(hit.getEtaModule());
767 phis.insert(hit.getPhiModule());
770 if (
etas.size() != 0) rmap_line << static_cast<int>(
det) <<
" " <<
static_cast<int>(
bec) <<
" " << lyr <<
" " << *phis.begin() <<
" " << *phis.rbegin() <<
" " << phis.size() <<
" " << *
etas.begin() <<
" " << *
etas.rbegin() <<
" " <<
etas.size() <<
"\n";
771 else rmap_line << static_cast<int>(
det) <<
" " <<
static_cast<int>(
bec) <<
" " << lyr <<
" 0 0 0 0 0 0\n";
775 return rmap_line.str();
783 std::string
det,
bec, lyr;
784 std::map <std::string, std::vector<std::string>> abrevs = { {
"pb",{
"pixel",
"barrel"}}, {
"pe",{
"pixel",
"endcap"}}, {
"sb",{
"strip",
"barrel"}}, {
"se",{
"strip",
"endcap"}} };
798 ATH_MSG_ERROR(
"Invalid KeyString: '" <<
m_keystring <<
"'." <<
"Accepted formats are 'strip,posEndcap,2', 'pixel,barrel,3', or 'plane 0'");
806 std::string plane =
s.substr(0,
s.find(
delimiter));
808 for (
unsigned i = 0;
i <
s.size();
i++){
809 std::string
reg =
s[
i].substr(0, 2);
810 std::vector<std::string>
zone = abrevs[
reg];
811 if (
s[
i].back() ==
'+')
zone[1] =
"posEndcap";
812 if (
s[
i].back() ==
'-')
zone[1] =
"negEndcap";
815 std::string lyr =
s[
i].substr(2);
821 ATH_MSG_ERROR(
"Invalid KeyString: '" <<
m_keystring <<
"'." <<
"Accepted formats are 'strip,posEndcap,2', 'pixel,barrel,3', or 'plane 0'");
843 ATH_MSG_ERROR(
"Invalid KeyString2: '" <<
m_keystring2 <<
"'." <<
"Accepted formats are 'strip,posEndcap,2', 'pixel,barrel,3', or 'plane 0'");
851 std::string plane =
s.substr(0,
s.find(
delimiter));
853 for (
unsigned i = 0;
i <
s.size();
i++){
854 std::string
reg =
s[
i].substr(0, 2);
855 std::vector<std::string>
zone = abrevs[
reg];
856 if (
s[
i].back() ==
'+')
zone[1] =
"posEndcap";
857 if (
s[
i].back() ==
'-')
zone[1] =
"negEndcap";
860 std::string lyr =
s[
i].substr(2);
866 ATH_MSG_ERROR(
"Invalid KeyString2: '" <<
m_keystring2 <<
"'." <<
"Accepted formats are 'strip,posEndcap,2', 'pixel,barrel,3', or 'plane 0'");
874 std::stringstream subrmap_line;
875 std::set<int>
etas, phis;
877 std::vector<Module*> mods;
878 for (
auto*
mod: allmods)
881 for(
int lyr = 0; lyr <=
max; lyr++)
890 phis.insert(
mod->phi);
893 if (
etas.size() != 0) subrmap_line << static_cast<int>(
det) <<
" " <<
static_cast<int>(
bec) <<
" " << lyr <<
" " << *phis.begin() <<
" " << *phis.rbegin() <<
" " << phis.size() <<
" " << *
etas.begin() <<
" " << *
etas.rbegin() <<
" " <<
etas.size() <<
"\n";
894 else subrmap_line << static_cast<int>(
det) <<
" " <<
static_cast<int>(
bec) <<
" " << lyr <<
" 0 0 0 0 0 0\n";
898 return subrmap_line.str();