1 #include <framework/gearbox/Const.h>
2 #include <framework/logging/Logger.h>
3 #include <tracking/trackFindingVXD/displayInterfaceTF/CollectorTFInfo.h>
4 #include <gtest/gtest.h>
88 uint sector_map_size = 40;
91 std::map<std::pair<unsigned int, unsigned int>, std::vector<int>> sectors_display_all_pass;
92 std::vector<int> sectors_display_friends;
96 sectors_display_friends.clear();
97 sectors_display_all_pass.insert(std::make_pair(std::make_pair(
pass_sector_id_single, 0), sectors_display_friends));
99 sectors_display_friends.clear();
100 sectors_display_friends.push_back(0);
101 sectors_display_all_pass.insert(std::make_pair(std::make_pair(
pass_sector_id_single, 1), sectors_display_friends));
103 sectors_display_friends.clear();
104 sectors_display_friends.push_back(1);
105 sectors_display_all_pass.insert(std::make_pair(std::make_pair(
pass_sector_id_single, 2), sectors_display_friends));
108 sectors_display_friends.clear();
109 for (uint u = 3; u < sector_map_size; u++) {
110 sectors_display_all_pass.insert(std::make_pair(std::make_pair(
pass_sector_id_single, u), sectors_display_friends));
126 uint anz_clusters_svd = 15;
127 uint anz_clusters_pxd = 15;
130 for (uint i = 0; i < anz_clusters_svd; i++) {
138 for (uint i = 0; i < anz_clusters_pxd; i++) {
150 std::map<unsigned int, std::vector<int>> sector_map;
151 uint size_of_sectors = 10;
153 for (uint i = 0; i < size_of_sectors; i++) {
154 sector_map.insert(std::make_pair(i, std::vector<int>()));
159 std::map<std::pair<unsigned int, unsigned int>, std::vector<int>> sectors_display_all_pass;
160 std::vector<int> sectors_display_friends;
164 B2DEBUG(1,
"PassNr. " << i <<
"Size of Sector Map: " << sector_map.size());
166 for (
auto& akt_sector : sector_map) {
167 sectors_display_friends.clear();
170 for (
auto& akt_friend : akt_sector.second) {
171 sectors_display_friends.push_back(akt_friend);
173 sectors_display_all_pass.insert(std::make_pair(std::make_pair(i, akt_sector.first), sectors_display_friends));
184 uint anz_clusters_svd = 10;
185 uint anz_clusters_pxd = 10;
188 for (uint i = 0; i < anz_clusters_svd; i++) {
191 m_collector.importCluster(index,
"", -1, std::vector<int>(), std::vector<int>(), Const::SVD, i);
193 for (uint i = 0; i < anz_clusters_pxd; i++) {
196 m_collector.importCluster(index,
"", -1, std::vector<int>(), std::vector<int>(), Const::PXD, i);
268 for (uint i = 0; i < anz_hits; i++) {
273 m_collector.importHit(index,
"", -1, std::vector<int>(), std::vector<int>(), std::vector<int>(), 1, TVector3(),
305 std::vector<int> hits = {1, 2, 3};
308 for (uint i = 0; i < anz_cell; i++) {
311 m_collector.importCell(index,
"", -1, std::vector<int>(), std::vector<int>(), hits);
318 std::vector<std::pair<int, unsigned int>> cells0 = {std::make_pair(0, 0)};
319 std::vector<std::pair<int, unsigned int>> cells1 = {std::make_pair(1, 1)};
320 std::vector<std::pair<int, unsigned int>> cells23 = {std::make_pair(2, 2), std::make_pair(3, 3)};
321 std::vector<std::pair<int, unsigned int>> cells79 = {std::make_pair(7, 7), std::make_pair(9, 9)};
322 std::vector<std::pair<int, unsigned int>> cells8 = {std::make_pair(8, 8)};
334 std::vector<std::pair<int, unsigned int>>({}));
336 std::vector<std::pair<int, unsigned int>>({}));
338 std::vector<std::pair<int, unsigned int>>({}));
340 std::vector<std::pair<int, unsigned int>>({}));
342 std::vector<std::pair<int, unsigned int>>({}));
349 std::vector<std::pair<int, unsigned int>> cells = {std::make_pair(1, 1), std::make_pair(2, 2), std::make_pair(3, 3)};
352 for (uint i = 0; i < anz_tfc; i++) {
356 m_collector.importTC(index,
"", -1, std::vector<int>(), std::vector<int>(), cells);
365 for (uint i = 0; i <
m_collector.m_cellTF.size(); i++) {
366 B2INFO(
"* Cell ID: " << i <<
"; active: " <<
m_collector.m_cellTF[i].getActive() <<
", died_at: " <<
367 m_collector.m_cellTF[i].getDiedAt() <<
"; m_assigned_hits_ids - size: " <<
m_collector.m_cellTF[i].getAssignedHits().size() <<
368 "; Neighbours - size: " <<
m_collector.m_cellTF[i].getNeighbours().size() <<
"; State: " <<
m_collector.m_cellTF[i].getState() <<
369 "; use counter: " <<
m_collector.m_cellTF[i].getUseCounter());
377 for (uint i = 0; i <
m_collector.m_hitTF.size(); i++) {
378 B2INFO(
"* Hit ID: " << i <<
"; active: " <<
m_collector.m_hitTF[i].getActive() <<
", died_at: " <<
379 m_collector.m_hitTF[i].getDiedAt() <<
"; m_assigned_cluster - size: " <<
m_collector.m_hitTF[i].getAssignedCluster().size() <<
380 "; UseTC IDs - size: " <<
m_collector.m_hitTF[i].getUseCounterTCIDs().size() <<
381 "; use counter: " <<
m_collector.m_hitTF[i].getUseCounter() <<
"; SectorID: " <<
m_collector.m_hitTF[i].getSectorID());
388 for (uint i = 0; i <
m_collector.m_clustersTF.size(); i++) {
389 B2INFO(
"* Cluster ID: " << i <<
"; active: " <<
m_collector.m_clustersTF[i].getActive() <<
", died_at: " <<
390 m_collector.m_clustersTF[i].getDiedAt() <<
"; Real Cluster ID: " <<
391 m_collector.m_clustersTF[i].getRealClusterID() <<
"; Detector Type: "
392 <<
m_collector.m_clustersTF[i].getDetectorType() <<
"; use counter: " <<
m_collector.m_clustersTF[i].getUseCounter());
400 for (uint i = 0; i <
m_collector.m_tfCandTF.size(); i++) {
401 B2INFO(
"* TC ID: " << i <<
"; active: " <<
m_collector.m_tfCandTF[i].getActive() <<
", died_at: " <<
402 m_collector.m_tfCandTF[i].getDiedAt() <<
"; Own ID: " <<
403 m_collector.m_tfCandTF[i].getOwnID() <<
"; AssignedCells - size: " <<
404 m_collector.m_tfCandTF[i].getAssignedCell().size());
413 B2INFO(
"* Sector ID: " << akt_sector.second.getSectorID() <<
"; active: " << akt_sector.second.getActive() <<
", died_at: " <<
414 akt_sector.second.getDiedAt() <<
"; is only friend: " <<
415 akt_sector.second.getIsOnlyFriend() <<
"; Friends - size: "
416 << akt_sector.second.getFriends().size() <<
"; use counter: " << akt_sector.second.getUseCounter());
570 bool false_item =
false;
571 bool true_item =
true;
573 import_sectors_loop();
575 import_clusters_loop();
577 m_collector.updateSectors(4, 1,
"geloescht", 0, std::vector<int>(), std::vector<int>(), 0);
579 m_collector.updateClusters(3,
"geloescht", 0, std::vector<int>(), std::vector<int>(), 0);
581 ASSERT_EQ(
"geloescht", m_collector.m_clustersTF[3].getDiedAt());
582 ASSERT_EQ(false_item, m_collector.m_clustersTF[3].getActive());
586 m_collector.updateHit(1,
"geloescht", 0, std::vector<int>(), std::vector<int>(), -1, -1, std::vector<int>());
588 ASSERT_EQ(
"geloescht", m_collector.m_hitTF[1].getDiedAt());
589 ASSERT_EQ(false_item, m_collector.m_hitTF[1].getActive());
590 ASSERT_EQ(true_item, m_collector.m_hitTF[2].getActive());
593 m_collector.updateCell(3,
"geloescht", 2, std::vector<int>(), std::vector<int>(), -1, -1, 0, std::vector<int>());
595 ASSERT_EQ(
"geloescht", m_collector.m_cellTF[3].getDiedAt());
596 ASSERT_EQ(false_item, m_collector.m_cellTF[3].getActive());
597 ASSERT_EQ(true_item, m_collector.m_cellTF[1].getActive());
601 m_collector.updateTC(2,
"geloescht", 4, std::vector<int>(), std::vector<int>());
603 ASSERT_EQ(
"geloescht", m_collector.m_tfCandTF[2].getDiedAt());
604 ASSERT_EQ(false_item, m_collector.m_tfCandTF[2].getActive());
605 ASSERT_EQ(true_item, m_collector.m_tfCandTF[1].getActive());