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
|
#include "ITKWrap.hpp"
#include <itkNoiseImageFilter.h>
template <int dims, DataClass dataclass, class PixelClass>
static Array NoiseFilter(const ArrayVector & arg)
{
if (arg.size() < 2) throw Exception("noise filter requires at least two arguments");
typedef itk::Image<PixelClass, dims> ITKType;
typename ITKType::Pointer imageIn = CreateITKFromArray<dims,dataclass,PixelClass>(arg[0]);
typedef itk::NoiseImageFilter<ITKType, ITKType> FilterType;
typename FilterType::Pointer filter = FilterType::New();
typename ITKType::SizeType radius;
if (arg[1].isScalar())
{
for (int i=0;i<dims;i++)
radius[i] = arg[1].asDouble();
}
else
{
for (int i=0;i<dims;i++)
radius[i] = arg[1].get(i+1).asDouble();
}
filter->SetRadius(radius);
filter->SetInput(imageIn);
filter->Update();
return CreateArrayFromITK<dims,dataclass,PixelClass>(filter->GetOutput());
}
//@@Signature
//gfunction noisefilter NoiseFilterFunction
//input x radius
//output y
ArrayVector NoiseFilterFunction(int nargout, const ArrayVector& arg)
{
ITKCases(NoiseFilter,arg);
}
|