Cluster2D.h

Go to the documentation of this file.

/* Copyright (C) 2024 Facility for Antiproton and Ion Research in Europe, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dominik Smith [committer], Alexandru Bercuci */
 
#pragma once
 
#include "DigiRec.h"
#include "compat/RTypes.h"
 
#include <cstdint>
#include <vector>  // for vector
 
class CbmTrdDigi;
 
namespace cbm::algo::trd
{
  class Cluster2D {
 
   public:
    enum eClusterDef
    {
      kFasp = 5  
        ,
      kStart  
        ,
      kStop  
    };
 
    Cluster2D() = delete;
    Cluster2D(const Cluster2D& ref);
    Cluster2D(int32_t address, int32_t idx, const CbmTrdDigi* digi, uint16_t chT, uint16_t chR, int32_t r,
              int32_t time);
    virtual ~Cluster2D(){};
 
    void AddDigiIdxPair(int32_t index, const CbmTrdDigi* digi)
    {
      fDigiIndices.push_back(index);
      fDigis.push_back(digi);
    }
 
    void AddDigiIdxPairs(const std::vector<int32_t>& indices, const std::vector<const CbmTrdDigi*> digis)
    {
      fDigiIndices.insert(fDigiIndices.end(), indices.begin(), indices.end());
      fDigis.insert(fDigis.end(), digis.begin(), digis.end());
    }
 
    int32_t GetNofDigis() const { return fDigiIndices.size(); }
 
    int32_t GetDigi(int32_t index) const { return fDigiIndices[index]; }
 
    const std::vector<int32_t>& GetDigiIndices() const { return fDigiIndices; }
 
    std::vector<const CbmTrdDigi*>& GetDigis() { return fDigis; }
 
    const std::vector<DigiRec>& GetRecDigis() const { return fRecDigis; }
 
    void ClearDigis()
    {
      fDigiIndices.clear();
      fDigis.clear();
      fRecDigis.clear();
    }
 
    int32_t GetAddress() const { return fAddress; }
 
    bool AddDigi(int32_t idx, const CbmTrdDigi* digi, uint16_t chT = 0xffff, uint16_t chR = 0, int32_t dt = 0);
 
    uint16_t GetNCols() const { return fNCols; }
    uint16_t GetNRows() const { return fNRows & 0x1f; }
    uint16_t GetNRowsRaw() const { return fNRows; }
    uint16_t GetEndCh() const { return fStartCh + fNCols - 1; }
    uint16_t GetRow() const { return GetNRows(); }
    uint16_t GetSize() const { return GetNCols(); }
    uint16_t GetStartCh() const { return fStartCh; }
    uint32_t GetStartTime() const { return fStartTime; }
    bool HasFaspDigis() const { return true; }
    bool HasStart() const { return TESTBIT(fNRows, kStart); }
    bool HasStop() const { return TESTBIT(fNRows, kStop); }
 
    int32_t IsChannelInRange(uint16_t chT, uint16_t chR) const;
 
    bool Merge(Cluster2D* second);
 
    bool Finalize(const size_t numCols);
 
    void SetNCols(uint16_t ncols) { fNCols = ncols; }
    void SetNRows(uint16_t nrows)
    {
      fNRows &= (7 << 5);
      fNRows |= (nrows & 0x1f);
    }
    void SetStart(bool set = true) { set ? SETBIT(fNRows, kStart) : CLRBIT(fNRows, kStart); }
    void SetStop(bool set = true) { set ? SETBIT(fNRows, kStop) : CLRBIT(fNRows, kStop); }
 
   private:
    std::vector<DigiRec> fRecDigis;         
    std::vector<const CbmTrdDigi*> fDigis;  
    std::vector<int32_t> fDigiIndices;      
 
    int32_t fAddress    = 0;     
    uint8_t fNCols      = 0;     //< number of columns with charge above threshold
    uint8_t fNRows      = 0x1f;  //< cluster row info plus extra meta data. Use dedicated getters for the correct value
    uint16_t fStartCh   = 0xffff;      //< channel address of first channel
    uint32_t fStartTime = 0xffffffff;  
  };
}  // namespace cbm::algo::trd