tof/RecoSetupUnit.cxx

Go to the documentation of this file.

/* Copyright (C) 2025 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer] */

 
#include "algo/detectors/tof/RecoSetupUnit.h"
 
#include <numeric>
#include <unordered_set>
 
using cbm::algo::tof::RecoSetupUnit;
using cbm::algo::tof::TrkStationIdMap;
 
// ---------------------------------------------------------------------------------------------------------------------
//
TrkStationIdMap::TrkStationIdMap(std::vector<SmTypeInfo>&& smTypeRpcMap, std::vector<int>&& trkStationIdMap)
  : fSmTypeRpcMap(std::move(smTypeRpcMap))
  , fTrkStationIdMap(std::move(trkStationIdMap))
{
  // Validate consistency
  // - a difference in offsets with the next SmType must be equal to nSm * nRpc
  // - including the "end" SmTypeInfo, which has an offset == fTrkStationId.size()
  auto ValidateConsSmTypes = [](SmTypeInfo thisSmType, SmTypeInfo nextSmType, uint16_t iThisSmType) {
    uint32_t offsetDiff = nextSmType.offset - thisSmType.offset;
    uint32_t nRpcsSmt   = thisSmType.nSm * thisSmType.nRpc;
    if (nRpcsSmt != offsetDiff) {
      std::stringstream msg;
      msg << "Mapping issue for iSmType=" << iThisSmType << ": num of RPCs ((nSm=)" << thisSmType.nSm << " x (nRpc=)"
          << thisSmType.nRpc << " = " << nRpcsSmt << ") differs from the offset difference relative to the next "
          << "supermodule type ((next.offset=)" << nextSmType.offset << " - (this.offset=)" << thisSmType.offset
          << " = " << offsetDiff;
      throw std::runtime_error(msg.str());
    }
  };
 
  SmTypeInfo thisSmt = fSmTypeRpcMap[0];
  for (uint16_t iSmType = 0; iSmType < GetNofSmTypes() - 1; ++iSmType) {
    SmTypeInfo nextSmt{fSmTypeRpcMap[iSmType + 1]};
    ValidateConsSmTypes(thisSmt, nextSmt, iSmType);
    thisSmt = nextSmt;
  }
  ValidateConsSmTypes(thisSmt,
                      SmTypeInfo{.offset = static_cast<uint16_t>(fTrkStationIdMap.size()), .nSm = 0, .nRpc = 0},
                      GetNofSmTypes() - 1);
}
 
 
// ---------------------------------------------------------------------------------------------------------------------
//
RecoSetupUnit::RecoSetupUnit(std::pair<std::vector<GeoVolume>, std::vector<GeoVolume>>&& stationVolumes,
                             TrkStationIdMap&& trkStationIdMap)
  : TrackingSetupIfs<RecoSetupUnit>(std::move(stationVolumes))
  , fTrkStationIdMap(std::move(trkStationIdMap))
{
}
 
// ---------------------------------------------------------------------------------------------------------------------
//
bool RecoSetupUnit::IsInitialized() const
{
  // TOF tracking detector is initialized, if:
  // - the fvTrackingStationId contain as much stations, as defined in stationFullVolumes
 
  // Collect all knownStationIds (> -1)
  std::unordered_set<int> knownStationIds;
  for (int stationId : fTrkStationIdMap.GetTrkStationIdMap()) {
    if (stationId > -1) knownStationIds.insert(stationId);
  }
 
  bool res{true};
  for (int stationId : knownStationIds) {
    if (stationId > GetNofTrackingStations()) {
      LOG(error) << "\t - found stationID (" << stationId << ") greater than max. stationID provided by geometry ("
                 << (GetNofTrackingStations() - 1) << ')';
      res = false;
    }
  }
 
  if (knownStationIds.size() != static_cast<size_t>(GetNofTrackingStations())) {
    LOG(error) << "\t - number of tracking stations provided by geometry (" << GetNofTrackingStations()
               << ") differs from one retrieved from digitization parameters (" << knownStationIds.size() << ')';
    res = false;
  }
 
  return res;
}