106 lines
1.9 KiB
C++
106 lines
1.9 KiB
C++
/*
|
|
* This file is part of Pastie
|
|
*
|
|
* Pastie 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 3 of the License, or
|
|
* any later version.
|
|
*
|
|
* Pastie 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 Pastie. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
/**
|
|
* @copyright Kitware Inc.
|
|
* @license http://www.gnu.org/licenses/gpl.txt GNU Public License
|
|
* @author Steffen Vogel <steffen.vogel@rwth-aachen.de>
|
|
* @link http://www.steffenvogel.de
|
|
* @package CTK
|
|
*/
|
|
|
|
#include <QDir>
|
|
#include <QtDebug>
|
|
|
|
#include <opencv2/core.hpp>
|
|
|
|
#include "source.h"
|
|
|
|
using namespace cv;
|
|
|
|
Source::Source() :
|
|
VideoCapture(CAP_ANY)
|
|
{
|
|
if (!isOpened())
|
|
throw Exception();
|
|
|
|
setSize(Size(640, 480));
|
|
setFPS(15);
|
|
|
|
connect(&tmr, SIGNAL(timeout()), this, SLOT(tick()));
|
|
}
|
|
|
|
Size Source::getSize()
|
|
{
|
|
return Size(
|
|
get(CAP_PROP_FRAME_WIDTH),
|
|
get(CAP_PROP_FRAME_HEIGHT)
|
|
);
|
|
}
|
|
|
|
bool Source::setSize(const Size s)
|
|
{
|
|
set(CAP_PROP_FRAME_WIDTH, s.width);
|
|
return set(CAP_PROP_FRAME_HEIGHT, s.height);
|
|
}
|
|
|
|
double Source::getFPS()
|
|
{
|
|
return fps;
|
|
}
|
|
|
|
bool Source::setFPS(double f)
|
|
{
|
|
fps = f;
|
|
|
|
if (tmr.isActive())
|
|
tmr.start(1000.0 / fps);
|
|
|
|
return set(CAP_PROP_FPS, fps);
|
|
}
|
|
|
|
Image * Source::getSnapshot()
|
|
{
|
|
static unsigned no;
|
|
|
|
Mat mat;
|
|
QString fileName = QString("%1/snapshot_%2.png")
|
|
.arg(QDir::tempPath()).arg(no++);
|
|
|
|
read(mat);
|
|
|
|
return new Image(mat, fileName);
|
|
}
|
|
|
|
void Source::play(bool run)
|
|
{
|
|
if (run)
|
|
tmr.start(1000.0 / fps);
|
|
else
|
|
tmr.stop();
|
|
}
|
|
|
|
void Source::tick()
|
|
{
|
|
Mat m;
|
|
|
|
if (read(m)) {
|
|
last.setSourceMat(m);
|
|
emit newFrame(&last);
|
|
}
|
|
}
|
|
|