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 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkImageAppendComponents.h"
#include "vtkAlgorithmOutput.h"
#include "vtkDataSetAttributes.h"
#include "vtkImageData.h"
#include "vtkImageProgressIterator.h"
#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkObjectFactory.h"
#include "vtkStreamingDemandDrivenPipeline.h"
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkImageAppendComponents);
//------------------------------------------------------------------------------
void vtkImageAppendComponents::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
//------------------------------------------------------------------------------
void vtkImageAppendComponents::ReplaceNthInputConnection(int idx, vtkAlgorithmOutput* input)
{
if (idx < 0 || idx >= this->GetNumberOfInputConnections(0))
{
vtkErrorMacro("Attempt to replace connection idx "
<< idx << " of input port " << 0 << ", which has only "
<< this->GetNumberOfInputConnections(0) << " connections.");
return;
}
if (!input || !input->GetProducer())
{
vtkErrorMacro("Attempt to replace connection index "
<< idx << " for input port " << 0 << " with "
<< (!input ? "a null input." : "an input with no producer."));
return;
}
this->SetNthInputConnection(0, idx, input);
}
//------------------------------------------------------------------------------
// The default vtkImageAlgorithm semantics are that SetInput() puts
// each input on a different port, we want all the image inputs to
// go on the first port.
void vtkImageAppendComponents::SetInputData(int idx, vtkDataObject* input)
{
this->SetInputDataInternal(idx, input);
}
//------------------------------------------------------------------------------
vtkDataObject* vtkImageAppendComponents::GetInput(int idx)
{
if (this->GetNumberOfInputConnections(0) <= idx)
{
return nullptr;
}
return vtkImageData::SafeDownCast(this->GetExecutive()->GetInputData(0, idx));
}
//------------------------------------------------------------------------------
// This method tells the output it will have more components
int vtkImageAppendComponents::RequestInformation(vtkInformation* vtkNotUsed(request),
vtkInformationVector** inputVector, vtkInformationVector* outputVector)
{
// get the info objects
vtkInformation* outInfo = outputVector->GetInformationObject(0);
vtkInformation* inScalarInfo;
int idx1, num;
num = 0;
for (idx1 = 0; idx1 < this->GetNumberOfInputConnections(0); ++idx1)
{
inScalarInfo =
vtkDataObject::GetActiveFieldInformation(inputVector[0]->GetInformationObject(idx1),
vtkDataObject::FIELD_ASSOCIATION_POINTS, vtkDataSetAttributes::SCALARS);
if (inScalarInfo && inScalarInfo->Has(vtkDataObject::FIELD_NUMBER_OF_COMPONENTS()))
{
num += inScalarInfo->Get(vtkDataObject::FIELD_NUMBER_OF_COMPONENTS());
}
}
vtkDataObject::SetPointDataActiveScalarInfo(outInfo, -1, num);
return 1;
}
//------------------------------------------------------------------------------
// This templated function executes the filter for any type of data.
template <class T>
void vtkImageAppendComponentsExecute(vtkImageAppendComponents* self, vtkImageData* inData,
vtkImageData* outData, int outComp, int outExt[6], int id, T*)
{
vtkImageIterator<T> inIt(inData, outExt);
vtkImageProgressIterator<T> outIt(outData, outExt, self, id);
int numIn = inData->GetNumberOfScalarComponents();
int numSkip = outData->GetNumberOfScalarComponents() - numIn;
int i;
// Loop through output pixels
while (!outIt.IsAtEnd())
{
T* inSI = inIt.BeginSpan();
T* outSI = outIt.BeginSpan() + outComp;
T* outSIEnd = outIt.EndSpan();
while (outSI < outSIEnd)
{
// now process the components
for (i = 0; i < numIn; ++i)
{
*outSI = *inSI;
++outSI;
++inSI;
}
outSI = outSI + numSkip;
}
inIt.NextSpan();
outIt.NextSpan();
}
}
//------------------------------------------------------------------------------
int vtkImageAppendComponents::FillInputPortInformation(int i, vtkInformation* info)
{
info->Set(vtkAlgorithm::INPUT_IS_REPEATABLE(), 1);
return this->Superclass::FillInputPortInformation(i, info);
}
//------------------------------------------------------------------------------
// This method is passed a input and output regions, and executes the filter
// algorithm to fill the output from the inputs.
// It just executes a switch statement to call the correct function for
// the regions data types.
void vtkImageAppendComponents::ThreadedRequestData(vtkInformation* vtkNotUsed(request),
vtkInformationVector** vtkNotUsed(inputVector), vtkInformationVector* vtkNotUsed(outputVector),
vtkImageData*** inData, vtkImageData** outData, int outExt[6], int id)
{
int idx1, outComp;
outComp = 0;
for (idx1 = 0; idx1 < this->GetNumberOfInputConnections(0); ++idx1)
{
if (inData[0][idx1] != nullptr)
{
// this filter expects that input is the same type as output.
if (inData[0][idx1]->GetScalarType() != outData[0]->GetScalarType())
{
vtkErrorMacro(<< "Execute: input" << idx1 << " ScalarType ("
<< inData[0][idx1]->GetScalarType() << "), must match output ScalarType ("
<< outData[0]->GetScalarType() << ")");
return;
}
switch (inData[0][idx1]->GetScalarType())
{
vtkTemplateMacro(vtkImageAppendComponentsExecute(
this, inData[0][idx1], outData[0], outComp, outExt, id, static_cast<VTK_TT*>(nullptr)));
default:
vtkErrorMacro(<< "Execute: Unknown ScalarType");
return;
}
outComp += inData[0][idx1]->GetNumberOfScalarComponents();
}
}
}
VTK_ABI_NAMESPACE_END
|