// vim: set tabstop=4 shiftwidth=4 noexpandtab /* Gwenview - A simple image viewer for KDE Copyright 2000-2004 Aurélien Gâteau This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. */ #include // TQt #include #include #include #include // KDE #include #include #include #include #include // Local #include "imageutils/imageutils.h" #include "imageutils/jpegcontent.h" using namespace std; const char* ORIENT6_FILE="orient6.jpg"; const int ORIENT6_WIDTH=128; // This size is the size *after* orientation const int ORIENT6_HEIGHT=256; // has been applied const char* CUT_FILE="cut.jpg"; const TQString ORIENT6_COMMENT="a comment"; const TQString ORIENT1_VFLIP_FILE="test_orient1_vflip.jpg"; const char* THUMBNAIL_FILE="test_thumbnail.jpg"; const char* TMP_FILE="tmp.jpg"; class TestEnvironment { public: TestEnvironment() { bool result; TQFile in(ORIENT6_FILE); result=in.open(IO_ReadOnly); Q_ASSERT(result); TQFileInfo info(in); int size=info.size()/2; char* data=new char[size]; int readSize=in.readBlock(data, size); Q_ASSERT(size==readSize); TQFile out(CUT_FILE); result=out.open(IO_WriteOnly); Q_ASSERT(result); int wroteSize=out.writeBlock(data, size); Q_ASSERT(size==wroteSize); delete []data; } ~TestEnvironment() { TQDir::current().remove(CUT_FILE); } }; typedef TQMap MetaInfoMap; MetaInfoMap getMetaInfo(const TQString& path) { KFileMetaInfo fmi(path); TQStringList list=fmi.supportedKeys(); TQStringList::ConstIterator it=list.begin(); MetaInfoMap map; for ( ; it!=list.end(); ++it) { KFileMetaInfoItem item=fmi.item(*it); map[*it]=item.string(); } return map; } void compareMetaInfo(const TQString& path1, const TQString& path2, const TQStringList& ignoredKeys) { MetaInfoMap mim1=getMetaInfo(path1); MetaInfoMap mim2=getMetaInfo(path2); Q_ASSERT(mim1.keys()==mim2.keys()); TQValueList keys=mim1.keys(); TQValueList::ConstIterator it=keys.begin(); for ( ; it!=keys.end(); ++it) { TQString key=*it; if (ignoredKeys.tqcontains(key)) continue; if (mim1[key]!=mim2[key]) { kdError() << "Meta info differs. Key:" << key << ", V1:" << mim1[key] << ", V2:" << mim2[key] << endl; } } } void testResetOrientation() { ImageUtils::JPEGContent content; bool result; // Test resetOrientation without transform result=content.load(ORIENT6_FILE); Q_ASSERT(result); content.resetOrientation(); result=content.save(TMP_FILE); Q_ASSERT(result); result=content.load(TMP_FILE); Q_ASSERT(result); Q_ASSERT(content.orientation() == ImageUtils::NORMAL); // Test resetOrientation with transform result=content.load(ORIENT6_FILE); Q_ASSERT(result); content.resetOrientation(); content.transform(ImageUtils::ROT_90); result=content.save(TMP_FILE); Q_ASSERT(result); result=content.load(TMP_FILE); Q_ASSERT(result); Q_ASSERT(content.orientation() == ImageUtils::NORMAL); } /** * This function tests JPEGContent::transform() by applying a ROT_90 * transformation, saving, reloading and applying a ROT_270 to undo the ROT_90. * Saving and reloading are necessary because lossless transformation only * happens in JPEGContent::save() */ void testTransform() { bool result; TQImage finalImage, expectedImage; ImageUtils::JPEGContent content; result = content.load(ORIENT6_FILE); Q_ASSERT(result); content.transform(ImageUtils::ROT_90); result = content.save(TMP_FILE); Q_ASSERT(result); result = content.load(TMP_FILE); Q_ASSERT(result); content.transform(ImageUtils::ROT_270); result = content.save(TMP_FILE); Q_ASSERT(result); result = finalImage.load(TMP_FILE); Q_ASSERT(result); result = expectedImage.load(ORIENT6_FILE); Q_ASSERT(result); Q_ASSERT(finalImage == expectedImage); } void testSetComment() { TQString comment = "test comment"; ImageUtils::JPEGContent content; bool result; result = content.load(ORIENT6_FILE); Q_ASSERT(result); content.setComment(comment); Q_ASSERT(content.comment() == comment); result = content.save(TMP_FILE); Q_ASSERT(result); result = content.load(TMP_FILE); Q_ASSERT(result); Q_ASSERT(content.comment() == comment); } int main(int argc, char* argv[]) { TestEnvironment testEnv; bool result; KAboutData aboutData("testjpegcontent", "testjpegcontent", "0"); KCmdLineArgs::init( argc, argv, &aboutData ); KApplication kapplication; // Reading info ImageUtils::JPEGContent content; result=content.load(ORIENT6_FILE); Q_ASSERT(result); Q_ASSERT(content.orientation() == 6); Q_ASSERT(content.comment() == ORIENT6_COMMENT); Q_ASSERT(content.size() == TQSize(ORIENT6_WIDTH, ORIENT6_HEIGHT)); // thumbnail() TQImage thumbnail=content.thumbnail(); result=thumbnail.save(THUMBNAIL_FILE, "JPEG"); Q_ASSERT(result); testResetOrientation(); testTransform(); testSetComment(); // Test that rotating a file a lot of times does not cause findJxform() to fail result = content.load(ORIENT6_FILE); Q_ASSERT(result); // 12*4 + 1 is the same as 1, since rotating four times brings you back for(int loop=0; loop< 12*4 + 1; ++loop) { content.transform(ImageUtils::ROT_90); } result = content.save(TMP_FILE); Q_ASSERT(result); result = content.load(TMP_FILE); Q_ASSERT(result); Q_ASSERT(content.size() == TQSize(ORIENT6_HEIGHT, ORIENT6_WIDTH)); // Check the other meta info are still here TQStringList ignoredKeys; ignoredKeys << "Qt::Orientation" << "Comment"; compareMetaInfo(ORIENT6_FILE, ORIENT1_VFLIP_FILE, ignoredKeys); // Test that loading and manipulating a truncated file does not crash result=content.load(CUT_FILE); Q_ASSERT(result); Q_ASSERT(content.orientation() == 6); Q_ASSERT(content.comment() == ORIENT6_COMMENT); content.transform(ImageUtils::VFLIP); kdWarning() << "# Next function should output errors about incomplete image" << endl; content.save(TMP_FILE); kdWarning() << "#" << endl; }