AugOriginalCounts.h

Go to the documentation of this file.

/* 
   Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORKBPHYS_AUGORIGINALCOUNTS_H
#define DERIVATIONFRAMEWORKBPHYS_AUGORIGINALCOUNTS_H
 
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
 
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "StoreGate/WriteDecorHandleKey.h"
 
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/VertexContainer.h"
#include "xAODEventInfo/EventInfo.h"
 
#include <string>
 
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/VertexContainer.h"
#include <string>
#include "xAODEventInfo/EventInfo.h"
#include <StoreGate/WriteDecorHandleKey.h>
 
namespace DerivationFramework {
  class AugOriginalCounts : public AthAlgTool, public IAugmentationTool {
  public:
    AugOriginalCounts(const std::string& t, const std::string& n,
              const IInterface* p);
    virtual StatusCode addBranches() const override;
    virtual StatusCode initialize() override;
    private:
    SG::WriteDecorHandleKey<xAOD::EventInfo> m_OrigPVNTracks{this, "DO_NOT_SET1", "", "internal property"};
    SG::WriteDecorHandleKey<xAOD::EventInfo> m_OrigNTracksKeys{this, "DO_NOT_SET2", "", "internal property"};
    SG::WriteDecorHandleKey<xAOD::EventInfo> m_OrigNTracksLRTKeys{this, "DO_NOT_SET22", "", "internal property"};
    SG::WriteDecorHandleKey<xAOD::EventInfo> m_OrigNtype0{this, "DO_NOT_SET3", "", "internal property"};
    SG::WriteDecorHandleKey<xAOD::EventInfo> m_OrigNtype1{this, "DO_NOT_SET4", "", "internal property"};
    SG::WriteDecorHandleKey<xAOD::EventInfo> m_OrigNtype2{this, "DO_NOT_SET5", "", "internal property"};
    SG::WriteDecorHandleKey<xAOD::EventInfo> m_OrigNtype3{this, "DO_NOT_SET6", "", "internal property"};
    SG::WriteDecorHandleKey<xAOD::EventInfo> m_OrigNtypeUnknown{this, "DO_NOT_SET7", "", "internal property"};
 
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_OrigSqrtPt2Sum{this, "DO_NOT_SET8", "", "internal property"};
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_d_nPVTracks{this, "DO_NOT_SET9", "", "internal property"};
    SG::ReadHandleKey<xAOD::TrackParticleContainer> m_TrackContainername;
    SG::ReadHandleKey<xAOD::TrackParticleContainer> m_TrackContainerLRTname;
    SG::ReadHandleKey<xAOD::VertexContainer> m_PVContainername;
    bool        m_addPVCountsByType;
    bool        m_addNTracksToPVs;
    bool        m_addSqrtPt2SumToPVs;
  };
}
 
#endif // DERIVATIONFRAMEWORKBPHYS_AUGORIGINALCOUNTS_H