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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkDataSetWriter.h"
#include "vtkAlgorithmOutput.h"
#include "vtkDataSet.h"
#include "vtkErrorCode.h"
#include "vtkImageData.h"
#include "vtkInformation.h"
#include "vtkObjectFactory.h"
#include "vtkPolyData.h"
#include "vtkPolyDataWriter.h"
#include "vtkRectilinearGrid.h"
#include "vtkRectilinearGridWriter.h"
#include "vtkStructuredGrid.h"
#include "vtkStructuredGridWriter.h"
#include "vtkStructuredPointsWriter.h"
#include "vtkUnstructuredGrid.h"
#include "vtkUnstructuredGridWriter.h"
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkDataSetWriter);
void vtkDataSetWriter::WriteData()
{
int type;
vtkDataWriter* writer;
vtkAlgorithmOutput* input = this->GetInputConnection(0, 0);
vtkDebugMacro(<< "Writing vtk dataset...");
type = this->GetInput()->GetDataObjectType();
if (type == VTK_POLY_DATA)
{
vtkPolyDataWriter* pwriter = vtkPolyDataWriter::New();
pwriter->SetInputConnection(input);
writer = pwriter;
}
else if (type == VTK_STRUCTURED_POINTS || type == VTK_IMAGE_DATA || type == VTK_UNIFORM_GRID)
{
vtkStructuredPointsWriter* spwriter = vtkStructuredPointsWriter::New();
spwriter->SetInputConnection(input);
writer = spwriter;
}
else if (type == VTK_STRUCTURED_GRID)
{
vtkStructuredGridWriter* sgwriter = vtkStructuredGridWriter::New();
sgwriter->SetInputConnection(input);
writer = sgwriter;
}
else if (type == VTK_UNSTRUCTURED_GRID)
{
vtkUnstructuredGridWriter* ugwriter = vtkUnstructuredGridWriter::New();
ugwriter->SetInputConnection(input);
writer = ugwriter;
}
else if (type == VTK_RECTILINEAR_GRID)
{
vtkRectilinearGridWriter* rgwriter = vtkRectilinearGridWriter::New();
rgwriter->SetInputConnection(input);
writer = rgwriter;
}
else
{
vtkErrorMacro(<< "Cannot write dataset type: " << type);
return;
}
writer->SetFileName(this->FileName);
writer->SetScalarsName(this->ScalarsName);
writer->SetVectorsName(this->VectorsName);
writer->SetNormalsName(this->NormalsName);
writer->SetTensorsName(this->TensorsName);
writer->SetTCoordsName(this->TCoordsName);
writer->SetHeader(this->Header);
writer->SetLookupTableName(this->LookupTableName);
writer->SetFieldDataName(this->FieldDataName);
writer->SetFileType(this->FileType);
writer->SetDebug(this->Debug);
writer->SetWriteToOutputString(this->WriteToOutputString);
writer->SetFileVersion(this->FileVersion);
writer->Write();
if (writer->GetErrorCode() == vtkErrorCode::OutOfDiskSpaceError)
{
this->SetErrorCode(vtkErrorCode::OutOfDiskSpaceError);
}
if (this->WriteToOutputString)
{
delete[] this->OutputString;
this->OutputStringLength = writer->GetOutputStringLength();
this->OutputString = writer->RegisterAndGetOutputString();
}
writer->Delete();
}
int vtkDataSetWriter::FillInputPortInformation(int, vtkInformation* info)
{
info->Set(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkDataSet");
return 1;
}
vtkDataSet* vtkDataSetWriter::GetInput()
{
return vtkDataSet::SafeDownCast(this->Superclass::GetInput());
}
vtkDataSet* vtkDataSetWriter::GetInput(int port)
{
return vtkDataSet::SafeDownCast(this->Superclass::GetInput(port));
}
void vtkDataSetWriter::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
VTK_ABI_NAMESPACE_END
|
