src/Pstream/gamma/IPread.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 1991-2009 OpenCFD Ltd.
   6      \\/     M anipulation  |
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of OpenFOAM.
  10
  11     OpenFOAM is free software; you can redistribute it and/or modify it
  12     under the terms of the GNU General Public License as published by the
  13     Free Software Foundation; either version 2 of the License, or (at your
  14     option) any later version.
  15
  16     OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
  17     ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  18     FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  19     for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with OpenFOAM; if not, write to the Free Software Foundation,
  23     Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  24
  25 Description
  26     Read token and binary block from IPstream
  27
  28 \*---------------------------------------------------------------------------*/
  29
  30 #include "IPstream.H"
  31 #include "long.H"
  32 #include "PstreamGlobals.H"
  33
  34 extern "C"
  35 {
  36 #   include <linux/gamma/libgamma.h>
  37 }
  38
  39 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  40
  41 namespace Foam
  42 {
  43
  44 // * * * * * * * * * * * * * * * * Constructor * * * * * * * * * * * * * * * //
  45
  46 IPstream::IPstream
  47 (
  48     const commsTypes commsType,
  49     const int fromProcNo,
  50     const label bufSize,
  51     streamFormat format,
  52     versionNumber version
  53 )
  54 :
  55     Pstream(commsType, bufSize),
  56     Istream(format, version),
  57     fromProcNo_(fromProcNo),
  58     messageSize_(0)
  59 {
  60     // Blocking read.
  61
  62     setOpened();
  63     setGood();
  64
  65     if (Pstream::debug)
  66     {
  67         Pout<< "IPstream::IPstream : Starting receive from " << fromProcNo_
  68             << " recvIndex:" << PstreamGlobals::recvIndex[fromProcNo_]
  69             << Foam::endl;
  70     }
  71
  72     PstreamGlobals::gammaWait(fromProcNo_);
  73
  74     label ready = PstreamGlobals::consumeIndex[fromProcNo_];
  75     messageSize_ = PstreamGlobals::recvBufLen[ready][fromProcNo_];
  76
  77     if (!bufSize)
  78     {
  79         if (Pstream::debug)
  80         {
  81             Pout<< "IPstream::IPstream : sizing buffer to " << messageSize_
  82                 << endl;
  83         }
  84
  85         buf_.setSize(messageSize_);
  86     }
  87
  88     PstreamGlobals::copyReceive(fromProcNo_, buf_.begin(), buf_.size());
  89
  90     if (Pstream::debug)
  91     {
  92         Pout<< "IPstream::IPstream : Received " << messageSize_
  93             << " from " << fromProcNo_
  94             << Foam::endl;
  95     }
  96 }
  97
  98
  99 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 100
 101 label IPstream::read
 102 (
 103     const commsTypes commsType,
 104     const int fromProcNo,
 105     char* buf,
 106     const std::streamsize bufSize
 107 )
 108 {
 109     // Blocking read.
 110     label messageSize;
 111
 112     if (Pstream::debug)
 113     {
 114         Pout<< "IPstream::read : Starting receive from " << fromProcNo
 115             << " recvIndex:" << PstreamGlobals::recvIndex[fromProcNo]
 116             << Foam::endl;
 117     }
 118
 119     PstreamGlobals::gammaWait(fromProcNo);
 120     messageSize = PstreamGlobals::copyReceive(fromProcNo, buf, bufSize);
 121
 122     if (Pstream::debug)
 123     {
 124         Pout<< "IPstream::read : Received " << messageSize
 125             << " from " << fromProcNo
 126             << Foam::endl;
 127     }
 128
 129     return messageSize;
 130 }
 131
 132
 133 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 134
 135 } // End namespace Foam
 136
 137 // ************************************************************************* //