src/XGUI/XGUI_CompressFiles.cpp

   1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or
  18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
  19 //
  20
  21 #include "XGUI_CompressFiles.h"
  22
  23 #include <QFile>
  24 #include <QFileInfo>
  25 #include <QDir>
  26 #include <QDataStream>
  27
  28
  29
  30
  31 bool XGUI_CompressFiles::compress(const QString& theFile, std::list<std::string> theFiles)
  32 {
  33   QFile aOutFile(theFile);
  34   if (aOutFile.open(QIODevice::WriteOnly)) {
  35     QDataStream aOutStream(&aOutFile);
  36
  37     std::list<std::string>::const_iterator aIt;
  38     for (aIt = theFiles.cbegin(); aIt != theFiles.cend(); ++aIt) {
  39       QString aPathName((*aIt).c_str());
  40       QFile aSrcFile(aPathName);
  41       QFileInfo aInfo(aSrcFile);
  42       quint64 aSize = aInfo.size();
  43       if (aSrcFile.open(QIODevice::ReadOnly)) {
  44         QString aName =
  45           aPathName.right(aPathName.length() - aPathName.lastIndexOf(QDir::separator()) - 1);
  46
  47         aOutStream << aName << aSize << qCompress(aSrcFile.readAll());
  48         aSrcFile.close();
  49       }
  50       else {
  51         aOutFile.close();
  52         return false;
  53       }
  54     }
  55     aOutFile.close();
  56   }
  57   else
  58     return false;
  59
  60   return true;
  61 }
  62
  63
  64 bool XGUI_CompressFiles::uncompress(const QString& theFile, const QString& theDir)
  65 {
  66   QFile aInFile(theFile);
  67   QDir aDir(theDir);
  68   if (!aDir.exists())
  69     aDir.mkpath(theDir);
  70   if (!aDir.isEmpty()) {
  71     if (!aDir.removeRecursively())
  72       return false;
  73     aDir.mkdir(theDir);
  74   }
  75
  76   if (aInFile.open(QIODevice::ReadOnly)) {
  77     QDataStream aInStream(&aInFile);
  78     do {
  79       QString aFileName;
  80       quint64 aSize;
  81       aInStream >> aFileName;
  82       aInStream >> aSize;
  83
  84       QFile aOutFile(theDir + QDir::separator() + aFileName);
  85       if (aOutFile.open(QIODevice::WriteOnly)) {
  86         QByteArray aContent;
  87         aContent.resize(aSize);
  88         aInStream >> aContent;
  89         aOutFile.write(qUncompress(aContent));
  90         aOutFile.close();
  91       }
  92     } while (!aInStream.atEnd());
  93     aInFile.close();
  94   }
  95   return true;
  96 }