    --------------------------------------------------------------------------------
    -- |
    -- Module      :  Graphics.Rendering.OpenGL.GL.PixelRectangles.Convolution
    -- Copyright   :  (c) Sven Panne 2002-2009
    -- License     :  BSD-style (see the file libraries/OpenGL/LICENSE)
    --
    -- Maintainer  :  sven.panne@aedion.de
    -- Stability   :  stable
    -- Portability :  portable
    --
    -- This module corresponds to a part of section 3.6.1 (Pixel Storage Modes) of
    -- the OpenGL 2.1 specs.
    --
    --------------------------------------------------------------------------------

    module Graphics.Rendering.OpenGL.GL.PixelRectangles.Convolution (
       ConvolutionTarget(..), convolution,
       convolutionFilter1D, getConvolutionFilter1D,
       convolutionFilter2D, getConvolutionFilter2D,
       separableFilter2D, getSeparableFilter2D,
       copyConvolutionFilter1D, copyConvolutionFilter2D,
       convolutionWidth, convolutionHeight,
       maxConvolutionWidth, maxConvolutionHeight,
       ConvolutionBorderMode(..), convolutionBorderMode,
       convolutionFilterScale, convolutionFilterBias,
    ) where

    import Data.StateVar
    import Foreign.Marshal.Alloc
    import Foreign.Marshal.Utils
    import Foreign.Ptr
    import Foreign.Storable
    import Graphics.Rendering.OpenGL.GL.Capability
    import Graphics.Rendering.OpenGL.GL.CoordTrans
    import Graphics.Rendering.OpenGL.GL.PeekPoke
    import Graphics.Rendering.OpenGL.GL.PixelData
    import Graphics.Rendering.OpenGL.GL.PixelRectangles.Rasterization
    import Graphics.Rendering.OpenGL.GL.Texturing.PixelInternalFormat
    import Graphics.Rendering.OpenGL.GL.VertexSpec
    import Graphics.Rendering.OpenGL.GLU.ErrorsInternal
    import Graphics.Rendering.OpenGL.Raw.ARB.Compatibility (
       glConvolutionFilter1D, glConvolutionFilter2D, glConvolutionParameterfv,
       glConvolutionParameteri, glCopyConvolutionFilter1D,
       glCopyConvolutionFilter2D, glGetConvolutionFilter,
       glGetConvolutionParameterfv, glGetConvolutionParameteriv,
       glGetSeparableFilter, glSeparableFilter2D, gl_CONSTANT_BORDER,
       gl_CONVOLUTION_1D, gl_CONVOLUTION_2D, gl_CONVOLUTION_BORDER_COLOR,
       gl_CONVOLUTION_BORDER_MODE, gl_CONVOLUTION_FILTER_BIAS,
       gl_CONVOLUTION_FILTER_SCALE, gl_CONVOLUTION_FORMAT, gl_CONVOLUTION_HEIGHT,
       gl_CONVOLUTION_WIDTH, gl_MAX_CONVOLUTION_HEIGHT, gl_MAX_CONVOLUTION_WIDTH,
       gl_REDUCE, gl_REPLICATE_BORDER, gl_SEPARABLE_2D )
    import Graphics.Rendering.OpenGL.Raw.Core31

    --------------------------------------------------------------------------------

    data ConvolutionTarget =
         Convolution1D
       | Convolution2D
       | Separable2D
       deriving ( Eq, Ord, Show )

    marshalConvolutionTarget :: ConvolutionTarget -> GLenum
    marshalConvolutionTarget x = case x of
       Convolution1D -> gl_CONVOLUTION_1D
       Convolution2D -> gl_CONVOLUTION_2D
       Separable2D -> gl_SEPARABLE_2D

    convolutionTargetToEnableCap :: ConvolutionTarget -> EnableCap
    convolutionTargetToEnableCap x = case x of
       Convolution1D -> CapConvolution1D
       Convolution2D -> CapConvolution2D
       Separable2D -> CapSeparable2D

    --------------------------------------------------------------------------------

    convolution :: ConvolutionTarget -> StateVar Capability
    convolution = makeCapability . convolutionTargetToEnableCap

    --------------------------------------------------------------------------------

    convolutionFilter1D :: PixelInternalFormat -> GLsizei -> PixelData a -> IO ()
    convolutionFilter1D int w pd =
       withPixelData pd $
          glConvolutionFilter1D
             (marshalConvolutionTarget Convolution1D)
             (marshalPixelInternalFormat' int) w

    --------------------------------------------------------------------------------

    getConvolutionFilter1D :: PixelData a -> IO ()
    getConvolutionFilter1D = getConvolutionFilter Convolution1D

    getConvolutionFilter :: ConvolutionTarget -> PixelData a -> IO ()
    getConvolutionFilter t pd =
       withPixelData pd $ glGetConvolutionFilter (marshalConvolutionTarget t)

    --------------------------------------------------------------------------------

    convolutionFilter2D :: PixelInternalFormat -> Size -> PixelData a -> IO ()
    convolutionFilter2D int (Size w h) pd =
       withPixelData pd $
          glConvolutionFilter2D
             (marshalConvolutionTarget Convolution2D)
             (marshalPixelInternalFormat' int) w h

    --------------------------------------------------------------------------------

    getConvolutionFilter2D :: PixelData a -> IO ()
    getConvolutionFilter2D = getConvolutionFilter Convolution2D

    --------------------------------------------------------------------------------

    separableFilter2D ::
       PixelInternalFormat -> Size -> PixelData a -> PixelData a -> IO ()
    separableFilter2D int (Size w h) pdRow pdCol =
       withPixelData pdRow $ \f1 d1 p1 ->
          withPixelData pdCol $ \f2 d2 p2 ->
             if f1 == f2 && d1 == d2
                then glSeparableFilter2D
                        (marshalConvolutionTarget Separable2D)
                        (marshalPixelInternalFormat' int) w h f1 d1 p1 p2
                else recordInvalidValue

    --------------------------------------------------------------------------------

    getSeparableFilter2D :: PixelData a -> PixelData a -> IO ()
    getSeparableFilter2D pdRow pdCol =
       withPixelData pdRow $ \f1 d1 p1 ->
          withPixelData pdCol $ \f2 d2 p2 ->
             if f1 == f2 && d1 == d2
                then glGetSeparableFilter
                        (marshalConvolutionTarget Separable2D) f1 d1 p1 p2 nullPtr
                else recordInvalidValue

    --------------------------------------------------------------------------------

    copyConvolutionFilter1D :: PixelInternalFormat -> Position -> GLsizei -> IO ()
    copyConvolutionFilter1D int (Position x y) =
       glCopyConvolutionFilter1D
          (marshalConvolutionTarget Convolution1D) (marshalPixelInternalFormat' int)
          x y

    --------------------------------------------------------------------------------

    copyConvolutionFilter2D :: PixelInternalFormat -> Position -> Size -> IO ()
    copyConvolutionFilter2D int (Position x y) (Size w h) =
       glCopyConvolutionFilter2D
          (marshalConvolutionTarget Convolution2D) (marshalPixelInternalFormat' int)
          x y w h

    --------------------------------------------------------------------------------

    data ConvolutionParameter =
         ConvolutionBorderColor
       | ConvolutionBorderMode
       | ConvolutionFilterScale
       | ConvolutionFilterBias
       | ConvolutionFormat
       | ConvolutionWidth
       | ConvolutionHeight
       | MaxConvolutionWidth
       | MaxConvolutionHeight
       deriving ( Eq, Ord, Show )

    marshalConvolutionParameter :: ConvolutionParameter -> GLenum
    marshalConvolutionParameter x = case x of
       ConvolutionBorderColor -> gl_CONVOLUTION_BORDER_COLOR
       ConvolutionBorderMode -> gl_CONVOLUTION_BORDER_MODE
       ConvolutionFilterScale -> gl_CONVOLUTION_FILTER_SCALE
       ConvolutionFilterBias -> gl_CONVOLUTION_FILTER_BIAS
       ConvolutionFormat -> gl_CONVOLUTION_FORMAT
       ConvolutionWidth -> gl_CONVOLUTION_WIDTH
       ConvolutionHeight -> gl_CONVOLUTION_HEIGHT
       MaxConvolutionWidth -> gl_MAX_CONVOLUTION_WIDTH
       MaxConvolutionHeight -> gl_MAX_CONVOLUTION_HEIGHT

    --------------------------------------------------------------------------------

    convolutionWidth :: ConvolutionTarget -> GettableStateVar GLsizei
    convolutionWidth t = convolutionParameteri t ConvolutionWidth

    convolutionHeight :: ConvolutionTarget -> GettableStateVar GLsizei
    convolutionHeight t = convolutionParameteri t ConvolutionHeight

    maxConvolutionWidth :: ConvolutionTarget -> GettableStateVar GLsizei
    maxConvolutionWidth t = convolutionParameteri t MaxConvolutionWidth

    maxConvolutionHeight :: ConvolutionTarget -> GettableStateVar GLsizei
    maxConvolutionHeight t = convolutionParameteri t MaxConvolutionHeight

    convolutionParameteri ::
       ConvolutionTarget -> ConvolutionParameter -> GettableStateVar GLsizei
    convolutionParameteri t p =
       makeGettableStateVar (getConvolutionParameteri fromIntegral t p)

    getConvolutionParameteri ::
       (GLint -> a) -> ConvolutionTarget -> ConvolutionParameter -> IO a
    getConvolutionParameteri f t p =
       alloca $ \buf -> do
          glGetConvolutionParameteriv
             (marshalConvolutionTarget t) (marshalConvolutionParameter p) buf
          peek1 f buf

    --------------------------------------------------------------------------------

    data ConvolutionBorderMode' =
         Reduce'
       | ConstantBorder'
       | ReplicateBorder'

    marshalConvolutionBorderMode' :: ConvolutionBorderMode' -> GLint
    marshalConvolutionBorderMode' x = fromIntegral $ case x of
       Reduce' -> gl_REDUCE
       ConstantBorder' -> gl_CONSTANT_BORDER
       ReplicateBorder' -> gl_REPLICATE_BORDER

    unmarshalConvolutionBorderMode' :: GLint -> ConvolutionBorderMode'
    unmarshalConvolutionBorderMode' x
       | y == gl_REDUCE = Reduce'
       | y == gl_CONSTANT_BORDER = ConstantBorder'
       | y == gl_REPLICATE_BORDER = ReplicateBorder'
       | otherwise = error ("unmarshalConvolutionBorderMode': illegal value " ++ show x)
       where y = fromIntegral x

    --------------------------------------------------------------------------------

    data ConvolutionBorderMode =
         Reduce
       | ConstantBorder (Color4 GLfloat)
       | ReplicateBorder
       deriving ( Eq, Ord, Show )

    --------------------------------------------------------------------------------

    convolutionBorderMode :: ConvolutionTarget -> StateVar ConvolutionBorderMode
    convolutionBorderMode t =
       makeStateVar (getConvolutionBorderMode t) (setConvolutionBorderMode t)

    getConvolutionBorderMode :: ConvolutionTarget -> IO ConvolutionBorderMode
    getConvolutionBorderMode t = do
       mode <- getConvolutionParameteri
                  unmarshalConvolutionBorderMode' t ConvolutionBorderMode
       case mode of
          Reduce' -> return Reduce
          ConstantBorder' -> do
             c <- getConvolutionParameterC4f t ConvolutionBorderColor
             return $ ConstantBorder c
          ReplicateBorder' -> return ReplicateBorder

    setConvolutionBorderMode :: ConvolutionTarget -> ConvolutionBorderMode -> IO ()
    setConvolutionBorderMode t mode = do
       let setBM = setConvolutionParameteri
                      marshalConvolutionBorderMode' t ConvolutionBorderMode
       case mode of
          Reduce -> setBM Reduce'
          ConstantBorder c -> do
             setBM ConstantBorder'
             convolutionParameterC4f t ConvolutionBorderColor c
          ReplicateBorder -> setBM ReplicateBorder'

    setConvolutionParameteri ::
       (a -> GLint) -> ConvolutionTarget -> ConvolutionParameter -> a -> IO ()
    setConvolutionParameteri f t p x =
       glConvolutionParameteri
          (marshalConvolutionTarget t) (marshalConvolutionParameter p) (f x)

    --------------------------------------------------------------------------------

    convolutionFilterScale :: ConvolutionTarget -> StateVar (Color4 GLfloat)
    convolutionFilterScale = convolutionC4f ConvolutionFilterScale

    convolutionFilterBias :: ConvolutionTarget -> StateVar (Color4 GLfloat)
    convolutionFilterBias = convolutionC4f ConvolutionFilterBias

    convolutionC4f ::
       ConvolutionParameter -> ConvolutionTarget -> StateVar (Color4 GLfloat)
    convolutionC4f p t =
       makeStateVar (getConvolutionParameterC4f t p) (convolutionParameterC4f t p)

    getConvolutionParameterC4f ::
       ConvolutionTarget -> ConvolutionParameter -> IO (Color4 GLfloat)
    getConvolutionParameterC4f t p =
       alloca $ \buf -> do
          glGetConvolutionParameterfv
             (marshalConvolutionTarget t) (marshalConvolutionParameter p) (castPtr buf)
          peek buf

    convolutionParameterC4f ::
       ConvolutionTarget -> ConvolutionParameter -> Color4 GLfloat -> IO ()
    convolutionParameterC4f t p c =
       with c $ \ptr ->
          glConvolutionParameterfv
             (marshalConvolutionTarget t) (marshalConvolutionParameter p) (castPtr ptr)