You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
323 lines
11 KiB
323 lines
11 KiB
/***************************************************************************
|
|
* Copyright (C) 2004 by Jean-Baptiste Mardelle *
|
|
* bj@altern.org *
|
|
* *
|
|
* 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 <tqfile.h>
|
|
#include <config.h>
|
|
#include "tdefile_theora.h"
|
|
|
|
#include <kdebug.h>
|
|
#include <tdelocale.h>
|
|
#include <kgenericfactory.h>
|
|
|
|
#include "theora/theora.h"
|
|
#include "vorbis/codec.h"
|
|
|
|
ogg_stream_state t_stream_state;
|
|
ogg_stream_state v_stream_state;
|
|
|
|
int theora_p=0;
|
|
int vorbis_p=0;
|
|
|
|
static int queue_page(ogg_page *page)
|
|
{
|
|
if(theora_p)
|
|
ogg_stream_pagein(&t_stream_state,page);
|
|
if(vorbis_p)
|
|
ogg_stream_pagein(&v_stream_state,page);
|
|
return 0;
|
|
}
|
|
|
|
static int buffer_data(FILE *in,ogg_sync_state *oy)
|
|
{
|
|
char *buffer=ogg_sync_buffer(oy,4096);
|
|
int bytes=fread(buffer,1,4096,in);
|
|
ogg_sync_wrote(oy,bytes);
|
|
return(bytes);
|
|
}
|
|
|
|
typedef KGenericFactory<theoraPlugin> theoraFactory;
|
|
|
|
K_EXPORT_COMPONENT_FACTORY(tdefile_theora, theoraFactory( "tdefile_theora" ))
|
|
|
|
theoraPlugin::theoraPlugin(TQObject *parent, const char *name,
|
|
const TQStringList &args)
|
|
: KFilePlugin(parent, name, args)
|
|
{
|
|
// kdDebug(7034) << "theora plugin\n";
|
|
|
|
KFileMimeTypeInfo* info = addMimeTypeInfo( "video/x-theora" );
|
|
|
|
KFileMimeTypeInfo::GroupInfo* group = 0;
|
|
KFileMimeTypeInfo::ItemInfo* item;
|
|
|
|
// video group
|
|
|
|
group = addGroupInfo(info, "Video", i18n("Video Details"));
|
|
setAttributes(group, 0);
|
|
item = addItemInfo(group, "Length", i18n("Length"), TQVariant::Int);
|
|
setUnit(item, KFileMimeTypeInfo::Seconds);
|
|
setHint(item, KFileMimeTypeInfo::Length);
|
|
item = addItemInfo(group, "Resolution", i18n("Resolution"), TQVariant::Size);
|
|
setHint(item, KFileMimeTypeInfo::Size);
|
|
setUnit(item, KFileMimeTypeInfo::Pixels);
|
|
item = addItemInfo(group, "FrameRate", i18n("Frame Rate"), TQVariant::Int);
|
|
setUnit(item, KFileMimeTypeInfo::FramesPerSecond);
|
|
item = addItemInfo(group, "TargetBitrate", i18n("Target Bitrate"), TQVariant::Int);
|
|
setUnit(item, KFileMimeTypeInfo::Bitrate);
|
|
item = addItemInfo(group, "Quality", i18n("Quality"), TQVariant::Int);
|
|
|
|
// audio group
|
|
|
|
group = addGroupInfo(info, "Audio", i18n("Audio Details"));
|
|
setAttributes(group, 0);
|
|
addItemInfo(group, "Channels", i18n("Channels"), TQVariant::Int);
|
|
|
|
item = addItemInfo(group, "SampleRate", i18n("Sample Rate"), TQVariant::Int);
|
|
setUnit(item, KFileMimeTypeInfo::Hertz);
|
|
}
|
|
|
|
bool theoraPlugin::readInfo( KFileMetaInfo& info, uint what)
|
|
{
|
|
// most of the ogg stuff was borrowed from libtheora/examples/player_example.c
|
|
FILE *fp;
|
|
|
|
ogg_sync_state o_sync_state;
|
|
ogg_packet o_packet;
|
|
ogg_page o_page;
|
|
|
|
theora_info t_info;
|
|
theora_comment t_comment;
|
|
theora_state t_state;
|
|
vorbis_info v_info;
|
|
vorbis_comment v_comment;
|
|
|
|
theora_p=0;
|
|
vorbis_p=0;
|
|
int theora_serial=0;
|
|
int stateflag=0;
|
|
|
|
ogg_int64_t duration=0;
|
|
|
|
// libtheora is still a bit unstable and sadly the init_ functions don't
|
|
// take care of things the way one would expect. So, let's do some explicit
|
|
// clearing of these fields.
|
|
|
|
memset(&t_info, 0, sizeof(theora_info));
|
|
memset(&t_comment, 0, sizeof(theora_comment));
|
|
memset(&t_state, 0, sizeof(theora_state));
|
|
|
|
bool readTech = false;
|
|
|
|
if (what & (KFileMetaInfo::Fastest |
|
|
KFileMetaInfo::DontCare |
|
|
KFileMetaInfo::TechnicalInfo))
|
|
readTech = true;
|
|
|
|
if ( info.path().isEmpty() ) // remote file
|
|
return false;
|
|
|
|
fp = fopen(TQFile::encodeName(info.path()),"rb");
|
|
if (!fp)
|
|
{
|
|
kdDebug(7034) << "Unable to open " << TQFile::encodeName(info.path()).data() << endl;
|
|
return false;
|
|
}
|
|
|
|
ogg_sync_init(&o_sync_state);
|
|
|
|
/* init supporting Vorbis structures needed in header parsing */
|
|
vorbis_info_init(&v_info);
|
|
vorbis_comment_init(&v_comment);
|
|
|
|
/* init supporting Theora structures needed in header parsing */
|
|
theora_comment_init(&t_comment);
|
|
theora_info_init(&t_info);
|
|
|
|
while(!stateflag && buffer_data(fp,&o_sync_state)!=0)
|
|
{
|
|
while (ogg_sync_pageout(&o_sync_state,&o_page)>0)
|
|
{
|
|
ogg_stream_state stream_test;
|
|
/* is this a mandated initial header? If not, stop parsing */
|
|
if(!ogg_page_bos(&o_page))
|
|
{
|
|
queue_page(&o_page);
|
|
stateflag=1;
|
|
break;
|
|
}
|
|
|
|
ogg_stream_init(&stream_test,ogg_page_serialno(&o_page));
|
|
ogg_stream_pagein(&stream_test,&o_page);
|
|
ogg_stream_packetout(&stream_test,&o_packet);
|
|
|
|
/* identify the codec: try theora */
|
|
if(!theora_p && theora_decode_header(&t_info,&t_comment,&o_packet)>=0)
|
|
{
|
|
/* it is theora */
|
|
memcpy(&t_stream_state,&stream_test,sizeof(stream_test));
|
|
theora_serial=ogg_page_serialno(&o_page);
|
|
theora_p=1;
|
|
}
|
|
else if(!vorbis_p && vorbis_synthesis_headerin(&v_info,&v_comment,&o_packet)>=0)
|
|
{
|
|
/* it is vorbis */
|
|
memcpy(&v_stream_state,&stream_test,sizeof(stream_test));
|
|
vorbis_p=1;
|
|
}
|
|
else
|
|
{
|
|
/* whatever it is, we don't care about it */
|
|
ogg_stream_clear(&stream_test);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* we're expecting more header packets. */
|
|
bool corruptedHeaders=false;
|
|
|
|
while((theora_p && theora_p<3) || (vorbis_p && vorbis_p<3))
|
|
{
|
|
int ret;
|
|
/* look for further theora headers */
|
|
while(theora_p && (theora_p<3) && (ret=ogg_stream_packetout(&t_stream_state,&o_packet)))
|
|
{
|
|
if(ret<0)
|
|
{
|
|
kdDebug(7034)<<"Error parsing Theora stream headers; corrupt stream?\n"<<endl;
|
|
corruptedHeaders=true;
|
|
}
|
|
if(theora_decode_header(&t_info,&t_comment,&o_packet))
|
|
{
|
|
kdDebug(7034)<<"Error parsing Theora stream headers; corrupt stream?"<<endl;
|
|
corruptedHeaders=true;
|
|
}
|
|
theora_p++;
|
|
if(theora_p==3)
|
|
break;
|
|
}
|
|
|
|
/* look for more vorbis header packets */
|
|
while(vorbis_p && (vorbis_p<3) && (ret=ogg_stream_packetout(&v_stream_state,&o_packet)))
|
|
{
|
|
if(ret<0)
|
|
{
|
|
kdDebug(7034)<<"Error parsing Vorbis stream headers; corrupt stream"<<endl;
|
|
corruptedHeaders=true;
|
|
}
|
|
if(vorbis_synthesis_headerin(&v_info,&v_comment,&o_packet))
|
|
{
|
|
kdDebug(7034)<<"Error parsing Vorbis stream headers; corrupt stream?"<<endl;
|
|
corruptedHeaders=true;
|
|
}
|
|
vorbis_p++;
|
|
if(vorbis_p==3)
|
|
break;
|
|
}
|
|
/* The header pages/packets will arrive before anything else we
|
|
care about, or the stream is not obeying spec */
|
|
|
|
if(ogg_sync_pageout(&o_sync_state,&o_page)>0)
|
|
{
|
|
queue_page(&o_page);
|
|
/* demux into the appropriate stream */
|
|
}
|
|
else
|
|
{
|
|
int ret=buffer_data(fp,&o_sync_state); /* someone needs more data */
|
|
if(ret==0)
|
|
{
|
|
kdDebug(7034)<<"End of file while searching for codec headers."<<endl;
|
|
corruptedHeaders=true;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* and now we have it all. initialize decoders */
|
|
if(theora_p && !corruptedHeaders)
|
|
{
|
|
theora_decode_init(&t_state,&t_info);
|
|
}
|
|
else
|
|
{
|
|
/* tear down the partial theora setup */
|
|
theora_info_clear(&t_info);
|
|
theora_comment_clear(&t_comment);
|
|
|
|
vorbis_info_clear(&v_info);
|
|
vorbis_comment_clear(&v_comment);
|
|
ogg_sync_clear(&o_sync_state);
|
|
fclose(fp);
|
|
return false;
|
|
}
|
|
//queue_page(&o_page);
|
|
|
|
while (buffer_data(fp,&o_sync_state))
|
|
{
|
|
while (ogg_sync_pageout(&o_sync_state,&o_page)>0)
|
|
{
|
|
// The following line was commented out by Scott Wheeler <wheeler@kde.org>
|
|
// We don't actually need to store all of the pages / packets in memory since
|
|
// (a) libtheora doesn't use them anyway in the one call that we make after this
|
|
// that usese t_state and (b) it basically buffers the entire file to memory if
|
|
// we queue them up like this and that sucks where a typical file size is a few
|
|
// hundred megs.
|
|
|
|
// queue_page(&o_page);
|
|
}
|
|
if (theora_serial==ogg_page_serialno(&o_page))
|
|
duration=(ogg_int64_t) theora_granule_time(&t_state,ogg_page_granulepos(&o_page));
|
|
}
|
|
|
|
if (readTech)
|
|
{
|
|
int stream_fps=0;
|
|
if (t_info.fps_denominator!=0)
|
|
stream_fps=t_info.fps_numerator/t_info.fps_denominator;
|
|
KFileMetaInfoGroup videogroup = appendGroup(info, "Video");
|
|
appendItem(videogroup, "Length", int (duration));
|
|
appendItem(videogroup, "Resolution", TQSize(t_info.frame_width,t_info.frame_height));
|
|
appendItem(videogroup, "FrameRate", stream_fps);
|
|
appendItem(videogroup, "Quality", (int) t_info.quality);
|
|
|
|
KFileMetaInfoGroup audiogroup = appendGroup(info, "Audio");
|
|
appendItem(audiogroup, "Channels", v_info.channels);
|
|
appendItem(audiogroup, "SampleRate", int(v_info.rate));
|
|
}
|
|
fclose(fp);
|
|
|
|
if (vorbis_p)
|
|
{
|
|
ogg_stream_clear(&v_stream_state);
|
|
vorbis_comment_clear(&v_comment);
|
|
vorbis_info_clear(&v_info);
|
|
}
|
|
|
|
ogg_stream_clear(&t_stream_state);
|
|
theora_clear(&t_state);
|
|
theora_comment_clear(&t_comment);
|
|
theora_info_clear(&t_info);
|
|
ogg_sync_clear(&o_sync_state);
|
|
|
|
return true;
|
|
}
|
|
|
|
#include "tdefile_theora.moc"
|
|
|