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_;
+      }
+    };
+  };
+}