/*---------------------------------------------------------------------------*
 *                                   IT++			             *
 *---------------------------------------------------------------------------*
 * Copyright (c) 1995-2004 by Tony Ottosson, Thomas Eriksson, Pål Frenger,   *
 * Tobias Ringström, and Jonas Samuelsson.                                   *
 *                                                                           *
 * Permission to use, copy, modify, and distribute this software and its     *
 * documentation under the terms of the GNU General Public License is hereby *
 * granted. No representations are made about the suitability of this        *
 * software for any purpose. It is provided "as is" without expressed or     *
 * implied warranty. See the GNU General Public License for more details.    *
 *---------------------------------------------------------------------------*/

/*!
  \file
  \brief Numerical Integration
  \author Tony Ottosson

  1.1

  2004/10/14 16:41:22 
*/

#include "base/integration.h"

#include "base/specmat.h"
#include "base/matfunc.h"
#include "base/scalfunc.h"
#include "base/help_functions.h"

namespace itpp {



  double quadstep(double (*f)(double), const double a, const double b, const double fa, const double fm, const double fb, const double is)
  {
    double Q, m, h, fml, fmr, i1, i2;
    vec x(2), y(2);

    m = (a+b)/2; h = (b-a)/4;
    x = vec_2(a+h, b-h);
    y = vec_function(f, x);
    fml = y(0); fmr = y(1);

    i1 = h/1.5 * (fa + 4*fm + fb);
    i2 = h/3 * (fa + 4*(fml + fmr) + 2*fm + fb);
    i1 = (16*i2 - i1)/15;

    if( (is + (i1-i2) == is) || (m<=a) || (b<=m) ) {
      if( (m <= a) || (b<=m) ) {
	it_warning("Interval contains no more machine number. Required tolerance may not be met");
      }
      Q = i1;
      return Q;
    } else {
      Q = quadstep(f, a, m, fa, fml, fm, is) + quadstep(f, m, b, fm, fmr, fb, is);
    }
    return Q;
  }


  double quad(double (*f)(double), const double a, const double b, const double tol)
  {
    vec x(3), y(3), yy(5);
    double Q, fa, fm, fb, is;

    x = vec_3(a, (a+b)/2, b);
    y = vec_function(f, x);
    fa = y(0); fm = y(1); fb = y(2);
    yy = vec_function(f, a + vec(".9501 .2311 .6068 .4860 .8913")*(b-a));
    is = (b-a)/8*(sum(y)+sum(yy));

    if (is == 0.0)
      is = b-a;

    is = is*tol/std::numeric_limits<double>::epsilon();
    Q = quadstep(f, a, b, fa, fm, fb, is);

    return Q;
  }


  //--------------------- quadl() ----------------------------------------

  double quadlstep(double (*f)(double), const double a, const double b, const double fa, const double fb, const double is)
  {
    double Q, h, m, alpha, beta, mll, ml, mr, mrr, fmll, fml, fm, fmr, fmrr, i1, i2;
    vec x(5), y(5);

    h=(b-a)/2; m=(a+b)/2;
    alpha=sqrt(2.0/3); beta=1.0/sqrt(5.0);
    mll=m-alpha*h; ml=m-beta*h; mr=m+beta*h; mrr=m+alpha*h;
    x(0)=mll; x(1) = ml; x(2) = m; x(3) = mr; x(4) = mrr;

    y=vec_function(f, x);
    
    fmll=y(0); fml=y(1); fm=y(2); fmr=y(3); fmrr=y(4);
    
    i2 = (h/6)*(fa+fb+5*(fml+fmr));
    i1 = (h/1470)*(77*(fa+fb)+432*(fmll+fmrr)+625*(fml+fmr)+672*fm);

    if( (is + (i1-i2) == is) || (mll<=a) || (b<=mrr) ) {
      if( (m <= a) || (b<=m) ) {
	it_warning("Interval contains no more machine number. Required tolerance may not be met");
      }
      Q = i1;
      return Q;
    } else {
      Q=quadlstep(f, a, mll, fa, fmll, is) + quadlstep(f, mll, ml, fmll, fml, is) + quadlstep(f, ml, m, fml, fm, is) +
      quadlstep(f, m, mr, fm, fmr, is) + quadlstep(f, mr, mrr, fmr, fmrr, is) + quadlstep(f, mrr, b, fmrr, fb, is);
    }
    return Q;
  }

  double quadl(double (*f)(double), const double a, const double b, const double tol)
  {
    double Q, m, h, alpha, beta, x1, x2, x3, fa, fb, i1, i2, is, s, erri1, erri2, R;
    vec x(13), y(13);
    double tol2 = tol;

    m=(a+b)/2; h=(b-a)/2;
    
    alpha=std::sqrt(2.0/3); beta=1/sqrt(5.0);

    x1=.942882415695480; x2=.641853342345781; x3=.236383199662150;
    x(0)=a; x(1)=m-x1*h; x(2)=m-alpha*h; x(3)=m-x2*h;
    x(4)=m-beta*h; x(5)=m-x3*h; x(6)=m; x(7)=m+x3*h; 
    x(8)=m+beta*h; x(9)=m+x2*h; x(10)=m+alpha*h; 
    x(11)=m+x1*h; x(12)=b;

    y=vec_function(f, x);

    fa=y(0); fb=y(12);
    i2=(h/6)*(y(0)+y(12)+5*(y(4)+y(8)));
    i1=(h/1470)*(77*(y(0)+y(12))+432*(y(2)+y(10))+ 625*(y(4)+y(8))+672*y(6));

    is=h*(.0158271919734802*(y(0)+y(12))+.0942738402188500*(y(1)+y(11))+.155071987336585*(y(2)+y(10))+ 
	  .188821573960182*(y(3)+y(9))+.199773405226859 *(y(4)+y(8))+.224926465333340*(y(5)+y(7))+.242611071901408*y(6));

    s=sign(is);
    if (s == 0.0)
      s = 1;
    
    erri1 = std::abs(i1-is);
    erri2 = std::abs(i2-is);

    R=1;
    if (erri2 != 0.0)
      R = erri1/erri2;

    if (R>0 && R<1)
      tol2=tol2/R;

    is = s*std::abs(is)*tol2/std::numeric_limits<double>::epsilon();
    if (is == 0.0)
      is=b-a;

    Q = quadlstep(f, a, b, fa, fb, is);

    return Q;
  }


} // namespace itpp