/*****************************************************************************
    This routine must ensure that the psVectorFitPolynomial1D works correctly.
 *****************************************************************************/
#include <stdio.h>
#include "pslib.h"
#include "psTest.h"
#include "psMemory.h"
#include "psVector.h"
#include "psImage.h"
#include "psFunctions.h"
#include "psMinimize.h"
#include <math.h>
#define NUM_DATA 20
#define POLY_ORDER 10
#define A 2.0
#define B 3.0
#define C 2.0
#define D 4.0
#define E 5.0
#define ERROR_TOLERANCE 0.10
#define IGNORE (ERROR_TOLERANCE * NUM_DATA)

double setData(double x)
{
    return(A + (B * x) + (C * x * x) + (D * x * x * x) + (E * x * x * x * x));
}

int t00()
{
    psPolynomial1D *myPoly = NULL;
    psVector *x = NULL;
    psVector *y = NULL;
    psVector *yErr = NULL;
    int i = 0;
    int currentId = psMemGetId();
    int testStatus = true;
    int memLeaks = 0;

    myPoly = psPolynomial1DAlloc(POLY_ORDER+1, PS_POLYNOMIAL_CHEB);
    x = psVectorAlloc(NUM_DATA, PS_TYPE_F64);
    y = psVectorAlloc(NUM_DATA, PS_TYPE_F64);
    yErr = psVectorAlloc(NUM_DATA, PS_TYPE_F64);

    for (i=0;i<NUM_DATA;i++) {
        x->data.F64[i] = (double) i;
        y->data.F64[i] = setData(x->data.F64[i]);
        yErr->data.F64[i] = 1.0;
    }
    p_psNormalizeVectorF64(x);

    printPositiveTestHeader(stdout,
                            "psMinimize functions",
                            "psVectorFitPolynomial1D(): CHEB, equal errors in yErr");

    psVectorFitPolynomial1D(myPoly, x, y, yErr);

    //  Remove for testing since print out differs from platforms
    //    for (i=0;i<POLY_ORDER+1;i++) {
    //        printf("Polynomial coefficient %d is %0.1f\n", i, myPoly->coeff[i]);
    //    }

    // We don't test the first or last few data items.
    for (i=IGNORE;i<NUM_DATA-IGNORE;i++) {
        double expectData = y->data.F64[i];
        double actualData = psPolynomial1DEval(x->data.F64[i], myPoly);
        if (fabs(actualData-expectData) > fabs(ERROR_TOLERANCE * expectData)) {
            printf("ERROR: Fitted data %d: (%.1f %.1f), expected was (%.1f)\n",
                   i, x->data.F64[i], actualData, expectData);
            testStatus = false;
        }
        //      Remove for testing since print out differs from platforms
        //        } else {
        //            printf("Fitted data %d: (%.1f %.1f), expected was (%.1f)\n",
        //                   i, x->data.F64[i], actualData, expectData);
        //        }
    }

    psMemCheckCorruption(1);
    psFree(myPoly);
    psFree(x);
    psFree(y);
    psFree(yErr);
    psMemCheckCorruption(1);
    memLeaks = psMemCheckLeaks(currentId,NULL,stderr);
    if (0 != memLeaks) {
        psAbort(__func__,"Memory Leaks! (%d leaks)", memLeaks);
    }

    printFooter(stdout,
                "psMinimize functions",
                "psVectorFitPolynomial1D(): CHEB, equal errors in yErr",
                testStatus);
    return (!testStatus);
}


