// lsfit_file.java  input plot?d_gl type file
//                  num-var num-var1 num-var2 ...
//        data      var1 var2 ... result   max var7
// java -cp . lsfit_file data-file highest order-1-to-4

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

class lsfit_file
{
  int np; // highest power of fit, default 2
  int nvar; // number of independent variables 1 to 7
  int nx, ny, nz, nt, nu, nv, nw; // number of each variable
  String varnum[] = {"nx", "ny", "nz", "nt", "nv", "nu", "nw"};
  int nss[] = new int[7]; // nx, ny, nz, ...
  int dof; // number of data points nx*ny*nz...
  double x, y, z, t, u, v, w, U; // read a line
  double D[] = new double[8]; // x, y, z, ...
  double xmin, xmax, ymin, ymax, zmin, zmax, tmin, tmax;
  double umin, umax, vmin, vmax, wmin, wmax;
  FileReader file_in;
  BufferedReader in;
  String input_line;
  int mode = 0;
  String intStr;
  boolean lastl;
  boolean debug = false; 
  int index, last, len;
  double maxerr, err, Uf;

  lsfit_file(String file_name, int npp)
  {
    System.out.println("about to open "+file_name);
    np = npp;
    try
    {
      file_in = new FileReader(file_name);
      in = new BufferedReader(file_in);
    } // end try
    catch(Exception e)
    {
      System.out.println("lsfit_file unable to open file "+file_name);
      return;
    } // end catch

    // read first line
    try
    {
      input_line = in.readLine();
      if(debug) System.out.println("input_line="+input_line);
      len = input_line.length();
      index = 0;
      if(debug) System.out.println("index="+index);
      last = input_line.indexOf(' ',index+1);
      if(debug) System.out.println("last="+last);
      intStr = input_line.substring(index,last);
      if(debug) System.out.println("intStr="+intStr);
      nvar = Integer.parseInt(intStr);
      if(debug) System.out.println("nvar="+nvar);

      for(int i=0; i<nvar; i++)
      {
	index = input_line.indexOf(' ',last); // nss[i]
	if(debug) System.out.println("index="+index);
	last = input_line.indexOf(' ',index+1);
        if(last<0) last = len;
	if(debug) System.out.println("last="+last);
        intStr = input_line.substring(index+1,last);
	if(debug) System.out.println("intStr="+intStr);
	nss[i] = Integer.parseInt(intStr);
	if(debug) System.out.println("nsss="+nss[i]);
      }
    } // end try
    catch(Exception e)
    {
      System.out.println("read first line exception"+e);
      return;
    } // end catch
    dof = 1;
    for(int i=0; i<nvar; i++)
    {
      dof = dof * nss[i];
      System.out.println(varnum[i]+" ="+nss[i]);
    }
    System.out.println("number of samples to fit "+dof);
    System.out.println("number of variables "+nvar);
    System.out.println("highest power "+np);

    lsfit LSd = new lsfit(nvar, np, dof); // dimension, power, points
    LSd.debug = false; // to turn off debug printing in lsfit.java
    lastl = false;
    // loop read data
    for(int k=0; k<dof; k++)
    {
      try
      {
        input_line = in.readLine();
        if(debug) System.out.println("input_line="+input_line);
        len = input_line.length();
        index = 0;
	if(debug) System.out.println("index="+index);
        last = input_line.indexOf(' ',index+1);
	if(debug) System.out.println("last="+last);
        intStr = input_line.substring(index,last);
        if(debug) System.out.println("intStr="+intStr);
        D[0] = Double.parseDouble(intStr);
        if(debug) System.out.println("D[0]="+D[0]);
        for(int i=0; i<nvar; i++)
        {
	  index = input_line.indexOf(' ',last); // D[i+1]
	  if(debug) System.out.println("index="+index);
	  last = input_line.indexOf(' ',index+1);
          if(last<0) last = len;
	  if(debug) System.out.println("last="+last);
          intStr = input_line.substring(index+1,last);
	  if(debug) System.out.println("intStr="+intStr);
	  D[i+1] = Double.parseDouble(intStr);
	  if(debug) System.out.println("D["+(i+1)+"]="+D[i+1]);
        }
      } // end try
      catch(Exception e)
      {
        System.out.println("read data line exception"+e);
        return;
      } // end catch

      if(k==dof-1) lastl = true;
      if(nvar==2)
        LSd.fit2_data(D[0], D[1], D[2], lastl); // x, y, U
      else if(nvar==3)  
        LSd.fit3_data(D[0], D[1], D[2], D[3], lastl); // x,y,z,U
      else if(nvar==4)  
	LSd.fit4_data(D[0], D[1], D[2], D[3], D[4], lastl); // x,y,z,t,U
      else if(nvar==5)  
	LSd.fit5_data(D[0], D[1], D[2], D[3], D[4], D[5], lastl); 
                      // x,y,z,t,u,U
      else if(nvar==6)  
        LSd.fit6_data(D[0], D[1], D[2], D[3], D[4], D[5], D[6], lastl);
                      // x,y,z,t,u,v,U
      else if(nvar==7)  
	LSd.fit7_data(D[0], D[1], D[2], D[3], D[4], D[5], D[6], D[7], lastl);
                      // x,y,z,t,u,v,w,U
      // end if
    } // end loop k

    System.out.println("fit "+nvar+" variables to "+np+" power");
    for(int i=0; i<LSd.nnn; i++)
    {
      if(nvar==2)
        System.out.println("C["+i+"]="+LSd.C[i]+"  x^"+LSd.Apwr[i]+
                                                 " y^"+LSd.Bpwr[i]);
      else if(nvar==3)
        System.out.println("C["+i+"]="+LSd.C[i]+"  x^"+LSd.Apwr[i]+
                               " y^"+LSd.Bpwr[i]+" z^"+LSd.Cpwr[i]);
      else if(nvar==4)
        System.out.println("C["+i+"]="+LSd.C[i]+"  x^"+LSd.Apwr[i]+
                               " y^"+LSd.Bpwr[i]+" z^"+LSd.Cpwr[i]+
                               " t^"+LSd.Dpwr[i]);
      else if(nvar==5)
        System.out.println("C["+i+"]="+LSd.C[i]+"  x^"+LSd.Apwr[i]+
                               " y^"+LSd.Bpwr[i]+" z^"+LSd.Cpwr[i]+
                               " t^"+LSd.Dpwr[i]+" u^"+LSd.Epwr[i]);
      else if(nvar==6)
        System.out.println("C["+i+"]="+LSd.C[i]+"  x^"+LSd.Apwr[i]+
                               " y^"+LSd.Bpwr[i]+" z^"+LSd.Cpwr[i]+
                               " t^"+LSd.Dpwr[i]+" u^"+LSd.Epwr[i]+
                               " v^"+LSd.Fpwr[i]);
      else if(nvar==7)
        System.out.println("C["+i+"]="+LSd.C[i]+"  x^"+LSd.Apwr[i]+
                               " y^"+LSd.Bpwr[i]+" z^"+LSd.Cpwr[i]+
                               " t^"+LSd.Dpwr[i]+" u^"+LSd.Epwr[i]+
                               " v^"+LSd.Fpwr[i]+" w^"+LSd.Gpwr[i]);
      // end if
    }
    // check
    maxerr = 0.0;
    //  read data again
    try
    {
      file_in.close();
      file_in = new FileReader(file_name);
      in = new BufferedReader(file_in);
      input_line = in.readLine(); // skip counts
      if(debug) System.out.println("input_line="+input_line);
      len = input_line.length();
    } // end try
    catch(Exception e)
    {
      System.out.println("lsfit_file unable to open file "+file_name);
      return;
    } // end catch
    for(int k=0; k<dof; k++)
    {
      try
      {
        input_line = in.readLine();
        if(debug) System.out.println("input_line="+input_line);
        len = input_line.length();
        index = 0;
	if(debug) System.out.println("index="+index);
        last = input_line.indexOf(' ',index+1);
	if(debug) System.out.println("last="+last);
        intStr = input_line.substring(index,last);
	if(debug) System.out.println("intStr="+intStr);
        D[0] = Double.parseDouble(intStr);
        if(debug) System.out.println("D[0]="+D[0]);
        for(int i=0; i<nvar; i++)
        {
	  index = input_line.indexOf(' ',last); // D[i+1]
	  if(debug) System.out.println("index="+index);
	  last = input_line.indexOf(' ',index+1);
          if(last<0) last = len;
	  if(debug) System.out.println("last="+last);
          intStr = input_line.substring(index+1,last);
	  if(debug) System.out.println("intStr="+intStr);
	  D[i+1] = Double.parseDouble(intStr);
	  if(debug) System.out.println("D["+(i+1)+"]="+D[i+1]);
        }
      } // end try
      catch(Exception e)
      {
        System.out.println("read data line exception"+e);
        return;
      } // end catch

      if(nvar==2)
      {
        Uf = LSd.eval2_poly(D[0], D[1]);
        U = D[2];
      }
      else if(nvar==3)
      {
	Uf = LSd.eval3_poly(D[0], D[1], D[2]);
        U = D[3];
      }
      else if(nvar==4)
      {
	Uf = LSd.eval4_poly(D[0], D[1], D[2], D[3]);
        U = D[4];
      }
      else if(nvar==5)
      {
	Uf = LSd.eval5_poly(D[0], D[1], D[2], D[3], D[4]);
        U = D[5];
      }
      else if(nvar==6)
      {
	Uf = LSd.eval6_poly(D[0], D[1], D[2], D[3], D[4], D[5]);
        U = D[6];
      }
      else if(nvar==7)
      {
	Uf = LSd.eval7_poly(D[0], D[1], D[2], D[3], D[4], D[5], D[6]);
        U = D[7];
      }
      // end if
      err = Math.abs(Uf-U);
      maxerr = Math.max(maxerr,err);
    } // end loop k
    System.out.println("lsfit error on data, maxerr="+maxerr);
  } // end lsfit_file

  public static void main(String args[])
  {
    int npp = 2; // only second power unless specified 
    String file_name = new String("test2d.dat");
    System.out.println("lsfit_file.java  running");
    if(args.length>0)
    {
      file_name = args[0];
    }
    if(args.length>1)
    {
      npp = Integer.parseInt(args[1]);
    }
    new lsfit_file(file_name, npp);
    System.out.println("lsfit_file.java  ends");
  }
} // end class lsfit_file