source: XIOS/dev/branch_yushan_merged/src/buffer_client.cpp @ 1176

Last change on this file since 1176 was 1176, checked in by yushan, 4 years ago

save modif

  • Property copyright set to
    Software name : XIOS (Xml I/O Server)
    http://forge.ipsl.jussieu.fr/ioserver
    Creation date : January 2009
    Licence : CeCCIL version2
    see license file in root directory : Licence_CeCILL_V2-en.txt
    or http://www.cecill.info/licences/Licence_CeCILL_V2-en.html
    Holder : CEA/LSCE (Laboratoire des Sciences du CLimat et de l'Environnement)
    CNRS/IPSL (Institut Pierre Simon Laplace)
    Project Manager : Yann Meurdesoif
    yann.meurdesoif@cea.fr
  • Property svn:eol-style set to native
File size: 2.6 KB
Line 
1#include "xios_spl.hpp"
2#include "exception.hpp"
3#include "log.hpp"
4#include "buffer_out.hpp"
5#include "buffer_client.hpp"
6#include "cxios.hpp"
7#include "mpi.hpp"
8#include "tracer.hpp"
9
10namespace xios
11{
12  size_t CClientBuffer::maxRequestSize = 0;
13
14  CClientBuffer::CClientBuffer(MPI_Comm interComm, int serverRank, StdSize bufferSize, StdSize maxBufferedEvents)
15    : interComm(interComm)
16    , serverRank(serverRank)
17    , bufferSize(bufferSize)
18    , current(0)
19    , count(0)
20    , bufferedEvents(0)
21    , maxBufferedEvents(maxBufferedEvents)
22    , pending(false)
23  {
24    buffer[0] = new char[bufferSize]; // transform it with MPI_ALLOC_MEM later
25    buffer[1] = new char[bufferSize];
26    retBuffer = new CBufferOut(buffer[current], bufferSize);
27    #pragma omp critical (_output)
28    info(10) << "CClientBuffer: allocated 2 x " << bufferSize << " bytes for server " << serverRank << " with a maximum of " << maxBufferedEvents << " buffered events" << endl;
29  }
30
31  CClientBuffer::~CClientBuffer()
32  {
33   delete [] buffer[0];
34   delete [] buffer[1];
35   delete retBuffer;
36  }
37
38  int CClientBuffer::remain(void)
39  {
40    return bufferSize - count;
41  }
42
43  bool CClientBuffer::isBufferFree(int size)
44  {
45    if (size > maxRequestSize) maxRequestSize = size;
46
47    if (size > bufferSize)
48      ERROR("bool CClientBuffer::isBufferFree(int size)",
49            << "The requested size (" << size << " bytes) is too big to fit the buffer (" << bufferSize << " bytes), please increase the client buffer size." << endl);
50
51    return (size <= remain() && bufferedEvents < maxBufferedEvents);
52  }
53
54
55  CBufferOut* CClientBuffer::getBuffer(int size)
56  {
57    if (size <= remain())
58    {
59      retBuffer->realloc(buffer[current] + count, size);
60      count += size;
61      bufferedEvents++;
62      return retBuffer;
63    }
64    else
65    {
66      ERROR("CBufferOut* CClientBuffer::getBuffer(int size)",
67            << "Not enough space in buffer, this should not have happened...");
68      return NULL;
69    }
70  }
71
72  bool CClientBuffer::checkBuffer(void)
73  {
74    MPI_Status status;
75    int flag;
76
77    if (pending)
78    {
79      traceOff();
80      MPI_Test(&request, &flag, &status);
81      traceOn();
82      if (flag == true) pending = false;
83    }
84
85    if (!pending)
86    {
87      if (count > 0)
88      {
89        MPI_Issend(buffer[current], count, MPI_CHAR, serverRank, 20, interComm, &request);
90        #pragma omp critical (_output)
91        pending = true;
92        if (current == 1) current = 0;
93        else current = 1;
94        count = 0;
95        bufferedEvents = 0;
96      }
97    }
98
99    return pending;
100  }
101
102  bool CClientBuffer::hasPendingRequest(void)
103  {
104    return (pending || count > 0);
105  }
106}
Note: See TracBrowser for help on using the repository browser.