LArHitContainerBuilder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef LARG4CODE_LARHITCONTAINERBUILDER_H
#define LARG4CODE_LARHITCONTAINERBUILDER_H
 
#include "LArSimEvent/LArHit.h"
#include "LArSimEvent/LArHitContainer.h"
 
#include "G4Types.hh"
 
#include <map>
#include <memory>
#include <set>
#include <string>
#include <vector>

class LArHitContainerBuilder : public LArHitContainer
{
public:
  using LArHitContainer::LArHitContainer;
  using hit_ptr_t = std::unique_ptr<LArHit>;
 
  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>;
  using timeBins_t = std::map<G4int, hits_t>;
 
  ~LArHitContainerBuilder() override = default;

  void RegisterSource(const std::string& sourceName);
  void AddHit(const std::string& sourceName, hit_ptr_t hit, G4int timeBin);
  void Finalize();
 
private:
  struct Partition
  {
    std::string sourceName;
    timeBins_t timeBins;
  };
 
  static void AddHit(timeBins_t& timeBins, hit_ptr_t hit, G4int timeBin);
  void Finalize(timeBins_t& timeBins);
  Partition& FindOrCreatePartition(const std::string& sourceName);
 
  std::vector<Partition> m_partitions;
  timeBins_t m_directHits;
};
 
#endif