PXDStateBasicVarSet.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
#include <tracking/trackFindingCDC/varsets/VarSet.h>
#include <tracking/trackFindingCDC/varsets/VarNames.h>
#include <tracking/trackFindingCDC/varsets/FixedSizeNamedFloatTuple.h>
 
#include <tracking/ckf/pxd/filters/states/BasePXDStateFilter.h>
#include <tracking/ckf/pxd/utilities/PXDKalmanStepper.h>
 
namespace Belle2 {
  constexpr
  static char const* const pxdStateBasicVarNames[] = {
    "distance",
    "xy_distance",
    "z_distance",
 
    "mSoP_distance",
    "mSoP_xy_distance",
    "mSoP_z_distance",
 
    "same_hemisphere",
 
    "layer",
    "number",
 
    "arcLengthOfHitPosition",
    "arcLengthOfCenterPosition",
 
    "pt",
    "tan_lambda",
    "phi",
 
    "sigma_uv",
    "residual_over_sigma",
 
    "residual",
 
    "chi2",
 
    "cluster_charge",
    "cluster_seed_charge",
    "cluster_size",
    "cluster_size_u",
    "cluster_size_v",
 
    "ladder",
    "sensor",
    "segment",
    "id",
 
    "last_layer",
    "last_ladder",
    "last_sensor",
    "last_segment",
    "last_id",
 
    "last_cluster_charge",
    "last_cluster_seed_charge",
    "last_cluster_size",
    "last_cluster_size_u",
    "last_cluster_size_v",
  };

  class PXDStateBasicVarNames : public TrackFindingCDC::VarNames<BasePXDStateFilter::Object> {
 
  public:
    static const size_t nVars = TrackFindingCDC::size(pxdStateBasicVarNames);

    constexpr
    static char const* getName(int iName)
    {
      return pxdStateBasicVarNames[iName];
    }
  };

  class PXDStateBasicVarSet : public TrackFindingCDC::VarSet<PXDStateBasicVarNames> {
 
  public:
    virtual bool extract(const BasePXDStateFilter::Object* pair) override;
 
  private:
    PXDKalmanStepper m_kalmanStepper;
  };

}