src/dash/projectsaver.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by David Cuadrado                                  *
   3  *   krawek@gmail.com                                                      *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21
  22 #include "projectsaver.h"
  23
  24 // Qt
  25 #include <QDir>
  26
  27 // DLib
  28 #include <dcore/zipper.h>
  29 #include <dcore/algorithm.h>
  30 #include <dcore/debug.h>
  31
  32 // YAMF
  33 #include <yamf/model/project.h>
  34 #include <yamf/model/scene.h>
  35 #include <yamf/model/frame.h>
  36 #include <yamf/model/library.h>
  37 #include <yamf/model/command/manager.h>
  38
  39 namespace Dash {
  40
  41 static bool deleteProjectDir(const QString &projectDir)
  42 {
  43         if(!projectDir.startsWith(".") && QFile::exists( projectDir ) )
  44         {
  45                 QDir dir(projectDir);
  46
  47                 if( (dir.exists("audio") && dir.exists("video") && dir.exists("images")) || dir.exists("project") )
  48                 {
  49                         dDebug() << "Removing " << dir.absolutePath() << "...";
  50
  51                         dir.remove("project");
  52                         dir.remove("library");
  53
  54                         foreach(QString scene, dir.entryList(QStringList() << "scene*", QDir::Files ))
  55                         {
  56                                 dir.remove(scene);
  57                         }
  58
  59                         foreach( QString subdir, QStringList() << "audio" << "video" << "images" << "svg" )
  60                         {
  61                                 if( dir.exists(subdir) )
  62                                 {
  63                                         dir.cd(subdir);
  64                                         foreach(QString file, dir.entryList() )
  65                                         {
  66                                                 QString absolute = dir.absolutePath() + "/" + file;
  67
  68                                                 if( !file.startsWith(".") )
  69                                                 {
  70                                                         QFileInfo finfo(absolute);
  71
  72                                                         if( finfo.isFile() )
  73                                                         {
  74                                                                 QFile::remove(absolute);
  75                                                         }
  76                                                 }
  77                                         }
  78                                         dir.cdUp();
  79                                         dir.rmdir(subdir);
  80                                 }
  81                         }
  82
  83                         if( ! dir.rmdir(dir.absolutePath()) )
  84                         {
  85                                 dError("project") << "Cannot remove project data directory!";
  86                         }
  87                 }
  88                 return true;
  89         }
  90
  91         return false;
  92 }
  93
  94
  95 ProjectSaver::ProjectSaver()
  96 {
  97 }
  98
  99
 100 ProjectSaver::~ProjectSaver()
 101 {
 102 }
 103
 104 bool ProjectSaver::save(YAMF::Model::Project *project, const QString &path)
 105 {
 106         QDir projectDir = QDir::tempPath()+"/dash."+QString::number(DCore::Algorithm::random());
 107
 108         if ( !projectDir.exists() )
 109         {
 110                 if ( ! projectDir.mkpath(projectDir.absolutePath()) )
 111                 {
 112                         dError() << "Cannot create: " << projectDir.absolutePath();
 113                         return false;
 114                 }
 115         }
 116
 117         QString current = QDir::currentPath();
 118         QDir::setCurrent(projectDir.absolutePath());
 119
 120         QStringList filesToZip;
 121         {
 122                 // Save project
 123                 QFile prj(projectDir.absolutePath()+"/project");
 124
 125                 if ( prj.open(QIODevice::WriteOnly | QIODevice::Text))
 126                 {
 127                         QTextStream ts(&prj);
 128
 129                         QDomDocument doc;
 130                         doc.appendChild(project->toXml(doc));
 131
 132                         ts << doc.toString();
 133
 134                         prj.close();
 135
 136                         QFileInfo finfo(prj);
 137                         filesToZip << finfo.absoluteFilePath();
 138                 }
 139         }
 140
 141         // Save scenes
 142         {
 143                 int index = 0;
 144                 foreach ( YAMF::Model::Scene *scene, project->scenes().visualValues() )
 145                 {
 146                         QDomDocument doc;
 147                         doc.appendChild(scene->toXml(doc));
 148
 149                         QFile scn(projectDir.path()+"/scene"+QString::number(index));
 150
 151                         if ( scn.open(QIODevice::WriteOnly | QIODevice::Text) )
 152                         {
 153                                 QTextStream st(&scn);
 154                                 st << doc.toString();
 155                                 scn.close();
 156                                 index++;
 157
 158                                 QFileInfo finfo(scn);
 159                                 filesToZip << finfo.absoluteFilePath();
 160                         }
 161                 }
 162         }
 163
 164         // Save library
 165         {
 166                 QString libraryPath = projectDir.path()+"/library";
 167                 project->library()->save(libraryPath);
 168
 169                 filesToZip << libraryPath;
 170
 171                 QStringList dirs;
 172                 dirs << "audio" << "video" << "images" << "svg";
 173
 174                 foreach(QString dir, dirs)
 175                 {
 176                         QString path = dir;
 177                         if( QFile::exists(projectDir.absoluteFilePath(dir)) )
 178                         {
 179                                 filesToZip << path;
 180                         }
 181                 }
 182         }
 183
 184         DCore::Zipper zipper;
 185
 186         D_SHOW_VAR(QDir::currentPath());
 187         bool ok = zipper.write(filesToZip, path);
 188         QDir::setCurrent(current);
 189
 190         if( ! ok )
 191                 return false;
 192
 193         deleteProjectDir(projectDir.absolutePath());
 194
 195         return true;
 196 }
 197
 198 bool ProjectSaver::load(YAMF::Model::Project *project, const QString &path)
 199 {
 200         QDir projectDir = QDir::tempPath()+"/dash."+QString::number(DCore::Algorithm::random());
 201
 202         DCore::Zipper zipper;
 203
 204         if( ! zipper.read(path, projectDir.absolutePath() ) )
 205                 return false;
 206
 207         project->clear();
 208         project->setOpen(false);
 209
 210         // Load project
 211         if( projectDir.exists("project") )
 212         {
 213                 QFile prj(projectDir.absoluteFilePath("project"));
 214
 215                 if ( prj.open(QIODevice::ReadOnly | QIODevice::Text) )
 216                 {
 217                         project->fromXml(QString::fromLocal8Bit(prj.readAll()));
 218                         prj.close();
 219                 }
 220         }
 221
 222         // Cargar libreria!
 223         {
 224                 project->loadLibrary(projectDir.path()+"/library");
 225         }
 226
 227         // Load scenes
 228         {
 229                 QStringList scenes = projectDir.entryList(QStringList() << "scene*", QDir::Readable | QDir::Files);
 230
 231                 foreach(QString scenePath, scenes)
 232                 {
 233                         scenePath = projectDir.absoluteFilePath(scenePath);
 234                         YAMF::Model::Scene *scene = project->createScene();
 235
 236                         QFile f(scenePath);
 237
 238                         if ( f.open(QIODevice::ReadOnly | QIODevice::Text) )
 239                         {
 240                                 QString xml = QString::fromLocal8Bit(f.readAll());
 241                                 scene->fromXml(xml);
 242
 243                                 f.close();
 244                         }
 245                 }
 246         }
 247
 248         project->setOpen(true);
 249         deleteProjectDir(projectDir.absolutePath());
 250
 251         project->commandManager()->clear();
 252
 253         return true;
 254 }
 255
 256 }