LUCID_PileUpTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef LUCID_DIGITIZATION_TOOL_H
#define LUCID_DIGITIZATION_TOOL_H
 
#include "PileUpTools/PileUpToolBase.h"
 
#include "LUCID_DigitizationToolBox.h"
 
#include "LUCID_SimEvent/LUCID_SimHitCollection.h"
#include "LUCID_RawEvent/LUCID_DigitContainer.h"
 
#include "xAODEventInfo/EventInfo.h"   // SubEventIterator
#include "xAODEventInfo/EventAuxInfo.h"// SubEventIterator
 
#include "AthenaKernel/IAthRNGSvc.h"
#include "PileUpTools/PileUpMergeSvc.h"
 
#include "Gaudi/Property.h"
#include "GaudiKernel/ServiceHandle.h"
 
#include <vector>
#include <utility> /* pair */
 
namespace CLHEP {
  class HepRandomEngine;
}
 
class LUCID_PileUpTool: public PileUpToolBase, public LUCID_DigitizationToolBox {
 
public:
 
  LUCID_PileUpTool(const std::string& type,
                   const std::string& name,
                   const IInterface* parent);
 
  virtual StatusCode initialize() override final;
  virtual StatusCode finalize() override final;

  virtual StatusCode prepareEvent(const EventContext& ctx, const unsigned int nInputEvents) override final;

  virtual StatusCode processBunchXing(int bunchXing,
                                      SubEventIterator bSubEvents,
                                      SubEventIterator eSubEvents) override final;
  //  virtual bool toProcess(int bunchXing) const override final;

  virtual StatusCode mergeEvent(const EventContext& ctx) override final;

  virtual StatusCode processAllSubEvents(const EventContext& ctx) override final;
 
private:
 
  ServiceHandle<PileUpMergeSvc> m_mergeSvc{this, "mergeSvc", "PileUpMergeSvc", ""};
  ServiceHandle<IAthRNGSvc> m_randomSvc{this, "RndmSvc", "AthRNGSvc", ""};
  Gaudi::Property<std::string> m_randomStreamName{this, "RandomStreamName", "LUCIDRndEng", ""};
 
  std::string m_dataObjectName;
  std::string m_SimHitCollectionName;
  std::string m_digitsContainerName;
  std::string m_key_digitCnt;
 
  LUCID_SimHitCollection* m_mergedhitList{};
};
 
#endif