superVtmWriter2D.h

Go to the documentation of this file.

/*  This file is part of the OpenLB library
 *
 *  Copyright (C) 2016 Albert Mink, Mathias J. 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.
*/
 
#ifndef SUPER_VTM_WRITER_2D_H
#define SUPER_VTM_WRITER_2D_H
 
#include <sstream>
#include <vector>
#include "io/ostreamManager.h"
#include "functors/lattice/superBaseF2D.h"
 
namespace olb {
 
template<typename T, typename OUT_T=float, typename W=T>
class SuperVTMwriter2D {
public:
  SuperVTMwriter2D( std::string name, int overlap = 1, bool binary = true );
  void write(int iT=0);
  void write(std::vector<T> rMin, std::vector<T> rMax, int iT=0);
  void write(SuperF2D<T,W>& f, int iT=0);
  void write(SuperF2D<T,W>& f, std::vector<T> rMin, std::vector<T> rMax, int iT=0);
  void write(std::shared_ptr<SuperF2D<T,W>> ptr_f, int iT=0);
  void write(std::shared_ptr<SuperF2D<T,W>> ptr_f, std::vector<T> rMin, std::vector<T> rMax, int iT=0);
  //   the master pvd file, where all vti are linked!
  void createMasterFile();
  void addFunctor(SuperF2D<T,W>& f);
  void addFunctor(SuperF2D<T,W>& f, const std::string& functorName);
  void clearAddedFunctors();
  std::string getName() const;
private:
  void extremes(int nMinOut[], int nMaxOut[], CuboidDecomposition2D<T> const& cGeometry, LoadBalancer<T>& load, std::vector<T> rMin, std::vector<T> rMax, int iCloc);
  void preambleVTI(const std::string& fullName, int x0, int y0, int x1, int y1,
                   T originX, T originY, T delta);
  void closeVTI(const std::string& fullNamePiece);
  void preamblePVD(const std::string& fullNamePVD);
  void closePVD(const std::string& fullNamePVD);
  void preambleVTM(const std::string& fullNamePVD);
  void closeVTM(const std::string& fullNamePVD);
  void dataVTM(int iC, const std::string& fullNamePVD, const std::string& namePiece);
  void dataPVDmaster(int iT, const std::string& fullNamePVDMaster,
                     const std::string& namePiece);
  void dataArray(const std::string& fullName, SuperF2D<T,W>& f,
                 int iC, int nxMin, int nxMax, int nyMin, int nyMax);
  void dataArrayBinary(const std::string& fullName, SuperF2D<T,W>& f,
                       int iC, int nxMin, int nxMax, int nyMin, int nyMax);
  void closePiece(const std::string& fullNamePiece);
private:
  mutable OstreamManager clout;
  bool _createFile;
  std::string _name;
  std::vector< SuperF2D<T,W>* > _pointerVec;
  int _overlap;
  bool _binary =true;
};
 
template <typename T, typename W>
void writeVTK(SuperF2D<T,W>& f, int iT=0) {
  SuperVTMwriter2D<T> writer("");
  writer.write(f, iT);
}
 
}  // namespace olb
 
 
#endif