FPGATrackSimLLPDoubletHoughTransformTool.h

Go to the documentation of this file.

// Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
 
#ifndef FPGATrackSimLLPDOUBLETHOUGHTRANSFORMTOOL_H
#define FPGATrackSimLLPDOUBLETHOUGHTRANSFORMTOOL_H
 
 
#include "TFile.h"
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
 
#include "FPGATrackSimObjects/FPGATrackSimTypes.h"
#include "FPGATrackSimObjects/FPGATrackSimVectors.h"
#include "FPGATrackSimObjects/FPGATrackSimRoad.h"
#include "FPGATrackSimObjects/FPGATrackSimHit.h"
#include "FPGATrackSimObjects/FPGATrackSimTrackPars.h"
#include "FPGATrackSimHough/IFPGATrackSimRoadFinderTool.h"
 
 
/*
Implments HoughT for d0, q/pt using hit doublets
Hough space x -> d0 , y -> q/pt
*/
 
class FPGATrackSimLLPDoubletHoughTransformTool : public extends <AthAlgTool, IFPGATrackSimRoadFinderTool >
{
 
public:
    using base_class::base_class;
 
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;
 
    virtual StatusCode getRoads(const std::vector<std::shared_ptr<const FPGATrackSimHit>> & hits, std::vector<std::shared_ptr<const FPGATrackSimRoad>> & roads) override;
    virtual int getSubRegion() const override{return -1;}
    //TODO mabye this should have a subRegion
 
private:
 
    Gaudi::Property <unsigned> m_imageSize_x { this, "nBins_x", 0, ""};
    Gaudi::Property <unsigned> m_imageSize_y { this, "nBins_y", 0, ""};
    Gaudi::Property <int> m_threshold { this, "m_threshold", 8, "Threshold to be allied to bins of abs(d0) > 50 (mm)"};
    Gaudi::Property <int> m_threshold50 { this, "m_threshold50", 8, "Threshold to be allied to bins of abs(d0) < (mm)"};
    Gaudi::Property <bool> m_traceHits { this, "traceHits", true, "Trace each hit that goes in a bin. Disabling this will save memory/time since each bin doesn't have to store all its hits but the roads created won't have hits from convolution, etc."};
    Gaudi::Property <float> m_d0_range { this, "d0_range", 120, "The d0 span: (-range, range) in mm"};
    Gaudi::Property <float> m_qOverPt_range { this, "qOverPt_range", 0.002, "The q/pT span: (-range, range) in e/MeV"};
    Gaudi::Property <bool> m_continuous { this, "continuous", true, "assure that there is continuity of the line (i.e. middle bins in d0 are filled when one q/pT step would result in a hole)"};
 
 
    typedef vector2D<std::pair<int, std::unordered_set<std::shared_ptr<const FPGATrackSimHit>>>> Image;
    std::vector<FPGATrackSimRoad> m_roads;
 
    const double m_acceptedDistanceBetweenLayersMin = 200; // min R disstance for hits pair filtering
    const double m_acceptedDistanceBetweenLayersMax = 600;
 
    double m_step_x = 0; // helpers (accumulator granularity)
    double m_step_y = 0;
 
    StatusCode fillImage(std::shared_ptr<const FPGATrackSimHit> &hit1, std::shared_ptr<const FPGATrackSimHit> &hit2, Image& image) const;
    bool passThreshold(Image const & image, int x, int y) const;
    bool isLocalMaxima(Image const & image, int x, int y) const;
 
    FPGATrackSimRoad createRoad(const std::unordered_set<std::shared_ptr<const FPGATrackSimHit>> & hits, unsigned x, unsigned y) const;
    unsigned m_event = 0;
    void drawImage(Image const & image) const;
 
    float xtod0(int x) const { return x*m_step_x - m_d0_range; }
    float ytoqoverpt(int y) const { return y*m_step_y - m_qOverPt_range; }
    int m_eventsProcessed = 0;
    int m_roadsGenerated = 0;
 
};
 
#endif //