/*
// Copyright 2015 2016 Intel Corporation All Rights Reserved.
//
// The source code, information and material ("Material") contained herein is
// owned by Intel Corporation or its suppliers or licensors, and title
// to such Material remains with Intel Corporation or its suppliers or
// licensors. The Material contains proprietary information of Intel
// or its suppliers and licensors. The Material is protected by worldwide
// copyright laws and treaty provisions. No part of the Material may be used,
// copied, reproduced, modified, published, uploaded, posted, transmitted,
// distributed or disclosed in any way without Intel's prior express written
// permission. No license under any patent, copyright or other intellectual
// property rights in the Material is granted to or conferred upon you,
// either expressly, by implication, inducement, estoppel or otherwise.
// Any license under such intellectual property rights must be express and
// approved by Intel in writing.
//
// Unless otherwise agreed by Intel in writing,
// you may not remove or alter this notice or any other notice embedded in
// Materials by Intel or Intel's suppliers or licensors in any way.
*/

//   A simple example of performing linear filtering of an image using one of predefined convolution kernels
// implemented with Intel IPP functions:
//     ippiFilterPrewittHorizBorderGetBufferSize
//     ippiFilterPrewittHorizBorder_8u16s_C1R
//     ippiFilterPrewittVertBorder_8u16s_C1R
//     ippiFilterRobertsDownBorder_8u16s_C1R
//     ippiFilterRobertsUpBorder_8u16s_C1R
//     ippiFilterScharrHorizMaskBorder_8u16s_C1R

#include <stdio.h>
#include "ipp.h"

#define WIDTH  128  /* image width  */
#define HEIGHT 128 /* image height */

/* Next two defines are created to simplify code reading and understanding */
#define EXIT_MAIN exitLine:                                  /* Label for Exit */
#define check_sts(st) if((st) != ippStsNoErr) goto exitLine; /* Go to Exit if IPP function returned status different from ippStsNoErr */

/* Results of ippMalloc() are not validated because Intel(R) Integrated Performance Primitives functions perform bad arguments check and will return an appropriate status  */

int main(void)
{
    IppStatus status = ippStsNoErr;
    Ipp8u* pSrc = NULL; Ipp16s *pDst = NULL;  /* Pointers to source/destination images */
    int srcStep = 0, dstStep = 0;             /* Steps, in bytes, through the source/destination images */
    IppiSize srcImageSize = { WIDTH + 1, HEIGHT + 1 }; /* size of source image in pixels */
    IppiSize roiSize = { WIDTH, HEIGHT }; /* Size of source/destination ROI in pixels */
    Ipp8u *pBuffer = NULL;                /* Pointer to the work buffer */
    int iTmpBufSize = 0;                  /* Common work buffer size */
    IppiMaskSize mask = ippMskSize3x3;               /* Predefined mask of IppiMaskSize type. */
    IppDataType srcDataType = ipp8u, dstDataType = ipp16s;
    IppiBorderType borderType = (IppiBorderType)(ippBorderRepl | ippBorderInMemTop | ippBorderInMemRight);
    Ipp8u borderValue = 254;
    int numChannels = 1;

    /* memory allocation */
    pSrc = ippiMalloc_8u_C1(srcImageSize.width, srcImageSize.height, &srcStep);
    pDst = ippiMalloc_16s_C1(roiSize.width, roiSize.height, &dstStep);

    /* Filters and image using a horizontal Prewitt filter */
    check_sts( status = ippiFilterPrewittHorizBorderGetBufferSize(roiSize, mask, srcDataType, dstDataType, numChannels, &iTmpBufSize) )

    pBuffer = ippsMalloc_8u(iTmpBufSize);

    check_sts( status = ippiFilterPrewittHorizBorder_8u16s_C1R(pSrc + srcStep, srcStep, pDst, dstStep, roiSize, mask, borderType, borderValue, pBuffer) )

    ippsFree(pBuffer);
    pBuffer = NULL;

    /* Filters and image using a vertical Prewitt filter */
    check_sts( status = ippiFilterPrewittVertBorderGetBufferSize(roiSize, mask, srcDataType, dstDataType, numChannels, &iTmpBufSize) )

    pBuffer = ippsMalloc_8u(iTmpBufSize);

    check_sts( status = ippiFilterPrewittVertBorder_8u16s_C1R(pSrc + srcStep, srcStep, pDst, dstStep, roiSize, mask, borderType, borderValue, pBuffer) )

    ippsFree(pBuffer);
    pBuffer = NULL;

    /* Filters an image using a horizontal Roberts kernel. */
    check_sts( status = ippiFilterRobertsDownBorderGetBufferSize(roiSize, mask, srcDataType, dstDataType, numChannels, &iTmpBufSize) )

    pBuffer = ippsMalloc_8u(iTmpBufSize);

    check_sts( status = ippiFilterRobertsDownBorder_8u16s_C1R(pSrc + srcStep, srcStep, pDst, dstStep, roiSize, mask, borderType, borderValue, pBuffer) )

    ippsFree(pBuffer);
    pBuffer = NULL;

    /* Filters an image using a vertical Roberts edge filter */
    check_sts( status = ippiFilterRobertsUpBorderGetBufferSize(roiSize, mask, srcDataType, dstDataType, numChannels, &iTmpBufSize) )

    pBuffer = ippsMalloc_8u(iTmpBufSize);

    check_sts( status = ippiFilterRobertsUpBorder_8u16s_C1R(pSrc + srcStep, srcStep, pDst, dstStep, roiSize, mask, borderType, borderValue, pBuffer) )

    ippsFree(pBuffer);
    pBuffer = NULL;

    /* Filters an image using a horizontal Scharr kernel */
    check_sts( status = ippiFilterScharrHorizMaskBorderGetBufferSize(roiSize, mask, srcDataType, dstDataType, numChannels, &iTmpBufSize) )

    pBuffer = ippsMalloc_8u(iTmpBufSize);

    check_sts( status = ippiFilterScharrHorizMaskBorder_8u16s_C1R(pSrc + srcStep, srcStep, pDst, dstStep, roiSize, mask, borderType, borderValue, pBuffer) )

EXIT_MAIN
    ippsFree(pBuffer);
    ippiFree(pSrc);
    ippiFree(pDst);
    printf("Exit status %d (%s)\n", (int)status, ippGetStatusString(status));
    return (int)status;
}