Mercurial > hg > orthanc-stone
annotate Framework/Messaging/MessagingToolbox.cpp @ 33:12987d11be33
fix
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Mon, 19 Dec 2016 11:29:23 +0100 |
parents | 9aace933cb64 |
children | a865c7992a87 |
rev | line source |
---|---|
0 | 1 /** |
2 * Stone of Orthanc | |
3 * Copyright (C) 2012-2016 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 "MessagingToolbox.h" | |
34 | |
16 | 35 #include "../../Resources/Orthanc/Core/Images/Image.h" |
36 #include "../../Resources/Orthanc/Core/Images/ImageProcessing.h" | |
37 #include "../../Resources/Orthanc/Core/Images/JpegReader.h" | |
38 #include "../../Resources/Orthanc/Core/Images/PngReader.h" | |
39 #include "../../Resources/Orthanc/Core/OrthancException.h" | |
40 #include "../../Resources/Orthanc/Core/Toolbox.h" | |
41 #include "../../Resources/Orthanc/Core/Logging.h" | |
0 | 42 |
43 #include <boost/lexical_cast.hpp> | |
44 #include <json/reader.h> | |
45 | |
46 #if defined(__native_client__) | |
47 # include <boost/math/special_functions/round.hpp> | |
48 #else | |
22
b01d46e5a2b3
support for sandboxed environments
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
16
diff
changeset
|
49 # include <boost/date_time/posix_time/posix_time.hpp> |
0 | 50 # include <boost/date_time/microsec_time_clock.hpp> |
51 #endif | |
52 | |
53 namespace OrthancStone | |
54 { | |
55 namespace MessagingToolbox | |
56 { | |
57 #if defined(__native_client__) | |
58 static pp::Core* core_ = NULL; | |
59 | |
60 void Timestamp::Initialize(pp::Core* core) | |
61 { | |
62 if (core == NULL) | |
63 { | |
64 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
65 } | |
66 | |
67 core_ = core; | |
68 } | |
69 | |
70 Timestamp::Timestamp() | |
71 { | |
72 if (core_ == NULL) | |
73 { | |
74 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
75 } | |
76 | |
77 time_ = core_->GetTimeTicks(); | |
78 } | |
79 | |
80 int Timestamp::GetMillisecondsSince(const Timestamp& other) | |
81 { | |
82 double difference = time_ - other.time_; | |
83 return static_cast<int>(boost::math::iround(difference * 1000.0)); | |
84 } | |
85 #else | |
86 Timestamp::Timestamp() | |
87 { | |
88 time_ = boost::posix_time::microsec_clock::local_time(); | |
89 } | |
90 | |
91 int Timestamp::GetMillisecondsSince(const Timestamp& other) | |
92 { | |
93 boost::posix_time::time_duration difference = time_ - other.time_; | |
94 return static_cast<int>(difference.total_milliseconds()); | |
95 } | |
96 #endif | |
97 | |
98 static bool ParseVersion(std::string& version, | |
99 unsigned int& major, | |
100 unsigned int& minor, | |
101 unsigned int& patch, | |
102 const Json::Value& info) | |
103 { | |
104 if (info.type() != Json::objectValue || | |
105 !info.isMember("Version") || | |
106 info["Version"].type() != Json::stringValue) | |
107 { | |
108 return false; | |
109 } | |
110 | |
111 version = info["Version"].asString(); | |
112 if (version == "mainline") | |
113 { | |
114 // Some arbitrary high values Orthanc versions will never reach ;) | |
115 major = 999; | |
116 minor = 999; | |
117 patch = 999; | |
118 return true; | |
119 } | |
120 | |
121 std::vector<std::string> tokens; | |
122 Orthanc::Toolbox::TokenizeString(tokens, version, '.'); | |
123 | |
124 if (tokens.size() != 2 && | |
125 tokens.size() != 3) | |
126 { | |
127 return false; | |
128 } | |
129 | |
130 int a, b, c; | |
131 try | |
132 { | |
133 a = boost::lexical_cast<int>(tokens[0]); | |
134 b = boost::lexical_cast<int>(tokens[1]); | |
135 | |
136 if (tokens.size() == 3) | |
137 { | |
138 c = boost::lexical_cast<int>(tokens[2]); | |
139 } | |
140 else | |
141 { | |
142 c = 0; | |
143 } | |
144 } | |
145 catch (boost::bad_lexical_cast&) | |
146 { | |
147 return false; | |
148 } | |
149 | |
150 if (a < 0 || | |
151 b < 0 || | |
152 c < 0) | |
153 { | |
154 return false; | |
155 } | |
156 else | |
157 { | |
158 major = static_cast<unsigned int>(a); | |
159 minor = static_cast<unsigned int>(b); | |
160 patch = static_cast<unsigned int>(c); | |
161 return true; | |
162 } | |
163 } | |
164 | |
165 void ParseJson(Json::Value& target, | |
166 const std::string& source) | |
167 { | |
168 Json::Reader reader; | |
169 if (!reader.parse(source, target)) | |
170 { | |
171 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadFileFormat); | |
172 } | |
173 } | |
174 | |
175 void RestApiGet(Json::Value& target, | |
31
9aace933cb64
sharing code with the Orthanc core
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
22
diff
changeset
|
176 OrthancPlugins::IOrthancConnection& orthanc, |
0 | 177 const std::string& uri) |
178 { | |
179 std::string tmp; | |
180 orthanc.RestApiGet(tmp, uri); | |
181 ParseJson(target, tmp); | |
182 } | |
183 | |
184 | |
31
9aace933cb64
sharing code with the Orthanc core
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
22
diff
changeset
|
185 bool HasWebViewerInstalled(OrthancPlugins::IOrthancConnection& orthanc) |
0 | 186 { |
187 try | |
188 { | |
189 Json::Value json; | |
190 RestApiGet(json, orthanc, "/plugins/web-viewer"); | |
191 return json.type() == Json::objectValue; | |
192 } | |
193 catch (Orthanc::OrthancException&) | |
194 { | |
195 return false; | |
196 } | |
197 } | |
198 | |
199 | |
31
9aace933cb64
sharing code with the Orthanc core
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
22
diff
changeset
|
200 bool CheckOrthancVersion(OrthancPlugins::IOrthancConnection& orthanc) |
0 | 201 { |
202 Json::Value json; | |
203 std::string version; | |
204 unsigned int major, minor, patch; | |
205 | |
206 try | |
207 { | |
208 RestApiGet(json, orthanc, "/system"); | |
209 } | |
210 catch (Orthanc::OrthancException&) | |
211 { | |
212 LOG(ERROR) << "Cannot connect to your Orthanc server"; | |
213 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
214 } | |
215 | |
216 if (!ParseVersion(version, major, minor, patch, json)) | |
217 { | |
218 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
219 } | |
220 | |
221 LOG(WARNING) << "Version of the Orthanc core (must be above 1.1.0): " << version; | |
222 | |
223 // Stone is only compatible with Orthanc >= 1.1.0, otherwise deadlocks might occur | |
224 if (major < 1 || | |
225 (major == 1 && minor < 1)) | |
226 { | |
227 return false; | |
228 } | |
229 | |
230 try | |
231 { | |
232 RestApiGet(json, orthanc, "/plugins/web-viewer"); | |
233 } | |
234 catch (Orthanc::OrthancException&) | |
235 { | |
236 // The Web viewer is not installed, this is OK | |
237 LOG(WARNING) << "The Web viewer plugin is not installed, progressive download is disabled"; | |
238 return true; | |
239 } | |
240 | |
241 if (!ParseVersion(version, major, minor, patch, json)) | |
242 { | |
243 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
244 } | |
245 | |
246 LOG(WARNING) << "Version of the Web viewer plugin (must be above 2.2): " << version; | |
247 | |
248 return (major >= 3 || | |
249 (major == 2 && minor >= 2)); | |
250 } | |
251 | |
252 | |
31
9aace933cb64
sharing code with the Orthanc core
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
22
diff
changeset
|
253 Orthanc::ImageAccessor* DecodeFrame(OrthancPlugins::IOrthancConnection& orthanc, |
0 | 254 const std::string& instance, |
255 unsigned int frame, | |
256 Orthanc::PixelFormat targetFormat) | |
257 { | |
258 std::string uri = ("instances/" + instance + "/frames/" + | |
259 boost::lexical_cast<std::string>(frame)); | |
260 | |
261 std::string compressed; | |
262 | |
263 switch (targetFormat) | |
264 { | |
265 case Orthanc::PixelFormat_RGB24: | |
266 orthanc.RestApiGet(compressed, uri + "/preview"); | |
267 break; | |
268 | |
269 case Orthanc::PixelFormat_Grayscale16: | |
270 orthanc.RestApiGet(compressed, uri + "/image-uint16"); | |
271 break; | |
272 | |
273 case Orthanc::PixelFormat_SignedGrayscale16: | |
274 orthanc.RestApiGet(compressed, uri + "/image-int16"); | |
275 break; | |
276 | |
277 default: | |
278 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
279 } | |
280 | |
281 std::auto_ptr<Orthanc::PngReader> result(new Orthanc::PngReader); | |
282 result->ReadFromMemory(compressed); | |
283 | |
284 if (targetFormat == Orthanc::PixelFormat_SignedGrayscale16) | |
285 { | |
286 if (result->GetFormat() == Orthanc::PixelFormat_Grayscale16) | |
287 { | |
288 result->SetFormat(Orthanc::PixelFormat_SignedGrayscale16); | |
289 } | |
290 else | |
291 { | |
292 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
293 } | |
294 } | |
295 | |
296 return result.release(); | |
297 } | |
298 | |
299 | |
31
9aace933cb64
sharing code with the Orthanc core
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
22
diff
changeset
|
300 Orthanc::ImageAccessor* DecodeJpegFrame(OrthancPlugins::IOrthancConnection& orthanc, |
0 | 301 const std::string& instance, |
302 unsigned int frame, | |
303 unsigned int quality, | |
304 Orthanc::PixelFormat targetFormat) | |
305 { | |
306 if (quality <= 0 || | |
307 quality > 100) | |
308 { | |
309 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
310 } | |
311 | |
312 // This requires the official Web viewer plugin to be installed! | |
313 std::string uri = ("web-viewer/instances/jpeg" + | |
314 boost::lexical_cast<std::string>(quality) + | |
315 "-" + instance + "_" + | |
316 boost::lexical_cast<std::string>(frame)); | |
317 | |
318 Json::Value encoded; | |
319 RestApiGet(encoded, orthanc, uri); | |
320 | |
321 if (encoded.type() != Json::objectValue || | |
322 !encoded.isMember("Orthanc") || | |
323 encoded["Orthanc"].type() != Json::objectValue) | |
324 { | |
325 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
326 } | |
327 | |
328 Json::Value& info = encoded["Orthanc"]; | |
329 if (!info.isMember("PixelData") || | |
330 !info.isMember("Stretched") || | |
331 !info.isMember("Compression") || | |
332 info["Compression"].type() != Json::stringValue || | |
333 info["PixelData"].type() != Json::stringValue || | |
334 info["Stretched"].type() != Json::booleanValue || | |
335 info["Compression"].asString() != "Jpeg") | |
336 { | |
337 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
338 } | |
339 | |
340 bool isSigned = false; | |
341 bool isStretched = info["Stretched"].asBool(); | |
342 | |
343 if (info.isMember("IsSigned")) | |
344 { | |
345 if (info["IsSigned"].type() != Json::booleanValue) | |
346 { | |
347 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
348 } | |
349 else | |
350 { | |
351 isSigned = info["IsSigned"].asBool(); | |
352 } | |
353 } | |
354 | |
355 std::string jpeg; | |
356 Orthanc::Toolbox::DecodeBase64(jpeg, info["PixelData"].asString()); | |
357 | |
358 std::auto_ptr<Orthanc::JpegReader> reader(new Orthanc::JpegReader); | |
359 reader->ReadFromMemory(jpeg); | |
360 | |
361 if (reader->GetFormat() == Orthanc::PixelFormat_RGB24) // This is a color image | |
362 { | |
363 if (targetFormat != Orthanc::PixelFormat_RGB24) | |
364 { | |
365 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
366 } | |
367 | |
368 if (isSigned || isStretched) | |
369 { | |
370 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); | |
371 } | |
372 else | |
373 { | |
374 return reader.release(); | |
375 } | |
376 } | |
377 | |
378 if (reader->GetFormat() != Orthanc::PixelFormat_Grayscale8) | |
379 { | |
380 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); | |
381 } | |
382 | |
383 if (!isStretched) | |
384 { | |
385 if (targetFormat != reader->GetFormat()) | |
386 { | |
387 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
388 } | |
389 | |
390 return reader.release(); | |
391 } | |
392 | |
393 int32_t stretchLow = 0; | |
394 int32_t stretchHigh = 0; | |
395 | |
396 if (!info.isMember("StretchLow") || | |
397 !info.isMember("StretchHigh") || | |
398 info["StretchLow"].type() != Json::intValue || | |
399 info["StretchHigh"].type() != Json::intValue) | |
400 { | |
401 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
402 } | |
403 | |
404 stretchLow = info["StretchLow"].asInt(); | |
405 stretchHigh = info["StretchHigh"].asInt(); | |
406 | |
407 if (stretchLow < -32768 || | |
408 stretchHigh > 65535 || | |
409 (stretchLow < 0 && stretchHigh > 32767)) | |
410 { | |
411 // This range cannot be represented with a uint16_t or an int16_t | |
412 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); | |
413 } | |
414 | |
415 // Decode a grayscale JPEG 8bpp image coming from the Web viewer | |
416 std::auto_ptr<Orthanc::ImageAccessor> image | |
11 | 417 (new Orthanc::Image(targetFormat, reader->GetWidth(), reader->GetHeight(), false)); |
0 | 418 |
419 float scaling = static_cast<float>(stretchHigh - stretchLow) / 255.0f; | |
420 float offset = static_cast<float>(stretchLow) / scaling; | |
421 | |
422 Orthanc::ImageProcessing::Convert(*image, *reader); | |
423 Orthanc::ImageProcessing::ShiftScale(*image, offset, scaling); | |
424 | |
425 #if 0 | |
426 /*info.removeMember("PixelData"); | |
427 std::cout << info.toStyledString();*/ | |
428 | |
429 int64_t a, b; | |
430 Orthanc::ImageProcessing::GetMinMaxValue(a, b, *image); | |
431 std::cout << stretchLow << "->" << stretchHigh << " = " << a << "->" << b << std::endl; | |
432 #endif | |
433 | |
434 return image.release(); | |
435 } | |
436 } | |
437 } |