/usr/include/dolfin/io/VTKWriter.h is in libdolfin-dev 2017.2.0.post0-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 | // Copyright (C) 2010 Garth N. Wells
//
// This file is part of DOLFIN.
//
// DOLFIN is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// DOLFIN 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with DOLFIN. If not, see <http://www.gnu.org/licenses/>.
//
// First added: 2010-07-19
// Last changed:
#ifndef __VTK_WRITER_H
#define __VTK_WRITER_H
#include <cstdint>
#include <string>
#include <vector>
#include "Encoder.h"
namespace dolfin
{
class Function;
class Mesh;
/// Write VTK Mesh representation
class VTKWriter
{
public:
/// Mesh writer
static void write_mesh(const Mesh& mesh, std::size_t cell_dim,
std::string file,
bool binary, bool compress);
/// Cell data writer
static void write_cell_data(const Function& u, std::string file,
bool binary, bool compress);
/// Form (compressed) base64 encoded string for VTK
template<typename T>
static std::string encode_stream(const std::vector<T>& data,
bool compress);
//friend class VTKFile;
private:
// Write cell data (ascii)
static std::string ascii_cell_data(const Mesh& mesh,
const std::vector<std::size_t>& offset,
const std::vector<double>& values,
std::size_t dim, std::size_t rank);
// Write cell data (base64)
static std::string base64_cell_data(const Mesh& mesh,
const std::vector<std::size_t>& offset,
const std::vector<double>& values,
std::size_t dim, std::size_t rank,
bool compress);
// Mesh writer (ascii)
static void write_ascii_mesh(const Mesh& mesh, std::size_t cell_dim,
std::string file);
// Mesh writer (base64)
static void write_base64_mesh(const Mesh& mesh, std::size_t cell_dim,
std::string file, bool compress);
// Get VTK cell type
static std::uint8_t vtk_cell_type(const Mesh& mesh, std::size_t cell_dim);
// Compute base64 encoded stream for VTK
template<typename T>
static std::string encode_inline_base64(const std::vector<T>& data);
// Compute compressed base64 encoded stream for VTK
template<typename T>
static std::string encode_inline_compressed_base64(const std::vector<T>&
data);
};
//--------------------------------------------------------------------------
template<typename T>
std::string VTKWriter::encode_stream(const std::vector<T>& data,
bool compress)
{
std::stringstream stream;
if (compress)
{
#ifdef HAS_ZLIB
return encode_inline_compressed_base64(data);
#else
warning("zlib must be configured to enable compressed VTK output. Using uncompressed base64 encoding instead.");
return encode_inline_base64(data);
#endif
}
else
return encode_inline_base64(data);
}
//--------------------------------------------------------------------------
template<typename T>
std::string VTKWriter::encode_inline_base64(const std::vector<T>& data)
{
std::stringstream stream;
const std::uint32_t size = data.size()*sizeof(T);
Encoder::encode_base64(&size, 1, stream);
Encoder::encode_base64(data, stream);
return stream.str();
}
//--------------------------------------------------------------------------
#ifdef HAS_ZLIB
template<typename T>
std::string VTKWriter::encode_inline_compressed_base64(const std::vector<T>&
data)
{
std::stringstream stream;
std::uint32_t header[4];
header[0] = 1;
header[1] = data.size()*sizeof(T);
header[2] = 0;
// Compress data
std::vector<unsigned char> compressed_data
= Encoder::compress_data(data);
// Length of compressed data
header[3] = compressed_data.size();
// Encode header
Encoder::encode_base64(&header[0], 4, stream);
// Encode data
Encoder::encode_base64(compressed_data.data(),
compressed_data.size(), stream);
return stream.str();
}
#endif
//--------------------------------------------------------------------------
}
#endif
|