k9copy/k9author/k9menu.cpp

//
// C++ Implementation: k9menu
//
// Description:
//
//
// Author: Jean-Michel PETIT <k9copy@free.fr>, (C) 2007
//
// Copyright: See COPYING file that comes with this distribution
//
//
#include "k9menu.h"
#include "k9menubutton.h"
#include "k9process.h"
#include <tqpainter.h>
#include <tqbitmap.h>
#include <tdeapplication.h>
#include <tqfile.h>
#include <tqdom.h>
#include "k9tools.h"

k9Menu::k9Menu(TQObject *parent, const char *name)
        : TQObject(parent, name),m_format(PAL) {
    m_buttons.setAutoDelete(false);
    m_canvas=new TQCanvas(this);
    TQPixmap pix(720,576);
    pix.fill(TQt::black);
    m_canvas->setBackgroundPixmap(pix);
    m_startScript=m_startScript2=m_endScript="";
}


k9Menu::~k9Menu() {}


#include "k9menu.moc"



k9MenuButton *k9Menu::addButton() {
    k9MenuButton *button=new k9MenuButton(m_canvas);
    m_buttons.append(button);
    return button;

}



void k9Menu::createMenus(TQDomElement *_rootNode) {
  
    m_config=new k9Config();
    int height=m_format==PAL?576:480;

    TQImage img;

    //create menu image
    TQCanvasItemList l=m_canvas->allItems();
    for (TQCanvasItemList::Iterator it=l.begin(); it!=l.end(); ++it) {
        if ( (*it)->rtti() > 1001  || (*it)->rtti()==TQCanvasItem::Rtti_Rectangle ) 
            (*it)->hide();
    }


    TQPixmap pixbg(720,height);
    TQPainter pbg(&pixbg);
    m_canvas->resize(720,height);
    m_canvas->update();
    m_canvas->drawArea(TQRect(0,0,720,height),&pbg);

    //create the hilite picture
    int lineWidth=2;
    img.create(720,height,8,3,TQImage::IgnoreEndian);
    TQPixmap pixhi;
    pixhi.convertFromImage(img,0);
    pixhi.fill(TQt::white);
    TQPainter phi(&pixhi);
    TQPen penhi;
    penhi.setColor(m_config->getPrefButtonHiliteColor());
    penhi.setWidth(lineWidth*2);
    phi.setPen(penhi);
    phi.setBrush(TQt::NoBrush);
    //create the hilite mask
    TQBitmap mask(720,height);
    mask.fill(TQt::color0);
    TQPainter pmk(&mask);
    TQPen penmk;
    penmk.setColor(TQt::color1);
    penmk.setWidth(lineWidth*2);
    pmk.setPen(penmk);
    pmk.setBrush(TQt::NoBrush);

    //draw buttons
    for (k9MenuButton *button=m_buttons.first();button;button=m_buttons.next()) {
        //draw hilight
        phi.drawRect(button->getLeft()+lineWidth,button->getTop()+lineWidth,button->getWidth()-lineWidth,button->getHeight()-lineWidth);
        pmk.drawRect(button->getLeft()+lineWidth,button->getTop()+lineWidth,button->getWidth()-lineWidth,button->getHeight()-lineWidth);
            
    }
    pbg.end();
    phi.end();
    pmk.end();
    pixhi.setMask(mask);
    //save menu image

    TQString bgFileName=m_workDir+TDEApplication::randomString(8)+".jpg";
    TQString hiFileName=m_workDir+TDEApplication::randomString(8)+".png";
    TQString m2vFileName= m_workDir+TDEApplication::randomString(8)+".m2v";
    TQString mpaFileName= m_workDir+TDEApplication::randomString(8)+".m2v";
    TQString mpgFileName= m_workDir+TDEApplication::randomString(8)+".mpg";

    pixbg.save(bgFileName,"JPEG",-1);
    pixhi.save(hiFileName,"PNG",-1);

    convertJpegToMpeg(bgFileName,m2vFileName);
    createAudio(mpaFileName);
    multiplex(mpaFileName,m2vFileName,mpgFileName);
    spumux(hiFileName,mpgFileName);
    appendMenu(_rootNode);

//    mask.save(m_workDir+"/mask.bmp","BMP",-1);

    TQFile::remove(bgFileName);
    TQFile::remove(m2vFileName);
    TQFile::remove(mpaFileName);
    TQFile::remove(mpgFileName);
    TQFile::remove(hiFileName);
    delete m_config;
}

void k9Menu::convertJpegToMpeg(const TQString &_imageJpg,const TQString &_imageMpg) {
    k9Process *process=new k9Process(0,0);
    process->setWorkingDirectory(m_workDir);
    process->setUseShell(true);
    if (m_format==NTSC) {
        *process << "jpeg2yuv" << "-n" << "50" << "-I" << "p" << "-f" << "29.97" << "-j" << _imageJpg << "|";
        *process << "mpeg2enc" << "-n" << "n"  <<"-f" << "8" << "-o" << _imageMpg;
    } else {
        *process << "jpeg2yuv" << "-n" << "50" << "-I" << "p" << "-f" << "25" << "-j" << _imageJpg << "|";
        *process << "mpeg2enc" << "-n" << "p"  <<"-f" << "8" << "-o" << _imageMpg;
    }
    process->start();
    process->sync();
    if (!process->normalExit()) {
        tqDebug("error converting jpeg to mpeg");
    }
    tqDebug("%s", process->debug().latin1());
}

