Mercurial > hg > orthanc-webviewer
annotate Plugin/ViewerToolbox.cpp @ 76:07d2e87e450b
improvement
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 25 Sep 2015 11:10:23 +0200 |
parents | d7bd116777eb |
children | abdde1dfb3eb |
rev | line source |
---|---|
0 | 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 Affero General Public License | |
8 * as published by the Free Software Foundation, either version 3 of | |
9 * the License, or (at your option) any later version. | |
10 * | |
11 * This program is distributed in the hope that it will be useful, but | |
12 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 * Affero General Public License for more details. | |
15 * | |
16 * You should have received a copy of the GNU Affero General Public License | |
17 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
18 **/ | |
19 | |
20 | |
21 #include "ViewerToolbox.h" | |
22 | |
26 | 23 #include "../Orthanc/Core/OrthancException.h" |
24 #include "../Orthanc/Core/Toolbox.h" | |
0 | 25 |
26 #include <json/reader.h> | |
27 #include <zlib.h> | |
28 #include <stdexcept> | |
29 #include <boost/lexical_cast.hpp> | |
30 #include <sys/stat.h> | |
31 | |
32 namespace OrthancPlugins | |
33 { | |
34 bool GetStringFromOrthanc(std::string& content, | |
35 OrthancPluginContext* context, | |
36 const std::string& uri) | |
37 { | |
38 OrthancPluginMemoryBuffer answer; | |
39 | |
40 if (OrthancPluginRestApiGet(context, &answer, uri.c_str())) | |
41 { | |
42 return false; | |
43 } | |
44 | |
45 if (answer.size) | |
46 { | |
47 try | |
48 { | |
49 content.assign(reinterpret_cast<const char*>(answer.data), answer.size); | |
50 } | |
51 catch (std::bad_alloc&) | |
52 { | |
53 OrthancPluginFreeMemoryBuffer(context, &answer); | |
54 throw Orthanc::OrthancException("Not enough memory"); | |
55 } | |
56 } | |
57 | |
58 OrthancPluginFreeMemoryBuffer(context, &answer); | |
59 return true; | |
60 } | |
61 | |
62 | |
63 bool GetJsonFromOrthanc(Json::Value& json, | |
64 OrthancPluginContext* context, | |
65 const std::string& uri) | |
66 { | |
67 OrthancPluginMemoryBuffer answer; | |
68 | |
69 if (OrthancPluginRestApiGet(context, &answer, uri.c_str())) | |
70 { | |
71 return false; | |
72 } | |
73 | |
74 if (answer.size) | |
75 { | |
76 try | |
77 { | |
78 const char* data = reinterpret_cast<const char*>(answer.data); | |
79 Json::Reader reader; | |
80 if (!reader.parse(data, data + answer.size, json, | |
81 false /* don't collect comments */)) | |
82 { | |
83 return false; | |
84 } | |
85 } | |
86 catch (std::runtime_error&) | |
87 { | |
88 OrthancPluginFreeMemoryBuffer(context, &answer); | |
89 return false; | |
90 } | |
91 } | |
92 | |
93 OrthancPluginFreeMemoryBuffer(context, &answer); | |
94 return true; | |
95 } | |
96 | |
97 | |
98 | |
99 | |
100 bool TokenizeVector(std::vector<float>& result, | |
101 const std::string& value, | |
102 unsigned int expectedSize) | |
103 { | |
104 std::vector<std::string> tokens; | |
105 Orthanc::Toolbox::TokenizeString(tokens, value, '\\'); | |
106 | |
107 if (tokens.size() != expectedSize) | |
108 { | |
109 return false; | |
110 } | |
111 | |
112 result.resize(tokens.size()); | |
113 | |
114 for (size_t i = 0; i < tokens.size(); i++) | |
115 { | |
116 try | |
117 { | |
118 result[i] = boost::lexical_cast<float>(tokens[i]); | |
119 } | |
120 catch (boost::bad_lexical_cast&) | |
121 { | |
122 return false; | |
123 } | |
124 } | |
125 | |
126 return true; | |
127 } | |
128 | |
129 | |
130 bool CompressUsingDeflate(std::string& compressed, | |
131 const void* uncompressed, | |
132 size_t uncompressedSize, | |
133 uint8_t compressionLevel) | |
134 { | |
135 if (uncompressedSize == 0) | |
136 { | |
137 compressed.clear(); | |
138 return true; | |
139 } | |
140 | |
141 uLongf compressedSize = compressBound(uncompressedSize); | |
142 compressed.resize(compressedSize); | |
143 | |
144 int error = compress2 | |
145 (reinterpret_cast<uint8_t*>(&compressed[0]), | |
146 &compressedSize, | |
147 const_cast<Bytef *>(static_cast<const Bytef *>(uncompressed)), | |
148 uncompressedSize, | |
149 compressionLevel); | |
150 | |
151 if (error == Z_OK) | |
152 { | |
153 compressed.resize(compressedSize); | |
154 return true; | |
155 } | |
156 else | |
157 { | |
158 compressed.clear(); | |
159 return false; | |
160 } | |
161 } | |
162 | |
163 | |
164 | |
165 const char* GetMimeType(const std::string& path) | |
166 { | |
167 size_t dot = path.find_last_of('.'); | |
168 | |
169 std::string extension = (dot == std::string::npos) ? "" : path.substr(dot); | |
170 std::transform(extension.begin(), extension.end(), extension.begin(), tolower); | |
171 | |
172 if (extension == ".html") | |
173 { | |
174 return "text/html"; | |
175 } | |
176 else if (extension == ".css") | |
177 { | |
178 return "text/css"; | |
179 } | |
180 else if (extension == ".js") | |
181 { | |
182 return "application/javascript"; | |
183 } | |
184 else if (extension == ".gif") | |
185 { | |
186 return "image/gif"; | |
187 } | |
188 else if (extension == ".svg") | |
189 { | |
190 return "image/svg+xml"; | |
191 } | |
192 else if (extension == ".json") | |
193 { | |
194 return "application/json"; | |
195 } | |
196 else if (extension == ".xml") | |
197 { | |
198 return "application/xml"; | |
199 } | |
200 else if (extension == ".png") | |
201 { | |
202 return "image/png"; | |
203 } | |
204 else if (extension == ".jpg" || extension == ".jpeg") | |
205 { | |
206 return "image/jpeg"; | |
207 } | |
208 else | |
209 { | |
210 return "application/octet-stream"; | |
211 } | |
212 } | |
213 | |
214 | |
215 bool ReadConfiguration(Json::Value& configuration, | |
216 OrthancPluginContext* context) | |
217 { | |
33
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
218 std::string s; |
0 | 219 |
220 { | |
33
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
221 char* tmp = OrthancPluginGetConfiguration(context); |
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
222 if (tmp == NULL) |
0 | 223 { |
33
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
224 OrthancPluginLogError(context, "Error while retrieving the configuration from Orthanc"); |
0 | 225 return false; |
226 } | |
227 | |
33
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
228 s.assign(tmp); |
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
229 OrthancPluginFreeString(context, tmp); |
0 | 230 } |
231 | |
232 Json::Reader reader; | |
33
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
233 if (reader.parse(s, configuration)) |
0 | 234 { |
33
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
235 return true; |
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
236 } |
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
237 else |
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
238 { |
d7bd116777eb
Use of OrthancPluginGetConfiguration
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
26
diff
changeset
|
239 OrthancPluginLogError(context, "Unable to parse the configuration"); |
0 | 240 return false; |
241 } | |
242 } | |
243 | |
244 | |
245 std::string GetStringValue(const Json::Value& configuration, | |
246 const std::string& key, | |
247 const std::string& defaultValue) | |
248 { | |
249 if (configuration.type() != Json::objectValue || | |
250 !configuration.isMember(key) || | |
251 configuration[key].type() != Json::stringValue) | |
252 { | |
253 return defaultValue; | |
254 } | |
255 else | |
256 { | |
257 return configuration[key].asString(); | |
258 } | |
259 } | |
260 | |
261 | |
262 int GetIntegerValue(const Json::Value& configuration, | |
263 const std::string& key, | |
264 int defaultValue) | |
265 { | |
266 if (configuration.type() != Json::objectValue || | |
267 !configuration.isMember(key) || | |
268 configuration[key].type() != Json::intValue) | |
269 { | |
270 return defaultValue; | |
271 } | |
272 else | |
273 { | |
274 return configuration[key].asInt(); | |
275 } | |
276 } | |
277 | |
278 | |
279 bool ReadFile(std::string& content, | |
280 const std::string& path) | |
281 { | |
282 struct stat s; | |
283 if (stat(path.c_str(), &s) != 0 || | |
284 !(s.st_mode & S_IFREG)) | |
285 { | |
286 // Either the path does not exist, or it is not a regular file | |
287 return false; | |
288 } | |
289 | |
290 FILE* fp = fopen(path.c_str(), "rb"); | |
291 if (fp == NULL) | |
292 { | |
293 return false; | |
294 } | |
295 | |
296 long size; | |
297 | |
298 if (fseek(fp, 0, SEEK_END) == -1 || | |
299 (size = ftell(fp)) < 0) | |
300 { | |
301 fclose(fp); | |
302 return false; | |
303 } | |
304 | |
305 content.resize(size); | |
306 | |
307 if (fseek(fp, 0, SEEK_SET) == -1) | |
308 { | |
309 fclose(fp); | |
310 return false; | |
311 } | |
312 | |
313 bool ok = true; | |
314 | |
315 if (size > 0 && | |
316 fread(&content[0], size, 1, fp) != 1) | |
317 { | |
318 ok = false; | |
319 } | |
320 | |
321 fclose(fp); | |
322 | |
323 return ok; | |
324 } | |
325 } |