CollectionCollector.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 T>
using CollectionCollectorAlgorithm =
    algorithms::Algorithm<typename algorithms::Input<std::vector<const T>>,
                          typename algorithms::Output<T>>;
 
template <class T>
class CollectionCollector : public CollectionCollectorAlgorithm<T>, public WithPodConfig<NoConfig> {
 
public:
  CollectionCollector(std::string_view name)
      : CollectionCollectorAlgorithm<T>{name,
                                        {"inputCollections"},
                                        {"outputCollection"},
                                        "Merge content of collections into one subset collection"} {
  }
 
  void init() final {};
 
  void process(const typename CollectionCollector::Input& input,
               const typename CollectionCollector::Output& output) const final {
 
    const auto [in_collections] = input;
    auto [out_collection]       = output;
 
    out_collection->setSubsetCollection();
 
    for (const auto& collection : in_collections) {
      for (const auto& hit : *collection) {
        out_collection->push_back(hit);
      }
    }
    //Log how many hits were collected from N input collections
    this->debug("Collected {} hits from {} input collections", out_collection->size(),
                in_collections.size());
  }
};
} // namespace eicrecon