source: XMLIO_V2/dev/common/src/buffer_client.cpp @ 300

#include "xmlioserver_spl.hpp"
#include "exception.hpp"
#include "buffer_out.hpp"
#include "buffer_client.hpp"
#include <mpi.h>

namespace xmlioserver
{
  int CClientBuffer::bufferSizeByServer=50*1024*1024 ;
//  int CClientBuffer::bufferSizeByServer=1024*10 ;

  CClientBuffer::CClientBuffer(MPI_Comm interComm_,int serverRank_)
  {
    interComm=interComm_ ;
    serverRank=serverRank_ ;
    bufferSize=bufferSizeByServer/2 ;
    buffer[0]=new char[bufferSize] ; // transform it with MPI_ALLOC_MEM later
    buffer[1]=new char[bufferSize] ;
    current=0 ;
    count=0 ;
    pending=false ;
    retBuffer=new CBufferOut(buffer[current],bufferSize) ;
  }
  
  CClientBuffer::~CClientBuffer()
  {
   delete [] buffer[0] ;
   delete [] buffer[1] ;
   delete retBuffer ;
  }
  
  int CClientBuffer::remain(void)
  {
    return bufferSize-count ;
  }
  
  bool CClientBuffer::isBufferFree(int size)
  {
    if (size>bufferSize) ERROR("CClientBuffer::hasSpace(int size)",
                               <<"request size is too big for buffer, increase buffer client size");
 
    if (size<=remain()) return true ;
    else return false ;
  }
    
  
  CBufferOut*  CClientBuffer::getBuffer(int size)
  {
    if (size<=remain())
    {
      retBuffer->realloc(buffer[current]+count,size) ;
      count+=size ;
      return retBuffer ;
    }
    else
    {
       ERROR("CBufferOut*  CClientBuffer::getSpace(int size) ;",
               <<"No ennough space in buffer, that may not happen...");
       return NULL ;
    }
 
  }  
  
  bool CClientBuffer::checkBuffer(void)
  {
    MPI_Status status ;
    int flag ;
    
    if (pending)
    {
      MPI_Test(&request,&flag,&status) ;
      if (flag==true) pending=false ;
    }

    if (!pending)
    {
      if (count>0)
      {
        MPI_Issend(buffer[current],count,MPI_CHAR,serverRank,20,interComm,&request) ;
        pending=true ;
        if (current==1) current=0 ;
        else current=1 ;
        count=0 ;
      }
    }
    return pending ;
  }
  
  bool CClientBuffer::hasPendingRequest(void)
  {
    if (pending) return true ;
    else if (count>0) return true ;
    else return false ;
  }
    
  
  
}