1 files changed, 138 insertions, 0 deletions
diff --git a/src/particles/superParticleSysVTUout.h b/src/particles/superParticleSysVTUout.h
new file mode 100644
index 0000000..a6c90b0
--- /dev/null
+++ b/src/particles/superParticleSysVTUout.h
@@ -0,0 +1,138 @@
+/*  This file is part of the OpenLB library
+ *
+ *  Copyright (C) 2013 Thomas Henn, 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 SUPERPARTICLESYSVTUOUT_H
+#define SUPERPARTICLESYSVTUOUT_H
+
+#include <stddef.h>
+#include <sys/types.h>
+#include <cstdint>
+#include <fstream>
+#include <iostream>
+#include <string>
+#include <bitset>
+
+#include "core/singleton.h"
+#include "io/base64.h"
+#include "io/base64.hh"
+#include "io/fileName.h"
+#include "io/ostreamManager.h"
+#include "superParticleSystem3D.h"
+
+namespace olb {
+
+template<typename T, template<typename U> class PARTICLETYPE>
+class SuperParticleSystem3D;
+
+template<typename T, template<typename U> class PARTICLETYPE>
+class SuperParticleSysVtuWriter {
+public:
+  //  SuperParticleSysVtuWriter() = default;
+  SuperParticleSysVtuWriter(SuperParticleSystem3D<T, PARTICLETYPE>&,
+                            std::string const,
+                            unsigned short properties,
+                            bool binary = true);
+  SuperParticleSysVtuWriter(const SuperParticleSysVtuWriter<T, PARTICLETYPE>& rhs);
+  SuperParticleSysVtuWriter(const SuperParticleSysVtuWriter<T, PARTICLETYPE>&& rhs);
+
+  void write(int iT = 0);
+
+  int numofpsys();
+
+  void set(unsigned short);
+
+  enum particleProperties
+    : unsigned short {velocity = 1, radius = 2, mass = 4, force = 8, cuboid = 16, active = 32};
+
+//private:
+protected:
+  ///  performes <VTKFile ...>, <ImageData ...> and <PieceExtent ...>
+  void preambleVTU(const std::string& fullName);
+  ///  performes </ImageData> and </VTKFile>
+  void closeVTU(const std::string& fullNamePiece);
+  ///  performes <VTKFile ...> and <Collection>
+  void preamblePVD(const std::string& fullNamePVD);
+  ///  performes </Collection> and </VTKFile>
+  void closePVD(const std::string& fullNamePVD);
+  ///  performes <DataSet timestep= ... file=namePiece />
+  void dataPVD(int iT, int iR, const std::string& fullNamePVD,
+               const std::string& namePiece);
+  ///  performes <DataSet timestep= ... file=namePiece />
+  void dataPVDmaster(int iT, int iR, const std::string& fullNamePVDMaster,
+                     const std::string& namePiece);
+  ///  writes functors stored at pointerVec
+  void dataArray(const std::string& fullName);
+  ///  writes functors stored at pointerVec
+  void dataArrayBinary(const std::string& fullName);
+  ///  performes <VTKFile...> and <ImageData ...>
+  void preambleOneFile(const std::string& fullName);
+  ///  writes instantaniously given functor, without adding to _pointerVec
+  void writePieceToOneFile(const std::string& fullName);
+
+  void createMasterFile();
+
+//private:
+protected:
+  SuperParticleSystem3D<T, PARTICLETYPE>& _psys;
+  std::string _name;
+  unsigned short _properties;
+  bool _binary;
+  bool _haveMaster;
+  mutable OstreamManager clout;
+
+};
+
+template<typename T>
+class SuperParticleSysVtuWriterMag : public SuperParticleSysVtuWriter<T, MagneticParticle3D> {
+public:
+  const int pPropActive = 0;
+  const int pPropCuboid = 1;
+  const int pPropMass = 2;
+  const int pPropRadius = 3;
+  const int pPropVelocity = 4;
+  const int pPropForce = 5;
+  const int pPropAVel = 6;
+  const int pPropTorque = 7;
+  const int pPropMoment = 8;
+
+  SuperParticleSysVtuWriterMag(SuperParticleSystem3D<T, MagneticParticle3D>&,
+                               std::string const,
+                               const std::bitset<9>& properties,
+                               bool binary = true);
+  SuperParticleSysVtuWriterMag(const SuperParticleSysVtuWriterMag<T>& rhs);
+  SuperParticleSysVtuWriterMag(const SuperParticleSysVtuWriterMag<T>&& rhs);
+
+  void write(int iT = 0);
+  void set(int);
+
+private:
+  void dataArray(const std::string& fullName);
+  void dataArrayBinary(const std::string& fullName);
+
+  std::bitset<9> _properties;
+
+};
+
+}  // namespace OLB
+
+#endif /* SUPERPARTICLESYSVTUOUT_H */