AttractorsPartitionConfig.cpp

Go to the documentation of this file.

/* 
 * @file AttractorsPartitionConfig.cpp
 * @author nikoapos
 */
 
#include "SEImplementation/Configuration/AttractorsPartitionConfig.h"
#include "SEImplementation/Configuration/PartitionStepConfig.h"
#include "SEImplementation/Configuration/MinAreaPartitionConfig.h"
#include "SEImplementation/Configuration/DetectionFrameConfig.h"
 
#include "SEImplementation/Partition/AttractorsPartitionStep.h"
 
using namespace Euclid::Configuration;
namespace po = boost::program_options;
 
namespace SourceXtractor {
 
static const std::string USE_ATTRACTORS_PARTITION {"use-attractors-partition"};
 
AttractorsPartitionConfig::AttractorsPartitionConfig(long manager_id) : Configuration(manager_id) {
  declareDependency<PartitionStepConfig>();
  declareDependency<DetectionFrameConfig>();
 
  // this is used to enforce the order the PartitionSteps are added and performed
  ConfigManager::getInstance(manager_id).registerDependency<AttractorsPartitionConfig, MinAreaPartitionConfig>();
}
 
auto AttractorsPartitionConfig::getProgramOptions() -> std::map<std::string, OptionDescriptionList> {
  return { {"Extraction", {
      {USE_ATTRACTORS_PARTITION.c_str(), po::value<bool>()->default_value(false),
              "Enables the use of attractors for partitioning"}
  }}};
}
 
void AttractorsPartitionConfig::initialize(const UserValues& args) {
  if (getDependency<DetectionFrameConfig>().getDetectionFrames().size() > 0) {
    if (args.at(USE_ATTRACTORS_PARTITION).as<bool>()) {
      getDependency<PartitionStepConfig>().addPartitionStepCreator(
              [](std::shared_ptr<SourceFactory> source_factory) {
                return std::make_shared<AttractorsPartitionStep>(source_factory);
              }
      );
    }
  }
}
 
} // SourceXtractor namespace