FPGATrackSimNNTrackTool.h

Go to the documentation of this file.

// Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
 
#ifndef FPGATrackSimNNTRACKTOOL_H
#define FPGATrackSimNNTRACKTOOL_H
 
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "FPGATrackSimObjects/FPGATrackSimRoad.h"
#include "FPGATrackSimObjects/FPGATrackSimHit.h"
#include "FPGATrackSimObjects/FPGATrackSimTrack.h"
#include "FPGATrackSimObjects/FPGATrackSimMultiTruth.h"
#include "FPGATrackSimObjects/FPGATrackSimTruthTrack.h"
#include "FPGATrackSimBanks/FPGATrackSimSectorBank.h"
 
#include "GaudiKernel/ITHistSvc.h"
 
#include "FPGATrackSimMaps/IFPGATrackSimMappingSvc.h"
#include "FPGATrackSimBanks/IFPGATrackSimBankSvc.h"
#include "FPGATrackSimMaps/FPGATrackSimPlaneMap.h"
 
#include "lwtnn/LightweightGraph.hh"
#include "lwtnn/parse_json.hh"
 
 
class IFPGATrackSimMappingSvc;
 
class FPGATrackSimNNTrackTool : public AthAlgTool
{
  public:
 
        // AthAlgTool
 
        FPGATrackSimNNTrackTool(const std::string&, const std::string&, const IInterface*);
 
        virtual StatusCode initialize() override;
    StatusCode getTracks(const std::vector<std::shared_ptr<const FPGATrackSimRoad>> &roads, std::vector<FPGATrackSimTrack> &tracks, 
                 const FPGATrackSimNNMap *nnMap);
 
        static float getXScale() { return 1015;};
        static float getYScale() { return 1015;};
        static float getZScale() { return 3000;};
 
    // Flags
 
    Gaudi::Property <double> m_NNCut { this, "NNCut", 0.0, " NN output value to cut on when selecting good tracks"};
    Gaudi::Property <double> m_chi2_scalefactor { this, "Chi2ScaleFactor", 40/(1-0.1), "Scale factor to use in converting to a chi2, Nominal chi2ndof cut is 40 and we want to use NN>0.0075 (or NN<(1-0.0075)"};
 
 
    private:
 
        ServiceHandle<IFPGATrackSimMappingSvc>   m_FPGATrackSimMapping{this, "FPGATrackSimMappingSvc","FPGATrackSimMappingSvc"};
        ServiceHandle<ITHistSvc> m_tHistSvc{this, "THistSvc","THistSvc"};
 
    std::vector<float> m_x; // x position of hit in road
        std::vector<float> m_y; // y pos
        std::vector<float> m_z; // z pos
    std::vector<float> m_barcodefrac; // truth barcode fraction for the hit
    std::vector<int> m_barcode; // truth barcode for the hit
    std::vector<int> m_eventindex; // event index for the hit
    std::vector<unsigned int> m_isPixel; // is hit pixel? if 0 it is strip
    std::vector<unsigned int> m_layer; // layer ID
    std::vector<unsigned int> m_isBarrel; // is hit in barrel? if 0 it is endcap
    std::vector<unsigned int> m_etawidth;
    std::vector<unsigned int> m_phiwidth;
    std::vector<unsigned int> m_etamodule;
    std::vector<unsigned int> m_phimodule;
    std::vector<unsigned int> m_ID; // ID hash for hit
 
    std::vector<float> m_truth_d0;
    std::vector<float> m_truth_z0;
    std::vector<float> m_truth_pt;
    std::vector<float> m_truth_eta;
    std::vector<float> m_truth_phi;
    std::vector<float> m_truth_pdg;
    std::vector<int> m_truth_q;
    std::vector<int> m_truth_barcode;
    std::vector<int> m_truth_eventindex;
 
    // NN stuff
        std::vector<const char*> m_input_node_names;
        std::vector<int64_t> m_input_node_dims;
        std::vector<const char*> m_output_node_names;
 
    void compute_truth(FPGATrackSimTrack & newtrk) const;
 
};
 
 
#endif // FPGATrackSimNNTRACKTOOL_H