FlexibleModelFittingModel.cpp

Go to the documentation of this file.

/*
 * FlexibleModelFittingModel.cpp
 *
 *  Created on: Oct 9, 2018
 *      Author: mschefer
 */
 
#include "AlexandriaKernel/memory_tools.h"
 
#include "ModelFitting/Models/TransformedModel.h"
 
#include "ModelFitting/Parameters/ManualParameter.h"
#include "ModelFitting/Parameters/EngineParameter.h"
#include "ModelFitting/Parameters/DependentParameter.h"
#include "ModelFitting/Engine/EngineParameterManager.h"
 
#include "ModelFitting/Models/OnlySmooth.h"
#include "ModelFitting/Models/OldSharp.h"
#include "ModelFitting/Models/AutoSharp.h"
 
#include "ModelFitting/Models/CircularlySymmetricModelComponent.h"
 
#include "ModelFitting/Parameters/ExpSigmoidConverter.h"
#include "ModelFitting/Parameters/SigmoidConverter.h"
 
#include "ModelFitting/Models/CompactExponentialModel.h"
#include "ModelFitting/Models/CompactSersicModel.h"
 
#include "SEImplementation/Image/ImageInterfaceTraits.h"
 
#include "SEImplementation/Plugin/FlexibleModelFitting/FlexibleModelFittingParameter.h"
#include "SEImplementation/Plugin/FlexibleModelFitting/FlexibleModelFittingParameterManager.h"
 
#include "Configuration/ConfigManager.h"
#include "SEImplementation/Configuration/SamplingConfig.h"
 
#ifdef WITH_ONNX_MODELS
#include "SEImplementation/Plugin/FlexibleModelFitting/OnnxCompactModel.h"
#endif
 
#include "SEImplementation/Plugin/FlexibleModelFitting/FlexibleModelFittingModel.h"
 
