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
|
/*=========================================================================
Program: Visualization Toolkit
Module: $RCSfile: vtkMultiBlockMergeFilter.cxx,v $
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.
=========================================================================*/
#include "vtkMultiBlockMergeFilter.h"
#include "vtkCellData.h"
#include "vtkDataObject.h"
#include "vtkMultiBlockDataSet.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkStreamingDemandDrivenPipeline.h"
vtkCxxRevisionMacro(vtkMultiBlockMergeFilter, "$Revision: 1.5 $");
vtkStandardNewMacro(vtkMultiBlockMergeFilter);
vtkMultiBlockMergeFilter::vtkMultiBlockMergeFilter()
{
}
vtkMultiBlockMergeFilter::~vtkMultiBlockMergeFilter()
{
}
int vtkMultiBlockMergeFilter::RequestData(
vtkInformation *vtkNotUsed(request),
vtkInformationVector **inputVector,
vtkInformationVector *outputVector)
{
vtkInformation* info = outputVector->GetInformationObject(0);
vtkMultiBlockDataSet *output = vtkMultiBlockDataSet::SafeDownCast(
info->Get(vtkDataObject::DATA_OBJECT()));
if (!output) {return 0;}
int numInputs = inputVector[0]->GetNumberOfInformationObjects();
if (numInputs<0)
{
vtkErrorMacro("Too many inputs to algorithm.")
return 0;
}
unsigned int usNInputs = (unsigned int)numInputs;
int first = 1;
for (int idx = 0; idx < numInputs; ++idx)
{
vtkMultiBlockDataSet* input = 0;
vtkInformation* inInfo = inputVector[0]->GetInformationObject(idx);
if (inInfo)
{
input = vtkMultiBlockDataSet::SafeDownCast(
inInfo->Get(vtkDataObject::DATA_OBJECT()));
}
if (input)
{
if (first)
{
//shallow copy first input to output to start off with
//cerr << "Copy first input" << endl;
output->ShallowCopy(vtkMultiBlockDataSet::SafeDownCast(input));
first = 0;
}
else
{
//merge in the inputs data sets into the proper places in the output
for (unsigned int blk = 0; blk < input->GetNumberOfBlocks(); blk++)
{
//inputs are allowed to have either 1 or N datasets in each group
unsigned int dsId = 0;
if (input->GetNumberOfDataSets(blk) == usNInputs)
{
dsId = idx;
}
//cerr << "Copying blk " << blk << " index " << idx << endl;
vtkDataObject* inblk = input->GetDataSet(blk, dsId);
vtkDataObject* dsCopy = inblk->NewInstance();
dsCopy->ShallowCopy(inblk);
//output will always have N datasets in each group
if (output->GetNumberOfDataSets(blk) != usNInputs)
{
output->SetNumberOfDataSets(blk, numInputs);
}
output->SetDataSet(blk, idx, dsCopy);
dsCopy->Delete();
}
}
}
}
return !first;
}
void vtkMultiBlockMergeFilter::AddInput(vtkDataObject* input)
{
this->AddInput(0, input);
}
void vtkMultiBlockMergeFilter::AddInput(int index, vtkDataObject* input)
{
if(input)
{
this->AddInputConnection(index, input->GetProducerPort());
}
}
int vtkMultiBlockMergeFilter::FillInputPortInformation(
int, vtkInformation *info)
{
info->Set(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkMultiBlockDataSet");
info->Set(vtkAlgorithm::INPUT_IS_REPEATABLE(), 1);
info->Set(vtkAlgorithm::INPUT_IS_OPTIONAL(), 1);
return 1;
}
void vtkMultiBlockMergeFilter::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os,indent);
}
|
