fei_EqnBuffer.hpp

#ifndef _fei_EqnBuffer_hpp_
#define _fei_EqnBuffer_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.                    */
/*--------------------------------------------------------------------*/

#include "fei_fwd.hpp"

#include <vector>

class EqnBuffer {
 public:
  EqnBuffer();

  EqnBuffer(const EqnBuffer& src);

  virtual ~EqnBuffer();

   EqnBuffer* deepCopy();

   EqnBuffer& operator=(const EqnBuffer& src);

   int getNumEqns() {return(eqnNumbers_.size());};

   std::vector<int>& eqnNumbers() {return(eqnNumbers_);};

   std::vector<fei::CSVec*>& eqns() {return(eqns_);};

   int getNumRHSs() {return(numRHSs_);};

   void setNumRHSs(int n);

   int addRHS(int eqnNumber, int rhsIndex, double value, bool accumulate=true);

   std::vector<std::vector<double>*>* rhsCoefsPtr() {return(&rhsCoefs_);};

   int getEqnIndex(int eqn);

   int isInIndices(int eqn);

   int addEqn(int eqnNumber, const double* coefs, const int* indices,
               int len, bool accumulate, bool create_indices_union=false);

   int addEqns(EqnBuffer& inputEqns, bool accumulate);

   int getCoef(int eqnNumber, int colIndex, double& coef);

   int removeIndex(int eqnNumber, int colIndex);

   int getCoefAndRemoveIndex(int eqnNumber, int colIndex, double& coef);

   void resetCoefs();

   int addIndices(int eqnNumber, const int* indices, int len);

   int newCoefData_, newRHSData_;

 private:
   void deleteMemory();
   int insertNewEqn(int eqn, int insertPoint);

   int internalAddEqn(int index, const double* coefs,
                       const int* indices, int len, bool accumulate);

   std::vector<int> eqnNumbers_; //list of equation-numbers

   std::vector<fei::CSVec*> eqns_;

   std::vector<int> indices_union_; //union of all equation-indices

   int numRHSs_;     //number of right-hand-side coefficients per equation
   std::vector<std::vector<double>*> rhsCoefs_; //list of vector-pointers, each 
                                          //vector is of length numRHSs_
   bool setNumRHSsCalled_;
   bool rhsCoefsAllocated_;

   std::vector<double> dummyCoefs_;
};

std::ostream& operator<<(std::ostream& os, EqnBuffer& eq);

#endif