Salome HOME
05bb314ca0848d97fbe9a91311de37d7a3f5a2d1
[modules/hydro.git] / src / HYDROData / HYDROData_Image.cxx
1
2 #include "HYDROData_Image.h"
3
4 #include "HYDROData_Document.h"
5 #include "HYDROData_Lambert93.h"
6 #include "HYDROData_Tool.h"
7 #include "HYDROData_OperationsFactory.h"
8
9 #include <TDataStd_RealArray.hxx>
10 #include <TDataStd_ByteArray.hxx>
11 #include <TDataStd_Integer.hxx>
12 #include <TDataStd_IntegerArray.hxx>
13 #include <TDataStd_ReferenceList.hxx>
14 #include <TDataStd_Name.hxx>
15 #include <TDataStd_UAttribute.hxx>
16 #include <TDataStd_AsciiString.hxx>
17
18 #include <ImageComposer_Operator.h>
19 #include <ImageComposer_MetaTypes.h>
20
21 #include <QStringList>
22
23 static const Standard_GUID GUID_SELF_SPLITTED("997995aa-5c19-40bf-9a60-ab4b70ad04d8");
24 static const Standard_GUID GUID_HAS_LOCAL_POINTS("FD8841AA-FC44-42fa-B6A7-0F682CCC6F27");
25 static const Standard_GUID GUID_HAS_GLOBAL_POINTS("330D0E81-742D-4ea3-92D4-484877CFA7C1");
26
27 #define PYTHON_IMAGE_ID "KIND_IMAGE"
28
29 IMPLEMENT_STANDARD_HANDLE(HYDROData_Image, HYDROData_Entity)
30 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Image, HYDROData_Entity)
31
32 HYDROData_Image::HYDROData_Image()
33 : HYDROData_Entity()
34 {
35 }
36
37 HYDROData_Image::~HYDROData_Image()
38 {
39 }
40
41 QStringList HYDROData_Image::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
42 {
43   QStringList aResList;
44
45   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
46   if ( aDocument.IsNull() )
47     return aResList;
48                              
49   QString aDocName = aDocument->GetDocPyName();
50   QString anImageName = GetName();
51
52   aResList << QString( "%1 = %2.CreateObject( %3 );" )
53               .arg( anImageName ).arg( aDocName ).arg( PYTHON_IMAGE_ID );
54   aResList << QString( "%1.SetName( \"%2\" );" )
55               .arg( anImageName ).arg( anImageName );
56
57   QString aFilePath = GetFilePath();
58   if ( !aFilePath.isEmpty() )
59   {
60     aResList << QString( "" );
61     aResList << QString( "%1.LoadImage( \"%2\" );" )
62                 .arg( anImageName ).arg( aFilePath );
63     aResList << QString( "" );
64
65     // Dump transformation points for image
66
67     QString aGap = QString().fill( ' ', anImageName.size() + 16 );
68
69     bool anIsByTwoPoints = IsByTwoPoints();
70
71     QPoint aLocalPointA, aLocalPointB, aLocalPointC;
72     if ( GetLocalPoints( aLocalPointA, aLocalPointB, aLocalPointC ) )
73     {
74       aResList << QString( "%1.SetLocalPoints( QPoint( %2, %3 )," )
75                   .arg( anImageName ).arg( aLocalPointA.x() ).arg( aLocalPointA.y() );
76       aResList << QString( aGap             + "QPoint( %1, %2 )," )
77                   .arg( aLocalPointB.x() ).arg( aLocalPointB.y() );
78       aResList << QString( aGap             + "QPoint( %1, %2 ) );" )
79                   .arg( aLocalPointC.x() ).arg( aLocalPointC.y() );
80       aResList << QString( "" );
81     }
82
83     HYDROData_Image::TransformationMode aTransformationMode;
84     QPointF aTrsfPointA, aTrsfPointB, aTrsfPointC;
85     if ( GetGlobalPoints( aTransformationMode, aTrsfPointA, aTrsfPointB, aTrsfPointC ) )
86     {
87       aResList << QString( "%1.SetGlobalPoints( %2," )
88                   .arg( anImageName ).arg( aTransformationMode );
89       aResList << QString( aGap             +  "QPointF( %1, %2 )," )
90                   .arg( aTrsfPointA.x() ).arg( aTrsfPointA.y() );
91       aResList << QString( aGap             +  "QPointF( %1, %2 )," )
92                   .arg( aTrsfPointB.x() ).arg( aTrsfPointB.y() );
93       aResList << QString( aGap             +  "QPointF( %1, %2 ) );" )
94                   .arg( aTrsfPointC.x() ).arg( aTrsfPointC.y() );
95
96       if ( aTransformationMode == ReferenceImage )
97       {
98         Handle(HYDROData_Image) aRefImg = GetTrsfReferenceImage();
99         setPythonReferenceObject( theTreatedObjects, aResList, aRefImg, "SetTrsfReferenceImage" );
100       }
101     }
102   }
103   else
104   {
105     // Image is composed from other image(s)
106
107     QString anOperatorName = OperatorName();
108     if ( !anOperatorName.isEmpty() )
109     {
110       aResList << QString( "" );
111
112       aResList << QString( "%1.SetOperatorName( \"%2\" );" )
113                   .arg( anImageName ).arg( anOperatorName );
114
115       ImageComposer_Operator* anImageOp = 
116         HYDROData_OperationsFactory::Factory()->Operator( OperatorName() );
117       if ( anImageOp )
118       {
119         // Dump operation arguments
120         QString anOpArgsArrayName;
121         QStringList anOpArgs = anImageOp->dumpArgsToPython( anOpArgsArrayName );
122         if ( !anOpArgs.isEmpty() )
123         {
124           aResList << QString( "" );
125           aResList << anOpArgs;
126
127           aResList << QString( "" );
128           aResList << QString( "%1.SetArgs( %2 );" )
129                       .arg( anImageName ).arg( anOpArgsArrayName );
130         }
131       }
132     }
133     
134     int aNbReferences = NbReferences();
135     if ( aNbReferences > 0 )
136     {
137       aResList << QString( "" );
138
139       for ( int i = 0; i < aNbReferences; ++i )
140       {
141         Handle(HYDROData_Image) aRefImg = Handle(HYDROData_Image)::DownCast( Reference( i ) );
142         setPythonReferenceObject( theTreatedObjects, aResList, aRefImg, "AppendReference" );
143       }
144     }
145
146     // Necessary to update image in case of composed operator
147     aResList << QString( "" );
148     aResList << QString( "%1.Update();" ).arg( anImageName );
149   }
150
151   return aResList;
152 }
153
154 void HYDROData_Image::Update()
155 {
156   HYDROData_OperationsFactory* aFactory = HYDROData_OperationsFactory::Factory();
157
158   ImageComposer_Operator* anOp = aFactory->Operator( OperatorName() );
159   if ( anOp ) // Update image if there is an operation
160   {
161     // Fill by arguments and process the operation
162     QVariant anObj1, anObj2;
163     int aNbReferences = NbReferences();
164
165     if ( aNbReferences > 0 )
166     {
167       // First referenced object
168       Handle(HYDROData_Entity) aRefObj = Reference( 0 );
169       if ( !aRefObj.IsNull() )
170       {
171         anObj1 = aRefObj->GetDataVariant();
172         if ( !anObj1.isNull() && anObj1.canConvert<ImageComposer_Image>() )
173         {
174           ImageComposer_Image anImage = anObj1.value<ImageComposer_Image>();
175           QTransform aTransform = anImage.transform();
176           SetTrsf( aTransform );
177         }
178       }
179     }
180
181     if ( aNbReferences > 1 )
182     {
183       // Second referenced object
184       Handle(HYDROData_Entity) aRefObj = Reference( 1 );
185       if ( !aRefObj.IsNull() )
186         anObj2 = aRefObj->GetDataVariant();
187     }
188
189     ImageComposer_Image aResImg = anOp->process( anObj1, anObj2 );
190     SetImage( aResImg );
191     SetTrsf( aResImg.transform() );
192   }
193   else // Update image if it positioned relatively to other image
194   {
195     UpdateTrsf();
196   }
197
198   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
199   if ( !aDocument.IsNull() )
200   {
201     // Change the states of this and all depended images
202     SetToUpdate( true );
203     HYDROData_Tool::SetMustBeUpdatedImages( aDocument );
204     SetToUpdate( false );
205   }
206 }
207
208 QVariant HYDROData_Image::GetDataVariant()
209 {
210   QTransform aTransform = Trsf();
211
212   ImageComposer_Image anImage = Image();
213   anImage.setTransform( aTransform );
214
215   QVariant aVarData;
216   aVarData.setValue<ImageComposer_Image>( anImage );
217   
218   return aVarData;
219 }
220
221 HYDROData_SequenceOfObjects HYDROData_Image::GetAllReferenceObjects() const
222 {
223   HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
224
225   Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
226   if ( !aRefImage.IsNull() )
227     aResSeq.Append( aRefImage );
228
229   HYDROData_SequenceOfObjects aSeqRefObjects = GetReferenceObjects( 0 );
230   aResSeq.Append( aSeqRefObjects );
231
232   return aResSeq;
233 }
234
235 void HYDROData_Image::SetImage(const QImage& theImage)
236 {
237   if (theImage.isNull()) {
238     // for empty image remove all previously stored attributes
239     myLab.ForgetAttribute(TDataStd_IntegerArray::GetID());
240     myLab.ForgetAttribute(TDataStd_ByteArray::GetID());
241     return;
242   }
243   // store width, height, bytes per line and format in integer array 
244   Handle(TDataStd_IntegerArray) aParams;
245   if (!myLab.FindAttribute(TDataStd_IntegerArray::GetID(), aParams)) {
246     aParams = TDataStd_IntegerArray::Set(myLab, 1, 4);
247   }
248   aParams->SetValue(1, theImage.width());
249   aParams->SetValue(2, theImage.height());
250   aParams->SetValue(3, theImage.bytesPerLine());
251   aParams->SetValue(4, (int)(theImage.format()));
252   // store data of image in byte array
253   const char* aData = (const char*)(theImage.bits());
254   SaveByteArray(0, aData, theImage.byteCount());
255 }
256
257 bool HYDROData_Image::LoadImage(const QString& theFilePath)
258 {
259   QImage anImage( theFilePath );
260   SetImage( anImage );
261   return !anImage.isNull();
262 }
263
264 QImage HYDROData_Image::Image()
265 {
266   Handle(TDataStd_IntegerArray) aParams;
267   if (!myLab.FindAttribute(TDataStd_IntegerArray::GetID(), aParams))
268     return QImage(); // return empty image if there is no array
269   int aLen = 0;
270   uchar* anArray = (uchar*)ByteArray(0, aLen);
271   if (!aLen)
272     return QImage(); // return empty image if there is no array
273   QImage aResult(anArray, aParams->Value(1), aParams->Value(2),
274                  aParams->Value(3), QImage::Format(aParams->Value(4)));
275   return aResult;
276 }
277
278 void HYDROData_Image::SetFilePath(const QString& theFilePath)
279 {
280   TCollection_AsciiString anAsciiStr( theFilePath.toStdString().c_str() );
281   TDataStd_AsciiString::Set( myLab.FindChild( DataTag_FilePath ), anAsciiStr );
282 }
283
284 QString HYDROData_Image::GetFilePath() const
285 {
286   QString aRes;
287
288   Handle(TDataStd_AsciiString) anAsciiStr;
289   if ( myLab.FindChild( DataTag_FilePath ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
290     aRes = QString( anAsciiStr->Get().ToCString() );
291
292   return aRes;
293 }
294
295 void HYDROData_Image::SetTrsf(const QTransform& theTrsf)
296 {
297   // locate 9 coeffs of matrix into the real array
298   Handle(TDataStd_RealArray) anArray;
299   if (!myLab.FindAttribute(TDataStd_RealArray::GetID(), anArray)) {
300     if (theTrsf.isIdentity()) return; // no need to store identity transformation
301     anArray = TDataStd_RealArray::Set(myLab, 1, 9);
302   }
303   anArray->SetValue(1, theTrsf.m11());
304   anArray->SetValue(2, theTrsf.m12());
305   anArray->SetValue(3, theTrsf.m13());
306   anArray->SetValue(4, theTrsf.m21());
307   anArray->SetValue(5, theTrsf.m22());
308   anArray->SetValue(6, theTrsf.m23());
309   anArray->SetValue(7, theTrsf.m31());
310   anArray->SetValue(8, theTrsf.m32());
311   anArray->SetValue(9, theTrsf.m33());
312 }
313
314 QTransform HYDROData_Image::Trsf() const
315 {
316   // get 9 coeffs of matrix from the real array
317   Handle(TDataStd_RealArray) anArray;
318   if (!myLab.FindAttribute(TDataStd_RealArray::GetID(), anArray))
319     return QTransform(); // return identity if there is no array
320   QTransform aTrsf(
321     anArray->Value(1), anArray->Value(2), anArray->Value(3), 
322     anArray->Value(4), anArray->Value(5), anArray->Value(6), 
323     anArray->Value(7), anArray->Value(8), anArray->Value(9));
324   return aTrsf;
325 }
326
327 void HYDROData_Image::UpdateTrsf()
328 {
329   QPoint aPointA, aPointB, aPointC;
330   if ( !GetLocalPoints( aPointA, aPointB, aPointC ) )
331     return;
332
333   TransformationMode aTrsfMode;
334   QPointF aTrsfPointA, aTrsfPointB, aTrsfPointC;
335   if ( !GetGlobalPoints( aTrsfMode, aTrsfPointA, aTrsfPointB, aTrsfPointC ) )
336     return;
337
338   QTransform aRefTransform;
339   Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
340
341   bool anIsRefImage = aTrsfMode == ReferenceImage;
342   if ( anIsRefImage )
343   {
344     if ( aRefImage.IsNull() )
345       return;
346
347     aRefTransform = aRefImage->Trsf();
348   }
349
350   bool anIsByTwoPoints = IsByTwoPoints();
351
352   // Convert lambert coordinates to cartesian
353   if ( aTrsfMode == ManualGeodesic )
354   {
355     double aXCart = 0, aYCart = 0;
356
357     HYDROData_Lambert93::toXY( aTrsfPointA.y(), aTrsfPointA.x(), aXCart, aYCart );
358     aTrsfPointA = QPointF( aXCart, aYCart );
359
360     HYDROData_Lambert93::toXY( aTrsfPointB.y(), aTrsfPointB.x(), aXCart, aYCart );
361     aTrsfPointB = QPointF( aXCart, aYCart );
362
363     if ( !anIsByTwoPoints )
364     {
365       HYDROData_Lambert93::toXY( aTrsfPointC.y(), aTrsfPointC.x(), aXCart, aYCart );
366       aTrsfPointC = QPointF( aXCart, aYCart );
367     }
368   }
369
370   // generate third points if needed
371   if ( anIsByTwoPoints )
372   {
373     aPointC = generateThirdPoint( aPointA, aPointB, true ).toPoint();
374     aTrsfPointC = generateThirdPoint( aTrsfPointA, aTrsfPointB, anIsRefImage );
375   }
376
377   int xa = aPointA.x();
378   int ya = aPointA.y();
379   int xb = aPointB.x();
380   int yb = aPointB.y();
381   int xc = aPointC.x();
382   int yc = aPointC.y();
383
384   double xta = aTrsfPointA.x();
385   double yta = aTrsfPointA.y();
386   double xtb = aTrsfPointB.x();
387   double ytb = aTrsfPointB.y();
388   double xtc = aTrsfPointC.x();
389   double ytc = aTrsfPointC.y();
390
391   // first, check that three input points don't belong to a single line
392   if( ( yb - ya ) * ( xc - xa ) == ( yc - ya ) * ( xb - xa ) )
393     return;
394
395   // the same check for the reference points
396   if( anIsRefImage && ( ( ytb - yta ) * ( xtc - xta ) == ( ytc - yta ) * ( xtb - xta ) ) )
397     return;
398
399   QTransform aTransform1( xa, ya, 1, xb, yb, 1, xc, yc, 1 );
400   QTransform aTransform2( xta, yta, 1, xtb, ytb, 1, xtc, ytc, 1 );
401
402   bool anIsInvertible = false;
403   QTransform aTransform1Inverted = aTransform1.inverted( &anIsInvertible );
404   if( !anIsInvertible )
405     return;
406
407   QTransform aResTransform = aTransform1Inverted * aTransform2;
408   if( anIsRefImage )
409     aResTransform *= aRefTransform;
410
411   SetTrsf( aResTransform );
412 }
413
414 bool HYDROData_Image::IsByTwoPoints() const
415 {
416   if ( !HasLocalPoints() || !HasGlobalPoints() )
417     return false;
418
419   QPoint aPointA, aPointB, aPointC;
420   GetLocalPoints( aPointA, aPointB, aPointC );
421
422   return aPointC.x() < 0 && aPointC.y() < 0; 
423 }
424
425 bool HYDROData_Image::HasReferences() const
426 {
427   Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
428   int aNbReferences = NbReferences();
429
430   return !aRefImage.IsNull() || aNbReferences > 0;
431 }
432
433 void HYDROData_Image::RemoveAllReferences()
434 {
435   if ( !HasReferences() )
436     return;
437
438   Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
439   if ( !aRefImage.IsNull() )
440   {
441     RemoveTrsfReferenceImage();
442   }
443   else
444   {
445     ClearReferences();
446     SetOperatorName( "" );
447     SetArgs( "" );
448     SetIsSelfSplitted( false );
449   }
450
451   SetToUpdate( false );
452
453   bool anIsByTwoPoints = IsByTwoPoints();
454
455   QImage anImage = Image();
456   if ( anImage.isNull() )
457     return;
458
459   // Set local points to default position
460   QPoint aLocalPointA = QPoint( 0, 0 );
461   QPoint aLocalPointB = QPoint( anImage.width(), 0 );
462   QPoint aLocalPointC = anIsByTwoPoints ? QPoint( INT_MIN, INT_MIN ) : QPoint( 0, anImage.height() );
463
464   SetLocalPoints( aLocalPointA, aLocalPointB, aLocalPointC, false );
465
466   // Calculate global points
467   QTransform aTransform = Trsf();
468
469   QPointF aTrsfPointA = QPointF( aTransform.map( aLocalPointA ) );
470   QPointF aTrsfPointB = QPointF( aTransform.map( aLocalPointB ) );
471   QPointF aTrsfPointC = anIsByTwoPoints ? QPointF( INT_MIN, INT_MIN ) : 
472                                           QPointF( aTransform.map( aLocalPointC ) );
473
474   SetGlobalPoints( ManualCartesian, aTrsfPointA, aTrsfPointB, aTrsfPointC );
475 }
476
477 void HYDROData_Image::SetLocalPoints( const QPoint& thePointA,
478                                       const QPoint& thePointB,
479                                       const QPoint& thePointC,
480                                       const bool    theIsUpdate )
481 {
482   Handle(TDataStd_RealArray) anArray;
483   if ( !myLab.FindChild( DataTag_TrsfPoints ).FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
484     anArray = TDataStd_RealArray::Set( myLab.FindChild( DataTag_TrsfPoints ), 1, 12 );
485
486   anArray->SetValue( 1, thePointA.x() );
487   anArray->SetValue( 2, thePointA.y() );
488   anArray->SetValue( 3, thePointB.x() );
489   anArray->SetValue( 4, thePointB.y() );
490   anArray->SetValue( 5, thePointC.x() );
491   anArray->SetValue( 6, thePointC.y() );
492
493   TDataStd_UAttribute::Set( myLab.FindChild( DataTag_TrsfPoints ), GUID_HAS_LOCAL_POINTS );
494
495   if ( theIsUpdate )
496     UpdateTrsf();
497 }
498
499 bool HYDROData_Image::GetLocalPoints( QPoint& thePointA,
500                                       QPoint& thePointB,
501                                       QPoint& thePointC ) const
502 {
503   if ( !HasLocalPoints() )
504     return false;
505
506   Handle(TDataStd_RealArray) anArray;
507   myLab.FindChild( DataTag_TrsfPoints ).FindAttribute( TDataStd_RealArray::GetID(), anArray );
508
509   thePointA = QPointF( anArray->Value( 1 ), anArray->Value( 2 ) ).toPoint();
510   thePointB = QPointF( anArray->Value( 3 ), anArray->Value( 4 ) ).toPoint();
511   thePointC = QPointF( anArray->Value( 5 ), anArray->Value( 6 ) ).toPoint();
512
513   return true;
514 }
515
516 bool HYDROData_Image::HasLocalPoints() const
517 {
518   TDF_Label aLabel = myLab.FindChild( DataTag_TrsfPoints, false );
519   if ( aLabel.IsNull() || !aLabel.IsAttribute( GUID_HAS_LOCAL_POINTS ) )
520     return false;
521
522   Handle(TDataStd_RealArray) anArray;
523   return aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray );
524 }
525
526
527 void HYDROData_Image::SetGlobalPoints( const TransformationMode& theMode,
528                                        const QPointF&            thePointA,
529                                        const QPointF&            thePointB,
530                                        const QPointF&            thePointC,
531                                        const bool                theIsUpdate )
532 {
533   Handle(TDataStd_RealArray) anArray;
534   if ( !myLab.FindChild( DataTag_TrsfPoints ).FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
535     anArray = TDataStd_RealArray::Set( myLab.FindChild( DataTag_TrsfPoints ), 1, 12 );
536
537   anArray->SetValue( 7,  thePointA.x() );
538   anArray->SetValue( 8,  thePointA.y() );
539   anArray->SetValue( 9,  thePointB.x() );
540   anArray->SetValue( 10, thePointB.y() );
541   anArray->SetValue( 11, thePointC.x() );
542   anArray->SetValue( 12, thePointC.y() );
543   
544   SetTrsfMode( theMode );
545
546   TDataStd_UAttribute::Set( myLab.FindChild( DataTag_TrsfPoints ), GUID_HAS_GLOBAL_POINTS );
547
548   if ( theIsUpdate )
549     UpdateTrsf();
550
551   /*
552   if( anIsRefImage )
553   {
554     aCPointA = QPointF( aTransform.map( aPointA ) );
555     aCPointB = QPointF( aTransform.map( aPointB ) );
556     aCPointC = QPointF( aTransform.map( aPointC ) );
557
558     // compute Lambert93 points
559     xca = aCPointA.x();
560     yca = aCPointA.y();
561     xcb = aCPointB.x();
562     ycb = aCPointB.y();
563     xcc = aCPointC.x();
564     ycc = aCPointC.y();
565
566     double xla = 0, yla = 0, xlb = 0, ylb = 0, xlc = 0, ylc = 0;
567     HYDROData_Lambert93::toGeo( xca, yca, yla, xla );
568     HYDROData_Lambert93::toGeo( xcb, ycb, ylb, xlb );
569     HYDROData_Lambert93::toGeo( xcc, ycc, ylc, xlc );
570
571     aLPointA = QPointF( xla * 3600.0, yla * 3600.0 ); // convert degrees to seconds
572     aLPointB = QPointF( xlb * 3600.0, ylb * 3600.0 ); // convert degrees to seconds
573     aLPointC = QPointF( xlc * 3600.0, ylc * 3600.0 ); // convert degrees to seconds
574   }
575   */
576 }
577
578 bool HYDROData_Image::GetGlobalPoints( TransformationMode& theMode,
579                                        QPointF&            thePointA,
580                                        QPointF&            thePointB,
581                                        QPointF&            thePointC ) const
582 {
583   if ( !HasGlobalPoints() )
584     return false;
585
586   theMode = GetTrsfMode();
587
588   Handle(TDataStd_RealArray) anArray;
589   myLab.FindChild( DataTag_TrsfPoints ).FindAttribute( TDataStd_RealArray::GetID(), anArray );
590
591   thePointA = QPointF( anArray->Value( 7  ), anArray->Value( 8  ) );
592   thePointB = QPointF( anArray->Value( 9  ), anArray->Value( 10 ) );
593   thePointC = QPointF( anArray->Value( 11 ), anArray->Value( 12 ) );
594
595   return true;
596 }
597
598 bool HYDROData_Image::HasGlobalPoints() const
599 {
600   TDF_Label aLabel = myLab.FindChild( DataTag_TrsfPoints, false );
601   if ( aLabel.IsNull() || !aLabel.IsAttribute( GUID_HAS_GLOBAL_POINTS ) )
602     return false;
603
604   Handle(TDataStd_RealArray) anArray;
605   return aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray );
606 }
607
608 void HYDROData_Image::SetReferencePoints( const Handle(HYDROData_Image)& theRefImage,
609                                           const QPointF&                 thePointA,
610                                           const QPointF&                 thePointB,
611                                           const QPointF&                 thePointC,
612                                           const bool                     theIsUpdate )
613 {
614   SetTrsfReferenceImage( theRefImage );
615   SetGlobalPoints( ReferenceImage, thePointA, thePointB, thePointC, theIsUpdate );
616 }
617
618 bool HYDROData_Image::GetReferencePoints( Handle(HYDROData_Image)& theRefImage,
619                                           QPointF&                 thePointA,
620                                           QPointF&                 thePointB,
621                                           QPointF&                 thePointC ) const
622 {
623   if ( !HasReferencePoints() )
624     return false;
625
626   theRefImage = GetTrsfReferenceImage();
627
628   TransformationMode aMode;
629   GetGlobalPoints( aMode, thePointA, thePointB, thePointC );
630
631   return true;
632 }
633
634 bool HYDROData_Image::HasReferencePoints() const
635 {
636   if ( !HasGlobalPoints() )
637     return false;
638
639   Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
640   if ( aRefImage.IsNull() )
641     return false;
642
643   TransformationMode aTrsfMode = GetTrsfMode();
644   if ( aTrsfMode != ReferenceImage )
645     return false;
646
647   return true;
648 }
649
650 void HYDROData_Image::SetTrsfMode( const TransformationMode& theMode )
651 {
652   TDataStd_Integer::Set( myLab.FindChild( DataTag_TrsfMode ), (int)theMode );
653 }
654
655 HYDROData_Image::TransformationMode HYDROData_Image::GetTrsfMode() const
656 {
657   TransformationMode aResMode = ManualGeodesic;
658
659   TDF_Label aLabel = myLab.FindChild( DataTag_TrsfPoints, false );
660   if ( !aLabel.IsNull() )
661   {
662     Handle(TDataStd_Integer) aMode;
663     if ( myLab.FindChild( DataTag_TrsfMode ).FindAttribute( TDataStd_Integer::GetID(), aMode ) )
664       aResMode = (TransformationMode)aMode->Get();
665   }
666
667   return aResMode;
668 }
669
670 void HYDROData_Image::SetTrsfReferenceImage( const Handle(HYDROData_Image)& theRefImage )
671 {
672   SetReferenceObject( theRefImage, DataTag_TrsfImage );
673 }
674
675 Handle(HYDROData_Image) HYDROData_Image::GetTrsfReferenceImage() const
676 {
677   return Handle(HYDROData_Image)::DownCast( GetReferenceObject( DataTag_TrsfImage ) );
678 }
679
680 void HYDROData_Image::RemoveTrsfReferenceImage()
681 {
682   return RemoveReferenceObject( DataTag_TrsfImage );
683 }
684
685 void HYDROData_Image::AppendReference( const Handle(HYDROData_Entity)& theReferenced )
686 {
687   AddReferenceObject( theReferenced, 0 );
688 }
689
690 int HYDROData_Image::NbReferences() const
691 {
692   return NbReferenceObjects( 0 );
693 }
694
695 Handle(HYDROData_Entity) HYDROData_Image::Reference( const int theIndex ) const
696 {
697   return GetReferenceObject( 0, theIndex );
698 }
699
700 void HYDROData_Image::ChangeReference(
701     const int theIndex, Handle(HYDROData_Entity) theReferenced)
702 {
703   SetReferenceObject( theReferenced, 0, theIndex );
704 }
705
706 void HYDROData_Image::RemoveReference(const int theIndex)
707 {
708   RemoveReferenceObject( 0, theIndex );
709 }
710
711 void HYDROData_Image::ClearReferences()
712 {
713   ClearReferenceObjects( 0 );
714 }
715
716 void HYDROData_Image::SetOperatorName(const QString theOpName)
717 {
718   TDataStd_Name::Set(myLab.FindChild(DataTag_Operator),
719     TCollection_ExtendedString(theOpName.toLatin1().constData()));
720 }
721
722 QString HYDROData_Image::OperatorName() const
723 {
724   Handle(TDataStd_Name) aName;
725   if (myLab.FindChild(DataTag_Operator).
726         FindAttribute(TDataStd_Name::GetID(), aName)) {
727     TCollection_AsciiString aStr(aName->Get());
728     return QString(aStr.ToCString());
729   }
730   return QString();
731 }
732
733 void HYDROData_Image::SetArgs(const QByteArray& theArgs)
734 {
735   SaveByteArray(DataTag_Operator, theArgs.constData(), theArgs.length());
736 }
737
738 QByteArray HYDROData_Image::Args() const
739 {
740   int aLen = 0;
741   const char* aData = ByteArray(DataTag_Operator, aLen);
742   if (!aLen)
743     return QByteArray();
744   return QByteArray(aData, aLen);
745 }
746
747 void HYDROData_Image::SetIsSelfSplitted(bool theFlag)
748 {
749   if (theFlag) {
750     TDataStd_UAttribute::Set(myLab, GUID_SELF_SPLITTED);
751   } else {
752     myLab.ForgetAttribute(GUID_SELF_SPLITTED);
753   }
754 }
755
756 bool HYDROData_Image::IsSelfSplitted() const
757 {
758   return myLab.IsAttribute(GUID_SELF_SPLITTED);
759 }
760
761 QPointF HYDROData_Image::generateThirdPoint( const QPointF& thePointA,
762                                              const QPointF& thePointB,
763                                              const bool&    theIsLocal ) const
764 {
765   // Rotate vector to 90 degrees : clockwise - for local
766   //                               anticlockwise - for global
767   const double aTheta = theIsLocal ? -M_PI_2 : M_PI_2;
768
769   QPointF aResPoint;
770
771   // Move to (0,0) for correct rotation
772   double x = thePointB.x() - thePointA.x();
773   double y = thePointB.y() - thePointA.y();
774
775   aResPoint.setX( x * cos( aTheta ) - y * sin( aTheta ) );
776   aResPoint.setY( x * sin( aTheta ) + y * cos( aTheta ) );
777
778   // Move back to origin position
779   aResPoint.setX( aResPoint.x() + thePointA.x() );
780   aResPoint.setY( aResPoint.y() + thePointA.y() );
781
782   return aResPoint;
783 }
784