mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-10-01 02:35:48 -04:00
139 lines
3.3 KiB
C++
139 lines
3.3 KiB
C++
#include <opencv/cv.h>
|
|
#include <opencv/highgui.h>
|
|
|
|
#include <QTimer>
|
|
#include <QPainter>
|
|
#include "QVideoDevice.h"
|
|
#include "VideoProcessor.h"
|
|
|
|
QVideoInputDevice::QVideoInputDevice(QWidget *parent)
|
|
:QObject(parent)
|
|
{
|
|
_timer = NULL ;
|
|
_capture_device = NULL ;
|
|
_video_processor = NULL ;
|
|
_echo_output_device = NULL ;
|
|
_estimated_bw = 0 ;
|
|
_total_encoded_size = 0 ;
|
|
_last_bw_estimate_TS = time(NULL) ;
|
|
}
|
|
|
|
void QVideoInputDevice::stop()
|
|
{
|
|
if(_timer != NULL)
|
|
{
|
|
QObject::disconnect(_timer,SIGNAL(timeout()),this,SLOT(grabFrame())) ;
|
|
_timer->stop() ;
|
|
delete _timer ;
|
|
_timer = NULL ;
|
|
}
|
|
if(_capture_device != NULL)
|
|
{
|
|
cvReleaseCapture(&_capture_device) ;
|
|
_capture_device = NULL ;
|
|
}
|
|
}
|
|
void QVideoInputDevice::start()
|
|
{
|
|
// make sure everything is re-initialised
|
|
//
|
|
stop() ;
|
|
|
|
// Initialise la capture
|
|
static const int cam_id = 0 ;
|
|
_capture_device = cvCaptureFromCAM(cam_id);
|
|
|
|
if(_capture_device == NULL)
|
|
{
|
|
std::cerr << "Cannot initialise camera. Something's wrong." << std::endl;
|
|
return ;
|
|
}
|
|
|
|
_timer = new QTimer ;
|
|
QObject::connect(_timer,SIGNAL(timeout()),this,SLOT(grabFrame())) ;
|
|
|
|
_timer->start(50) ; // 10 images per second.
|
|
}
|
|
|
|
void QVideoInputDevice::grabFrame()
|
|
{
|
|
IplImage *img=cvQueryFrame(_capture_device);
|
|
|
|
if(img == NULL)
|
|
{
|
|
std::cerr << "(EE) Cannot capture image from camera. Something's wrong." << std::endl;
|
|
return ;
|
|
}
|
|
// get the image data
|
|
|
|
if(img->nChannels != 3)
|
|
{
|
|
std::cerr << "(EE) expected 3 channels. Got " << img->nChannels << std::endl;
|
|
return ;
|
|
}
|
|
|
|
// convert to RGB and copy to new buffer, because cvQueryFrame tells us to not modify the buffer
|
|
cv::Mat img_rgb;
|
|
cv::cvtColor(cv::Mat(img), img_rgb, CV_BGR2RGB);
|
|
|
|
QImage image = QImage(img_rgb.data,img_rgb.cols,img_rgb.rows,QImage::Format_RGB888);
|
|
|
|
if(_video_processor != NULL)
|
|
{
|
|
uint32_t encoded_size ;
|
|
|
|
_video_processor->processImage(image,0,encoded_size) ;
|
|
|
|
std::cerr << "Encoded size = " << encoded_size << std::endl;
|
|
_total_encoded_size += encoded_size ;
|
|
|
|
time_t now = time(NULL) ;
|
|
|
|
if(now > _last_bw_estimate_TS)
|
|
{
|
|
_estimated_bw = uint32_t(0.75*_estimated_bw + 0.25 * (_total_encoded_size / (float)(now - _last_bw_estimate_TS))) ;
|
|
|
|
_total_encoded_size = 0 ;
|
|
_last_bw_estimate_TS = now ;
|
|
|
|
std::cerr << "new bw estimate: " << _estimated_bw << std::endl;
|
|
}
|
|
|
|
emit networkPacketReady() ;
|
|
}
|
|
if(_echo_output_device != NULL)
|
|
_echo_output_device->showFrame(image) ;
|
|
}
|
|
|
|
bool QVideoInputDevice::getNextEncodedPacket(RsVOIPDataChunk& chunk)
|
|
{
|
|
if(_video_processor)
|
|
return _video_processor->nextEncodedPacket(chunk) ;
|
|
else
|
|
return false ;
|
|
}
|
|
|
|
QVideoInputDevice::~QVideoInputDevice()
|
|
{
|
|
stop() ;
|
|
}
|
|
|
|
|
|
QVideoOutputDevice::QVideoOutputDevice(QWidget *parent)
|
|
: QLabel(parent)
|
|
{
|
|
showFrameOff() ;
|
|
}
|
|
|
|
void QVideoOutputDevice::showFrameOff()
|
|
{
|
|
setPixmap(QPixmap(":/images/video-icon-big.png").scaled(320,256,Qt::KeepAspectRatio,Qt::SmoothTransformation)) ;
|
|
}
|
|
|
|
void QVideoOutputDevice::showFrame(const QImage& img)
|
|
{
|
|
std::cerr << "img.size = " << img.width() << " x " << img.height() << std::endl;
|
|
setPixmap(QPixmap::fromImage(img).scaled( QSize(height()*640/480,height()),Qt::IgnoreAspectRatio,Qt::SmoothTransformation)) ;
|
|
}
|
|
|