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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright (c) Sandia Corporation
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkPExodusIIWriter.h"
#include "vtkArrayIteratorIncludes.h"
#include "vtkCellArray.h"
#include "vtkCellData.h"
#include "vtkCompositeDataIterator.h"
#include "vtkCompositeDataSet.h"
#include "vtkDataObject.h"
#include "vtkDoubleArray.h"
#include "vtkFieldData.h"
#include "vtkFloatArray.h"
#include "vtkIdList.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkIntArray.h"
#include "vtkModelMetadata.h"
#include "vtkObjectFactory.h"
#include "vtkPointData.h"
#include "vtkStreamingDemandDrivenPipeline.h"
#include "vtkThreshold.h"
#include "vtkUnstructuredGrid.h"
#include "vtkMultiProcessController.h"
#include "vtk_exodusII.h"
#include <cctype>
#include <ctime>
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkPExodusIIWriter);
//------------------------------------------------------------------------------
vtkPExodusIIWriter::vtkPExodusIIWriter() = default;
vtkPExodusIIWriter::~vtkPExodusIIWriter() = default;
void vtkPExodusIIWriter::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
//------------------------------------------------------------------------------
int vtkPExodusIIWriter::CheckParameters()
{
vtkMultiProcessController* c = vtkMultiProcessController::GetGlobalController();
int numberOfProcesses = c ? c->GetNumberOfProcesses() : 1;
int myRank = c ? c->GetLocalProcessId() : 0;
if (this->GhostLevel > 0)
{
vtkWarningMacro(<< "ExodusIIWriter ignores ghost level request");
}
return this->Superclass::CheckParametersInternal(numberOfProcesses, myRank);
}
//------------------------------------------------------------------------------
int vtkPExodusIIWriter::RequestUpdateExtent(
vtkInformation* request, vtkInformationVector** inputVector, vtkInformationVector* outputVector)
{
this->Superclass::RequestUpdateExtent(request, inputVector, outputVector);
vtkMultiProcessController* c = vtkMultiProcessController::GetGlobalController();
if (c)
{
int numberOfProcesses = c->GetNumberOfProcesses();
int myRank = c->GetLocalProcessId();
vtkInformation* info = inputVector[0]->GetInformationObject(0);
info->Set(vtkStreamingDemandDrivenPipeline::UPDATE_PIECE_NUMBER(), myRank);
info->Set(vtkStreamingDemandDrivenPipeline::UPDATE_NUMBER_OF_PIECES(), numberOfProcesses);
}
return 1;
}
void vtkPExodusIIWriter::CheckBlockInfoMap()
{
// if we're multiprocess we need to make sure the block info map matches
if (this->NumberOfProcesses > 1)
{
int maxId = -1;
std::map<int, Block>::const_iterator iter;
for (iter = this->BlockInfoMap.begin(); iter != this->BlockInfoMap.end(); ++iter)
{
if (iter->first > maxId)
{
maxId = iter->first;
}
}
vtkMultiProcessController* c = vtkMultiProcessController::GetGlobalController();
int globalMaxId;
c->AllReduce(&maxId, &globalMaxId, 1, vtkCommunicator::MAX_OP);
maxId = globalMaxId;
for (int i = 1; i <= maxId; i++)
{
Block& b = this->BlockInfoMap[i]; // ctor called (init all to 0/-1) if not preset
int globalType;
c->AllReduce(&b.Type, &globalType, 1, vtkCommunicator::MAX_OP);
if (b.Type != 0 && b.Type != globalType)
{
vtkWarningMacro(<< "The type associated with ID's across processors doesn't match");
}
else
{
b.Type = globalType;
}
int globalNodes;
c->AllReduce(&b.NodesPerElement, &globalNodes, 1, vtkCommunicator::MAX_OP);
if (b.NodesPerElement != globalNodes &&
// on a processor with no data, b.NodesPerElement == 0.
b.NodesPerElement != 0)
{
vtkWarningMacro(<< "NodesPerElement associated with ID's across "
"processors doesn't match: "
<< b.NodesPerElement << " != " << globalNodes);
}
else
{
b.NodesPerElement = globalNodes;
}
}
}
}
//------------------------------------------------------------------------------
int vtkPExodusIIWriter::GlobalContinueExecuting(int localContinue)
{
vtkMultiProcessController* c = vtkMultiProcessController::GetGlobalController();
int globalContinue = localContinue;
if (c)
{
c->AllReduce(&localContinue, &globalContinue, 1, vtkCommunicator::MIN_OP);
}
return globalContinue;
}
//------------------------------------------------------------------------------
unsigned int vtkPExodusIIWriter::GetMaxNameLength()
{
unsigned int maxName = this->Superclass::GetMaxNameLength();
vtkMultiProcessController* c = vtkMultiProcessController::GetGlobalController();
unsigned int globalMaxName = 0;
c->AllReduce(&maxName, &globalMaxName, 1, vtkCommunicator::MAX_OP);
return maxName;
}
VTK_ABI_NAMESPACE_END
|
