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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkXMLCompositeDataSetWriterHelper.h"
#include "vtkDataObject.h"
#include "vtkLogger.h"
#include "vtkObjectFactory.h"
#include "vtkXMLDataObjectWriter.h"
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkXMLCompositeDataSetWriterHelper);
vtkCxxSetObjectMacro(vtkXMLCompositeDataSetWriterHelper, Writer, vtkXMLWriterBase);
//----------------------------------------------------------------------------
vtkXMLCompositeDataSetWriterHelper::vtkXMLCompositeDataSetWriterHelper()
: Writer(nullptr)
{
}
//----------------------------------------------------------------------------
vtkXMLCompositeDataSetWriterHelper::~vtkXMLCompositeDataSetWriterHelper()
{
this->SetWriter(nullptr);
}
//----------------------------------------------------------------------------
std::string vtkXMLCompositeDataSetWriterHelper::WriteDataSet(
const std::string& path, const std::string& prefix, vtkDataObject* data)
{
if (!data)
{
return {};
}
bool isEmpty = true;
for (int attrType = vtkDataObject::POINT;
attrType < vtkDataObject::NUMBER_OF_ATTRIBUTE_TYPES && isEmpty; ++attrType)
{
isEmpty = (data->GetNumberOfElements(attrType) == 0);
}
if (isEmpty)
{
return {};
}
auto writer = this->GetWriter(data->GetDataObjectType());
if (!writer)
{
vtkLogF(WARNING, "skipping dataset of unsupported type '%s'.", data->GetClassName());
return {};
}
std::string fname = prefix + "." + writer->GetDefaultFileExtension();
writer->SetInputDataObject(data);
writer->SetFileName(path.empty() ? fname.c_str() : (path + "/" + fname).c_str());
writer->Write();
writer->SetInputDataObject(nullptr);
vtkLogF(TRACE, "wrote leaf %s", fname.c_str());
return fname;
}
//----------------------------------------------------------------------------
vtkXMLWriterBase* vtkXMLCompositeDataSetWriterHelper::GetWriter(int dataType)
{
auto iter = this->WriterCache.find(dataType);
if (iter != this->WriterCache.end())
{
return iter->second;
}
if (auto writer = vtkXMLDataObjectWriter::NewWriter(dataType))
{
// Copy settings to the writer.
writer->SetDebug(this->Writer->GetDebug());
writer->SetByteOrder(this->Writer->GetByteOrder());
writer->SetCompressor(this->Writer->GetCompressor());
writer->SetBlockSize(this->Writer->GetBlockSize());
writer->SetDataMode(this->Writer->GetDataMode());
writer->SetEncodeAppendedData(this->Writer->GetEncodeAppendedData());
writer->SetHeaderType(this->Writer->GetHeaderType());
writer->SetIdType(this->Writer->GetIdType());
writer->SetWriteTimeValue(this->Writer->GetWriteTimeValue());
this->WriterCache[dataType].TakeReference(writer);
return writer;
}
return nullptr;
}
//----------------------------------------------------------------------------
void vtkXMLCompositeDataSetWriterHelper::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
VTK_ABI_NAMESPACE_END
|
