Salome HOME
Small correction for writing of data to file.
[modules/hydro.git] / src / HYDROData / HYDROData_Image.cxx
index 62e1ba5311df9861433fd9a91be4aa2c33ad1edb..930d2a0e0733b9c04f8aa5c4ee5761e3cc7fc69a 100644 (file)
@@ -2,11 +2,16 @@
 #include <HYDROData_Iterator.h>
 
 #include <TDataStd_RealArray.hxx>
-#include <TDataStd_IntegerArray.hxx>
 #include <TDataStd_ByteArray.hxx>
+#include <TDataStd_IntegerArray.hxx>
 #include <TDataStd_ReferenceList.hxx>
+#include <TDataStd_Name.hxx>
+#include <TDataStd_UAttribute.hxx>
+#include <TDataStd_AsciiString.hxx>
 #include <TDF_ListIteratorOfLabelList.hxx>
 
+static const Standard_GUID GUID_MUST_BE_UPDATED("80f2bb81-3873-4631-8ddd-940d2119f000");
+
 IMPLEMENT_STANDARD_HANDLE(HYDROData_Image, HYDROData_Object)
 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Image, HYDROData_Object)
 
@@ -36,50 +41,41 @@ void HYDROData_Image::SetImage(const QImage& theImage)
   aParams->SetValue(3, theImage.bytesPerLine());
   aParams->SetValue(4, (int)(theImage.format()));
   // store data of image in byte array
-  Handle(TDataStd_ByteArray) aData;
-  int aLen = theImage.byteCount();
-  if (!myLab.FindAttribute(TDataStd_ByteArray::GetID(), aData)) {
-    aData = TDataStd_ByteArray::Set(myLab, 1, aLen);
-  }
-  // copy bytes one by one
-  const uchar* aBits = theImage.bits();
-  if (aData->Length() != aLen) {
-    Handle(TColStd_HArray1OfByte) aNewData = new TColStd_HArray1OfByte(1, aLen);
-    for(int a = 0; a < aLen; a++)
-      aNewData->SetValue(a + 1, aBits[a]);
-    aData->ChangeArray(aNewData);
-  } else {
-    for(int a = 0; a < aLen; a++)
-      aData->SetValue(a + 1, aBits[a]);
-  }
-
+  const char* aData = (const char*)(theImage.bits());
+  SaveByteArray(0, aData, theImage.byteCount());
 }
 
 QImage HYDROData_Image::Image()
 {
   Handle(TDataStd_IntegerArray) aParams;
-  Handle(TDataStd_ByteArray) aData;
-  if (!myLab.FindAttribute(TDataStd_IntegerArray::GetID(), aParams) ||
-      !myLab.FindAttribute(TDataStd_ByteArray::GetID(), aData))
+  if (!myLab.FindAttribute(TDataStd_IntegerArray::GetID(), aParams))
+    return QImage(); // return empty image if there is no array
+  int aLen = 0;
+  uchar* anArray = (uchar*)ByteArray(0, aLen);
+  if (!aLen)
     return QImage(); // return empty image if there is no array
-  /*
-  // make uchar array one by one
-  int aLen = aData->Upper();
-  uchar* anArray = new uchar[aLen];
-  for(int a = 0; a < aLen; a++)
-    anArray[a] = aData->Value(a + 1);
-  // recreate image from integer parameters and array of bytes
-  QImage aResult(anArray, aParams->Value(1), aParams->Value(2),
-                 aParams->Value(3), QImage::Format(aParams->Value(4)));
-  delete [] anArray; <- this is wrong, because QImage references to this array
-  */
-  uchar* anArray = (uchar*)(void*)(&(aData->InternalArray()->ChangeArray1().ChangeValue(1)));
   QImage aResult(anArray, aParams->Value(1), aParams->Value(2),
                  aParams->Value(3), QImage::Format(aParams->Value(4)));
-
   return aResult;
 }
 