void k9Menu::createAudio(const TQString & _audioFile) {
    k9Process *process=new k9Process(0,0);
    process->setWorkingDirectory(m_workDir);
    process->setUseShell(true);
    if (k9Tools::checkProgram("toolame"))
	    *process << "dd" << "if=/dev/zero" << "bs=4" << "count=1920" << "|" << "toolame"  << "-b" << "128"  << "-s" << "48" << "/dev/stdin" << _audioFile;
    else
	    *process << "dd" << "if=/dev/zero" << "bs=4" << "count=1920" << "|" << "twolame" <<"-r" << "-b" << "128"  << "-s" << "48000" << "/dev/stdin" << _audioFile;

    process->start();
    process->sync();
    if (!process->normalExit()) {
        tqDebug("error creating audio stream");
    }
}
void k9Menu::multiplex(const TQString &_audio, const TQString &_video, const TQString _result) {
    k9Process *process=new k9Process(0,0);
    process->setWorkingDirectory(m_workDir);
    process->setUseShell(true);
    *process << "mplex" <<  "-f" << "8" << "-o" << _result << _video << _audio;
    process->start();
    process->sync();
    if (!process->normalExit()) {
        tqDebug("error multiplexing audio and video");
    }
}

void k9Menu::spumux(const TQString &_hiFileName,const TQString &_mpgFileName) {
    TQDomElement stream,spu;

    TQDomDocument xml;

    TQDomElement root = xml.createElement( "subpictures" );
    xml.appendChild( root );

    stream=xml.createElement("stream");
    root.appendChild(stream);

    spu=xml.createElement("spu");
    stream.appendChild(spu);

    spu.setAttribute("start","00:00:00.0");
    spu.setAttribute("end","00:00:00.0");
    spu.setAttribute("highlight",_hiFileName);
    spu.setAttribute("select",_hiFileName);

    //spu.setAttribute("autooutline","infer");
    spu.setAttribute("force","yes");
 //   spu.setAttribute("autoorder","rows");
    spu.setAttribute("xoffset","0");
    spu.setAttribute("yoffset","0");

    int i=1;
    for (k9MenuButton *button=m_buttons.first();button;button=m_buttons.next()) {
        TQDomElement eButton = xml.createElement("button");
        spu.appendChild(eButton);
        eButton.setAttribute("name",TQString("button%1").arg(i));
        eButton.setAttribute("x0", TQString::number(button->getLeft()));
        eButton.setAttribute("y0",TQString::number(button->getTop()));
        eButton.setAttribute("x1",TQString::number(button->getLeft()+button->getWidth()));
        eButton.setAttribute("y1",TQString::number(button->getTop()+button->getHeight()));
        i++;
    }

    TQString spuFileName(m_workDir + TDEApplication::randomString(8)+ ".xml");
    TQFile file( spuFileName );
    if ( file.open( IO_WriteOnly ) ) {
        TQTextStream stream( &file );
        xml.save(stream,1);
        file.close();
    }


    k9Process *process=new k9Process(0,0);
    process->setWorkingDirectory(m_workDir);
    process->setUseShell(true);
//    *process << "mplex" <<  "-f" << "8" << "-o" << _result << _video << _audio;
    *process << "spumux" << spuFileName << "<" << _mpgFileName << ">" << m_menuFileName;
    process->start();
    process->sync();
    if (!process->normalExit()) {
        tqDebug("error multiplexing audio and video");
    }

    file.remove();
}

void k9Menu::appendMenu(TQDomElement *_rootNode) {
    TQDomDocument doc=_rootNode->ownerDocument();
    TQDomNodeList l=_rootNode->elementsByTagName("menus");
    TQDomNode menus;
    if (l.count()==0) {
        menus=doc.createElement("menus");
        _rootNode->appendChild(menus);
    } else {
        menus=l.item(0);
    }
   
    TQDomElement pgc=doc.createElement("pgc");
    menus.appendChild(pgc);
    TQDomElement pre=doc.createElement("pre");
    pgc.appendChild(pre);
    TQDomText txt=doc.createTextNode(m_startScript +m_startScript2);
    pre.appendChild(txt);

    TQDomElement vob=doc.createElement("vob");
    pgc.appendChild(vob);
    vob.setAttribute("file",m_menuFileName);
    vob.setAttribute("pause","inf");
    int i=1;
    for (k9MenuButton *button=m_buttons.first();button;button=m_buttons.next()) {
        TQDomElement eButton = doc.createElement("button");
        pgc.appendChild(eButton); 
        TQDomText eButtonTxt=doc.createTextNode(button->getScript());
        eButton.appendChild(eButtonTxt);
        eButton.setAttribute("name",TQString("button%1").arg(i));
        i++;
    }
    

}

TQString k9Menu::getWorkDir() const {
    return m_workDir;
}


void k9Menu::setWorkDir(const TQString& _value) {
    m_workDir = _value;
}


void k9Menu::setFormat(const eFormat& _value) {
    m_format = _value;
}


TQString k9Menu::getMenuFileName() const {
    return m_menuFileName;
}


void k9Menu::setMenuFileName(const TQString& _value) {
    m_menuFileName = _value;
}




TQCanvas* k9Menu::getCanvas() const {
    return m_canvas;
}


void k9Menu::setCanvas(TQCanvas* _value) {
    m_canvas = _value;
}

void k9Menu::setStartScript(const TQString &_value) {
    m_startScript=_value;
}


TQString k9Menu::getStartScript() const {
    return m_startScript;
}

void k9Menu::setStartScript2(const TQString &_value) {
    m_startScript2=_value;
}


TQString k9Menu::getStartScript2() const {
    return m_startScript2;
}


TQString k9Menu::getEndScript() const {
    return m_endScript;
}


void k9Menu::setEndScript(const TQString& _value) {
    m_endScript = _value;
}