Scrooge.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/trackFindingVXD/trackSetEvaluator/OverlapResolverNodeInfo.h>
 
#include <vector>
#include <algorithm>
 
namespace Belle2 {
  class Scrooge {
  public:
    void performSelection(std::vector <OverlapResolverNodeInfo>& overlapResolverNodeInfo)
    {
      //sort the vector according to the QI supplied.
      std::sort(overlapResolverNodeInfo.begin(), overlapResolverNodeInfo.end(),
      [](OverlapResolverNodeInfo const & lhs, OverlapResolverNodeInfo const & rhs) -> bool {
        return lhs.qualityIndicator > rhs.qualityIndicator;
      });
 
      //kill all tracks, that have overlaps and lower QI:
      auto endOfOverlapResolverNodeInfo   = overlapResolverNodeInfo.cend();
      for (auto trackIter = overlapResolverNodeInfo.begin(); trackIter != endOfOverlapResolverNodeInfo; trackIter++) {
        if (!trackIter->activityState) continue;
        for (auto testTrackIter = trackIter; testTrackIter != endOfOverlapResolverNodeInfo; testTrackIter++) {
          if (std::find(trackIter->overlaps.begin(), trackIter->overlaps.end(), testTrackIter->trackIndex) !=
              trackIter->overlaps.end()) {
            testTrackIter->activityState = 0.;
          }
        }
      }
    }
  };
}