namespace SourceXtractor {
 
using namespace ModelFitting;
using Euclid::make_unique;
using namespace Euclid::Configuration;
 
static const double MODEL_MIN_SIZE = 4.0;
static const double MODEL_SIZE_FACTOR = 1.2;
 
// Reference for Sersic related quantities:
// See https://ned.ipac.caltech.edu/level5/March05/Graham/Graham2.html
 
 
// Note about pixel coordinates:
 
// The model fitting is made in pixel coordinates of the detection image, or if no detection image is used,
// pixel coordinate of the "reference" image, which is usually the first measurement image
 
// Internally we use a coordinate system with (0,0) at the center of the first pixel. But for compatibility with
// SExtractor 2, all pixel coordinates visible to the end user need to follow the FITS convention of (1,1) being the
// center of the coordinate system.
 
// The ModelFitting module uses the more common standard of (0, 0) being the corner of the first pixel.
 
// So we first convert the Python parameter to our internal coordinates, then do the transformation of coordinate,
// subtract the offset to the image cut-out and shift the result by 0.5 pixels
 
 
void FlexibleModelFittingPointModel::addForSource(FlexibleModelFittingParameterManager& manager,
                                         const SourceInterface& source,
                                         std::vector<ModelFitting::ConstantModel>& /* constant_models */,
                                         std::vector<ModelFitting::PointModel>& point_models,
                                         std::vector<std::shared_ptr<ModelFitting::ExtendedModel<ImageInterfaceTypePtr>>>&,
                                         double /* model_base_size */,
                                         std::tuple<double, double, double, double> /*jacobian*/,
                                         std::shared_ptr<CoordinateSystem> reference_coordinates,
                                         std::shared_ptr<CoordinateSystem> coordinates, PixelCoordinate offset) const  {
 
  auto pixel_x = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_x - offset.m_x + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
 
  auto pixel_y = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_y - offset.m_y + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
  point_models.emplace_back(pixel_x, pixel_y, manager.getParameter(source, m_flux));
}
 
void FlexibleModelFittingExponentialModel::addForSource(FlexibleModelFittingParameterManager& manager,
                          const SourceInterface& source,
                          std::vector<ModelFitting::ConstantModel>& /* constant_models */,
                          std::vector<ModelFitting::PointModel>& /*point_models*/,
                          std::vector<std::shared_ptr<ModelFitting::ExtendedModel<ImageInterfaceTypePtr>>>& extended_models,
                          double model_base_size,
                          std::tuple<double, double, double, double> jacobian,
                          std::shared_ptr<CoordinateSystem> reference_coordinates,
                          std::shared_ptr<CoordinateSystem> coordinates, PixelCoordinate offset) const {
 
  auto pixel_x = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_x - offset.m_x + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
 
  auto pixel_y = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_y - offset.m_y + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
  auto x_scale = std::make_shared<ManualParameter>(1); // we don't scale the x coordinate
 
  auto i0 = createDependentParameter(
     [](double flux, double radius, double aspect) { return flux / (2 * M_PI * 0.35513 * radius * radius * aspect); },
     manager.getParameter(source, m_flux), manager.getParameter(source, m_effective_radius),
     manager.getParameter(source, m_aspect_ratio));
 
  auto k = createDependentParameter(
      [](double eff_radius) { return 1.678 / eff_radius; },
      manager.getParameter(source, m_effective_radius));
 
  int size = std::max(MODEL_MIN_SIZE, MODEL_SIZE_FACTOR * model_base_size);
  extended_models.emplace_back(std::make_shared<CompactExponentialModel<ImageInterfaceTypePtr>>(
      2.0, i0, k, x_scale, manager.getParameter(source, m_aspect_ratio), manager.getParameter(source, m_angle),
      size, size, pixel_x, pixel_y, manager.getParameter(source, m_flux), jacobian));
}
 
void FlexibleModelFittingDevaucouleursModel::addForSource(FlexibleModelFittingParameterManager& manager,
                          const SourceInterface& source,
                          std::vector<ModelFitting::ConstantModel>& /* constant_models */,
                          std::vector<ModelFitting::PointModel>& /*point_models*/,
                          std::vector<std::shared_ptr<ModelFitting::ExtendedModel<ImageInterfaceTypePtr>>>& extended_models,
                          double model_base_size,
                          std::tuple<double, double, double, double> jacobian,
                          std::shared_ptr<CoordinateSystem> reference_coordinates,
                          std::shared_ptr<CoordinateSystem> coordinates, PixelCoordinate offset) const {
 
  auto pixel_x = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_x - offset.m_x + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
 
  auto pixel_y = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_y - offset.m_y + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
  auto n = std::make_shared<ManualParameter>(4); // Sersic index for Devaucouleurs
  auto x_scale = std::make_shared<ManualParameter>(1); // we don't scale the x coordinate
 
  auto i0 = createDependentParameter(
     [](double flux, double radius, double aspect) { return flux / (2 * M_PI * 0.001684925 * radius * radius * aspect); },
     manager.getParameter(source, m_flux), manager.getParameter(source, m_effective_radius),
     manager.getParameter(source, m_aspect_ratio));
 
  auto k = createDependentParameter(
      [](double eff_radius) { return 7.669 / pow(eff_radius, .25); },
      manager.getParameter(source, m_effective_radius));
 
  int size = std::max(MODEL_MIN_SIZE, MODEL_SIZE_FACTOR * model_base_size);
  extended_models.emplace_back(std::make_shared<CompactSersicModel<ImageInterfaceTypePtr>>(
      3.0, i0, k, n, x_scale, manager.getParameter(source, m_aspect_ratio), manager.getParameter(source, m_angle),
      size, size, pixel_x, pixel_y, manager.getParameter(source, m_flux), jacobian));
}
 
static double computeBn(double n) {
  // Using approximation from MacArthur, L.A., Courteau, S., & Holtzman, J.A. 2003, ApJ, 582, 689
 
  // The approximation only works for n >= 0.36, so we clamp the value to avoid numerical problems
  n = std::max(0.36, n);
 
  return 2 * n - 1.0 / 3.0 + 4 / (405 * n)
      + 46 / (25515 * n * n) + 131 / (1148175 * n * n * n) - 2194697 / (30690717750 * n * n * n * n);
}
 
void FlexibleModelFittingSersicModel::addForSource(FlexibleModelFittingParameterManager& manager,
                          const SourceInterface& source,
                          std::vector<ModelFitting::ConstantModel>& /* constant_models */,
                          std::vector<ModelFitting::PointModel>& /*point_models*/,
                          std::vector<std::shared_ptr<ModelFitting::ExtendedModel<ImageInterfaceTypePtr>>>& extended_models,
                          double model_base_size,
                          std::tuple<double, double, double, double> jacobian,
                          std::shared_ptr<CoordinateSystem> reference_coordinates,
                          std::shared_ptr<CoordinateSystem> coordinates, PixelCoordinate offset) const {
  auto pixel_x = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_x - offset.m_x + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
  auto pixel_y = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_y - offset.m_y + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
  auto x_scale = std::make_shared<ManualParameter>(1); // we don't scale the x coordinate
 
  auto i0 = createDependentParameter(
      [](double flux, double radius, double aspect, double n) { return flux / (2 * M_PI * pow(computeBn(n), -2*n) * n * std::tgamma(2*n) * radius * radius *  aspect); },
     manager.getParameter(source, m_flux), manager.getParameter(source, m_effective_radius),
     manager.getParameter(source, m_aspect_ratio), manager.getParameter(source, m_sersic_index));
 
  auto k = createDependentParameter(
      [](double eff_radius, double n) { return computeBn(n) / pow(eff_radius, 1.0 / n); },
      manager.getParameter(source, m_effective_radius), manager.getParameter(source, m_sersic_index));
 
  int size = std::max(MODEL_MIN_SIZE, MODEL_SIZE_FACTOR * model_base_size);
  extended_models.emplace_back(std::make_shared<CompactSersicModel<ImageInterfaceTypePtr>>(
      3.0, i0, k, manager.getParameter(source, m_sersic_index), x_scale, manager.getParameter(source, m_aspect_ratio),
      manager.getParameter(source, m_angle), size, size, pixel_x, pixel_y, manager.getParameter(source, m_flux), jacobian));
}
 
void FlexibleModelFittingConstantModel::addForSource(FlexibleModelFittingParameterManager& manager,
                          const SourceInterface& source,
                          std::vector<ModelFitting::ConstantModel>& constant_models,
                          std::vector<ModelFitting::PointModel>& /* point_models */,
                          std::vector<std::shared_ptr<ModelFitting::ExtendedModel<ImageInterfaceTypePtr>>>&,
                          double /* model_base_size */,
                          std::tuple<double, double, double, double> /* jacobian */,
                          std::shared_ptr<CoordinateSystem> /* reference_coordinates */,
                          std::shared_ptr<CoordinateSystem> /* coordinates */, PixelCoordinate /* offset */) const {
  constant_models.emplace_back(manager.getParameter(source, m_value));
}
 
#ifdef WITH_ONNX_MODELS
 
void FlexibleModelFittingOnnxModel::addForSource(FlexibleModelFittingParameterManager& manager,
                          const SourceInterface& source,
                          std::vector<ModelFitting::ConstantModel>& /* constant_models */,
                          std::vector<ModelFitting::PointModel>& /*point_models*/,
                          std::vector<std::shared_ptr<ModelFitting::ExtendedModel<ImageInterfaceTypePtr>>>& extended_models,
                          double model_base_size,
                          std::tuple<double, double, double, double> jacobian,
                          std::shared_ptr<CoordinateSystem> reference_coordinates,
                          std::shared_ptr<CoordinateSystem> coordinates, PixelCoordinate offset) const {
 
  auto pixel_x = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_x - offset.m_x + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
 
  auto pixel_y = createDependentParameter(
      [reference_coordinates, coordinates, offset](double x, double y) {
        return coordinates->worldToImage(reference_coordinates->imageToWorld(ImageCoordinate(x-1, y-1))).m_y - offset.m_y + 0.5;
      }, manager.getParameter(source, m_x), manager.getParameter(source, m_y));
 
  auto y_scale = createDependentParameter(
      [](double scale, double ratio) {
        return scale * ratio;
      }, manager.getParameter(source, m_scale), manager.getParameter(source, m_aspect_ratio));
 
  std::map<std::string, std::shared_ptr<BasicParameter>> params;
  for (auto it : m_params) {
    params[it.first] = manager.getParameter(source, it.second);
  }
 
  int size = std::max(MODEL_MIN_SIZE, MODEL_SIZE_FACTOR * model_base_size);
  extended_models.emplace_back(std::make_shared<OnnxCompactModel<ImageInterfaceTypePtr>>(m_models,
      manager.getParameter(source, m_scale), y_scale, manager.getParameter(source, m_angle),
      size, size, pixel_x, pixel_y, manager.getParameter(source, m_flux), params, jacobian));
}
 
FlexibleModelFittingOnnxModel::FlexibleModelFittingOnnxModel(
    std::vector<std::shared_ptr<OnnxModel>> models,
    std::shared_ptr<FlexibleModelFittingParameter> x,
    std::shared_ptr<FlexibleModelFittingParameter> y,
    std::shared_ptr<FlexibleModelFittingParameter> flux,
    std::shared_ptr<FlexibleModelFittingParameter> aspect_ratio,
    std::shared_ptr<FlexibleModelFittingParameter> angle,
    std::shared_ptr<FlexibleModelFittingParameter> scale,
    std::map<std::string, std::shared_ptr<FlexibleModelFittingParameter>> params)
        : m_x(x),
          m_y(y),
          m_flux(flux),
          m_aspect_ratio(aspect_ratio),
          m_angle(angle),
          m_scale(scale),
          m_params(params),
          m_models(models){
 
  std::sort(m_models.begin(), m_models.end(),
      [](const std::shared_ptr<OnnxModel>& a, const std::shared_ptr<OnnxModel>& b) -> bool {
          return a->getOutputShape()[2] < b->getOutputShape()[2];
      });
}
 
#endif
 
}