ostreamManager.cpp

Go to the documentation of this file.

/*  This file is part of the OpenLB library
 *
 *  Copyright (C) 2011 Lukas Baron, Mathias Krause
 *  E-mail contact: info@openlb.net
 *  The most recent release of OpenLB can be downloaded at
 *  <http://www.openlb.net/>
 *
 *  This program is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU General Public License
 *  as published by the Free Software Foundation; either version 2
 *  of the License, or (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public
 *  License along with this program; if not, write to the Free
 *  Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 *  Boston, MA  02110-1301, USA.
*/
 
#include "ostreamManager.h"
#include "core/singleton.h"
 
#include "communication/mpiManager.h"
#include "communication/ompManager.h"
 
namespace olb {
 
bool OMBuf::multiOutput = 0;
 
OMBuf::OMBuf()
  : output(nullptr), text("")
{ }
 
OMBuf::~OMBuf()
{ }
 
OMBuf::OMBuf(const OMBuf& rhs)
  : output(rhs.output), text(rhs.text)
{ }
 
OMBuf& OMBuf::operator=(const OMBuf& rhs)
{
  output = rhs.output;
  text = rhs.text;
 
  return *this;
}
 
OMBuf::OMBuf(std::ostream& str, std::string classname)
  : output(&str), text(classname)
{ }
 
void OMBuf::setMultiOutput(bool b)
{
  (*this).multiOutput = b;
}
 
int OMBuf::sync()
{
#ifdef PARALLEL_MODE_MPI
  #ifdef PARALLEL_MODE_OMP
    if (multiOutput==true) {
      *output << "["
              << text << ":" << singleton::mpi().getRank()
                      << ":" << singleton::omp().getRank()
              << "] " << str();
    }
    else {    // multiOutput==false
      if (singleton::mpi().getRank() == 0 && singleton::omp().getRank() == 0) {
        *output << "[" << text << "] " << str();
      }
    }
  #else
    if (multiOutput==true) {
      *output << "["
              << text << ":" << singleton::mpi().getRank()
              << "] " << str();
    }
    else {    // multiOutput==false
      if (singleton::mpi().getRank()==0) {
        *output << "[" << text << "] " << str();
      }
    }
  #endif
#elif PARALLEL_MODE_OMP
  if (multiOutput==true) {
    *output << "["
            << text << ":" << singleton::omp().getRank()
            << "] " << str();
  }
  else {    // multiOutput==false
    if (singleton::omp().getRank()==0) {
      *output << "[" << text << "] " << str();
    }
  }
#else
  *output << "[" << text << "] " << str();
#endif
  str("");
  output->flush();
  return 0;
}
 
// class OstreamManager /////////////////////////////////
 
OstreamManager::OstreamManager(std::string classname)
  : std::ostream(&buffer), buffer(std::cout, classname)
{ }
 
OstreamManager::OstreamManager(std::ostream& str, std::string classname)
  : std::ostream(&buffer), buffer(str, classname)
{ }
 
OstreamManager::OstreamManager(const OstreamManager& rhs)
  : std::ostream(&buffer), buffer(rhs.buffer)
{ }
 
OstreamManager& OstreamManager::operator=(const OstreamManager& rhs)
{
  buffer = rhs.buffer;
  return *this;
}
 
void OstreamManager::setMultiOutput(bool b)
{
  buffer.setMultiOutput(b);
}
 
}