trd/Clusterizer.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], Etienne Bechtel, Florian Uhlig */
 
#pragma once
 
#include "CbmTrdDigi.h"
#include "Cluster.h"
#include "HitFinderPars.h"
 
#include <tuple>
#include <vector>
 
namespace cbm::algo::trd
{
 
 
  class Clusterizer {
   public:
    typedef std::tuple<int, const CbmTrdDigi*, double>
      inputType;  //digi index, pointer to digi (null if processed), digi time
 
    Clusterizer() = default;
 
    Clusterizer(HitFinderModPar par) : fParams(par){};
 
    virtual ~Clusterizer(){};
 
    std::vector<Cluster> operator()(const std::vector<std::pair<CbmTrdDigi, int32_t>>& inVec) const;
 
   protected:
   private:
    HitFinderModPar fParams;  
 
    bool TryAddDigi(std::vector<std::vector<inputType*>>* digimap, int chan,
                    std::vector<std::vector<inputType*>::iterator>* lastPos,
                    std::vector<std::pair<int, const CbmTrdDigi*>>* cluster, const double digiTime) const;
 
    void addClusters(std::vector<std::pair<int, const CbmTrdDigi*>> cluster, std::vector<Cluster>* clustersOut) const;
  };
 
}  // namespace cbm::algo::trd