FlexibleModelFittingTask.h

Go to the documentation of this file.

/*
 * FlexibleModelFittingTask.h
 *
 *  Created on: Sep 17, 2018
 *      Author: mschefer
 */
 
#ifndef _SEIMPLEMENTATION_PLUGIN_FLEXIBLEMODELFITTING_FLEXIBLEMODELFITTINGTASK_H_
#define _SEIMPLEMENTATION_PLUGIN_FLEXIBLEMODELFITTING_FLEXIBLEMODELFITTINGTASK_H_
 
#include "ModelFitting/Models/FrameModel.h"
 
#include "SEImplementation/Image/ImagePsf.h"
 
#include "SEFramework/Image/VectorImage.h"
#include "SEFramework/Task/GroupTask.h"
 
#include "SEImplementation/Configuration/SamplingConfig.h"
#include "SEImplementation/Plugin/FlexibleModelFitting/FlexibleModelFittingParameter.h"
#include "SEImplementation/Plugin/FlexibleModelFitting/FlexibleModelFittingFrame.h"
#include "SEImplementation/Plugin/FlexibleModelFitting/FlexibleModelFittingPrior.h"
 
namespace SourceXtractor {
 
class FlexibleModelFittingTask : public GroupTask {
 
public:
  FlexibleModelFittingTask(const std::string &least_squares_engine,
      unsigned int max_iterations, double modified_chi_squared_scale,
      std::vector<std::shared_ptr<FlexibleModelFittingParameter>> parameters,
      std::vector<std::shared_ptr<FlexibleModelFittingFrame>> frames,
      std::vector<std::shared_ptr<FlexibleModelFittingPrior>> priors,
      double scale_factor=1.0
      );
 
  virtual ~FlexibleModelFittingTask();
 
  void computeProperties(SourceGroupInterface& group) const override;
 
private:
 
  bool isFrameValid(SourceGroupInterface& group, int frame_index) const;
 
  std::shared_ptr<VectorImage<SeFloat>> createImageCopy(SourceGroupInterface& group, int frame_index) const;
  std::shared_ptr<VectorImage<SeFloat>> createWeightImage(SourceGroupInterface& group, int frame_index) const;
 
  ModelFitting::FrameModel<ImagePsf, std::shared_ptr<VectorImage<SourceXtractor::SeFloat>>> createFrameModel(
      SourceGroupInterface& group,
      double pixel_scale, FlexibleModelFittingParameterManager& manager, std::shared_ptr<FlexibleModelFittingFrame> frame) const;
 
  void createParameters() const;
 
  void updateCheckImages(SourceGroupInterface& group,
      double pixel_scale, FlexibleModelFittingParameterManager& manager) const;
 
  SeFloat computeChiSquaredForFrame(std::shared_ptr<const Image<SeFloat>> image,
      std::shared_ptr<const Image<SeFloat>> model, std::shared_ptr<const Image<SeFloat>> weights, int& data_points) const;
  SeFloat computeChiSquared(SourceGroupInterface& group,
      double pixel_scale, FlexibleModelFittingParameterManager& manager, int& total_data_points) const;
 
  void setDummyProperty(SourceGroupInterface& group, FlexibleModelFittingParameterManager& parameter_manager, Flags flags) const;
 
  // Task configuration
  std::string m_least_squares_engine;
  unsigned int m_max_iterations;
  double m_modified_chi_squared_scale;
 
  std::vector<std::shared_ptr<FlexibleModelFittingParameter>> m_parameters;
  std::vector<std::shared_ptr<FlexibleModelFittingFrame>> m_frames;
  std::vector<std::shared_ptr<FlexibleModelFittingPrior>> m_priors;
 
  double m_scale_factor;
};
 
}
 
 
 
#endif /* _SEIMPLEMENTATION_PLUGIN_FLEXIBLEMODELFITTING_FLEXIBLEMODELFITTINGTASK_H_ */