1 #include <HYDROData_Image.h>
2 #include <HYDROData_Iterator.h>
4 #include <TDataStd_RealArray.hxx>
5 #include <TDataStd_IntegerArray.hxx>
6 #include <TDataStd_ByteArray.hxx>
7 #include <TDataStd_ReferenceList.hxx>
8 #include <TDF_ListIteratorOfLabelList.hxx>
10 IMPLEMENT_STANDARD_HANDLE(HYDROData_Image, HYDROData_Object)
11 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Image, HYDROData_Object)
13 HYDROData_Image::HYDROData_Image()
17 HYDROData_Image::~HYDROData_Image()
21 void HYDROData_Image::SetImage(const QImage& theImage)
23 if (theImage.isNull()) {
24 // for empty image remove all previously stored attributes
25 myLab.ForgetAttribute(TDataStd_IntegerArray::GetID());
26 myLab.ForgetAttribute(TDataStd_ByteArray::GetID());
29 // store width, height, bytes per line and format in integer array
30 Handle(TDataStd_IntegerArray) aParams;
31 if (!myLab.FindAttribute(TDataStd_IntegerArray::GetID(), aParams)) {
32 aParams = TDataStd_IntegerArray::Set(myLab, 1, 4);
34 aParams->SetValue(1, theImage.width());
35 aParams->SetValue(2, theImage.height());
36 aParams->SetValue(3, theImage.bytesPerLine());
37 aParams->SetValue(4, (int)(theImage.format()));
38 // store data of image in byte array
39 Handle(TDataStd_ByteArray) aData;
40 int aLen = theImage.byteCount();
41 if (!myLab.FindAttribute(TDataStd_ByteArray::GetID(), aData)) {
42 aData = TDataStd_ByteArray::Set(myLab, 1, aLen);
44 // copy bytes one by one
45 const uchar* aBits = theImage.bits();
46 if (aData->Length() != aLen) {
47 Handle(TColStd_HArray1OfByte) aNewData = new TColStd_HArray1OfByte(1, aLen);
48 for(int a = 0; a < aLen; a++)
49 aNewData->SetValue(a + 1, aBits[a]);
50 aData->ChangeArray(aNewData);
52 for(int a = 0; a < aLen; a++)
53 aData->SetValue(a + 1, aBits[a]);
58 QImage HYDROData_Image::Image()
60 Handle(TDataStd_IntegerArray) aParams;
61 Handle(TDataStd_ByteArray) aData;
62 if (!myLab.FindAttribute(TDataStd_IntegerArray::GetID(), aParams) ||
63 !myLab.FindAttribute(TDataStd_ByteArray::GetID(), aData))
64 return QImage(); // return empty image if there is no array
66 // make uchar array one by one
67 int aLen = aData->Upper();
68 uchar* anArray = new uchar[aLen];
69 for(int a = 0; a < aLen; a++)
70 anArray[a] = aData->Value(a + 1);
71 // recreate image from integer parameters and array of bytes
72 QImage aResult(anArray, aParams->Value(1), aParams->Value(2),
73 aParams->Value(3), QImage::Format(aParams->Value(4)));
74 delete [] anArray; <- this is wrong, because QImage references to this array
76 uchar* anArray = (uchar*)(void*)(&(aData->InternalArray()->ChangeArray1().ChangeValue(1)));
77 QImage aResult(anArray, aParams->Value(1), aParams->Value(2),
78 aParams->Value(3), QImage::Format(aParams->Value(4)));
83 void HYDROData_Image::SetTrsf(const QTransform& theTrsf)
85 // locate 9 coeffs of matrix into the real array
86 Handle(TDataStd_RealArray) anArray;
87 if (!myLab.FindAttribute(TDataStd_RealArray::GetID(), anArray)) {
88 if (theTrsf.isIdentity()) return; // no need to store identity transformation
89 anArray = TDataStd_RealArray::Set(myLab, 1, 9);
91 anArray->SetValue(1, theTrsf.m11());
92 anArray->SetValue(2, theTrsf.m12());
93 anArray->SetValue(3, theTrsf.m13());
94 anArray->SetValue(4, theTrsf.m21());
95 anArray->SetValue(5, theTrsf.m22());
96 anArray->SetValue(6, theTrsf.m23());
97 anArray->SetValue(7, theTrsf.m31());
98 anArray->SetValue(8, theTrsf.m32());
99 anArray->SetValue(9, theTrsf.m33());
102 QTransform HYDROData_Image::Trsf()
104 // get 9 coeffs of matrix from the real array
105 Handle(TDataStd_RealArray) anArray;
106 if (!myLab.FindAttribute(TDataStd_RealArray::GetID(), anArray))
107 return QTransform(); // return identity if there is no array
109 anArray->Value(1), anArray->Value(2), anArray->Value(3),
110 anArray->Value(4), anArray->Value(5), anArray->Value(6),
111 anArray->Value(7), anArray->Value(8), anArray->Value(9));
115 void HYDROData_Image::AppendReference(Handle(HYDROData_Image) theReferenced)
117 Handle(TDataStd_ReferenceList) aRefs;
118 if (!myLab.FindAttribute(TDataStd_ReferenceList::GetID(), aRefs))
119 aRefs = TDataStd_ReferenceList::Set(myLab);
120 aRefs->Append(theReferenced->Label());
123 int HYDROData_Image::NbReferences()
125 Handle(TDataStd_ReferenceList) aRefs;
126 if (!myLab.FindAttribute(TDataStd_ReferenceList::GetID(), aRefs))
128 return aRefs->Extent();
131 Handle(HYDROData_Image) HYDROData_Image::Reference(const int theIndex) const
133 Handle(TDataStd_ReferenceList) aRefs;
134 if (!myLab.FindAttribute(TDataStd_ReferenceList::GetID(), aRefs))
135 return Handle(HYDROData_Image)();
136 if (theIndex < 0 || theIndex >= aRefs->Extent())
137 return Handle(HYDROData_Image)();
139 TDF_ListIteratorOfLabelList anIter(aRefs->List());
140 for(int anIndex = 0; anIndex != theIndex; anIter.Next(), anIndex++);
141 const TDF_Label& aRefLab = anIter.Value();
142 return Handle(HYDROData_Image)::DownCast(HYDROData_Iterator::Object(aRefLab));
145 void HYDROData_Image::ChangeReference(
146 const int theIndex, Handle(HYDROData_Image) theReferenced)
148 Handle(TDataStd_ReferenceList) aRefs;
149 if (!myLab.FindAttribute(TDataStd_ReferenceList::GetID(), aRefs))
150 aRefs = TDataStd_ReferenceList::Set(myLab);
151 if (theIndex >= aRefs->Extent()) { // for too big index append it just to the end
152 AppendReference(theReferenced);
153 } else { // remove and insert new
154 TDF_ListIteratorOfLabelList anIter(aRefs->List());
156 for(; anIndex != theIndex; anIter.Next(), anIndex++);
157 const TDF_Label& aRemovedLab = anIter.Value();
159 aRefs->Remove(aRemovedLab);
161 aRefs->InsertBefore(theReferenced->Label(), anIter.Value());
163 aRefs->Append(theReferenced->Label());
167 void HYDROData_Image::RemoveReference(const int theIndex)
169 Handle(TDataStd_ReferenceList) aRefs;
170 if (!myLab.FindAttribute(TDataStd_ReferenceList::GetID(), aRefs))
171 return; // no references, nothing to remove
172 if (aRefs->Extent() == 1 && theIndex == 0) { // remove all if only one
176 TDF_ListIteratorOfLabelList anIter(aRefs->List());
178 for(; anIndex != theIndex && anIter.More(); anIter.Next(), anIndex++);
180 aRefs->Remove(anIter.Value());
183 void HYDROData_Image::ClearReferences()
185 myLab.ForgetAttribute(TDataStd_ReferenceList::GetID());