Reco_mumu.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
#ifndef DERIVATIONFRAMEWORK_Reco_mumu_H
#define DERIVATIONFRAMEWORK_Reco_mumu_H
 
#include <string>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "JpsiUpsilonTools/ICandidateSearch.h"
#include "JpsiUpsilonTools/PrimaryVertexRefitter.h"
#include "xAODEventInfo/EventInfo.h"
namespace Trk {
  class V0Tools;
}
 
 

namespace DerivationFramework {
 
  class Reco_mumu : public extends<AthAlgTool, IAugmentationTool> {
    public: 
      Reco_mumu(const std::string& t, const std::string& n, const IInterface* p);
 
      StatusCode initialize() override;
     
      StatusCode addBranches(const EventContext& ctx) const override;
      
    private:
      ToolHandle<Trk::V0Tools>                    m_v0Tools{this, "V0Tools", "Trk::V0Tools"};
      ToolHandle<Analysis::ICandidateSearch>      m_jpsiFinder{this,"JpsiFinder", "Analysis::JpsiFinder"};
      ToolHandle<Analysis::PrimaryVertexRefitter> m_pvRefitter{this, "PVRefitter", "Analysis::PrimaryVertexRefitter"};
      SG::ReadHandleKey<xAOD::EventInfo> m_eventInfo_key{this, "EventInfo", "EventInfo", "Input event information"};
      
      
    // minimum number of tracks for PV to be considered for PV association
   
      Gaudi::Property<bool>         m_refitPV{this, "RefitPV", false};
      Gaudi::Property<int>          m_PV_max{this, "MaxPVrefit", 1};
      Gaudi::Property<int>          m_DoVertexType{this, "DoVertexType", 1};
      Gaudi::Property<unsigned int> m_PV_minNTracks{this, "MinNTracksInPV", 0};
      Gaudi::Property<bool>        m_do3d{this, "Do3d" , false};
      Gaudi::Property<bool>        m_checkCollections{this, "CheckCollections", false};
      
      SG::ReadHandleKeyArray<xAOD::VertexContainer> m_CollectionsToCheck{this, "CheckVertexContainers", {}};
      SG::ReadHandleKey<xAOD::VertexContainer> m_pvContainerKey{this,"PVContainerName", "PrimaryVertices"};
     
      SG::WriteHandleKey<xAOD::VertexContainer> m_refContainerKey{this, "RefPVContainerName","RefittedPrimaryVertices" };
      SG::WriteHandleKey<xAOD::VertexContainer> m_outContainerKey{this, "OutputVtxContainerName", "OniaCandidates"};
  }; 
}
 
#endif // DERIVATIONFRAMEWORK_Reco_mumu_H