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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkOpenFOAMReader.h"
#include "vtkCellData.h"
#include "vtkCompositeDataSet.h"
#include "vtkDataSetMapper.h"
#include "vtkInformation.h"
#include "vtkMultiBlockDataSet.h"
#include "vtkPointData.h"
#include "vtkPolyData.h"
#include "vtkSmartPointer.h"
#include "vtkTestUtilities.h"
#include "vtkUnstructuredGrid.h"
namespace
{
// Get named block of specified type
template <class Type>
Type* findBlock(vtkMultiBlockDataSet* mb, const char* blockName)
{
Type* dataset = nullptr;
const unsigned int nblocks = (mb ? mb->GetNumberOfBlocks() : 0u);
for (unsigned int blocki = 0; !dataset && blocki < nblocks; ++blocki)
{
vtkDataObject* obj = mb->GetBlock(blocki);
if (strcmp(mb->GetMetaData(blocki)->Get(vtkCompositeDataSet::NAME()), blockName) == 0)
{
dataset = Type::SafeDownCast(obj);
}
if (!dataset)
{
dataset = findBlock<Type>(vtkMultiBlockDataSet::SafeDownCast(obj), blockName);
}
}
return dataset;
}
} // End anonymous namespace
int TestOpenFOAMReaderLagrangianSerial(int argc, char* argv[])
{
// Read file name.
char* filename = vtkTestUtilities::ExpandDataFileName(
argc, argv, "Data/OpenFOAM/simplifiedSiwek-serial/simplifiedSiwek-serial.foam");
// Read the file
vtkNew<vtkOpenFOAMReader> reader;
reader->SetFileName(filename);
delete[] filename;
reader->Update();
reader->SetTimeValue(0.005);
// Re-read with everything selected
reader->EnableAllPatchArrays();
reader->Update();
reader->Print(std::cout);
// reader->GetOutput()->Print(std::cout);
auto* allBlocks = vtkMultiBlockDataSet::SafeDownCast(reader->GetOutput());
auto* lagrangianBlocks = findBlock<vtkMultiBlockDataSet>(allBlocks, "lagrangian");
if (!lagrangianBlocks)
{
std::cout << "No lagrangian blocks!\n";
return 1;
}
long nClouds = 0;
long nParticles = 0;
const int nLagrangianFields = reader->GetNumberOfLagrangianArrays();
std::cout << "----- Have " << nLagrangianFields << " Lagrangian fields" << std::endl;
// Bad name, but it is what we have
for (int i = 0; i < reader->GetNumberOfPatchArrays(); ++i)
{
std::string displayName(reader->GetPatchArrayName(i));
auto slash = displayName.rfind('/');
if (slash != std::string::npos && displayName.compare(0, ++slash, "lagrangian/") == 0)
{
std::string cloudName(displayName.substr(slash));
std::cout << " Display " << displayName << " = Cloud <" << cloudName << ">" << std::endl;
auto* cloudData = findBlock<vtkPolyData>(lagrangianBlocks, cloudName.c_str());
if (cloudData)
{
++nClouds;
nParticles += cloudData->GetNumberOfPoints();
}
}
}
std::cout << " Read " << nParticles << " particles from " << nClouds << " clouds" << std::endl;
int retVal = (nParticles != 0);
return !retVal;
}
|
