// test_lsfit5.java

import java.io.*;
import java.text.*;

public class test_lsfit5
{
  int nx = 6;
  int ny = 6;
  int nz = 6;
  int nt = 6;
  int nu = 6;
  double hx, hy, hz, ht, hu;
  double xg[] = new double[nx]; // X grid, does not need to be uniform
  double yg[] = new double[ny];
  double zg[] = new double[nz];
  double tg[] = new double[nt];
  double ug[] = new double[nu];
  double Ug[][][][][] = new double[nx][ny][nz][nt][nu]; 
  double Ua[][][][][] = new double[nx][ny][nz][nt][nu]; 
  double xmin = -1.0;
  double xmax =  1.0;
  double ymin = -1.0;
  double ymax =  1.0;
  double zmin = -1.0;
  double zmax =  1.0;
  double tmin = -1.0;
  double tmax =  1.0;
  double umin = -1.0;
  double umax =  1.0;
  double x, y, z, t, u, U;
  double err, maxerr;
  boolean last = false;
  DecimalFormat f4 = new DecimalFormat("0.000");
  lsfit LS1 = new lsfit(5, 1, 0); // dimension, power, points
  lsfit LS2 = new lsfit(5, 2, 0); // dimension, power, points

  public test_lsfit5()
  {
    System.out.println("test_lsfit5.java");
    System.out.println(" ");
    // initialize hx, hy, hy, ht, hu, xg, yg, zg, tg, Ug
    hx = (xmax-xmin)/(double)(nx-1);
    hy = (ymax-ymin)/(double)(ny-1);
    hz = (zmax-zmin)/(double)(nz-1);
    ht = (tmax-tmin)/(double)(nt-1);
    hu = (umax-umin)/(double)(nu-1);
    for(int i=0; i<nx; i++) xg[i] = xmin+hx*(double)i;
    for(int ii=0; ii<ny; ii++) yg[ii] = ymin+hy*(double)ii;
    for(int iii=0; iii<nz; iii++) zg[iii] = zmin+hz*(double)iii;
    for(int iiii=0; iiii<nt; iiii++) tg[iiii] = tmin+ht*(double)iiii;
    for(int iiiii=0; iiiii<nu; iiiii++) ug[iiiii] = umin+hu*(double)iiiii;
    System.out.println("nx="+nx+", ny="+ny+", nz="+nz+", nt="+nt+", nu="+nu);
    System.out.println("hx="+hx+", hy="+hy+", hz="+hz);
    System.out.println("ht="+ht+", hy="+hu);

    for(int k=1; k<=2; k++)
    {
      // set up test data
      for(int i=0; i<nx; i++)
      {
        x = xg[i];
        for(int ii=0; ii<ny; ii++)
        {
	  y = yg[ii];
	  for(int iii=0; iii<nz; iii++)
	  {
	    z = zg[iii];
	    for(int iiii=0; iiii<nt; iiii++)
	    {
	      t = tg[iiii];
	      for(int iiiii=0; iiiii<nu; iiiii++)
	      {
	        u = ug[iiiii];
	        if(k==1) Ua[i][ii][iii][iiii][iiiii] = u1(x,y,z,t,u);
	        if(k==2) Ua[i][ii][iii][iiii][iiiii] = u2(x,y,z,t,u);
	        if(i==0 || ii==0 || i==nx-1 || ii==ny-1 || iii==0 || iii==nz-1 ||
                   iiii==0 || iiii==nt-1 || iiiii==0 || iiiii==nu-1)
	        {
		    Ug[i][ii][iii][iiii][iiiii] = Ua[i][ii][iii][iiii][iiiii];
	        }
	        else // not boundary
	        {
	          Ug[i][ii][iii][iiii][iiiii] = 0.0;
	        }
	      } // iiiii
	    } // iiii
	  } // iii
        } // ii
      } // i
      System.out.println("Ua, Ug initialized");

      System.out.println("test_lsfit5.java (5,"+k+") data fit");
      LS1.debug = false;
      LS2.debug = false;

      last = false;
      for(int i=0; i<nx; i++)
      {
        x = xg[i];
        for(int ii=0; ii<ny; ii++)
        {
	  y = yg[ii];
	  for(int iii=0; iii<nz; iii++)
	  {
	    z = zg[iii];
	    for(int iiii=0; iiii<nt; iiii++)
	    {
	      t = tg[iiii];
	      for(int iiiii=0; iiiii<nu; iiiii++)
	      {
	        u = ug[iiiii];
                if(k==1) U = u1(x,y,z,t,u);
                if(k==2) U = u2(x,y,z,t,u);
	        if(i==nx-1 && ii==ny-1 && iii==nz-1 && iiii==nt-1 &&
                   iiiii==nu-1) last = true;
	        if(k==1) LS1.fit5_data(x, y, z, t, u, U, last);
	        if(k==2) LS2.fit5_data(x, y, z, t, u, U, last);
	      } // iiiii
	    } // iiii
	  } // iii
        } // ii
      } // i
      System.out.println("test_lsfit5.java (5,"+k+") data fit coefficients");
      if(k==1)
      {
        for(int i=0; i<LS1.nnn; i++)
	  System.out.println("C["+i+"]="+LS1.C[i]+"  x^"+LS1.Apwr[i]+
                             " y^"+LS1.Bpwr[i]+" z^"+LS1.Cpwr[i]+
                             " t^"+LS1.Dpwr[i]+" u^"+LS1.Epwr[i]);
      }
      if(k==2)
      {
        for(int i=0; i<LS2.nnn; i++)
	  System.out.println("C["+i+"]="+LS2.C[i]+"  x^"+LS2.Apwr[i]+
                             " y^"+LS2.Bpwr[i]+" z^"+LS2.Cpwr[i]+
                             " t^"+LS2.Dpwr[i]+" u^"+LS2.Epwr[i]);
      }

      // check
      maxerr = 0.0;
      for(int i=0; i<nx; i++)
      {
        x = xg[i];
        for(int ii=0; ii<ny; ii++)
        {
	  y = yg[ii];
	  for(int iii=0; iii<nz; iii++)
	  {
	    z = zg[iii];
	    for(int iiii=0; iiii<nt; iiii++)
	    {
	      t = tg[iiii];
	      for(int iiiii=0; iiiii<nu; iiiii++)
	      {
	        u = ug[iiiii];
	        if(k==1) U = LS1.eval5_poly(x, y, z, t, u);
	        if(k==2) U = LS2.eval5_poly(x, y, z, t, u);
                err = Math.abs(U-Ua[i][ii][iii][iiii][iiiii]);
                maxerr = Math.max(maxerr,err);
	      } // iiiii
	    } // iiii
	  } // iii
        } // ii
      } // i
      System.out.println("test_lsfit5.java (5,"+k+") data fit maxerr="+maxerr);
    } // k Ug and Ua set up for Bpoly

    // boundary fit
    lsfit LSb = new lsfit(5, 2, 0); // dimension, power, points
    LSb.debug = false;
    LSb.fit5_bpoly(nx, ny, nz, nt, nu, xg, yg, zg, tg, ug, Ug);
    System.out.println("test_lsfit5.java (5,2) data fit coefficients");
    for(int i=0; i<LSb.nnn; i++)
	System.out.println("C["+i+"]="+LSb.C[i]+"  x^"+LSb.Apwr[i]+
                           " y^"+LSb.Bpwr[i]+" z^"+LSb.Cpwr[i]+
                           " t^"+LSb.Dpwr[i]+" u^"+LSb.Epwr[i]);

    // check boundary
    for(int i=0; i<nx; i++)
    {
      x = xg[i];
      for(int ii=0; ii<ny; ii++)
      {
	y = yg[ii];
	for(int iii=0; iii<nz; iii++)
	{
	  z = zg[iii];
	  for(int iiii=0; iiii<nt; iiii++)
	  {
	    t = tg[iiii];
	    for(int iiiii=0; iiiii<nu; iiiii++)
	    {
	      u = ug[iiii];
	      if(i==0 || ii==0 || i==nx-1 || ii==ny-1 || iii==0 || iii==nz-1 ||
                 iiii==0 || iiii==nt-1 || iiiii==0 || iiiii==nu-1)
	      {
		U = LSb.eval5_poly(x, y, z, t, u);
                err = Math.abs(U-Ua[i][ii][iii][iiii][iiiii]);
                maxerr = Math.max(maxerr,err);
	      }
	    } // iiiii
	  } // iiii
	} // iii
      } // ii
    } // i
    System.out.println("test_lsfit5.java, on boundary maxerr="+maxerr);

    // inside boundary fit
    // check boundary
    for(int i=1; i<nx-1; i++)
    {
      x = xg[i];
      for(int ii=1; ii<ny-1; ii++)
      {
	y = yg[ii];
	for(int iii=1; iii<nz-1; iii++)
	{
	  z = zg[iii];
	  for(int iiii=1; iiii<nt-1; iiii++)
	  {
	    t = tg[iiii];
	    for(int iiiii=1; iiiii<nu-1; iiiii++)
	    {
	      u = ug[iiiii];
	      U = LSb.eval5_poly(x, y, z, t, u);
              err = Math.abs(U-Ua[i][ii][iii][iiii][iiiii]);
              maxerr = Math.max(maxerr,err);
	    } // iiiii
	  } // iiii
	} // iii
      } // ii
    } // i
    System.out.println("test_lsfit5.java inside boundary, maxerr="+maxerr);

    System.out.println("finished test_lsfit5.java");
  } // end test_lsfit5

    double u2(double x, double y, double z, double t, double u)
  {
    return 1.0 + 2.0*x + 3.0*y + 4.0*z + 5.0*t + 6.0*u +
           7.0*x*x + 8.0*x*y + 9.0*x*z + 10.0*x*t + 11.0*x*u +
          12.0*y*y + 13.0*y*z + 14.0*y*t + 15.0*y*u + 16.0*z*z +
          17.0*z*t + 18.0*z*u + 19.0*t*t + 20.0*t*u + 21.0*u*u;
  }

    double u1(double x, double y, double z, double t, double u)
  {
    return 1.0 + 2.0*x + 3.0*y + 4.0*z + 5.0*t + 6.0*u;
  }

  public static void main (String[] args)
  {
    new test_lsfit5();
  } // end main of test_lsfit5.java
} // end class test_lsfit5