MultiThresholdPartitionConfig.cpp

Go to the documentation of this file.

/*
 * MultiThresholdPartitionConfig.cpp
 *
 *  Created on: Jan 18, 2017
 *      Author: mschefer
 */
 
#include "SEImplementation/Configuration/MultiThresholdPartitionConfig.h"
#include "SEImplementation/Configuration/MinAreaPartitionConfig.h"
#include "SEImplementation/Configuration/PartitionStepConfig.h"
#include "SEImplementation/Configuration/RngConfig.h"
#include "SEImplementation/Configuration/DetectionFrameConfig.h"
 
#include "SEImplementation/Partition/MultiThresholdPartitionStep.h"
 
using namespace Euclid::Configuration;
namespace po = boost::program_options;
 
namespace SourceXtractor {
 
static const std::string MTHRESH_USE {"partition-multithreshold"};
static const std::string MTHRESH_THRESHOLDS_NB {"partition-threshold-count"};
static const std::string MTHRESH_MIN_AREA {"partition-minimum-area"};
static const std::string MTHRESH_MIN_CONTRAST {"partition-minimum-contrast"};
 
MultiThresholdPartitionConfig::MultiThresholdPartitionConfig(long manager_id) : Configuration(manager_id) {
  declareDependency<PartitionStepConfig>();
  declareDependency<RngConfig>();
  declareDependency<DetectionFrameConfig>();
 
  // this is used to enforce the order the PartitionSteps are added and performed
  ConfigManager::getInstance(manager_id).registerDependency<MultiThresholdPartitionConfig, MinAreaPartitionConfig>();
}
 
auto MultiThresholdPartitionConfig::getProgramOptions() -> std::map<std::string, OptionDescriptionList> {
  return { {"Multi-thresholding", {
      {MTHRESH_USE.c_str(), po::value<bool>()->default_value(true), "activates/deactivates multithreshold partitioning"},
      {MTHRESH_THRESHOLDS_NB.c_str(), po::value<int>()->default_value(32), "# of thresholds"},
      {MTHRESH_MIN_AREA.c_str(), po::value<int>()->default_value(3), "min area in pixels to consider partitioning"},
      {MTHRESH_MIN_CONTRAST.c_str(), po::value<double>()->default_value(0.005), "min contrast of for partitioning"}
  }}};
}
 
void MultiThresholdPartitionConfig::initialize(const UserValues& args) {
  // We can only use this if we have a detection image and it was on by default
  if (getDependency<DetectionFrameConfig>().getDetectionFrames().size() > 0) {
    if (args.at(MTHRESH_USE).as<bool>()) {
      auto threshold_nb = args.at(MTHRESH_THRESHOLDS_NB).as<int>();
      auto min_area = args.at(MTHRESH_MIN_AREA).as<int>();
      auto min_contrast = args.at(MTHRESH_MIN_CONTRAST).as<double>();
      auto seed = getDependency<RngConfig>().getSeed();
 
      if (min_area <= 0) {
          throw Elements::Exception() << "Invalid " << MTHRESH_MIN_AREA << " value: " << min_area;
      }
      if (threshold_nb <= 0) {
          throw Elements::Exception() << "Invalid " << MTHRESH_THRESHOLDS_NB << " value: " << threshold_nb;
      }
 
      getDependency<PartitionStepConfig>().addPartitionStepCreator(
        [=](std::shared_ptr<SourceFactory> source_factory) {
          return std::make_shared<MultiThresholdPartitionStep>(
              source_factory, min_contrast, threshold_nb, min_area, seed);
          });
    }
  }
}
 
} // SourceXtractor namespace