fei_VectorTraits_LinSysCore.hpp

/*--------------------------------------------------------------------*/
/*    Copyright 2005 Sandia Corporation.                              */
/*    Under the terms of Contract DE-AC04-94AL85000, there is a       */
/*    non-exclusive license for use of this work by or on behalf      */
/*    of the U.S. Government.  Export of this program may require     */
/*    a license from the United States Government.                    */
/*--------------------------------------------------------------------*/

#ifndef _fei_VectorTraits_LinSysCore_hpp_
#define _fei_VectorTraits_LinSysCore_hpp_


#include <fei_VectorTraits.hpp>
#include <fei_LinearSystemCore.hpp>

namespace fei {

  template<>
  struct VectorTraits<LinearSystemCore>  {

    static const char* typeName()
      { return("LinearSystemCore"); }

    static int setValues(LinearSystemCore* vec, int firstLocalOffset,
       double scalar, bool isSolnVector=false)
      {
  if (isSolnVector) {
    //LinearSystemCore doesn't have a 'resetSolnVector()'.
    return(-1);
  }
  int err = vec->resetRHSVector(scalar);
  return(err);
      }

    static int putValuesIn(LinearSystemCore* vec,
         int firstLocalOffset,
         int numValues, const int* indices, const double* values,
                     bool sum_into,
         bool isSolnVector=false,
         int vectorIndex=0)
    {
      int err = 0;
      if (isSolnVector) {
        if (sum_into) {
          return(-97);//LinearSystemCore allows 'put' (overwrite) operations on
          //the soln-vector, but not 'sumInto'.
        }
        else {
          err = vec->putInitialGuess(indices, values, numValues);
        }
      }
      else {
        if (sum_into) {
          err = vec->sumIntoRHSVector(numValues, values, indices);
        }
        else {
          err = vec->putIntoRHSVector(numValues, values, indices);
        }
      }
      return(err);
    }

    static int copyOut(LinearSystemCore* vec,
           int firstLocalOffset,
           int numValues, const int* indices, double* values,
           bool isSolnVector=false,
           int vectorIndex=0)
      {
  int err = 0;
  if (isSolnVector) {
    for(int i=0; i<numValues; ++i) {
      if (vec->getSolnEntry(indices[i], values[i]) != 0) return(-1);
    }
  }
  else {
    err = vec->getFromRHSVector(numValues, values, indices);
  }

  return(err);
      }

    static int update(LinearSystemCore* vec,
          double a,
          const LinearSystemCore* x,
          double b)
    { return(-1); }

  };//struct VectorTraits
}//namespace fei

#endif // _fei_VectorTraits_LinSysCore_hpp_