evsl.c

Go to the documentation of this file.

#include <stdlib.h>
#include "def.h"
#include "struct.h"
#include "internal_proto.h"

evslData evsldata;

evslStat evslstat;

int EVSLStart() {
  evsldata.n = 0;
  evsldata.ifGenEv = 0;
  evsldata.Amv = NULL;
  evsldata.Bmv = NULL;
  evsldata.Bsol = NULL;
  evsldata.LTsol = NULL;
  evsldata.ds = NULL;

  StatsReset();

  /* do a dummy timer call to improve accuracy of timing */
  evsl_timer();
  
  return 0;
}

int EVSLFinish() {
  if (evsldata.Amv) {
    free(evsldata.Amv);
  }
  if (evsldata.Bmv) {
    free(evsldata.Bmv);
  }
  if (evsldata.Bsol) {
    free(evsldata.Bsol);
  }
  if (evsldata.LTsol) {
    free(evsldata.LTsol);
  }
  return 0;
}

int SetAMatrix(csrMat *A) {
  evsldata.n = A->ncols;
  if (!evsldata.Amv) {
    Calloc(evsldata.Amv, 1, EVSLMatvec);
  }
  evsldata.Amv->func = matvec_csr;
  evsldata.Amv->data = (void *) A;
   
  return 0;
}

int SetBMatrix(csrMat *B) {
  evsldata.n = B->ncols;
  if (!evsldata.Bmv) {
    Calloc(evsldata.Bmv, 1, EVSLMatvec);
  }
  evsldata.Bmv->func = matvec_csr;
  evsldata.Bmv->data = (void *) B;

  return 0;
}

int SetAMatvec(int n, MVFunc func, void *data) {
  evsldata.n = n;
  if (!evsldata.Amv) {
    Calloc(evsldata.Amv, 1, EVSLMatvec);
  }
  evsldata.Amv->func = func;
  evsldata.Amv->data = data;

  return 0;
}

int SetBMatvec(int n, MVFunc func, void *data) {
  evsldata.n = n;
  if (!evsldata.Bmv) {
    Calloc(evsldata.Bmv, 1, EVSLMatvec);
  }
  evsldata.Bmv->func = func;
  evsldata.Bmv->data = data;

  return 0;
}


int SetBSol(SolFuncR func, void *data) {
  if (!evsldata.Bsol) {
    Calloc(evsldata.Bsol, 1, EVSLBSol);
  }

  evsldata.Bsol->func = func;
  evsldata.Bsol->data = data;

  return 0;
}


int SetStdEig() {
  evsldata.ifGenEv = 0;

  return 0;
}

int SetGenEig() {
  evsldata.ifGenEv = 1;

  return 0;
}

int SetASigmaBSol(ratparams *rat, SolFuncC *func, SolFuncC allf, void **data) {
  int i,num;
  num = rat->num;
  Calloc(rat->ASIGBsol, num, EVSLASIGMABSol);

  for (i=0; i<num; i++) {
    rat->ASIGBsol[i].func = func ? func[i] : allf;
    rat->ASIGBsol[i].data = data[i];
  }

  return 0;
}


int SetLTSol(SolFuncR func, void *data) {
  if (!evsldata.LTsol) {
    Calloc(evsldata.LTsol, 1, EVSLLTSol);
  }
  evsldata.LTsol->func = func;
  evsldata.LTsol->data = data;
  return 0;
}

void SetDiagScal(double *ds) {
  evsldata.ds = ds;
}