fei_CSVec.hpp

Go to the documentation of this file.

#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