/* 
// 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 computing the affine coefficients for the transform
// that rotates an image using Intel IPP functions:
//     ippiGetRotateTransform
//     ippiWarpAffineLinearInit
//     ippiWarpGetBufferSize
//     ippiWarpAffineLinear_8u_C3R


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

#define WIDTH   64  /* source image width */
#define HEIGHT  64  /* source image height */
#define NUM_CHN  3

/* 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;
    IppiWarpSpec* pSpec = NULL; /* Pointer to the specification structure */
    IppiSize srcSize = { WIDTH, HEIGHT }, dstSize = { WIDTH, HEIGHT }; /* Size of source/destination images */
    int srcStep, dstStep; /* Steps, in bytes, through the source/destination images */
    Ipp8u* pSrc = NULL, *pDst = NULL;   /* Pointers to source/destination images */
    double angle = 30., xShift = 2., yShift = 1.;
    double coeffs[2][3] = {0};
    IppiPoint dstOffset = { 0, 0 }; /* Offset of the destination image ROI with respect to the destination image origin */
    int specSize = 0, initSize = 0, bufSize = 0; /* Work buffer size */
    Ipp8u* pBuffer = NULL;
    IppiBorderType   borderType = ippBorderConst;
    IppiWarpDirection direction = ippWarpForward; /* Transformation direction */
    Ipp64f pBorderValue[NUM_CHN] = {0};

    pSrc = ippiMalloc_8u_C3(srcSize.width, srcSize.height, &srcStep);
    pDst = ippiMalloc_8u_C3(dstSize.width, dstSize.height, &dstStep);

    check_sts( status = ippiGetRotateTransform(angle, xShift, yShift, (double(*)[3])coeffs) )

    /* Spec and init buffer sizes */
    check_sts( status = ippiWarpAffineGetSize(srcSize, dstSize, ipp8u, coeffs, ippLinear, direction, borderType, &specSize, &initSize) )

    pSpec = (IppiWarpSpec*)ippsMalloc_8u(specSize);

    /* Filter initialization */
    check_sts( status = ippiWarpAffineLinearInit(srcSize, dstSize, ipp8u, coeffs, direction, NUM_CHN, borderType, pBorderValue, 0, pSpec) )

    /* Get work buffer size */
    check_sts( status = ippiWarpGetBufferSize(pSpec, dstSize, &bufSize) )

    pBuffer = ippsMalloc_8u(bufSize);

    /* WarpAffine processing */
    check_sts( status = ippiWarpAffineLinear_8u_C3R(pSrc, srcStep, pDst, dstStep, dstOffset, dstSize, pSpec, pBuffer) )

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