cart_comm.h

Go to the documentation of this file.

//                                                         //
// Copyright (c) 2003-2014 by The University of Queensland //
// Centre for Geoscience Computing                         //
// http://earth.uq.edu.au/centre-geoscience-computing      //
//                                                         //
// Primary Business: Brisbane, Queensland, Australia       //
// Licensed under the Open Software License version 3.0    //
// http://www.opensource.org/licenses/osl-3.0.php          //
//                                                         //

#ifndef __CARTCOMM_H
#define __CARTCOMM_H

//--- MPI ---
#include <mpi.h>

//--- project includes ---
#include "tml/comm/comm.h"

//--- STL ---
#include <vector>
using std::vector;
class TML_CartComm : public TML_Comm
{
private:
  int m_ndims;
  vector<int> m_dims;
 
 public:
  // constructor
  TML_CartComm(TML_Comm *comm, vector<unsigned int> dims, vector<bool> circular);
  TML_CartComm(TML_Comm *comm, unsigned int ndims, const vector<int> &dims, const vector<bool> &circular);
  TML_CartComm(TML_Comm*,unsigned int,int*,int*);

  vector<int> get_coords(int); 
  vector<int> get_coords() const; 
  vector<int> get_all_dims() const; 
  int get_dim(int); 
  int get_ndim() const {return m_ndims;}; 

  template <typename T,typename P> void shift(T,P&,int,int,int=0);
  template <typename T,typename P> void shift_array(T*,int,P*,int,int,int,int=0);
  template <typename T,typename P> void shift_cont(T,P&,int,int,int=0);

  template <typename T,typename P> void shift_packed(T,P&,int,int,int=0);
  template <typename T,typename P> void shift_array_packed(T*,int,P*,int,int,int,int=0);
  template <typename T,typename P> void shift_cont_packed(T,P&,int,int,int=0);
  // shift_replace ops?
};

#include "tml/comm/cart_comm.hpp"

#endif // __CARTCOMM_H