CbmRichGeoHandler.h

Go to the documentation of this file.

/* Copyright (C) 2015-2024 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Semen Lebedev, Martin Beyer, Andrey Lebedev [committer], Florian Uhlig */

 
#pragma once
 
#include "CbmRichDetectorData.h"
 
#include <RtypesCore.h>
#include <TString.h>
 
#include <iostream>
#include <map>
#include <vector>
 
class CbmRichGeoHandler {
  enum class CbmRichGeoSetup
  {
    Rich,
    MiniRich,
  };
 
 private:
  CbmRichGeoHandler() { Init(); }
 
 public:
  virtual ~CbmRichGeoHandler() = default;

  CbmRichGeoHandler(const CbmRichGeoHandler&) = delete;

  CbmRichGeoHandler& operator=(const CbmRichGeoHandler&) = delete;

  static CbmRichGeoHandler& GetInstance()
  {
    static CbmRichGeoHandler instance;
    return instance;
  }

  std::vector<Int_t> GetPixelAddresses() { return fPixelAddresses; }

  Int_t GetPixelAddressByPath(const std::string& path);

  CbmRichPixelData* GetPixelDataByAddress(Int_t address);

  Int_t GetRandomPixelAddress();

  std::vector<Int_t> GetPmtIds() { return fPmtIds; }

  CbmRichPmtData* GetPmtDataById(Int_t id);

  std::vector<Int_t> GetNeighbourPixels(Int_t address, Int_t N, Bool_t horizontal = true, Bool_t vertical = true,
                                        Bool_t diagonal = true);

  std::vector<Int_t> GetDirectNeighbourPixels(Int_t address, Bool_t horizontal = true, Bool_t vertical = true)
  {
    return GetNeighbourPixels(address, 1, horizontal, vertical, false);
  }

  std::vector<Int_t> GetDiagonalNeighbourPixels(Int_t address)
  {
    return GetNeighbourPixels(address, 1, false, false, true);
  }

  std::vector<Int_t> GetNxNNeighbourPixels(Int_t address, Int_t n)
  {
    return GetNeighbourPixels(address, n, true, true, true);
  }
 
 private:
  bool fFallback{false};  // Fallback mode for deriving CBM RICH addresses
 
  std::map<std::string, Int_t> fPixelPathToAddress{};
  std::map<Int_t, std::unique_ptr<CbmRichPixelData>> fPixelAddressToData{};
  std::vector<Int_t> fPixelAddresses{};
 
  std::map<std::string, Int_t> fPmtPathToId{};
  std::map<Int_t, std::unique_ptr<CbmRichPmtData>> fPmtIdToData{};
  std::vector<Int_t> fPmtIds{};

  void Init();

  CbmRichGeoSetup GetDetectorSetup(const TString& nodePath)
  {
    return nodePath.Contains("mcbm") ? CbmRichGeoSetup::MiniRich : CbmRichGeoSetup::Rich;
  }

  Int_t CreateAddressRich(const std::string& nodePath, Int_t& channelAddr, Int_t& pmtId);

  Int_t CreateAddressMiniRich(const std::string& nodePath, Int_t& channelAddr, Int_t& pmtId);
};