SplitSourcesGrouping.cpp

Go to the documentation of this file.

 
#include "SEImplementation/Grouping/SplitSourcesGrouping.h"
#include "SEImplementation/Property/SourceId.h"
 
namespace SourceXtractor {
 
SplitSourcesGrouping::SplitSourcesGrouping(std::shared_ptr<SourceGroupFactory> group_factory, unsigned int hard_limit)
    : m_group_factory(group_factory), m_hard_limit(hard_limit)
{
}
 
std::set<PropertyId> SplitSourcesGrouping::requiredProperties() const {
  return {};
}

void SplitSourcesGrouping::receiveSource(std::unique_ptr<SourceInterface> source) {
  auto source_id = source->getProperty<SourceId>().getDetectionId();
 
  if (m_source_groups.find(source_id) == m_source_groups.end()) {
    auto new_group = m_group_factory->createSourceGroup();
    m_source_groups[source_id] = std::move(new_group);
  }
 
  if (m_hard_limit > 0 && m_source_groups.at(source_id)->size() >= m_hard_limit) {
      // the stored group has reached the hard limit
      // send the current group to processing
      sendSource(std::move(m_source_groups.at(source_id)));
 
      // and replace it with a new empty one
      auto new_group = m_group_factory->createSourceGroup();
      m_source_groups[source_id] = std::move(new_group);
    }
 
  m_source_groups.at(source_id)->addSource(std::move(source));
}

void SplitSourcesGrouping::receiveProcessSignal(const ProcessSourcesEvent& event) {
  std::vector<unsigned int> groups_to_process;
 
  // We iterate through all the SourceGroups we have
  for (auto const& it : m_source_groups) {
    // We look at its Sources and if we find at least one that needs to be processed
    for (auto& source : *it.second) {
      if (event.m_selection_criteria->mustBeProcessed(source)) {
        groups_to_process.push_back(it.first);
        break;
      }
    }
  }
 
  // For each SourceGroup that we put in groups_to_process,
  for (auto group_id : groups_to_process) {
    // we remove it from our list of stored SourceGroups and notify our observers
    sendSource(std::move(m_source_groups[group_id]));
    m_source_groups.erase(group_id);
  }
}
 
} // SourceXtractor namespace