mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-10-01 02:35:48 -04:00
framecatcher utility for capturing frames from videos (libxine dependency)
- initial ci git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@3787 b45a01b8-16f6-495d-af2f-9b41ad6348cc
This commit is contained in:
chrisparker126
parent
f6e4257442
commit
cb4f44ea39
373
retroshare-gui/src/util/filefunctions.cpp
Normal file
373
retroshare-gui/src/util/filefunctions.cpp
Normal file
@ -0,0 +1,373 @@
|
|||||||
|
/****************************************************************
|
||||||
|
* This file is distributed under the following license:
|
||||||
|
*
|
||||||
|
* Copyright (c) 2006 Raul E.
|
||||||
|
* Copyright (c) 2010 Christopher Evi-Parker
|
||||||
|
*
|
||||||
|
* 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 Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||||||
|
*
|
||||||
|
*************************************************************************/
|
||||||
|
|
||||||
|
|
||||||
|
#include "filefunctions.h"
|
||||||
|
|
||||||
|
QString filefunctions::cfilename=NULL;
|
||||||
|
QTreeWidget *filefunctions::treeObj=NULL;
|
||||||
|
QStringList filefunctions::tempimages;
|
||||||
|
QString filefunctions::codec=NULL;
|
||||||
|
int filefunctions::length=0;
|
||||||
|
int filefunctions::fwidth=0;
|
||||||
|
int filefunctions::fheight=0;
|
||||||
|
|
||||||
|
filefunctions::filefunctions(){
|
||||||
|
|
||||||
|
// clear tempfiles
|
||||||
|
for ( int i=0; i<filefunctions::tempimages.size();i++){
|
||||||
|
QFile::remove (filefunctions::tempimages.at(i));
|
||||||
|
}
|
||||||
|
filefunctions::tempimages.clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
filefunctions::~filefunctions(){
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
QString filefunctions::getCodec(){
|
||||||
|
return filefunctions::codec;
|
||||||
|
}
|
||||||
|
QString filefunctions::getImageWidth(){
|
||||||
|
return QString::number(filefunctions::fwidth);
|
||||||
|
}
|
||||||
|
QString filefunctions::getImageHeight(){
|
||||||
|
return QString::number(filefunctions::fheight);
|
||||||
|
}
|
||||||
|
int filefunctions::getLength(){ //ms
|
||||||
|
return filefunctions::length;
|
||||||
|
}
|
||||||
|
|
||||||
|
void filefunctions::open(){
|
||||||
|
vo_driver="auto";
|
||||||
|
//ao_driver="auto";
|
||||||
|
tempdir=QDir::tempPath ();
|
||||||
|
xine= xine_new();
|
||||||
|
xine_init(xine);
|
||||||
|
|
||||||
|
if((vo_port=xine_open_video_driver(xine,vo_driver,XINE_VISUAL_TYPE_NONE, NULL))==NULL)
|
||||||
|
qWarning("ERROR - OPEN VIDEO DRIVER");
|
||||||
|
//ao_port=xine_open_audio_driver(xine,ao_driver,NULL);
|
||||||
|
//stream = xine_stream_new(xine,ao_port,vo_port);
|
||||||
|
stream = xine_stream_new(xine,NULL,vo_port);
|
||||||
|
|
||||||
|
if(!xine_open(stream,cfilename.toLocal8Bit())){
|
||||||
|
const int errorCode= xine_get_error(stream);
|
||||||
|
QMessageBox::critical(0,"QFrameCatcher","ERROR.\n" + getXineError(errorCode));
|
||||||
|
return ;
|
||||||
|
}
|
||||||
|
if(xine_get_pos_length(stream,0,0,&length)){
|
||||||
|
//std::cout << length<< std::endl;
|
||||||
|
}
|
||||||
|
xine_play(stream,0,0);
|
||||||
|
|
||||||
|
|
||||||
|
const uint8_t option= 1;
|
||||||
|
|
||||||
|
switch (option){
|
||||||
|
case 1 :
|
||||||
|
getFrames(); // Capture x Frames
|
||||||
|
break;
|
||||||
|
case 2 :
|
||||||
|
getFramesTime(); // Capture every x seconds
|
||||||
|
break;
|
||||||
|
default :
|
||||||
|
getFrames();
|
||||||
|
}
|
||||||
|
|
||||||
|
xine_close(stream);
|
||||||
|
xine_dispose(stream);
|
||||||
|
xine_close_video_driver(xine,vo_port);
|
||||||
|
//xine_close_audio_driver(xine,ao_port);
|
||||||
|
}
|
||||||
|
void filefunctions::saveRGBImage(uchar *rgb32BitData,int pos,int width,int height){
|
||||||
|
|
||||||
|
currImage = new QImage(rgb32BitData, width, height,QImage::Format_RGB32);
|
||||||
|
|
||||||
|
//delete rgb32BitData;
|
||||||
|
|
||||||
|
}
|
||||||
|
void filefunctions::savebuffer(QByteArray &ba , QImage *image){
|
||||||
|
QBuffer buffer(&ba);
|
||||||
|
buffer.open(QIODevice::WriteOnly);
|
||||||
|
image->save(&buffer, "PNG");
|
||||||
|
}
|
||||||
|
bool filefunctions::saveimage(const QByteArray &ba , const QString &filename){
|
||||||
|
bool correct=true;
|
||||||
|
QFile image(filename);
|
||||||
|
if(!image.open(QIODevice::WriteOnly) )
|
||||||
|
return false;
|
||||||
|
|
||||||
|
if(image.write(ba)==-1)
|
||||||
|
correct=false;
|
||||||
|
|
||||||
|
image.close();
|
||||||
|
return correct;
|
||||||
|
}
|
||||||
|
void filefunctions::getFramesTime(){
|
||||||
|
xine_play(stream,0,0);
|
||||||
|
codec=xine_get_meta_info(stream, XINE_META_INFO_VIDEOCODEC);
|
||||||
|
|
||||||
|
const uint32_t chunk= (30) * 1000; // mseconds
|
||||||
|
|
||||||
|
|
||||||
|
// Progress Dialog
|
||||||
|
QString progresdialogtext =QApplication::translate("xine","Generating Thumbnails...",0,QApplication::UnicodeUTF8);
|
||||||
|
QString cancelbuttontext=QApplication::translate("xine","&Cancel",0,QApplication::UnicodeUTF8);
|
||||||
|
|
||||||
|
progress = new QProgressDialog(progresdialogtext , cancelbuttontext , 0, length,0,Qt::WindowSystemMenuHint);
|
||||||
|
|
||||||
|
|
||||||
|
for (int32_t frame=chunk ; frame<length ; frame=frame+chunk){
|
||||||
|
xine_play(stream,0,frame); // 1 frame 2 mseconds
|
||||||
|
SaveFrame(frame);
|
||||||
|
progress->setValue(frame);
|
||||||
|
qApp->processEvents();
|
||||||
|
if(progress->wasCanceled())
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
progress->setValue(length);
|
||||||
|
delete progress;
|
||||||
|
}
|
||||||
|
void filefunctions::getFrames(){
|
||||||
|
xine_play(stream,0,0);
|
||||||
|
codec=xine_get_meta_info(stream, XINE_META_INFO_VIDEOCODEC);
|
||||||
|
|
||||||
|
int frame = 20;
|
||||||
|
|
||||||
|
xine_play(stream,0,frame); // 1 frame 2 mseconds
|
||||||
|
SaveFrame(frame);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
void filefunctions::yuy2Toyv12 (uint8_t *y, uint8_t *u, uint8_t *v, uint8_t *input, int width, int height){
|
||||||
|
|
||||||
|
int i, j, w2;
|
||||||
|
|
||||||
|
w2 = width / 2;
|
||||||
|
|
||||||
|
for (i = 0; i < height; i += 2){
|
||||||
|
for (j = 0; j < w2; j++){
|
||||||
|
/*
|
||||||
|
* packed YUV 422 is: Y[i] U[i] Y[i+1] V[i]
|
||||||
|
*/
|
||||||
|
*(y++) = *(input++);
|
||||||
|
*(u++) = *(input++);
|
||||||
|
*(y++) = *(input++);
|
||||||
|
*(v++) = *(input++);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* down sampling
|
||||||
|
*/
|
||||||
|
|
||||||
|
for (j = 0; j < w2; j++){
|
||||||
|
/*
|
||||||
|
* skip every second line for U and V
|
||||||
|
*/
|
||||||
|
*(y++) = *(input++);
|
||||||
|
input++;
|
||||||
|
*(y++) = *(input++);
|
||||||
|
input++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
uchar * filefunctions::yv12ToRgb (uint8_t *src_y, uint8_t *src_u, uint8_t *src_v, int width, int height){
|
||||||
|
/*
|
||||||
|
* Create rgb data from yv12
|
||||||
|
*/
|
||||||
|
|
||||||
|
#define clip_8_bit(val) \
|
||||||
|
{ \
|
||||||
|
if (val < 0) \
|
||||||
|
val = 0; \
|
||||||
|
else \
|
||||||
|
if (val > 255) val = 255; \
|
||||||
|
}
|
||||||
|
|
||||||
|
int i, j;
|
||||||
|
|
||||||
|
int y, u, v;
|
||||||
|
int r, g, b;
|
||||||
|
|
||||||
|
int sub_i_uv;
|
||||||
|
int sub_j_uv;
|
||||||
|
|
||||||
|
int uv_width, uv_height;
|
||||||
|
|
||||||
|
uchar *rgb;
|
||||||
|
|
||||||
|
uv_width = width / 2;
|
||||||
|
uv_height = height / 2;
|
||||||
|
|
||||||
|
rgb = new uchar[(width * height * 4)]; //qt needs a 32bit align
|
||||||
|
if (!rgb) //qDebug ("Not enough memory!");
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
|
||||||
|
for (i = 0; i < height; ++i){
|
||||||
|
/*
|
||||||
|
* calculate u & v rows
|
||||||
|
*/
|
||||||
|
sub_i_uv = ((i * uv_height) / height);
|
||||||
|
|
||||||
|
for (j = 0; j < width; ++j){
|
||||||
|
/*
|
||||||
|
* calculate u & v columns
|
||||||
|
*/
|
||||||
|
sub_j_uv = ((j * uv_width) / width);
|
||||||
|
|
||||||
|
/***************************************************
|
||||||
|
*
|
||||||
|
* Colour conversion from
|
||||||
|
* http://www.inforamp.net/~poynton/notes/colour_and_gamma/ColorFAQ.html#RTFToC30
|
||||||
|
*
|
||||||
|
* Thanks to Billy Biggs <vektor@dumbterm.net>
|
||||||
|
* for the pointer and the following conversion.
|
||||||
|
*
|
||||||
|
* R' = [ 1.1644 0 1.5960 ] ([ Y' ] [ 16 ])
|
||||||
|
* G' = [ 1.1644 -0.3918 -0.8130 ] * ([ Cb ] - [ 128 ])
|
||||||
|
* B' = [ 1.1644 2.0172 0 ] ([ Cr ] [ 128 ])
|
||||||
|
*
|
||||||
|
* Where in xine the above values are represented as
|
||||||
|
*
|
||||||
|
* Y' == image->y
|
||||||
|
* Cb == image->u
|
||||||
|
* Cr == image->v
|
||||||
|
*
|
||||||
|
***************************************************/
|
||||||
|
|
||||||
|
y = src_y[(i * width) + j] - 16;
|
||||||
|
u = src_u[(sub_i_uv * uv_width) + sub_j_uv] - 128;
|
||||||
|
v = src_v[(sub_i_uv * uv_width) + sub_j_uv] - 128;
|
||||||
|
|
||||||
|
r = (int)((1.1644 * (double)y) + (1.5960 * (double)v));
|
||||||
|
g = (int)((1.1644 * (double)y) - (0.3918 * (double)u) - (0.8130 * (double)v));
|
||||||
|
b = (int)((1.1644 * (double)y) + (2.0172 * (double)u));
|
||||||
|
|
||||||
|
clip_8_bit (r);
|
||||||
|
clip_8_bit (g);
|
||||||
|
clip_8_bit (b);
|
||||||
|
|
||||||
|
rgb[(i * width + j) * 4 + 0] = b;
|
||||||
|
rgb[(i * width + j) * 4 + 1] = g;
|
||||||
|
rgb[(i * width + j) * 4 + 2] = r;
|
||||||
|
rgb[(i * width + j) * 4 + 3] = 0;
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return rgb;
|
||||||
|
}
|
||||||
|
void filefunctions::SaveFrame(int pos){
|
||||||
|
|
||||||
|
uint8_t *yuv = NULL, *y = NULL, *u = NULL, *v =NULL ;
|
||||||
|
|
||||||
|
int width, height, ratio, format;
|
||||||
|
|
||||||
|
// double desired_ratio, image_ratio;
|
||||||
|
|
||||||
|
if (!xine_get_current_frame(stream, &width, &height, &ratio, &format, NULL))
|
||||||
|
return;
|
||||||
|
|
||||||
|
yuv = new uint8_t[((width+8) * (height+1) * 2)];
|
||||||
|
if (yuv == NULL){
|
||||||
|
qWarning("Not enough memory to make screenshot!");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
xine_get_current_frame(stream, &width, &height, &ratio, &format, yuv);
|
||||||
|
fwidth=width;
|
||||||
|
fheight=height;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* convert to yv12 if necessary
|
||||||
|
*/
|
||||||
|
switch (format){
|
||||||
|
case XINE_IMGFMT_YUY2:
|
||||||
|
{
|
||||||
|
uint8_t *yuy2 = yuv;
|
||||||
|
|
||||||
|
yuv = new uint8_t[(width * height * 2)];
|
||||||
|
if (yuv == NULL){
|
||||||
|
qWarning("Not enough memory to make screenshot!");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
y = yuv;
|
||||||
|
u = yuv + width * height;
|
||||||
|
v = yuv + width * height * 5 / 4;
|
||||||
|
|
||||||
|
yuy2Toyv12 (y, u, v, yuy2, width, height);
|
||||||
|
|
||||||
|
delete [] yuy2;
|
||||||
|
//free (yuy2);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case XINE_IMGFMT_YV12:
|
||||||
|
y = yuv;
|
||||||
|
u = yuv + width * height;
|
||||||
|
v = yuv + width * height * 5 / 4;
|
||||||
|
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
{
|
||||||
|
qWarning("Screenshot: Format %s not supported!",(char *) &format);
|
||||||
|
delete [] yuv;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
uchar* rgb32BitData = yv12ToRgb (y, u, v, width, height);
|
||||||
|
delete [] yuv;
|
||||||
|
saveRGBImage(rgb32BitData,pos,width,height);
|
||||||
|
}
|
||||||
|
QTime filefunctions::postime(const int itime){
|
||||||
|
QTime time(0,0,0);
|
||||||
|
time=time.addMSecs(itime);
|
||||||
|
return time;
|
||||||
|
}
|
||||||
|
QString filefunctions::getXineError(int errorCode){
|
||||||
|
QString error;
|
||||||
|
switch (errorCode){
|
||||||
|
case XINE_ERROR_NO_INPUT_PLUGIN:
|
||||||
|
case XINE_ERROR_NO_DEMUX_PLUGIN: {
|
||||||
|
error = QApplication::translate("xine","No plugin found to handle this resource",0,QApplication::UnicodeUTF8);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case XINE_ERROR_DEMUX_FAILED: {
|
||||||
|
error = QApplication::translate("xine","Resource seems to be broken",0,QApplication::UnicodeUTF8);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case XINE_ERROR_MALFORMED_MRL: {
|
||||||
|
error = QApplication::translate("xine","Requested resource does not exist",0,QApplication::UnicodeUTF8);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case XINE_ERROR_INPUT_FAILED: {
|
||||||
|
error = QApplication::translate("xine","Resource can not be opened",0,QApplication::UnicodeUTF8);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
default: {
|
||||||
|
error = QApplication::translate("xine","Unknown error",0,QApplication::UnicodeUTF8);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return error;
|
||||||
|
}
|
||||||
67
retroshare-gui/src/util/filefunctions.h
Normal file
67
retroshare-gui/src/util/filefunctions.h
Normal file
@ -0,0 +1,67 @@
|
|||||||
|
#ifndef FILEFUNCTIONS_H
|
||||||
|
#define FILEFUNCTIONS_H
|
||||||
|
|
||||||
|
#include <xine.h>
|
||||||
|
#include <xine/xineutils.h>
|
||||||
|
#include <QString>
|
||||||
|
#include <QStringList>
|
||||||
|
#include <QDir>
|
||||||
|
#include <QTreeWidgetItem>
|
||||||
|
#include <QTime>
|
||||||
|
#include <QPixmap>
|
||||||
|
#include <QIcon>
|
||||||
|
#include <QImage>
|
||||||
|
#include <QByteArray>
|
||||||
|
#include <QBuffer>
|
||||||
|
#include <QMessageBox>
|
||||||
|
#include <QProgressDialog>
|
||||||
|
#include <QApplication>
|
||||||
|
|
||||||
|
|
||||||
|
class filefunctions{
|
||||||
|
|
||||||
|
private:
|
||||||
|
xine_t *xine;
|
||||||
|
xine_stream_t *stream;
|
||||||
|
xine_video_port_t *vo_port;
|
||||||
|
xine_audio_port_t *ao_port;
|
||||||
|
xine_event_queue_t *event_queue;
|
||||||
|
char *vo_driver;
|
||||||
|
char *ao_driver;
|
||||||
|
|
||||||
|
QString tempdir;
|
||||||
|
|
||||||
|
static QString codec;
|
||||||
|
static int length; // length of the stream
|
||||||
|
static int fwidth;
|
||||||
|
static int fheight;
|
||||||
|
void saveRGBImage(uchar *,int,int,int);
|
||||||
|
QString getXineError(int);
|
||||||
|
QProgressDialog *progress;
|
||||||
|
void savebuffer(QByteArray &, QImage *);
|
||||||
|
bool saveimage(const QByteArray & , const QString &);
|
||||||
|
|
||||||
|
public:
|
||||||
|
filefunctions();
|
||||||
|
~filefunctions();
|
||||||
|
static QString cfilename;
|
||||||
|
static QTreeWidget *treeObj;
|
||||||
|
static QStringList tempimages;
|
||||||
|
QImage* currImage;
|
||||||
|
|
||||||
|
static QString getCodec();
|
||||||
|
static QString getImageWidth();
|
||||||
|
static QString getImageHeight();
|
||||||
|
static int getLength();
|
||||||
|
|
||||||
|
void open();
|
||||||
|
void getFrames();
|
||||||
|
void getFramesTime();
|
||||||
|
|
||||||
|
void yuy2Toyv12 (uint8_t *, uint8_t *, uint8_t *, uint8_t *, int , int );
|
||||||
|
uchar * yv12ToRgb (uint8_t *, uint8_t *, uint8_t *, int , int );
|
||||||
|
void SaveFrame(int);
|
||||||
|
QTime postime(const int);
|
||||||
|
|
||||||
|
};
|
||||||
|
#endif
|
||||||
Loading…
Reference in New Issue
Block a user