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