CutSelector.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - 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/findlets/base/Findlet.h>
 
#include <framework/core/ModuleParamList.h>
#include <tracking/trackFindingCDC/utilities/WeightedRelation.h>
#include <tracking/trackFindingCDC/utilities/StringManipulation.h>
 
#include <algorithm>
#include <cmath>
#include <string>
#include <vector>
 
namespace Belle2 {
  namespace TrackFindingCDC {
    template <class ACollectorItem, class ACollectionItem>
    class CutSelector :
      public Findlet<WeightedRelation<ACollectorItem, const ACollectionItem>> {
    public:
      using WeightedRelationItem = WeightedRelation<ACollectorItem, const ACollectionItem>;
 
      using Super = Findlet<WeightedRelation<ACollectorItem, const ACollectionItem>>;
 
      void exposeParameters(ModuleParamList* moduleParamList, const std::string& prefix) override
      {
        Super::exposeParameters(moduleParamList, prefix);
        moduleParamList->addParameter(prefixed(prefix, "cutValue"), m_param_cutValue,
                                      "Value to cut at.",
                                      m_param_cutValue);
      }
 
      void apply(std::vector<WeightedRelationItem>& weightedRelations) override
      {
        const auto& lessThanCutOrNaN = [this](const WeightedRelationItem & relationItem) {
          return std::isnan(relationItem.getWeight()) or relationItem.getWeight() < m_param_cutValue;
        };
 
        // Erase all items with a weight of NAN
        weightedRelations.erase(std::remove_if(weightedRelations.begin(),
                                               weightedRelations.end(),
                                               lessThanCutOrNaN),
                                weightedRelations.end());
      }
 
      void setCutValue(Weight cutValue)
      {
        m_param_cutValue = cutValue;
      }
 
    private:
      Weight m_param_cutValue = NAN;
    };
  }
}