E:/dev/ssip2006/snake/snake.cpp

Go to the documentation of this file.

//#include "stdafx.h"

// STL includes:
#include <iostream>
using namespace std;

// OpenCV includes
#include "highgui.h"

// local includes
#include "snake.h"
#include "QtCvWrapper.h"


Snake::Snake(QtCvWrapper* parent, IplImage* inImg) :
        m_parent(parent),
        m_inImg(0),
        m_snakeImg(0),
        m_alpha((float)1.0),
        m_beta((float)1.0),
        m_gamma((float)1.2),
        m_numSegments(200),
        m_numIteration(20)
{
        m_inImg = cvCloneImage(inImg);

        m_winSize.width = 5;
        m_winSize.height = 5;

        m_snakeImg = cvCreateImage( cvSize(m_inImg->width, m_inImg->height), 
                IPL_DEPTH_8U, 3);
        m_crit = cvTermCriteria(CV_TERMCRIT_ITER, m_numIteration, 0.0);
        m_snakeCurve = new CvPoint[m_numSegments];
}

Snake::~Snake(void)
{
        delete[] m_snakeCurve;
}

void Snake::initSnakeCurve()
{
        m_snakeCurve = new CvPoint[m_numSegments];
        for(int i = 0; i < m_numSegments; i++)
        {
                m_snakeCurve[i].x = cvRound( 0.8 * m_inImg->width * cos(i * 6.28 / m_numSegments) / 2.0
                        + m_inImg->width / 2 );

                m_snakeCurve[i].y = cvRound( 0.8 * m_inImg->height * sin(i * 6.28 / m_numSegments) / 2.0
                        + m_inImg->height / 2 );
        }
}

void Snake::initSnakeCurve(CvPoint* pt)
{
        m_snakeCurve = pt;
        if(m_inImg->nChannels == 1)
                cvCvtColor(m_inImg, m_snakeImg, CV_GRAY2RGB);
        else if (m_inImg->nChannels == 3)
                cvCopy( m_inImg, m_snakeImg, NULL);

        for (int actSegment = 0; actSegment < m_numSegments; actSegment++)
        {
                if(actSegment < m_numSegments-1)
                        cvLine(m_snakeImg, pt[actSegment], pt[actSegment+1],
                        CV_RGB(255, 0, 0), 2, 1);
                else
                        cvLine(m_snakeImg, pt[actSegment], pt[0],
                        CV_RGB(0, 255, 0), 2, 1);
        }
}

IplImage* Snake::iterateSnakeCurve(int numIterations, bool showIterations)
{
        m_snakeImg = cvCreateImage( cvSize(m_inImg->width, m_inImg->height), 
                IPL_DEPTH_8U, 3);

        char wndName[] = "SNAKE";

        for (int i = 0; i < numIterations; i++)
        {
                cvSnakeImage(m_inImg, m_snakeCurve, m_numSegments, 
                        &m_alpha, &m_beta, &m_gamma, 1, m_winSize, m_crit, 1);

                if (showIterations)
                {
                        // clear the line with loading the inital image
                        if(m_inImg->nChannels == 1)
                                cvCvtColor(m_inImg, m_snakeImg, CV_GRAY2RGB);
                        else if (m_inImg->nChannels == 3)
                                cvCopy( m_inImg, m_snakeImg, NULL);
                        else
                                qDebug("SNAKE - iterate: Undefined number of channels in the input image!");

                        for (int actSegment = 0; actSegment < m_numSegments; actSegment++)
                        {
                                if(actSegment < m_numSegments-1)
                                        cvLine(m_snakeImg, m_snakeCurve[actSegment], m_snakeCurve[actSegment+1],
                                        CV_RGB(255, 0, 0), 2, 1);
                                else
                                        cvLine(m_snakeImg, m_snakeCurve[actSegment], m_snakeCurve[0],
                                        CV_RGB(0, 255, 0), 2, 1);
                        }
                        if (m_parent->iterate() == true)
                        {
                                cvNamedWindow( wndName, 1 );
                                cvResizeWindow( wndName, m_inImg->width,m_inImg->height );
                                cvShowImage( wndName, m_snakeImg );

                                if (m_parent->step() == true)
                                        cvWaitKey(0);
                                else
                                        cvWaitKey(200);
                        }
                }
        }
        cvDestroyWindow(wndName);
        return m_snakeImg;
}

---