Salome HOME
Separate action "Find bottom" for context menu.
[modules/hydro.git] / src / HYDROData / test_HYDROData_Image.cxx
1 // Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License, or (at your option) any later version.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 #include<test_HYDROData_Image.h>
24
25 #include <HYDROData_Document.h>
26 #include <HYDROData_Image.h>
27 #include <QPainter>
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);
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 }