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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
// Description:
// This tests reading of a simple ADIOS2 bp file.
#include "vtkADIOS2CoreImageReader.h"
#include "vtkActor.h"
#include "vtkAlgorithm.h"
#include "vtkCamera.h"
#include "vtkCellData.h"
#include "vtkColorTransferFunction.h"
#include "vtkCompositeRenderManager.h"
#include "vtkDataArray.h"
#include "vtkDataSetMapper.h"
#include "vtkExecutive.h"
#include "vtkImageData.h"
#include "vtkImageDataToPointSet.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkMultiBlockDataSet.h"
#include "vtkMultiBlockVolumeMapper.h"
#include "vtkMultiPieceDataSet.h"
#include "vtkPointData.h"
#include "vtkPolyDataMapper.h"
#include "vtkProperty.h"
#include "vtkRegressionTestImage.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkRenderer.h"
#include "vtkStreamingDemandDrivenPipeline.h"
#include "vtkUnsignedIntArray.h"
#include "vtkVolume.h"
#include "vtkVolumeProperty.h"
#include "vtkXMLPMultiBlockDataWriter.h"
#include "vtkNew.h"
#include "vtkTestUtilities.h"
#include <sstream> // istringstream
int TestADIOS2BPReaderMultiTimeSteps3D(int argc, char* argv[])
{
vtkNew<vtkADIOS2CoreImageReader> reader;
// Read the input data file
char* filePath =
vtkTestUtilities::ExpandDataFileName(argc, argv, "Data/ADIOS2/3D_7-point_24-step/gs.bp");
if (!reader->CanReadFile(filePath))
{
std::cerr << "Cannot read file " << reader->GetFileName() << std::endl;
return 0;
}
reader->SetFileName(filePath);
delete[] filePath;
reader->UpdateInformation();
auto& availVars = reader->GetAvilableVariables();
assert(availVars.size() == 3);
// Get the dimension
std::string varName = availVars.begin()->first;
// Enable multi time stesp
reader->SetTimeStepArray("step");
reader->SetDimensionArray("U");
reader->SetArrayStatus("step", false);
reader->SetActiveScalar(std::make_pair("U", vtkADIOS2CoreImageReader::VarType::CellData));
reader->Update();
vtkSmartPointer<vtkMultiBlockDataSet> output =
vtkMultiBlockDataSet::SafeDownCast(reader->GetOutput());
assert(output->GetNumberOfBlocks() == 1);
vtkSmartPointer<vtkMultiPieceDataSet> mpds =
vtkMultiPieceDataSet::SafeDownCast(output->GetBlock(0));
assert(mpds->GetNumberOfPieces() == 6);
vtkSmartPointer<vtkImageData> image0 = vtkImageData::SafeDownCast(mpds->GetPiece(0));
vtkSmartPointer<vtkImageData> image1 = vtkImageData::SafeDownCast(mpds->GetPiece(1));
// Use vtkXMLPMultiBlockDataWriter + vtkStreamingDemandDrivenPipeline::UPDATE_TIME_STEP()
// to write out the data if needed
vtkNew<vtkImageDataToPointSet> imageToPointset;
assert(image0->GetCellData()->GetNumberOfArrays() == 2);
image0->GetCellData()->SetActiveScalars("U");
imageToPointset->SetInputData(image0);
imageToPointset->Update();
// Since I fail to find a proper mapper to render two vtkImageDatas inside
// a vtkMultiPieceDataSet in a vtkMultiBlockDataSet, I render the image directly here
vtkNew<vtkDataSetMapper> mapper;
mapper->SetInputDataObject(imageToPointset->GetOutput());
mapper->ScalarVisibilityOn();
mapper->SetScalarRange(0, 2000);
mapper->SetScalarModeToUseCellData();
mapper->ColorByArrayComponent("U", 0);
vtkNew<vtkActor> actor;
actor->SetMapper(mapper);
actor->GetProperty()->EdgeVisibilityOn();
vtkNew<vtkRenderer> renderer;
renderer->AddActor(actor);
renderer->SetBackground(0.5, 0.5, 0.5);
renderer->GetActiveCamera()->Elevation(300);
renderer->GetActiveCamera()->Yaw(60);
renderer->ResetCamera();
vtkNew<vtkRenderWindow> rendWin;
rendWin->SetSize(600, 300);
rendWin->AddRenderer(renderer);
rendWin->Render();
// Do the test comparison
int retval = vtkRegressionTestImage(rendWin);
if (retval == vtkRegressionTester::DO_INTERACTOR)
{
vtkNew<vtkRenderWindowInteractor> iren;
iren->SetRenderWindow(rendWin);
iren->Initialize();
iren->Start();
retval = vtkRegressionTester::PASSED;
}
return !retval;
}
|
