PLearn 0.1
MPIStream.h
Go to the documentation of this file.
00001 // -*- C++ -*-
00002 
00003 // PLearn (A C++ Machine Learning Library)
00004 // Copyright (C) 1998 Pascal Vincent
00005 // Copyright (C) 1999-2002 Pascal Vincent, Yoshua Bengio and University of Montreal
00006 //
00007 
00008 // Redistribution and use in source and binary forms, with or without
00009 // modification, are permitted provided that the following conditions are met:
00010 // 
00011 //  1. Redistributions of source code must retain the above copyright
00012 //     notice, this list of conditions and the following disclaimer.
00013 // 
00014 //  2. Redistributions in binary form must reproduce the above copyright
00015 //     notice, this list of conditions and the following disclaimer in the
00016 //     documentation and/or other materials provided with the distribution.
00017 // 
00018 //  3. The name of the authors may not be used to endorse or promote
00019 //     products derived from this software without specific prior written
00020 //     permission.
00021 // 
00022 // THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS OR
00023 // IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
00024 // OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN
00025 // NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
00026 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
00027 // TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
00028 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
00029 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
00030 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
00031 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00032 // 
00033 // This file is part of the PLearn library. For more information on the PLearn
00034 // library, go to the PLearn Web site at www.plearn.org
00035 
00036 
00037  
00038 
00039 /* *******************************************************      
00040  * $Id: MPIStream.h 8232 2007-11-07 20:52:17Z nouiz $
00041  * AUTHORS: Pascal Vincent 
00042  * This file is part of the PLearn library.
00043  ******************************************************* */
00044 
00045 
00048 #ifndef MPIStream_INC
00049 #define MPIStream_INC
00050 
00051 #if USING_MPI
00052 
00053 #include <iostream>
00054 #include <plearn/sys/PLMPI.h>
00055 
00056 namespace PLearn {
00057 using namespace std;
00058 
00059 class MPIStreambuf : public streambuf
00060 {
00061 protected:
00062     int tag; 
00063     int peerrank; 
00064   
00065     char* inbuf;
00066     int inbuf_capacity;
00067   
00068 protected:
00069 
00071     inline void reserveInputBuffer(int buflen)
00072     {
00073         if(buflen>inbuf_capacity)
00074         {
00075             if(inbuf)
00076                 delete[] inbuf;
00077             inbuf = new char[buflen];
00078             inbuf_capacity = buflen;
00079         }
00080     }
00081 
00083     virtual int overflow(int c=EOF);
00084     virtual int underflow();
00085     // virtual int uflow();
00087     virtual int showmanyc();
00088     virtual streamsize xsputn(const char* s, streamsize n);
00089     virtual streamsize xsgetn(char* s, streamsize n);
00092     virtual streambuf* setbuf(char* p, int len);
00093     virtual int sync();
00094 
00095 public:
00096     MPIStreambuf(int the_peerrank, int inbufsize);
00097     virtual ~MPIStreambuf();
00098 
00099 };
00100   
00101 class MPIStream: public iostream
00102 {
00103 protected:
00104     char* outbuffer;
00105 
00106 public:
00107 
00108     MPIStream()
00109         :iostream(0), outbuffer(0)
00110     {}
00111 
00113     MPIStream(int the_peerrank, int inbufsize=200, int outbufsize=200)
00114         :iostream(0), outbuffer(0)
00115     { init(the_peerrank, inbufsize, outbufsize); }
00116 
00118     void init(int the_peerrank, int inbufsize= 0, int outbufsize= 0);
00119 
00120     ~MPIStream();
00121 };
00122 
00123 
00124 // implements an "array" with streams open to all nodes
00125 class MPIStreams
00126 {
00127 protected:
00128     MPIStream* mpistreams;
00129 
00130 public:
00131     MPIStreams(int inbufsize=200, int outbufsize=200);
00132 
00133     int size() const { return PLMPI::size; } 
00134     int length() const { return PLMPI::size; } 
00135 
00136     iostream& operator[](int i)
00137     {
00138 #ifdef BOUNDCHECK
00139         if(i<0 || i>=PLMPI::size)
00140             PLERROR("In MPIStreams OUT OF BOUND ACCESS");
00141 #endif      
00142         return mpistreams[i];
00143     }
00144 
00145     ~MPIStreams() 
00146     { if(mpistreams) delete[] mpistreams; }
00147 };
00148 
00149 } // end of namespace PLearn
00150 
00152 #endif 
00153 
00155 #endif
00156 
00157 
00158 /*
00159   Local Variables:
00160   mode:c++
00161   c-basic-offset:4
00162   c-file-style:"stroustrup"
00163   c-file-offsets:((innamespace . 0)(inline-open . 0))
00164   indent-tabs-mode:nil
00165   fill-column:79
00166   End:
00167 */
00168 // vim: filetype=cpp:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=79 :
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines