comparison UnitTestsSources/ImageTests.cpp @ 1602:292bce3f54ed

JpegWriter
author Sebastien Jodogne <s.jodogne@gmail.com>
date Tue, 01 Sep 2015 13:06:39 +0200
parents UnitTestsSources/PngTests.cpp@9f66a12eb8fc
children 1f5d6a2f9638
comparison
equal deleted inserted replaced
1601:6cccf1da35c6 1602:292bce3f54ed
1 /**
2 * Orthanc - A Lightweight, RESTful DICOM Store
3 * Copyright (C) 2012-2015 Sebastien Jodogne, Medical Physics
4 * Department, University Hospital of Liege, Belgium
5 *
6 * This program is free software: you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License as
8 * published by the Free Software Foundation, either version 3 of the
9 * License, or (at your option) any later version.
10 *
11 * In addition, as a special exception, the copyright holders of this
12 * program give permission to link the code of its release with the
13 * OpenSSL project's "OpenSSL" library (or with modified versions of it
14 * that use the same license as the "OpenSSL" library), and distribute
15 * the linked executables. You must obey the GNU General Public License
16 * in all respects for all of the code used other than "OpenSSL". If you
17 * modify file(s) with this exception, you may extend this exception to
18 * your version of the file(s), but you are not obligated to do so. If
19 * you do not wish to do so, delete this exception statement from your
20 * version. If you delete this exception statement from all source files
21 * in the program, then also delete it here.
22 *
23 * This program is distributed in the hope that it will be useful, but
24 * WITHOUT ANY WARRANTY; without even the implied warranty of
25 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
26 * General Public License for more details.
27 *
28 * You should have received a copy of the GNU General Public License
29 * along with this program. If not, see <http://www.gnu.org/licenses/>.
30 **/
31
32
33 #include "PrecompiledHeadersUnitTests.h"
34 #include "gtest/gtest.h"
35
36 #include <stdint.h>
37 #include "../Core/ImageFormats/ImageBuffer.h"
38 #include "../Core/ImageFormats/PngReader.h"
39 #include "../Core/ImageFormats/PngWriter.h"
40 #include "../Core/ImageFormats/JpegWriter.h"
41 #include "../Core/Toolbox.h"
42 #include "../Core/Uuid.h"
43
44
45 TEST(PngWriter, ColorPattern)
46 {
47 Orthanc::PngWriter w;
48 int width = 17;
49 int height = 61;
50 int pitch = width * 3;
51
52 std::vector<uint8_t> image(height * pitch);
53 for (int y = 0; y < height; y++)
54 {
55 uint8_t *p = &image[0] + y * pitch;
56 for (int x = 0; x < width; x++, p += 3)
57 {
58 p[0] = (y % 3 == 0) ? 255 : 0;
59 p[1] = (y % 3 == 1) ? 255 : 0;
60 p[2] = (y % 3 == 2) ? 255 : 0;
61 }
62 }
63
64 w.WriteToFile("UnitTestsResults/ColorPattern.png", width, height, pitch, Orthanc::PixelFormat_RGB24, &image[0]);
65
66 std::string f, md5;
67 Orthanc::Toolbox::ReadFile(f, "UnitTestsResults/ColorPattern.png");
68 Orthanc::Toolbox::ComputeMD5(md5, f);
69 ASSERT_EQ("604e785f53c99cae6ea4584870b2c41d", md5);
70 }
71
72 TEST(PngWriter, Gray8Pattern)
73 {
74 Orthanc::PngWriter w;
75 int width = 17;
76 int height = 256;
77 int pitch = width;
78
79 std::vector<uint8_t> image(height * pitch);
80 for (int y = 0; y < height; y++)
81 {
82 uint8_t *p = &image[0] + y * pitch;
83 for (int x = 0; x < width; x++, p++)
84 {
85 *p = y;
86 }
87 }
88
89 w.WriteToFile("UnitTestsResults/Gray8Pattern.png", width, height, pitch, Orthanc::PixelFormat_Grayscale8, &image[0]);
90
91 std::string f, md5;
92 Orthanc::Toolbox::ReadFile(f, "UnitTestsResults/Gray8Pattern.png");
93 Orthanc::Toolbox::ComputeMD5(md5, f);
94 ASSERT_EQ("5a9b98bea3d0a6d983980cc38bfbcdb3", md5);
95 }
96
97 TEST(PngWriter, Gray16Pattern)
98 {
99 Orthanc::PngWriter w;
100 int width = 256;
101 int height = 256;
102 int pitch = width * 2 + 16;
103
104 std::vector<uint8_t> image(height * pitch);
105
106 int v = 0;
107 for (int y = 0; y < height; y++)
108 {
109 uint16_t *p = reinterpret_cast<uint16_t*>(&image[0] + y * pitch);
110 for (int x = 0; x < width; x++, p++, v++)
111 {
112 *p = v;
113 }
114 }
115
116 w.WriteToFile("UnitTestsResults/Gray16Pattern.png", width, height, pitch, Orthanc::PixelFormat_Grayscale16, &image[0]);
117
118 std::string f, md5;
119 Orthanc::Toolbox::ReadFile(f, "UnitTestsResults/Gray16Pattern.png");
120 Orthanc::Toolbox::ComputeMD5(md5, f);
121 ASSERT_EQ("0785866a08bf0a02d2eeff87f658571c", md5);
122 }
123
124 TEST(PngWriter, EndToEnd)
125 {
126 Orthanc::PngWriter w;
127 unsigned int width = 256;
128 unsigned int height = 256;
129 unsigned int pitch = width * 2 + 16;
130
131 std::vector<uint8_t> image(height * pitch);
132
133 int v = 0;
134 for (unsigned int y = 0; y < height; y++)
135 {
136 uint16_t *p = reinterpret_cast<uint16_t*>(&image[0] + y * pitch);
137 for (unsigned int x = 0; x < width; x++, p++, v++)
138 {
139 *p = v;
140 }
141 }
142
143 std::string s;
144 w.WriteToMemory(s, width, height, pitch, Orthanc::PixelFormat_Grayscale16, &image[0]);
145
146 {
147 Orthanc::PngReader r;
148 r.ReadFromMemory(s);
149
150 ASSERT_EQ(r.GetFormat(), Orthanc::PixelFormat_Grayscale16);
151 ASSERT_EQ(r.GetWidth(), width);
152 ASSERT_EQ(r.GetHeight(), height);
153
154 v = 0;
155 for (unsigned int y = 0; y < height; y++)
156 {
157 const uint16_t *p = reinterpret_cast<const uint16_t*>((const uint8_t*) r.GetConstBuffer() + y * r.GetPitch());
158 ASSERT_EQ(p, r.GetConstRow(y));
159 for (unsigned int x = 0; x < width; x++, p++, v++)
160 {
161 ASSERT_EQ(*p, v);
162 }
163 }
164 }
165
166 {
167 Orthanc::Toolbox::TemporaryFile tmp;
168 Orthanc::Toolbox::WriteFile(s, tmp.GetPath());
169
170 Orthanc::PngReader r2;
171 r2.ReadFromFile(tmp.GetPath());
172
173 ASSERT_EQ(r2.GetFormat(), Orthanc::PixelFormat_Grayscale16);
174 ASSERT_EQ(r2.GetWidth(), width);
175 ASSERT_EQ(r2.GetHeight(), height);
176
177 v = 0;
178 for (unsigned int y = 0; y < height; y++)
179 {
180 const uint16_t *p = reinterpret_cast<const uint16_t*>((const uint8_t*) r2.GetConstBuffer() + y * r2.GetPitch());
181 ASSERT_EQ(p, r2.GetConstRow(y));
182 for (unsigned int x = 0; x < width; x++, p++, v++)
183 {
184 ASSERT_EQ(*p, v);
185 }
186 }
187 }
188 }
189
190
191
192
193 TEST(JpegWriter, Basic)
194 {
195 Orthanc::ImageBuffer img(16, 16, Orthanc::PixelFormat_Grayscale8);
196 Orthanc::ImageAccessor accessor = img.GetAccessor();
197 for (unsigned int y = 0, value = 0; y < img.GetHeight(); y++)
198 {
199 uint8_t* p = reinterpret_cast<uint8_t*>(accessor.GetRow(y));
200 for (unsigned int x = 0; x < img.GetWidth(); x++, p++)
201 {
202 *p = value++;
203 }
204 }
205
206 Orthanc::JpegWriter w;
207 w.WriteToFile("UnitTestsResults/hello.jpg", accessor);
208
209 std::string s;
210 w.WriteToMemory(s, accessor);
211 Orthanc::Toolbox::WriteFile(s, "UnitTestsResults/hello2.jpg");
212 }