Mercurial > hg > orthanc-wsi
diff Framework/Outputs/DicomPyramidWriter.cpp @ 0:4a7a53257c7d
initial commit
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Sat, 22 Oct 2016 21:48:33 +0200 |
parents | |
children | bc3ca410b765 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Framework/Outputs/DicomPyramidWriter.cpp Sat Oct 22 21:48:33 2016 +0200 @@ -0,0 +1,198 @@ +/** + * Orthanc - A Lightweight, RESTful DICOM Store + * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics + * Department, University Hospital of Liege, Belgium + * + * This program is free software: you can redistribute it and/or + * modify it under the terms of the GNU Affero General Public License + * as published by the Free Software Foundation, either version 3 of + * the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + **/ + + +#include "DicomPyramidWriter.h" + +#include "../DicomToolbox.h" + +#include "../Orthanc/Core/Logging.h" +#include "../Orthanc/Core/OrthancException.h" +#include "../Orthanc/OrthancServer/FromDcmtkBridge.h" + +#include <dcmtk/dcmdata/dcdeftag.h> +#include <boost/lexical_cast.hpp> + +namespace OrthancWSI +{ + void DicomPyramidWriter::FlushInternal(MultiframeDicomWriter& writer, + bool force) + { + if (writer.GetFramesCount() > 0 && + writer.GetSize() > 0 && + (force || (maxSize_ != 0 && writer.GetSize() >= maxSize_))) + { + countInstances_ += 1; + + std::string dicom; + writer.Flush(dicom, countInstances_); + target_.Write(dicom); + } + } + + + DcmItem* DicomPyramidWriter::CreateFunctionalGroup(unsigned int frame, + unsigned int x, + unsigned int y, + unsigned int totalWidth, + unsigned int totalHeight, + float physicalZ) const + { + float physicalX, physicalY; + volume_.GetLocation(physicalX, physicalY, x, y, totalWidth, totalHeight); + + std::string tmpX = boost::lexical_cast<std::string>(physicalX); + std::string tmpY = boost::lexical_cast<std::string>(physicalY); + std::string tmpZ = boost::lexical_cast<std::string>(physicalZ); + + std::auto_ptr<DcmItem> dimension(new DcmItem); + if (!dimension->putAndInsertUint32(DCM_DimensionIndexValues, frame).good()) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); + } + + // From Supp 145: The column position of the top left pixel of the + // Total Pixel Matrix is 1. The row position of the top left pixel + // of the Total Pixel Matrix is 1. + std::auto_ptr<DcmItem> position(new DcmItem); + if (!position->putAndInsertSint32(DCM_ColumnPositionInTotalImagePixelMatrix, x + 1).good() || + !position->putAndInsertSint32(DCM_RowPositionInTotalImagePixelMatrix, y + 1).good() || + !position->putAndInsertString(DCM_XOffsetInSlideCoordinateSystem, tmpX.c_str()).good() || + !position->putAndInsertString(DCM_YOffsetInSlideCoordinateSystem, tmpY.c_str()).good() || + !position->putAndInsertString(DCM_ZOffsetInSlideCoordinateSystem, tmpZ.c_str()).good()) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); + } + + + std::auto_ptr<DcmSequenceOfItems> sequencePosition(new DcmSequenceOfItems(DCM_PlanePositionSlideSequence)); + if (!sequencePosition->insert(position.release(), false, false).good()) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); + } + + std::auto_ptr<DcmSequenceOfItems> sequenceDimension(new DcmSequenceOfItems(DCM_FrameContentSequence)); + if (!sequenceDimension->insert(dimension.release(), false, false).good()) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); + } + + std::auto_ptr<DcmItem> item(new DcmItem); + if (!item->insert(sequencePosition.release(), false, false).good() || + !item->insert(sequenceDimension.release(), false, false).good()) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); + } + + return item.release(); + } + + + void DicomPyramidWriter::WriteRawTileInternal(const std::string& tile, + const Level& level, + unsigned int x, + unsigned int y) + { + if (x >= level.countTilesX_ || + y >= level.countTilesY_) + { + LOG(ERROR) << "Tile index out of range: " << x << "," << y + << " (max: " << level.countTilesX_ << "," << level.countTilesY_ << ")"; + throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); + } + + const unsigned int z = level.z_; + + { + boost::mutex::scoped_lock lock(mutex_); + + if (z >= writers_.size()) + { + writers_.resize(z + 1); + } + + MultiframeDicomWriter* writer = writers_[z]; + + if (writer == NULL) + { + writer = new MultiframeDicomWriter + (dataset_, GetImageCompression(), GetPixelFormat(), level.width_, level.height_, + GetTileWidth(), GetTileHeight()); + writers_[z] = writer; + } + + std::auto_ptr<DcmItem> functionalGroup(CreateFunctionalGroup(writer->GetFramesCount() + 1, + x * GetTileWidth(), + y * GetTileHeight(), + writer->GetTotalWidth(), + writer->GetTotalHeight(), + 0.0f /* TODO Z-plane */)); + + writer->AddFrame(tile, functionalGroup.release()); + FlushInternal(*writer, false); + + countTiles_ ++; + } + } + + + DicomPyramidWriter::DicomPyramidWriter(IFileTarget& target, + const DcmDataset& dataset, + Orthanc::PixelFormat pixelFormat, + ImageCompression compression, + unsigned int tileWidth, + unsigned int tileHeight, + size_t maxSize, // If "0", no automatic flushing + const ImagedVolumeParameters& volume) : + PyramidWriterBase(pixelFormat, compression, tileWidth, tileHeight), + target_(target), + dataset_(dataset), + maxSize_(maxSize), + countTiles_(0), + countInstances_(0), + volume_(volume) + { + } + + + DicomPyramidWriter::~DicomPyramidWriter() + { + LOG(WARNING) << "Closing the DICOM pyramid (" << countTiles_ << " tiles were written)"; + + for (size_t i = 0; i < writers_.size(); i++) + { + if (writers_[i] != NULL) + { + FlushInternal(*writers_[i], true); + delete writers_[i]; + } + } + } + + + void DicomPyramidWriter::Flush() + { + boost::mutex::scoped_lock lock(mutex_); + + for (size_t i = 0; i < writers_.size(); i++) + { + FlushInternal(*writers_[i], true); + } + } +}