CRMatrixTranspose.h

Go to the documentation of this file.

// This file os part of FVM
// Copyright (c) 2012 FVM Authors
// See LICENSE file for terms.

#ifndef _CRMATRIXTRANSPOSE_H_
#define _CRMATRIXTRANSPOSE_H_


#include "Matrix.h"
#include "CRConnectivity.h"
#include "Array.h"
#include "StorageSite.h"


template<class T_Coeff, class X, class B>
class CRMatrixTranspose : public Matrix
{
public:
  typedef T_Coeff Coeff;
  typedef Array<Coeff> CoeffArray;
  typedef Array<X> XArray;
  typedef Array<B> BArray;
  

  CRMatrixTranspose(const CRConnectivity& conn) :
    Matrix(),
    _conn(conn),
    _row(_conn.getRow()),
    _col(_conn.getCol()),
    _coeffPtr(new CoeffArray(_col.getLength())),
    _coeff(*_coeffPtr)
  {
    logCtor();
  }

  // meant to be used to create CRMatrixTranspose<T,Vec3,Vec3> from
  // CRMatrix<T,T,T>
  
  template<class X2, class B2>
  CRMatrixTranspose(const CRMatrixTranspose<Coeff,X2,B2>& m) :
    Matrix(),
    _conn(m.getConnectivity()),
    _row(_conn.getRow()),
    _col(_conn.getCol()),
    _coeffPtr(m.getCoeffPtr()),
    _coeff(*_coeffPtr)
  {
    logCtor();
  }

  
  DEFINE_TYPENAME("CRMatrixTranspose<"
                  +NumTypeTraits<Coeff>::getTypeName()+","
                  +NumTypeTraits<X>::getTypeName()+","
                  +NumTypeTraits<B>::getTypeName()
             +">");

  virtual void initAssembly()
  {
    _coeff.zero();
  }

  virtual void multiply(IContainer& yB, const IContainer& xB) const
  {
    yB.zero();
    this->multiplyAndAdd(yB,xB);
  }

  virtual void multiplyAndAdd(IContainer& yB, const IContainer& xB) const
  {
    BArray& y = dynamic_cast<BArray&>(yB);
    const XArray& x = dynamic_cast<const XArray&>(xB);
    
    const int nRows = _conn.getRowSite().getCount();
    for(int nr=0; nr<nRows; nr++)
    {
        for (int nb = _row[nr]; nb<_row[nr+1]; nb++)
        {
            const int j = _col[nb];
            y[nr] += _coeff[nb]*x[j];
        }
    }
  }

  
  const CRConnectivity& getConnectivity() const {return _conn;}

  Array<Coeff>& getCoeff() {return _coeff;}
  const Array<Coeff>& getCoeff() const {return _coeff;}
  shared_ptr<Array<Coeff> > getCoeffPtr() const {return _coeffPtr;}
  
private:
  const CRConnectivity& _conn;
  const Array<int>& _row;
  const Array<int>& _col;
  shared_ptr<Array<Coeff> > _coeffPtr;
  Array<Coeff>& _coeff;
};

#endif