Undistortion.cs

Go to the documentation of this file.

using System.Runtime.InteropServices;
using IDSImaging.Peak.Common.Types;
using IDSImaging.Peak.ICV.Algorithms.Calibration;
using IDSImaging.Peak.ICV.Backend;
using IDSImaging.Peak.ICV.Types;
using Microsoft.Win32.SafeHandles;
 
namespace IDSImaging.Peak.ICV.Algorithms.Transformations;
 
public class Undistortion : IDisposable
{
    private readonly Handle _handle;
    private readonly Backend.Types.peak_icv_intrinsic_parameters _newIntrinsicParameters;
 
    public Undistortion(IntrinsicParameters intrinsicParameters)
    {
        var handle = new Handle(intrinsicParameters, out _newIntrinsicParameters);
        _handle = handle;
    }
 
    public Undistortion(IntrinsicParameters intrinsicParameters, Metadata metadata)
    {
        Backend.Types.peak_icv_capture_information captureInformation = CaptureInformation.FromMetadata(metadata);
        var handle = new Handle(intrinsicParameters, captureInformation, out _newIntrinsicParameters);
        _handle = handle;
    }
 
    public Undistortion(CalibrationParameters calibrationParameters)
    {
        var handle = new Handle(calibrationParameters, out _newIntrinsicParameters);
        _handle = handle;
    }
 
    public Undistortion(CalibrationParameters calibrationParameters, Metadata metadata)
    {
        Backend.Types.peak_icv_capture_information captureInformation = CaptureInformation.FromMetadata(metadata);
        var handle = new Handle(calibrationParameters, captureInformation, out _newIntrinsicParameters);
        _handle = handle;
    }
 
    public Interpolation Interpolation
    {
        get
        {
            Interpolation interpolation = 0;
            DllHelper.ExecuteAndMapReturnCodes(() =>
                CInterface.peak_icv_Transform_Undistortion_GetInterpolation(_handle, out interpolation));
            return interpolation;
        }
 
        set
        {
            DllHelper.ExecuteAndMapReturnCodes(() =>
                CInterface.peak_icv_Transform_Undistortion_SetInterpolation(_handle, value));
        }
    }
 
    public void Dispose()
    {
        _handle.Dispose();
        GC.SuppressFinalize(this);
    }
 
    public UndistortedImage Process(Image image)
    {
        using var outputImage = new Image(image.PixelFormat, image.Size);
 
        DllHelper.ExecuteAndMapReturnCodes(() =>
            CInterface.peak_icv_Transform_Undistortion_Process(_handle, image._handle, outputImage._handle));
 
        return new UndistortedImage(outputImage, new IntrinsicParameters(_newIntrinsicParameters));
    }
 
    private sealed class Handle : SafeHandleZeroOrMinusOneIsInvalid
    {
        internal Handle(IntrinsicParameters intrinsicParameters,
            out Backend.Types.peak_icv_intrinsic_parameters newIntrinsicParameters) : base(true)
        {
            Backend.Types.peak_icv_intrinsic_parameters tmp = default;
 
            DllHelper.ExecuteAndMapReturnCodes(() =>
                CInterface.peak_icv_Transform_Undistortion_Create(
                    out handle,
                    intrinsicParameters.ToCType(),
                    (nuint) Marshal.SizeOf<Backend.Types.peak_icv_intrinsic_parameters>(),
                    out tmp));
 
            newIntrinsicParameters = tmp;
        }
 
        internal Handle(IntrinsicParameters intrinsicParameters,
            Backend.Types.peak_icv_capture_information captureInformation,
            out Backend.Types.peak_icv_intrinsic_parameters newIntrinsicParameters) : base(true)
        {
            Backend.Types.peak_icv_intrinsic_parameters tmp = default;
 
            DllHelper.ExecuteAndMapReturnCodes(() =>
                CInterface.peak_icv_Transform_Undistortion_Create_With_Image_CaptureInformation(
                    out handle,
                    intrinsicParameters.ToCType(),
                    (nuint) Marshal.SizeOf<Backend.Types.peak_icv_intrinsic_parameters>(),
                    captureInformation,
                    (nuint) Marshal.SizeOf(typeof(Backend.Types.peak_icv_capture_information)),
                    out tmp));
 
            newIntrinsicParameters = tmp;
        }
 
