Mercurial > hg > orthanc-stone
diff Framework/Volumes/ImageBuffer3D.h @ 0:351ab0da0150
initial commit
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 14 Oct 2016 15:34:11 +0200 |
parents | |
children | ff1e935768e7 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Framework/Volumes/ImageBuffer3D.h Fri Oct 14 15:34:11 2016 +0200 @@ -0,0 +1,163 @@ +/** + * Stone of Orthanc + * 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 General Public License as + * published by the Free Software Foundation, either version 3 of the + * License, or (at your option) any later version. + * + * In addition, as a special exception, the copyright holders of this + * program give permission to link the code of its release with the + * OpenSSL project's "OpenSSL" library (or with modified versions of it + * that use the same license as the "OpenSSL" library), and distribute + * the linked executables. You must obey the GNU General Public License + * in all respects for all of the code used other than "OpenSSL". If you + * modify file(s) with this exception, you may extend this exception to + * your version of the file(s), but you are not obligated to do so. If + * you do not wish to do so, delete this exception statement from your + * version. If you delete this exception statement from all source files + * in the program, then also delete it here. + * + * 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 + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + **/ + + +#pragma once + +#include "../Enumerations.h" +#include "../Toolbox/IThreadSafety.h" +#include "../Toolbox/SliceGeometry.h" +#include "../Toolbox/ParallelSlices.h" + +#include "../Orthanc/Core/Images/Image.h" + +#include <boost/thread/shared_mutex.hpp> + +#if defined(_WIN32) +# include <boost/thread/win32/mutex.hpp> +#endif + +namespace OrthancStone +{ + class ImageBuffer3D : public IThreadSafe + { + private: + typedef boost::shared_mutex Mutex; + typedef boost::unique_lock<Mutex> WriteLock; + typedef boost::shared_lock<Mutex> ReadLock; + + Mutex mutex_; + SliceGeometry axialGeometry_; + Vector voxelDimensions_; + Orthanc::Image image_; + Orthanc::PixelFormat format_; + unsigned int width_; + unsigned int height_; + unsigned int depth_; + + Orthanc::ImageAccessor GetAxialSliceAccessor(unsigned int slice, + bool readOnly); + + Orthanc::ImageAccessor GetCoronalSliceAccessor(unsigned int slice, + bool readOnly); + + Orthanc::Image* ExtractSagittalSlice(unsigned int slice) const; + + public: + ImageBuffer3D(Orthanc::PixelFormat format, + unsigned int width, + unsigned int height, + unsigned int depth); + + void Clear(); + + // Set the geometry of the first axial slice (i.e. the one whose + // depth == 0) + void SetAxialGeometry(const SliceGeometry& geometry); + + void SetVoxelDimensions(double x, + double y, + double z); + + Vector GetVoxelDimensions(VolumeProjection projection); + + void GetSliceSize(unsigned int& width, + unsigned int& height, + VolumeProjection projection); + + unsigned int GetWidth() const + { + return width_; + } + + unsigned int GetHeight() const + { + return height_; + } + + unsigned int GetDepth() const + { + return depth_; + } + + Orthanc::PixelFormat GetFormat() const + { + return format_; + } + + ParallelSlices* GetGeometry(VolumeProjection projection); + + + class SliceReader : public boost::noncopyable + { + private: + ReadLock lock_; + Orthanc::ImageAccessor accessor_; + std::auto_ptr<Orthanc::Image> sagittal_; // Unused for axial and coronal + + public: + SliceReader(ImageBuffer3D& that, + VolumeProjection projection, + unsigned int slice); + + const Orthanc::ImageAccessor& GetAccessor() const + { + return accessor_; + } + }; + + + class SliceWriter : public boost::noncopyable + { + private: + WriteLock lock_; + Orthanc::ImageAccessor accessor_; + std::auto_ptr<Orthanc::Image> sagittal_; // Unused for axial and coronal + + void Flush(); + + public: + SliceWriter(ImageBuffer3D& that, + VolumeProjection projection, + unsigned int slice); + + ~SliceWriter() + { + Flush(); + } + + Orthanc::ImageAccessor& GetAccessor() + { + return accessor_; + } + }; + }; +}