CbmTrdParSetGeo.cxx

Go to the documentation of this file.

#include "CbmTrdParSetGeo.h"
 
#include "CbmTrdGeoHandler.h"  // for CbmTrdGeoHandler
#include "CbmTrdParModGeo.h"   // for CbmTrdParModGeo
 
#include <FairLogger.h>  // for LOG, Logger
 
#include <TGenericClassInfo.h>  // for TGenericClassInfo
#include <TGeoManager.h>        // for TGeoManager, gGeoManager
#include <TGeoNode.h>           // for TGeoNode
#include <TObjArray.h>          // for TObjArray
#include <TString.h>            // for operator+, TString, Form, TString::kI...
 
#include <map>      // for map, map<>::const_iterator, operator!=
#include <stdio.h>  // for printf
#include <utility>  // for pair
 
//________________________________________________________________________________________
CbmTrdParSetGeo::CbmTrdParSetGeo(const char* name,
                                 const char* title,
                                 const char* context)
  : CbmTrdParSet(name, title, context) {
  LOG(debug) << "Constructor of CbmTrdParSetGeo";
  Init();
  LOG(debug) << "Constructor of CbmTrdParSetGeo --- finished";
}
 
//________________________________________________________________________________________
CbmTrdParSetGeo::~CbmTrdParSetGeo(void) {}
 
//________________________________________________________________________________________
Bool_t CbmTrdParSetGeo::Init() {
  CbmTrdGeoHandler geo;
  TGeoNode* topNode = gGeoManager->GetTopNode();
  TObjArray* nodes  = topNode->GetNodes();
  if (!nodes) {
    LOG(fatal) << "CbmTrdParSetGeo::Init: nodes is null!";
    return kFALSE;
  }
 
  for (Int_t iNode = 0; iNode < nodes->GetEntriesFast(); iNode++) {
    TGeoNode* node = static_cast<TGeoNode*>(nodes->At(iNode));
    if (!TString(node->GetName()).Contains("trd", TString::kIgnoreCase))
      continue;  // trd_vXXy top node, e.g. trd_v13a, trd_v14b
    TGeoNode* station = node;
 
    TObjArray* layers = station->GetNodes();
    for (Int_t iLayer = 0; iLayer < layers->GetEntriesFast(); iLayer++) {
      TGeoNode* layer = static_cast<TGeoNode*>(layers->At(iLayer));
      if (!TString(layer->GetName()).Contains("layer", TString::kIgnoreCase))
        continue;  // only layers
 
      TObjArray* modules = layer->GetNodes();
      for (Int_t iModule = 0; iModule < modules->GetEntriesFast(); iModule++) {
        TGeoNode* module = static_cast<TGeoNode*>(modules->At(iModule));
        TObjArray* parts = module->GetNodes();
        for (Int_t iPart = 0; iPart < parts->GetEntriesFast(); iPart++) {
          TGeoNode* part = static_cast<TGeoNode*>(parts->At(iPart));
          if (!TString(part->GetName()).Contains("gas", TString::kIgnoreCase))
            continue;  // only active gas volume
 
          // Put together the full path to the interesting volume, which
          // is needed to navigate with the geomanager to this volume.
          // Extract the geometry information (size, global position)
          // from this volume.
          TString path = TString("/") + topNode->GetName() + "/"
                         + station->GetName() + "/" + layer->GetName() + "/"
                         + module->GetName() + "/" + part->GetName();
 
          LOG(debug) << "Adding detector with path " << path.Data();
          // Generate a physical node which has all needed information
          gGeoManager->cd(path.Data());
          Int_t address = geo.GetModuleAddress();
          fModuleMap[address] =
            new CbmTrdParModGeo(Form("TRD_%d", address), path.Data());
        }
      }
    }
  }
  return true;
}
 
//________________________________________________________________________________________
void CbmTrdParSetGeo::Print(Option_t*) const {
  std::map<Int_t, CbmTrdParMod*>::const_iterator imod = fModuleMap.begin();
  CbmTrdParModGeo* mod(nullptr);
  while (imod != fModuleMap.end()) {
    mod = (CbmTrdParModGeo*) imod->second;
    printf("Module %4d [%p]:\n", imod->first, (void*) mod);
    printf("  %s[%s]\n", mod->GetName(), mod->GetPath());
    Double_t xyz[3];
    mod->GetXYZ(xyz);
    printf("  X[%.2f] Y[%.2f] Z[%.2f] DX[%.2f] DY[%.2f] DZ[%.2f]\n",
           xyz[0],
           xyz[1],
           xyz[2],
           mod->GetDX(),
           mod->GetDY(),
           mod->GetDZ());
    imod++;
  }
}
 
// //________________________________________________________________________________________
// void CbmTrdParSetGeo::putParams(FairParamList* l)
// {
//   if (!l) return;
//   TObjArray *snodes(nullptr), *pnodes(nullptr);
//   //for()
//   l->addObject("FairGeoNodes Sensitive List", snodes);
//   l->addObject("FairGeoNodes Passive List", pnodes);
// }
//
// //________________________________________________________________________________________
// Bool_t CbmTrdParSetGeo::getParams(FairParamList* l)
// {
//   if (!l) return kFALSE;
//   TObjArray *snodes(nullptr), *pnodes(nullptr);
//   if (!l->fillObject("FairGeoNodes Sensitive List", snodes)) return kFALSE;
//   if (!l->fillObject("FairGeoNodes Passive List", pnodes)) return kFALSE;
//
//   //CbmTrdParModGeo *geo(nullptr);
//   for (Int_t i=0; i < snodes->GetEntriesFast(); i++){
// //     fModPar[moduleId[i]] = new CbmTrdParModGeo(GetName());
// //     fModPar
//
//     ((FairGeoNode*)(*snodes)[i])->print();
//   }
//   return kTRUE;
// }
 
ClassImp(CbmTrdParSetGeo)