FPGATrackSimRoad.cxx

Go to the documentation of this file.

/*
 Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#include <vector>
 
#include "FPGATrackSimObjects/FPGATrackSimRoad.h"
#include "FPGATrackSimObjects/FPGATrackSimConstants.h"
 
using namespace std;
 
//FPGATrackSimRoad::~FPGATrackSimRoad() { }
 
size_t FPGATrackSimRoad::getNHits() const
{
    size_t n = 0;
    for (const auto& l : m_hits_trans) n += l.size();
    return n;
}
 
std::vector<size_t> FPGATrackSimRoad::getNHits_layer() const
{
    std::vector<size_t> out;
    out.reserve(m_hits_trans.size());
    for (const auto& l : m_hits_trans) out.push_back(l.size());
    return out;
}
 
size_t FPGATrackSimRoad::getNHitCombos() const
{
    size_t combos = 1;
    size_t l = m_hits_trans.size();
    std::vector<unsigned int> skipindices;
 
    // figure out indices of outer layers of stereo layers, where inner layer is also present to not
    // double count spacepoints
    for (size_t i = 1; i < l; ++i) {
        if (m_hits_trans[i - 1].size() > 0 && m_hits_trans[i].size() > 0) {
            if ((m_hits_trans[i - 1][0]->isStrip() && m_hits_trans[i][0]->isStrip()) &&
                (m_hits_trans[i - 1][0]->getPhysLayer() % 2 == 0) &&
                (m_hits_trans[i][0]->getPhysLayer() - m_hits_trans[i - 1][0]->getPhysLayer() == 1)) {
                skipindices.push_back(i);
                ++i;
            }
        }
    }
 
    // calculate number of combinations
    for (size_t i = 0; i < l; ++i) {
        size_t n = 0;
        for (size_t j = 0; j < m_hits_trans[i].size(); ++j) {
            // extra handling of spacepoints in combination calculation
            if (m_hits_trans[i][j]->getHitType() == HitType::spacepoint) {
                bool skip = false;
                size_t skipsize = skipindices.size();
                for (size_t k = 0; k < skipsize; ++k)
                    if (i == skipindices[k]) skip = true;
 
                if (skip) continue;
            }
            ++n;
        }
        if (n > 0) combos *= n;
    }
 
    return combos;
}
 
 
FPGATrackSimMultiTruth FPGATrackSimRoad::getTruth() const
{
    // get number of pixel layers from hits. Assumes strips follow pixels,
    // and all hits in a layer share the same det type
    unsigned nPixel;
    for (nPixel = 0; nPixel < m_hits_trans.size(); nPixel++)
        if (!m_hits_trans[nPixel].empty() && m_hits_trans[nPixel].front()->isStrip())
            break;
 
    std::map<FPGATrackSimMultiTruth::Barcode, layer_bitmask_t> layer_map;
 
    for (auto const& hits : m_hits_trans)
        for (auto const &h : hits)
            for (auto const& x : h->getTruth())
                layer_map[x.first] |= (1 << h->getLayer());
 
    FPGATrackSimMultiTruth mt;
    for (auto const& x : layer_map)
    {
        int w = 0, n = 0;
        for (unsigned i = 0; i < m_hits_trans.size(); i++)
        {
            if (!m_hits_trans[i].empty())
                n += (i < nPixel) ? 2 : 1; // double weight pixels
            if (x.second & (1 << i))
                w += (i < nPixel) ? 2 : 1; // double weight pixels
        }
        if (n == 0){
          throw std::range_error("divide by zero in FPGATrackSimRoad::getTruth");
        }
        mt.add(x.first, static_cast<float>(w) / n);
    }
 
    return mt;
}
 
 
std::unordered_set<std::shared_ptr<const FPGATrackSimHit>> FPGATrackSimRoad::getHits_flat() const {
    std::unordered_set<std::shared_ptr<const FPGATrackSimHit>> hits;
    for (const auto& layerHits : m_hits_trans)
        for (auto const& hit : layerHits)
            hits.insert(hit);
            // for (const auto& x : m_hits) hits.insert(x.begin(), x.end());
    return hits;
}
 
void FPGATrackSimRoad::repopulateTransHits() {  // this is needed if trying to read the hits from the road in a stored output file, call this first, otherwise not in Athena
    m_hits_trans.resize(m_hits.size());
    for (unsigned ilayer = 0; ilayer < m_hits.size(); ilayer++) {
       m_hits_trans[ilayer].resize(m_hits[ilayer].size());
       for (unsigned ihit = 0; ihit < m_hits[ilayer].size(); ihit++) {
          m_hits_trans[ilayer][ihit] = std::make_shared<const FPGATrackSimHit>(m_hits[ilayer][ihit]);
       }
    }
}
 
void FPGATrackSimRoad::setHits(unsigned layer, std::vector<std::shared_ptr<const FPGATrackSimHit>> && hits) {
    m_hits_trans[layer] = std::move(hits);
    m_hits[layer].clear();
    for (const auto& hit : m_hits_trans[layer])
        m_hits[layer].push_back(*hit);
} // ensure setNLayers is called first
 
void FPGATrackSimRoad::setHits(std::vector<std::vector<std::shared_ptr<const FPGATrackSimHit>>> &&hits){
    if (hits.size() != m_hits_trans.size()) setNLayers(hits.size());
    for (unsigned i = 0;i < hits.size();++i)
        setHits(i,std::move(hits[i]));
}
 
 
ostream& operator<<(ostream& os, const FPGATrackSimRoad& road)
{
    os << "road " << road.m_roadID
        << ": PID " << road.m_pid
        << ", sector " << road.m_sector
        << " hitLayers " << std::showbase << std::hex << road.m_hit_layers
        << " wcLayers " << std::showbase << std::hex << road.m_wildcard_layers
        << ", nHits " << road.getNHits();
    /*
    for (size_t l=0; l < road.m_hits.size(); l++)
    {
        for (FPGATrackSimHit const * hit : road.m_hits[l])
            os << "\n\t" << *hit;
    }
    */
 
    return os;
}