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 // Dash
  40 #include "dash/project.h"
  41
  42 namespace Dash {
  43
  44 static bool deleteProjectDir(const QString &projectDir)
  45 {
  46         if(!projectDir.startsWith(".") && QFile::exists( projectDir ) )
  47         {
  48                 QDir dir(projectDir);
  49
  50                 if( (dir.exists("audio") && dir.exists("video") && dir.exists("images")) || dir.exists("project") )
  51                 {
  52                         dDebug() << "Removing " << dir.absolutePath() << "...";
  53
  54                         dir.remove("project");
  55                         dir.remove("library");
  56                         dir.remove("palette");
  57
  58                         foreach(QString scene, dir.entryList(QStringList() << "scene*", QDir::Files ))
  59                         {
  60                                 dir.remove(scene);
  61                         }
  62
  63                         foreach( QString subdir, QStringList() << "audio" << "video" << "images" << "svg" )
  64                         {
  65                                 if( dir.exists(subdir) )
  66                                 {
  67                                         dir.cd(subdir);
  68                                         foreach(QString file, dir.entryList() )
  69                                         {
  70                                                 QString absolute = dir.absolutePath() + "/" + file;
  71
  72                                                 if( !file.startsWith(".") )
  73                                                 {
  74                                                         QFileInfo finfo(absolute);
  75
  76                                                         if( finfo.isFile() )
  77                                                         {
  78                                                                 QFile::remove(absolute);
  79                                                         }
  80                                                 }
  81                                         }
  82                                         dir.cdUp();
  83                                         dir.rmdir(subdir);
  84                                 }
  85                         }
  86
  87                         if( ! dir.rmdir(dir.absolutePath()) )
  88                         {
  89                                 dError("project") << "Cannot remove project data directory!";
  90                         }
  91                 }
  92                 return true;
  93         }
  94
  95         return false;
  96 }
  97
  98
  99 ProjectSaver::ProjectSaver()
 100 {
 101 }
 102
 103
 104 ProjectSaver::~ProjectSaver()
 105 {
 106 }
 107
 108 bool ProjectSaver::save(Dash::Project *project, const QString &path)
 109 {
 110         QDir projectDir = QDir::tempPath()+"/dash."+QString::number(DCore::Algorithm::random());
 111
 112         if ( !projectDir.exists() )
 113         {
 114                 if ( ! projectDir.mkpath(projectDir.absolutePath()) )
 115                 {
 116                         dError() << "Cannot create: " << projectDir.absolutePath();
 117                         return false;
 118                 }
 119         }
 120
 121         QString current = QDir::currentPath();
 122         QDir::setCurrent(projectDir.absolutePath());
 123
 124         QStringList filesToZip;
 125         {
 126                 // Save project
 127                 QFile prj(projectDir.absolutePath()+"/project");
 128
 129                 if ( prj.open(QIODevice::WriteOnly | QIODevice::Text))
 130                 {
 131                         QTextStream ts(&prj);
 132
 133                         QDomDocument doc;
 134                         doc.appendChild(project->toXml(doc));
 135
 136                         ts << doc.toString();
 137
 138                         prj.close();
 139
 140                         QFileInfo finfo(prj);
 141                         filesToZip << finfo.absoluteFilePath();
 142                 }
 143         }
 144
 145         // Save scenes
 146         {
 147                 int index = 0;
 148                 foreach ( YAMF::Model::Scene *scene, project->scenes().visualValues() )
 149                 {
 150                         QDomDocument doc;
 151                         doc.appendChild(scene->toXml(doc));
 152
 153                         QFile scn(projectDir.path()+"/scene"+QString::number(index));
 154
 155                         if ( scn.open(QIODevice::WriteOnly | QIODevice::Text) )
 156                         {
 157                                 QTextStream st(&scn);
 158                                 st << doc.toString();
 159                                 scn.close();
 160                                 index++;
 161
 162                                 QFileInfo finfo(scn);
 163                                 filesToZip << finfo.absoluteFilePath();
 164                         }
 165                 }
 166         }
 167
 168         // Save library
 169         {
 170                 QString libraryPath = projectDir.path()+"/library";
 171                 project->library()->save(libraryPath);
 172
 173                 filesToZip << libraryPath;
 174
 175                 QStringList dirs;
 176                 dirs << "audio" << "video" << "images" << "svg";
 177
 178                 foreach(QString dir, dirs)
 179                 {
 180                         QString path = dir;
 181                         if( QFile::exists(projectDir.absoluteFilePath(dir)) )
 182                         {
 183                                 filesToZip << path;
 184                         }
 185                 }
 186         }
 187
 188         // Save palette
 189         {
 190                 if( QFile::copy(project->projectPalette(), projectDir.path()+"/palette") )
 191                 {
 192                         filesToZip << "palette";
 193                 }
 194         }
 195
 196         DCore::Zipper zipper;
 197
 198         D_SHOW_VAR(QDir::currentPath());
 199         bool ok = zipper.write(filesToZip, path);
 200         QDir::setCurrent(current);
 201
 202         if( ! ok )
 203                 return false;
 204
 205         deleteProjectDir(projectDir.absolutePath());
 206
 207         return true;
 208 }
 209
 210 bool ProjectSaver::load(Dash::Project *project, const QString &path)
 211 {
 212         QDir projectDir = QDir::tempPath()+"/dash."+QString::number(DCore::Algorithm::random());
 213
 214         DCore::Zipper zipper;
 215
 216         if( ! zipper.read(path, projectDir.absolutePath() ) )
 217                 return false;
 218
 219         project->clear();
 220         project->setOpen(false);
 221
 222         // Load project
 223         if( projectDir.exists("project") )
 224         {
 225                 QFile prj(projectDir.absoluteFilePath("project"));
 226
 227                 if ( prj.open(QIODevice::ReadOnly | QIODevice::Text) )
 228                 {
 229                         project->fromXml(QString::fromLocal8Bit(prj.readAll()));
 230                         prj.close();
 231                 }
 232         }
 233
 234         // Cargar libreria!
 235         {
 236                 project->loadLibrary(projectDir.path()+"/library");
 237         }
 238
 239         // Load scenes
 240         {
 241                 QStringList scenes = projectDir.entryList(QStringList() << "scene*", QDir::Readable | QDir::Files);
 242
 243                 foreach(QString scenePath, scenes)
 244                 {
 245                         scenePath = projectDir.absoluteFilePath(scenePath);
 246                         YAMF::Model::Scene *scene = project->createScene();
 247
 248                         QFile f(scenePath);
 249
 250                         if ( f.open(QIODevice::ReadOnly | QIODevice::Text) )
 251                         {
 252                                 QString xml = QString::fromLocal8Bit(f.readAll());
 253                                 scene->fromXml(xml);
 254
 255                                 f.close();
 256                         }
 257                 }
 258         }
 259
 260         // Load palette
 261         {
 262                 QString path = projectDir.path()+"/palette";
 263                 if( QFile::exists( path ) )
 264                 {
 265                         project->importPalette(path);
 266                 }
 267         }
 268
 269         project->setOpen(true);
 270         deleteProjectDir(projectDir.absolutePath());
 271
 272         project->commandManager()->clear();
 273
 274         return true;
 275 }
 276
 277 }