// test_poly.java    polyprt  polycopy polyval  polyderiv  polyintegrate
//                   polyadd  polysub  polymul    polydiv
//                   polyshrink  polymake  polyroots  polyfit
//                   polyfindaroot  polyremovearoot  
//                   polynomial stored  p[0] + p[1]*x + ...p[pwr]x^pwr


public class test_poly
{
  public test_poly()
  {
    poly P = new poly(); // makes poly functions available P.polyprt etc

    int pwr = 3;
    int n = 10;
    double p[] = {1.2, 2.4, 3.6, 4.8}; // changed for later test
    double p1[] = {-1.0, 1.0}; // for product and root
    double p2[] = {-2.0, 1.0}; // p[0], p[1]*X, p[2]*X^2
    double p3[] = {-3.0, 1.0};
    double p4[] = {-4.0, 1.0};
    double p1r[] = {0.0, 0.0};
    double p12[] = {0.0, 0.0, 0.0};
    double p12r[] = {0.0, 0.0, 0.0};
    double p123[] = {0.0, 0.0, 0.0, 0.0};
    double p123r[] = {0.0, 0.0, 0.0, 0.0};
    double p1234[] = {0.0, 0.0, 0.0, 0.0, 0.0};

    double pck[] = new double[10];
    double pqck[] = new double[10];
    double pre[] = new double[10];
    double pderiv[] = new double[3];
    double pintgr[] = new double[5];
    double x[] = new double[n];
    double y[] = new double[n];
    double yc[] = new double[n];
    double err;
    double pa[] = {3.0, 5.0, 8.0, 12.0};
    int na = 3;
    double pb[] = {1.0, 2.0, 3.0, 4.0};
    int nb = 3;
    double xr[] = {1.0, 2.0, 3.0, 4.0};
    int nrt = 4; 
    double result[] = new double[10];
    int nr[] = new int[1]; // returned pwr
    int nq[] = new int[1];
    int npre[] = new int[1];

    System.out.println("test_poly.java running");
    System.out.println("test polyval");
    for(int i=0; i<n; i++)
    {
      x[i] = (double)(i+1)*0.25;
    }
    for(int i=0; i<n; i++)
    {
      y[i] = P.polyval(pwr, p, x[i]);
      yc[i] = y[i];
      System.out.println("i="+i+", x[i]="+x[i]+", y[i]="+y[i]);
    }

    System.out.println("computed values");
    for(int j=0; j<=pwr; j++) p[j] = -1.0;
    P.polyfit(pwr, n, x, y, p);
    for(int j=0; j<=pwr; j++) System.out.println("j="+j+", p[j]="+p[j]);

    for(int i=0; i<n; i++)
    {
      y[i] = P.polyval(pwr, p, x[i]);
      err = Math.abs(y[i]-yc[i]);
      System.out.println("i="+i+", x[i]="+x[i]+", y[i]="+y[i]+", err="+err);
    }

    System.out.println(" ");
    System.out.println("test polycopy");
    P.polyprt(nb, pb);
    P.polycopy(nb, pb, result);
    System.out.println("copied = ");
    P.polyprt(nb, result);

    System.out.println(" ");
    System.out.println("test polyadd");
    P.polyprt(na, pa);
    System.out.println("+");
    P.polyprt(nb, pb);
    System.out.println("sum =");
    P.polyadd(na, pa, nb, pb, nr, result);
    P.polyprt(nr[0], result);

    System.out.println(" ");
    System.out.println("test polysub");
    P.polyprt(na, pa);
    System.out.println("-");
    P.polyprt(nb, pb);
    System.out.println("difference =");
    P.polysub(na, pa, nb, pb, nr, result);
    P.polyprt(nr[0], result);

    System.out.println(" ");
    System.out.println("test polymul");
    P.polyprt(na, pa);
    System.out.println("*");
    P.polyprt(nb, pb);
    P.polymul(na, pa, nb, pb, nr, result);
    System.out.println("product =");
    P.polyprt(nr[0], result);

    System.out.println(" ");
    System.out.println("test polydiv");
    P.polyprt(nr[0], result);
    System.out.println("/");
    P.polyprt(nb, pb);
    P.polydiv(nr[0], result, nb, pb, nq, pqck, nr, pck);
    System.out.println("quotient =");
    P.polyprt(nq[0], pqck);
    System.out.println("remainder =");
    P.polyprt(nr[0], pck);

    System.out.println(" ");
    System.out.println("test polyderiv");
    p[0] = 4.0;
    p[1] = 3.0;
    p[2] = 2.0;
    p[3] = 1.0;

    P.polyprt(3, p);
    P.polyderiv(3, p, pderiv);
    System.out.println("derivative");
    P.polyprt(2, pderiv);
    System.out.println("integrate");
    P.polyintegrate(2, pderiv, pck);
    P.polyprt(3, pck);

    System.out.println(" ");
    System.out.println("test polyintegrate");
    P.polyprt(3, p);
    P.polyintegrate(3, p, pintgr);
    System.out.println("integral");
    P.polyprt(4, pintgr);
    System.out.println("derivative");
    P.polyderiv(4, pintgr, pck);
    P.polyprt(3, pck);

    System.out.println(" ");
    System.out.println("test polymake from roots xr");
    for(int i=0; i<1; i++) // nrt=1
       System.out.println("xr["+i+"]="+xr[i]);
    P.polymake(1, xr, nr, result);
    System.out.println("polynomial nr="+nr[0]+" p =");
    P.polyprt(nr[0], result);
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[0]));

    System.out.println("test polymake from roots xr");
    for(int i=0; i<2; i++)
       System.out.println("xr["+i+"]="+xr[i]);
    P.polymake(2, xr, nr, result);
    System.out.println("polynomial nr="+nr[0]+" p=");
    P.polyprt(nr[0], result);
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[0]));
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[1]));

    System.out.println("test polymake from roots xr");
    for(int i=0; i<3; i++)
       System.out.println("xr["+i+"]="+xr[i]);
    P.polymake(3, xr, nr, result);
    System.out.println("polynomial nr="+nr[0]+" p=");
    P.polyprt(nr[0], result);
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[0]));
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[1]));
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[2]));

    System.out.println("test polymake from roots xr");
    for(int i=0; i<nrt; i++)
       System.out.println("xr["+i+"]="+xr[i]);
    P.polymake(nrt, xr, nr, result);
    System.out.println("polynomial =");
    P.polyprt(nr[0], result);
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[0]));
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[1]));
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[2]));
    System.out.println("check 0 "+P.polyval(nr[0], result, xr[3]));
    P.polycopy(nr[0], result, pck); // made from roots 1.0, 2.0, 3.0, 4.0

    System.out.println(" ");
    System.out.println("test polyremovearoot 4.0 from pre ");
    P.polycopy(nr[0], result, pre); // made from roots 1.0, 2.0, 3.0, 4.0
    npre[0] = nr[0];
    x[0] = 4.0;
    P.polyprt(npre[0], pre);
    P.polyremovearoot(x[0], npre, pre);
    System.out.println("with root 4.0 removed = ");
    P.polyprt(npre[0], pre); // npre reduced

    System.out.println(" ");
    System.out.println("test polyfindaroot (x-1)*(x-2)*(x-3)*(x-4)");
    P.polyprt(nr[0], pck);
    x[0] = P.polyfindaroot (nr[0], pck);
    System.out.println("root x[0] = "+x[0]);
    System.out.println("remove that root ");
    P.polyremovearoot(x[0], nr, pck);

    System.out.println("test next polyfindaroot ");
    P.polyprt(nr[0], pck);
    x[1] = P.polyfindaroot (nr[0], pck);
    System.out.println("root x[1] = "+x[1]);
    System.out.println("remove that root ");
    P.polyremovearoot(x[1], nr, pck);

    System.out.println("test next polyfindaroot ");
    P.polyprt(nr[0], pck);
    x[2] = P.polyfindaroot (nr[0], pck);
    System.out.println("root x[2] = "+x[2]);
    System.out.println("remove that root ");
    P.polyremovearoot(x[2], nr, pck);

    System.out.println("test next polyfindaroot ");
    P.polyprt(nr[0], pck);
    x[3] = P.polyfindaroot (nr[0], pck);
    System.out.println("root x[3] = "+x[3]);

    System.out.println(" ");
    System.out.println("test (x-1)*(x-2)*(x-3)*(x-4)polymul then polyroot");
    P.polyprt(1, p1);
    System.out.println("*");
    P.polyprt(1, p2);
    P.polymul(1, p1, 1, p2, nr, p12); // nr[0] and p12 returned
    System.out.println("product p1*p2=");
    P.polyprt(nr[0], p12);
    P.polymul(2, p12, 1, p3, nr, p123);
    System.out.println("product p1*p2*p3=");
    P.polyprt(nr[0], p123);
    P.polymul(3, p123, 1, p4, nr, p1234); // nr[0] and p12 returned
    System.out.println("product p1*p2*p3*p4=");
    P.polyprt(nr[0], p1234);

    System.out.println(" ");
    System.out.println("test polyfindaroot p1234");
    P.polyprt(nr[0], p1234);

    x[0] = P.polyfindaroot (4, p1234);
    System.out.println("x[0] = "+x[0]);
    System.out.println("remove that root ");
    P.polyremovearoot(x[0], nr, p1234); // p1234 reurned reduced
    P.polyprt(nr[0], p1234);

    x[1] = P.polyfindaroot (3, p1234);
    System.out.println("x[1] = "+x[1]);
    System.out.println("remove that root ");
    P.polyremovearoot(x[1], nr, p1234); // p1234 reurned reduced
    P.polyprt(nr[0], p1234);

    x[2] = P.polyfindaroot (2, p1234);
    System.out.println("x[2] = "+x[2]);
    System.out.println("remove that root ");
    P.polyremovearoot(x[2], nr, p1234); // p1234 reurned reduced
    P.polyprt(nr[0], p1234);

    x[3] = P.polyfindaroot (1, p1234);
    System.out.println("x[3] = "+x[3]);
    System.out.println("remove that root ");
    P.polyremovearoot(x[3], nr, p1234); // p1234 reurned reduced
    P.polyprt(nr[0], p1234);

    System.out.println(" ");
    System.out.println("test polyfindaroot (x-50)*(x-0.01)");
    pck[0] = 0.5;
    pck[1] = -50.01;
    pck[2] = 1.0;
    nr[0]=2;
    P.polyprt(nr[0], pck);
    x[0] = P.polyfindaroot (nr[0], pck);
    System.out.println("root x[0] = "+x[0]);
    System.out.println("remove that root ");
    P.polyremovearoot(x[0], nr, pck);

    System.out.println("test next polyfindaroot ");
    P.polyprt(nr[0], pck);
    x[1] = P.polyfindaroot (nr[0], pck);
    System.out.println("root x[1] = "+x[1]);
    System.out.println("remove that root ");
    P.polyremovearoot(x[1], nr, pck);

    System.out.println("end test_poly.java");
  }

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