//
//              LAPACK++ 1.1 Linear Algebra Package 1.1
//               University of Tennessee, Knoxvilee, TN.
//            Oak Ridge National Laboratory, Oak Ridge, TN.
//        Authors: J. J. Dongarra, E. Greaser, R. Pozo, D. Walker
//                 (C) 1992-1996 All Rights Reserved
//
//                             NOTICE
//
// Permission to use, copy, modify, and distribute this software and
// its documentation for any purpose and without fee is hereby granted
// provided that the above copyright notice appear in all copies and
// that both the copyright notice and this permission notice appear in
// supporting documentation.
//
// Neither the Institutions (University of Tennessee, and Oak Ridge National
// Laboratory) nor the Authors make any representations about the suitability 
// of this software for any purpose.  This software is provided ``as is'' 
// without express or implied warranty.
//
// LAPACK++ was funded in part by the U.S. Department of Energy, the
// National Science Foundation and the State of Tennessee.

#include "lafnames.h"
#include LA_GEN_MAT_DOUBLE_H
#include LA_VECTOR_DOUBLE_H
#include "blaspp.h"
#include LA_SPD_MAT_DOUBLE_H
#include LA_SOLVE_DOUBLE_H  
#include LA_GENERATE_MAT_DOUBLE_H
#include LA_EXCEPTION_H
#include LA_UTIL_H


double residual(LaSpdMatDouble &A, LaVectorDouble &x, 
    const LaVectorDouble& b)
{
    int N = A.size(0);

    std::cout << "\tNorm_Inf(A*x-b)" << Norm_Inf(A*x-b) << std::endl;
    std::cout << "\tNorm_Inf(A) " << Norm_Inf(A) << std::endl;
    std::cout << "\tNorm_Inf(x) " << Norm_Inf(x) << std::endl;
    std::cout << "\tMacheps :" << Mach_eps_double() << std::endl;

        return Norm_Inf(A*x-b ) /
                ( Norm_Inf(A)* Norm_Inf(x) * N * Mach_eps_double());
}
    
int TestSpdLinearSolve(int N)
{
    LaSpdMatDouble A(N,N);
    LaVectorDouble x(N), b(N);

#ifndef HPPA
    const char fname[] = "TestSpdLinearSolve(LaGenMat, x, b) ";
#else
    char *fname = NULL;
#endif


    LaGenerateMatDouble(A);

    // save a snapshot of what A looked like before the solution
    LaSpdMatDouble old_A(A);


    b = 1.1;

    std::cerr << fname << ": testing LaLinearSolve(Spd,...) \n";
    LaLinearSolve(A, x, b);
    double norm;


    if ( (norm = Norm_Inf( old_A - A)) >  0.0)  // this is a hard test, not
                                         // necessary to worry about
                                         // round-off issues.  We
                                         // are testing to see A was
                                         // overwritten.
    {
        std::cerr << fname << ": overwrote 1st arg.\n";
        std::cerr << "       error norm: " << norm << std::endl;
        exit(1);
    }

#if 0
    std::cout << "A\n" << A << std::endl;
    std::cout << "old_A\n" << old_A << std::endl;
    std::cout << "x \n" << x << std::endl;
#endif

    double res = residual(A,x,b);
    if (res > 1)
    {
        std::cerr << fname << "resdiual " << res << " is to too high.\n";
        exit(1);
    }
    std::cerr << fname << ": LaLinearSolve(Spd) success.\n\n";

    // now try the in-place solver

    std::cerr << fname << ": testing LaLinearSolveIP(Spd,...) \n";
    LaLinearSolveIP(A, x, b);
    res = residual(old_A, x, b);
    if (res > 1)
    {
        std::cerr << fname << "resdiual " << res << " is to too high.\n";
        exit(1);
    }
    std::cerr << fname << ": LaLinearSolveIP(Spd) success.\n\n";
    return 0;
}


int main(int argc, char **argv)
{

    std::cout.precision(8);
    std::cout.setf(std::ios::scientific, std::ios::floatfield);

    if (argc < 3)
    {
        std::cerr << "Usage " << argv[0] << " M N " << std::endl;
        exit(1);
    }

    int M = atoi(argv[1]);
    //int N = atoi(argv[2]);

    TestSpdLinearSolve(M);

}