wintf 0.0.1

use windows::core::*;
use windows::Win32::Foundation::*;
use windows::Win32::Graphics::Direct2D::*;
use windows::Win32::Graphics::DirectComposition::*;
use windows::Win32::Graphics::Dxgi::Common::*;
use windows_numerics::*;

pub fn dcomp_create_desktop_device<P0>(renderingdevice: P0) -> Result<IDCompositionDesktopDevice>
where
    P0: Param<IUnknown>,
{
    unsafe { DCompositionCreateDevice3(renderingdevice) }
}

pub trait DCompositionDeviceExt {
    /// CreateVisual
    fn create_visual(&self) -> Result<IDCompositionVisual3>;
    /// Commit
    fn commit(&self) -> Result<()>;
    /// CreateSurface
    fn create_surface(
        &self,
        width: u32,
        height: u32,
        pixelformat: DXGI_FORMAT,
        alphamode: DXGI_ALPHA_MODE,
    ) -> Result<IDCompositionSurface>;
    /// GetFrameStatistics
    fn get_frame_statistics(&self) -> Result<DCOMPOSITION_FRAME_STATISTICS>;
    /// CreateAnimation
    fn create_animation(&self) -> Result<IDCompositionAnimation>;
    /// CreateMatrixTransform3D
    fn create_matrix_transform_3d(&self) -> Result<IDCompositionMatrixTransform3D>;
    /// CreateTransform3DGroup
    fn create_transform_3d_group(
        &self,
        transforms: &[Option<IDCompositionTransform3D>],
    ) -> Result<IDCompositionTransform3D>;
    fn create_rotate_transform_3d(&self) -> Result<IDCompositionRotateTransform3D>;
}

impl DCompositionDeviceExt for IDCompositionDevice3 {
    #[inline(always)]
    fn create_visual(&self) -> Result<IDCompositionVisual3> {
        unsafe { self.CreateVisual()?.cast() }
    }

    #[inline(always)]
    fn commit(&self) -> Result<()> {
        unsafe { self.Commit() }
    }

    #[inline(always)]
    fn create_surface(
        &self,
        width: u32,
        height: u32,
        pixelformat: DXGI_FORMAT,
        alphamode: DXGI_ALPHA_MODE,
    ) -> Result<IDCompositionSurface> {
        unsafe { self.CreateSurface(width, height, pixelformat, alphamode) }
    }

    #[inline(always)]
    fn get_frame_statistics(&self) -> Result<DCOMPOSITION_FRAME_STATISTICS> {
        unsafe { self.GetFrameStatistics() }
    }

    #[inline(always)]
    fn create_animation(&self) -> Result<IDCompositionAnimation> {
        unsafe { self.CreateAnimation() }
    }

    #[inline(always)]
    fn create_matrix_transform_3d(&self) -> Result<IDCompositionMatrixTransform3D> {
        unsafe { self.CreateMatrixTransform3D() }
    }

    #[inline(always)]
    fn create_transform_3d_group(
        &self,
        transforms: &[Option<IDCompositionTransform3D>],
    ) -> Result<IDCompositionTransform3D> {
        unsafe { self.CreateTransform3DGroup(transforms) }
    }

    #[inline(always)]
    fn create_rotate_transform_3d(&self) -> Result<IDCompositionRotateTransform3D> {
        unsafe { self.CreateRotateTransform3D() }
    }
}

pub trait DCompositionDesktopDeviceExt {
    /// CreateTargetForHwnd
    fn create_target_for_hwnd(&self, hwnd: HWND, topmost: bool) -> Result<IDCompositionTarget>;
}

impl DCompositionDesktopDeviceExt for IDCompositionDesktopDevice {
    #[inline(always)]
    fn create_target_for_hwnd(&self, hwnd: HWND, topmost: bool) -> Result<IDCompositionTarget> {
        unsafe { self.CreateTargetForHwnd(hwnd, topmost) }
    }
}

pub trait DCompositionTargetExt {
    /// SetRoot
    fn set_root<P0>(&self, visual: P0) -> Result<()>
    where
        P0: Param<IDCompositionVisual>;
}

impl DCompositionTargetExt for IDCompositionTarget {
    #[inline(always)]
    fn set_root<P0>(&self, visual: P0) -> Result<()>
    where
        P0: Param<IDCompositionVisual>,
    {
        unsafe { self.SetRoot(visual) }
    }
}

pub trait DCompositionVisualExt {
    /// SetBackFaceVisibility
    fn set_backface_visibility(&self, visibility: DCOMPOSITION_BACKFACE_VISIBILITY) -> Result<()>;
    /// SetOffsetX
    fn set_offset_x(&self, offset: f32) -> Result<()>;
    /// SetOffsetY
    fn set_offset_y(&self, offset: f32) -> Result<()>;
    /// SetOpacity - Visualの不透明度を設定 (0.0 = 完全透明, 1.0 = 完全不透明)
    fn set_opacity(&self, opacity: f32) -> Result<()>;
    /// AddVisual
    fn add_visual<P0, P1>(&self, visual: P0, insertabove: bool, referencevisual: P1) -> Result<()>
    where
        P0: Param<IDCompositionVisual>,
        P1: Param<IDCompositionVisual>;
    /// RemoveVisual - 指定されたVisualを親から削除 (R1)
    fn remove_visual<P0>(&self, visual: P0) -> Result<()>
    where
        P0: Param<IDCompositionVisual>;
    /// RemoveAllVisuals - 全ての子Visualを削除 (R1)
    fn remove_all_visuals(&self) -> Result<()>;
    /// SetContent
    fn set_content<P0>(&self, content: P0) -> Result<()>
    where
        P0: Param<IUnknown>;
    /// SetEffect
    fn set_effect<P0>(&self, effect: P0) -> Result<()>
    where
        P0: Param<IDCompositionEffect>;
}

