matpackII.h

Go to the documentation of this file.

/* Copyright (C) 2001-2012
   Stefan Buehler <sbuehler@ltu.se>
   Mattias Ekstroem <ekstrom@rss.chalmers.se>

   This program is free software; you can redistribute it and/or modify it
   under the terms of the GNU General Public License as published by the
   Free Software Foundation; either version 2, or (at your option) any
   later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
   USA. */

#ifndef matpackII_h
#define matpackII_h

#include "matpackI.h"


class Sparse {
public:
  // Constructors:
  Sparse();
  Sparse(Index r, Index c);
  
  // Insert functions
  void insert_row(Index r, Vector v);

  // Make identity function:
  void make_I( Index r, Index c);

  // Resize function:
  void resize(Index r, Index c);

  // Member functions:
  Index nrows() const;
  Index ncols() const;
  Index nnz()   const;

  const std::vector<Numeric> & data () const {return mdata;}
  const std::vector<Index> & rowind () const {return mrowind;}
  const std::vector<Index> & colptr () const {return mcolptr;}

  // Index Operators:
  Numeric& rw(Index r, Index c);
  Numeric  ro(Index r, Index c) const;
  Numeric  operator() (Index r, Index c) const;

  // Friends:
  friend std::ostream& operator<<(std::ostream& os, const Sparse& v);
  friend void abs (Sparse& A, const Sparse& B );
  friend void mult (VectorView y, const Sparse& M, ConstVectorView x );
  friend void mult (MatrixView A, const Sparse& B, ConstMatrixView C );
  friend void mult (Sparse& A, const Sparse& B, const Sparse& C );
  friend void add (Sparse& A, const Sparse& B, const Sparse& C );
  friend void sub (Sparse& A, const Sparse& B, const Sparse& C );
  friend void transpose (Sparse& A, const Sparse& B );

private:
  std::vector<Numeric> mdata;
  std::vector<Index> mrowind;
  std::vector<Index> mcolptr;
  Index mrr;
  Index mcr;
};


// Functions for general matrix operations
void abs(       Sparse& A,
          const Sparse& B );

void mult( VectorView y,
           const Sparse& M,
           ConstVectorView x );

void mult( MatrixView A,
           const Sparse& B,
           ConstMatrixView C );

void mult( Sparse& A,
           const Sparse& B,
           const Sparse& C );

void add( Sparse& A,
          const Sparse& B,
          const Sparse& C );

void sub( Sparse& A,
          const Sparse& B,
          const Sparse& C );

void transpose( Sparse& A,
                const Sparse& B );

#endif