GroupingFactory.cpp

Go to the documentation of this file.

 
#include "SEImplementation/Grouping/GroupingFactory.h"
 
#include "SEImplementation/Grouping/SplitSourcesGrouping.h"
#include "SEImplementation/Grouping/AssocGrouping.h"
#include "SEImplementation/Grouping/MoffatGrouping.h"
 
namespace SourceXtractor {
 
GroupingFactory::GroupingFactory(std::shared_ptr<SourceGroupFactory> source_group_factory)
  : m_algorithm(GroupingConfig::Algorithm::NO_GROUPING), m_source_group_factory(source_group_factory),
    m_hard_limit(0), m_moffat_max_distance(1000.0) {}
 
void GroupingFactory::reportConfigDependencies(Euclid::Configuration::ConfigManager& manager) const {
  manager.registerConfiguration<GroupingConfig>();
}
 
void GroupingFactory::configure(Euclid::Configuration::ConfigManager& manager)  {
  auto grouping_config = manager.getConfiguration<GroupingConfig>();
  m_algorithm = grouping_config.getAlgorithmOption();
  m_moffat_max_distance = grouping_config.getMoffatMaxDistance();
  switch (m_algorithm) {
    case GroupingConfig::Algorithm::NO_GROUPING:
      m_grouping_criteria = std::make_shared<NoGroupingCriteria>();
      break;
    case GroupingConfig::Algorithm::OVERLAPPING:
      m_grouping_criteria = std::make_shared<OverlappingBoundariesCriteria>();
      break;
    case GroupingConfig::Algorithm::SPLIT_SOURCES:
      m_grouping_criteria = std::make_shared<SplitSourcesCriteria>();
      break;
    case GroupingConfig::Algorithm::MOFFAT:
      m_grouping_criteria = std::make_shared<MoffatCriteria>(grouping_config.getMoffatThreshold(), m_moffat_max_distance);
      break;
    case GroupingConfig::Algorithm::ASSOC:
      m_grouping_criteria = std::make_shared<AssocCriteria>();
      break;
  }
  m_hard_limit = grouping_config.getHardLimit();
}
 
std::shared_ptr<SourceGroupingInterface> GroupingFactory::createGrouping() const {
  assert(m_grouping_criteria != nullptr);
  assert(m_source_group_factory != nullptr);
 
  // return optimized grouping if available, if not uses general grouping with criteria
  switch (m_algorithm) {
    case GroupingConfig::Algorithm::SPLIT_SOURCES:
      return std::make_shared<SplitSourcesGrouping>(m_source_group_factory, m_hard_limit);
    case GroupingConfig::Algorithm::ASSOC:
      return std::make_shared<AssocGrouping>(m_source_group_factory, m_hard_limit);
    case GroupingConfig::Algorithm::MOFFAT:
      return std::make_shared<MoffatGrouping>(
          m_grouping_criteria, m_source_group_factory, m_hard_limit, m_moffat_max_distance);
      break;
    default:
      return std::make_shared<SourceGrouping>(m_grouping_criteria, m_source_group_factory, m_hard_limit);
  }
}
 
} // SourceXtractor namespace