FilterMatching.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// Copyright (C) 2024 Simon Gardner
 
#pragma once
 
#include <spdlog/spdlog.h>
#include <algorithms/algorithm.h>
#include <string>
#include <string_view>
 
#include "services/log/Log_service.h"
#include "algorithms/interfaces/WithPodConfig.h"
 
namespace eicrecon {
 
template <class ToFilterObjectT, class FilterByObjectT>
using FilterMatchingAlgorithm =
    algorithms::Algorithm<typename algorithms::Input<typename ToFilterObjectT::collection_type,
                                                     typename FilterByObjectT::collection_type>,
                          typename algorithms::Output<typename ToFilterObjectT::collection_type,
                                                      typename ToFilterObjectT::collection_type>>;
 
template <typename ToFilterObjectT, auto ToFilterFunction, typename FilterByObjectT,
          auto FilterByFunction>
class FilterMatching : public FilterMatchingAlgorithm<ToFilterObjectT, FilterByObjectT>,
                       public WithPodConfig<NoConfig> {
 
public:
  FilterMatching(std::string_view name)
      : FilterMatchingAlgorithm<ToFilterObjectT, FilterByObjectT>{
            name,
            {"inputCollection", "inputMatchedCollection"},
            {"outputMatchedAssociations", "outputUnmatchedAssociations"},
            "Filter by matching to a collection"} {};
 
  void init() final {};
 
  void
  process(const typename FilterMatchingAlgorithm<ToFilterObjectT, FilterByObjectT>::Input& input,
          const typename FilterMatchingAlgorithm<ToFilterObjectT, ToFilterObjectT>::Output& output)
      const final {
 
    const auto [toFilterEntries, filterByEntries] = input;
    auto [is_matched, is_not_matched]             = output;
 
    is_matched->setSubsetCollection();
    is_not_matched->setSubsetCollection();
 
    for (const auto& matchedEntry : *toFilterEntries) {
 
      auto ref_value = ToFilterFunction(&matchedEntry);
 
      bool found_match = false;
 
      // Tries to find the association in the entries
      for (const auto& entry : *filterByEntries) {
        auto other_value = FilterByFunction(&entry);
        if (other_value == ref_value) {
          is_matched->push_back(matchedEntry);
          found_match = true;
          break;
        }
      }
 
      if (!found_match) {
        is_not_matched->push_back(matchedEntry);
      }
    }
  };
};
 
} // namespace eicrecon