// test_sparse.java    

import java.io.*;

public class test_sparse
{
  sparse A;    // the sparse simultaneous equations
               // norman A matrix with Y as last column
  sparse Ac;   // copy for checking
  double val;
  double X[];  // solution of A*X=Y simultaneous equations
  double Y[];  // extracted Y
  double Yc[]; // for checking
  double AA[][]; // for testing

  public test_sparse(int nrow) // constructor
  {
    System.out.println("test_sparse.java nrow= "+nrow);
    A = new sparse(nrow);
    Ac = new sparse(nrow);
    X = new double[nrow];
    Y = new double[nrow];
    Yc = new double[nrow];

    System.out.println("test put(0,4,3.14) and get in A");
    A.put(0,4,3.14);
    val = A.get(0,4);
    System.out.println("test put in row, 3.14 should be "+val);
    System.out.println("test put(5,0,6.28) and get in A");
    A.put(5,0,6.28);
    val = A.get(5,0);
    System.out.println("test put in row, 6.28 should be "+val);
    A.write_all();
    A = new sparse(nrow);

    for(int i=0; i<nrow; i++)
    {
      for(int j=i-1; j<=i+1; j++)
      {
	  val = 1.0+(double)(i*i+(j+1)*(j+1));
        if(i>=0 && j>=0 && i<nrow && j<nrow)
	{
          A.add(i, j, val);
          Ac.add(i, j, val);
        }
        A.add(i, nrow, 8.0+4.0*i);
        Ac.add(i, nrow, 8.0+4.0*i);
      }
    }
    System.out.println("initial set of equations in A");
    A.write_all();

    Y  = A.getRHS();      // for testing
    for(int i=0; i<nrow; i++)
    {
      System.out.println("i="+i+", Y[i]="+Y[i]);
    }


    System.out.println("calling sparse.simeq");
    A.simeq(X); // modifies, destroys, A
    for(int i=0; i<nrow; i++)
      System.out.println("X["+i+"]="+X[i]);
    System.out.println("A after simeq");
    A.write_all();

    System.out.println("Ac after simeq, should be initial");
    Ac.write_all();

    System.out.println(" ");
    System.out.println("test multiply and simeq");
    Yc = Ac.multiply(X);

    for(int i=0; i<nrow; i++)
    {
      if(Math.abs(Y[i]-Yc[i])> 0.00001)
      {
	System.out.println("either multiply or simeq failed");
        System.out.println("i="+i+", Y[i]="+Y[i]+" yet found "+Yc[i]);
      }
    }

    System.out.println(" ");
    System.out.println("simple test of setRHS, getRHS");
    for(int i=0; i<nrow; i++) Y[i] = (double)(i+1);
    A.setRHS(Y);
    A.write_all();
    Yc = A.getRHS();
    for(int i=0; i<nrow; i++)
    {
      if(Y[i]!=Yc[i])
      {
	System.out.println("either setRHS or getRHS failed");
        System.out.println("i="+i+", Y[i]="+Y[i]+" yet found "+X[i]);
      }
    }
    System.out.println("test_sparse finished");
  }

  public static void main (String[] args)
  {
    new test_sparse(6);
  }

} // end test_sparse.java