fei_CSVec.hpp

#ifndef _fei_CSVec_hpp_
#define _fei_CSVec_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_macros.hpp>
#include <vector>
#include <algorithm>
 
namespace fei {

class CSVec {
 public:
  CSVec(unsigned sz=0);
  virtual ~CSVec();
 
  CSVec& operator=(const CSVec& invec);
 
  std::vector<int>& indices() {return indices_;}
  const std::vector<int>& indices() const {return indices_;}
  std::vector<double>& coefs() {return coefs_;}
  const std::vector<double>& coefs() const {return coefs_;}
 
  size_t size() const {return indices_.size();}
 
  void clear() { indices_.clear(); coefs_.clear(); }
 
  bool operator==(const CSVec& rhs) const {
    return indices_==rhs.indices_ && coefs_==rhs.coefs_;
  }
 
  bool operator!=(const CSVec& rhs) const {
    return indices_!=rhs.indices_ || coefs_!=rhs.coefs_;
  }
 
  void subtract(const CSVec& rhs);
 
 private:
  std::vector<int> indices_;
  std::vector<double> coefs_;
};//class CSVec
 
inline
void add_entry(CSVec& vec, int eqn, double coef)
{
  std::vector<int>& v_ind = vec.indices();
  std::vector<double>& v_coef = vec.coefs();
 
  std::vector<int>::iterator
    iter = std::lower_bound(v_ind.begin(), v_ind.end(), eqn);
 
  size_t offset = iter - v_ind.begin();
 
  if (iter == v_ind.end() || *iter != eqn) {
    v_ind.insert(iter, eqn);
    v_coef.insert(v_coef.begin()+offset, coef);
  }
  else {
    v_coef[offset] += coef;
  }
}
 
 
void add_entries(CSVec& vec, int num, const int* eqns, const double* coefs);
 
void put_entry(CSVec& vec, int eqn, double coef);
 
double get_entry(const CSVec& vec, int eqn);
 
void remove_entry(CSVec& vec, int eqn);
 
void set_values(CSVec& vec, double scalar);

void add_CSVec_CSVec(const CSVec& u, CSVec& v);
 
}//namespace fei
 
#endif