CbmTrdClusterFinder.h

Go to the documentation of this file.

/* Copyright (C) 2010-2021 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Alexandru Bercuci, Pascal Raisig, Florian Uhlig [committer] */
 
#ifndef CBMTRDCLUSTERFINDER_H
#define CBMTRDCLUSTERFINDER_H
 
#include "CbmEvent.h"
#include "FairTask.h"
 
#include <RtypesCore.h>
 
#include <map>
#include <set>
#include <vector>
 
class CbmTrdCluster;
class CbmTrdDigi;
class CbmTrdParSetAsic;
class CbmTrdParSetGas;
class CbmTrdParSetGeo;
class CbmTrdParSetDigi;
class CbmTrdParSetGain;
class CbmTrdModuleRec;
class TClonesArray;
class TGeoPhysicalNode;
 
class CbmTrdClusterFinder : public FairTask {
  friend class CbmTrdModuleRecR;
  friend class CbmTrdModuleRec2D;
 
 public:
  enum ECbmTrdRecDef
  {
    kTime = 0,         
    kMultiHit,         
    kRowMerger,        
    kNeighbourCol,     
    kNeighbourRow,     
    kDumpClusters,     
    kFASP,             
    kOnlyEventDigis,   
    kDebugStatements,  
    kUseRecoHelpers    
  };
 
  CbmTrdClusterFinder();
 
  ~CbmTrdClusterFinder();
 
  static Float_t GetMinimumChargeTH() { return fgMinimumChargeTH; }
  static Bool_t HasDumpClusters() { return TESTBIT(fgConfig, kDumpClusters); }
  static Bool_t HasMultiHit() { return TESTBIT(fgConfig, kMultiHit); }
  static Bool_t HasNeighbourCol() { return TESTBIT(fgConfig, kNeighbourCol); }
  static Bool_t HasNeighbourRow() { return TESTBIT(fgConfig, kNeighbourRow); }
  static Bool_t HasRowMerger() { return TESTBIT(fgConfig, kRowMerger); }
  static Bool_t IsTimeBased() { return TESTBIT(fgConfig, kTime); }
  static Bool_t DoDebugPrintouts() { return TESTBIT(fgConfig, kDebugStatements); }
  static Bool_t UseRecoHelpers() { return TESTBIT(fgConfig, kUseRecoHelpers); }
 
  static Bool_t UseOnlyEventDigis() { return TESTBIT(fgConfig, kOnlyEventDigis); }
 
  //virtual InitStatus ReInit();
  virtual InitStatus Init();
  virtual void SetParContainers();
 
  virtual void Exec(Option_t* option);
 
  virtual void Finish();
 
 
  static void SetDumpClusters(Bool_t set = kTRUE)
  {
    set ? SETBIT(fgConfig, kDumpClusters) : CLRBIT(fgConfig, kDumpClusters);
  }
  static void SetRowMerger(Bool_t set = kTRUE) { set ? SETBIT(fgConfig, kRowMerger) : CLRBIT(fgConfig, kRowMerger); }
  static void SetMultiHit(Bool_t set = kTRUE) { set ? SETBIT(fgConfig, kMultiHit) : CLRBIT(fgConfig, kMultiHit); }
  static void SetNeighbourEnable(Bool_t col = kTRUE, Bool_t row = kFALSE)
  {
    col ? SETBIT(fgConfig, kNeighbourCol) : CLRBIT(fgConfig, kNeighbourCol);
    row ? SETBIT(fgConfig, kNeighbourRow) : CLRBIT(fgConfig, kNeighbourRow);
  }
  static void SetMinimumChargeTH(Float_t th) { fgMinimumChargeTH = th; }
  static void SetTimeBased(Bool_t set = kTRUE) { set ? SETBIT(fgConfig, kTime) : CLRBIT(fgConfig, kTime); }
 
  static void SetDoDebugPrintouts(Bool_t set = kTRUE)
  {
    set ? SETBIT(fgConfig, kDebugStatements) : CLRBIT(fgConfig, kDebugStatements);
  }
  static void SetUseRecoHelpers(Bool_t set = kTRUE)
  {
    set ? SETBIT(fgConfig, kUseRecoHelpers) : CLRBIT(fgConfig, kUseRecoHelpers);
  }
 
 
  static void SetUseOnlyEventDigis(Bool_t set = kTRUE)
  {
    set ? SETBIT(fgConfig, kOnlyEventDigis) : CLRBIT(fgConfig, kOnlyEventDigis);
  }
 
 protected:
  Bool_t AddCluster(CbmTrdCluster* c);
 
 private:
  CbmTrdClusterFinder(const CbmTrdClusterFinder&);
  CbmTrdClusterFinder& operator=(const CbmTrdClusterFinder&);
 
  Int_t AddClusters(TClonesArray* clusters, CbmEvent* event, Bool_t moveOwner = kTRUE);
 
  UInt_t addDigisToModules();
 
  UInt_t addDigisToModules(CbmEvent* event);
 
  void addDigiToModule(UInt_t digiIdx);
 
  void processDigisInModules(UInt_t ndigis, CbmEvent* event = nullptr, bool clr = true);
 
  CbmTrdModuleRec* AddModule(const CbmTrdDigi* digi);
 
 
  static Int_t fgConfig;             
  static Float_t fgMinimumChargeTH;  
 
 
  TClonesArray* fClusters = nullptr; 
  TClonesArray* fEvents = nullptr;
 
  //==================================================================
  std::map<Int_t, CbmTrdModuleRec*> fModules = {};       
  CbmTrdParSetAsic* fAsicPar                 = nullptr;  
  CbmTrdParSetGas* fGasPar                   = nullptr;  
  CbmTrdParSetDigi* fDigiPar                 = nullptr;  
  CbmTrdParSetGain* fGainPar                 = nullptr;  
  CbmTrdParSetGeo* fGeoPar                   = nullptr;  
 
  UInt_t fNrTs = 0;
 
  UInt_t fNrEvents = 0;
 
  UInt_t fNrDigis = 0;
 
  UInt_t fNrClusters = 0;
 
  Float_t fProcessTime = 0;
 
  ClassDef(CbmTrdClusterFinder, 2);
};
#endif