API Documentation

5eeebd3c7472c3c2e6618ff9a94eca7af8af593e

 All Classes Namespaces Functions Variables Enumerations Properties Pages
test_filemanager.cpp
1 /*
2 
3 Pencil2D - Traditional Animation Software
4 Copyright (C) 2012-2020 Matthew Chiawen Chang
5 
6 This program is free software; you can redistribute it and/or
7 modify it under the terms of the GNU General Public License
8 as published by the Free Software Foundation; version 2 of the License.
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 */
16 #include "catch.hpp"
17 
18 #include <QTemporaryDir>
19 #include <QTemporaryFile>
20 #include <QImage>
21 #include "qminiz.h"
22 #include "fileformat.h"
23 #include "filemanager.h"
24 #include "util.h"
25 #include "object.h"
26 #include "bitmapimage.h"
27 #include "layerbitmap.h"
28 
29 
30 TEST_CASE("FileManager Initial Test")
31 {
32  SECTION("Initial error code")
33  {
34  FileManager f;
35  REQUIRE(f.error() == Status::OK);
36  }
37 }
38 
39 TEST_CASE("FileManager invalid operations")
40 {
41  SECTION("Open a non-existing file")
42  {
43  FileManager fm;
44 
45  QString strDummyPath = "hahaha_blala.pcl";
46  Object* obj = fm.load(strDummyPath);
47 
48  REQUIRE(obj == nullptr);
49  REQUIRE(fm.error().code() == Status::FILE_NOT_FOUND);
50  }
51 
52  SECTION("Bad XML")
53  {
54  QString strBadXMLPath = QDir::tempPath() + "/bad.pcl";
55 
56  // make a fake xml.
57  QFile badXMLFile(strBadXMLPath);
58  badXMLFile.open(QIODevice::WriteOnly);
59 
60  QTextStream fout(&badXMLFile);
61  fout << "%% haha, this is not a xml file.";
62  badXMLFile.close();
63 
64  FileManager fm;
65  Object* pObj = fm.load(strBadXMLPath);
66 
67  REQUIRE(pObj == nullptr);
68  REQUIRE(fm.error().code() == Status::ERROR_INVALID_XML_FILE);
69  }
70 
71  SECTION("A valid xml, but doesn't follow pencil2d's rule")
72  {
73  QString strBadXMLPath = QDir::tempPath() + "/bad.pcl";
74 
75  QFile badXMLFile(strBadXMLPath);
76  badXMLFile.open(QIODevice::WriteOnly);
77 
78  QTextStream fout(&badXMLFile);
79  fout << "<!DOCTYPE BlahBlahRoot><document></document>";
80  badXMLFile.close();
81 
82  FileManager fm;
83  Object* pObj = fm.load(strBadXMLPath);
84 
85  REQUIRE(pObj == nullptr);
86  REQUIRE(fm.error().code() == Status::ERROR_INVALID_PENCIL_FILE);
87  }
88 }
89 
90 TEST_CASE("FileManager Loading XML Tests")
91 {
92  SECTION("Minimal working xml")
93  {
94  QTemporaryFile minimalDoc;
95  if (minimalDoc.open())
96  {
97  QFile minXML(minimalDoc.fileName());
98  minXML.open(QIODevice::WriteOnly);
99 
100  QTextStream fout(&minXML);
101  fout << "<!DOCTYPE PencilDocument><document>";
102  fout << " <object></object>";
103  fout << "</document>";
104  minXML.close();
105 
106  FileManager fm;
107  Object* o = fm.load(minimalDoc.fileName());
108 
109  REQUIRE(o != nullptr);
110  REQUIRE(fm.error().ok());
111  REQUIRE(o->getLayerCount() == 1); // have at least one cam layer
112 
113  delete o;
114  }
115  }
116 
117  SECTION("Xml with one bitmap layer")
118  {
119  QTemporaryFile tmpFile;
120  if (!tmpFile.open())
121  {
122  REQUIRE(false);
123  }
124  QFile theXML(tmpFile.fileName());
125  theXML.open(QIODevice::WriteOnly);
126 
127  QTextStream fout(&theXML);
128  fout << "<!DOCTYPE PencilDocument><document>";
129  fout << " <object>";
130  fout << " <layer name='MyLayer' id='5' visibility='1' type='1'></layer>";
131  fout << " </object>";
132  fout << "</document>";
133  theXML.close();
134 
135  FileManager fm;
136  Object* obj = fm.load(theXML.fileName());
137  REQUIRE(obj->getLayerCount() == 2); // one bitmap layer and one default cam layer
138  REQUIRE(obj->getLayer(0)->name() == "MyLayer");
139  REQUIRE(obj->getLayer(0)->id() == 5);
140  REQUIRE(obj->getLayer(0)->visible() == true);
141  REQUIRE(obj->getLayer(0)->type() == Layer::BITMAP);
142 
143  delete obj;
144  }
145 
146  SECTION("xml with one bitmap layer and one key")
147  {
148  QTemporaryFile tmpFile;
149  if (!tmpFile.open())
150  {
151  REQUIRE(false);
152  }
153  QFile theXML(tmpFile.fileName());
154  theXML.open(QIODevice::WriteOnly);
155 
156  QTextStream fout(&theXML);
157  fout << "<!DOCTYPE PencilDocument><document>";
158  fout << " <object>";
159  fout << " <layer name='GoodLayer' id='5' visibility='0' type='1' >";
160  fout << " <image frame='1' topLeftY='0' src='003.001.png' topLeftX='0' />";
161  fout << " </layer>";
162  fout << " </object>";
163  fout << "</document>";
164  theXML.close();
165 
166  FileManager fm;
167  Object* obj = fm.load(theXML.fileName());
168 
169  Layer* layer = obj->getLayer(0);
170  REQUIRE(layer->name() == "GoodLayer");
171  REQUIRE(layer->id() == 5);
172  REQUIRE(layer->visible() == false);
173  REQUIRE(layer->type() == Layer::BITMAP);
174 
175  REQUIRE(layer->getKeyFrameAt(1) != nullptr);
176  REQUIRE(layer->keyFrameCount() == 1);
177 
178  delete obj;
179  }
180 
181  SECTION("xml with 1 bitmap layer and 2 keys")
182  {
183  QTemporaryFile tmpFile;
184  if (!tmpFile.open())
185  {
186  REQUIRE(false);
187  }
188  QFile theXML(tmpFile.fileName());
189  theXML.open(QIODevice::WriteOnly);
190 
191  QTextStream fout(&theXML);
192  fout << "<!DOCTYPE PencilDocument><document>";
193  fout << " <object>";
194  fout << " <layer name='MyLayer' id='5' visibility='1' type='1' >";
195  fout << " <image frame='1' topLeftY='0' src='003.001.png' topLeftX='0' />";
196  fout << " <image frame='2' topLeftY='0' src='003.001.png' topLeftX='0' />";
197  fout << " </layer>";
198  fout << " </object>";
199  fout << "</document>";
200  theXML.close();
201 
202  FileManager fm;
203  Object* obj = fm.load(theXML.fileName());
204 
205  Layer* layer = obj->getLayer(0);
206  REQUIRE(layer->name() == "MyLayer");
207  REQUIRE(layer->id() == 5);
208  REQUIRE(layer->visible() == true);
209  REQUIRE(layer->type() == Layer::BITMAP);
210 
211  REQUIRE(layer->getKeyFrameAt(1) != nullptr);
212  delete obj;
213  }
214 }
215 
216 // Turn a Qt resource file into an actual file on disk
217 QString QtResourceToFile(QString rscPath, QString filename, QTemporaryDir& tempDir)
218 {
219  QFile fin(rscPath);
220  if (!fin.open(QFile::ReadOnly))
221  {
222  qWarning() << __FUNCTION__ << "Cannot open" << rscPath;
223  return "";
224  }
225  QByteArray content = fin.readAll();
226  fin.close();
227 
228  QString filePathOnDisk = tempDir.filePath(filename);
229  QFile fout(filePathOnDisk);
230  if (!fout.open(QFile::WriteOnly))
231  {
232  qWarning() << __FUNCTION__ << "Cannot write to" << filePathOnDisk;
233  }
234  fout.write(content);
235  fout.close();
236  return filePathOnDisk;
237 }
238 
239 TEST_CASE("FileManager Load PCLX")
240 {
241  SECTION("Empty PCLX")
242  {
243  QTemporaryDir tempDir;
244 
245  FileManager fm;
246  Object* o = fm.load(QtResourceToFile(":/empty.pclx", "empty.pclx", tempDir));
247  REQUIRE(o != nullptr);
248  if (o)
249  {
250  // file has 2 bitmap layers, 1 vector layers and 1 cam layers
251  REQUIRE(o->getLayerCount() == 4);
252  }
253  delete o;
254  }
255 
256  SECTION("Chinese Filename")
257  {
258  QTemporaryDir tempDir;
259 
260  FileManager fm;
261  Object* o = fm.load(QtResourceToFile(":/cjk-test.pclx", "許功蓋.pclx", tempDir));
262  REQUIRE(o != nullptr);
263  if (o)
264  {
265  // file has 2 bitmap layers, 1 vector layers and 1 cam layers
266  REQUIRE(o->getLayerCount() == 4);
267  }
268  delete o;
269  }
270 
271  SECTION("Japanese Filename")
272  {
273  QTemporaryDir tempDir;
274 
275  FileManager fm;
276  Object* o = fm.load(QtResourceToFile(":/cjk-test.pclx", "構わない.pclx", tempDir));
277  REQUIRE(o != nullptr);
278  if (o)
279  {
280  // file has 2 bitmap layers, 1 vector layers and 1 cam layers
281  REQUIRE(o->getLayerCount() == 4);
282  }
283  delete o;
284  }
285 
286  SECTION("Korean Filename")
287  {
288  QTemporaryDir tempDir;
289 
290  FileManager fm;
291  Object* o = fm.load(QtResourceToFile(":/cjk-test.pclx", "대박이야.pclx", tempDir));
292  REQUIRE(o != nullptr);
293  if (o)
294  {
295  // file has 2 bitmap layers, 1 vector layers and 1 cam layers
296  REQUIRE(o->getLayerCount() == 4);
297  }
298  delete o;
299  }
300 }
301 
302 TEST_CASE("FileManager File-saving")
303 {
304  // https://github.com/pencil2d/pencil/issues/939
305  SECTION("#939 Clear action not properly saved")
306  {
307  FileManager fm;
308 
309  // 1. create a animation with one red frame & save it
310  Object* o1 = new Object;
311  o1->init();
312  o1->createDefaultLayers();
313 
314  LayerBitmap* layer = dynamic_cast<LayerBitmap*>(o1->getLayer(2));
315  REQUIRE(layer->addNewKeyFrameAt(2));
316 
317  BitmapImage* b1 = layer->getBitmapImageAtFrame(2);
318  b1->drawRect(QRectF(0, 0, 10, 10), QPen(QColor(255, 0, 0)), QBrush(Qt::red), QPainter::CompositionMode_SourceOver, false);
319 
320  QTemporaryDir testDir("PENCIL_TEST_XXXXXXXX");
321  QString animationPath = testDir.path() + "/abc.pclx";
322  fm.save(o1, animationPath);
323  delete o1;
324 
325  // 2. load the animation, and then clear the red frame, save it.
326  Object* o2 = fm.load(animationPath);
327  layer = dynamic_cast<LayerBitmap*>(o2->getLayer(2));
328 
329  BitmapImage* b2 = layer->getBitmapImageAtFrame(2);
330  b2->clear();
331 
332  fm.save(o2, animationPath);
333  delete o2;
334 
335  // 3. load the animation again, check whether it's an empty frame
336  Object* o3 = fm.load(animationPath);
337  layer = dynamic_cast<LayerBitmap*>(o3->getLayer(2));
338 
339  BitmapImage* b3 = layer->getBitmapImageAtFrame(2);
340  REQUIRE(b3->bounds().isEmpty());
341 
342  delete o3;
343  }
344 
345  //https://github.com/pencil2d/pencil/issues/966
346  SECTION("#966 Moving more than 200 frames corrupts frames upon save")
347  {
348  FileManager fm;
349 
350  // 1. Create a animation with 150 frames & save it
351  Object* o1 = new Object;
352  o1->init();
353  o1->createDefaultLayers();
354 
355  LayerBitmap* layer = dynamic_cast<LayerBitmap*>(o1->getLayer(2));
356  for (int i = 100; i < 150; ++i)
357  {
358  layer->addNewKeyFrameAt(i);
359  auto bitmap = layer->getBitmapImageAtFrame(i);
360  bitmap->drawRect(QRectF(0, 0, 10, 10), QPen(QColor(255, 0, 0)), QBrush(Qt::red), QPainter::CompositionMode_SourceOver, false);
361  }
362 
363  QTemporaryDir testDir("PENCIL_TEST_XXXXXXXX");
364  QString animationPath = testDir.path() + "/abc.pclx";
365  fm.save(o1, animationPath);
366  delete o1;
367 
368  // 2. Load the animation back and then make some frames unloaded by active frame pool
369  Object* o2 = fm.load(animationPath);
370  o2->setActiveFramePoolSize(20);
371 
372  layer = dynamic_cast<LayerBitmap*>(o2->getLayer(2));
373  for (int i = 1; i < 150; ++i)
374  o2->updateActiveFrames(i);
375 
376  // 3. Move those unloaded frames around
377  for (int i = 100; i < 150; ++i)
378  layer->setFrameSelected(i, true);
379 
380  layer->moveSelectedFrames(-55);
381  fm.save(o2, animationPath);
382  delete o2;
383 
384  // 4. Check no lost frames
385  Object* o3 = fm.load(animationPath);
386  layer = dynamic_cast<LayerBitmap*>(o3->getLayer(2));
387  for (int i = 2; i < 150; ++i)
388  {
389  auto bitmap = layer->getBitmapImageAtFrame(i);
390  if (bitmap)
391  {
392  REQUIRE(bitmap->image()->width() > 1);
393  REQUIRE(bitmap->image()->height() > 1);
394  }
395  }
396  delete o3;
397  }
398 }
399 
400 TEST_CASE("Empty Sound Frames")
401 {
402  SECTION("Invalid src value")
403  {
404 
405  QTemporaryFile soundFrameDoc;
406  if (soundFrameDoc.open())
407  {
408  QFile newXML(soundFrameDoc.fileName());
409  newXML.open(QIODevice::WriteOnly);
410 
411  QTextStream fout(&newXML);
412  fout << "<!DOCTYPE PencilDocument><document>";
413  fout << " <object>";
414  fout << " <layer type='4' id='5' name='GoodLayer' visibility='1'>";
415  fout << " <sound frame='1' name='' src=''/>";
416  fout << " </layer>";
417  fout << " </object>";
418  fout << "</document>";
419  newXML.close();
420 
421 
422  FileManager fm;
423  Object* newObj = fm.load(soundFrameDoc.fileName());
424 
425  REQUIRE(newObj != nullptr);
426  REQUIRE(fm.error().ok());
427  REQUIRE(newObj->getLayerCount() == 2);
428  REQUIRE(newObj->getLayer(0)->type() == 4);
429  REQUIRE(newObj->getLayer(0)->id() == 5);
430  REQUIRE(newObj->getLayer(0)->name() == "GoodLayer");
431  REQUIRE(newObj->getLayer(0)->getVisibility() == true);
432  REQUIRE(newObj->getLayer(0)->getKeyFrameAt(1) == nullptr);
433 
434  delete newObj;
435  }
436  }
437 }
QDir::tempPath
QString tempPath()
QTemporaryDir::filePath
QString filePath(const QString &fileName) const const
Layer
Definition: layer.h:39
QFile::open
virtual bool open(QIODevice::OpenMode mode) override
QPainter::CompositionMode_SourceOver
CompositionMode_SourceOver
QRect::isEmpty
bool isEmpty() const const
QTemporaryFile::fileName
virtual QString fileName() const const override
Object
Definition: object.h:54
Generated on Sun Dec 20 2020 16:19:59 for Pencil2D by   doxygen 1.8.6