You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
129 lines
3.9 KiB
129 lines
3.9 KiB
/*=========================================================================
|
|
|
|
Program: Visualization Toolkit
|
|
Module: $RCSfile: vtkImageEuclideanToPolar.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 "vtkImageEuclideanToPolar.h"
|
|
|
|
#include "vtkImageData.h"
|
|
#include "vtkImageProgressIterator.h"
|
|
#include "vtkObjectFactory.h"
|
|
|
|
#include <math.h>
|
|
|
|
vtkCxxRevisionMacro(vtkImageEuclideanToPolar, "$Revision: 1.28 $");
|
|
vtkStandardNewMacro(vtkImageEuclideanToPolar);
|
|
|
|
//----------------------------------------------------------------------------
|
|
vtkImageEuclideanToPolar::vtkImageEuclideanToPolar()
|
|
{
|
|
this->SetNumberOfInputPorts(1);
|
|
this->SetNumberOfOutputPorts(1);
|
|
this->ThetaMaximum = 255.0;
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
// This templated function executes the filter for any type of data.
|
|
template <class T>
|
|
void vtkImageEuclideanToPolarExecute(vtkImageEuclideanToPolar *self,
|
|
vtkImageData *inData,
|
|
vtkImageData *outData,
|
|
int outExt[6], int id, T *)
|
|
{
|
|
vtkImageIterator<T> inIt(inData, outExt);
|
|
vtkImageProgressIterator<T> outIt(outData, outExt, self, id);
|
|
double X, Y, Theta, R;
|
|
double thetaMax = self->GetThetaMaximum();
|
|
|
|
// find the region to loop over
|
|
int maxC = inData->GetNumberOfScalarComponents();
|
|
|
|
// Loop through ouput pixels
|
|
while (!outIt.IsAtEnd())
|
|
{
|
|
T* inSI = inIt.BeginSpan();
|
|
T* outSI = outIt.BeginSpan();
|
|
T* outSIEnd = outIt.EndSpan();
|
|
while (outSI != outSIEnd)
|
|
{
|
|
// Pixel operation
|
|
X = (double)(*inSI);
|
|
Y = (double)(inSI[1]);
|
|
|
|
if ((X == 0.0) && (Y == 0.0))
|
|
{
|
|
Theta = 0.0;
|
|
R = 0.0;
|
|
}
|
|
else
|
|
{
|
|
Theta = atan2(Y, X) * thetaMax / 6.2831853;
|
|
if (Theta < 0.0)
|
|
{
|
|
Theta += thetaMax;
|
|
}
|
|
R = sqrt(X*X + Y*Y);
|
|
}
|
|
|
|
*outSI = (T)(Theta);
|
|
outSI[1] = (T)(R);
|
|
inSI += maxC;
|
|
outSI += maxC;
|
|
}
|
|
inIt.NextSpan();
|
|
outIt.NextSpan();
|
|
}
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
void vtkImageEuclideanToPolar::ThreadedExecute (vtkImageData *inData,
|
|
vtkImageData *outData,
|
|
int outExt[6], int id)
|
|
{
|
|
vtkDebugMacro(<< "Execute: inData = " << inData
|
|
<< ", outData = " << outData);
|
|
|
|
// this filter expects that input is the same type as output.
|
|
if (inData->GetScalarType() != outData->GetScalarType())
|
|
{
|
|
vtkErrorMacro(<< "Execute: input ScalarType, "
|
|
<< inData->GetScalarType()
|
|
<< ", must match out ScalarType "
|
|
<< outData->GetScalarType());
|
|
return;
|
|
}
|
|
|
|
// input must have at least two components
|
|
if (inData->GetNumberOfScalarComponents() < 2)
|
|
{
|
|
vtkErrorMacro(<< "Execute: input does not have at least two components");
|
|
return;
|
|
}
|
|
|
|
switch (inData->GetScalarType())
|
|
{
|
|
vtkTemplateMacro(
|
|
vtkImageEuclideanToPolarExecute( this,
|
|
inData, outData, outExt, id,
|
|
static_cast<VTK_TT *>(0)));
|
|
default:
|
|
vtkErrorMacro(<< "Execute: Unknown ScalarType");
|
|
return;
|
|
}
|
|
}
|
|
|
|
void vtkImageEuclideanToPolar::PrintSelf(ostream& os, vtkIndent indent)
|
|
{
|
|
this->Superclass::PrintSelf(os,indent);
|
|
|
|
os << indent << "Maximum Angle: " << this->ThetaMaximum << "\n";
|
|
}
|
|
|