int t01()
{
    psPolynomial1D *myPoly = NULL;
    psVector *x = NULL;
    psVector *y = NULL;
    int i = 0;
    int currentId = psMemGetId();
    int testStatus = true;
    int memLeaks = 0;

    myPoly = psPolynomial1DAlloc(POLY_ORDER+1, PS_POLYNOMIAL_CHEB);
    x = psVectorAlloc(NUM_DATA, PS_TYPE_F64);
    y = psVectorAlloc(NUM_DATA, PS_TYPE_F64);

    for (i=0;i<NUM_DATA;i++) {
        x->data.F64[i] = (double) i;
        y->data.F64[i] = setData(x->data.F64[i]);
    }
    p_psNormalizeVectorF64(x);

    printPositiveTestHeader(stdout,
                            "psMinimize functions",
                            "psVectorFitPolynomial1D(): CHEB, yErr is NULL");

    psVectorFitPolynomial1D(myPoly, x, y, NULL);

    // We don't test the first or last few data items.
    for (i=IGNORE;i<NUM_DATA-IGNORE;i++) {
        double expectData = y->data.F64[i];
        double actualData = psPolynomial1DEval(x->data.F64[i], myPoly);
        if (fabs(actualData-expectData) > fabs(ERROR_TOLERANCE * expectData)) {
            printf("ERROR: Fitted data %d: (%.1f %.1f), expected was (%.1f)\n",
                   i, x->data.F64[i], actualData, expectData);
            testStatus = false;
        }
        //      Remove for testing since print out differs from platforms
        //        } else {
        //            printf("Fitted data %d: (%.1f %.1f), expected was (%.1f)\n",
        //                   i, x->data.F64[i], actualData, expectData);
        //        }
    }

    psMemCheckCorruption(1);
    psFree(myPoly);
    psFree(x);
    psFree(y);
    psMemCheckCorruption(1);
    memLeaks = psMemCheckLeaks(currentId,NULL,stderr);
    if (0 != memLeaks) {
        psAbort(__func__,"Memory Leaks! (%d leaks)", memLeaks);
    }

    printFooter(stdout,
                "psMinimize functions",
                "psVectorFitPolynomial1D(): CHEB, yErr is NULL",
                testStatus);
    return (!testStatus);
}


int t02()
{
    psPolynomial1D *myPoly = NULL;
    psVector *x = NULL;
    psVector *y = NULL;
    int i = 0;
    int currentId = psMemGetId();
    int testStatus = true;
    int memLeaks = 0;

    myPoly = psPolynomial1DAlloc(POLY_ORDER+1, PS_POLYNOMIAL_CHEB);
    x = psVectorAlloc(NUM_DATA, PS_TYPE_F64);
    y = psVectorAlloc(NUM_DATA, PS_TYPE_F64);

    for (i=0;i<NUM_DATA;i++) {
        x->data.F64[i] = (double) i;
        y->data.F64[i] = setData(x->data.F64[i]);
    }
    p_psNormalizeVectorF64(x);

    printPositiveTestHeader(stdout,
                            "psMinimize functions",
                            "psVectorFitPolynomial1D(): CHEB, x, yErr is NULL");

    psVectorFitPolynomial1D(myPoly, NULL, y, NULL);

    // We don't test the first or last few data items.
    for (i=IGNORE;i<NUM_DATA-IGNORE;i++) {
        double expectData = y->data.F64[i];
        double actualData = psPolynomial1DEval(x->data.F64[i], myPoly);
        if (fabs(actualData-expectData) > fabs(ERROR_TOLERANCE * expectData)) {
            printf("ERROR: Fitted data %d: (%.1f %.1f), expected was (%.1f)\n",
                   i, x->data.F64[i], actualData, expectData);
            testStatus = false;
        }
        //      Remove for testing since print out differs from platforms
        //        } else {
        //            printf("Fitted data %d: (%.1f %.1f), expected was (%.1f)\n",
        //                   i, x->data.F64[i], actualData, expectData);
        //        }
    }

    psMemCheckCorruption(1);
    psFree(myPoly);
    psFree(x);
    psFree(y);
    psMemCheckCorruption(1);
    memLeaks = psMemCheckLeaks(currentId,NULL,stderr);
    if (0 != memLeaks) {
        psAbort(__func__,"Memory Leaks! (%d leaks)", memLeaks);
    }

    printFooter(stdout,
                "psMinimize functions",
                "psVectorFitPolynomial1D(): CHEB, x, yErr is NULL",
                testStatus);
    return (!testStatus);
}

int t03()
{
    psPolynomial1D *myPoly = NULL;
    int currentId = psMemGetId();
    int testStatus = true;
    int memLeaks = 0;

    printPositiveTestHeader(stdout,
                            "psMinimize functions",
                            "psVectorFitPolynomial1D(): CHEB, yErr is NULL");

    psLogMsg(__func__,PS_LOG_INFO,"Following should generate an error for null arguments.");
    myPoly = psVectorFitPolynomial1D(NULL, NULL, NULL, NULL);
    if (myPoly != NULL) {
        printf("ERROR: psVectorFitPolynomial1D() returned a non-NULL polynomial.\n");
        testStatus = false;
    }

    psMemCheckCorruption(1);
    memLeaks = psMemCheckLeaks(currentId,NULL,stderr);
    if (0 != memLeaks) {
        psAbort(__func__,"Memory Leaks! (%d leaks)", memLeaks);
    }
    printFooter(stdout,
                "psMinimize functions",
                "psVectorFitPolynomial1D(): CHEB, yErr is NULL",
                testStatus);
    return (!testStatus);
}

int main()
{
    t00();
    t01();
    t02();
    t03();
}
