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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright 2009 Sandia Corporation
// SPDX-License-Identifier: LicenseRef-BSD-3-Clause-Sandia-USGov
#include "vtkPBivariateLinearTableThreshold.h"
#include "vtkDataArrayCollection.h"
#include "vtkDataObject.h"
#include "vtkDoubleArray.h"
#include "vtkIdTypeArray.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkMultiProcessController.h"
#include "vtkObjectFactory.h"
#include "vtkTable.h"
#include <map>
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkPBivariateLinearTableThreshold);
vtkCxxSetObjectMacro(vtkPBivariateLinearTableThreshold, Controller, vtkMultiProcessController);
vtkPBivariateLinearTableThreshold::vtkPBivariateLinearTableThreshold()
{
this->Controller = nullptr;
this->SetController(vtkMultiProcessController::GetGlobalController());
}
vtkPBivariateLinearTableThreshold::~vtkPBivariateLinearTableThreshold()
{
this->SetController(nullptr);
}
void vtkPBivariateLinearTableThreshold::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
os << indent << "Controller: " << this->Controller << endl;
}
int vtkPBivariateLinearTableThreshold::RequestData(
vtkInformation* request, vtkInformationVector** inputVector, vtkInformationVector* outputVector)
{
this->Superclass::RequestData(request, inputVector, outputVector);
// single process?
if (!this->Controller || this->Controller->GetNumberOfProcesses() <= 1)
{
return 1;
}
vtkCommunicator* comm = this->Controller->GetCommunicator();
if (!comm)
{
vtkErrorMacro("Need a communicator.");
return 0;
}
vtkTable* outRowDataTable = vtkTable::GetData(outputVector, OUTPUT_ROW_DATA);
int numProcesses = this->Controller->GetNumberOfProcesses();
// 2) gather the selected data together
// for each column, make a new one and add it to a new table
vtkSmartPointer<vtkTable> gatheredTable = vtkSmartPointer<vtkTable>::New();
for (int i = 0; i < outRowDataTable->GetNumberOfColumns(); i++)
{
vtkAbstractArray* col = vtkArrayDownCast<vtkAbstractArray>(outRowDataTable->GetColumn(i));
if (!col)
continue;
vtkIdType myLength = col->GetNumberOfTuples();
vtkIdType totalLength = 0;
std::vector<vtkIdType> recvLengths(numProcesses, 0);
std::vector<vtkIdType> recvOffsets(numProcesses, 0);
// gathers all of the array lengths together
comm->AllGather(&myLength, recvLengths.data(), 1);
// compute the displacements
vtkIdType typeSize = col->GetDataTypeSize();
for (int j = 0; j < numProcesses; j++)
{
recvOffsets[j] = totalLength * typeSize;
totalLength += recvLengths[j];
recvLengths[j] *= typeSize;
}
// communicating this as a byte array :/
vtkAbstractArray* received = vtkAbstractArray::CreateArray(col->GetDataType());
received->SetNumberOfTuples(totalLength);
char* sendBuf = (char*)col->GetVoidPointer(0);
char* recvBuf = (char*)received->GetVoidPointer(0);
comm->AllGatherV(sendBuf, recvBuf, myLength * typeSize, recvLengths.data(), recvOffsets.data());
gatheredTable->AddColumn(received);
received->Delete();
}
outRowDataTable->ShallowCopy(gatheredTable);
return 1;
}
VTK_ABI_NAMESPACE_END
|
