MergeTrackRecordCollTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MCTRUTHSIMALGS_MERGETRACKRECORDCOLLTOOL_H
#define MCTRUTHSIMALGS_MERGETRACKRECORDCOLLTOOL_H
 
#include "PileUpTools/PileUpToolBase.h"
#include "TrackRecord/TrackRecordCollection.h"
 
#include "Gaudi/Property.h"
#include "GaudiKernel/ServiceHandle.h"
#include "PileUpTools/PileUpMergeSvc.h"
 
#include <string>

class MergeTrackRecordCollTool : public PileUpToolBase {
public:
  MergeTrackRecordCollTool(const std::string& type,
                           const std::string& name,
                           const IInterface* parent);
  virtual StatusCode initialize() override final;
  virtual StatusCode prepareEvent(const EventContext& ctx, unsigned int nInputEvents) override final;
  virtual StatusCode mergeEvent(const EventContext& ctx) override final;
  virtual StatusCode processBunchXing(int bunchXing,
                                      SubEventIterator bSubEvents,
                                      SubEventIterator eSubEvents) override final;
  virtual bool toProcess(int bunchXing) const override final;
  virtual StatusCode processAllSubEvents(const EventContext& ctx) override final;
private:
  ServiceHandle<PileUpMergeSvc> m_pMergeSvc{this, "PileUpMergeSvc", "PileUpMergeSvc", ""};
  StringProperty m_trRecCollKey{this, "TrackRecordCollKey", "MuonEntryLayer", ""};
  SG::WriteHandleKey<TrackRecordCollection> m_outputKey{ this, "TrackRecordCollOutputKey", "MuonExitLayer", "" };
  bool m_firstSubEvent{true};
};
#endif //MCTRUTHSIMALGS_MERGETRACKRECORDCOLLTOOL_H