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"

FileCopier::FileCopier()
{
    mFileIn = NULL;
    mFileOut = NULL;
}

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

bool FileCopier::Open(const QString &aFileIn, const QString &aFileOut)
{
    mFileIn = fopen(aFileIn.toAscii().data(),"rb");
    if(!mFileIn) return false;

    mFileOut = fopen(aFileOut.toAscii().data(),"wb");
    if(!mFileOut) { fclose(mFileIn); mFileIn = NULL; 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;

   fseeko64(mFileIn,aPosition,SEEK_SET);
   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 = fread(aBuff,1,aBuffSize,mFileIn);
       if(aReadSize<aBuffSize)
              {
                  print_log(QString("Erreur entrée : %1/%2").arg(aReadSize).arg(aBuffSize));
                  aStatus = false;
                  break;
              }

       size_t aWrittenSize = fwrite(aBuff,1,aBuffSize,mFileOut);
       fflush(mFileOut);
       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) fclose(mFileIn);
    if(mFileOut) fclose(mFileOut);

    mFileIn = NULL;
    mFileOut = NULL;
}