Line | Branch | Exec | Source |
---|---|---|---|
1 | // SPDX-FileCopyrightText: 2022-2023 Dennis Gläser <dennis.glaeser@iws.uni-stuttgart.de> | ||
2 | // SPDX-License-Identifier: MIT | ||
3 | /*! | ||
4 | * \file | ||
5 | * \ingroup VTK | ||
6 | * \copydoc GridFormat::VTSWriter | ||
7 | */ | ||
8 | #ifndef GRIDFORMAT_VTK_VTS_WRITER_HPP_ | ||
9 | #define GRIDFORMAT_VTK_VTS_WRITER_HPP_ | ||
10 | |||
11 | #include <array> | ||
12 | #include <ranges> | ||
13 | #include <ostream> | ||
14 | #include <utility> | ||
15 | #include <string> | ||
16 | #include <optional> | ||
17 | |||
18 | #include <gridformat/common/field.hpp> | ||
19 | #include <gridformat/common/field_storage.hpp> | ||
20 | #include <gridformat/common/lvalue_reference.hpp> | ||
21 | |||
22 | #include <gridformat/grid/grid.hpp> | ||
23 | #include <gridformat/grid/concepts.hpp> | ||
24 | #include <gridformat/grid/type_traits.hpp> | ||
25 | #include <gridformat/vtk/common.hpp> | ||
26 | #include <gridformat/vtk/xml.hpp> | ||
27 | |||
28 | namespace GridFormat { | ||
29 | |||
30 | /*! | ||
31 | * \ingroup VTK | ||
32 | * \brief Writer for .vts file format | ||
33 | */ | ||
34 | template<Concepts::StructuredGrid Grid> | ||
35 | class VTSWriter : public VTK::XMLWriterBase<Grid, VTSWriter<Grid>> { | ||
36 | using ParentType = VTK::XMLWriterBase<Grid, VTSWriter<Grid>>; | ||
37 | using CType = CoordinateType<Grid>; | ||
38 | static constexpr std::size_t dim = dimension<Grid>; | ||
39 | static_assert(dim <= 3); | ||
40 | |||
41 | public: | ||
42 | struct Domain { | ||
43 | std::array<std::size_t, dim> whole_extent; | ||
44 | }; | ||
45 | |||
46 | using Offset = std::array<std::size_t, dim>; | ||
47 | |||
48 | 14181 | explicit VTSWriter(LValueReferenceOf<const Grid> grid, | |
49 | VTK::XMLOptions xml_opts = {}) | ||
50 |
2/4✓ Branch 2 taken 7141 times.
✗ Branch 3 not taken.
✓ Branch 6 taken 7141 times.
✗ Branch 7 not taken.
|
42543 | : ParentType(grid.get(), ".vts", true, std::move(xml_opts)) |
51 | 14181 | {} | |
52 | |||
53 | 4112 | VTSWriter as_piece_for(Domain domain) const { | |
54 | 4112 | auto result = this->with(this->_xml_opts); | |
55 | 4112 | result._domain = std::move(domain); | |
56 | 4112 | result._offset = _offset; | |
57 | 4112 | return result; | |
58 | } | ||
59 | |||
60 | 4112 | VTSWriter with_offset(Offset offset) const { | |
61 | 4112 | auto result = this->with(this->_xml_opts); | |
62 | 4112 | result._offset = std::move(offset); | |
63 | 4112 | result._domain = _domain; | |
64 | 4112 | return result; | |
65 | } | ||
66 | |||
67 | private: | ||
68 | 9484 | VTSWriter _with(VTK::XMLOptions xml_opts) const override { | |
69 |
1/2✓ Branch 4 taken 4774 times.
✗ Branch 5 not taken.
|
9484 | return VTSWriter{this->grid(), std::move(xml_opts)}; |
70 | } | ||
71 | |||
72 | 4744 | void _write(std::ostream& s) const override { | |
73 |
2/4✓ Branch 1 taken 2396 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2396 times.
✗ Branch 5 not taken.
|
9488 | auto context = this->_get_write_context("StructuredGrid"); |
74 |
1/2✓ Branch 1 taken 2396 times.
✗ Branch 2 not taken.
|
4744 | _set_attributes(context); |
75 | |||
76 | 4744 | FieldStorage vtk_point_fields; | |
77 | 4744 | FieldStorage vtk_cell_fields; | |
78 |
2/4✓ Branch 1 taken 2396 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2396 times.
✗ Branch 5 not taken.
|
29588 | std::ranges::for_each(this->_point_field_names(), [&] (const std::string& name) { |
79 |
6/12✓ Branch 1 taken 2929 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2929 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 2929 times.
✗ Branch 8 not taken.
✓ Branch 15 taken 9721 times.
✗ Branch 16 not taken.
✓ Branch 18 taken 9721 times.
✗ Branch 19 not taken.
✓ Branch 21 taken 9721 times.
✗ Branch 22 not taken.
|
12650 | vtk_cell_fields.set(name, VTK::make_vtk_field(this->_get_point_field_ptr(name))); |
80 |
4/8✓ Branch 2 taken 2929 times.
✗ Branch 3 not taken.
✓ Branch 6 taken 2929 times.
✗ Branch 7 not taken.
✓ Branch 12 taken 9721 times.
✗ Branch 13 not taken.
✓ Branch 16 taken 9721 times.
✗ Branch 17 not taken.
|
12650 | this->_set_data_array(context, "Piece/PointData", name, vtk_cell_fields.get(name)); |
81 | }); | ||
82 |
2/4✓ Branch 1 taken 2396 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2396 times.
✗ Branch 5 not taken.
|
29588 | std::ranges::for_each(this->_cell_field_names(), [&] (const std::string& name) { |
83 |
6/12✓ Branch 1 taken 2929 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2929 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 2929 times.
✗ Branch 8 not taken.
✓ Branch 15 taken 9721 times.
✗ Branch 16 not taken.
✓ Branch 18 taken 9721 times.
✗ Branch 19 not taken.
✓ Branch 21 taken 9721 times.
✗ Branch 22 not taken.
|
12650 | vtk_cell_fields.set(name, VTK::make_vtk_field(this->_get_cell_field_ptr(name))); |
84 |
4/8✓ Branch 2 taken 2929 times.
✗ Branch 3 not taken.
✓ Branch 6 taken 2929 times.
✗ Branch 7 not taken.
✓ Branch 12 taken 9721 times.
✗ Branch 13 not taken.
✓ Branch 16 taken 9721 times.
✗ Branch 17 not taken.
|
12650 | this->_set_data_array(context, "Piece/CellData", name, vtk_cell_fields.get(name)); |
85 | }); | ||
86 | |||
87 | 2444 | const FieldPtr coords_field = std::visit([&] <typename T> (const Precision<T>&) { | |
88 | 2396 | return VTK::make_coordinates_field<T>(this->grid(), true); | |
89 |
1/2✓ Branch 1 taken 2396 times.
✗ Branch 2 not taken.
|
4744 | }, this->_xml_settings.coordinate_precision); |
90 |
3/4✓ Branch 2 taken 2396 times.
✗ Branch 3 not taken.
✓ Branch 6 taken 2120 times.
✓ Branch 7 taken 276 times.
|
10040 | this->_set_data_array(context, "Piece/Points", "Coordinates", *coords_field); |
91 | |||
92 |
1/2✓ Branch 2 taken 2120 times.
✗ Branch 3 not taken.
|
4192 | this->_write_xml(std::move(context), s); |
93 | 6400 | } | |
94 | |||
95 | 4744 | void _set_attributes(typename ParentType::WriteContext& context) const { | |
96 | 4744 | _set_domain_attributes(context); | |
97 | 4744 | _set_extent_attributes(context); | |
98 | 4744 | } | |
99 | |||
100 | 4744 | void _set_domain_attributes(typename ParentType::WriteContext& context) const { | |
101 | using VTK::CommonDetail::extents_string; | ||
102 |
2/2✓ Branch 1 taken 2072 times.
✓ Branch 2 taken 324 times.
|
4744 | if (_domain) |
103 |
3/6✓ Branch 2 taken 2072 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 2072 times.
✗ Branch 6 not taken.
✓ Branch 9 taken 2072 times.
✗ Branch 10 not taken.
|
12336 | this->_set_attribute(context, "", "WholeExtent", extents_string(_domain->whole_extent)); |
104 | else | ||
105 |
3/6✓ Branch 2 taken 324 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 324 times.
✗ Branch 6 not taken.
✓ Branch 9 taken 324 times.
✗ Branch 10 not taken.
|
1896 | this->_set_attribute(context, "", "WholeExtent", extents_string(this->grid())); |
106 | 4744 | } | |
107 | |||
108 | 4744 | void _set_extent_attributes(typename ParentType::WriteContext& context) const { | |
109 | using VTK::CommonDetail::extents_string; | ||
110 |
2/2✓ Branch 1 taken 2072 times.
✓ Branch 2 taken 324 times.
|
4744 | if (int i = 0; _offset) { |
111 | 4112 | auto begin = (*_offset); | |
112 |
1/2✓ Branch 2 taken 8 times.
✗ Branch 3 not taken.
|
4112 | auto end = GridFormat::extents(this->grid()); |
113 |
1/2✓ Branch 1 taken 2072 times.
✗ Branch 2 not taken.
|
9888 | std::ranges::for_each(end, [&] (std::integral auto& ex) { ex += begin[i++]; }); |
114 |
3/6✓ Branch 1 taken 2072 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2072 times.
✗ Branch 5 not taken.
✓ Branch 8 taken 2072 times.
✗ Branch 9 not taken.
|
12336 | this->_set_attribute(context, "Piece", "Extent", extents_string(begin, end)); |
115 | } else { | ||
116 |
3/6✓ Branch 2 taken 324 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 324 times.
✗ Branch 6 not taken.
✓ Branch 9 taken 324 times.
✗ Branch 10 not taken.
|
1896 | this->_set_attribute(context, "Piece", "Extent", extents_string(this->grid())); |
117 | } | ||
118 | 4744 | } | |
119 | |||
120 | std::optional<Domain> _domain; | ||
121 | std::optional<Offset> _offset; | ||
122 | }; | ||
123 | |||
124 | template<typename G> | ||
125 | VTSWriter(G&&, VTK::XMLOptions = {}) -> VTSWriter<std::remove_cvref_t<G>>; | ||
126 | |||
127 | } // namespace GridFormat | ||
128 | |||
129 | #endif // GRIDFORMAT_VTK_VTS_WRITER_HPP_ | ||
130 |