Salome HOME
lot 10 - warnings for DTM - untested
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_DataObject.cxx
index 116afcc2ee641cd2b2aa1a816ef9c668384788b3..f68c26e692f4999d207bfd8e927c067ed1a6b0d1 100644 (file)
@@ -23,6 +23,7 @@
 #include <HYDROData_Object.h>
 #include <HYDROData_ArtificialObject.h>
 #include <HYDROData_NaturalObject.h>
+#include <HYDROData_BCPolygon.h>
 
 #include <TDF_Tool.hxx>
 
@@ -175,6 +176,7 @@ QPixmap HYDROGUI_DataObject::icon( const int theId ) const
         Handle(HYDROData_Object) aFatherObj = anObject3D->GetObject();
         if ( !aFatherObj.IsNull() )
           anObjectKind = aFatherObj->GetKind();
+        anIcon = QObject::tr( QString("HYDRO_%1TYPE%2_ICO").arg( aNeedUpdate ).arg( anObjectKind ).toLatin1() );
       }
       else if ( !anObject.IsNull() )
       {
@@ -197,6 +199,19 @@ QPixmap HYDROGUI_DataObject::icon( const int theId ) const
         painter.end();
         return qpmD;
       }
+      else if (anObjectKind == KIND_BC_POLYGON)
+      {
+        Handle(HYDROData_BCPolygon) aBCObj = Handle(HYDROData_BCPolygon)::DownCast( aDataObject );
+        int aBT = aBCObj->GetBoundaryType();
+        if (aBT == 1)
+          anIcon = QObject::tr( QString("HYDRO_BC_POLYGON_TYPE1_ICO").toLatin1());
+        else if (aBT == 2)
+          anIcon = QObject::tr( QString("HYDRO_BC_POLYGON_TYPE2_ICO").toLatin1());
+        else if (aBT == 3)
+          anIcon = QObject::tr( QString("HYDRO_BC_POLYGON_TYPE3_ICO").toLatin1());
+        else
+          anIcon = QObject::tr( QString("HYDRO_BC_POLYGON_TYPE_UNDEF_ICO").toLatin1());
+      }
       else
         anIcon = QObject::tr( QString("HYDRO_%1TYPE%2_ICO").arg( aNeedUpdate ).arg( anObjectKind ).toLatin1() );
     }