#include <LightApp_DataOwner.h>
#include <LightApp_Displayer.h>
#include <LightApp_SelectionMgr.h>
+#include <HYDROData_StricklerTable.h>
#include <SUIT_Desktop.h>
#include <SUIT_ViewManager.h>
#include <TopoDS.hxx>
#include <TopoDS_Shape.hxx>
#include <TopoDS_Wire.hxx>
+#include <QSet>
+#include <HYDROData_LandCoverMap.h>
-#include <BRepTools.hxx>
+//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 )
{
- setName( tr( "IMPORT_LANDCOVER" ) );
+ setName( tr( "IMPORT_LANDCOVERMAP" ) );
}
HYDROGUI_ImportLandCoverMapOp::~HYDROGUI_ImportLandCoverMapOp()
{
erasePreview();
+ myImporter.Free();
+ myAttrV.clear();
}
void HYDROGUI_ImportLandCoverMapOp::startOperation()
module()->getApp()->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
}
- if ( !isApplyAndClose() ) {
- return;
- }
-
HYDROGUI_ImportLandCoverMapDlg* aPanel =
::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
if ( !aPanel ) {
anObjectName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_LANDCOVERMAP_NAME" ) );
aPanel->setObjectName( anObjectName );
- QString aFileName = aPanel->getFileName();
- if ( aFileName.isEmpty() )
+ myFileName = aPanel->getFileName();
+ if ( myFileName.isEmpty() )
{
abort();
return;
}
- QString anExt = aFileName.split('.', QString::SkipEmptyParts).back();
+ QString anExt = myFileName.split('.', QString::SkipEmptyParts).back();
if (anExt == "shp")
{
QApplication::setOverrideCursor(Qt::WaitCursor);
QStringList aPolygonsList;
- TopTools_SequenceOfShape aFaces;
- HYDROData_ShapeFile anImporter;
+ myPolygonFaces.Clear();
+ myImporter.Free();
SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( module()->getApp()->activeStudy() );
if ( !aStudy )
Handle(AIS_InteractiveContext) aCtx = NULL;
int aShapeTypeOfFile = -1;
- int aStat = anImporter.ImportLandCovers(aFileName, aPolygonsList, aFaces, aShapeTypeOfFile);
+
+ //Import polygons from SHP file as faces
+ //This faces should be added to the new LCM object
+
+ int aStat = myImporter.ImportPolygons(myFileName, aPolygonsList, myPolygonFaces, aShapeTypeOfFile);
if (aStat == 1)
{
aPanel->setPolygonNames(aPolygonsList);
}
}
- for ( int i = 1; i <= aFaces.Length(); i++ )
+ for ( int i = 1; i <= myPolygonFaces.Length(); i++ )
{
- TopoDS_Face aFace = TopoDS::Face(aFaces.Value( i ));
+ TopoDS_Face aFace = TopoDS::Face(myPolygonFaces.Value( i ));
if ( aViewManager && !aCtx.IsNull() )
{
HYDROGUI_Shape* aShape = new HYDROGUI_Shape( aCtx, NULL, getPreviewZLayer() );
UpdateZLayersOfHilightPresentationsOfDisplayedObjects( aCtx, Graphic3d_ZLayerId_TopOSD );
aCtx->UpdateCurrentViewer();
}
-
+ //
QApplication::restoreOverrideCursor();
- commitDocOperation();
+ aPanel->setFirstPageState(true);
}
else
{
else if (aStat == -2)
aMess += "Cannot open SHX file";
else
- aMess += "The shape type of file is " + anImporter.GetShapeTypeName(aShapeTypeOfFile);
+ aMess += "The shape type of file is " + myImporter.GetShapeTypeName(aShapeTypeOfFile);
SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "IMPORT_LANDCOVER" ), aMess);
commitDocOperation();
+ myImporter.Free();
//abort();
+ aPanel->setFirstPageState(false);
}
- anImporter.Free();
-
+
}
}
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;
+ }
+
+ //
+ 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 );
- if ( isApplyAndClose() )
- erasePreview();
+ erasePreview();
+ myImporter.Free();
+ myAttrV.clear();
return true;
}
}
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
+ HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
+ if ( !aPanel )
+ return;
+
+ if (theIndex == 1)
+ {
+ //Try to load DBF-database...
+ QString theDBFFileName;
+ theDBFFileName = myFileName.simplified().replace( ".shp", ".dbf", Qt::CaseInsensitive);
+ bool DBF_Stat = myImporter.DBF_OpenDBF(theDBFFileName);
+ // TODO:
+ // add MSG BOX if stat is bad
+ myFieldList = myImporter.DBF_GetFieldList();
+ aPanel->setAttributeNames(myFieldList);
+ }
+
+ if (theIndex == 2)
+ {
+ //std::vector<HYDROData_ShapeFile::DBF_AttrValue> myAttrV;
+ int Ind = myFieldList.indexOf(aPanel->getSelectedFieldName());
+ if (Ind == -1)
+ {
+ return;
+ }
+ //aPanel->setSecondPageState(true);
+ myAttrV.clear();
+ myImporter.DBF_GetAttributeList(Ind, myAttrV );
+
+ mySetOfAttrValues.clear();
+ for (size_t i = 0; i < myAttrV.size(); i++)
+ {
+ HYDROData_ShapeFile::DBF_AttrValue aV = myAttrV[i];
+ mySetOfAttrValues << QString(aV.myStrVal); //take myStrVal by now..
+ }
+
+ //Collect all strickler_types
+ QSet<QString> aSTSet;
+ Handle_HYDROData_Document aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
+ if ( aDoc )
+ {
+ HYDROData_Iterator It( aDoc, KIND_STRICKLER_TABLE );
+ for( ; It.More(); It.Next() )
+ {
+ Handle(HYDROData_StricklerTable) aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( It.Current() );
+ if ( !aStricklerTableObj.IsNull())
+ {
+ const QStringList& aStricklerList = aStricklerTableObj->GetTypes();
+ foreach (QString aStr, aStricklerList)
+ aSTSet << aStr;
+ }
+ }
+ }
+ aPanel->FillCorrTable(mySetOfAttrValues.toList(), aSTSet.toList());
+ }
}
{
}
+
+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();
+ }
+}
+
