Mercurial > hg > orthanc-stone
annotate Framework/Loaders/OrthancMultiframeVolumeLoader.cpp @ 937:86ac61a040c9
Added getters and notifications to allow clients of the loaders (DicomStructureSetLoader, OrthancSeriesVolumeProgressiveLoader and OrthancMultiframeVolumeLoader) to know when the loading is finished + added ability for SDL event loop to execute a callback repeatedly (used to check the view loading state)
author | Benjamin Golinvaux <bgo@osimis.io> |
---|---|
date | Wed, 31 Jul 2019 10:24:09 +0200 |
parents | 401808e7ff2e |
children | 32eaf4929b08 |
rev | line source |
---|---|
815 | 1 /** |
2 * Stone of Orthanc | |
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics | |
4 * Department, University Hospital of Liege, Belgium | |
5 * Copyright (C) 2017-2019 Osimis S.A., Belgium | |
6 * | |
7 * This program is free software: you can redistribute it and/or | |
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. | |
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 | |
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 | |
18 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
19 **/ | |
20 | |
21 | |
22 #include "OrthancMultiframeVolumeLoader.h" | |
23 | |
816 | 24 #include <Core/Endianness.h> |
815 | 25 #include <Core/Toolbox.h> |
26 | |
27 namespace OrthancStone | |
28 { | |
29 class OrthancMultiframeVolumeLoader::LoadRTDoseGeometry : public LoaderStateMachine::State | |
30 { | |
31 private: | |
32 std::auto_ptr<Orthanc::DicomMap> dicom_; | |
33 | |
34 public: | |
35 LoadRTDoseGeometry(OrthancMultiframeVolumeLoader& that, | |
36 Orthanc::DicomMap* dicom) : | |
37 State(that), | |
38 dicom_(dicom) | |
39 { | |
40 if (dicom == NULL) | |
41 { | |
42 throw Orthanc::OrthancException(Orthanc::ErrorCode_NullPointer); | |
43 } | |
44 | |
45 } | |
46 | |
47 virtual void Handle(const OrthancRestApiCommand::SuccessMessage& message) | |
48 { | |
49 // Complete the DICOM tags with just-received "Grid Frame Offset Vector" | |
50 std::string s = Orthanc::Toolbox::StripSpaces(message.GetAnswer()); | |
51 dicom_->SetValue(Orthanc::DICOM_TAG_GRID_FRAME_OFFSET_VECTOR, s, false); | |
52 | |
53 GetLoader<OrthancMultiframeVolumeLoader>().SetGeometry(*dicom_); | |
54 } | |
55 }; | |
56 | |
57 | |
58 static std::string GetSopClassUid(const Orthanc::DicomMap& dicom) | |
59 { | |
60 std::string s; | |
61 if (!dicom.CopyToString(s, Orthanc::DICOM_TAG_SOP_CLASS_UID, false)) | |
62 { | |
63 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadFileFormat, | |
64 "DICOM file without SOP class UID"); | |
65 } | |
66 else | |
67 { | |
68 return s; | |
69 } | |
70 } | |
71 | |
72 | |
73 class OrthancMultiframeVolumeLoader::LoadGeometry : public State | |
74 { | |
75 public: | |
76 LoadGeometry(OrthancMultiframeVolumeLoader& that) : | |
77 State(that) | |
78 { | |
79 } | |
80 | |
81 virtual void Handle(const OrthancRestApiCommand::SuccessMessage& message) | |
82 { | |
83 OrthancMultiframeVolumeLoader& loader = GetLoader<OrthancMultiframeVolumeLoader>(); | |
84 | |
85 Json::Value body; | |
86 message.ParseJsonBody(body); | |
87 | |
88 if (body.type() != Json::objectValue) | |
89 { | |
90 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
91 } | |
92 | |
93 std::auto_ptr<Orthanc::DicomMap> dicom(new Orthanc::DicomMap); | |
94 dicom->FromDicomAsJson(body); | |
95 | |
96 if (StringToSopClassUid(GetSopClassUid(*dicom)) == SopClassUid_RTDose) | |
97 { | |
98 // Download the "Grid Frame Offset Vector" DICOM tag, that is | |
99 // mandatory for RT-DOSE, but is too long to be returned by default | |
100 | |
101 std::auto_ptr<OrthancRestApiCommand> command(new OrthancRestApiCommand); | |
102 command->SetUri("/instances/" + loader.GetInstanceId() + "/content/" + | |
103 Orthanc::DICOM_TAG_GRID_FRAME_OFFSET_VECTOR.Format()); | |
104 command->SetPayload(new LoadRTDoseGeometry(loader, dicom.release())); | |
105 | |
106 Schedule(command.release()); | |
107 } | |
108 else | |
109 { | |
110 loader.SetGeometry(*dicom); | |
111 } | |
112 } | |
113 }; | |
114 | |
115 | |
116 | |
117 class OrthancMultiframeVolumeLoader::LoadTransferSyntax : public State | |
118 { | |
119 public: | |
120 LoadTransferSyntax(OrthancMultiframeVolumeLoader& that) : | |
121 State(that) | |
122 { | |
123 } | |
124 | |
125 virtual void Handle(const OrthancRestApiCommand::SuccessMessage& message) | |
126 { | |
127 GetLoader<OrthancMultiframeVolumeLoader>().SetTransferSyntax(message.GetAnswer()); | |
128 } | |
129 }; | |
925
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
130 |
815 | 131 class OrthancMultiframeVolumeLoader::LoadUncompressedPixelData : public State |
132 { | |
133 public: | |
134 LoadUncompressedPixelData(OrthancMultiframeVolumeLoader& that) : | |
135 State(that) | |
136 { | |
137 } | |
138 | |
139 virtual void Handle(const OrthancRestApiCommand::SuccessMessage& message) | |
140 { | |
141 GetLoader<OrthancMultiframeVolumeLoader>().SetUncompressedPixelData(message.GetAnswer()); | |
142 } | |
143 }; | |
925
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
144 |
815 | 145 const std::string& OrthancMultiframeVolumeLoader::GetInstanceId() const |
146 { | |
147 if (IsActive()) | |
148 { | |
149 return instanceId_; | |
150 } | |
151 else | |
152 { | |
153 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadSequenceOfCalls); | |
154 } | |
155 } | |
156 | |
157 void OrthancMultiframeVolumeLoader::ScheduleFrameDownloads() | |
158 { | |
159 if (transferSyntaxUid_.empty() || | |
160 !volume_->HasGeometry()) | |
161 { | |
162 return; | |
163 } | |
164 /* | |
165 1.2.840.10008.1.2 Implicit VR Endian: Default Transfer Syntax for DICOM | |
166 1.2.840.10008.1.2.1 Explicit VR Little Endian | |
167 1.2.840.10008.1.2.2 Explicit VR Big Endian | |
168 | |
169 See https://www.dicomlibrary.com/dicom/transfer-syntax/ | |
170 */ | |
171 if (transferSyntaxUid_ == "1.2.840.10008.1.2" || | |
172 transferSyntaxUid_ == "1.2.840.10008.1.2.1" || | |
173 transferSyntaxUid_ == "1.2.840.10008.1.2.2") | |
174 { | |
175 std::auto_ptr<OrthancRestApiCommand> command(new OrthancRestApiCommand); | |
176 command->SetHttpHeader("Accept-Encoding", "gzip"); | |
177 command->SetUri("/instances/" + instanceId_ + "/content/" + | |
178 Orthanc::DICOM_TAG_PIXEL_DATA.Format() + "/0"); | |
179 command->SetPayload(new LoadUncompressedPixelData(*this)); | |
180 Schedule(command.release()); | |
181 } | |
182 else | |
183 { | |
184 throw Orthanc::OrthancException( | |
185 Orthanc::ErrorCode_NotImplemented, | |
186 "No support for multiframe instances with transfer syntax: " + transferSyntaxUid_); | |
187 } | |
188 } | |
189 | |
190 | |
191 void OrthancMultiframeVolumeLoader::SetTransferSyntax(const std::string& transferSyntax) | |
192 { | |
193 transferSyntaxUid_ = Orthanc::Toolbox::StripSpaces(transferSyntax); | |
194 ScheduleFrameDownloads(); | |
195 } | |
196 | |
197 void OrthancMultiframeVolumeLoader::SetGeometry(const Orthanc::DicomMap& dicom) | |
198 { | |
199 DicomInstanceParameters parameters(dicom); | |
200 volume_->SetDicomParameters(parameters); | |
201 | |
202 Orthanc::PixelFormat format; | |
203 if (!parameters.GetImageInformation().ExtractPixelFormat(format, true)) | |
204 { | |
205 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); | |
206 } | |
207 | |
208 double spacingZ; | |
209 switch (parameters.GetSopClassUid()) | |
210 { | |
211 case SopClassUid_RTDose: | |
212 spacingZ = parameters.GetThickness(); | |
213 break; | |
214 | |
215 default: | |
216 throw Orthanc::OrthancException( | |
217 Orthanc::ErrorCode_NotImplemented, | |
218 "No support for multiframe instances with SOP class UID: " + GetSopClassUid(dicom)); | |
219 } | |
220 | |
221 const unsigned int width = parameters.GetImageInformation().GetWidth(); | |
222 const unsigned int height = parameters.GetImageInformation().GetHeight(); | |
223 const unsigned int depth = parameters.GetImageInformation().GetNumberOfFrames(); | |
224 | |
225 { | |
226 VolumeImageGeometry geometry; | |
227 geometry.SetSize(width, height, depth); | |
228 geometry.SetAxialGeometry(parameters.GetGeometry()); | |
229 geometry.SetVoxelDimensions(parameters.GetPixelSpacingX(), | |
230 parameters.GetPixelSpacingY(), spacingZ); | |
231 volume_->Initialize(geometry, format); | |
232 } | |
233 | |
234 volume_->GetPixelData().Clear(); | |
235 | |
236 ScheduleFrameDownloads(); | |
237 | |
238 BroadcastMessage(DicomVolumeImage::GeometryReadyMessage(*volume_)); | |
239 } | |
240 | |
241 | |
242 ORTHANC_FORCE_INLINE | |
925
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
243 static void CopyPixel(uint32_t& target, const void* source) |
815 | 244 { |
245 // TODO - check alignement? | |
246 target = le32toh(*reinterpret_cast<const uint32_t*>(source)); | |
247 } | |
925
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
248 |
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
249 ORTHANC_FORCE_INLINE |
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
250 static void CopyPixel(uint16_t& target, const void* source) |
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
251 { |
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
252 // TODO - check alignement? |
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
253 target = le16toh(*reinterpret_cast<const uint16_t*>(source)); |
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
254 } |
815 | 255 |
256 template <typename T> | |
257 void OrthancMultiframeVolumeLoader::CopyPixelData(const std::string& pixelData) | |
258 { | |
259 ImageBuffer3D& target = volume_->GetPixelData(); | |
260 | |
261 const unsigned int bpp = target.GetBytesPerPixel(); | |
262 const unsigned int width = target.GetWidth(); | |
263 const unsigned int height = target.GetHeight(); | |
264 const unsigned int depth = target.GetDepth(); | |
265 | |
266 if (pixelData.size() != bpp * width * height * depth) | |
267 { | |
268 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadFileFormat, | |
269 "The pixel data has not the proper size"); | |
270 } | |
271 | |
272 if (pixelData.empty()) | |
273 { | |
274 return; | |
275 } | |
276 | |
277 const uint8_t* source = reinterpret_cast<const uint8_t*>(pixelData.c_str()); | |
278 | |
279 for (unsigned int z = 0; z < depth; z++) | |
280 { | |
281 ImageBuffer3D::SliceWriter writer(target, VolumeProjection_Axial, z); | |
282 | |
283 assert (writer.GetAccessor().GetWidth() == width && | |
284 writer.GetAccessor().GetHeight() == height); | |
285 | |
286 for (unsigned int y = 0; y < height; y++) | |
287 { | |
288 assert(sizeof(T) == Orthanc::GetBytesPerPixel(target.GetFormat())); | |
289 | |
290 T* target = reinterpret_cast<T*>(writer.GetAccessor().GetRow(y)); | |
291 | |
292 for (unsigned int x = 0; x < width; x++) | |
293 { | |
294 CopyPixel(*target, source); | |
295 | |
296 target ++; | |
297 source += bpp; | |
298 } | |
299 } | |
300 } | |
301 } | |
302 | |
303 void OrthancMultiframeVolumeLoader::SetUncompressedPixelData(const std::string& pixelData) | |
304 { | |
305 switch (volume_->GetPixelData().GetFormat()) | |
306 { | |
307 case Orthanc::PixelFormat_Grayscale32: | |
308 CopyPixelData<uint32_t>(pixelData); | |
309 break; | |
925
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
310 case Orthanc::PixelFormat_Grayscale16: |
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
311 CopyPixelData<uint16_t>(pixelData); |
4639d0bf6390
Added support for Grayscale16 in the multiframe loader (Victor)
Benjamin Golinvaux <bgo@osimis.io>
parents:
921
diff
changeset
|
312 break; |
815 | 313 default: |
314 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); | |
315 } | |
316 | |
317 volume_->IncrementRevision(); | |
318 | |
937
86ac61a040c9
Added getters and notifications to allow clients of the loaders (DicomStructureSetLoader, OrthancSeriesVolumeProgressiveLoader and OrthancMultiframeVolumeLoader) to know when the loading is finished + added ability for SDL event loop to execute a callback repeatedly (used to check the view loading state)
Benjamin Golinvaux <bgo@osimis.io>
parents:
935
diff
changeset
|
319 pixelDataLoaded_ = true; |
815 | 320 BroadcastMessage(DicomVolumeImage::ContentUpdatedMessage(*volume_)); |
321 } | |
322 | |
323 | |
921
81d30cd93b65
Ability to ask the loader for the geometry in PULL mode (when subscribing to the messages is not possible) + small changes (removed const/ref qualifiers for boost::shared_ptr param, added traces, doc change)
Benjamin Golinvaux <bgo@osimis.io>
parents:
816
diff
changeset
|
324 OrthancMultiframeVolumeLoader::OrthancMultiframeVolumeLoader(boost::shared_ptr<DicomVolumeImage> volume, |
815 | 325 IOracle& oracle, |
326 IObservable& oracleObservable) : | |
327 LoaderStateMachine(oracle, oracleObservable), | |
328 IObservable(oracleObservable.GetBroker()), | |
937
86ac61a040c9
Added getters and notifications to allow clients of the loaders (DicomStructureSetLoader, OrthancSeriesVolumeProgressiveLoader and OrthancMultiframeVolumeLoader) to know when the loading is finished + added ability for SDL event loop to execute a callback repeatedly (used to check the view loading state)
Benjamin Golinvaux <bgo@osimis.io>
parents:
935
diff
changeset
|
329 volume_(volume), |
86ac61a040c9
Added getters and notifications to allow clients of the loaders (DicomStructureSetLoader, OrthancSeriesVolumeProgressiveLoader and OrthancMultiframeVolumeLoader) to know when the loading is finished + added ability for SDL event loop to execute a callback repeatedly (used to check the view loading state)
Benjamin Golinvaux <bgo@osimis.io>
parents:
935
diff
changeset
|
330 pixelDataLoaded_(false) |
815 | 331 { |
332 if (volume.get() == NULL) | |
333 { | |
334 throw Orthanc::OrthancException(Orthanc::ErrorCode_NullPointer); | |
335 } | |
336 } | |
337 | |
935
401808e7ff2e
Added traces in LoaderCache objects dtors + fuse to prevent dead weak ptrs to be calleds in VisitWdigets (in GuiAdapter.h)
Benjamin Golinvaux <bgo@osimis.io>
parents:
925
diff
changeset
|
338 OrthancMultiframeVolumeLoader::~OrthancMultiframeVolumeLoader() |
401808e7ff2e
Added traces in LoaderCache objects dtors + fuse to prevent dead weak ptrs to be calleds in VisitWdigets (in GuiAdapter.h)
Benjamin Golinvaux <bgo@osimis.io>
parents:
925
diff
changeset
|
339 { |
401808e7ff2e
Added traces in LoaderCache objects dtors + fuse to prevent dead weak ptrs to be calleds in VisitWdigets (in GuiAdapter.h)
Benjamin Golinvaux <bgo@osimis.io>
parents:
925
diff
changeset
|
340 LOG(TRACE) << "OrthancMultiframeVolumeLoader::~OrthancMultiframeVolumeLoader()"; |
401808e7ff2e
Added traces in LoaderCache objects dtors + fuse to prevent dead weak ptrs to be calleds in VisitWdigets (in GuiAdapter.h)
Benjamin Golinvaux <bgo@osimis.io>
parents:
925
diff
changeset
|
341 } |
815 | 342 |
343 void OrthancMultiframeVolumeLoader::LoadInstance(const std::string& instanceId) | |
344 { | |
345 Start(); | |
346 | |
347 instanceId_ = instanceId; | |
348 | |
349 { | |
350 std::auto_ptr<OrthancRestApiCommand> command(new OrthancRestApiCommand); | |
351 command->SetHttpHeader("Accept-Encoding", "gzip"); | |
352 command->SetUri("/instances/" + instanceId + "/tags"); | |
353 command->SetPayload(new LoadGeometry(*this)); | |
354 Schedule(command.release()); | |
355 } | |
356 | |
357 { | |
358 std::auto_ptr<OrthancRestApiCommand> command(new OrthancRestApiCommand); | |
359 command->SetUri("/instances/" + instanceId + "/metadata/TransferSyntax"); | |
360 command->SetPayload(new LoadTransferSyntax(*this)); | |
361 Schedule(command.release()); | |
362 } | |
363 } | |
364 } |