vtk_writer.hpp

// Copyright (C) 2011-2012 by the BEM++ Authors
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#ifndef bempp_vtk_writer_hpp
#define bempp_vtk_writer_hpp

#include "../common/common.hpp"

#include "../common/armadillo_fwd.hpp"
#include <memory>
#include <string>

namespace Bempp
{

class VtkWriter
{
public:
    enum OutputType {
      ASCII,
      BASE_64,
      APPENDED_RAW,
      APPENDED_BASE_64
    };

    enum DataType
    {
        CELL_DATA,
        VERTEX_DATA
    };

    virtual ~VtkWriter() {}

    void addCellData(const arma::Mat<double>& data, const std::string &name);
    void addCellData(const arma::Mat<float>& data, const std::string &name);

    void addVertexData(const arma::Mat<double>& data, const std::string &name);
    void addVertexData(const arma::Mat<float>& data, const std::string &name);

    virtual void clear() = 0;

    virtual std::string write (const std::string &name,
                               OutputType type = ASCII) = 0;

    virtual std::string pwrite(const std::string& name, const std::string& path,
                               const std::string& extendpath,
                               OutputType type = ASCII) = 0;

private:
    virtual void addCellDataDoubleImpl(
            const arma::Mat<double>& data, const std::string &name) = 0;
    virtual void addCellDataFloatImpl(
            const arma::Mat<float>& data, const std::string &name) = 0;

    virtual void addVertexDataDoubleImpl(
            const arma::Mat<double>& data, const std::string &name) = 0;
    virtual void addVertexDataFloatImpl(
            const arma::Mat<float>& data, const std::string &name) = 0;
};

inline void VtkWriter::addCellData(const arma::Mat<double>& data,
                                   const std::string& name)
{
    addCellDataDoubleImpl(data, name);
}

inline void VtkWriter::addCellData(const arma::Mat<float>& data,
                                   const std::string& name)
{
    addCellDataFloatImpl(data, name);
}

inline void VtkWriter::addVertexData(const arma::Mat<double>& data,
                                     const std::string& name)
{
    addVertexDataDoubleImpl(data, name);
}

inline void VtkWriter::addVertexData(const arma::Mat<float>& data,
                                     const std::string& name)
{
    addVertexDataFloatImpl(data, name);
}

} // namespace Bempp

#endif