sparvec.h

Go to the documentation of this file.

#ifndef ABA_SPARVEC_H
#define ABA_SPARVEC_H
#include <stdlib.h>

#include "abacus/array.h"
#include "abacus/buffer.h"

#ifdef ABACUS_PARALLEL
class ABA_MESSAGE;
#endif

  class  ABA_SPARVEC :  public ABA_ABACUSROOT  {
    public:

      ABA_SPARVEC(ABA_GLOBAL *glob, 
                  int size, 
                  double reallocFac = 10.0);

      ABA_SPARVEC(ABA_GLOBAL *glob, 
                  int size, 
                  const ABA_ARRAY<int> &s, 
                  const ABA_ARRAY<double> &c, 
                  double reallocFac = 10.0);

      ABA_SPARVEC(ABA_GLOBAL *glob, 
                  int size, 
                  int *s, 
                  double *c, 
                  double reallocFac = 10.0);

      ABA_SPARVEC(const ABA_SPARVEC& rhs);

      ~ABA_SPARVEC();
#ifdef ABACUS_PARALLEL

      ABA_SPARVEC(const ABA_GLOBAL *glob, ABA_MESSAGE &msg);

      void pack(ABA_MESSAGE &msg) const;
#endif

      const ABA_SPARVEC& operator=(const ABA_SPARVEC& rhs);

      friend ostream& operator<<(ostream& out, const ABA_SPARVEC& rhs);

      int support(int i) const; 

      double coeff(int i) const;

      double origCoeff(int i) const;

      void insert(int s, double c);

      void leftShift(ABA_BUFFER<int> &del);

      void copy (const ABA_SPARVEC &vec);

      void clear();

      void rename(ABA_ARRAY<int> &newName);

      int size() const;

      int nnz() const;

      double norm();

      void realloc();

      void realloc(int newSize);
    protected:

      void rangeCheck(int i) const;

      ABA_GLOBAL *glob_;

      int size_;

      int nnz_;

      double reallocFac_;

      int *support_;

      double *coeff_;
  };


inline int ABA_SPARVEC::support(int i) const
  {
#ifdef ABACUSSAFE
    rangeCheck(i);
#endif    
    return support_[i];
  }

inline double ABA_SPARVEC::coeff(int i) const
  {
#ifdef ABACUSSAFE
    rangeCheck(i);
#endif    
    return coeff_[i];
  }

inline void ABA_SPARVEC::insert(int s, double c)
  {

    if (nnz_ == size_) realloc();

    support_[nnz_] = s;
    coeff_[nnz_]   = c;
    ++nnz_;

  }

inline void ABA_SPARVEC::clear()
  {
    nnz_ = 0;
  }

inline int ABA_SPARVEC::size() const
  {
    return size_;
  }

inline int ABA_SPARVEC::nnz() const
  {
    return nnz_;
  }


#endif  // ABA_SPARVEC_H

---