File: Noise.cpp

package info (click to toggle)
freemat 4.2%2Bdfsg1-6
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 142,116 kB
  • sloc: ansic: 126,788; cpp: 62,015; python: 2,080; perl: 1,255; sh: 1,146; yacc: 1,019; lex: 239; makefile: 107
file content (37 lines) | stat: -rw-r--r-- 1,092 bytes parent folder | download | duplicates (2)
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);
}