GbtsSeedingTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSTRK_GBTSSEEDINGTOOL_SEEDINGTOOL_H
#define ACTSTRK_GBTSSEEDINGTOOL_SEEDINGTOOL_H 1
 
// ATHENA
#include "ActsToolInterfaces/ISeedingTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "InDetIdentifier/PixelID.h"
#include "PixelReadoutGeometry/PixelDetectorManager.h"
#include "ActsInterop/Logger.h"
#include "xAODInDetMeasurement/PixelClusterContainer.h"
 
// ACTS CORE
#include "Acts/Seeding2/GraphBasedTrackSeeder.hpp"
#include "Acts/Definitions/Units.hpp"
#include "Acts/EventData/SeedContainer2.hpp"
#include "Acts/EventData/SpacePointContainer2.hpp"
 
//for det elements, not sure which need: 
#include "TrigInDetToolInterfaces/ITrigL2LayerNumberTool.h"
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "AthenaMonitoringKernel/GenericMonitoringTool.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/ReadCondHandleKey.h"
 
#include "ActsToolInterfaces/IPixelSpacePointFormationTool.h"
 
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
 
#include "xAODInDetMeasurement/SpacePointContainer.h"
#include "xAODInDetMeasurement/SpacePointAuxContainer.h"
 
#include <numbers>
#include <memory>
 
namespace ActsTrk {
 
  class GbtsSeedingTool final: public extends<AthAlgTool, ActsTrk::ISeedingTool> {
 
  public:
 
    GbtsSeedingTool(const std::string& type, const std::string& name,
      const IInterface* parent);
 
    virtual ~GbtsSeedingTool() = default;
 
    virtual StatusCode initialize() override;
 
    // Interface
    StatusCode createSeeds(
      const EventContext& ctx,
      const std::vector<const xAOD::SpacePointContainer*>& spacePointCollections,
      const Eigen::Vector3f& beamSpotPos, float bFieldInZ,
      ActsTrk::SeedContainer& seedContainer) const override;
 
  private:

    void printGbtsConfig() const;

    StatusCode prepareConfiguration();

    const Acts::Logger &logger() const { return *m_logger; }

    std::unique_ptr<const Acts::Logger> m_logger;

    Acts::Experimental::GraphBasedTrackSeeder::Config m_finderCfg;

    Acts::Experimental::GbtsTrackingFilter::Config m_filterCfg;

    std::optional<Acts::Experimental::GraphBasedTrackSeeder> m_finder;

    std::optional<Acts::Experimental::GbtsTrackingFilter> m_filter;

    const std::vector<short>* m_sct_h2l = nullptr;
    const std::vector<short>* m_pix_h2l = nullptr;
    std::vector<bool> m_are_pixels;

    ToolHandle<ITrigL2LayerNumberTool> m_layerNumberTool {this, "layerNumberTool", "TrigL2LayerNumberToolITk"}; 
 
    // TODO: ADD NEW VERIABLES, DELETE OLD ONES AND CHANGE CURRENT ONES TO NEW VALUES
    // Config settings                                                              
    Gaudi::Property<std::string> m_connectorInputFile {this, "connectorInputFile","binTables_ITK_RUN4.txt", "input file for making connector object"};
    Gaudi::Property<std::string> m_lutFile {this, "lutInputFile", "gbts_ml_pixel_barrel_loose.lut", "file to LUT"}; 
    
