#ifndef V3D_DISTORTION_H
#define V3D_DISTORTION_H
#include "Math/v3d_linear.h"
#include "Math/v3d_linear_utils.h"
namespace V3D
{
struct StdDistortionFunction
{
double k1, k2, p1, p2;
StdDistortionFunction()
: k1(0), k2(0), p1(0), p2(0)
{ }
Vector2d operator()(Vector2d const& xu) const
{
double const r2 = xu[0]*xu[0] + xu[1]*xu[1];
double const r4 = r2*r2;
double const kr = 1 + k1*r2 + k2*r4;
Vector2d xd;
xd[0] = kr * xu[0] + 2*p1*xu[0]*xu[1] + p2*(r2 + 2*xu[0]*xu[0]);
xd[1] = kr * xu[1] + 2*p2*xu[0]*xu[1] + p1*(r2 + 2*xu[1]*xu[1]);
return xd;
}
Matrix2x2d derivativeWrtRadialParameters(Vector2d const& xu) const
{
double const r2 = xu[0]*xu[0] + xu[1]*xu[1];
double const r4 = r2*r2;
Matrix2x2d deriv;
deriv[0][0] = xu[0] * r2; deriv[0][1] = xu[0] * r4; deriv[1][0] = xu[1] * r2; deriv[1][1] = xu[1] * r4; return deriv;
}
Matrix2x2d derivativeWrtTangentialParameters(Vector2d const& xu) const
{
double const r2 = xu[0]*xu[0] + xu[1]*xu[1];
Matrix2x2d deriv;
deriv[0][0] = 2*xu[0]*xu[1]; deriv[0][1] = r2 + 2*xu[0]*xu[0]; deriv[1][0] = r2 + 2*xu[1]*xu[1]; deriv[1][1] = deriv[0][0]; return deriv;
}
Matrix2x2d derivativeWrtUndistortedPoint(Vector2d const& xu) const
{
double const r2 = xu[0]*xu[0] + xu[1]*xu[1];
double const r4 = r2*r2;
double const kr = 1 + k1*r2 + k2*r4;
double const dkr = 2*k1 + 4*k2*r2;
Matrix2x2d deriv;
deriv[0][0] = kr + xu[0] * xu[0] * dkr + 2*p1*xu[1] + 6*p2*xu[0]; deriv[0][1] = xu[0] * xu[1] * dkr + 2*p1*xu[0] + 2*p2*xu[1]; deriv[1][0] = deriv[0][1]; deriv[1][1] = kr + xu[1] * xu[1] * dkr + 6*p1*xu[1] + 2*p2*xu[0]; return deriv;
}
};
}
#endif