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.
k9copy/k9author/k9newdvd.cpp

440 lines
14 KiB

//
// C++ Implementation: k9newdvd
//
// Description:
//
//
// Author: Jean-Michel PETIT <k9copy@free.fr>, (C) 2007
//
// Copyright: See COPYING file that comes with this distribution
//
//
#include "k9newdvd.h"
#include "k9title.h"
#include "k9tools.h"
#include <tqfile.h>
#include <stdio.h>
#include <tqtextstream.h>
#include <kstandarddirs.h>
#include <tqapplication.h>
#include <ktempfile.h>
#include <kapplication.h>
#include <klocale.h>
#include <tqimage.h>
#include <kmessagebox.h>
#include "k9menu.h"
#include "k9menubutton.h"
#include "k9processlist.h"
#include <tqthread.h>
#include <tqfileinfo.h>
k9NewDVD::k9NewDVD(TQObject *parent, const char *name)
: TQObject(parent, name) {
m_workDir=locateLocal("tmp", "k9copy/" ) ;
m_rootMenu=new k9Menu(this);
m_format=PAL;
}
k9NewDVD::~k9NewDVD() {}
int k9NewDVDItems::compareItems(TQPtrCollection::Item item1,TQPtrCollection::Item item2) {
k9Title *_i1=(k9Title*) item1;
k9Title *_i2=(k9Title*) item2;
return (_i1->getNum() - _i2->getNum());
}
void k9NewDVD::execute() {
m_cancel=false;
m_error="";
m_config=new k9Config();
// connect(m_process, TQT_SIGNAL(receivedStderr(KProcess *, char *, int)),this, TQT_SLOT(getStderr(KProcess *, char *, int) ));
connect(&m_aviDecode,TQT_SIGNAL(drawFrame(TQImage*)),this,TQT_SLOT(drawImage(TQImage*)));
k9Tools::clearOutput(m_workDir+"dvd");
createXML();
disconnect(&m_aviDecode,TQT_SIGNAL(drawFrame(TQImage*)),this,TQT_SLOT(drawImage(TQImage*)));
delete m_config;
}
#include "k9newdvd.moc"
void k9NewDVD::drawImage(TQImage * _image) {
// m_progress->setImage(*_image);
}
void k9NewDVD::setFormat ( const eFormat& _value ) {
m_format = _value;
m_rootMenu->setFormat((k9Menu::eFormat)_value);
for (k9Title *title=m_titles.first();title;title=m_titles.next()) {
title->getMenu()->setFormat((k9Menu::eFormat)_value);
}
}
void k9NewDVD::createXML() {
m_rootMenu->setWorkDir(m_workDir);
TQString menuFileName=m_workDir+TDEApplication::randomString(8)+".mpg";
m_rootMenu->setMenuFileName(menuFileName);
m_xml=new TQDomDocument();
TQDomElement root = m_xml->createElement( "dvdauthor" );
root.setAttribute ("dest",m_workDir+"dvd");
m_xml->appendChild( root );
// Create vmgm menu
TQDomElement vmgm = m_xml->createElement("vmgm");
root.appendChild(vmgm);
m_processList->addProgress(i18n("Creating root menu"));
m_rootMenu->createMenus(&vmgm);
addTitles(root);
m_processList->execute();
m_totalEncodedSize=0;
m_offset=0;
m_lastvalue=0;
for ( TQStringList::Iterator it = m_tmpFiles.begin(); it != m_tmpFiles.end(); ++it ) {
TQString file= *it;
if (file.endsWith(".mpeg")) {
TQFileInfo f(file);
m_totalEncodedSize+=f.size();
}
}
m_totalEncodedSize/=1024*1024;
m_cancel=m_processList->getCancel();
bool error=false;
if (!m_cancel) {
TQString dvdAuthor(m_workDir+"/"+TDEApplication::randomString(8)+".xml");
TQFile file( dvdAuthor);
file.open(IO_WriteOnly);
TQTextStream stream( &file );
m_xml->save(stream,1);
file.close();
m_processList->clear();
k9Process *process=m_processList->addProcess(i18n("authoring"));
connect(process, TQT_SIGNAL(receivedStderr(KProcess *, char *, int)),this, TQT_SLOT(getStderr(KProcess *, char *, int) ));
*process << "dvdauthor" << "-x" << dvdAuthor;
m_processList->execute();
m_cancel=m_processList->getCancel();
error=m_processList->getError();
// else
// m_error=i18n("An error occured while running dvdauthor");
file.remove();
}
if (m_cancel)
m_error=i18n("The dvd authoring was canceled");
else if (error)
m_error=i18n("An error occured while running DVDAuthor:\n")+ m_stdout;
TQFile::remove(menuFileName);
for ( TQStringList::Iterator it = m_tmpFiles.begin(); it != m_tmpFiles.end(); ++it ) {
TQFile::remove(*it);
}
m_tmpFiles.clear();
if (m_error !="") {
KMessageBox::error( 0, m_error, i18n("Authoring"));
}
delete m_xml;
}
void k9NewDVD::addTitles (TQDomElement &_root) {
calcVideoBitrate();
for (k9Title *title=m_titles.first();title && !m_cancel;title=m_titles.next()) {
TQDomElement titleSet = m_xml->createElement("titleset");
_root.appendChild(titleSet);
TQDomElement pgc;
k9Menu *menu=title->getMenu();
menu->setWorkDir(m_workDir);
TQString menuFileName=m_workDir+TDEApplication::randomString(8)+".mpg";
m_tmpFiles << menuFileName,
menu->setMenuFileName(menuFileName);
m_processList->addProgress(i18n("Creating menu for title %1").arg(title->getNum()+1));
menu->createMenus(&titleSet);
TQDomElement eTitle=m_xml->createElement("titles");
titleSet.appendChild(eTitle);
TQDomElement e=m_xml->createElement("video");
e.setAttribute("aspect","16:9");
e.setAttribute("format",m_format==PAL?"PAL":"NTSC");
// if (l_track->getaspectRatio()!="4:3") {
e.setAttribute("widescreen","nopanscan");
// }setProgressWindow
eTitle.appendChild(e);
e=m_xml->createElement("audio");
e.setAttribute("format","ac3");
e.setAttribute("channels","2");
eTitle.appendChild(e);
pgc=m_xml->createElement("pgc");
eTitle.appendChild(pgc);
TQDomElement post=m_xml->createElement("post");
pgc.appendChild(post);
TQDomText txt=m_xml->createTextNode(title->getMenu()->getEndScript());
post.appendChild(txt);
TQPtrList <k9AviFile > *l=title->getFiles();
for (k9AviFile *aviFile= l->first();aviFile && !m_cancel;aviFile=l->next()) {
if ( aviFile->getPrevious()==NULL || aviFile->getBreakPrevious()) {
TQString cmd="",chapters="";
createMencoderCmd(cmd,chapters,aviFile);
e=m_xml->createElement("vob");
e.setAttribute("file",cmd);
e.setAttribute("chapters",chapters);
pgc.appendChild(e);
m_tmpFiles << cmd;
}
}
}
}
void k9NewDVD::setWorkDir ( const TQString& _value ) {
m_workDir = _value;
if (!m_workDir.endsWith("/"))
m_workDir +="/";
}
void k9NewDVD::createMencoderCmd(TQString &_cmd,TQString &_chapters, k9AviFile *_aviFile) {
// m_aviDecode.open(_aviFile->getFileName());
m_timer.start();
m_timer2.start();
m_timer3.start();
TQTime end;
k9AviFile *file=_aviFile;
bool bEnd;
_chapters="0";
do {
end=file->getEnd();
bEnd= (file->getNext()==NULL) || (file->getBreakNext());
file=file->getNext();
if (!bEnd) {
int lt=_aviFile->getStart().msecsTo(end);
TQTime t;
t=t.addMSecs(lt);
_chapters +="," + t.toString("hh:mm:ss");
}
} while (!bEnd);
TQString fileName= m_workDir + TDEApplication::randomString(8)+".mpeg";
TQString t1=_aviFile->getStart().toString("hh:mm:ss.zzz");
int length=_aviFile->getStart().msecsTo(end);
TQTime l;
l=l.addMSecs(length);
TQString t2=l.toString("hh:mm:ss.zzz");
TQString scale;
TQString fps;
switch (m_format) {
case PAL:
scale="720:576";
fps="25";
break;
case NTSC:
scale="720:480";
fps="30000/1001";
break;
}
k9Process *process=m_processList->addProcess(i18n("Encoding %1").arg(_aviFile->getFileName()));
m_processList->setFileName(process,_aviFile->getFileName());
TQTime t(0,0);
t.start();
m_timers[process]=t;
connect(process, TQT_SIGNAL(receivedStdout(KProcess *, char *, int)),this, TQT_SLOT(getStdout(KProcess *, char *, int) ));
//m_progress->setTitle(i18n("Encoding file"));
//m_process->clearArguments();
*process << "mencoder" << "-oac" << "lavc" << "-ovc" << "lavc" << "-of" << "mpeg" <<"-afm" <<"libmad";
*process << "-mpegopts" << "format=dvd" << "-vf" << "scale="+scale+",harddup" << "-srate" << "48000" << "-af" << "lavcresample=48000";
*process << "-lavcopts" << TQString("vcodec=mpeg2video:vrc_buf_size=1835:vrc_maxrate=9800:vbitrate=%1:keyint=15:acodec=%3:abitrate=%2:aspect=16/9").arg(m_videoBitrate).arg(m_config->getPrefAudioBitrate()).arg(m_config->getPrefAudioFormat().lower());
*process << "-ofps" << fps << "-o" << fileName << "-ss" << t1 << "-endpos" << t2 << _aviFile->getFileName();
tqDebug("%s", process->debug().ascii());
/*
if (!m_progress->execute()) {
m_cancel=true;
if (m_progress->getCanceled())
m_error=i18n("The dvd authoring was canceled");
else
m_error=i18n("An error occured while transcoding video");
}
*/
_cmd=fileName;
// m_aviDecode.close();
}
void k9NewDVD::getStderr(KProcess *_process, char *_buffer, int _length) {
TQCString tmp(_buffer,_length);
m_stdout=tmp;
int pos;
if (tmp.contains("STAT:")) {
pos=tmp.find("fixing VOBU");
if (pos!=-1) {
TQString tmp2=tmp;
// m_progress->setTitle(i18n("Authoring"));
// m_progress->setLabelText(i18n("Fixing VOBUS"));
int end=tmp2.find("%");
if (end!=-1) {
pos =end -2;
tmp2=tmp2.mid(pos,end-pos);
tmp2=tmp2.stripWhiteSpace();
// m_progress->setProgress(tmp2.toInt(),100);
}
} else {
pos=tmp.find("STAT: VOBU ");
if (pos !=-1) {
TQCString tmp2(_buffer+pos,_length-pos);
int vobu,mb;
sscanf(tmp2.data(),"STAT: VOBU %d at %dMB",&vobu,&mb);
if (mb <m_lastvalue)
m_offset+=m_lastvalue;
m_lastvalue=mb;
m_processList->setProgress((k9Process*)_process,mb+m_offset,m_totalEncodedSize);
}
}
//STAT: VOBU 16 at 3MB, 1 PGCS
}
}
void k9NewDVD::getStdout(KProcess *_process, char *_buffer, int _length) {
k9Process *process=(k9Process*) _process;
if (m_timers[process].elapsed() >500) {
TQCString tmp(_buffer,_length);
int pos=tmp.find("Pos:");
if (pos!=-1) {
TQString tmp2=tmp.mid(pos);
tmp2=tmp2.replace(":",": ").replace("(","").replace(")","").simplifyWhiteSpace();
TQStringList sl=TQStringList::split(" ",tmp2);
float position;
sscanf(sl[1].latin1(),"%fs",&position);
int frame;
sscanf(sl[2].latin1(),"%df",&frame);
int percent;
sscanf(sl[3].latin1(),"%d",&percent);
int fps;
sscanf(sl[4].latin1(),"%d",&fps);
m_processList->setProgress(process,percent,100);
m_processList->setPos(process,position);
//m_progress->setProgress(percent,100);
// if (percent>0 &&m_timer3.elapsed() >1000 ) {
if (percent>0 ) {
int elapsed=process->getElapsed();
TQTime time2(0,0);
time2=time2.addMSecs(elapsed);
TQTime time3(0,0);
float fprc=percent/100.0;
time3=time3.addMSecs((uint32_t)(elapsed*(1.0/fprc)));
m_processList->setText(process,time2.toString("hh:mm:ss") +" / " + time3.toString("hh:mm:ss"),1);
m_timer3.restart();
}
TQString text;//=i18n("filename") + " : " + m_aviDecode.getFileName();
text=i18n("fps")+ " : "+TQString::number(fps);
m_processList->setText(process,text,2);
/*
m_progress->setLabelText(text);
if (m_timer.elapsed() > 5000) {
m_timer.restart();
if (m_aviDecode.opened()) {
m_aviDecode.readFrame(position);
}
}
*/
}
m_timers[process].restart();
}
}
void k9NewDVD::appendTitle(k9Title *_title) {
m_config=new k9Config();
m_titles.append(_title);
m_titles.sort();
//create the menu button
k9MenuButton *btn=m_rootMenu->addButton();
_title->setButton(btn);
btn->setNum(_title->getNum()+1);
TQPixmap px(m_config->getPrefButtonWidth(),m_config->getPrefButtonHeight());
px.fill(TQt::black);
TQImage img=px.convertToImage();
btn->setImage(img);
int nbColumn=(720-50)/(m_config->getPrefButtonWidth()+50);
int top=(int) _title->getNum()/nbColumn ;
int left=_title->getNum() %nbColumn;
btn->setTop(top*(m_config->getPrefButtonHeight()+20) +50);
btn->setLeft(left*(m_config->getPrefButtonWidth()+50) +50);
btn->setWidth(m_config->getPrefButtonWidth());
btn->setHeight(m_config->getPrefButtonHeight());
btn->setScript(TQString("g1=0;jump titleset %1 menu;").arg(_title->getNum()+1));
btn->setTextPosition(k9MenuButton::RIGHT);
btn->setText(i18n("title %1").arg(_title->getNum()+1));
btn->setColor(m_config->getPrefButtonTextColor());
btn->setFont(m_config->getPrefButtonFont());
TQString script="\n";
for (k9Title *t = m_titles.first();t;t=m_titles.next()) {
script +=TQString("if (g6== %1) { g6=0; jump titleset %2 menu;}\n").arg(t->getNum()+1).arg(t->getNum()+1);
}
m_rootMenu->setStartScript2(script);
emit sigAddTitle();
delete m_config;
}
void k9NewDVD::setProcessList(k9ProcessList *_value) {
m_processList=_value;
}
k9NewDVD::eFormat k9NewDVD::getFormat() const {
return m_format;
}
k9Menu* k9NewDVD::getRootMenu() const {
return m_rootMenu;
}
void k9NewDVD::calcVideoBitrate() {
// bitrate video = (MB *8388.608) /SEC - bitrate audio
int length=0;
for (k9Title *title=m_titles.first();title;title=m_titles.next()) {
k9TitleItems *chapters=title->getFiles();
for (k9AviFile *chapter=chapters->first();chapter;chapter=chapters->next()) {
length+=chapter->getStart().msecsTo(chapter->getEnd());
}
}
int size=m_config->getPrefSize();
double sec=(double)length/1000.0;
// m_videoBitrate=(int)( (size * 8388.608)/sec - 192);
m_videoBitrate=8*((size*1024 - (m_config->getPrefAudioBitrate() * sec/8))/sec);
m_videoBitrate=TQMIN(m_videoBitrate,9800);
}
int k9NewDVD::getTotalTime() {
int total=0;
for (k9Title * title=m_titles.first();title;title=m_titles.next()) {
k9TitleItems *chapters=title->getFiles();
for (k9AviFile *chapter=chapters->first();chapter;chapter=chapters->next()) {
total+=chapter->getStart().secsTo(chapter->getEnd());
}
}
return total;
}
TQString k9NewDVD::getError() const {
return m_error;
}