src/HYDRO_tests/test_HYDROData_Image.cxx

   1 // Copyright (C) 2014-2015  EDF-R&D
   2 // This library is free software; you can redistribute it and/or
   3 // modify it under the terms of the GNU Lesser General Public
   4 // License as published by the Free Software Foundation; either
   5 // version 2.1 of the License, or (at your option) any later version.
   6 //
   7 // This library is distributed in the hope that it will be useful,
   8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
   9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  10 // Lesser General Public License for more details.
  11 //
  12 // You should have received a copy of the GNU Lesser General Public
  13 // License along with this library; if not, write to the Free Software
  14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  15 //
  16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  17 //
  18
  19 #include<test_HYDROData_Image.h>
  20
  21 #include <HYDROData_Document.h>
  22 #include <HYDROData_Image.h>
  23 #include <TestViewer.h>
  24 #include <QPainter>
  25 #include <QImage>
  26
  27 extern QString REF_DATA_PATH;
  28
  29 static QImage TestImage() {
  30   QImage aPic(50, 40, QImage::Format_RGB32);
  31   QPainter aPainter(&aPic);
  32   aPainter.drawEllipse(6, 7, 38, 30);
  33   aPainter.drawLine(0, 40, 10, 0);
  34   aPainter.drawLine(10, 0, 25, 35);
  35   aPainter.drawLine(25, 35, 40, 0);
  36   aPainter.drawLine(40, 0, 50, 40);
  37   return aPic;
  38 }
  39
  40 void test_HYDROData_Image::testQImage()
  41 {
  42   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  43
  44   Handle(HYDROData_Image) anImage =
  45     Handle(HYDROData_Image)::DownCast(aDoc->CreateObject(KIND_IMAGE));
  46   // empty image
  47   QImage anEmpty = anImage->Image();
  48   CPPUNIT_ASSERT(anEmpty.isNull());
  49
  50   // prepare Qt Image for testing
  51   QImage aPic(TestImage());
  52   anImage->SetImage(aPic);
  53   QImage aRestored = anImage->Image();
  54
  55   CPPUNIT_ASSERT(!aRestored.isNull());
  56   //aRestored.save("pic2.bmp");
  57   CPPUNIT_ASSERT(aPic == aRestored);
  58
  59   aDoc->Close();
  60 }
  61
  62 void test_HYDROData_Image::testTrsf()
  63 {
  64   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  65
  66   Handle(HYDROData_Image) anImage =
  67     Handle(HYDROData_Image)::DownCast(aDoc->CreateObject(KIND_IMAGE));
  68   // empty trsf
  69   QTransform anEmpty = anImage->Trsf();
  70   CPPUNIT_ASSERT(anEmpty.isIdentity());
  71
  72   // prepare Qt transformation for testing
  73   QTransform aTrsf;
  74   aTrsf.translate(50, 50);
  75   aTrsf.rotate(45);
  76   aTrsf.scale(0.5, 1.0);
  77
  78   anImage->SetTrsf(aTrsf);
  79   QTransform aRestored = anImage->Trsf();
  80
  81   CPPUNIT_ASSERT(!aRestored.isIdentity());
  82   CPPUNIT_ASSERT(aTrsf == aRestored);
  83
  84   aDoc->Close();
  85 }
  86
  87 void test_HYDROData_Image::testReferences()
  88 {
  89   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  90
  91   Handle(HYDROData_Image) anImage1 =
  92     Handle(HYDROData_Image)::DownCast(aDoc->CreateObject(KIND_IMAGE));
  93   Handle(HYDROData_Image) anImage2 =
  94     Handle(HYDROData_Image)::DownCast(aDoc->CreateObject(KIND_IMAGE));
  95   Handle(HYDROData_Image) anImage3 =
  96     Handle(HYDROData_Image)::DownCast(aDoc->CreateObject(KIND_IMAGE));
  97
  98   // check initially there is no references
  99   CPPUNIT_ASSERT_EQUAL(anImage3->NbReferences(), 0);
 100
 101   // append reference
 102   anImage3->AppendReference(anImage1);
 103   CPPUNIT_ASSERT_EQUAL(anImage3->NbReferences(), 1);
 104   CPPUNIT_ASSERT(IsEqual(anImage3->Reference(0), anImage1));
 105
 106   // change reference
 107   anImage3->ChangeReference(0, anImage2);
 108   CPPUNIT_ASSERT_EQUAL(anImage3->NbReferences(), 1);
 109   CPPUNIT_ASSERT(IsEqual(anImage3->Reference(0), anImage2));
 110
 111   // append one more reference
 112   anImage3->AppendReference(anImage1);
 113   CPPUNIT_ASSERT_EQUAL(anImage3->NbReferences(), 2);
 114   CPPUNIT_ASSERT(IsEqual(anImage3->Reference(0), anImage2));
 115   CPPUNIT_ASSERT(IsEqual(anImage3->Reference(1), anImage1));
 116
 117   // remove reference
 118   anImage3->RemoveReference(0);
 119   CPPUNIT_ASSERT_EQUAL(anImage3->NbReferences(), 1);
 120   CPPUNIT_ASSERT(IsEqual(anImage3->Reference(0), anImage1));
 121
 122   // remove reference
 123   anImage3->RemoveReference(0);
 124   CPPUNIT_ASSERT_EQUAL(anImage3->NbReferences(), 0);
 125
 126   // append two references once again and remove all references
 127   anImage3->ChangeReference(0, anImage1);
 128   anImage3->ChangeReference(1, anImage2);
 129   CPPUNIT_ASSERT_EQUAL(anImage3->NbReferences(), 2);
 130   CPPUNIT_ASSERT(IsEqual(anImage3->Reference(0), anImage1));
 131   CPPUNIT_ASSERT(IsEqual(anImage3->Reference(1), anImage2));
 132   anImage3->ClearReferences();
 133   CPPUNIT_ASSERT_EQUAL(anImage3->NbReferences(), 0);
 134
 135   aDoc->Close();
 136 }
 137
 138 void test_HYDROData_Image::testCopy()
 139 {
 140   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 141
 142   Handle(HYDROData_Image) anImage1 =
 143     Handle(HYDROData_Image)::DownCast(aDoc->CreateObject(KIND_IMAGE));
 144   // fill image 1
 145   anImage1->SetImage(TestImage());
 146
 147   QTransform aTrsf;
 148   aTrsf.translate(50, 50);
 149   aTrsf.rotate(45);
 150   aTrsf.scale(0.5, 1.0);
 151   anImage1->SetTrsf(aTrsf);
 152
 153   Handle(HYDROData_Image) anImage2 =
 154     Handle(HYDROData_Image)::DownCast(aDoc->CreateObject(KIND_IMAGE));
 155
 156   anImage1->AppendReference(anImage2);
 157
 158   // copy image to the new one
 159   Handle(HYDROData_Image) anImage3 =
 160     Handle(HYDROData_Image)::DownCast(aDoc->CreateObject(KIND_IMAGE));
 161   anImage1->CopyTo(anImage3, true);
 162
 163   // check all fields are correctly copied
 164   CPPUNIT_ASSERT(anImage1->Image() == anImage3->Image());
 165   CPPUNIT_ASSERT(anImage1->Trsf() == anImage3->Trsf());
 166   CPPUNIT_ASSERT_EQUAL(anImage3->NbReferences(), 1);
 167   CPPUNIT_ASSERT(IsEqual(anImage3->Reference(0), anImage2));
 168
 169   aDoc->Close();
 170 }
 171
 172 void test_HYDROData_Image::test_static_loadECW()
 173 {
 174   QImage ecwImage;
 175   HYDROData_Image::ECW_FileInfo* ECWInfo = new HYDROData_Image::ECW_FileInfo;
 176   QString refECWPath = REF_DATA_PATH + "/01-2009-0865-6525-LA93.ecw";
 177   QString refPNGPath = REF_DATA_PATH + "/01-2009-0865-6525-LA93.png";
 178   //
 179   CPPUNIT_ASSERT ( HYDROData_Image::OpenECW(refECWPath.toLatin1().data(), ecwImage, ECWInfo) );
 180   CPPUNIT_ASSERT_EQUAL (10000, ECWInfo->myXSize);
 181   CPPUNIT_ASSERT_EQUAL (10000, ECWInfo->myYSize);
 182   CPPUNIT_ASSERT_DOUBLES_EQUAL (0.5, ECWInfo->myCellIncrementX, 0.0001);
 183   CPPUNIT_ASSERT_DOUBLES_EQUAL (-0.5, ECWInfo->myCellIncrementY, 0.0001);
 184   CPPUNIT_ASSERT_DOUBLES_EQUAL (865000, ECWInfo->myOriginX, 0.0001);
 185   CPPUNIT_ASSERT_DOUBLES_EQUAL (6525000, ECWInfo->myOriginY, 0.0001);
 186   //QImage pngImage(refPNGPath);
 187   //CPPUNIT_ASSERT( ecwImage == pngImage);
 188 }
 189
 190 void test_HYDROData_Image::test_loadECW()
 191 {
 192   QImage ecwImage;
 193   QString refECWPath = REF_DATA_PATH + "/01-2009-0865-6525-LA93.ecw";
 194   QString refPNGPath = REF_DATA_PATH + "/01-2009-0865-6525-LA93.png";
 195   //
 196   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 197   Handle(HYDROData_Image) ecwHImage = Handle(HYDROData_Image)::DownCast(aDoc->CreateObject(KIND_IMAGE));
 198   //
 199   CPPUNIT_ASSERT (ecwHImage->LoadImageECW(refECWPath));
 200   //QImage pngImage(refPNGPath);
 201   //CPPUNIT_ASSERT( ecwHImage->Image() == pngImage);
 202
 203   aDoc->Close();
 204 }
 205
 206
 207 void test_HYDROData_Image::testDiff()
 208 {
 209   QImage im1, im2;
 210   im1.load( REF_DATA_PATH + "/LandCover_Triangles.png" );
 211   im2.load( REF_DATA_PATH + "/LandCover_Triangles_Split.png" );
 212
 213   QImage im3 = TestViewer::diff( im1, im2 );
 214   CPPUNIT_ASSERT_IMAGES3( &im3, "diff_image", false );
 215 }