#include "sparse.h"

#define ARRAY_SIZE 10

static void SetupA(SparseArray& A) {
  int row;
  int col;
  int diff;

  for (row=0; row<ARRAY_SIZE; row++) {
    for (col=0; col<ARRAY_SIZE; col++) {
      diff = row-col;
      if (diff >= -1 && diff <= 1) {
	A.Store(row,col,row+(0.01*col));
      }
    }
  }
}


static void SetupB(SparseArray& B) {
  int row;
  int col;
  int diff;

  for (row=0; row<ARRAY_SIZE; row++) {
    for (col=0; col<ARRAY_SIZE; col++) {
      diff = row - (ARRAY_SIZE-(col+1));
      if (diff >= -1 && diff <= 1) {
	B.Store(row,col,row+(0.01*col));
      }
    }
  }
}


static void Add(SparseArray& Sum,SparseArray& A,SparseArray& B) {
  int row,col;
  double aval;
  double bval;
  double sum;

  for (row=0; row<ARRAY_SIZE; row++) {
    for (col=0; col<ARRAY_SIZE; col++) {
      if (A.Access(row,col) != NULL) {
	aval = A.Read(row,col);
      } else {
	aval = 0.0;
      }
      if (B.Access(row,col) != NULL) {
	bval = B.Read(row,col);
      } else {
	bval = 0.0;
      }
      sum = aval + bval;
      if (sum != 0.0) {
	Sum.Store(row,col,sum);
      }
    }
  }
}


static void Output(SparseArray& A,SparseArray& B,SparseArray& C) {
  ofstream outfile("add.out",ios::out);

  outfile << "A:\n";
  A.WriteDense(outfile);
  outfile << "\nplus B:\n";
  B.WriteDense(outfile);
  outfile << "\nequals C:\n";
  C.WriteDense(outfile);
}


void main() {
  SparseArray A(ARRAY_SIZE,ARRAY_SIZE,0.0);
  SparseArray B(ARRAY_SIZE,ARRAY_SIZE,0.0);
  SparseArray C(ARRAY_SIZE,ARRAY_SIZE,0.0);

  SetupA(A);
  SetupB(B);
  Add(C,A,B);
  Output(A,B,C);
}