Equidistant.hpp

// This file is part of the AliceVision project.
// Copyright (c) 2020 AliceVision contributors.
// This Source Code Form is subject to the terms of the Mozilla Public License,
// v. 2.0. If a copy of the MPL was not distributed with this file,
// You can obtain one at https://mozilla.org/MPL/2.0/.
 
#pragma once
 
#include <aliceVision/numeric/numeric.hpp>
#include <aliceVision/numeric/projection.hpp>
#include <aliceVision/camera/cameraCommon.hpp>
#include <aliceVision/camera/IntrinsicScaleOffsetDisto.hpp>
#include <aliceVision/geometry/Pose3.hpp>
 
#include "DistortionFisheye1.hpp"
 
#include <memory>
#include <algorithm>
 
namespace aliceVision {
namespace camera {
 
class Equidistant : public IntrinsicScaleOffsetDisto
{
  public:
    Equidistant()
      : Equidistant(1, 1, 1.0, 0.0, 0.0)
    {}
 
    Equidistant(unsigned int w,
                unsigned int h,
                double focalLengthPix,
                double offsetX,
                double offsetY,
                std::shared_ptr<Distortion> distortion = nullptr)
      : IntrinsicScaleOffsetDisto(w, h, focalLengthPix, focalLengthPix, offsetX, offsetY, distortion),
        _circleRadius(std::min(w, h) * 0.5),
        _circleCenter(w / 2.0, h / 2.0)
    {}
 
    Equidistant(unsigned int w,
                unsigned int h,
                double focalLengthPix,
                double offsetX,
                double offsetY,
                double circleRadiusPix,
                std::shared_ptr<Distortion> distortion = nullptr)
      : IntrinsicScaleOffsetDisto(w, h, focalLengthPix, focalLengthPix, offsetX, offsetY, distortion),
        _circleRadius(circleRadiusPix != 0.0 ? circleRadiusPix : std::min(w, h) * 0.5),
        _circleCenter(w / 2.0, h / 2.0)
    {}
 
    ~Equidistant() override = default;
 
    Equidistant* clone() const override { return new Equidistant(*this); }
 
    static std::shared_ptr<Equidistant> cast(std::shared_ptr<IntrinsicBase> sptr);
 
    void assign(const IntrinsicBase& other) override { *this = dynamic_cast<const Equidistant&>(other); }
 
    bool isValid() const override { return _scale(0) > 0 && IntrinsicBase::isValid(); }
 
    EINTRINSIC getType() const override;
 
    Vec2 transformProject(const Eigen::Matrix4d& pose, const Vec4& pt, bool applyDistortion = true) const override;
 
    Vec2 transformProject(const geometry::Pose3& pose, const Vec4& pt3D, bool applyDistortion = true) const
    {
        return transformProject(pose.getHomogeneous(), pt3D, applyDistortion);
    }
 
    Vec2 project(const Vec4& pt, bool applyDistortion = true) const override;
 
    Eigen::Matrix<double, 2, 3> getDerivativeTransformProjectWrtPoint3(const Eigen::Matrix4d& pose, const Vec4& pt) const override;
 
    Eigen::Matrix<double, 2, 2> getDerivativeTransformProjectWrtScale(const Eigen::Matrix4d& pose, const Vec4& pt) const;
 
    Eigen::Matrix<double, 2, 2> getDerivativeTransformProjectWrtPrincipalPoint(const Eigen::Matrix4d& pose, const Vec4& pt) const;
 
    Eigen::Matrix<double, 2, Eigen::Dynamic> getDerivativeTransformProjectWrtParams(const Eigen::Matrix4d& pose, const Vec4& pt3D) const override;
 
    Vec3 toUnitSphere(const Vec2& pt) const override;
 
    Eigen::Matrix<double, 3, 2> getDerivativetoUnitSphereWrtPoint(const Vec2& pt) const;
 
    Eigen::Matrix<double, 3, 2> getDerivativetoUnitSphereWrtScale(const Vec2& pt) const;
 
    Eigen::Matrix<double, 3, Eigen::Dynamic> getDerivativeBackProjectUnitWrtParams(const Vec2& pt2D) const override;
 
    Eigen::Matrix<double, 2, Eigen::Dynamic> getDerivativeTransformProjectWrtDistortion(const Eigen::Matrix4d& pose, const Vec4& pt) const override;
 
    Eigen::Matrix<double, 3, Eigen::Dynamic> getDerivativeBackProjectUnitWrtDistortion(const Vec2& pt2D) const override;
 
    double imagePlaneToCameraPlaneError(double value) const override;
 
    // Transform a point from the camera plane to the image plane
    Vec2 cam2ima(const Vec2& p) const override;
 
    Eigen::Matrix2d getDerivativeCam2ImaWrtPoint() const override;
 
    // Transform a point from the image plane to the camera plane
    Vec2 ima2cam(const Vec2& p) const override;
 
    Eigen::Matrix2d getDerivativeIma2CamWrtPoint() const override;
 
    Eigen::Matrix2d getDerivativeIma2CamWrtPrincipalPoint() const override;
 
    bool isVisibleRay(const Vec3& ray) const override;
 
    inline double getCircleRadius() const { return _circleRadius; }
 
    inline void setCircleRadius(double radius) { _circleRadius = radius; }
 
    inline double getCircleCenterX() const { return _circleCenter(0); }
 
    inline void setCircleCenterX(double x) { _circleCenter(0) = x; }
 
    inline double getCircleCenterY() const { return _circleCenter(1); }
 
    inline void setCircleCenterY(double y) { _circleCenter(1) = y; }
 
    inline Vec2 getCircleCenter() const { return _circleCenter; }
 
    double getHorizontalFov() const override;
 
    double getVerticalFov() const override;
 
    double pixelProbability() const override;
 
  protected:
    double _circleRadius{0.0};
    Vec2 _circleCenter{0.0, 0.0};
};
 
}  // namespace camera
}  // namespace aliceVision