forked from Kitware/VTK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
vtkPBivariateLinearTableThreshold.cxx
119 lines (94 loc) · 3.93 KB
/
vtkPBivariateLinearTableThreshold.cxx
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
/*=========================================================================
Program: Visualization Toolkit
Module: vtkPBivariateLinearTableThreshold.cxx
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
/*-------------------------------------------------------------------------
Copyright 2009 Sandia Corporation.
Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------*/
#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>
vtkStandardNewMacro(vtkPBivariateLinearTableThreshold);
vtkCxxSetObjectMacro(vtkPBivariateLinearTableThreshold, Controller, vtkMultiProcessController);
vtkPBivariateLinearTableThreshold::vtkPBivariateLinearTableThreshold()
{
this->Controller = 0;
this->SetController(vtkMultiProcessController::GetGlobalController());
}
vtkPBivariateLinearTableThreshold::~vtkPBivariateLinearTableThreshold()
{
this->SetController(0);
}
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 = vtkAbstractArray::SafeDownCast(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[0], 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[0], &recvOffsets[0]);
gatheredTable->AddColumn(received);
received->Delete();
}
outRowDataTable->ShallowCopy(gatheredTable);
return 1;
}