// test_lsfit6.java

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

public class test_lsfit6
{
  int nx = 6;
  int ny = 6;
  int nz = 6;
  int nt = 6;
  int nu = 6;
  int nv = 6;
  double hx, hy, hz, ht, hu, hv;
  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 vg[] = new double[nv];
  double Ug[][][][][][] = new double[nx][ny][nz][nt][nu][nv]; 
  double Ua[][][][][][] = new double[nx][ny][nz][nt][nu][nv]; 
  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 vmin = -1.0;
  double vmax =  1.0;
  double x, y, z, t, u, v, U;
  double err, maxerr;
  boolean last = false;
  DecimalFormat f4 = new DecimalFormat("0.000");

  public test_lsfit6()
  {
    System.out.println("test_lsfit6.java");
    System.out.println(" ");
    // initialize hx, hy, hy, ht, hu, hv, xg, yg, zg, tg, ,ug, vg, 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);
    hv = (vmax-vmin)/(double)(nv-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;
    for(int iiiiii=0; iiiiii<nv; iiiiii++) vg[iiiiii] = vmin+hv*(double)iiiiii;
    System.out.println("nx="+nx+", ny="+ny+", nz="+nz+", nt="+nt+
                       ", nu="+nu+", nv="+nv);
    System.out.println("hx="+hx+", hy="+hy+", hz="+hz);
    System.out.println("ht="+ht+", hy="+hu+", hv="+hv);

    lsfit LS1 = new lsfit(6, 1, 0); // dimension, power, points
    lsfit LS2 = new lsfit(6, 2, 0); // dimension, power, points
    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];
	        for(int iiiiii=0; iiiiii<nv; iiiiii++)
	        {
	          v = vg[iiiiii];
	          if(k==1) Ua[i][ii][iii][iiii][iiiii][iiiiii] =
                             u1(x,y,z,t,u,v);
	          if(k==2) Ua[i][ii][iii][iiii][iiiii][iiiiii] = 
                             u2(x,y,z,t,u,v);
	          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 || iiiiii==0 || iiiiii==nv-1)
	          {
		    Ug[i][ii][iii][iiii][iiiii][iiiiii] = 
                      Ua[i][ii][iii][iiii][iiiii][iiiiii];
	          }
	          else // not boundary
	          {
	            Ug[i][ii][iii][iiii][iiiii][iiiiii] = 0.0;
	          }
		} // iiiiii
	      } // iiiii
	    } // iiii
	  } // iii
        } // ii
      } // i
      System.out.println("Ua, Ug initialized");

      System.out.println("test_lsfit6.java (6,"+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];
	        for(int iiiiii=0; iiiiii<nv; iiiiii++)
	        {
	          v = vg[iiiiii];
                  if(k==1) U = u1(x,y,z,t,u,v);
                  if(k==2) U = u2(x,y,z,t,u,v);
	          if(i==nx-1 && ii==ny-1 && iii==nz-1 && iiii==nt-1 &&
                     iiiii==nu-1 && iiiiii==nv-1) last = true;
	          if(k==1) LS1.fit6_data(x, y, z, t, u, v, U, last);
	          if(k==2) LS2.fit6_data(x, y, z, t, u, v, U, last);
		} // iiiiii
	      } // iiiii
	    } // iiii
	  } // iii
        } // ii
      } // i
      System.out.println("test_lsfit6.java (6,"+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]+
                             " v^"+LS1.Fpwr[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]+
                             " v^"+LS2.Fpwr[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];
	        for(int iiiiii=0; iiiiii<nv; iiiiii++)
	        {
	          v = vg[iiiii];
	          if(k==1) U = LS1.eval6_poly(x, y, z, t, u, v);
	          if(k==2) U = LS2.eval6_poly(x, y, z, t, u, v);
                  if(k==1) err = Math.abs(U-u1(x,y,z,t,u,v));
                  if(k==2) err = Math.abs(U-u2(x,y,z,t,u,v));

                  // err = Math.abs(U-Ua[i][ii][iii][iiii][iiiii][iiiiii]);
                  maxerr = Math.max(maxerr,err);
		} // iiiiii
	      } // iiiii
	    } // iiii
	  } // iii
        } // ii
      } // i
      System.out.println("test_lsfit6.java (6,"+k+") data fit maxerr="+maxerr);
    } // k Ug and Ua set up for Bpoly

    // boundary fit
    lsfit LSb = new lsfit(6, 2, 0); // dimension, power, points
    LSb.debug = false;
    LSb.fit6_bpoly(nx, ny, nz, nt, nu, nv, xg, yg, zg, tg, ug, vg, Ug);
    System.out.println("test_lsfit6.java (6,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]+
                           " v^"+LSb.Fpwr[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];
	      for(int iiiiii=0; iiiiii<nv; iiiiii++)
	      {
	        v = vg[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 ||
                   iiiiii==0 || iiiiii==nv-1)
	        {
		    U = LSb.eval6_poly(x, y, z, t, u, v);
                  err = Math.abs(U-Ua[i][ii][iii][iiii][iiiii][iiiiii]);
                  maxerr = Math.max(maxerr,err);
	        }
	      } // iiiiii
	    } // iiiii
	  } // iiii
	} // iii
      } // ii
    } // i
    System.out.println("test_lsfit6.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];
	      for(int iiiiii=1; iiiiii<nu-1; iiiiii++)
	      {
	        v = vg[iiiiii];
	        U = LSb.eval6_poly(x, y, z, t, u, v);
                err = Math.abs(U-Ua[i][ii][iii][iiii][iiiii][iiiiii]);
                maxerr = Math.max(maxerr,err);
	      } // iiiiii
	    } // iiiii
	  } // iiii
	} // iii
      } // ii
    } // i
    System.out.println("test_lsfit6.java inside boundary, maxerr="+maxerr);

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

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

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

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