IlluminationModel.h

#ifndef MTF_ILLUMINATION_MODEL_H
#define MTF_ILLUMINATION_MODEL_H

#include "mtf/Macros/common.h"
#include "mtf/Utilities/excpUtils.h"

#define ilm_not_implemeted(func_name) \
    throw mtf::utils::FunctonNotImplemented(cv::format("%s :: %s :: Not implemented Yet", name.c_str(), #func_name))

_MTF_BEGIN_NAMESPACE

struct ILMParams{
    int resx, resy;
    ILMParams(int _resx, int _resy) :resx(_resx), resy(_resy){}
    ILMParams(const ILMParams *ilm_params = nullptr) :
        resx(MTF_RES), resy(MTF_RES){
        if(ilm_params){
            resx = ilm_params->resx;
            resy = ilm_params->resy;
        }
    }
};

class IlluminationModel{

public:
    enum class PixHessType{ Constant, Diagonal, General };
    string name;
    bool apply_on_init;
    IlluminationModel(const ILMParams *ilm_params):
        resx(MTF_RES), resy(MTF_RES), d2f_dg2_type(PixHessType::General){
        if(ilm_params) {
            if(ilm_params->resx <= 0 || ilm_params->resy <= 0) {
                throw utils::InvalidArgument("IlluminationModel::Invalid sampling resolution provided");
            }
            resx = ilm_params->resx;
            resy = ilm_params->resy;            
        }
        n_pix = resx*resy;
    }
    virtual ~IlluminationModel(){}
    virtual int getStateSize() const = 0;
    virtual void initialize(double *p){}
    virtual void apply(double *g, const double *I, const double *p) = 0;
    virtual void invert(double *inv_p, const double *p) = 0;
    virtual void update(double *new_p, const double *old_p, const double *dp) = 0;
    virtual void cmptParamJacobian(double *df_dp, const double *df_dg,
        const double *I, const double *p) = 0;
    virtual void cmptPixJacobian(double *df_dI, const double *df_dg,
        const double *I, const double *p) = 0;
    virtual void cmptParamHessian(double *d2f_dp2, const double *d2f_dg2,
        const double *df_dg, const double *I, const double *p) = 0;
    virtual void cmptPixHessian(double *d2f_dI2, const double *d2f_dg2,
        const double *df_dg, const double *I, const double *p) = 0;
    virtual void cmptCrossHessian(double *d2f_dp_dI, const double *d2f_dg2,
        const double *df_dg, const double *I, const double *p) = 0;
    virtual void cmptParamHessian(double *d2f_dp2, const double *d2f_dg2,
        const double *I, const double *p) = 0;
    virtual void cmptPixHessian(double *d2f_dI2, const double *d2f_dg2,
        const double *I, const double *p) = 0;
    virtual void cmptCrossHessian(double *d2f_dp_dI, const double *d2f_dg2,
        const double *I, const double *p) = 0;

    virtual void setPixHessType(PixHessType _d2f_dg2_type){
        d2f_dg2_type = _d2f_dg2_type;
    }
    virtual PixHessType getPixHessType() = 0;
    virtual void parseSamplerSigma(VectorXd &out_sigma, const VectorXd &in_sigma) = 0;
    virtual void parseSamplerMean(VectorXd &out_mean, const VectorXd &in_mean) = 0;

protected:

    int resx, resy, n_pix;
    PixHessType d2f_dg2_type;
};

_MTF_END_NAMESPACE

#endif