include/distributed_data.h

Go to the documentation of this file.

#if !defined (DISTRIBUTED_DATA_H)
#define DISTRIBUTED_DATA_H

#include "petscvec.h"
#include "petscmat.h"
#include <string>

namespace ddata
{
class DistributedData{

 protected:
  int num_cores;
  int core_id;  
 
 public:
  DistributedData();
  ~DistributedData();
};

/***************************************************************/
 class Vector:public DistributedData{

 private: 
  int num_total_rows;
  int num_local_rows;
  int local_row_begin;
  int local_row_end;
  int num_ghosts;

 public: 
  Vec data; 
  Vector();
  Vector(int num_total_rows);
  Vector(int num_total_rows, int num_ghosts);
  ~Vector();
  void set_size(int num_total_rows);
  void set_ghost_size(int num_total_rows, int num_ghosts);
  void print(std::string file_name) const;
  void view(void) const;
  int id(void) const {return core_id;}
  int size(void) const {return num_total_rows;}
  int lsize(void) const {return num_local_rows;}
};


/***************************************************************/
class Matrix:public DistributedData{
 private:
  int num_total_rows;
  int num_local_rows;
  int num_total_cols;
  int num_local_cols;

 public: 
  Mat data;
  Matrix(int num_total_rows, int num_total_cols);  
  ~Matrix();
  
};
}
#endif

---