171 lines
5.3 KiB
C++
171 lines
5.3 KiB
C++
//============================================================================
|
|
// Copyright (c) Kitware, Inc.
|
|
// All rights reserved.
|
|
// See LICENSE.txt for details.
|
|
//
|
|
// This software is distributed WITHOUT ANY WARRANTY; without even
|
|
// the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
// PURPOSE. See the above copyright notice for more information.
|
|
//============================================================================
|
|
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
#include <vtkm/io/ImageReaderPNG.h>
|
|
#include <vtkm/io/ImageReaderPNM.h>
|
|
#include <vtkm/io/ImageWriterPNG.h>
|
|
#include <vtkm/io/ImageWriterPNM.h>
|
|
#include <vtkm/rendering/Canvas.h>
|
|
#include <vtkm/rendering/Color.h>
|
|
|
|
#include <string>
|
|
|
|
namespace
|
|
{
|
|
|
|
using namespace vtkm::io;
|
|
using namespace vtkm::rendering;
|
|
|
|
void TestFilledImage(vtkm::cont::DataSet& dataSet,
|
|
const std::string& fieldName,
|
|
const vtkm::rendering::Canvas& canvas)
|
|
{
|
|
VTKM_TEST_ASSERT(dataSet.HasPointField(fieldName), "Point Field Not Found: " + fieldName);
|
|
|
|
auto pointField = dataSet.GetPointField(fieldName);
|
|
VTKM_TEST_ASSERT(pointField.GetNumberOfValues() == canvas.GetWidth() * canvas.GetHeight(),
|
|
"wrong image dimensions");
|
|
VTKM_TEST_ASSERT(pointField.GetData().template IsType<vtkm::cont::ArrayHandle<vtkm::Vec4f_32>>(),
|
|
"wrong ArrayHandle type");
|
|
auto pixelPortal = pointField.GetData()
|
|
.template AsArrayHandle<vtkm::cont::ArrayHandle<vtkm::Vec4f_32>>()
|
|
.ReadPortal();
|
|
|
|
auto colorPortal = canvas.GetColorBuffer().ReadPortal();
|
|
|
|
VTKM_TEST_ASSERT(test_equal_portals(pixelPortal, colorPortal));
|
|
}
|
|
|
|
void TestCreateImageDataSet(const vtkm::rendering::Canvas& canvas)
|
|
{
|
|
std::cout << "TestCreateImageDataSet" << std::endl;
|
|
auto dataSet = canvas.GetDataSet("pixel-color");
|
|
TestFilledImage(dataSet, "pixel-color", canvas);
|
|
}
|
|
|
|
void TestReadAndWritePNG(const vtkm::rendering::Canvas& canvas,
|
|
std::string filename,
|
|
vtkm::io::ImageWriterBase::PixelDepth pixelDepth)
|
|
{
|
|
std::cout << "TestReadAndWritePNG - " << filename << std::endl;
|
|
bool throws = false;
|
|
try
|
|
{
|
|
vtkm::io::ImageWriterPNG writer(filename);
|
|
vtkm::cont::DataSet dataSet;
|
|
writer.WriteDataSet(dataSet);
|
|
}
|
|
catch (const vtkm::cont::Error&)
|
|
{
|
|
throws = true;
|
|
}
|
|
VTKM_TEST_ASSERT(throws, "Fill Image did not throw with empty data");
|
|
|
|
{
|
|
vtkm::io::ImageWriterPNG writer(filename);
|
|
writer.SetPixelDepth(pixelDepth);
|
|
writer.WriteDataSet(canvas.GetDataSet());
|
|
}
|
|
{
|
|
vtkm::io::ImageReaderPNG reader(filename);
|
|
vtkm::cont::DataSet dataSet = reader.ReadDataSet();
|
|
}
|
|
{
|
|
vtkm::io::ImageWriterPNG writer(filename);
|
|
writer.SetPixelDepth(pixelDepth);
|
|
writer.WriteDataSet(canvas.GetDataSet());
|
|
}
|
|
{
|
|
vtkm::io::ImageReaderPNG reader(filename);
|
|
vtkm::cont::DataSet dataSet = reader.ReadDataSet();
|
|
TestFilledImage(dataSet, reader.GetPointFieldName(), canvas);
|
|
}
|
|
}
|
|
|
|
void TestReadAndWritePNM(const vtkm::rendering::Canvas& canvas,
|
|
std::string filename,
|
|
vtkm::io::ImageWriterBase::PixelDepth pixelDepth)
|
|
{
|
|
std::cout << "TestReadAndWritePNM - " << filename << std::endl;
|
|
bool throws = false;
|
|
try
|
|
{
|
|
vtkm::io::ImageWriterPNM writer(filename);
|
|
vtkm::cont::DataSet dataSet;
|
|
writer.WriteDataSet(dataSet);
|
|
}
|
|
catch (const vtkm::cont::Error&)
|
|
{
|
|
throws = true;
|
|
}
|
|
VTKM_TEST_ASSERT(throws, "Fill Image did not throw with empty data");
|
|
|
|
{
|
|
vtkm::io::ImageWriterPNM writer(filename);
|
|
writer.SetPixelDepth(pixelDepth);
|
|
writer.WriteDataSet(canvas.GetDataSet());
|
|
}
|
|
{
|
|
vtkm::io::ImageReaderPNM reader(filename);
|
|
vtkm::cont::DataSet dataSet = reader.ReadDataSet();
|
|
}
|
|
{
|
|
vtkm::io::ImageWriterPNM writer(filename);
|
|
writer.SetPixelDepth(pixelDepth);
|
|
writer.WriteDataSet(canvas.GetDataSet());
|
|
}
|
|
{
|
|
vtkm::io::ImageReaderPNM reader(filename);
|
|
vtkm::cont::DataSet dataSet = reader.ReadDataSet();
|
|
TestFilledImage(dataSet, reader.GetPointFieldName(), canvas);
|
|
}
|
|
}
|
|
|
|
void TestBaseImageMethods(const vtkm::rendering::Canvas& canvas)
|
|
{
|
|
TestCreateImageDataSet(canvas);
|
|
}
|
|
|
|
void TestPNMImage(const vtkm::rendering::Canvas& canvas)
|
|
{
|
|
TestReadAndWritePNM(canvas, "pnmRGB8Test.png", vtkm::io::ImageWriterBase::PixelDepth::PIXEL_8);
|
|
TestReadAndWritePNM(canvas, "pnmRGB16Test.png", vtkm::io::ImageWriterBase::PixelDepth::PIXEL_16);
|
|
}
|
|
|
|
void TestPNGImage(const vtkm::rendering::Canvas& canvas)
|
|
{
|
|
TestReadAndWritePNG(canvas, "pngRGB8Test.png", vtkm::io::ImageWriterBase::PixelDepth::PIXEL_8);
|
|
TestReadAndWritePNG(canvas, "pngRGB16Test.png", vtkm::io::ImageWriterBase::PixelDepth::PIXEL_16);
|
|
}
|
|
|
|
void TestImage()
|
|
{
|
|
vtkm::rendering::Canvas canvas(16, 16);
|
|
canvas.SetBackgroundColor(vtkm::rendering::Color::red);
|
|
canvas.Clear();
|
|
// Line from top left to bottom right, ensures correct transposedness
|
|
canvas.AddLine(-0.9, 0.9, 0.9, -0.9, 2.0f, vtkm::rendering::Color::black);
|
|
vtkm::Bounds colorBarBounds(-0.8, -0.6, -0.8, 0.8, 0, 0);
|
|
canvas.AddColorBar(colorBarBounds, vtkm::cont::ColorTable("inferno"), false);
|
|
canvas.BlendBackground();
|
|
canvas.SaveAs("baseline.ppm");
|
|
|
|
TestBaseImageMethods(canvas);
|
|
TestPNMImage(canvas);
|
|
TestPNGImage(canvas);
|
|
}
|
|
}
|
|
|
|
int UnitTestImageWriter(int argc, char* argv[])
|
|
{
|
|
return vtkm::cont::testing::Testing::Run(TestImage, argc, argv);
|
|
}
|