File: RelabelComponents.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 (29 lines) | stat: -rw-r--r-- 1,047 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
#include "ITKWrap.hpp"
#include <itkRelabelComponentImageFilter.h>

template <int dims, DataClass dataclass, class PixelClass>
static Array TRelabelComponentsFunction(Array A, int minsize)
{
  typedef itk::Image<PixelClass, dims> ITKType;
  typename ITKType::Pointer imageIn = CreateITKFromArray<dims,dataclass,PixelClass>(A);
  typedef itk::RelabelComponentImageFilter<ITKType, ITKType> FilterType;
  typename FilterType::Pointer filter = FilterType::New();
  filter->SetInput(imageIn);
  filter->SetMinimumObjectSize(minsize);
  filter->Update();
  return CreateArrayFromITK<dims,dataclass,PixelClass>(filter->GetOutput());
}

//@@Signature
//gfunction relabelcomponents RelabelComponentsFunction
//input x minsize
//output y
ArrayVector RelabelComponentsFunction(int nargout, const ArrayVector& arg)
{
  int minsize = 0;
  if (arg.size() >= 2) minsize = arg[1].asInteger();
  if (arg[0].is2D())
    return TRelabelComponentsFunction<2,Int32,int32>(arg[0],minsize);
  else
    return TRelabelComponentsFunction<3,Int32,int32>(arg[0],minsize);
}