src/OpenFOAM/db/IOstreams/Pstreams/PstreamCommsStruct.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 \*---------------------------------------------------------------------------*/
  26
  27 #include "Pstream.H"
  28 #include "boolList.H"
  29
  30 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  31
  32 Foam::Pstream::commsStruct::commsStruct()
  33 :
  34     above_(-1),
  35     below_(0),
  36     allBelow_(0),
  37     allNotBelow_(0)
  38 {}
  39
  40
  41 Foam::Pstream::commsStruct::commsStruct
  42 (
  43     const label above,
  44     const labelList& below,
  45     const labelList& allBelow,
  46     const labelList& allNotBelow
  47 )
  48 :
  49     above_(above),
  50     below_(below),
  51     allBelow_(allBelow),
  52     allNotBelow_(allNotBelow)
  53 {}
  54
  55
  56 Foam::Pstream::commsStruct::commsStruct
  57 (
  58     const label nProcs,
  59     const label myProcID,
  60     const label above,
  61     const labelList& below,
  62     const labelList& allBelow
  63 )
  64 :
  65     above_(above),
  66     below_(below),
  67     allBelow_(allBelow),
  68     allNotBelow_(nProcs - allBelow.size() - 1)
  69 {
  70     boolList inBelow(nProcs, false);
  71
  72     forAll(allBelow, belowI)
  73     {
  74         inBelow[allBelow[belowI]] = true;
  75     }
  76
  77     label notI = 0;
  78     forAll(inBelow, procI)
  79     {
  80         if ((procI != myProcID) && !inBelow[procI])
  81         {
  82             allNotBelow_[notI++] = procI;
  83         }
  84     }
  85     if (notI != allNotBelow_.size())
  86     {
  87         FatalErrorIn("commsStruct") << "problem!" << Foam::abort(FatalError);
  88     }
  89 }
  90
  91
  92 // * * * * * * * * * * * * * * * Member Operators  * * * * * * * * * * * * * //
  93
  94 bool Foam::Pstream::commsStruct::operator==(const commsStruct& comm) const
  95 {
  96     return
  97     (
  98         (above_ == comm.above())
  99      && (below_ == comm.below())
 100      && (allBelow_ == allBelow())
 101      && (allNotBelow_ == allNotBelow())
 102     );
 103 }
 104
 105
 106 bool Foam::Pstream::commsStruct::operator!=(const commsStruct& comm) const
 107 {
 108     return !operator==(comm);
 109 }
 110
 111
 112 // * * * * * * * * * * * * * * * Ostream Operator  * * * * * * * * * * * * * //
 113
 114 Foam::Ostream& Foam::operator<<(Ostream& os, const Pstream::commsStruct& comm)
 115 {
 116     os  << comm.above_ << token::SPACE
 117         << comm.below_ << token::SPACE
 118         << comm.allBelow_ << token::SPACE
 119         << comm.allNotBelow_;
 120
 121     os.check
 122     (
 123         "Ostream& operator<<(Ostream&, const commsStruct&)"
 124     );
 125
 126     return os;
 127 }
 128
 129
 130 // ************************************************************************* //