1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestPNGReaderReadFromMemory.cxx
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm 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 "vtkImageData.h"
#include "vtkImageViewer.h"
#include "vtkPNGReader.h"
#include "vtkRegressionTestImage.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkRenderer.h"
#include "vtksys/FStream.hxx"
#include "vtksys/SystemTools.hxx"
#include <fstream>
#include <vector>
int TestPNGReaderReadFromMemory(int argc, char* argv[])
{
if (argc <= 1)
{
cout << "Usage: " << argv[0] << " <png file>" << endl;
return EXIT_FAILURE;
}
std::string filename = argv[1];
// Open the file
vtksys::ifstream stream(filename.c_str(), std::ios::in | std::ios::binary);
if (!stream.is_open())
{
std::cerr << "Could not open file " << filename << std::endl;
}
// Get file size
unsigned long len = vtksys::SystemTools::FileLength(filename);
// Load the file into a buffer
std::vector<char> buffer = std::vector<char>(std::istreambuf_iterator<char>(stream), {});
// Initialize reader
vtkNew<vtkPNGReader> pngReader;
pngReader->SetMemoryBuffer(buffer.data());
pngReader->SetMemoryBufferLength(len);
// Visualize
vtkNew<vtkImageViewer> imageViewer;
imageViewer->SetInputConnection(pngReader->GetOutputPort());
imageViewer->SetColorWindow(256);
imageViewer->SetColorLevel(127.5);
vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
imageViewer->SetupInteractor(renderWindowInteractor);
imageViewer->Render();
vtkRenderWindow* renWin = imageViewer->GetRenderWindow();
int retVal = vtkRegressionTestImage(renWin);
if (retVal == vtkRegressionTester::DO_INTERACTOR)
{
renderWindowInteractor->Start();
}
return !retVal;
}
|