        internal Handle(CalibrationParameters calibrationParameters,
            out Backend.Types.peak_icv_intrinsic_parameters newIntrinsicParameters) : base(true)
        {
            Backend.Types.peak_icv_intrinsic_parameters tmp = default;
 
            DllHelper.ExecuteAndMapReturnCodes(() =>
                CInterface.peak_icv_Transform_Undistortion_Create(
                    out handle,
                    calibrationParameters.IntrinsicParameters.ToCType(),
                    (nuint) Marshal.SizeOf<Backend.Types.peak_icv_intrinsic_parameters>(),
                    out tmp));
 
            newIntrinsicParameters = tmp;
        }
 
        internal Handle(CalibrationParameters calibrationParameters,
            Backend.Types.peak_icv_capture_information captureInformation,
            out Backend.Types.peak_icv_intrinsic_parameters newIntrinsicParameters) : base(true)
        {
            Backend.Types.peak_icv_intrinsic_parameters tmp = default;
 
            DllHelper.ExecuteAndMapReturnCodes(() =>
                CInterface.peak_icv_Transform_Undistortion_Create_With_Image_CaptureInformation(
                    out handle,
                    calibrationParameters.IntrinsicParameters.ToCType(),
                    (nuint) Marshal.SizeOf<Backend.Types.peak_icv_intrinsic_parameters>(),
                    captureInformation,
                    (nuint) Marshal.SizeOf(typeof(Backend.Types.peak_icv_capture_information)),
                    out tmp));
 
            newIntrinsicParameters = tmp;
        }
 
        protected override bool ReleaseHandle()
        {
            return CInterface.peak_icv_Transform_Undistortion_Destroy(handle) ==
                   Backend.Types.peak_icv_status.PEAK_ICV_STATUS_SUCCESS;
        }
    }
 
    private static class CInterface
    {
        [DllImport(Global.LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern Backend.Types.peak_icv_status peak_icv_Transform_Undistortion_Create(
            out IntPtr undistortionHandle,
            Backend.Types.peak_icv_intrinsic_parameters intrinsicParameters,
            nuint sizeOfIntrinsicParameters,
            out Backend.Types.peak_icv_intrinsic_parameters newIntrinsicParameters);
 
        [DllImport(Global.LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern Backend.Types.peak_icv_status
            peak_icv_Transform_Undistortion_Create_With_Image_CaptureInformation(
                out IntPtr undistortionHandle,
                Backend.Types.peak_icv_intrinsic_parameters intrinsicParameters,
                nuint sizeOfIntrinsicParameters,
                Backend.Types.peak_icv_capture_information captureInformation,
                nuint captureInformationSize,
                out Backend.Types.peak_icv_intrinsic_parameters newIntrinsicParameters);
 
        [DllImport(Global.LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern Backend.Types.peak_icv_status
            peak_icv_Transform_Undistortion_IncreaseUseCount(
                Handle undistortionHandle);
 
        [DllImport(Global.LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern Backend.Types.peak_icv_status peak_icv_Transform_Undistortion_Destroy(
            IntPtr undistortionHandle);
 
        [DllImport(Global.LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern Backend.Types.peak_icv_status peak_icv_Transform_Undistortion_Process(
            Handle undistortionHandle,
            Image.Handle inputImageHandle,
            Image.Handle outputImageHandle);
 
        [DllImport(Global.LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern Backend.Types.peak_icv_status
            peak_icv_Transform_Undistortion_GetInterpolation(
                Handle undistortionHandle,
                out Interpolation interpolation);
 
        [DllImport(Global.LibraryName, CallingConvention = CallingConvention.Cdecl)]
        public static extern Backend.Types.peak_icv_status
            peak_icv_Transform_Undistortion_SetInterpolation(
                Handle undistortionHandle,
                Interpolation interpolation);
    }
}