module OpenCV.Calib3d
( FundamentalMatMethod(..)
, WhichImage(..)
, findFundamentalMat
, computeCorrespondEpilines
) where
import "base" Data.Int
import "base" Data.Word
import "base" Foreign.C.Types
import qualified "inline-c" Language.C.Inline as C
import qualified "inline-c-cpp" Language.C.Inline.Cpp as C
import "this" OpenCV.Internal.C.Inline ( openCvCtx )
import "this" OpenCV.Internal.C.Types
import "this" OpenCV.Internal.Calib3d.Constants
import "this" OpenCV.Core.Types
import "this" OpenCV.Internal.Core.Types
import "this" OpenCV.Internal.Core.Types.Mat
import "this" OpenCV.Internal.Exception
import "this" OpenCV.TypeLevel
import "transformers" Control.Monad.Trans.Except
import qualified "vector" Data.Vector as V
C.context openCvCtx
C.include "opencv2/core.hpp"
C.include "opencv2/calib3d.hpp"
C.using "namespace cv"
data FundamentalMatMethod
= FM_7Point
| FM_8Point
| FM_Ransac !(Maybe Double) !(Maybe Double)
| FM_Lmeds !(Maybe Double)
deriving (Show, Eq)
marshalFundamentalMatMethod :: FundamentalMatMethod -> (Int32, CDouble, CDouble)
marshalFundamentalMatMethod = \case
FM_7Point -> (c'CV_FM_7POINT, 0, 0)
FM_8Point -> (c'CV_FM_8POINT, 0, 0)
FM_Ransac p1 p2 -> (c'CV_FM_RANSAC, maybe 3 realToFrac p1, maybe 0.99 realToFrac p2)
FM_Lmeds p2 -> (c'CV_FM_LMEDS, 0, maybe 0.99 realToFrac p2)
data WhichImage = Image1 | Image2 deriving (Show, Eq)
marshalWhichImage :: WhichImage -> Int32
marshalWhichImage = \case
Image1 -> 1
Image2 -> 2
findFundamentalMat
:: (IsPoint2 point2 CDouble)
=> V.Vector (point2 CDouble)
-> V.Vector (point2 CDouble)
-> FundamentalMatMethod
-> CvExcept ( Maybe ( Mat ('S '[ 'D, 'S 3 ]) ('S 1) ('S Double)
, Mat ('S '[ 'D, 'D ]) ('S 1) ('S Word8 )
)
)
findFundamentalMat pts1 pts2 method = do
(fm, pointMask) <- c'findFundamentalMat
catchE (Just . (, unsafeCoerceMat pointMask) <$> coerceMat fm)
(\case CoerceMatError _msgs -> pure Nothing
otherError -> throwE otherError
)
where
c'findFundamentalMat = unsafeWrapException $ do
fm <- newEmptyMat
pointMask <- newEmptyMat
handleCvException (pure (fm, pointMask)) $
withPtr fm $ \fmPtr ->
withPtr pointMask $ \pointMaskPtr ->
withArrayPtr (V.map toPoint pts1) $ \pts1Ptr ->
withArrayPtr (V.map toPoint pts2) $ \pts2Ptr ->
[cvExcept|
cv::_InputArray pts1 = cv::_InputArray($(Point2d * pts1Ptr), $(int32_t c'numPts1));
cv::_InputArray pts2 = cv::_InputArray($(Point2d * pts2Ptr), $(int32_t c'numPts2));
*$(Mat * fmPtr) =
cv::findFundamentalMat
( pts1
, pts2
, $(int32_t c'method)
, $(double c'p1)
, $(double c'p2)
, *$(Mat * pointMaskPtr)
);
|]
c'numPts1 = fromIntegral $ V.length pts1
c'numPts2 = fromIntegral $ V.length pts2
(c'method, c'p1, c'p2) = marshalFundamentalMatMethod method
computeCorrespondEpilines
:: (IsPoint2 point2 CDouble)
=> V.Vector (point2 CDouble)
-> WhichImage
-> Mat (ShapeT [3, 3]) ('S 1) ('S Double)
-> CvExcept (Mat ('S ['D, 'S 1]) ('S 3) ('S Double))
computeCorrespondEpilines points whichImage fm = unsafeWrapException $ do
epilines <- newEmptyMat
handleCvException (pure $ unsafeCoerceMat epilines) $
withArrayPtr (V.map toPoint points) $ \pointsPtr ->
withPtr fm $ \fmPtr ->
withPtr epilines $ \epilinesPtr -> do
[cvExcept|
cv::_InputArray points =
cv::_InputArray( $(Point2d * pointsPtr)
, $(int32_t c'numPoints)
);
cv::computeCorrespondEpilines
( points
, $(int32_t c'whichImage)
, *$(Mat * fmPtr)
, *$(Mat * epilinesPtr)
);
|]
where
c'numPoints = fromIntegral $ V.length points
c'whichImage = marshalWhichImage whichImage