diff options
Diffstat (limited to 'tests/writeImageToFile.cpp')
| -rw-r--r-- | tests/writeImageToFile.cpp | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/tests/writeImageToFile.cpp b/tests/writeImageToFile.cpp new file mode 100644 index 0000000..4614bef --- /dev/null +++ b/tests/writeImageToFile.cpp @@ -0,0 +1,137 @@ +// Copyright (C) 2009-2012 Colin MacDonald
+// No rights reserved: this software is in the public domain.
+
+#if defined(_MSC_VER)
+#define _CRT_SECURE_NO_WARNINGS 1
+#endif // _MSC_VER
+
+#include "testUtils.h"
+
+using namespace irr;
+using namespace core;
+using namespace scene;
+using namespace video;
+using namespace io;
+using namespace gui;
+
+//! Tests IVideoDriver::writeImageToFile() using IWriteFile
+bool writeImageToFile(void)
+{
+ IrrlichtDevice *device = createDevice( EDT_BURNINGSVIDEO, dimension2d<u32>(160, 120), 32);
+ if (!device)
+ return true; // Treat a failure to create a driver as benign; this saves a lot of #ifdefs
+
+ IVideoDriver* driver = device->getVideoDriver();
+ ISceneManager * smgr = device->getSceneManager();
+
+ // Draw a cube background so that we can check that the pixels' alpha is working.
+ ISceneNode * cube = smgr->addCubeSceneNode(50.f, 0, -1, vector3df(0, 0, 60));
+ cube->setMaterialTexture(0, driver->getTexture("../media/wall.bmp"));
+ cube->setMaterialFlag(video::EMF_LIGHTING, false);
+ (void)smgr->addCameraSceneNode();
+
+ driver->beginScene(video::ECBF_COLOR | video::ECBF_DEPTH, SColor(255,100,101,140));
+ smgr->drawAll();
+
+ // Test for benign handling of offscreen pixel values as well as onscreen ones.
+ for(s32 x = -10; x < 170; ++x)
+ {
+ s32 y = 120 * x / 160;
+ driver->drawPixel((u32)x, (u32)y, SColor(255, 255 * x / 640, 255 * (640 - x) / 640, 0));
+ y = 120 - y;
+ driver->drawPixel((u32)x, (u32)y, SColor(255 * x / 640, 0, 255, 255));
+ }
+
+ driver->endScene();
+
+ bool result = false;
+ IWriteFile * writtenFile = 0;
+ IWriteFile * memoryFile = 0;
+ const char * writtenFilename = 0;
+ const u32 BUFFER_SIZE = 160 * 120 * 4;
+ c8 * buffer = 0;
+ const char * referenceFilename = 0;
+ video::ECOLOR_FORMAT format;
+
+ irr::video::IImage * screenshot = driver->createScreenShot(video::ECF_R8G8B8);
+ if (!screenshot)
+ {
+ logTestString("Failed to take screenshot\n");
+ assert_log(false);
+ goto cleanup;
+ }
+
+ format = screenshot->getColorFormat();
+ if (format != video::ECF_R8G8B8)
+ {
+ irr::video::IImage * fixedScreenshot = driver->createImage(video::ECF_R8G8B8, screenshot->getDimension());
+ screenshot->copyTo(fixedScreenshot);
+ screenshot->drop();
+ screenshot = 0;
+
+ if (!fixedScreenshot)
+ {
+ logTestString("Failed to convert screenshot to ECF_A8R8G8B8\n");
+ assert_log(false);
+ goto cleanup;
+ }
+
+ screenshot = fixedScreenshot;
+ }
+
+ buffer = new c8[BUFFER_SIZE];
+ writtenFilename = "results/Burning's Video-writeImageToFile.png";
+ memoryFile = device->getFileSystem()->createMemoryWriteFile(buffer, BUFFER_SIZE, writtenFilename, false);
+ if (!driver->writeImageToFile(screenshot, memoryFile))
+ {
+ logTestString("Failed to write png to memory file\n");
+ assert_log(false);
+ goto cleanup;
+ }
+
+ writtenFile = device->getFileSystem()->createAndWriteFile(memoryFile->getFileName());
+ if (!writtenFile)
+ {
+ logTestString("Can't open %s for writing.\n", writtenFilename);
+ assert_log(false);
+ goto cleanup;
+ }
+
+ if ((size_t)memoryFile->getPos() != writtenFile->write(buffer, (size_t)memoryFile->getPos()))
+ {
+ logTestString("Error while writing to %s.\n", writtenFilename);
+ assert_log(false);
+ goto cleanup;
+ }
+
+ writtenFile->drop();
+ writtenFile = 0;
+
+ referenceFilename = "media/Burning's Video-drawPixel.png";
+
+ if ( fuzzyCompareImages(driver,writtenFilename, referenceFilename) < 99.9)
+ {
+ logTestString("File written from memory is not the same as the reference file. %s:%d\n" , __FILE__, __LINE__);
+// assert_log(false);
+ goto cleanup;
+ }
+
+ result = true;
+
+cleanup:
+ if ( screenshot )
+ screenshot->drop();
+
+ if(writtenFile)
+ writtenFile->drop();
+
+ if(memoryFile)
+ memoryFile->drop();
+
+ delete [] buffer;
+
+ device->drop();
+
+ return result;
+}
+
|