OpenCVHelpers.cpp

Go to the documentation of this file.

// -*- mode: c++; indent-tabs-mode: nil; c-basic-offset: 4; coding: iso-8859-1; -*-

/*
ACT-CV - Machine Vision for Cognitive Modeling
Copyright (c) 2008 Marc Halbruegge

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/


#include "OpenCVHelpers.h"

using namespace std;

#define _USE_MATH_DEFINES
#include <cmath>

#ifndef _PI
# ifdef M_PI
#  define _PI M_PI
# else
#  define _PI 3.1415926536
#  ifndef M_PI
#   define M_PI 3.1415926536
#  endif
# endif
#endif

ostream& operator << (ostream &o, const CvPoint &p) {
    o << p.x << " " << p.y;
    return o;
}

void PrintLine(const Linie &l, IplImage *img, CvScalar color) {
    cvLine(img, 
        cvPoint((int)l.GetLinks(),(int)l.GetOben()), 
        cvPoint((int)l.GetRechts(),(int)l.GetUnten()), 
        color, 
        1, // Dicke 
        CV_AA, 
        0);
}

void PrintArrow(const Linie &l, IplImage *img, CvScalar color) {
    cvLine(img, 
        cvPoint((int)l.GetLinks(),(int)l.GetOben()), 
        cvPoint((int)l.GetRechts(),(int)l.GetUnten()), 
        color, 
        1, // Dicke 
        CV_AA, 
        0);

    if (l.GetLen()>3) {
        CvPoint p,q;
        q.x=(int)l.GetRechts();
        q.y=(int)l.GetUnten();

        double angle = atan2(l.GetOben() - l.GetUnten(), l.GetLinks() - l.GetRechts());

        p.x = (int) (q.x + 5 * cos(angle + M_PI / 4));
        p.y = (int) (q.y + 5 * sin(angle + M_PI / 4));
        cvLine( img, p, q, color, 1, CV_AA, 0 );
        p.x = (int) (q.x + 5 * cos(angle - M_PI / 4));
        p.y = (int) (q.y + 5 * sin(angle - M_PI / 4));
        cvLine( img, p, q, color, 1, CV_AA, 0 );
    }
}