Tsqr_MpiDatatype.hpp

#ifndef __TSQR_MpiDatatype_hpp
#define __TSQR_MpiDatatype_hpp

#include <mpi.h>
#include <stdexcept>


namespace TSQR {
  namespace MPI {

    MPI_Datatype 
    cloneRawDatatype (MPI_Datatype in, const bool needsFree);

    template< class Datum >
    class MpiDatatype {
    public:
      MpiDatatype ();

      MpiDatatype (const MpiDatatype& rhs) 
      {
  clone (this->type_, this->needsFree_, rhs);
      }

      MpiDatatype& operator= (const MpiDatatype& rhs)
      {
  if (this != &rhs)
    {
      if (needsFree_)
        {
    // Return value doesn't matter...
    (void) MPI_Type_free (&type_);
    needsFree_ = false;
        }
      clone (this->type_, this->needsFree_, rhs);
    }
  return *this;
      }

      ~MpiDatatype () 
      { 
  if (needsFree_)
    {
      // Return value doesn't matter...
      (void) MPI_Type_free (&type_);
      needsFree_ = false;
    }
      }

      MPI_Datatype get() const { return type_; }

    private:
      static void 
      clone (MPI_Datatype& newType, 
       bool& needsFree,
       const MpiDatatype& rhs)
      {
  newType = cloneRawDatatype (rhs.get(), rhs.needsFree_);
  needsFree = rhs.needsFree_;
      }

      MPI_Datatype type_;

      bool needsFree_;
    };

  } // namespace MPI
} // namespace TSQR

#endif // __TSQR_MpiDatatype_hpp