TestFrameSource.h

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
*/


#ifndef TESTFRAMESOURCE_H_
#define TESTFRAMESOURCE_H_

#if HAVE_CONFIG_H
#include <config.h>
#endif

#include <cv.h>
#include <highgui.h>

#include <boost/random/uniform_real_distribution.hpp>
#include <boost/random/taus88.hpp>

#include <mhexcptn.h>

#include <general/OpenCVHelpers.h>
#include <vision/FrameSource.h>
#include <vision/FrameObserver.h>

class TestFrameSource : public FrameSource {
    boost::random::taus88 rgen_;
    boost::random::uniform_real_distribution<double> rv_;

protected:    
    CvSize size;

    GCPointer<ImageHolder> curImg;    
    
    int n;
    
public:
    TestFrameSource(CvSize s) : size(s), rv_(20.0,80.0), n(0) {};
    virtual ~TestFrameSource();
    
    IplImage *GetNextFrame();
    
    int GetFrameNr() const;
};

class TestLineFrameSource : public TestFrameSource {
    
    CvPoint centralPoint;
    
    CvPoint lineEnds[4];
    
    boost::random::taus88 rgen_;
    boost::random::uniform_real_distribution<double> rv_;
    
protected:
    CvPoint GetCentralPoint() const {
        return centralPoint;
    }

public:
    TestLineFrameSource(CvSize s) : TestFrameSource(s), rv_(0.0,255.0) {
        centralPoint = cvPoint(s.width/2, s.height/2);
        for (int i=0;i<4;i++) {
            lineEnds[i]=cvPoint(s.width*i/3, s.height-1); 
        }
        
    }

    IplImage *GetNextFrame();
    
    void SetCentralXPos(int x) {
        centralPoint.x = x;
    }
    
    int GetCentralXPos() const {
        return centralPoint.x;
    }
};

class TestLineFSStaticErr : public TestLineFrameSource {
    
    int frameCounter;

    boost::random::taus88 rgen_;
    boost::random::uniform_real_distribution<double> rv_;

    int curLineEnd;
public:
    TestLineFSStaticErr(CvSize s) : TestLineFrameSource(s),
        rv_(0.0, s.width-1), frameCounter(0) {
        curLineEnd = static_cast<int>(rv_(rgen_));
    }
        
    IplImage *GetNextFrame();
};

class StillImage : public FrameSource, public FrameObsSupport {
    GCPointer<ImageHolder> curImg;    
    int n;
public:
    StillImage(const char *filename) {
        IplImage *img = cvLoadImage(filename, CV_LOAD_IMAGE_UNCHANGED);
        if (img==nullptr) throw FileLineException(__FILE__,__LINE__);

        curImg = new ImageHolder(img);
    }
    IplImage *GetNextFrame() {
        n++;
        return curImg->GetImage();
    }

    void FireNotification() {
        IplImage *img = GetNextFrame();
        FireFrameNotification(img, GetFrameNr());
    } 

    
    int GetFrameNr() const {
        return n;
    }
};

#endif /*TESTFRAMESOURCE_H_*/