File: test_hdf5_3dimage.cc

package info (click to toggle)
mia 2.4.7-16
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 17,332 kB
  • sloc: cpp: 162,690; python: 1,317; sh: 321; xml: 127; makefile: 41; csh: 24; ansic: 9
file content (108 lines) | stat: -rw-r--r-- 3,104 bytes parent folder | download | duplicates (3)
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
/* -*- mia-c++  -*-
 *
 * This file is part of MIA - a toolbox for medical image analysis
 * Copyright (c) Leipzig, Madrid 1999-2017 Gert Wollny
 *
 * MIA is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with MIA; if not, see <http://www.gnu.org/licenses/>.
 *
 */


#define VSTREAM_DOMAIN "test-HDF5-3dimage"
#include <mia/internal/autotest.hh>

#include <boost/mpl/vector.hpp>
#include <boost/test/unit_test.hpp>

#include <addons/hdf5/hdf5_3dimage.hh>

using namespace std;
using namespace mia;
using namespace hdf5_3dimage;

template <typename T>
struct __fill_image {
       static void apply(T3DImage<T>& image)
       {
              int v = 1;

              for (auto i = image.begin(); i != image.end(); ++i)
                     *i = v++;
       }
};

template <>
struct __fill_image<bool> {
       static void apply(C3DBitImage& image)
       {
              bool v = false;

              for (auto i = image.begin(); i != image.end(); ++i) {
                     *i = v;
                     v = !v;
              }
       }
};

typedef boost::mpl::vector<bool,
        int8_t,
        uint8_t,
        int16_t,
        uint16_t,
        int32_t,
        uint32_t,
        int64_t,
        uint64_t,
        float,
        double
        > test_types;


BOOST_AUTO_TEST_CASE_TEMPLATE( test_simple_write_read, T, test_types )
{
       C3DBounds size (2, 3, 4);
       T3DImage<T> *image = new T3DImage<T>(size);
       image->set_attribute("int", PAttribute(new CIntAttribute(2)));
       const C3DFVector voxel_size(2, 2.5, 2.5);
       image->set_voxel_size(voxel_size);
       __fill_image<T>::apply(*image);
       CHDF53DImageIOPlugin io;
       CHDF53DImageIOPlugin::Data images;
       images.push_back(P3DImage(image));
       stringstream filename;
       filename << "testimage-" << __type_descr<T>::value << ".h5";
       cvdebug() << "test with " << filename.str() << "\n";
       BOOST_REQUIRE(io.save(filename.str(), images));
       auto loaded = io.load(filename.str());
       BOOST_REQUIRE(loaded);
       BOOST_REQUIRE(loaded->size() == 1u);
       const auto& ploaded = dynamic_cast<const T3DImage<T>&>(*(*loaded)[0]);
       auto iv = image->begin();
       auto ev = image->end();
       auto il = ploaded.begin();

       while (iv != ev) {
              BOOST_CHECK_EQUAL(*il, *iv);
              ++iv;
              ++il;
       }

       BOOST_CHECK_EQUAL(ploaded.get_voxel_size(), voxel_size);
       auto int_attr = ploaded.template get_attribute_as<int>("int");
       BOOST_CHECK_EQUAL(int_attr, 2);
       unlink(filename.str().c_str());
}