#include "sparse.h"

#define ARRAY_SIZE 10

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

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


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

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


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

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


static void Output(SparseArray A,SparseArray B,SparseArray C) {
  printf("A:\n");
  WriteDense(A,stdout);
  printf("\nplus B:\n");
  WriteDense(B,stdout);
  printf("\nequals C:\n");
  WriteDense(C,stdout);
}


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

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