SVDStateVarSet.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2017 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Nils Braun                                               *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/varsets/VarSet.h>
#include <tracking/trackFindingCDC/varsets/VarNames.h>
#include <tracking/trackFindingCDC/varsets/FixedSizeNamedFloatTuple.h>
 
#include <tracking/ckf/svd/filters/states/BaseSVDStateFilter.h>
 
#include <tracking/trackFindingVXD/trackQualityEstimators/QualityEstimatorCircleFit.h>
#include <tracking/trackFindingVXD/trackQualityEstimators/QualityEstimatorRiemannHelixFit.h>
#include <tracking/trackFindingVXD/trackQualityEstimators/QualityEstimatorTripletFit.h>
 
namespace Belle2 {
  constexpr
  static char const* const svdStateVarNames[] = {
    "seed_cdc_hits",
    "seed_svd_hits",
    "seed_lowest_svd_layer",
    "seed_lowest_cdc_layer",
 
    "quality_index_triplet",
    "quality_index_circle",
    "quality_index_helix",
 
    "cluster_1_charge",
    "cluster_2_charge",
    "mean_rest_cluster_charge",
    "min_rest_cluster_charge",
    "std_rest_cluster_charge",
 
    "cluster_1_seed_charge",
    "cluster_2_seed_charge",
    "mean_rest_cluster_seed_charge",
    "min_rest_cluster_seed_charge",
    "std_rest_cluster_seed_charge",
 
    "cluster_1_size",
    "cluster_2_size",
    "mean_rest_cluster_size",
    "min_rest_cluster_size",
    "std_rest_cluster_size",
 
    "cluster_1_snr",
    "cluster_2_snr",
    "mean_rest_cluster_snr",
    "min_rest_cluster_snr",
    "std_rest_cluster_snr",
 
    "cluster_1_charge_over_size",
    "cluster_2_charge_over_size",
    "mean_rest_cluster_charge_over_size",
    "min_rest_cluster_charge_over_size",
    "std_rest_cluster_charge_over_size",
  };
 
  class SVDStateVarNames : public TrackFindingCDC::VarNames<BaseSVDStateFilter::Object> {
 
  public:
    static const size_t nVars = TrackFindingCDC::size(svdStateVarNames);
 
    constexpr
    static char const* getName(int iName)
    {
      return svdStateVarNames[iName];
    }
  };
 
  class SVDStateVarSet : public TrackFindingCDC::VarSet<SVDStateVarNames> {
 
  public:
    virtual bool extract(const BaseSVDStateFilter::Object* object) override;
 
  private:
    QualityEstimatorCircleFit m_qualityCircle;
    QualityEstimatorRiemannHelixFit m_qualityHelix;
    QualityEstimatorTripletFit m_qualityTriplet;
  };
}