AssocModeTaskFactory.cpp

Go to the documentation of this file.

#include "SEImplementation/Plugin/AssocMode/AssocModeTaskFactory.h"
#include "SEFramework/Property/PropertyId.h"
#include "SEFramework/Task/Task.h"
#include "SEImplementation/Plugin/AssocMode/AssocMode.h"
#include "SEImplementation/Plugin/AssocMode/AssocModeConfig.h"
#include "SEImplementation/Plugin/AssocMode/AssocModeTask.h"
#include "SEImplementation/Plugin/AssocMode/AssocModeDummyTask.h"
#include <NdArray/NdArray.h>
#include <iostream>
#include <sstream>

namespace SourceXtractor {

AssocModeTaskFactory::AssocModeTaskFactory() : m_assoc_mode(AssocModeConfig::AssocMode::UNKNOWN), m_assoc_radius(0.) {}

void AssocModeTaskFactory::reportConfigDependencies(Euclid::Configuration::ConfigManager& manager) const {
  manager.registerConfiguration<AssocModeConfig>();
}

void AssocModeTaskFactory::configure(Euclid::Configuration::ConfigManager& manager) {
  auto config = manager.getConfiguration<AssocModeConfig>();

  m_catalogs               = config.getCatalogs();
  m_assoc_radius           = config.getAssocRadius();
  m_assoc_mode             = config.getAssocMode();
  m_add_property_instances = !config.getColumnsIdx().empty();
}

std::shared_ptr<Task> AssocModeTaskFactory::createTask(const PropertyId& property_id) const {
  if (property_id.getTypeId() == typeid(AssocMode)) {
    if (m_catalogs.empty()) {
      return std::make_shared<AssocModeDummyTask>();
    } else {
      return std::make_shared<AssocModeTask>(m_catalogs, m_assoc_mode, m_assoc_radius);
    }
  } else {
    return nullptr;
  }
}

void AssocModeTaskFactory::registerPropertyInstances(OutputRegistry& registry) {
  using Euclid::NdArray::NdArray;

  if (!m_add_property_instances) {
    return;
  }
  registry.registerColumnConverter<AssocMode, NdArray<SeFloat>>(
      "assoc_values", [](const AssocMode& prop) { return prop.getAssocValues(); }, "", "Assoc catalog values");
}

}  // namespace SourceXtractor