Mercurial > hg > orthanc
view UnitTestsSources/ImageProcessingTests.cpp @ 2851:859e880ac9a8
merge
author | am@osimis.io |
---|---|
date | Mon, 01 Oct 2018 14:19:45 +0200 |
parents | be1dbb1dcdd6 |
children | 22524fd06225 |
line wrap: on
line source
/** * Orthanc - A Lightweight, RESTful DICOM Store * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics * Department, University Hospital of Liege, Belgium * Copyright (C) 2017-2018 Osimis S.A., 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/>. **/ #include "PrecompiledHeadersUnitTests.h" #include "gtest/gtest.h" #include "../Core/DicomFormat/DicomImageInformation.h" #include "../Core/Images/Image.h" #include "../Core/Images/ImageProcessing.h" #include "../Core/Images/ImageTraits.h" #include <memory> using namespace Orthanc; TEST(DicomImageInformation, ExtractPixelFormat1) { // Cardiac/MR* DicomMap m; m.SetValue(DICOM_TAG_ROWS, "24", false); m.SetValue(DICOM_TAG_COLUMNS, "16", false); m.SetValue(DICOM_TAG_BITS_ALLOCATED, "16", false); m.SetValue(DICOM_TAG_SAMPLES_PER_PIXEL, "1", false); m.SetValue(DICOM_TAG_BITS_STORED, "12", false); m.SetValue(DICOM_TAG_HIGH_BIT, "11", false); m.SetValue(DICOM_TAG_PIXEL_REPRESENTATION, "0", false); m.SetValue(DICOM_TAG_PHOTOMETRIC_INTERPRETATION, "MONOCHROME2", false); DicomImageInformation info(m); PixelFormat format; ASSERT_TRUE(info.ExtractPixelFormat(format, false)); ASSERT_EQ(PixelFormat_Grayscale16, format); } TEST(DicomImageInformation, ExtractPixelFormat2) { // Delphine CT DicomMap m; m.SetValue(DICOM_TAG_ROWS, "24", false); m.SetValue(DICOM_TAG_COLUMNS, "16", false); m.SetValue(DICOM_TAG_BITS_ALLOCATED, "16", false); m.SetValue(DICOM_TAG_SAMPLES_PER_PIXEL, "1", false); m.SetValue(DICOM_TAG_BITS_STORED, "16", false); m.SetValue(DICOM_TAG_HIGH_BIT, "15", false); m.SetValue(DICOM_TAG_PIXEL_REPRESENTATION, "1", false); m.SetValue(DICOM_TAG_PHOTOMETRIC_INTERPRETATION, "MONOCHROME2", false); DicomImageInformation info(m); PixelFormat format; ASSERT_TRUE(info.ExtractPixelFormat(format, false)); ASSERT_EQ(PixelFormat_SignedGrayscale16, format); } namespace { template <typename T> class TestImageTraits : public ::testing::Test { private: std::auto_ptr<Image> image_; protected: virtual void SetUp() { image_.reset(new Image(ImageTraits::PixelTraits::GetPixelFormat(), 7, 9, false)); } virtual void TearDown() { image_.reset(NULL); } public: typedef T ImageTraits; ImageAccessor& GetImage() { return *image_; } }; template <typename T> class TestIntegerImageTraits : public TestImageTraits<T> { }; } typedef ::testing::Types< ImageTraits<PixelFormat_Grayscale8>, ImageTraits<PixelFormat_Grayscale16>, ImageTraits<PixelFormat_SignedGrayscale16> > IntegerFormats; TYPED_TEST_CASE(TestIntegerImageTraits, IntegerFormats); typedef ::testing::Types< ImageTraits<PixelFormat_Grayscale8>, ImageTraits<PixelFormat_Grayscale16>, ImageTraits<PixelFormat_SignedGrayscale16>, ImageTraits<PixelFormat_RGB24>, ImageTraits<PixelFormat_BGRA32> > AllFormats; TYPED_TEST_CASE(TestImageTraits, AllFormats); TYPED_TEST(TestImageTraits, SetZero) { ImageAccessor& image = this->GetImage(); memset(image.GetBuffer(), 128, image.GetHeight() * image.GetWidth()); switch (image.GetFormat()) { case PixelFormat_Grayscale8: case PixelFormat_Grayscale16: case PixelFormat_SignedGrayscale16: ImageProcessing::Set(image, 0); break; case PixelFormat_RGB24: case PixelFormat_BGRA32: ImageProcessing::Set(image, 0, 0, 0, 0); break; default: ASSERT_TRUE(0); } typename TestFixture::ImageTraits::PixelType zero, value; TestFixture::ImageTraits::PixelTraits::SetZero(zero); for (unsigned int y = 0; y < image.GetHeight(); y++) { for (unsigned int x = 0; x < image.GetWidth(); x++) { TestFixture::ImageTraits::GetPixel(value, image, x, y); ASSERT_TRUE(TestFixture::ImageTraits::PixelTraits::IsEqual(zero, value)); } } } TYPED_TEST(TestIntegerImageTraits, SetZeroFloat) { ImageAccessor& image = this->GetImage(); memset(image.GetBuffer(), 128, image.GetHeight() * image.GetWidth()); unsigned int c = 0; for (unsigned int y = 0; y < image.GetHeight(); y++) { for (unsigned int x = 0; x < image.GetWidth(); x++, c++) { TestFixture::ImageTraits::SetFloatPixel(image, c, x, y); } } c = 0; for (unsigned int y = 0; y < image.GetHeight(); y++) { for (unsigned int x = 0; x < image.GetWidth(); x++, c++) { ASSERT_FLOAT_EQ(c, TestFixture::ImageTraits::GetFloatPixel(image, x, y)); } } }