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
|
// SPDX-FileCopyrightText: Copyright © DUNE Project contributors, see file LICENSE.md in module root
// SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception
// -*- tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_PYTHON_GRID_VTK_HH
#define DUNE_PYTHON_GRID_VTK_HH
#include <type_traits>
#include <utility>
#include <dune/grid/io/file/vtk/vtkwriter.hh>
#include <dune/grid/io/file/vtk/subsamplingvtkwriter.hh>
#include <dune/python/common/typeregistry.hh>
#include <dune/python/common/getdimension.hh>
#include <dune/python/grid/enums.hh>
#include <dune/python/grid/object.hh>
#include <dune/python/pybind11/pybind11.h>
namespace Dune
{
namespace Python
{
// External Forward Declarations
// -----------------------------
template< class GridFunction >
struct GridFunctionTraits;
// addToVTKWriter
// --------------
template< class GridFunction, class Writer = VTKWriter< std::decay_t< decltype( gridView( std::declval< const GridFunction & >() ) ) > > >
void addToVTKWriter ( const GridFunction &gf, std::string name, Writer &vtkWriter, VTKDataType dataType )
{
typedef typename GridFunctionTraits< GridFunction >::Range Range;
switch( dataType )
{
case VTKDataType::CellData:
{
VTK::FieldInfo info( std::move( name ), VTK::FieldInfo::Type::scalar, GetDimension<Range>::value );
vtkWriter.addCellData( gf, info );
break;
}
case VTKDataType::PointData:
{
VTK::FieldInfo info( std::move( name ), VTK::FieldInfo::Type::scalar, GetDimension<Range>::value );
vtkWriter.addVertexData( gf, info );
break;
}
case VTKDataType::CellVector:
{
VTK::FieldInfo info( std::move( name ), VTK::FieldInfo::Type::vector, GetDimension<Range>::value );
vtkWriter.addCellData( gf, info );
break;
}
case VTKDataType::PointVector:
{
VTK::FieldInfo info( std::move( name ), VTK::FieldInfo::Type::vector, GetDimension<Range>::value );
vtkWriter.addVertexData( gf, info );
break;
}
default:
DUNE_THROW( InvalidStateException, "Invalid vtk data type" );
}
}
// registerVTKWriter
// -----------------
template<class GridView>
void registerVTKWriter(pybind11::handle scope)
{
{
typedef VTKWriter< GridView > Writer;
auto cls = insertClass<Writer>(scope, "VTKWriter",
GenerateTypeName("VTKWriter", MetaType<GridView>()) ).first;
cls.def( "write",
[] ( Writer &writer, const std::string &name, Dune::VTK::OutputType outputType ) {
writer.write( name, outputType );
},
pybind11::arg("name"),
pybind11::arg("outputType")=VTK::appendedraw );
cls.def( "write",
[] ( Writer &writer, const std::string &name, int number, Dune::VTK::OutputType outputType ) {
std::stringstream s; s << name << std::setw(5) << std::setfill('0') << number;
writer.write( s.str(), outputType );
},
pybind11::arg("name"),
pybind11::arg("number"),
pybind11::arg("outputType")=VTK::appendedraw );
}
{
typedef SubsamplingVTKWriter< GridView > Writer;
auto cls = insertClass< Writer, VTKWriter<GridView> >(scope, "SubsamplingVTKWriter",
GenerateTypeName("SubsamplingVTKWriter", MetaType<GridView>()) ).first;
cls.def( "write",
[] ( Writer &writer, const std::string &name, Dune::VTK::OutputType outputType ) {
writer.write( name, outputType );
},
pybind11::arg("name"),
pybind11::arg("outputType")=VTK::appendedraw );
cls.def( "write",
[] ( Writer &writer, const std::string &name, int number ) {
std::stringstream s; s << name << std::setw(5) << std::setfill('0') << number;
writer.write( s.str() );
},
pybind11::arg("name"),
pybind11::arg("number") );
cls.def( "write",
[] ( Writer &writer, const std::string &name, int number, Dune::VTK::OutputType outputType ) {
std::stringstream s; s << name << std::setw(5) << std::setfill('0') << number;
writer.write( s.str(), outputType );
},
pybind11::arg("name"),
pybind11::arg("number"),
pybind11::arg("outputType")=VTK::appendedraw );
}
}
} // namespace Python
} // namespace Dune
#endif // #ifndef DUNE_PYTHON_GRID_VTK_HH
|
