Mercurial > hg > orthanc-wsi
annotate Framework/Outputs/DicomPyramidWriter.cpp @ 35:ed649650d9d3
back to mainline
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 28 Oct 2016 09:57:03 +0200 |
parents | 7a88c614be04 |
children | 7a3853d51c45 |
rev | line source |
---|---|
0 | 1 /** |
2 * Orthanc - A Lightweight, RESTful DICOM Store | |
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 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 | |
16
7a88c614be04
preparing for precompiled headers
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
7
diff
changeset
|
21 #include "../PrecompiledHeadersWSI.h" |
0 | 22 #include "DicomPyramidWriter.h" |
23 | |
24 #include "../DicomToolbox.h" | |
25 | |
26 #include "../Orthanc/Core/Logging.h" | |
27 #include "../Orthanc/Core/OrthancException.h" | |
28 #include "../Orthanc/OrthancServer/FromDcmtkBridge.h" | |
29 | |
30 #include <dcmtk/dcmdata/dcdeftag.h> | |
31 #include <boost/lexical_cast.hpp> | |
32 | |
33 namespace OrthancWSI | |
34 { | |
35 void DicomPyramidWriter::FlushInternal(MultiframeDicomWriter& writer, | |
36 bool force) | |
37 { | |
38 if (writer.GetFramesCount() > 0 && | |
39 writer.GetSize() > 0 && | |
40 (force || (maxSize_ != 0 && writer.GetSize() >= maxSize_))) | |
41 { | |
42 countInstances_ += 1; | |
43 | |
44 std::string dicom; | |
45 writer.Flush(dicom, countInstances_); | |
46 target_.Write(dicom); | |
47 } | |
48 } | |
49 | |
50 | |
51 DcmItem* DicomPyramidWriter::CreateFunctionalGroup(unsigned int frame, | |
52 unsigned int x, | |
53 unsigned int y, | |
54 unsigned int totalWidth, | |
55 unsigned int totalHeight, | |
56 float physicalZ) const | |
57 { | |
58 float physicalX, physicalY; | |
59 volume_.GetLocation(physicalX, physicalY, x, y, totalWidth, totalHeight); | |
60 | |
61 std::string tmpX = boost::lexical_cast<std::string>(physicalX); | |
62 std::string tmpY = boost::lexical_cast<std::string>(physicalY); | |
63 std::string tmpZ = boost::lexical_cast<std::string>(physicalZ); | |
64 | |
65 std::auto_ptr<DcmItem> dimension(new DcmItem); | |
66 if (!dimension->putAndInsertUint32(DCM_DimensionIndexValues, frame).good()) | |
67 { | |
68 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
69 } | |
70 | |
71 // From Supp 145: The column position of the top left pixel of the | |
72 // Total Pixel Matrix is 1. The row position of the top left pixel | |
73 // of the Total Pixel Matrix is 1. | |
74 std::auto_ptr<DcmItem> position(new DcmItem); | |
75 if (!position->putAndInsertSint32(DCM_ColumnPositionInTotalImagePixelMatrix, x + 1).good() || | |
76 !position->putAndInsertSint32(DCM_RowPositionInTotalImagePixelMatrix, y + 1).good() || | |
77 !position->putAndInsertString(DCM_XOffsetInSlideCoordinateSystem, tmpX.c_str()).good() || | |
78 !position->putAndInsertString(DCM_YOffsetInSlideCoordinateSystem, tmpY.c_str()).good() || | |
79 !position->putAndInsertString(DCM_ZOffsetInSlideCoordinateSystem, tmpZ.c_str()).good()) | |
80 { | |
81 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
82 } | |
83 | |
84 | |
85 std::auto_ptr<DcmSequenceOfItems> sequencePosition(new DcmSequenceOfItems(DCM_PlanePositionSlideSequence)); | |
86 if (!sequencePosition->insert(position.release(), false, false).good()) | |
87 { | |
88 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
89 } | |
90 | |
91 std::auto_ptr<DcmSequenceOfItems> sequenceDimension(new DcmSequenceOfItems(DCM_FrameContentSequence)); | |
92 if (!sequenceDimension->insert(dimension.release(), false, false).good()) | |
93 { | |
94 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
95 } | |
96 | |
97 std::auto_ptr<DcmItem> item(new DcmItem); | |
98 if (!item->insert(sequencePosition.release(), false, false).good() || | |
99 !item->insert(sequenceDimension.release(), false, false).good()) | |
100 { | |
101 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
102 } | |
103 | |
104 return item.release(); | |
105 } | |
106 | |
107 | |
108 void DicomPyramidWriter::WriteRawTileInternal(const std::string& tile, | |
109 const Level& level, | |
110 unsigned int x, | |
111 unsigned int y) | |
112 { | |
113 if (x >= level.countTilesX_ || | |
114 y >= level.countTilesY_) | |
115 { | |
116 LOG(ERROR) << "Tile index out of range: " << x << "," << y | |
117 << " (max: " << level.countTilesX_ << "," << level.countTilesY_ << ")"; | |
118 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
119 } | |
120 | |
121 const unsigned int z = level.z_; | |
122 | |
123 { | |
124 boost::mutex::scoped_lock lock(mutex_); | |
125 | |
126 if (z >= writers_.size()) | |
127 { | |
128 writers_.resize(z + 1); | |
129 } | |
130 | |
131 MultiframeDicomWriter* writer = writers_[z]; | |
132 | |
133 if (writer == NULL) | |
134 { | |
135 writer = new MultiframeDicomWriter | |
136 (dataset_, GetImageCompression(), GetPixelFormat(), level.width_, level.height_, | |
137 GetTileWidth(), GetTileHeight()); | |
138 writers_[z] = writer; | |
139 } | |
140 | |
141 std::auto_ptr<DcmItem> functionalGroup(CreateFunctionalGroup(writer->GetFramesCount() + 1, | |
142 x * GetTileWidth(), | |
143 y * GetTileHeight(), | |
144 writer->GetTotalWidth(), | |
145 writer->GetTotalHeight(), | |
146 0.0f /* TODO Z-plane */)); | |
147 | |
148 writer->AddFrame(tile, functionalGroup.release()); | |
149 FlushInternal(*writer, false); | |
150 | |
151 countTiles_ ++; | |
152 } | |
153 } | |
154 | |
155 | |
156 DicomPyramidWriter::DicomPyramidWriter(IFileTarget& target, | |
157 const DcmDataset& dataset, | |
158 Orthanc::PixelFormat pixelFormat, | |
159 ImageCompression compression, | |
160 unsigned int tileWidth, | |
161 unsigned int tileHeight, | |
162 size_t maxSize, // If "0", no automatic flushing | |
163 const ImagedVolumeParameters& volume) : | |
164 PyramidWriterBase(pixelFormat, compression, tileWidth, tileHeight), | |
165 target_(target), | |
166 dataset_(dataset), | |
167 maxSize_(maxSize), | |
168 countTiles_(0), | |
169 countInstances_(0), | |
170 volume_(volume) | |
171 { | |
172 } | |
173 | |
174 | |
175 DicomPyramidWriter::~DicomPyramidWriter() | |
176 { | |
177 LOG(WARNING) << "Closing the DICOM pyramid (" << countTiles_ << " tiles were written)"; | |
178 | |
179 for (size_t i = 0; i < writers_.size(); i++) | |
180 { | |
181 if (writers_[i] != NULL) | |
182 { | |
7
bc3ca410b765
Fix freeze if the target Orthanc is not accepting images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
183 try |
bc3ca410b765
Fix freeze if the target Orthanc is not accepting images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
184 { |
bc3ca410b765
Fix freeze if the target Orthanc is not accepting images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
185 FlushInternal(*writers_[i], true); |
bc3ca410b765
Fix freeze if the target Orthanc is not accepting images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
186 } |
bc3ca410b765
Fix freeze if the target Orthanc is not accepting images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
187 catch (Orthanc::OrthancException&) |
bc3ca410b765
Fix freeze if the target Orthanc is not accepting images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
188 { |
bc3ca410b765
Fix freeze if the target Orthanc is not accepting images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
189 LOG(ERROR) << "Cannot push the pending tiles to the DICOM pyramid while finalizing"; |
bc3ca410b765
Fix freeze if the target Orthanc is not accepting images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
190 } |
bc3ca410b765
Fix freeze if the target Orthanc is not accepting images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
0
diff
changeset
|
191 |
0 | 192 delete writers_[i]; |
193 } | |
194 } | |
195 } | |
196 | |
197 | |
198 void DicomPyramidWriter::Flush() | |
199 { | |
200 boost::mutex::scoped_lock lock(mutex_); | |
201 | |
202 for (size_t i = 0; i < writers_.size(); i++) | |
203 { | |
204 FlushInternal(*writers_[i], true); | |
205 } | |
206 } | |
207 } |