channel.h

Go to the documentation of this file.

#ifndef _CHANNEL_H_
#define _CHANNEL_H_

#include <mpi.h>
#include <cassert>
#include <vector>
#include <valarray>

#include "communicator.h"

static const int max_dimension = 8;

// forward declaration
class ChannelSet;


class Channel {
 public:
  typedef char                          element_type;    
  typedef std::valarray<element_type>   container_type;

  Channel();                
  Channel(const int count); 
  virtual ~Channel();       

  int start();              
  int wait();               

  inline element_type& operator[](unsigned int idx) { return m_buf[idx]; }
  inline element_type operator[](unsigned int idx) const { return m_buf[idx]; }
  inline const element_type *ptr() { return &m_buf[0]; }

 private:
  container_type m_buf;   

  MPI_Request m_request;  
  MPI_Status  m_status;   

  friend class Communicator_impl;
  friend class ChannelSet;
};


class ChannelSet {
 public:
  ChannelSet(int nchannel = 8); 

  int append(Channel *const p); 

  int start();                  
  int wait();                   

 private:
  std::vector<MPI_Request> m_array;  
  unsigned int             m_nreq;   
};
#endif /* _CHANNEL_H_ */