Mercurial > hg > orthanc-stone
view OrthancStone/Sources/Toolbox/SubvoxelReader.h @ 2086:40476d5e0cfd
merge
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 07 Nov 2023 17:03:54 +0100 |
parents | 07964689cb0b |
children | c23eef785569 |
line wrap: on
line source
/** * Stone of Orthanc * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics * Department, University Hospital of Liege, Belgium * Copyright (C) 2017-2023 Osimis S.A., Belgium * Copyright (C) 2021-2023 Sebastien Jodogne, ICTEAM UCLouvain, Belgium * * This program is free software: you can redistribute it and/or * modify it under the terms of the GNU Lesser 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this program. If not, see * <http://www.gnu.org/licenses/>. **/ #pragma once #include "../Volumes/ImageBuffer3D.h" #include "GeometryToolbox.h" #include <Images/ImageTraits.h> #include <boost/noncopyable.hpp> #include <cmath> namespace OrthancStone { namespace Internals { class SubvoxelReaderBase : public boost::noncopyable { private: const ImageBuffer3D& source_; unsigned int width_; unsigned int height_; unsigned int depth_; public: explicit SubvoxelReaderBase(const ImageBuffer3D& source) : source_(source), width_(source.GetWidth()), height_(source.GetHeight()), depth_(source.GetDepth()) { } ORTHANC_FORCE_INLINE const Orthanc::ImageAccessor& GetSource() const { return source_.GetInternalImage(); } ORTHANC_FORCE_INLINE unsigned int GetWidth() const { return width_; } ORTHANC_FORCE_INLINE unsigned int GetHeight() const { return height_; } ORTHANC_FORCE_INLINE unsigned int GetDepth() const { return depth_; } ORTHANC_FORCE_INLINE unsigned int ComputeRow(unsigned int y, unsigned int z) const { /** * The "(depth_ - 1 - z)" comes from the fact that * "ImageBuffer3D" class stores its slices in DECREASING * z-order along the normal. This computation makes the * "SubvoxelReader" class use the same convention as * "ImageBuffer3D::GetVoxelXXX()". * * WARNING: Until changeset 1782:f053c80ea411, "z" was * directly used, causing this class to have a slice order * that was reversed between "SubvoxelReader" and * "ImageBuffer3D". This notably made * "DicomVolumeImageMPRSlicer" and "DicomVolumeImageReslicer" * inconsistent in sagittal and coronal views (the texture was * flipped along the Y-axis in the canvas). **/ return (depth_ - 1 - z) * height_ + y; } }; } template <Orthanc::PixelFormat Format, ImageInterpolation Interpolation> class SubvoxelReader; template <Orthanc::PixelFormat Format> class SubvoxelReader<Format, ImageInterpolation_Nearest> : public Internals::SubvoxelReaderBase { public: typedef Orthanc::PixelTraits<Format> Traits; typedef typename Traits::PixelType PixelType; explicit SubvoxelReader(const ImageBuffer3D& source) : SubvoxelReaderBase(source) { } inline bool GetValue(PixelType& target, float x, float y, float z) const; inline bool GetFloatValue(float& target, float x, float y, float z) const; }; template <Orthanc::PixelFormat Format> class SubvoxelReader<Format, ImageInterpolation_Bilinear> : public Internals::SubvoxelReaderBase { public: typedef Orthanc::PixelTraits<Format> Traits; typedef typename Traits::PixelType PixelType; explicit SubvoxelReader(const ImageBuffer3D& source) : SubvoxelReaderBase(source) { } inline bool Sample(float& f00, float& f01, float& f10, float& f11, unsigned int ux, unsigned int uy, unsigned int uz) const; inline bool GetValue(PixelType& target, float x, float y, float z) const; inline bool GetFloatValue(float& target, float x, float y, float z) const; }; template <Orthanc::PixelFormat Format> class SubvoxelReader<Format, ImageInterpolation_Trilinear> : public Internals::SubvoxelReaderBase { private: SubvoxelReader<Format, ImageInterpolation_Bilinear> bilinear_; public: typedef Orthanc::PixelTraits<Format> Traits; typedef typename Traits::PixelType PixelType; explicit SubvoxelReader(const ImageBuffer3D& source) : SubvoxelReaderBase(source), bilinear_(source) { } inline bool GetValue(PixelType& target, float x, float y, float z) const; inline bool GetFloatValue(float& target, float x, float y, float z) const; }; template <Orthanc::PixelFormat Format> bool SubvoxelReader<Format, ImageInterpolation_Nearest>::GetValue(PixelType& target, float x, float y, float z) const { if (x < 0 || y < 0 || z < 0) { return false; } else { unsigned int ux = static_cast<unsigned int>(std::floor(x)); unsigned int uy = static_cast<unsigned int>(std::floor(y)); unsigned int uz = static_cast<unsigned int>(std::floor(z)); if (ux < GetWidth() && uy < GetHeight() && uz < GetDepth()) { Orthanc::ImageTraits<Format>::GetPixel(target, GetSource(), ux, ComputeRow(uy, uz)); return true; } else { return false; } } } template <Orthanc::PixelFormat Format> bool SubvoxelReader<Format, ImageInterpolation_Nearest>::GetFloatValue(float& target, float x, float y, float z) const { PixelType value; if (GetValue(value, x, y, z)) { target = Traits::PixelToFloat(value); return true; } else { return false; } } template <Orthanc::PixelFormat Format> bool SubvoxelReader<Format, ImageInterpolation_Bilinear>::Sample(float& f00, float& f01, float& f10, float& f11, unsigned int ux, unsigned int uy, unsigned int uz) const { if (ux < GetWidth() && uy < GetHeight() && uz < GetDepth()) { f00 = Orthanc::ImageTraits<Format>::GetFloatPixel(GetSource(), ux, ComputeRow(uy, uz)); } else { // Pixel is out of the volume return false; } if (ux + 1 < GetWidth()) { f01 = Orthanc::ImageTraits<Format>::GetFloatPixel(GetSource(), ux + 1, ComputeRow(uy, uz)); } else { f01 = f00; } if (uy + 1 < GetHeight()) { f10 = Orthanc::ImageTraits<Format>::GetFloatPixel(GetSource(), ux, ComputeRow(uy + 1, uz)); } else { f10 = f00; } if (ux + 1 < GetWidth() && uy + 1 < GetHeight()) { f11 = Orthanc::ImageTraits<Format>::GetFloatPixel(GetSource(), ux + 1, ComputeRow(uy + 1, uz)); } else { f11 = f00; } return true; } template <Orthanc::PixelFormat Format> bool SubvoxelReader<Format, ImageInterpolation_Bilinear>::GetFloatValue(float& target, float x, float y, float z) const { x -= 0.5f; y -= 0.5f; if (x < 0 || y < 0 || z < 0) { return false; } else { unsigned int ux = static_cast<unsigned int>(std::floor(x)); unsigned int uy = static_cast<unsigned int>(std::floor(y)); unsigned int uz = static_cast<unsigned int>(std::floor(z)); float f00, f01, f10, f11; if (Sample(f00, f01, f10, f11, ux, uy, uz)) { float ax = x - static_cast<float>(ux); float ay = y - static_cast<float>(uy); target = GeometryToolbox::ComputeBilinearInterpolationUnitSquare(ax, ay, f00, f01, f10, f11); return true; } else { return false; } } } template <Orthanc::PixelFormat Format> bool SubvoxelReader<Format, ImageInterpolation_Bilinear>::GetValue(PixelType& target, float x, float y, float z) const { float value; if (GetFloatValue(value, x, y, z)) { Traits::FloatToPixel(target, value); return true; } else { return false; } } template <Orthanc::PixelFormat Format> bool SubvoxelReader<Format, ImageInterpolation_Trilinear>::GetFloatValue(float& target, float x, float y, float z) const { x -= 0.5f; y -= 0.5f; z -= 0.5f; if (x < 0 || y < 0 || z < 0) { return false; } else { unsigned int ux = static_cast<unsigned int>(std::floor(x)); unsigned int uy = static_cast<unsigned int>(std::floor(y)); unsigned int uz = static_cast<unsigned int>(std::floor(z)); float f000, f001, f010, f011; if (bilinear_.Sample(f000, f001, f010, f011, ux, uy, uz)) { const float ax = x - static_cast<float>(ux); const float ay = y - static_cast<float>(uy); float f100, f101, f110, f111; if (bilinear_.Sample(f100, f101, f110, f111, ux, uy, uz + 1)) { const float az = z - static_cast<float>(uz); target = GeometryToolbox::ComputeTrilinearInterpolationUnitSquare (ax, ay, az, f000, f001, f010, f011, f100, f101, f110, f111); } else { target = GeometryToolbox::ComputeBilinearInterpolationUnitSquare (ax, ay, f000, f001, f010, f011); } return true; } else { return false; } } } template <Orthanc::PixelFormat Format> bool SubvoxelReader<Format, ImageInterpolation_Trilinear>::GetValue(PixelType& target, float x, float y, float z) const { float value; if (GetFloatValue(value, x, y, z)) { Traits::FloatToPixel(target, value); return true; } else { return false; } } }