BasicTrackVarSet.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/trackingUtilities/varsets/VarSet.h>
#include <tracking/trackingUtilities/varsets/VarNames.h>
 
// BOOST accumulators
#include <boost/accumulators/accumulators.hpp>
#include <boost/accumulators/statistics/count.hpp>
#include <boost/accumulators/statistics/min.hpp>
#include <boost/accumulators/statistics/max.hpp>
#include <boost/accumulators/statistics/sum.hpp>
#include <boost/accumulators/statistics/mean.hpp>
#include <boost/accumulators/statistics/variance.hpp>
 
namespace bacc = boost::accumulators;
 
namespace Belle2 {
  namespace TrackingUtilities {
    class CDCTrack;
  }
  namespace TrackFindingCDC {

    constexpr
    static char const* const basicTrackVarNames[] = {
      "pt",
      "size",
      "hits_per_layer",
 
      "sz_slope",
      "z0",
      "s_range",
      "avg_hit_dist",
      "has_matching_segment",
 
      "cont_layer_mean",
      "cont_layer_variance",
      "cont_layer_max",
      "cont_layer_min",
      "cont_layer_first",
      "cont_layer_last",
      "cont_layer_max_vs_last",
      "cont_layer_first_vs_min",
      "cont_layer_count",
      "cont_layer_occupancy",
 
      "super_layer_mean",
      "super_layer_variance",
      "super_layer_max",
      "super_layer_min",
      "super_layer_first",
      "super_layer_last",
      "super_layer_max_vs_last",
      "super_layer_first_vs_min",
      "super_layer_count",
      "super_layer_occupancy",
 
      "drift_length_mean",
      "drift_length_variance",
      "drift_length_max",
      "drift_length_min",
      "drift_length_sum",
 
      "adc_mean",
      "adc_variance",
      "adc_max",
      "adc_min",
      "adc_sum",
 
      "tot_mean",
      "tot_variance",
      "tot_max",
      "tot_min",
      "tot_sum",
 
      "empty_s_mean",
      "empty_s_variance",
      "empty_s_max",
      "empty_s_min",
      "empty_s_sum",
    };

    struct BasicTrackVarSetNames : public TrackingUtilities::VarNames<TrackingUtilities::CDCTrack> {

      // we shouldn't use public member variables but we do want to rewrite all related code using setters/getters
      // at least tell cppcheck that everything is fine
      // cppcheck-suppress duplInheritedMember
      static const size_t nVars = TrackingUtilities::size(basicTrackVarNames);

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

    class BasicTrackVarSet : public TrackingUtilities::VarSet<BasicTrackVarSetNames> {
 
    public:
      bool extract(const TrackingUtilities::CDCTrack* track) override;

      using statistics_set = bacc::features<bacc::tag::count,
            bacc::tag::sum,
            bacc::tag::min,
            bacc::tag::max,
            bacc::tag::mean,
            bacc::tag::lazy_variance>;

      using statistics_accumulator = bacc::accumulator_set<double, statistics_set>;
    };
  }
}