1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
|
/////////////////////////////////////////////////////////////
// //
// Copyright (c) 2003-2011 by The University of Queensland //
// Earth Systems Science Computational Centre (ESSCC) //
// http://www.uq.edu.au/esscc //
// //
// Primary Business: Brisbane, Queensland, Australia //
// Licensed under the Open Software License version 3.0 //
// http://www.opensource.org/licenses/osl-3.0.php //
// //
/////////////////////////////////////////////////////////////
#ifndef ESYS_LSMVTKXMLWRITER_H
#define ESYS_LSMVTKXMLWRITER_H
#include <Geometry/SimpleParticle.h>
#include <Geometry/BasicInteraction.h>
#include <iostream>
#include <boost/shared_ptr.hpp>
namespace esys
{
namespace lsm
{
class ParticleDataVisitor
{
typedef SimpleParticle Particle;
typedef BasicInteraction Connection;
public:
ParticleDataVisitor();
void visitSimpleParticle(const Particle &particle);
void visitParticle(const Particle &particle);
void visitBasicInteraction(const Connection &connection);
void visitConnection(const Connection &connection);
size_t getNumParticles() const;
size_t getNumConnections() const;
int getIndex(int particleId) const;
void writeCentrePoints(std::ostream &oStream) const;
void writeRadii(std::ostream &oStream) const;
void writeTags(std::ostream &oStream) const;
void writeIds(std::ostream &oStream) const;
void writeParticleIndexConnections(std::ostream &oStream) const;
void writeConnectionTags(std::ostream &oStream) const;
private:
class Impl;
typedef boost::shared_ptr<Impl> ImplPtr;
ImplPtr m_implPtr;
};
/**
*
*/
class VtkXmlWriter
{
public:
VtkXmlWriter();
virtual ~VtkXmlWriter();
void setData(const ParticleDataVisitor &particleData);
size_t getNumParticles() const;
size_t getNumConnections() const;
virtual void writePoints(std::ostream &oStream);
virtual void writePointData(std::ostream &oStream);
virtual void writeCells(std::ostream &oStream);
virtual void writeCellData(std::ostream &oStream);
virtual void write(std::ostream &oStream);
virtual void writeToFile(const std::string &fileName);
private:
class Impl;
typedef boost::shared_ptr<Impl> ImplPtr;
ImplPtr m_implPtr;
};
};
};
#endif
|