LArCalibrationHitContainerBuilder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef LARG4CODE_LARCALIBRATIONHITCONTAINERBUILDER_H
#define LARG4CODE_LARCALIBRATIONHITCONTAINERBUILDER_H
 
#include "CaloSimEvent/CaloCalibrationHit.h"
#include "CaloSimEvent/CaloCalibrationHitContainer.h"
#include "CaloSimEvent/SrCaloCalibrationHitContainer.h"
 
#include <memory>
#include <set>
#include <string>
#include <vector>

template <class HitContainerT>
class LArCalibrationHitContainerBuilderBase : public HitContainerT
{
public:
  using HitContainerT::HitContainerT;
  using hit_ptr_t = std::unique_ptr<CaloCalibrationHit>;
 
  class LessHit {
  public:
    bool operator()(const hit_ptr_t& lhs, const hit_ptr_t& rhs) const
    {
      return lhs->Less(rhs.get());
    }
  };
 
  using hits_t = std::set<hit_ptr_t, LessHit>;
 
  ~LArCalibrationHitContainerBuilderBase() override = default;

  void RegisterSource(const std::string& sourceName);
  void AddHit(const std::string& sourceName, hit_ptr_t hit);
  void Finalize();
 
private:
  struct Partition
  {
    std::string sourceName;
    hits_t hits;
  };
 
  static void AddHit(hits_t& hits, hit_ptr_t hit);
  void Finalize(hits_t& hits);
  Partition& FindOrCreatePartition(const std::string& sourceName);
 
  std::vector<Partition> m_partitions;
  hits_t m_directHits;
};
 
using LArCalibrationHitContainerBuilder =
  LArCalibrationHitContainerBuilderBase<CaloCalibrationHitContainer>;
using LArSrCalibrationHitContainerBuilder =
  LArCalibrationHitContainerBuilderBase<SrCaloCalibrationHitContainer>;
 
#endif