local_stencil.hpp

Go to the documentation of this file.

#ifndef PARALUTION_LOCAL_STENCIL_HPP_
#define PARALUTION_LOCAL_STENCIL_HPP_

#include "operator.hpp"
#include "local_vector.hpp"
#include "base_stencil.hpp"
#include "stencil_types.hpp"

namespace paralution {

template <typename ValueType>
class LocalVector;
template <typename ValueType>
class GlobalVector;

// Local Stencil
template <typename ValueType>
class LocalStencil : public Operator<ValueType> {

public:

  LocalStencil();
  LocalStencil(unsigned int type);
  virtual ~LocalStencil();

  virtual void info() const;

  virtual int get_ndim(void) const;
  virtual int get_nrow(void) const;
  virtual int get_ncol(void) const;
  virtual int get_nnz(void) const;

  virtual void SetGrid(const int size);

  virtual void Clear();

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

  virtual void MoveToAccelerator(void);
  virtual void MoveToHost(void);

protected:

  virtual bool is_host(void) const {return true;};
  virtual bool is_accel(void) const {return false;};

private:
  
  std::string object_name_ ;

  BaseStencil<ValueType> *stencil_;

  HostStencil<ValueType> *stencil_host_;
  AcceleratorStencil<ValueType> *stencil_accel_;

  
  friend class LocalVector<ValueType>;
  friend class GlobalVector<ValueType>;

};


}

#endif // PARALUTION_LOCAL_STENCIL_HPP_