MITK-IGT
IGT Extension of MITK
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
mitkUSImageLoggingFilterTest.cpp
Go to the documentation of this file.
1/*============================================================================
2
3The Medical Imaging Interaction Toolkit (MITK)
4
5Copyright (c) German Cancer Research Center (DKFZ)
6All rights reserved.
7
8Use of this source code is governed by a 3-clause BSD license that can be
9found in the LICENSE file.
10
11============================================================================*/
12
13#include "mitkUSImageLoggingFilter.h"
14#include <mitkTestingMacros.h>
15#include <mitkTestFixture.h>
16#include <mitkTestingConfig.h>
17#include <mitkIOUtil.h>
18
19#include <mitkIOMimeTypes.h>
20#include <mitkCoreServices.h>
21#include <mitkIMimeTypeProvider.h>
22
23#include "mitkImageGenerator.h"
24
25#include "itksys/SystemTools.hxx"
26
27#include "Poco/File.h"
28
29class mitkUSImageLoggingFilterTestSuite : public mitk::TestFixture
30{
31 CPPUNIT_TEST_SUITE(mitkUSImageLoggingFilterTestSuite);
32 MITK_TEST(TestInstantiation);
33 MITK_TEST(TestSetFileExtension);
34 MITK_TEST(TestSetWrongFileExtension);
35 MITK_TEST(TestSavingValidTestImage);
36 MITK_TEST(TestSavingAfterMupltipleUpdateCalls);
37 MITK_TEST(TestFilterWithEmptyImages);
38 MITK_TEST(TestFilterWithInvalidPath);
39 //MITK_TEST(TestJpgFileExtension); //bug 19614
40 CPPUNIT_TEST_SUITE_END();
41
42private:
43
44 mitk::USImageLoggingFilter::Pointer m_TestFilter;
45 std::string m_TemporaryTestDirectory;
46 mitk::Image::Pointer m_RandomRestImage1;
47 mitk::Image::Pointer m_RandomRestImage2;
48 mitk::Image::Pointer m_RandomSingleSliceImage;
49 mitk::Image::Pointer m_RealTestImage;
50
51public:
52
53 void setUp() override
54 {
55 m_TestFilter = mitk::USImageLoggingFilter::New();
56 m_TemporaryTestDirectory = mitk::IOUtil::GetTempPath();
57 m_RandomRestImage1 = mitk::ImageGenerator::GenerateRandomImage<float>(100, 100, 100, 1, 0.2, 0.3, 0.4);
58 m_RandomRestImage2 = mitk::ImageGenerator::GenerateRandomImage<float>(100, 100, 100, 1, 0.2, 0.3, 0.4);
59 m_RandomSingleSliceImage = mitk::ImageGenerator::GenerateRandomImage<float>(100, 100, 1, 1, 0.2, 0.3, 0.4);
60 m_RealTestImage = mitk::IOUtil::Load<mitk::Image>(GetTestDataFilePath("Pic3D.nrrd"));
61 }
62
63 void tearDown() override
64 {
65 m_TestFilter = nullptr;
66 m_RandomRestImage1 = nullptr;
67 m_RandomRestImage2 = nullptr;
68 m_RealTestImage = nullptr;
69 m_RandomSingleSliceImage = nullptr;
70 }
71
72 void TestInstantiation()
73 {
74 CPPUNIT_ASSERT_MESSAGE("Testing instantiation",m_TestFilter.IsNotNull());
75 }
76
77 void TestSavingValidTestImage()
78 {
79 //######################## Test with valid test images ################################
80 m_TestFilter->SetInput(m_RandomRestImage1);
81 m_TestFilter->SetInput("secondImage",m_RandomRestImage2);
82 m_TestFilter->Update();
83 MITK_TEST_OUTPUT(<<"Tested method Update() with valid data.");
84 std::vector<std::string> filenames;
85 std::string csvFileName;
86 m_TestFilter->SaveImages(m_TemporaryTestDirectory,filenames,csvFileName);
87 MITK_TEST_OUTPUT(<<"Tested method SaveImages(...).");
88 CPPUNIT_ASSERT_MESSAGE("Testing if correct number of images was saved",filenames.size() == 1);
89 CPPUNIT_ASSERT_MESSAGE("Testing if image file exists",Poco::File(filenames.at(0).c_str()).exists());
90 CPPUNIT_ASSERT_MESSAGE("Testing if csv file exists",Poco::File(csvFileName.c_str()).exists());
91
92 //clean up
93 std::remove(filenames.at(0).c_str());
94 std::remove(csvFileName.c_str());
95 }
96
97 void TestSavingAfterMupltipleUpdateCalls()
98 {
99 //######################## Test multiple calls of update ################################
100 m_TestFilter->SetInput(m_RandomRestImage1);
101 m_TestFilter->SetInput("secondImage",m_RandomRestImage2);
102
103 for(int i=0; i<5; i++)
104 {
105 m_TestFilter->Update();
106 std::stringstream testmessage;
107 testmessage << "testmessage" << i;
108 m_TestFilter->AddMessageToCurrentImage(testmessage.str());
109 itksys::SystemTools::Delay(50);
110 }
111 MITK_TEST_OUTPUT(<<"Call Update() 5 times.");
112
113 std::vector<std::string> filenames;
114 std::string csvFileName;
115 m_TestFilter->SaveImages(m_TemporaryTestDirectory,filenames,csvFileName);
116 MITK_TEST_OUTPUT(<<"Tested method SaveImages(...).");
117 CPPUNIT_ASSERT_MESSAGE("Testing if correct number of images was saved",filenames.size() == 5);
118 CPPUNIT_ASSERT_MESSAGE("Testing if file 1 exists",Poco::File(filenames.at(0).c_str()).exists());
119 CPPUNIT_ASSERT_MESSAGE("Testing if file 2 exists",Poco::File(filenames.at(1).c_str()).exists());
120 CPPUNIT_ASSERT_MESSAGE("Testing if file 3 exists",Poco::File(filenames.at(2).c_str()).exists());
121 CPPUNIT_ASSERT_MESSAGE("Testing if file 4 exists",Poco::File(filenames.at(3).c_str()).exists());
122 CPPUNIT_ASSERT_MESSAGE("Testing if file 5 exists",Poco::File(filenames.at(4).c_str()).exists());
123 CPPUNIT_ASSERT_MESSAGE("Testing if csv file exists",Poco::File(csvFileName.c_str()).exists());
124
125 //clean up
126 for(size_t i=0; i<filenames.size(); i++) std::remove(filenames.at(i).c_str());
127 std::remove(csvFileName.c_str());
128 }
129
130 void TestFilterWithEmptyImages()
131 {
132 //create empty test images
133 mitk::Image::Pointer testImage = mitk::Image::New();
134 mitk::Image::Pointer testImage2 = mitk::Image::New();
135
136 //set both as input for the filter
137 m_TestFilter->SetInput(testImage);
138 CPPUNIT_ASSERT_MESSAGE("Testing SetInput(...) for first input.",m_TestFilter->GetNumberOfInputs()==1);
139 m_TestFilter->SetInput("secondImage",testImage2);
140 CPPUNIT_ASSERT_MESSAGE("Testing SetInput(...) for second input.",m_TestFilter->GetNumberOfInputs()==2);
141
142 //images are empty, but update method should not crash
143 CPPUNIT_ASSERT_NO_THROW_MESSAGE("Tested method Update() with invalid data.",m_TestFilter->Update());
144 }
145
146 void TestFilterWithInvalidPath()
147 {
148 #ifdef WIN32
149 std::string filename = "XV:/342INVALID<>"; //invalid filename for windows
150 #else
151 std::string filename = "/dsfdsf:$�$342INVALID"; //invalid filename for linux
152 #endif
153
154 m_TestFilter->SetInput(m_RealTestImage);
155 m_TestFilter->Update();
156 CPPUNIT_ASSERT_THROW_MESSAGE("Testing if correct exception if thrown if an invalid path is given.",
157 m_TestFilter->SaveImages(filename),
158 mitk::Exception);
159 }
160
161 void TestJpgFileExtension()
162 {
163 CPPUNIT_ASSERT_MESSAGE("Testing setting of jpg extension.",m_TestFilter->SetImageFilesExtension(".jpg"));
164 m_TestFilter->SetInput(m_RandomSingleSliceImage);
165 m_TestFilter->Update();
166
167 std::vector<std::string> filenames;
168 std::string csvFileName;
169 m_TestFilter->SaveImages(m_TemporaryTestDirectory,filenames,csvFileName);
170 CPPUNIT_ASSERT_MESSAGE("Testing if correct number of images was saved",filenames.size() == 1);
171 CPPUNIT_ASSERT_MESSAGE("Testing if jpg image file exists",Poco::File(filenames.at(0).c_str()).exists());
172 CPPUNIT_ASSERT_MESSAGE("Testing if csv file exists",Poco::File(csvFileName.c_str()).exists());
173
174 //clean up
175 std::remove(filenames.at(0).c_str());
176 std::remove(csvFileName.c_str());
177 }
178
179 void TestSetFileExtension()
180 {
181 CPPUNIT_ASSERT_MESSAGE("Testing if PIC extension can be set.",m_TestFilter->SetImageFilesExtension("PIC"));
182 CPPUNIT_ASSERT_MESSAGE("Testing if bmp extension can be set.",m_TestFilter->SetImageFilesExtension("bmp"));
183 CPPUNIT_ASSERT_MESSAGE("Testing if gdc extension can be set.",m_TestFilter->SetImageFilesExtension("gdcm"));
184 CPPUNIT_ASSERT_MESSAGE("Testing if dcm extension can be set.",m_TestFilter->SetImageFilesExtension("dcm"));
185 CPPUNIT_ASSERT_MESSAGE("Testing if dc3 extension can be set.",m_TestFilter->SetImageFilesExtension("dc3"));
186 CPPUNIT_ASSERT_MESSAGE("Testing if ima extension can be set.",m_TestFilter->SetImageFilesExtension(".ima"));
187 CPPUNIT_ASSERT_MESSAGE("Testing if img extension can be set.",m_TestFilter->SetImageFilesExtension("img"));
188 CPPUNIT_ASSERT_MESSAGE("Testing if gip extension can be set.",m_TestFilter->SetImageFilesExtension("gipl"));
189 CPPUNIT_ASSERT_MESSAGE("Testing if gipl.gz extension can be set.",m_TestFilter->SetImageFilesExtension(".gipl.gz"));
190 CPPUNIT_ASSERT_MESSAGE("Testing if jpg extension can be set.",m_TestFilter->SetImageFilesExtension("jpg"));
191 CPPUNIT_ASSERT_MESSAGE("Testing if jpe extension can be set.",m_TestFilter->SetImageFilesExtension("jpeg"));
192 CPPUNIT_ASSERT_MESSAGE("Testing if pic extension can be set.",m_TestFilter->SetImageFilesExtension("pic"));
193 }
194
195 void TestSetWrongFileExtension()
196 {
197
198 CPPUNIT_ASSERT_MESSAGE("Testing if wrong obj extension is recognized",!m_TestFilter->SetImageFilesExtension("obj "));
199 CPPUNIT_ASSERT_MESSAGE("Testing if wrong stl extension is recognized",!m_TestFilter->SetImageFilesExtension("stl "));
200 CPPUNIT_ASSERT_MESSAGE("Testing if wrong pvtp extension is recognized",!m_TestFilter->SetImageFilesExtension("pvtp"));
201 CPPUNIT_ASSERT_MESSAGE("Testing if wrong vtp extension is recognized",!m_TestFilter->SetImageFilesExtension("vtp "));
202 CPPUNIT_ASSERT_MESSAGE("Testing if wrong vtk extension is recognized",!m_TestFilter->SetImageFilesExtension("vtk "));
203
204 }
205
206};
207
208MITK_TEST_SUITE_REGISTRATION(mitkUSImageLoggingFilter)