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
|
/*=========================================================================
Program: Visualization Toolkit
Module: vtkPartitionedDataSet.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.
=========================================================================*/
#include "vtkPartitionedDataSet.h"
#include "vtkDataSet.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
vtkStandardNewMacro(vtkPartitionedDataSet);
//----------------------------------------------------------------------------
vtkPartitionedDataSet::vtkPartitionedDataSet() {}
//----------------------------------------------------------------------------
vtkPartitionedDataSet::~vtkPartitionedDataSet() {}
//----------------------------------------------------------------------------
vtkPartitionedDataSet* vtkPartitionedDataSet::GetData(vtkInformation* info)
{
return info ? vtkPartitionedDataSet::SafeDownCast(info->Get(DATA_OBJECT())) : nullptr;
}
//----------------------------------------------------------------------------
vtkPartitionedDataSet* vtkPartitionedDataSet::GetData(vtkInformationVector* v, int i)
{
return vtkPartitionedDataSet::GetData(v->GetInformationObject(i));
}
//----------------------------------------------------------------------------
void vtkPartitionedDataSet::SetNumberOfPartitions(unsigned int numPartitions)
{
this->Superclass::SetNumberOfChildren(numPartitions);
}
//----------------------------------------------------------------------------
unsigned int vtkPartitionedDataSet::GetNumberOfPartitions()
{
return this->Superclass::GetNumberOfChildren();
}
//----------------------------------------------------------------------------
vtkDataSet* vtkPartitionedDataSet::GetPartition(unsigned int idx)
{
return vtkDataSet::SafeDownCast(this->GetPartitionAsDataObject(idx));
}
//----------------------------------------------------------------------------
vtkDataObject* vtkPartitionedDataSet::GetPartitionAsDataObject(unsigned int idx)
{
return this->Superclass::GetChild(idx);
}
//----------------------------------------------------------------------------
void vtkPartitionedDataSet::SetPartition(unsigned int idx, vtkDataObject* partition)
{
if (partition && partition->IsA("vtkCompositeDataSet"))
{
vtkErrorMacro("Partition cannot be a vtkCompositeDataSet.");
return;
}
this->Superclass::SetChild(idx, partition);
}
//----------------------------------------------------------------------------
void vtkPartitionedDataSet::RemoveNullPartitions()
{
unsigned int next = 0;
for (unsigned int cc = 0; cc < this->GetNumberOfPartitions(); ++cc)
{
auto ds = this->GetPartition(cc);
if (ds)
{
if (next < cc)
{
this->SetPartition(next, ds);
if (this->HasChildMetaData(cc))
{
this->SetChildMetaData(next, this->GetChildMetaData(cc));
}
this->SetPartition(cc, nullptr);
this->SetChildMetaData(cc, nullptr);
}
next++;
}
}
this->SetNumberOfPartitions(next);
}
//----------------------------------------------------------------------------
void vtkPartitionedDataSet::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
|