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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkMemoryLimitImageDataStreamer.h"
#include "vtkAlgorithmOutput.h"
#include "vtkCommand.h"
#include "vtkExtentTranslator.h"
#include "vtkImageData.h"
#include "vtkInformation.h"
#include "vtkInformationExecutivePortKey.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkPipelineSize.h"
#include "vtkStreamingDemandDrivenPipeline.h"
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkMemoryLimitImageDataStreamer);
//------------------------------------------------------------------------------
vtkMemoryLimitImageDataStreamer::vtkMemoryLimitImageDataStreamer()
{
// Set a default memory limit of 50 mebibytes
this->MemoryLimit = 50 * 1024;
}
//------------------------------------------------------------------------------
void vtkMemoryLimitImageDataStreamer::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
os << indent << "MemoryLimit (in kibibytes): " << this->MemoryLimit << endl;
}
//------------------------------------------------------------------------------
vtkTypeBool vtkMemoryLimitImageDataStreamer ::ProcessRequest(
vtkInformation* request, vtkInformationVector** inputVector, vtkInformationVector* outputVector)
{
if (request->Has(vtkStreamingDemandDrivenPipeline::REQUEST_UPDATE_EXTENT()))
{
if (this->CurrentDivision == 0)
{
// we must set the extent on the input
vtkInformation* outInfo = outputVector->GetInformationObject(0);
// get the requested update extent
int outExt[6];
outInfo->Get(vtkStreamingDemandDrivenPipeline::UPDATE_EXTENT(), outExt);
vtkInformation* inInfo = inputVector[0]->GetInformationObject(0);
vtkExtentTranslator* translator = this->GetExtentTranslator();
translator->SetWholeExtent(outExt);
vtkPipelineSize* sizer = vtkPipelineSize::New();
this->NumberOfStreamDivisions = 1;
unsigned long oldSize, size = 0;
float ratio;
translator->SetPiece(0);
// watch for the limiting case where the size is the maximum size
// represented by an unsigned long. In that case we do not want to do
// the ratio test. We actual test for size < 0.5 of the max unsigned
// long which would indicate that oldSize is about at max unsigned
// long.
unsigned long maxSize;
maxSize = (((unsigned long)0x1) << (8 * sizeof(unsigned long) - 1));
// we also have to watch how many pieces we are creating. Since
// NumberOfStreamDivisions is an int, it cannot be more that say 2^31
// (which is a bit much anyhow) so we also stop if the number of pieces
// is too large.
int count = 0;
// double the number of pieces until the size fits in memory
// or the reduction in size falls to 20%
do
{
oldSize = size;
translator->SetNumberOfPieces(this->NumberOfStreamDivisions);
translator->PieceToExtentByPoints();
int inExt[6];
translator->GetExtent(inExt);
// set the update extent
inInfo->Set(vtkStreamingDemandDrivenPipeline::UPDATE_EXTENT(), inExt, 6);
// set a hint not to combine with previous requests
inInfo->Set(
vtkStreamingDemandDrivenPipeline::UPDATE_EXTENT_INITIALIZED(), VTK_UPDATE_EXTENT_REPLACE);
// then propagate it
vtkExecutive* exec = vtkExecutive::PRODUCER()->GetExecutive(inInfo);
int index = vtkExecutive::PRODUCER()->GetPort(inInfo);
vtkStreamingDemandDrivenPipeline* sddp =
vtkStreamingDemandDrivenPipeline::SafeDownCast(exec);
sddp->PropagateUpdateExtent(index);
// then reset the INITIALIZED flag to the default value COMBINE
inInfo->Set(
vtkStreamingDemandDrivenPipeline::UPDATE_EXTENT_INITIALIZED(), VTK_UPDATE_EXTENT_COMBINE);
size = sizer->GetEstimatedSize(this, 0, 0);
// watch for the first time through
if (!oldSize)
{
ratio = 0.5;
}
// otherwise the normal ratio calculation
else
{
ratio = size / (float)oldSize;
}
this->NumberOfStreamDivisions = this->NumberOfStreamDivisions * 2;
count++;
} while (size > this->MemoryLimit && (size < maxSize && ratio < 0.8) && count < 29);
// undo the last *2
this->NumberOfStreamDivisions = this->NumberOfStreamDivisions / 2;
sizer->Delete();
}
return this->Superclass::ProcessRequest(request, inputVector, outputVector);
}
return this->Superclass::ProcessRequest(request, inputVector, outputVector);
}
VTK_ABI_NAMESPACE_END
|
