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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkHyperTreeGridExtractGhostCells.h"
#include "vtkBitArray.h"
#include "vtkCellData.h"
#include "vtkHyperTree.h"
#include "vtkHyperTreeGrid.h"
#include "vtkHyperTreeGridNonOrientedCursor.h"
#include "vtkInformation.h"
#include "vtkObjectFactory.h"
#include "vtkUnsignedCharArray.h"
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkHyperTreeGridExtractGhostCells);
//------------------------------------------------------------------------------
vtkHyperTreeGridExtractGhostCells::vtkHyperTreeGridExtractGhostCells()
{
this->AppropriateOutput = true;
}
//------------------------------------------------------------------------------
void vtkHyperTreeGridExtractGhostCells::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
os << indent << "OutputGhostArrayName: "
<< (this->OutputGhostArrayName ? this->OutputGhostArrayName : "(nullptr)") << std::endl;
}
//------------------------------------------------------------------------------
int vtkHyperTreeGridExtractGhostCells::ProcessTrees(
vtkHyperTreeGrid* input, vtkDataObject* outputDO)
{
// Downcast output data object to hyper tree grid
vtkHyperTreeGrid* output = vtkHyperTreeGrid::SafeDownCast(outputDO);
if (!output)
{
vtkErrorMacro("Incorrect type of output: " << outputDO->GetClassName());
return 0;
}
output->ShallowCopy(input);
// Retrieve and copy input mask if it exists
this->InMask = input->HasMask() ? input->GetMask() : nullptr;
if (this->InMask)
{
this->OutMask->DeepCopy(this->InMask);
}
else
{
this->OutMask->SetNumberOfTuples(output->GetNumberOfCells());
}
output->SetMask(this->OutMask);
// Retrieve ghost array
if (input->HasAnyGhostCells())
{
this->InGhost = input->GetGhostCells();
}
else
{
vtkWarningMacro(<< "Input does not have a ghost array. Output HTG will be empty.");
}
// Iterate over
vtkIdType inIndex = 0;
vtkHyperTreeGrid::vtkHyperTreeGridIterator it;
output->InitializeTreeIterator(it);
vtkNew<vtkHyperTreeGridNonOrientedCursor> outCursor;
while (it.GetNextTree(inIndex))
{
if (this->CheckAbort())
{
break;
}
output->InitializeNonOrientedCursor(outCursor, inIndex, true);
if (!this->InGhost)
{
// Input has no ghost cell, mask the whole tree
this->OutMask->InsertValue(outCursor->GetGlobalNodeIndex(), 1);
}
else
{
this->RecursivelyMaskNonGhost(outCursor);
}
}
this->OutMask->Squeeze();
// Copy the input ghost array and rename it in output
if (this->InGhost)
{
vtkNew<vtkUnsignedCharArray> ghostCopy;
ghostCopy->ShallowCopy(this->InGhost);
if (!this->OutputGhostArrayName)
{
ghostCopy->SetName("GhostType");
}
else
{
ghostCopy->SetName(this->OutputGhostArrayName);
}
output->GetCellData()->AddArray(ghostCopy);
output->GetCellData()->RemoveArray(this->InGhost->GetName());
}
return 1;
}
//------------------------------------------------------------------------------
bool vtkHyperTreeGridExtractGhostCells::RecursivelyMaskNonGhost(
vtkHyperTreeGridNonOrientedCursor* cursor)
{
vtkIdType currentId = cursor->GetGlobalNodeIndex();
cursor->SetMask(false);
if (this->InMask && this->InMask->GetValue(currentId))
{
cursor->SetMask(true);
return false;
}
bool hasGhosts = false;
if (cursor->IsLeaf())
{
hasGhosts = this->InGhost->GetTuple1(currentId);
}
else
{
for (int child = 0; child < cursor->GetNumberOfChildren(); ++child)
{
cursor->ToChild(child);
// Coarse cell is masked if it does not have any ghost leaf
hasGhosts |= this->RecursivelyMaskNonGhost(cursor);
cursor->ToParent();
}
}
cursor->SetMask(!hasGhosts);
return hasGhosts;
}
VTK_ABI_NAMESPACE_END
|
