QtFrameSource.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) 2010 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
*/


#pragma once

#include "FrameSource.h"

#include <QtGui>
#include <QtWebKit>
#include <QWebPage>

class QtFrameSource: public FrameSource {
    int width_;
    int height_;

    QWebPage page_;
    IplImage* image_;
    int n_;

    int mouseX_;
    int mouseY_;

    std::vector<RegionInfo> regions_;
    bool walkTree(const QWebElement& elem); 

    QtFrameSource(const char* url, int width=960, int height=768);

public:
    ~QtFrameSource();

    virtual IplImage* GetNextFrame();
    virtual int GetFrameNr() const;
    virtual size_t GetNumInfos() const;
    virtual const RegionInfo& GetInfo(size_t idx) const;

    void SetMousePos(int x, int y);
    void DoClick(int button);
    void DoPressKey(int key);

    static FrameSource* QtSourceFactory(const char *url);
};

class QtCaptureSource : public FrameSource {
    friend class QtFrameSource;

    int width_;
    int height_;

    IplImage* image_;
    int n_;

    QtCaptureSource(int width=640, int height=480);

public:
    ~QtCaptureSource();

    IplImage* GetNextFrame();
    int GetFrameNr() const;
};