VidePub Code

/*
   Copyright (C) 2011 Arnaud Champenois arthelion92@gmail.com

    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 3 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, see <http://www.gnu.org/licenses/>.
*/
#include "filecopier.h"
#include "mainwindow.h"
#include <QFile>

FileCopier::FileCopier()
{
}

FileCopier::~FileCopier()
{
    Close();
}

bool FileCopier::Open(const QString &aFileIn, const QString &aFileOut)
{
    mFileIn.setFileName(aFileIn);
    if(!mFileIn.open(QIODevice::ReadOnly))
        return false;

    mFileOut.setFileName(aFileOut);
    if(!mFileOut.open(QIODevice::WriteOnly))
        return false;

    print_log(QString("Open :")+aFileIn+QString(" to ")+aFileOut);

    return true;
}

bool FileCopier::Write(const VideoSegment &aSegment, QProgressDialog *aDial)
{
   size_t aBuffSize = 3000*188; // Packets MPEG
   int64_t aRestSize = aSegment.mEnd.mFilePos-aSegment.mBegin.mFilePos;
   char *aBuff = new char[aBuffSize];
   int64_t aPosition = aSegment.mBegin.mFilePos;
   int64_t aMSDelta = (aSegment.mEnd.mMsPos*aBuffSize)/aSegment.mEnd.mFilePos;
   int64_t aMSPos = aSegment.mBegin.mMsPos;

   mFileIn.seek(aPosition);
   bool aStatus = true;

   while(aRestSize)
   {
       if(aDial)
       {
           aDial->setValue(aMSPos-aSegment.mBegin.mMsPos);
           MainWindow::FlushAll();
           if(aDial->wasCanceled())
           {
               aStatus = false;
               break;
           }
       }
		
	   aMSPos += aMSDelta;
	   
       if(aBuffSize > aRestSize)
           aBuffSize = aRestSize;
       size_t aReadSize = mFileIn.read(aBuff,aBuffSize);
       if(aReadSize<aBuffSize)
              {
                  print_log(QString("Erreur entrée : %1/%2").arg(aReadSize).arg(aBuffSize));
                  aStatus = false;
                  break;
              }

#ifdef DEBUG
       if(aBuff[0] != 0x47)
           print_log("MPEG start incorrect");
       if(aBuff[188] != 0x47)
           print_log("MPEG end incorrect");
#endif

       size_t aWrittenSize = mFileOut.write(aBuff,aBuffSize);
       if(aWrittenSize<aBuffSize)
       {
           print_log(QString("Erreur sortie : %1/%2").arg(aWrittenSize).arg(aBuffSize));
           aStatus = false;
           break;
       }

       aPosition += aWrittenSize;
       aRestSize -= aWrittenSize;
   }

  delete aBuff;
   return aStatus;
}

void FileCopier::Close()
{
    if(mFileIn.isOpen()) mFileIn.close();
    if(mFileOut.isOpen()) mFileOut.close();
}