MercuryDPM  Trunk
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
ParticleVtkWriter.cc
Go to the documentation of this file.
1 //Copyright (c) 2013-2020, The MercuryDPM Developers Team. All rights reserved.
2 //For the list of developers, see <http://www.MercuryDPM.org/Team>.
3 //
4 //Redistribution and use in source and binary forms, with or without
5 //modification, are permitted provided that the following conditions are met:
6 // * Redistributions of source code must retain the above copyright
7 // notice, this list of conditions and the following disclaimer.
8 // * Redistributions in binary form must reproduce the above copyright
9 // notice, this list of conditions and the following disclaimer in the
10 // documentation and/or other materials provided with the distribution.
11 // * Neither the name MercuryDPM nor the
12 // names of its contributors may be used to endorse or promote products
13 // derived from this software without specific prior written permission.
14 //
15 //THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
16 //ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 //WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 //DISCLAIMED. IN NO EVENT SHALL THE MERCURYDPM DEVELOPERS TEAM BE LIABLE FOR ANY
19 //DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
20 //(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
21 //LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
22 //ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 //(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
24 //SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 
27 #include "DPMBase.h"
28 
29 void ParticleVtkWriter::writeVTKPositions(std::fstream& file) const
30 {
31  file << "<Points>\n";
32  file << " <DataArray type=\"Float32\" Name=\"Position\" NumberOfComponents=\"3\" format=\"ascii\">\n";
33  for (const auto& p: handler_)
34  {
35 #ifdef MERCURY_USE_MPI
36  if (particleMustBeWritten(p))
37  {
38  file << '\t' << p->getPosition() << '\n';
39  }
40 #else
41  file << '\t' << p->getPosition() << '\n';
42 #endif
43 
44  }
45  file << " </DataArray>\n";
46  file << "</Points>\n";
47 }
48 
49 void ParticleVtkWriter::writeVTKIndSpecies(std::fstream& file) const
50 {
51  file << " <DataArray type=\"Float32\" Name=\"speciesType\" format=\"ascii\">\n";
52  // Add species type
53  for (const auto& p: handler_)
54  {
55 #ifdef MERCURY_USE_MPI
56  if (particleMustBeWritten(p))
57  {
58  file << '\t' << p->getIndSpecies() << '\n';
59  }
60 #else
61  file << '\t' << p->getIndSpecies() << '\n';
62 #endif
63  }
64  file << " </DataArray>\n";
65 }
66 
67 void ParticleVtkWriter::writeExtraFields(std::fstream& file) const
68 {
69  //check if this type of Interaction has extra fields
70  if (handler_.getSize() != 0)
71  {
72  for (unsigned i = 0; i < handler_.getLastObject()->getNumberOfFieldsVTK(); i++)
73  {
74  file << " <DataArray type=\"" << handler_.getLastObject()->getTypeVTK(i) << "\" Name=\""
75  << handler_.getLastObject()->getNameVTK(i) << "\" format=\"ascii\">\n";
76  // Add species type
77  for (const auto& p: handler_)
78  {
79  for (auto f : p->getFieldVTK(i)) {
80 #ifdef MERCURY_USE_MPI
81  if (particleMustBeWritten(p))
82  {
83  file << '\t' << f << '\n';
84  }
85 #else
86  file << '\t' << f << '\n';
87 #endif
88  }
89  }
90  file << " </DataArray>\n";
91  }
92  }
93 }
void writeVTKPositions(std::fstream &file) const
unsigned int getSize() const
Gets the size of the particleHandler (including mpi and periodic particles)
Definition: BaseHandler.h:655
void writeVTKIndSpecies(std::fstream &file) const
ParticleHandler & handler_
particle handler from which the particles should be written
Definition: BaseVTKWriter.h:72
bool particleMustBeWritten(BaseParticle *particle) const
const std::complex< Mdouble > i
Definition: ExtendedMath.h:50
void writeExtraFields(std::fstream &file) const
virtual std::string getTypeVTK(unsigned i) const
virtual std::string getNameVTK(unsigned i) const
T * getLastObject()
Gets a pointer to the last Object in this BaseHandler.
Definition: BaseHandler.h:634
virtual unsigned getNumberOfFieldsVTK() const