Salome HOME
LCM // Import/Export of SHP p.3
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_ImportLandCoverMapOp.cxx
index 5490f57fd530b1866b24c009303ce17fa0064d15..e8303bac36a5df471c005c7fa7e9918e55f1ef38 100644 (file)
 #include <TopoDS.hxx>
 #include <TopoDS_Shape.hxx>
 #include <TopoDS_Wire.hxx>
-#include <qset.h>
+#include <QSet>
+
+#include <HYDROData_LandCoverMap.h>
+
+
+//TODO deny to create null-LCM (selected poltgons on the first page = 0)
+//TODO erase onBack/onFinish if not needed...
 
 HYDROGUI_ImportLandCoverMapOp::HYDROGUI_ImportLandCoverMapOp( HYDROGUI_Module* theModule )
 : HYDROGUI_Operation( theModule )
@@ -74,6 +80,8 @@ HYDROGUI_ImportLandCoverMapOp::HYDROGUI_ImportLandCoverMapOp( HYDROGUI_Module* t
 HYDROGUI_ImportLandCoverMapOp::~HYDROGUI_ImportLandCoverMapOp()
 {
   erasePreview();
+  myImporter.Free();
+  myAttrV.clear();
 }
 
 void HYDROGUI_ImportLandCoverMapOp::startOperation()
@@ -85,10 +93,6 @@ void HYDROGUI_ImportLandCoverMapOp::startOperation()
                        module()->getApp()->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
   }
 
-  if ( !isApplyAndClose() ) {
-    return;
-  }
-
   HYDROGUI_ImportLandCoverMapDlg* aPanel = 
     ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
   if ( !aPanel ) {
@@ -184,7 +188,7 @@ void HYDROGUI_ImportLandCoverMapOp::onFileSelected()
       }
       //
       QApplication::restoreOverrideCursor();
-      commitDocOperation();
+      aPanel->setFirstPageState(true); 
     }
     else
     {
@@ -203,6 +207,7 @@ void HYDROGUI_ImportLandCoverMapOp::onFileSelected()
       commitDocOperation();
       myImporter.Free();
       //abort();
+      aPanel->setFirstPageState(false);
     }
     
   }
@@ -228,72 +233,63 @@ bool HYDROGUI_ImportLandCoverMapOp::processApply( int& theUpdateFlags,
                                                 QString& theErrorMsg,
                                                 QStringList& theBrowseObjectsEntries )
 {
-
+  //TODO rewrite this!
   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
   if ( !aPanel ) {
     return false;
   }
-  
-  QStringList aSelectedtPolygons = aPanel->getSelectedPolygonNames();
-  aPanel->removePolygonNames( aSelectedtPolygons );
+
+  Handle(HYDROData_LandCoverMap) aLCM =  Handle(HYDROData_LandCoverMap)::DownCast( doc()->CreateObject( KIND_LAND_COVER_MAP ) );
+   
+  HYDROData_MapOfFaceToStricklerType aMapFace2ST;
+
+  /*QStringList aSelectedPolygons = aPanel->getSelectedPolygonNames();
+  aPanel->removePolygonNames( aSelectedPolygons );
     
-  if (!aSelectedtPolygons.empty())
+  foreach ( QString aName, aSelectedPolygons )
   {
-    Handle(HYDROData_LandCoverMap) aLCM =  Handle(HYDROData_LandCoverMap)::DownCast( doc()->CreateObject( KIND_LAND_COVER_MAP ) );
-    TopoDS_Shape aResShape;
-    if (aSelectedtPolygons.size() > 1) 
-    {
-      TopoDS_Compound cmp;
-      BRep_Builder BB;
-      BB.MakeCompound(cmp);
-
-      foreach ( QString aName, aSelectedtPolygons ) {
-        TopoDS_Shape aShape = myPolygonName2PrsShape.value( aName )->getTopoShape();
-        if ( aShape.IsNull() ) 
-          continue;
-        BB.Add(cmp, aShape);
-        HYDROGUI_Shape* aShapeToDelete = myPolygonName2PrsShape.take( aName );
-        delete aShapeToDelete;
-      }
-      aResShape = cmp;
-    }
-    else
-    {         
-      TopoDS_Shape aShape = myPolygonName2PrsShape.value( aSelectedtPolygons.first() )->getTopoShape();
-      if ( !aShape.IsNull() ) 
-      {
-        HYDROGUI_Shape* aShapeToDelete = myPolygonName2PrsShape.take( aSelectedtPolygons.first() );
-        delete aShapeToDelete;
-        aResShape = aShape;
-      }
-    }
-    if( !aLCM.IsNull() ) 
-    {
-      QString aLCName = aPanel->getObjectName() + "_polygon";
-      int i = 0;
-      for( ;HYDROGUI_Tool::FindObjectByName(module(), aLCName); i++)
-        aLCName = aPanel->getObjectName() + "_polygon_" + QString::number(i);
-      aLCM->SetName( aLCName );
-      //TODO add color
-      //aLCM->SetFillingColor( aLCM->DefaultFillingColor() );
-      //aLCM->SetBorderColor( aLCM->DefaultBorderColor() );
-      
-      //aLCM->SetShape(aResShape);
-      aLCM->Show();
-            
-      //erasePreview();
-      
-      module()->setIsToUpdate( aLCM );
-      
-    }
+    TopoDS_Shape aShape = myPolygonName2PrsShape.value( aName )->getTopoShape();
+    if ( aShape.IsNull() ) 
+      continue;
+    aMapFace2ST.Add( TopoDS::Face( aShape ), "TODO temp" ); //TODO get from tableW
+    HYDROGUI_Shape* aShapeToDelete = myPolygonName2PrsShape.take( aName );
+    delete aShapeToDelete;
+  }*/
+  QStringList aAttrV_T;
+  QStringList aSTL;
+  aPanel->GetAttribute2StricklerCorr(aAttrV_T, aSTL);
+
+  QVector<int> aSelIndices = aPanel->getSelectedPolygonIndices();
+  foreach ( int Ind, aSelIndices ) //check i-base.
+  {
+    TopoDS_Shape aShape = myPolygonFaces(Ind + 1);
+    if ( aShape.IsNull() ) 
+      continue;
+    HYDROData_ShapeFile::DBF_AttrValue aV = myAttrV[Ind];
+    int StricklerTypesInd = aAttrV_T.indexOf(QString(aV.myStrVal));
+      QString aST = aSTL.at(StricklerTypesInd);
+    aMapFace2ST.Add( TopoDS::Face( aShape ), aST ); //TODO get from tableW
+    //HYDROGUI_Shape* aShapeToDelete = myPolygonName2PrsShape.take( aName );
+    //delete aShapeToDelete;
   }
-  module()->update( UF_Model | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init );
 
-  if ( isApplyAndClose() )
-    erasePreview();
+  //
+  aLCM->StoreLandCovers(aMapFace2ST);
 
+  if( !aLCM.IsNull() ) 
+  {
+    aLCM->SetName( "temp_name" );
+    aLCM->SetColor( QColor(0,100,100) );
+    aLCM->Show();
+    module()->setIsToUpdate( aLCM );
+    
+  }
+  
+  module()->update( UF_Model | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init );
+
+  erasePreview();
   myImporter.Free();
+  myAttrV.clear();
 
   return true;
 }
