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
|
/* Copyright (c) 2008-2022 the MRtrix3 contributors.
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*
* Covered Software is provided under this License on an "as is"
* basis, without warranty of any kind, either expressed, implied, or
* statutory, including, without limitation, warranties that the
* Covered Software is free of defects, merchantable, fit for a
* particular purpose or non-infringing.
* See the Mozilla Public License v. 2.0 for more details.
*
* For more details, see http://www.mrtrix.org/.
*/
#ifdef MRTRIX_PNG_SUPPORT
#include "image_io/png.h"
#include "header.h"
#include "image_helpers.h"
#include "file/png.h"
namespace MR
{
namespace ImageIO
{
void PNG::load (const Header& header, size_t)
{
segsize = header.datatype().bytes() * voxel_count (header) * files.size();
addresses.resize (1);
addresses[0].reset (new uint8_t [segsize]);
if (is_new) {
memset (addresses[0].get(), 0x00, segsize);
DEBUG ("allocated memory for PNG image \"" + header.name() + "\"");
} else {
DEBUG (std::string("loading PNG image") + (files.size() > 1 ? "s" : "") + " \"" + header.name() + "\"");
size_t slice_bytes = (header.datatype().bits() * header.size(0) * header.size(1) + 7) / 8;
if (header.ndim() == 4)
slice_bytes *= header.size (3);
for (size_t i = 0; i != files.size(); ++i) {
File::PNG::Reader png (files[i].name);
if (png.get_width() != header.size(0) ||
png.get_height() != header.size(1) ||
png.get_output_bitdepth() != int(header.datatype().bits()) ||
((header.ndim() > 3 && png.get_channels() != header.size(3)) || (header.ndim() <= 3 && png.get_channels() > 1))) {
Exception e ("Inconsistent image properties within series \"" + header.name() + "\"");
e.push_back ("Series: " + str(header.size(0)) + "x" + str(header.size(1)) + " x " + str(header.datatype().bits()) + " bits, " + (header.ndim() > 3 ? str(header.size(3)) : "1") + " volumes");
e.push_back ("File \"" + files[i].name + ": " + str(png.get_width()) + "x" + str(png.get_height()) + " x " + str(png.get_bitdepth()) + "(->" + str(png.get_output_bitdepth()) + ") bits, " + str(png.get_channels()) + " channels");
throw e;
}
png.load (addresses[0].get() + (i * slice_bytes));
}
}
}
void PNG::unload (const Header& header)
{
if (addresses.size()) {
if (writable) {
size_t slice_bytes = (header.datatype().bits() * header.size(0) * header.size(1) + 7) / 8;
if (header.ndim() == 4)
slice_bytes *= header.size (3);
for (size_t i = 0; i != files.size(); i++) {
File::PNG::Writer png (header, files[i].name);
png.save (addresses[0].get() + (i * slice_bytes));
}
}
DEBUG ("deleting buffer for PNG image \"" + header.name() + "\"...");
addresses[0].release();
}
}
}
}
#endif
|
