channel.h

Go to the documentation of this file.

#ifndef CHANNEL_INCLUDED
#define CHANNEL_INCLUDED

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

#include "Communicator/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;  
  std::vector<MPI_Status> m_status;  
  unsigned int m_nreq;               
};
#endif /* _CHANNEL_H_ */