CbmStsAnaBase.cxx

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dario Ramirez [committer] */
 
#include "CbmStsAnaBase.h"
 
#include "CbmStsModule.h"
#include "CbmStsStation.h"
#include "FairRootManager.h"
 
#include <TGeoBBox.h>          // for Geometry loading from file
#include <TGeoManager.h>       // for Geometry loading from file
#include <TGeoMatrix.h>        // for Geometry loading from file
#include <TGeoNode.h>          // for Geometry loading from file
#include <TGeoPhysicalNode.h>  // for Geometry loading from file
#include <TGeoVolume.h>        // for Geometry loading from file
 
#include <cassert>
 
 
void CbmStsAnaBase::SetCutMap(CbmCutMap* cuts_map)
{
  assert(cuts_map != nullptr);
  fAnalysisCuts = cuts_map;
}
 
void CbmStsAnaBase::LoadSetup()
{
  CbmStsSetup* sts_setup = CbmStsSetup::Instance();
  if (!sts_setup->IsInit()) {
    sts_setup->Init(nullptr);
  }
 
  if (sts_setup == nullptr) {
    LOG(warning) << "Something is wrong with the setup ...";
    return;
  }
 
  nb_sts_station_ = sts_setup->GetNofStations();
  assert(nb_sts_station_ != 0);
 
  for (unsigned short unit_idx = 0; unit_idx < nb_sts_station_; unit_idx++) {
    CbmStsStation* sts_unit  = sts_setup->GetStation(unit_idx);
    fStsGeoInfo[unit_idx]    = std::vector<double>(5, 0);
    fStsGeoInfo[unit_idx][0] = sts_unit->GetXmin();
    fStsGeoInfo[unit_idx][1] = sts_unit->GetXmax();
    fStsGeoInfo[unit_idx][2] = sts_unit->GetYmin();
    fStsGeoInfo[unit_idx][3] = sts_unit->GetYmax();
    fStsGeoInfo[unit_idx][4] = sts_unit->GetZ();
  }
 
  for (unsigned short module_idx = 0; module_idx < sts_setup->GetNofModules(); module_idx++) {  // Loop over modules
    CbmStsModule* sts_module = sts_setup->GetModule(module_idx);
    assert(sts_module);
    int32_t address = sts_module->GetAddress();
    assert(sts_module->GetNofDaughters() == 1);
 
    CbmStsElement* sts_sensor     = sts_module->GetDaughter(0);
    TGeoPhysicalNode* sensor_node = sts_sensor->GetPnode();
    TGeoVolume* sensor_volume     = sensor_node->GetVolume();
    TGeoBBox* sensor_shape        = (TGeoBBox*) sensor_volume->GetShape();
    TGeoMatrix* sensor_matrix     = sensor_node->GetMatrix();
 
    // const double* local = sensor_shape->GetOrigin();
    const double* trans     = sensor_matrix->GetTranslation();
    fStsGeoInfo[address]    = std::vector<double>(6, 0);
    fStsGeoInfo[address][0] = trans[0] - sensor_shape->GetDX();  // Xmin
    fStsGeoInfo[address][1] = trans[0] + sensor_shape->GetDX();  // Xmax
    fStsGeoInfo[address][2] = trans[1] - sensor_shape->GetDY();  // Ymin
    fStsGeoInfo[address][3] = trans[1] + sensor_shape->GetDY();  // Ymax
    fStsGeoInfo[address][4] = trans[2] - sensor_shape->GetDZ();  // Zmin
    fStsGeoInfo[address][5] = trans[2] + sensor_shape->GetDZ();  // Zmax
 
    if (address > 8) {
      switch (2 * (int) sensor_shape->GetDY()) {
        case 2: fFirstZStrip[address] = 2048 - 32; break;
        case 4: fFirstZStrip[address] = 2048 - 88; break;
        case 6: fFirstZStrip[address] = 2048 - 134; break;
        case 12: fFirstZStrip[address] = 2048 - 274; break;
        default:
          LOG(warning) << Form("Unknown sensor shape 0x%x: [%0.3f , %0.3f , %0.3f]", address, sensor_shape->GetDX(),
                               sensor_shape->GetDY(), sensor_shape->GetDZ());
          break;
      }
    }
  }
 
  LOG(debug) << "Loaded setup information:\n";
  for (auto& [address, geo] : fStsGeoInfo) {
    LOG(debug) << Form("0x%x: [%+0.3f, %+0.3f] ^ [%+0.3f, %+0.3f] ^ [%+0.3f, %+0.3f]", address, geo[0], geo[1], geo[2],
                       geo[3], geo[4], geo[5]);
  }
}
 
void CbmStsAnaBase::SaveToFile()
{
  FairRootManager* ioman = FairRootManager::Instance();
  auto sink              = ioman->GetSink();
 
  if (!sink) {
    LOG(fatal) << "Check the configuration: sink is invalid!";
    return;
  }
 
  LOG(info) << Form("Task histograms will be saved to %s ...", sink->GetFileName().Data());
 
  for (auto& [name, gr] : fG1D) {
    if (gr == nullptr) continue;
    sink->WriteObject(gr.get(), name.c_str());
  }
  for (auto& [name, h] : fH1D) {
    if (h == nullptr) continue;
    sink->WriteObject(h.get(), name.c_str());
  }
  for (auto& [name, h] : fH2D) {
    if (h == nullptr) continue;
    sink->WriteObject(h.get(), name.c_str());
  }
  for (auto& [name, h] : fH2DShared) {
    if (h == nullptr) continue;
    sink->WriteObject(h.get(), name.c_str());
  }
}