CATFinderParameters.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
/* ROOT headers. */
#include <TObject.h>
 
/* C++ headers. */
#include <cstdint>
 
namespace Belle2 {

  class CATFinderParameters : public TObject {
 
  public:

    CATFinderParameters() = default;

    ~CATFinderParameters() = default;

    void setTDCOffset(float tdcOffset) { m_tdcOffset = tdcOffset; }

    void setTDCScale(float tdcScale) { m_tdcScale = tdcScale; }

    void setADCClip(float adcClip) { m_adcClip = adcClip; }

    void setSLayerScale(float slayerScale) { m_slayerScale = slayerScale; }

    void setCLayerScale(float clayerScale) { m_clayerScale = clayerScale; }

    void setLayerScale(float layerScale) { m_layerScale = layerScale; }

    void setSpatialCoordinatesScale(float spatialCoordinatesScale) { m_spatialCoordinatesScale = spatialCoordinatesScale; }

    void setNInputFeatures(unsigned int nInputFeatures) { m_nInputFeatures = nInputFeatures; }

    void setLatentSpaceNDim(unsigned int latentSpaceNDim) { m_latentSpaceNDim = latentSpaceNDim; }

    void setTBeta(float tBeta) { m_tBeta = tBeta; }

    void setTDistance(float tDistance) { m_tDistance = tDistance; }

    void setMaxRadius(float maxRadius) { m_maxRadius = maxRadius; }

    void setMinNumberHits(unsigned int minNumberHits) { m_minNumberHits = minNumberHits; }

    void setInputTFeaturesName(const std::string& inputTFeaturesName) { m_inputTFeaturesName = inputTFeaturesName; }

    void setOutputTBetaName(const std::string& outputTBetaName) { m_outputTBetaName = outputTBetaName; }

    void setOutputTCoordinatesName(const std::string& outputTCoordinatesName) { m_outputTCoordinatesName = outputTCoordinatesName; }

    void setOutputTMomentumName(const std::string& outputTMomentumName) { m_outputTMomentumName = outputTMomentumName; }

    void setOutputTVertexName(const std::string& outputTVertexName) { m_outputTVertexName = outputTVertexName; }

    void setOutputTChargeName(const std::string& outputTChargeName) { m_outputTChargeName = outputTChargeName; }

    float getTDCOffset() const { return m_tdcOffset; }

    float getTDCScale() const { return m_tdcScale; }

    float getADCClip() const { return m_adcClip; }

    float getSLayerScale() const { return m_slayerScale; }

    float getCLayerScale() const { return m_clayerScale; }

    float getLayerScale() const { return m_layerScale; }

    float getSpatialCoordinatesScale() const { return m_spatialCoordinatesScale; }

    unsigned int getNInputFeatures() const { return m_nInputFeatures; }

    unsigned int getLatentSpaceNDim() const { return m_latentSpaceNDim; }

    float getTBeta() const { return m_tBeta; }

    float getTDistance() const { return m_tDistance; }

    float getMaxRadius() const { return m_maxRadius; }

    unsigned int getMinNumberHits() const { return m_minNumberHits; }

    const std::string& getInputTFeaturesName() const { return m_inputTFeaturesName; }

    const std::string& getOutputTBetaName() const { return m_outputTBetaName; }

    const std::string& getOutputTCoordinatesName() const { return m_outputTCoordinatesName; }

    const std::string& getOutputTMomentumName() const { return m_outputTMomentumName; }

    const std::string& getOutputTVertexName() const { return m_outputTVertexName; }

    const std::string& getOutputTChargeName() const { return m_outputTChargeName; }
 
  private:
 
    // Input feature normalization

    float m_tdcOffset = 0.;

    float m_tdcScale = 0.;

    float m_adcClip = 0.;

    float m_slayerScale = 0.;

    float m_clayerScale = 0.;

    float m_layerScale = 0.;

    float m_spatialCoordinatesScale = 0.;
 
    // GNN architecture

    unsigned int m_nInputFeatures = 0;

    unsigned int m_latentSpaceNDim = 0;
 
    // Object condensation

    float m_tBeta = 0.;

    float m_tDistance = 0.;

    float m_maxRadius = 0.;
 
    // Track selection

    unsigned int m_minNumberHits = 0;
 
    // Tensor names

    std::string m_inputTFeaturesName;

    std::string m_outputTBetaName;

    std::string m_outputTCoordinatesName;

    std::string m_outputTMomentumName;

    std::string m_outputTVertexName;

    std::string m_outputTChargeName;

    ClassDef(CATFinderParameters, 1);
 
  };

}