+void HYDROData_Image::SetFilePath(const QString& theFilePath)
+{
+  TCollection_AsciiString anAsciiStr( theFilePath.toStdString().c_str() );
+  TDataStd_AsciiString::Set( myLab.FindChild( DataTag_FilePath ), anAsciiStr );
+}
+
+QString HYDROData_Image::GetFilePath() const
+{
+  QString aRes;
+
+  Handle(TDataStd_AsciiString) anAsciiStr;
+  if ( myLab.FindChild( DataTag_FilePath ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
+    aRes = QString( anAsciiStr->Get().ToCString() );
+
+  return aRes;
+}
+
 void HYDROData_Image::SetTrsf(const QTransform& theTrsf)
 {
   // locate 9 coeffs of matrix into the real array
@@ -112,6 +108,49 @@ QTransform HYDROData_Image::Trsf()
   return aTrsf;
 }
 
+void HYDROData_Image::SetTrsfPoints(const QPoint& thePointAIn,
+                                    const QPoint& thePointBIn,
+                                    const QPoint& thePointCIn,
+                                    const QPointF& thePointAOut,
+                                    const QPointF& thePointBOut,
+                                    const QPointF& thePointCOut)
+{
+  Handle(TDataStd_RealArray) anArray;
+  if (!myLab.FindChild(DataTag_TrsfPoints).FindAttribute(TDataStd_RealArray::GetID(), anArray)) {
+    anArray = TDataStd_RealArray::Set(myLab.FindChild(DataTag_TrsfPoints), 1, 12);
+  }
+  anArray->SetValue(1, thePointAIn.x());
+  anArray->SetValue(2, thePointAIn.y());
+  anArray->SetValue(3, thePointBIn.x());
+  anArray->SetValue(4, thePointBIn.y());
+  anArray->SetValue(5, thePointCIn.x());
+  anArray->SetValue(6, thePointCIn.y());
+  anArray->SetValue(7, thePointAOut.x());
+  anArray->SetValue(8, thePointAOut.y());
+  anArray->SetValue(9, thePointBOut.x());
+  anArray->SetValue(10, thePointBOut.y());
+  anArray->SetValue(11, thePointCOut.x());
+  anArray->SetValue(12, thePointCOut.y());
+}
+
+void HYDROData_Image::TrsfPoints(QPoint& thePointAIn,
+                                 QPoint& thePointBIn,
+                                 QPoint& thePointCIn,
+                                 QPointF& thePointAOut,
+                                 QPointF& thePointBOut,
+                                 QPointF& thePointCOut)
+{
+  Handle(TDataStd_RealArray) anArray;
+  if (myLab.FindChild(DataTag_TrsfPoints).FindAttribute(TDataStd_RealArray::GetID(), anArray)) {
+    thePointAIn = QPointF( anArray->Value(1), anArray->Value(2) ).toPoint();
+    thePointBIn = QPointF( anArray->Value(3), anArray->Value(4) ).toPoint();
+    thePointCIn = QPointF( anArray->Value(5), anArray->Value(6) ).toPoint();
+    thePointAOut = QPointF( anArray->Value(7), anArray->Value(8) );
+    thePointBOut = QPointF( anArray->Value(9), anArray->Value(10) );
+    thePointCOut = QPointF( anArray->Value(11), anArray->Value(12) );
+  }
+}
+
 void HYDROData_Image::AppendReference(Handle(HYDROData_Image) theReferenced)
 {
   Handle(TDataStd_ReferenceList) aRefs;
@@ -184,3 +223,48 @@ void HYDROData_Image::ClearReferences()
 {
   myLab.ForgetAttribute(TDataStd_ReferenceList::GetID());
 }
+
+void HYDROData_Image::SetOperatorName(const QString theOpName)
+{
+  TDataStd_Name::Set(myLab.FindChild(DataTag_Operator),
+    TCollection_ExtendedString(theOpName.toLatin1().constData()));
+}
+
+QString HYDROData_Image::OperatorName()
+{
+  Handle(TDataStd_Name) aName;
+  if (myLab.FindChild(DataTag_Operator).
+        FindAttribute(TDataStd_Name::GetID(), aName)) {
+    TCollection_AsciiString aStr(aName->Get());
+    return QString(aStr.ToCString());
+  }
+  return QString();
+}
+
+void HYDROData_Image::SetArgs(const QByteArray& theArgs)
+{
+  SaveByteArray(DataTag_Operator, theArgs.constData(), theArgs.length());
+}
+
+QByteArray HYDROData_Image::Args()
+{
+  int aLen = 0;
+  const char* aData = ByteArray(DataTag_Operator, aLen);
+  if (!aLen)
+    return QByteArray();
+  return QByteArray(aData, aLen);
+}
+
+void HYDROData_Image::MustBeUpdated(bool theFlag)
+{
+  if (theFlag) {
+    TDataStd_UAttribute::Set(myLab, GUID_MUST_BE_UPDATED);
+  } else {
+    myLab.ForgetAttribute(GUID_MUST_BE_UPDATED);
+  }
+}
+
+bool HYDROData_Image::MustBeUpdated()
+{
+  return myLab.IsAttribute(GUID_MUST_BE_UPDATED);
+}