/* 
// Copyright 2015 2018 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 detecting lines in the source image using
// the probabilistic Hough transform using Intel(R) Integrated Primitives (Intel(R) IPP) functions:
//     ippiHoughProbLineGetSize_8u_C1R
//     ippiHoughProbLineInit_8u32f_C1R
//     ippiHoughProbLine_8u32f_C1R


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

#define WIDTH  800  /* image width  */
#define HEIGHT 600  /* 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 Intel(R) IPP function returned status different from ippStsNoErr */

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

int main(void)
{
    int i = 0;
    IppStatus status = ippStsNoErr;
    Ipp8u* pSrc = NULL;    /* Pointers to source images */
    int srcStep = 0;       /* Steps, in bytes, through the source images */
    IppiSize roiSize = { WIDTH, HEIGHT };    /* Size of source ROI in pixels */
    int bufferSize = 0, specSize = 0;/* Size of the working buffer and spec structure */
    Ipp8u *pBuffer = NULL;
    IppiHoughProbSpec*  pSpec = NULL;
    IppiPoint pLineP[128]; /* Pointer to output array of detected lines */
    int threshold = 20, lineLength = 40, ineGap = 4, maxLineCount = 128;
    IppPointPolar delta = { 1, (Ipp32f)IPP_PI / 180.f / 3.f }; /* Step of discretization */
    int  lineCount = 0; /* Number of detected lines. */

    pSrc = ippiMalloc_8u_C1(roiSize.width, roiSize.height, &srcStep);

    check_sts( status = ippiSet_8u_C1R(0, pSrc, srcStep, roiSize) ) /* Zero image */

    for (i = 0; i<70; i++) {     /* Draw line (100,100)-(170,170) */
        pSrc[(i + 100)*srcStep + i + 100] = 255;
    }

    /* Compute the size of the working buffer and spec structure */
    check_sts( status = ippiHoughProbLineGetSize_8u_C1R(roiSize, delta, &specSize, &bufferSize) )

    pBuffer = (Ipp8u*)ippsMalloc_8u(bufferSize);
    pSpec = (IppiHoughProbSpec*)ippsMalloc_8u(specSize);

    /* Initialize the specification structure */
    check_sts( status = ippiHoughProbLineInit_8u32f_C1R(roiSize, delta, ippAlgHintNone, pSpec) )

    /* Detect lines in the source image using the probabilistic Hough transform. */
    check_sts( status = ippiHoughProbLine_8u32f_C1R(pSrc, srcStep, roiSize, threshold, lineLength, ineGap, pLineP, maxLineCount, &lineCount, pBuffer, pSpec) )

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