CbmGeoHelper.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 "CbmGeoHelper.h"
 
#include <Logger.h>
 
#include <TGeoManager.h>
#include <TGeoNode.h>
#include <TString.h>
 
#include <iomanip>
#include <limits>
#include <regex>
#include <set>
#include <string>
 
using cbm::GeoHelper;
 
// ---------------------------------------------------------------------------------------------------------------------
//
std::vector<TString> GeoHelper::CollectNodes(const TString& detector, const std::regex& component, const TString& path,
                                             TGeoNode* pNode)
{
  std::vector<TString> vPaths;
 
  if (!pNode) {
    return vPaths;
  }
 
  TString nodePath = path + (path.IsNull() ? "" : "/") + pNode->GetName();
  std::string nodeName{pNode->GetName()};
  std::smatch sMatch;
  if (std::regex_search(nodeName, sMatch, component)) {
    if (nodePath.Contains(detector, TString::kIgnoreCase)) {
      vPaths.push_back(nodePath);
    }
  }
 
  for (int iNode = 0; iNode < pNode->GetNdaughters(); ++iNode) {
    TGeoNode* pDaughter = pNode->GetDaughter(iNode);
    auto daughterPaths  = CollectNodes(detector, component, nodePath, pDaughter);
    vPaths.insert(vPaths.end(), daughterPaths.begin(), daughterPaths.end());
  }
 
  return vPaths;
}