CMatchBookKeeper.cxx

Go to the documentation of this file.

#include "larreco/RecoAlg/CMTool/CMToolBase/CMatchBookKeeper.h"

namespace cmtool {

  CMatchBookKeeper::CMatchBookKeeper()
  {
    Reset();
  }

  void CMatchBookKeeper::Reset()
  {
    _register.clear();
  }

  void CMatchBookKeeper::Match(const std::vector<unsigned int>& matched_indexes, const float& score)
  {
    _register.insert(std::make_pair(score, matched_indexes));
  }

  std::vector<std::vector<unsigned int>> CMatchBookKeeper::GetResult() const
  {
    std::vector<std::vector<unsigned int>> res;

    PassResult(res);

    return res;
  }

  void CMatchBookKeeper::PassResult(std::vector<std::vector<unsigned int>>& result) const
  {
    result.clear();

    // Rough guess: assume half of registered pairs are good
    result.reserve((unsigned int)(_register.size() / 2));

    std::vector<bool> used_index;

    for (auto riter = _register.rbegin(); riter != _register.rend(); ++riter) {

      bool valid_set = true;

      for (auto const& index : (*riter).second) {

        if (index >= used_index.size())

          used_index.resize(index + 1, false);

        else if (used_index.at(index))
          valid_set = false;
      }

      if (valid_set) {

        result.push_back((*riter).second);

        for (auto& index : (*riter).second)

          used_index.at(index) = true;
      }
    }
  }

}