NnPixelClusterSplitProbTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/

//NnPixelClusterSplitProbTool.h
// (c) ATLAS Detector software
// Estimate cluster split probability by using multivariate techniquew
// 
// @author Giacinto Piacquadio (CERN PH-ADE-ID)
//
 
#ifndef SiClusterizationTool_NnPixelClusterSplitProbTool_H
#define SiClusterizationTool_NnPixelClusterSplitProbTool_H
 
 
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "InDetRecToolInterfaces/IPixelClusterSplitProbTool.h"
//Beam Spot Condition
#include "BeamSpotConditionsData/BeamSpotData.h"
#include "TrkParameters/TrackParameters.h"
 
 
namespace InDet {
 
  class NnClusterizationFactory;
 
  class NnPixelClusterSplitProbTool final : public extends<AthAlgTool, IPixelClusterSplitProbTool>
  {
  public:
    
    NnPixelClusterSplitProbTool(const std::string& t, const std::string& n, const IInterface*  p);
 
    virtual ~NnPixelClusterSplitProbTool() = default;
    
    StatusCode initialize() override;
 
    virtual InDet::PixelClusterSplitProb splitProbability(
        const InDet::PixelCluster& origCluster) const override;
 
    virtual InDet::PixelClusterSplitProb splitProbability(
        const InDet::PixelCluster& origCluster,
        const Trk::TrackParameters& trackParameters) const override;
 
   private:
    
    InDet::PixelClusterSplitProb compileSplitProbability(std::vector<double>& vectorOfProbs ) const;
    
    ToolHandle<NnClusterizationFactory> m_NnClusterizationFactory{ this, "NnClusterizationFactory", "InDet::NnClusterizationFactory/NnClusterizationFactory" };
    SG::ReadCondHandleKey<InDet::BeamSpotData> m_beamSpotKey { this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot" };
 
    DoubleArrayProperty m_priorMultiplicityContent{ this, "PriorMultiplicityContent", {2793337, 82056, 19944} };
 
    BooleanProperty m_useBeamSpotInfo{ this, "useBeamSpotInfo", true };
 
  };
  
}
 
 
#endif