host_matrix_hyb.hpp

Go to the documentation of this file.

#ifndef PARALUTION_HOST_MATRIX_HYB_HPP_
#define PARALUTION_HOST_MATRIX_HYB_HPP_

#include "../base_vector.hpp"
#include "../base_matrix.hpp"
#include "../matrix_formats.hpp"

namespace paralution {

template <typename ValueType>
class HostMatrixHYB : public HostMatrix<ValueType> {

public:

  HostMatrixHYB();
  HostMatrixHYB(const Paralution_Backend_Descriptor local_backend);
  virtual ~HostMatrixHYB();

  inline int get_ell_max_row(void) const { return this->mat_.ELL.max_row; }
  inline int get_ell_nnz(void) const { return this->ell_nnz_; }
  inline int get_coo_nnz(void) const { return this->coo_nnz_; }

  virtual void info(void) const;
  virtual unsigned int get_mat_format(void) const { return  HYB; }

  virtual void Clear(void);
  virtual void AllocateHYB(const int ell_nnz, const int coo_nnz, const int ell_max_row,
                           const int nrow, const int ncol);

  virtual bool ConvertFrom(const BaseMatrix<ValueType> &mat);

  virtual void CopyFrom(const BaseMatrix<ValueType> &mat);
  virtual void CopyTo(BaseMatrix<ValueType> *mat) const;

  virtual void Apply(const BaseVector<ValueType> &in, BaseVector<ValueType> *out) const;
  virtual void ApplyAdd(const BaseVector<ValueType> &in, const ValueType scalar,
                        BaseVector<ValueType> *out) const;

private:

  MatrixHYB<ValueType, int> mat_;
  int ell_nnz_;
  int coo_nnz_;

  friend class BaseVector<ValueType>;
  friend class HostVector<ValueType>;
  friend class HostMatrixCSR<ValueType>;
  friend class HostMatrixCOO<ValueType>;
  friend class HostMatrixELL<ValueType>;
  friend class HostMatrixDENSE<ValueType>;

  friend class GPUAcceleratorMatrixHYB<ValueType>;
  friend class OCLAcceleratorMatrixHYB<ValueType>;
  friend class MICAcceleratorMatrixHYB<ValueType>;

};


}

#endif // PARALUTION_HOST_MATRIX_HYB_HPP_