impl DCompositionVisualExt for IDCompositionVisual3 {
    #[inline(always)]
    fn set_backface_visibility(&self, visibility: DCOMPOSITION_BACKFACE_VISIBILITY) -> Result<()> {
        unsafe { self.SetBackFaceVisibility(visibility) }
    }

    #[inline(always)]
    fn set_offset_x(&self, offset: f32) -> Result<()> {
        unsafe { self.SetOffsetX2(offset) }
    }

    #[inline(always)]
    fn set_offset_y(&self, offset: f32) -> Result<()> {
        unsafe { self.SetOffsetY2(offset) }
    }

    #[inline(always)]
    fn set_opacity(&self, opacity: f32) -> Result<()> {
        unsafe { self.SetOpacity2(opacity) }
    }

    #[inline(always)]
    fn add_visual<P0, P1>(&self, visual: P0, insertabove: bool, referencevisual: P1) -> Result<()>
    where
        P0: Param<IDCompositionVisual>,
        P1: Param<IDCompositionVisual>,
    {
        unsafe { self.AddVisual(visual, insertabove, referencevisual) }
    }

    #[inline(always)]
    fn remove_visual<P0>(&self, visual: P0) -> Result<()>
    where
        P0: Param<IDCompositionVisual>,
    {
        unsafe { self.RemoveVisual(visual) }
    }

    #[inline(always)]
    fn remove_all_visuals(&self) -> Result<()> {
        unsafe { self.RemoveAllVisuals() }
    }

    #[inline(always)]
    fn set_content<P0>(&self, content: P0) -> Result<()>
    where
        P0: Param<IUnknown>,
    {
        unsafe { self.SetContent(content) }
    }

    #[inline(always)]
    fn set_effect<P0>(&self, effect: P0) -> Result<()>
    where
        P0: Param<IDCompositionEffect>,
    {
        unsafe { self.SetEffect(effect) }
    }
}

pub trait DCompositionSurfaceExt {
    /// BeginDraw
    fn begin_draw(&self, updaterect: Option<&RECT>) -> Result<(ID2D1DeviceContext3, POINT)>;
    /// EndDraw
    fn end_draw(&self) -> Result<()>;
    /// SuspendDraw
    fn suspend_draw(&self) -> Result<()>;
    /// ResumeDraw
    fn resume_draw(&self) -> Result<()>;
    /// Scroll
    fn scroll(
        &self,
        scrollrect: Option<&RECT>,
        cliprect: Option<&RECT>,
        offsetx: i32,
        offsety: i32,
    ) -> windows_core::Result<()>;
}

impl DCompositionSurfaceExt for IDCompositionSurface {
    #[inline(always)]
    fn begin_draw(&self, updaterect: Option<&RECT>) -> Result<(ID2D1DeviceContext3, POINT)> {
        let updaterect = updaterect.map(|r| r as *const _);
        let mut updateoffset = POINT::default();
        let dc: ID2D1DeviceContext = unsafe { self.BeginDraw(updaterect, &mut updateoffset)? };
        Ok((dc.cast()?, updateoffset))
    }

    #[inline(always)]
    fn end_draw(&self) -> Result<()> {
        unsafe { self.EndDraw() }
    }
    #[inline(always)]
    fn suspend_draw(&self) -> Result<()> {
        unsafe { self.SuspendDraw() }
    }
    #[inline(always)]
    fn resume_draw(&self) -> Result<()> {
        unsafe { self.ResumeDraw() }
    }
    #[inline(always)]
    fn scroll(
        &self,
        scrollrect: Option<&RECT>,
        cliprect: Option<&RECT>,
        offsetx: i32,
        offsety: i32,
    ) -> windows_core::Result<()> {
        let scrollrect = scrollrect.map(|r| r as *const _);
        let cliprect = cliprect.map(|r| r as *const _);
        unsafe { self.Scroll(scrollrect, cliprect, offsetx, offsety) }
    }
}

pub trait DCompositionRotateTransform3DExt {
    /// SetAngle
    fn set_angle_animation<P0>(&self, animation: P0) -> Result<()>
    where
        P0: Param<IDCompositionAnimation>;
    /// SetAngle2
    fn set_angle(&self, angle: f32) -> Result<()>;
    /// SetAxisZ2
    fn set_axis_z(&self, axis_z: f32) -> Result<()>;
    /// SetAxisY2
    fn set_axis_y(&self, axis_y: f32) -> Result<()>;
}

impl DCompositionRotateTransform3DExt for IDCompositionRotateTransform3D {
    #[inline(always)]
    fn set_angle_animation<P0>(&self, animation: P0) -> Result<()>
    where
        P0: Param<IDCompositionAnimation>,
    {
        unsafe { self.SetAngle(animation) }
    }

    #[inline(always)]
    fn set_angle(&self, angle: f32) -> Result<()> {
        unsafe { self.SetAngle2(angle) }
    }

    #[inline(always)]
    fn set_axis_z(&self, axis_z: f32) -> Result<()> {
        unsafe { self.SetAxisZ2(axis_z) }
    }

    #[inline(always)]
    fn set_axis_y(&self, axis_y: f32) -> Result<()> {
        unsafe { self.SetAxisY2(axis_y) }
    }
}

pub trait DCompositionMatrixTransform3DExt {
    /// SetMatrix
    fn set_matrix(&self, matrix: &Matrix4x4) -> Result<()>;
}

impl DCompositionMatrixTransform3DExt for IDCompositionMatrixTransform3D {
    #[inline(always)]
    fn set_matrix(&self, matrix: &Matrix4x4) -> Result<()> {
        unsafe { self.SetMatrix(matrix) }
    }
}