@@ -378,11 +374,11 @@ void HYDROGUI_ImportLandCoverMapOp::abortOperation()
   }
 
   erasePreview();
-
+  myImporter.Free();
   HYDROGUI_Operation::abortOperation();
 }
 
-
+//TODO take "Use dbf attrs as ST.." into account
 void HYDROGUI_ImportLandCoverMapOp::onNext( const int theIndex )
 {  
   //TODO add acceptor
@@ -404,16 +400,20 @@ void HYDROGUI_ImportLandCoverMapOp::onNext( const int theIndex )
 
   if (theIndex == 2)
   {
-    std::vector<HYDROData_ShapeFile::DBF_AttrValue> theAttrV;
+    //std::vector<HYDROData_ShapeFile::DBF_AttrValue> myAttrV;
     int Ind = myFieldList.indexOf(aPanel->getSelectedFieldName());
     if (Ind == -1)
-      return; //TODO add acceptor here!!
-    myImporter.DBF_GetAttributeList(Ind, theAttrV ); 
+    {
+      return;
+    }
+    //aPanel->setSecondPageState(true);
+    myAttrV.clear();
+    myImporter.DBF_GetAttributeList(Ind, myAttrV ); 
 
     mySetOfAttrValues.clear();
-    for (size_t i = 0; i < theAttrV.size(); i++)
+    for (size_t i = 0; i < myAttrV.size(); i++)
     {
-      HYDROData_ShapeFile::DBF_AttrValue aV = theAttrV[i];
+      HYDROData_ShapeFile::DBF_AttrValue aV = myAttrV[i];
       mySetOfAttrValues << QString(aV.myStrVal);  //take myStrVal by now..
     }
 
@@ -443,3 +443,36 @@ void HYDROGUI_ImportLandCoverMapOp::onBack( const int theIndex )
 {
 
 }
+
+void HYDROGUI_ImportLandCoverMapOp::onApply()
+{
+
+  QApplication::setOverrideCursor( Qt::WaitCursor );
+  int anUpdateFlags = 0;
+  QString anErrorMsg;
+  QStringList aBrowseObjectsEntries;
+
+  bool aResult = false;
+  try
+  {
+    aResult = processApply( anUpdateFlags, anErrorMsg, aBrowseObjectsEntries );
+  }
+  catch (...)
+  {
+    aResult = false;
+  }
+  
+  QApplication::restoreOverrideCursor();
+
+  if ( aResult )
+  {
+    module()->update( anUpdateFlags );
+    commit();
+    browseObjects( aBrowseObjectsEntries );
+  }
+  else
+  {
+    abort();
+  }
+}
+