AFPSiDLinRegTool.h

Go to the documentation of this file.

/*
    Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
 
#ifndef AFP_LOCRECO_AFPSIDLINREGTOOL_H
#define AFP_LOCRECO_AFPSIDLINREGTOOL_H
 
 
 
// FrameWork includes
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
 
// xAOD includes
#include "xAODForward/AFPSiHitsCluster.h"
#include "xAODForward/AFPSiHitsClusterContainer.h"
#include "xAODForward/AFPTrackContainer.h"
#include "xAODForward/AFPTrack.h"
#include "xAODForward/AFPTrackRecoAlgID.h"
 
// AFP_LocReco includes
#include "AFP_LocReco/IAFPSiDLocRecoTrackAlgTool.h"
 
// STL includes
#include <string>
#include <list>
#include <vector>
#include <tuple>
#include <memory>
#include <utility>
 
 
class AFPSiDLinRegTool
: public extends<AthAlgTool, IAFPSiDLocRecoTrackAlgTool>
{
 
public:
    AFPSiDLinRegTool(const std::string& type, const std::string& name, const IInterface* parent);
 
    virtual ~AFPSiDLinRegTool() override {}
 
    virtual StatusCode initialize() override;
 
 
    StatusCode reconstructTracks(std::unique_ptr<xAOD::AFPTrackContainer>& outputContainer, const EventContext& ctx) const override;
 
    const std::string& outputContainerName () const override {return m_tracksContainerName;}
 
private:
    Gaudi::Property<std::string> m_tracksContainerName{this, "tracksContainerName", "AFPTrackContainer", "Name of the container in which tracks are saved"};
 
    Gaudi::Property<int> m_stationID{this, "stationID", 0, "ID number of station for which tracks should be reconstructed"};
    
    Gaudi::Property<double> m_allowedDistanceBetweenClustersInTrack{this, "allowedDistanceBetweenClustersInTrack", 0.5, "Maximum allowed distance between clusters to be considered coming from the same proton"};
    
    Gaudi::Property<std::string> m_trackSelection{this, "trackSelection", "Medium", "Track selection, allowed values are \"Loose\" (accept all track candidates), \"Medium\" (accept only tracks with clusters in at least #m_minimalNumberOfPlanesInTrack planes), or \"Tight\" (reject all with more than a single cluster per plane and less than #m_minimalNumberOfPlanesInTrack planes)"};
    
    Gaudi::Property<unsigned int>  m_minimalNumberOfPlanesInTrack{this, "minimalNumberOfPlanesInTrack", 2, "Minimal numer of planes required for track candidate"};
    
    // function to select tracks
    std::function<bool(const std::vector<const xAOD::AFPSiHitsCluster*>&)> m_selectionFunction;
    
    SG::ReadHandleKey<xAOD::AFPSiHitsClusterContainer> m_hitsClusterContainerKey{this, "AFPSiHitsClusterContainerKey", "AFPSiHitsClusterContainer", "Name of the container with clusters of hits from which tracks are to be reconstructed"};
    
    std::vector<const xAOD::AFPSiHitsCluster*> findClusterAroundElement(std::list<const xAOD::AFPSiHitsCluster*>& toJoin, const xAOD::AFPSiHitsCluster* init) const;
    
    bool areNeighbours(const xAOD::AFPSiHitsCluster* a, const xAOD::AFPSiHitsCluster* b) const;
    
    std::pair<double, double> linearRegression(const std::vector<std::pair<double, double>>& YX) const;
    };
 
 
#endif