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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkPConvertToMultiBlockDataSet.h"
#include "vtkCommunicator.h"
#include "vtkMultiBlockDataSet.h"
#include "vtkMultiProcessController.h"
#include "vtkObjectFactory.h"
#include "vtkPartitionedDataSet.h"
#include "vtkPartitionedDataSetCollection.h"
#include "vtkSmartPointer.h"
VTK_ABI_NAMESPACE_BEGIN
vtkObjectFactoryNewMacro(vtkPConvertToMultiBlockDataSet);
vtkCxxSetObjectMacro(vtkPConvertToMultiBlockDataSet, Controller, vtkMultiProcessController);
//----------------------------------------------------------------------------
vtkPConvertToMultiBlockDataSet::vtkPConvertToMultiBlockDataSet()
: Controller(nullptr)
{
this->SetController(vtkMultiProcessController::GetGlobalController());
}
//----------------------------------------------------------------------------
vtkPConvertToMultiBlockDataSet::~vtkPConvertToMultiBlockDataSet()
{
this->SetController(nullptr);
}
//----------------------------------------------------------------------------
int vtkPConvertToMultiBlockDataSet::RequestData(
vtkInformation* request, vtkInformationVector** inputVector, vtkInformationVector* outputVector)
{
vtkSmartPointer<vtkPartitionedDataSetCollection> input;
if (auto pdc = vtkPartitionedDataSetCollection::GetData(inputVector[0], 0))
{
input = pdc;
}
else if (auto pd = vtkPartitionedDataSet::GetData(inputVector[0], 0))
{
input = vtkSmartPointer<vtkPartitionedDataSetCollection>::New();
input->SetPartitionedDataSet(0, pd);
}
const int numRanks = this->Controller ? this->Controller->GetNumberOfProcesses() : 1;
if (numRanks == 1 || input == nullptr || input->GetNumberOfPartitionedDataSets() == 0)
{
// nothing to do.
return this->Superclass::RequestData(request, inputVector, outputVector);
}
auto output = vtkMultiBlockDataSet::GetData(outputVector, 0);
// ensure that we have exactly the same number of partitions on all ranks.
vtkNew<vtkPartitionedDataSetCollection> clone;
clone->CompositeShallowCopy(input);
const auto count = input->GetNumberOfPartitionedDataSets();
std::vector<unsigned int> piece_counts(count);
for (unsigned int cc = 0; cc < count; ++cc)
{
piece_counts[cc] = clone->GetPartitionedDataSet(cc)
? clone->GetPartitionedDataSet(cc)->GetNumberOfPartitions()
: 0;
}
std::vector<unsigned int> result(piece_counts.size());
this->Controller->AllReduce(
piece_counts.data(), result.data(), static_cast<vtkIdType>(count), vtkCommunicator::MAX_OP);
for (unsigned int cc = 0; cc < count; ++cc)
{
if (result[cc] > 0)
{
if (clone->GetPartitionedDataSet(cc) == nullptr)
{
clone->SetPartitionedDataSet(cc, vtkNew<vtkPartitionedDataSet>{});
}
clone->GetPartitionedDataSet(cc)->SetNumberOfPartitions(result[cc]);
}
}
return this->Execute(clone, output) ? 1 : 0;
}
//----------------------------------------------------------------------------
void vtkPConvertToMultiBlockDataSet::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
VTK_ABI_NAMESPACE_END
|
