WeightImageConfig.h

Go to the documentation of this file.

/*
 * WeightImageConfig.h
 *
 *  Created on: Oct 7, 2016
 *      Author: mschefer
 */

#ifndef _SEIMPLEMENTATION_CONFIGURATION_WEIGHTIMAGECONFIG_H_
#define _SEIMPLEMENTATION_CONFIGURATION_WEIGHTIMAGECONFIG_H_

#include "Configuration/Configuration.h"
#include "SEFramework/Image/Image.h"

namespace SourceXtractor {

class WeightImageConfig : public Euclid::Configuration::Configuration {

public:

  enum class WeightType {
    WEIGHT_TYPE_NONE,
    WEIGHT_TYPE_FROM_BACKGROUND,
    WEIGHT_TYPE_RMS,
    WEIGHT_TYPE_VARIANCE,
    WEIGHT_TYPE_WEIGHT,
  };

  virtual ~WeightImageConfig() = default;
  explicit WeightImageConfig(long manager_id);

  std::map<std::string, Configuration::OptionDescriptionList> getProgramOptions() override;
  void initialize(const UserValues& args) override;

  std::shared_ptr<WeightImage> getWeightImage(size_t index = 0) const {
    try {
      return m_weight_images.at(index);
    } catch(...) {
      return nullptr;
    }
  }

  WeightType getWeightType() const {
    return m_weight_type;
  }

  bool isWeightAbsolute() const {
    return m_absolute_weight;
  }

  WeightImage::PixelType getWeightThreshold(size_t index = 0) const {
    try {
      return m_scaled_weight_thresholds.at(index);
    } catch(...) {
      return m_weight_threshold;
    }
  }

  bool symmetryUsage() const {
    return m_symmetry_usage;
  }

  static std::shared_ptr<WeightImage> convertWeightMap(std::shared_ptr<WeightImage> weight_image, WeightType weight_type, WeightImage::PixelType scaling = 1);

private:

  std::vector<std::shared_ptr<WeightImage>> m_weight_images;
  std::vector<WeightImage::PixelType> m_scaled_weight_thresholds;

  WeightType m_weight_type;
  bool m_absolute_weight;
  WeightImage::PixelType m_weight_scaling;
  WeightImage::PixelType m_weight_threshold;
  bool m_symmetry_usage;
};


}


#endif /* _SEIMPLEMENTATION_CONFIGURATION_WEIGHTIMAGECONFIG_H_ */