55 lines
1.2 KiB
C++
55 lines
1.2 KiB
C++
#include "morph_openclose.h"
|
|
|
|
#include "morph_erode.h"
|
|
#include "morph_dilate.h"
|
|
|
|
MorphOpenClose::MorphOpenClose(PNM* img) :
|
|
MorphologicalOperator(img), m_type(Open)
|
|
{
|
|
}
|
|
|
|
MorphOpenClose::MorphOpenClose(PNM* img, ImageViewer* iv) :
|
|
MorphologicalOperator(img, iv), m_type(Open)
|
|
{
|
|
}
|
|
|
|
PNM* MorphOpenClose::transform()
|
|
{
|
|
int size = getParameter("size").toInt();;
|
|
SE shape = (SE) getParameter("shape").toInt();
|
|
m_type = (Type) getParameter("type").toInt();
|
|
|
|
if (m_type == Open)
|
|
{
|
|
return dilate(erode(image, size, shape), size, shape);
|
|
}
|
|
else if (m_type == Close)
|
|
{
|
|
return erode(dilate(image, size, shape), size, shape);
|
|
}
|
|
else
|
|
{
|
|
return image;
|
|
}
|
|
}
|
|
|
|
PNM* MorphOpenClose::erode(PNM* image, int size, SE shape)
|
|
{
|
|
MorphErode e(image, getSupervisor());
|
|
e.setParameter("silent", true);
|
|
e.setParameter("shape", shape);
|
|
e.setParameter("size", size);
|
|
|
|
return e.transform();
|
|
}
|
|
|
|
PNM* MorphOpenClose::dilate(PNM* image, int size, SE shape)
|
|
{
|
|
MorphDilate e(image, getSupervisor());
|
|
e.setParameter("silent", true);
|
|
e.setParameter("shape", shape);
|
|
e.setParameter("size", size);
|
|
|
|
return e.transform();
|
|
}
|