    // SeedFinderGbts option
    Gaudi::Property<bool> m_LRTmode {this, "LRTmode", false, "whether strip or pixel hits are used"};
    Gaudi::Property<bool> m_useML {this, "useML", true, "use the cluster width of the spacepoint"};
    Gaudi::Property<bool> m_matchBeforeCreate {this, "matchBeforeCreate", true, "need to check what this does"};
    Gaudi::Property<bool> m_useOldTunings {this, "useOldTunings", false, "use the tunings for 900MeV cut"};
    Gaudi::Property<float> m_tau_ratio_cut {this, "cut_tau_ratio_max",0.007, "tau cut for doublets and triplets"}; 
    Gaudi::Property<float> m_tau_ratio_precut {this, "precut_tau_ratio_max",0.009f, "not sure"}; 
    Gaudi::Property<float> m_etaBinWidthOverride {this, "etaBinOverride", 0.0f, "apply custom binning for connections"};
    Gaudi::Property<float> m_nMaxPhiSlice {this, "nMaxPhiSlice",53, "used to calculate phi slices"}; 
    Gaudi::Property<bool> m_beamSpotCorrection {this, "beamSpotCorrection", true, "apply primary vertex corrections to spacepoints"};
    Gaudi::Property<float> m_minPt {this, "minPt", 1000.0, "Lower cutoff for seeds"};
    Gaudi::Property<float> m_phiSliceWidth {this, "phiSliceWidth",0, "initialised in loadSpacePoints function"};
    Gaudi::Property<float> m_tau_ratio_corr{this, "tau_ratio_correction", 0.006, "not sure"};
    Gaudi::Property<bool> m_validateTriplets{this, "ValidateTriplets", true, "not sure"};
    Gaudi::Property<bool> m_useAdaptiveCuts{this, "UseAdaptiveCuts", true, "not sure"};
 
    // BuildTheGraph() options
    Gaudi::Property<bool> m_useEtaBinning {this, "useEtaBinning",true, "bool to use eta binning from geometry structure"}; 
    Gaudi::Property<bool> m_doubletFilterRZ {this, "doubletFilterRZ",true, "bool applies new Z cuts on doublets"}; 
    Gaudi::Property<float> m_minDeltaRadius {this, "minDeltaRadius",2.0, " min dr for doublet"}; 
    Gaudi::Property<int> m_nMaxEdges {this, "MaxEdges",3000000, " max number of Gbts edges/doublets"};
 
    // triplet validation options
    Gaudi::Property<float> m_d0_max{this, "d0_max", 3.0, "not sure"};
 
    // GbtsTrackingFilter
    Gaudi::Property<float> m_sigmaMS {this, "sigmaMS", 0.016, "process noise from multiple scattering"};
    Gaudi::Property<float> m_radLen {this, "radLen", 0.025, "not sure"};
    Gaudi::Property<float> m_sigma_x {this, "sigma_x", 0.08, "measurement resolution for residual on y direction"}; //was 0.22
    Gaudi::Property<float> m_sigma_y {this, "sigma_y", 0.25, "measurement resolution on r-z plane"};// was 1.7
    Gaudi::Property<float> m_weight_x {this, "weight_x", 0.5, "penalty weight for track"};
    Gaudi::Property<float> m_weight_y {this, "weight_y", 0.5, "penalty weight for track"};
    Gaudi::Property<float> m_maxDChi2_x {this, "maxDChi2_x", 5.0, "gate threshold for chi2 test"}; //was 35.0
    Gaudi::Property<float> m_maxDChi2_y {this, "maxDChi2_y", 6.0, "gate threshold for chi2 test"};//was 31.0
    Gaudi::Property<float> m_add_hit {this, "add_hit", 14.0, "reward added to tracks for each accepted hit before chi2"};
    Gaudi::Property<float> m_max_curvature {this, "max_curvature", 1e-3f, "not sure"};
    Gaudi::Property<float> m_max_z0 {this, "max_z0", 170.0, "not sure"};
    Gaudi::Property<float> m_edge_mask_min_eta {this, "edge_mask_min_eta", 1.5, "not sure"};
    Gaudi::Property<float> m_hit_share_threshold {this, "hit_share_threshold", 0.49, "not sure"};
 
    // GbtsDataStorage
    Gaudi::Property<float> m_max_endcap_clusterwidth {this, "max_endcap_clusterwidth", 0.35, "discards any spacepoints which dr/dz cant be accurately modelled"};
  };
  
} // namespace
 
#endif