Mercurial > hg > orthanc-wsi
annotate Framework/Algorithms/PyramidReader.cpp @ 16:7a88c614be04
preparing for precompiled headers
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Thu, 27 Oct 2016 11:01:26 +0200 |
| parents | 4a7a53257c7d |
| children | 91fc9583b2de |
| 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:
0
diff
changeset
|
21 #include "../PrecompiledHeadersWSI.h" |
| 0 | 22 #include "PyramidReader.h" |
| 23 | |
| 24 #include "../ImageToolbox.h" | |
| 25 #include "../Orthanc/Core/Logging.h" | |
| 26 #include "../Orthanc/Core/OrthancException.h" | |
| 27 | |
| 28 #include <cassert> | |
| 29 | |
| 30 namespace OrthancWSI | |
| 31 { | |
| 32 class PyramidReader::SourceTile : public boost::noncopyable | |
| 33 { | |
| 34 private: | |
| 35 PyramidReader& that_; | |
| 36 unsigned int tileX_; | |
| 37 unsigned int tileY_; | |
| 38 bool hasRawTile_; | |
| 39 std::string rawTile_; | |
| 40 | |
| 41 std::auto_ptr<Orthanc::ImageAccessor> decoded_; | |
| 42 | |
| 43 bool IsRepaintNeeded() const | |
| 44 { | |
| 45 return (that_.parameters_.IsRepaintBackground() && | |
| 46 ((tileX_ + 1) * that_.sourceTileWidth_ > that_.levelWidth_ || | |
| 47 (tileY_ + 1) * that_.sourceTileHeight_ > that_.levelHeight_)); | |
| 48 } | |
| 49 | |
| 50 void RepaintBackground() | |
| 51 { | |
| 52 assert(decoded_.get() != NULL); | |
| 53 LOG(INFO) << "Repainting background of tile (" | |
| 54 << tileX_ << "," << tileY_ << ") at level " << that_.level_; | |
| 55 | |
| 56 if ((tileY_ + 1) * that_.sourceTileHeight_ > that_.levelHeight_) | |
| 57 { | |
| 58 // Bottom overflow | |
| 59 assert(tileY_ * that_.sourceTileHeight_ < that_.levelHeight_); | |
| 60 | |
| 61 unsigned int bottom = that_.levelHeight_ - tileY_ * that_.sourceTileHeight_; | |
| 62 Orthanc::ImageAccessor a = decoded_->GetRegion(0, bottom, | |
| 63 that_.sourceTileWidth_, | |
| 64 that_.sourceTileHeight_ - bottom); | |
| 65 ImageToolbox::Set(a, | |
| 66 that_.parameters_.GetBackgroundColorRed(), | |
| 67 that_.parameters_.GetBackgroundColorGreen(), | |
| 68 that_.parameters_.GetBackgroundColorBlue()); | |
| 69 | |
| 70 } | |
| 71 | |
| 72 if ((tileX_ + 1) * that_.sourceTileWidth_ > that_.levelWidth_) | |
| 73 { | |
| 74 // Right overflow | |
| 75 assert(tileX_ * that_.sourceTileWidth_ < that_.levelWidth_); | |
| 76 | |
| 77 unsigned int right = that_.levelWidth_ - tileX_ * that_.sourceTileWidth_; | |
| 78 Orthanc::ImageAccessor a = decoded_->GetRegion(right, 0, | |
| 79 that_.sourceTileWidth_ - right, | |
| 80 that_.sourceTileHeight_); | |
| 81 ImageToolbox::Set(a, | |
| 82 that_.parameters_.GetBackgroundColorRed(), | |
| 83 that_.parameters_.GetBackgroundColorGreen(), | |
| 84 that_.parameters_.GetBackgroundColorBlue()); | |
| 85 } | |
| 86 } | |
| 87 | |
| 88 | |
| 89 public: | |
| 90 SourceTile(PyramidReader& that, | |
| 91 unsigned int tileX, | |
| 92 unsigned int tileY) : | |
| 93 that_(that), | |
| 94 tileX_(tileX), | |
| 95 tileY_(tileY) | |
| 96 { | |
| 97 if (!that_.parameters_.IsForceReencode() && | |
| 98 !IsRepaintNeeded() && | |
| 99 that_.source_.ReadRawTile(rawTile_, that_.level_, tileX, tileY)) | |
| 100 { | |
| 101 hasRawTile_ = true; | |
| 102 } | |
| 103 else | |
| 104 { | |
| 105 hasRawTile_ = false; | |
| 106 decoded_.reset(that_.source_.DecodeTile(that_.level_, tileX, tileY)); | |
| 107 if (decoded_.get() == NULL) | |
| 108 { | |
| 109 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
| 110 } | |
| 111 | |
| 112 RepaintBackground(); | |
| 113 } | |
| 114 } | |
| 115 | |
| 116 bool HasRawTile() const | |
| 117 { | |
| 118 return hasRawTile_; | |
| 119 } | |
| 120 | |
| 121 const std::string& GetRawTile() const | |
| 122 { | |
| 123 if (hasRawTile_) | |
| 124 { | |
| 125 return rawTile_; | |
| 126 } | |
| 127 else | |
| 128 { | |
| 129 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
| 130 } | |
| 131 } | |
| 132 | |
| 133 const Orthanc::ImageAccessor& GetDecodedTile() | |
| 134 { | |
| 135 if (decoded_.get() == NULL) | |
| 136 { | |
| 137 if (!hasRawTile_) | |
| 138 { | |
| 139 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
| 140 } | |
| 141 | |
| 142 decoded_.reset(ImageToolbox::DecodeTile(rawTile_, that_.source_.GetImageCompression())); | |
| 143 if (decoded_.get() == NULL) | |
| 144 { | |
| 145 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
| 146 } | |
| 147 | |
| 148 RepaintBackground(); | |
| 149 } | |
| 150 | |
| 151 return *decoded_; | |
| 152 } | |
| 153 }; | |
| 154 | |
| 155 | |
| 156 Orthanc::ImageAccessor& PyramidReader::GetOutsideTile() | |
| 157 { | |
| 158 if (outside_.get() == NULL) | |
| 159 { | |
| 160 outside_.reset(ImageToolbox::Allocate(source_.GetPixelFormat(), targetTileWidth_, targetTileHeight_)); | |
| 161 ImageToolbox::Set(*outside_, | |
| 162 parameters_.GetBackgroundColorRed(), | |
| 163 parameters_.GetBackgroundColorGreen(), | |
| 164 parameters_.GetBackgroundColorBlue()); | |
| 165 } | |
| 166 | |
| 167 return *outside_; | |
| 168 } | |
| 169 | |
| 170 | |
| 171 void PyramidReader::CheckTileSize(const Orthanc::ImageAccessor& tile) const | |
| 172 { | |
| 173 if (tile.GetWidth() != sourceTileWidth_ || | |
| 174 tile.GetHeight() != sourceTileHeight_) | |
| 175 { | |
| 176 LOG(ERROR) << "One tile in the input image has size " << tile.GetWidth() << "x" << tile.GetHeight() | |
| 177 << " instead of required " << source_.GetTileWidth() << "x" << source_.GetTileHeight(); | |
| 178 throw Orthanc::OrthancException(Orthanc::ErrorCode_IncompatibleImageSize); | |
| 179 } | |
| 180 } | |
| 181 | |
| 182 | |
| 183 void PyramidReader::CheckTileSize(const std::string& tile) const | |
| 184 { | |
| 185 if (parameters_.IsSafetyCheck()) | |
| 186 { | |
| 187 std::auto_ptr<Orthanc::ImageAccessor> decoded(ImageToolbox::DecodeTile(tile, source_.GetImageCompression())); | |
| 188 CheckTileSize(*decoded); | |
| 189 } | |
| 190 } | |
| 191 | |
| 192 | |
| 193 PyramidReader::SourceTile& PyramidReader::AccessSourceTile(const Location& location) | |
| 194 { | |
| 195 Cache::iterator found = cache_.find(location); | |
| 196 if (found != cache_.end()) | |
| 197 { | |
| 198 return *found->second; | |
| 199 } | |
| 200 else | |
| 201 { | |
| 202 SourceTile *tile = new SourceTile(*this, location.first, location.second); | |
| 203 cache_[location] = tile; | |
| 204 return *tile; | |
| 205 } | |
| 206 } | |
| 207 | |
| 208 | |
| 209 PyramidReader::Location PyramidReader::MapTargetToSourceLocation(unsigned int tileX, | |
| 210 unsigned int tileY) | |
| 211 { | |
| 212 return std::make_pair(tileX / (sourceTileWidth_ / targetTileWidth_), | |
| 213 tileY / (sourceTileHeight_ / targetTileHeight_)); | |
| 214 } | |
| 215 | |
| 216 | |
| 217 PyramidReader::PyramidReader(ITiledPyramid& source, | |
| 218 unsigned int level, | |
| 219 unsigned int targetTileWidth, | |
| 220 unsigned int targetTileHeight, | |
| 221 const DicomizerParameters& parameters) : | |
| 222 source_(source), | |
| 223 level_(level), | |
| 224 levelWidth_(source.GetLevelWidth(level)), | |
| 225 levelHeight_(source.GetLevelHeight(level)), | |
| 226 sourceTileWidth_(source.GetTileWidth()), | |
| 227 sourceTileHeight_(source.GetTileHeight()), | |
| 228 targetTileWidth_(targetTileWidth), | |
| 229 targetTileHeight_(targetTileHeight), | |
| 230 parameters_(parameters) | |
| 231 { | |
| 232 if (sourceTileWidth_ % targetTileWidth_ != 0 || | |
| 233 sourceTileHeight_ % targetTileHeight_ != 0) | |
| 234 { | |
| 235 LOG(ERROR) << "When resampling the tile size, it must be a integer divisor of the original tile size"; | |
| 236 throw Orthanc::OrthancException(Orthanc::ErrorCode_IncompatibleImageSize); | |
| 237 } | |
| 238 } | |
| 239 | |
| 240 | |
| 241 PyramidReader::~PyramidReader() | |
| 242 { | |
| 243 for (Cache::iterator it = cache_.begin(); it != cache_.end(); ++it) | |
| 244 { | |
| 245 assert(it->second != NULL); | |
| 246 delete it->second; | |
| 247 } | |
| 248 } | |
| 249 | |
| 250 | |
| 251 const std::string* PyramidReader::GetRawTile(unsigned int tileX, | |
| 252 unsigned int tileY) | |
| 253 { | |
| 254 if (sourceTileWidth_ != targetTileWidth_ || | |
| 255 sourceTileHeight_ != targetTileHeight_) | |
| 256 { | |
| 257 return NULL; | |
| 258 } | |
| 259 | |
| 260 SourceTile& source = AccessSourceTile(MapTargetToSourceLocation(tileX, tileY)); | |
| 261 if (source.HasRawTile()) | |
| 262 { | |
| 263 CheckTileSize(source.GetRawTile()); | |
| 264 return &source.GetRawTile(); | |
| 265 } | |
| 266 else | |
| 267 { | |
| 268 return NULL; | |
| 269 } | |
| 270 } | |
| 271 | |
| 272 | |
| 273 Orthanc::ImageAccessor PyramidReader::GetDecodedTile(unsigned int tileX, | |
| 274 unsigned int tileY) | |
| 275 { | |
| 276 if (tileX * targetTileWidth_ >= levelWidth_ || | |
| 277 tileY * targetTileHeight_ >= levelHeight_) | |
| 278 { | |
| 279 // Accessing a tile out of the source image | |
| 280 return GetOutsideTile(); | |
| 281 } | |
| 282 | |
| 283 SourceTile& source = AccessSourceTile(MapTargetToSourceLocation(tileX, tileY)); | |
| 284 const Orthanc::ImageAccessor& tile = source.GetDecodedTile(); | |
| 285 | |
| 286 CheckTileSize(tile); | |
| 287 | |
| 288 assert(sourceTileWidth_ % targetTileWidth_ == 0 && | |
| 289 sourceTileHeight_ % targetTileHeight_ == 0); | |
| 290 | |
| 291 unsigned int xx = tileX % (sourceTileWidth_ / targetTileWidth_); | |
| 292 unsigned int yy = tileY % (sourceTileHeight_ / targetTileHeight_); | |
| 293 | |
| 294 const uint8_t* bytes = | |
| 295 reinterpret_cast<const uint8_t*>(tile.GetConstRow(yy * targetTileHeight_)) + | |
| 296 GetBytesPerPixel(tile.GetFormat()) * xx * targetTileWidth_; | |
| 297 | |
| 298 Orthanc::ImageAccessor region; | |
| 299 region.AssignReadOnly(tile.GetFormat(), | |
| 300 targetTileWidth_, | |
| 301 targetTileHeight_, | |
| 302 tile.GetPitch(), bytes); | |
| 303 | |
| 304 return region; | |
| 305 } | |
| 306 } |