HgtdTimedClusteringTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSTRK_DATAPREPARATION_HGTD_TIMEDCLUSTERING_TOOL_H
#define ACTSTRK_DATAPREPARATION_HGTD_TIMEDCLUSTERING_TOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "ActsToolInterfaces/IHGTDClusteringTool.h"
 
#include "HGTD_Identifier/HGTD_ID.h"
#include "HGTD_ReadoutGeometry/HGTD_DetectorManager.h"
 
#include "Acts/Definitions/Units.hpp"
 
namespace Hgtd {
  struct UnpackedHgtdRDO {
    UnpackedHgtdRDO(int ncl, int row, int col, float toa, int tot, Identifier id)
      : NCL(ncl), ROW(row), COL(col), TOA(toa), TOT(tot), ID(id)
    {};
    
    int        NCL{0};
    int        ROW{0};
    int        COL{0};
    float      TOA{0.f};
    int        TOT{0};
    Identifier ID {};
  };
}
 
namespace ActsTrk {
 
class HgtdTimedClusteringTool :
    public extends<AthAlgTool, IHGTDClusteringTool> {
public:
  struct Cluster {
    std::vector<Identifier> ids;
    std::vector<int> tots;
    std::vector<double> times;
  };
  
  using Cell = Hgtd::UnpackedHgtdRDO;
  using CellCollection = std::vector<Cell>;
  using ClusterCollection = std::vector<Cluster>;
  
public:
    HgtdTimedClusteringTool(const std::string& type,
                const std::string& name,
                const IInterface* parent);
 
    virtual StatusCode initialize() override;
    virtual StatusCode clusterize(const EventContext& ctx,
                  const RawDataCollection& RDOs,
                  ClusterContainer& container) const override;
 
private:
  // N.B. the cluster is added to the container
  StatusCode makeCluster(const EventContext& ctx,
             const typename HgtdTimedClusteringTool::Cluster &cluster,
             xAOD::HGTDCluster& xaodcluster) const;
  
 
private:
    const HGTD_DetectorManager* m_hgtd_det_mgr{nullptr};
    const HGTD_ID* m_hgtd_id{nullptr};
 
  Gaudi::Property<double> m_timeTollerance {this, "TimeTollerance", 0.035 * Acts::UnitConstants::ns};
  Gaudi::Property<bool> m_addCorners {this, "AddCorners", true};
};
 
}
 
#endif