/* 
// 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 image foreground/background segmentation
// using Intel IPP functions:
//     ippiFGMMGetBufferSize_8u_C3R
//     ippiFGMMInit_8u_C3R
//     ippiFGMMForeground_8u_C3R
//     ippiFGMMBackground_8u_C3R


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

#define WIDTH  32      /* image width  */
#define HEIGHT 32      /* image height */
#define NUM_FRAME 100
#define MAX_N_RECT 3   /* number of rectangles */

static struct {
    int x;
    int y;
    int w;
    int h;
    int dx;
    int dy;
    Ipp8u color[3];
} rect[MAX_N_RECT];  /* Parameters of rectangle */

/* 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)
{
    int r = 0, frame = 0;
    IppStatus status = ippStsNoErr;
    Ipp8u* pSrc = NULL, *pDstF = NULL, *pDstB = NULL; /* Pointers to source/foreground/background images */
    int srcStep = 0, dstFStep = 0, dstBStep = 0;      /* Steps, in bytes, through the source/foreground/background images */
    IppiSize roiSize = { WIDTH, HEIGHT }; /* Size of source ROI in pixels */
    int specSize = 0;                     /* Size of the working buffer and spec structure */
    IppFGMMState_8u_C3R*  pState = NULL;  /* Pointer to the state */
    IppFGMModel           pModel = {0};   /* Pointer to the parameters structure */
    Ipp8u value[3] = {0,0,0};
    int maxNGauss = 5; /* Maximum number of allowed Gaussians */

    pSrc  = ippiMalloc_8u_C3(roiSize.width, roiSize.height, &srcStep);
    pDstF = ippiMalloc_8u_C1(roiSize.width, roiSize.height, &dstFStep);
    pDstB = ippiMalloc_8u_C3(roiSize.width, roiSize.height, &dstBStep);

    check_sts( status = ippiFGMMGetBufferSize_8u_C3R(roiSize, maxNGauss, &specSize) )

    pState = (IppFGMMState_8u_C3R*)ippsMalloc_8u(specSize);

    pModel.numFrames = 50;
    pModel.varInit = 15.0;
    pModel.varMin = 4.0;
    pModel.varMax = 75.0;
    pModel.varWBRatio = 4.0f*4.0f;
    pModel.bckgThr = 0.9f;
    pModel.varNGRatio = 3.0f*3.0f;
    pModel.reduction = 0.05f;
    pModel.shadowValue = 127;
    pModel.shadowFlag = 1;
    pModel.shadowRatio = 0.5;
    pModel.maxNGauss = maxNGauss;

    check_sts( status = ippiFGMMInit_8u_C3R(roiSize, maxNGauss, &pModel, pState) )

    for (r = 0; r<MAX_N_RECT; r++)
    {
        rect[r].x = 0;
        rect[r].y = 0;
        rect[r].w = 10 + r;
        rect[r].h = 5 + r;
        rect[r].dx = 1;
        rect[r].dy = 1;
        rect[r].color[0] = (Ipp8u)(10 * r);
        rect[r].color[1] = (Ipp8u)(20 * r);
        rect[r].color[2] = (Ipp8u)( 3 * r);
    }
    for (frame = 0; frame < NUM_FRAME; frame++)
    {
        Ipp8u valueRect[3] = { 0, 0, 0 };
        IppiSize roiRectSize = {0};
        /* Draw rectangle to simulate moving object */
        check_sts( status = ippiSet_8u_C3R(value, pSrc, srcStep, roiSize) ) /* zero image */

        for (r = 0; r<MAX_N_RECT; r++)
        {
            valueRect[0] = (Ipp8u)(rect[r].color[0] + frame);
            valueRect[1] = (Ipp8u)(rect[r].color[1] + frame);
            valueRect[2] = (Ipp8u)(rect[r].color[2] + frame);
            roiRectSize.width  = rect[r].w;
            roiRectSize.height = rect[r].h;
            check_sts( status = ippiSet_8u_C3R(valueRect, pSrc + srcStep*rect[r].y + 3 * rect[r].x, srcStep, roiRectSize) )
            if (status<0) break;

            if (rect[r].x + rect[r].w >= roiSize.width){
                rect[r].dx = -1;
            }
            else if (rect[r].x <= 0){
                rect[r].dx = 1;
            }
            if (rect[r].y + rect[r].h >= roiSize.height){
                rect[r].dy = -1;
            }
            else if (rect[r].y == 0){
                rect[r].dy = 1;
            }
            rect[r].x += rect[r].dx;
            rect[r].y += rect[r].dy;
        }
        check_sts( status = ippiFGMMForeground_8u_C3R(pSrc, srcStep, pDstF, dstFStep, roiSize, pState, &pModel, -1) )

        check_sts( status = ippiFGMMBackground_8u_C3R(pDstB, dstBStep, roiSize, pState) )
    }

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