Name validator is added to the Calculation Case dialog.

[modules/hydro.git] / src / HYDROGUI / HYDROGUI_DataObject.cxx

diff --git a/src/HYDROGUI/HYDROGUI_DataObject.cxx b/src/HYDROGUI/HYDROGUI_DataObject.cxx
index f0b47e22b4beb40f0440981ae3271517a0b6a97f..133a99d36ff72d7e2fa2ba00abadb27e70af2e3f 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DataObject.cxx
+++ b/src/HYDROGUI/HYDROGUI_DataObject.cxx
@@ -23,13 +23,16 @@
 #include "HYDROGUI_DataObject.h"
 
 #include <HYDROData_Image.h>
+#include <HYDROData_Zone.h>
+#include <HYDROData_Object.h>
+#include <HYDROData_Bathymetry.h>
 
 #include <SUIT_DataObject.h>
 
 #include <TDF_Tool.hxx>
 
 HYDROGUI_DataObject::HYDROGUI_DataObject( SUIT_DataObject* theParent, 
-                                          Handle(HYDROData_Object) theData,
+                                          Handle(HYDROData_Entity) theData,
                                           const QString& theParentEntry )
 : CAM_DataObject( theParent ),
   LightApp_DataObject( theParent ),
@@ -72,14 +75,67 @@ QFont HYDROGUI_DataObject::font( const int theId ) const
   return aFont;
 }
 
-QString HYDROGUI_DataObject::dataObjectEntry( const Handle(HYDROData_Object)& theObject )
+QString HYDROGUI_DataObject::text( const int theColumnId ) const
+{
+  QString aRes;
+  if( !myData.IsNull() )
+  {
+    if ( theColumnId == RefObjectId || theColumnId == BathymetryId )
+    {
+      Handle(HYDROData_Zone) aZone = Handle(HYDROData_Zone)::DownCast( myData );
+      if ( !aZone.IsNull() )
+      {
+        HYDROData_SequenceOfObjects::Iterator anIter( aZone->GetGeometryObjects() );
+        for ( ; anIter.More(); anIter.Next() )
+        {
+          Handle(HYDROData_Object) aRefGeomObj =
+            Handle(HYDROData_Object)::DownCast( anIter.Value() );
+          if ( !aRefGeomObj.IsNull() )
+          {
+            switch ( theColumnId )
+            {
+              case RefObjectId:
+                // Get Ref.Object name
+                aRes += aRefGeomObj->GetName() + ", ";
+                break;
+              case BathymetryId:
+                // Get bathymetry name
+                aRes += aRefGeomObj->GetBathymetry()->GetName();
+            }
+          }
+        }
+      }
+      if ( aRes.length() > 1 )
+      {
+        aRes.remove( aRes.length() - 2, 2 );
+      }
+    }
+    else
+    {
+      aRes = LightApp_DataObject::text( theColumnId );
+    }
+  }
+  return aRes;
+}
+
+QColor HYDROGUI_DataObject::color( const ColorRole theColorRole, const int theColumnId ) const
+{
+  //TODO: Implement red color for bathymetry conflicts in case creation dialog
+  return LightApp_DataObject::color( theColorRole, theColumnId );
+}
+
+
+QString HYDROGUI_DataObject::dataObjectEntry( const Handle(HYDROData_Entity)& theObject,
+                                              const bool theWithPrefix )
 {
   QString aEntryStr = QString::null;
   if( !theObject.IsNull() )
   {
     TCollection_AsciiString aLabEntr;
     TDF_Tool::Entry( theObject->Label(), aLabEntr );
-    aEntryStr = HYDROGUI_DataObject::entryPrefix() + QString( aLabEntr.ToCString() );
+    aEntryStr = aLabEntr.ToCString();
+    if( theWithPrefix )
+      aEntryStr.prepend( HYDROGUI_DataObject::entryPrefix() );
   }
   return aEntryStr;
 }