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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
|
/*=========================================================================
Program: Visualization Toolkit
Module: VTXSchemaManager.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.
=========================================================================*/
/*
* VTXSchemaManager.cxx
*
* Created on: May 31, 2019
* Author: William F Godoy godoywf@ornl.gov
*/
#include "VTXSchemaManager.h"
#include "schema/vtk/VTXvtkVTI.h"
#include "schema/vtk/VTXvtkVTU.h"
#include <vtk_pugixml.h>
#include <vtksys/SystemTools.hxx>
#include "common/VTXHelper.h"
namespace vtx
{
VTK_ABI_NAMESPACE_BEGIN
// PUBLIC
void VTXSchemaManager::Update(
const std::string& streamName, const size_t /*step*/, const std::string& schemaName)
{
// can't do it in the constructor as it need MPI initialized
if (!this->ADIOS)
{
this->ADIOS.reset(new adios2::ADIOS(helper::MPIGetComm()));
}
if (!this->IO && !this->Engine)
{
this->StreamName = streamName;
this->SchemaName = schemaName;
const std::string fileName = helper::GetFileName(this->StreamName);
this->IO = this->ADIOS->DeclareIO(fileName);
this->IO.SetEngine(helper::GetEngineType(fileName));
this->Engine = this->IO.Open(fileName, adios2::Mode::Read);
InitReader();
}
else
{
// TODO: check if variables changed
}
}
void VTXSchemaManager::Fill(vtkMultiBlockDataSet* multiBlock, const size_t step)
{
this->Reader->Fill(multiBlock, step);
}
// PRIVATE
const std::set<std::string> VTXSchemaManager::SupportedTypes = { "ImageData", "UnstructuredGrid" };
// TODO: , "StructuredGrid", "PolyData" };
void VTXSchemaManager::InitReader()
{
if (InitReaderXMLVTK())
{
return;
}
// else if( InitReaderOther() ) {}
// here we can make it extensible by trying to find other schema types
// for now we stick with VTK XML schemas
}
bool VTXSchemaManager::InitReaderXMLVTK()
{
pugi::xml_document xmlDocument;
std::string xmlContents;
bool isSchemaFile = false;
// check if it's file, not optimizing with MPI_Bcast
std::string xmlFileName;
if (vtksys::SystemTools::FileIsDirectory(this->Engine.Name())) // BP4
{
xmlFileName = this->Engine.Name() + "/" + this->SchemaName;
}
else if (vtksys::SystemTools::FileIsDirectory(this->Engine.Name() + ".dir")) // BP3
{
xmlFileName = this->Engine.Name() + ".dir/" + this->SchemaName;
}
if (!xmlFileName.empty())
{
if (vtksys::SystemTools::FileExists(xmlFileName))
{
xmlContents = helper::FileToString(xmlFileName);
xmlDocument =
helper::XMLDocument(xmlContents, true, "when reading " + this->SchemaName + " file");
isSchemaFile = true;
}
}
if (!isSchemaFile)
{
const adios2::Attribute<std::string> vtkXMLAttribute =
this->IO.InquireAttribute<std::string>(this->SchemaName);
if (!vtkXMLAttribute)
{
throw std::runtime_error("ERROR: neither " + this->SchemaName +
" file or bp attribute was found in " + this->Engine.Name() + "\n");
}
const std::vector<std::string> vtkAttributes = vtkXMLAttribute.Data();
xmlContents = vtkAttributes.front();
xmlDocument =
helper::XMLDocument(xmlContents, true, "when reading " + this->SchemaName + " attribute");
}
constexpr bool isDebug = true;
constexpr bool isMandatory = true;
constexpr bool isUnique = true;
const pugi::xml_node vtkXMLFileNode = helper::XMLNode("VTKFile", xmlDocument, isDebug,
"when reading VTKFile node in " + this->Engine.Name(), isMandatory, isUnique);
const pugi::xml_attribute typeXML = helper::XMLAttribute("type", vtkXMLFileNode, true,
"when reading type xml attribute in vtk.xml " + this->Engine.Name(), isMandatory);
const std::string type = std::string(typeXML.value());
if (VTXSchemaManager::SupportedTypes.count(type) == 0)
{
throw std::runtime_error("ERROR: ADIOS2Reader only supports types= " +
helper::SetToCSV(VTXSchemaManager::SupportedTypes) + " when reading type xml attribute in " +
this->SchemaName + " from " + this->Engine.Name() + "\n");
}
if (type == "ImageData")
{
this->Reader.reset(new schema::VTXvtkVTI(xmlContents, this->IO, this->Engine));
}
else if (type == "UnstructuredGrid")
{
this->Reader.reset(new schema::VTXvtkVTU(xmlContents, this->IO, this->Engine));
}
const bool success = static_cast<bool>(this->Reader);
return success;
}
VTK_ABI_NAMESPACE_END
} // end vtx namespace
|
