FrameObsTest.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 FRAMEOBSTEST_H_
#define FRAMEOBSTEST_H_

#include <sstream>

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

#include "TestCase.h"
#include "TestFrameSource.h"

namespace tests {

class DummyFrameSource : public TestFrameSource, public FrameObsSupport {
public:
    DummyFrameSource(CvSize s) : TestFrameSource(s) {}
    void FireNotification() {
        IplImage *img = GetNextFrame();
        FireFrameNotification(img, GetFrameNr());
    } 
};

class DummyFrameObserver : public FrameObs {
    int *counter;
    std::string name;
public:
    DummyFrameObserver(int *c = 0, std::string n="DummyFrameObserver") : counter(c), name(n) {}
    void FrameNotify(IplImage *img, int frameNr) {
        if (counter) (*counter)++;
    }
    const char* GetName() const {
        return name.c_str();
    }
};

class FrameObsTestDifferent : public TestCase {
public:
    const char* GetName() const {
        return "FrameObsTestDifferent";
    }
    
    void Run() {
        int counter=0;
        DummyFrameSource src(cvSize(100,100));
        std::vector<GCPointer<DummyFrameObserver> > obs(5);
        
        
        for (int i=0;i<obs.size();i++) {
            std::ostringstream ostr;
            ostr << "obs" << i;
            
            obs[i] = new DummyFrameObserver(&counter, ostr.str());
            src.AddObserver(&*obs[i]);
        }
        src.FireNotification();

        AssertEqual(counter, obs.size());

        for (int i=0;i<obs.size();i++) {
            counter = 0;
            src.RemoveObserver(&*obs[i]);
            src.FireNotification();

            AssertEqual(counter, obs.size() - i - 1);
        }

    }
};
    
class FrameObsTestEqualNames : public TestCase {
public:
    const char* GetName() const {
        return "FrameObsTestEqualNames";
    }
    
    void Run() {
        int counter=0;
        DummyFrameSource src(cvSize(100,100));
        std::vector<GCPointer<DummyFrameObserver> > obs(5);
        
        
        for (int i=0;i<obs.size();i++) {
            obs[i] = new DummyFrameObserver(&counter);
            src.AddObserver(&*obs[i]);
        }
        src.FireNotification();

        AssertEqual(counter, 1);

        for (int i=0;i<obs.size();i++) {
            counter = 0;
            src.RemoveObserver(&*obs[i]);
            src.FireNotification();

            AssertEqual(counter, 0);
        }

    }
};

}; // end of namespace tests

#endif /*ACTCVOBSTEST_H_*/