PXDStateBasicVarSet.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/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",
 
    "ladder",
    "sensor",
    "segment",
    "id",
 
    "last_layer",
    "last_ladder",
    "last_sensor",
    "last_segment",
    "last_id",
  };
 
  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;
  };
}