superVtmWriter3D.h

Go to the documentation of this file.

/*  This file is part of the OpenLB library
 *
 *  Copyright (C) 2016-2017 Albert Mink, Maximilian Gaedtke, Markus Morhard 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_3D_H
#define SUPER_VTM_WRITER_3D_H
 
#include <sstream>
#include <vector>
#include "io/ostreamManager.h"
#include "functors/lattice/superBaseF3D.h"
 
namespace olb {
 
template<typename T, typename OUT_T=float, typename W=T>
class SuperVTMwriter3D {
public:
  SuperVTMwriter3D( const std::string& name, int overlap = 1, bool binary=true, bool compress=true );
  SuperVTMwriter3D( CuboidGeometry3D<T>& cGeometry,
                    const std::string& name, int overlap = 1, bool binary=true, bool compress=true );
  void write(int iT=0);
  void writePVD(int iT);
  void writeGlobalVTI(int iT, int iC);
  void writeVTI(int iT, int iCloc);
 
  void write(SuperF3D<T,W>& f, int iT=0);
  void write(std::shared_ptr<SuperF3D<T,W>> ptr_f, int iT=0);
 
  //   the master pvd file, where all vti are linked!
  void createMasterFile();
 
  void addFunctor(SuperF3D<T,W>& f);
  void addFunctor(SuperF3D<T,W>& f, const std::string& functorName);
 
  void clearAddedFunctors();
 
  std::string getName() const;
 
private:
  CuboidGeometry3D<T>* _cGeometry = nullptr;
 
  void preambleVTI(const std::string& fullName, const Vector<int,3> extent0, const Vector<int,3> extent1,
                   T origin[], 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, SuperF3D<T,W>& f,
                 int iC, const Vector<int,3> extent1);
  void closePiece(const std::string& fullNamePiece);
 
  OstreamManager clout;
  bool _createFile;
  std::string const _name;
  std::vector< SuperF3D<T,W>* > _pointerVec;
  int _overlap;
  bool _binary;
  bool _compress;
 
};
 
 
}  // namespace olb
 
 
#endif