#include "HYDROGUI_Tool.h"
#include "HYDROGUI_ImportLandCoverMapDlg.h"
#include "HYDROGUI_Shape.h"
-#include <HYDROData_LandCoverMap.h>
#include <HYDROGUI_ZLayers.h>
-
#include <HYDROGUI_DataObject.h>
#include <HYDROData_Iterator.h>
#include <HYDROData_ShapeFile.h>
#include <HYDROData_Profile.h>
+#include <HYDROData_LandCoverMap.h>
+#include <HYDROData_StricklerTable.h>
#include <SUIT_Desktop.h>
#include <SUIT_FileDlg.h>
-#include <LightApp_Application.h>
-
-#include <QApplication>
-#include <QFile>
-#include <QFileInfo>
#include <SUIT_MessageBox.h>
+#include <SUIT_Desktop.h>
+#include <SUIT_ViewManager.h>
#include <OCCViewer_ViewManager.h>
#include <OCCViewer_ViewModel.h>
#include <SalomeApp_Study.h>
-
#include <LightApp_Application.h>
#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 <BRep_Builder.hxx>
#include <TopoDS.hxx>
#include <TopoDS_Shape.hxx>
#include <TopoDS_Wire.hxx>
-#include <qset.h>
+
+#include <QSet>
+#include <QFile>
+#include <QFileInfo>
+
+#include <QApplication>
+
+#define MAX_LCM_NAME_INDEX 1000
HYDROGUI_ImportLandCoverMapOp::HYDROGUI_ImportLandCoverMapOp( HYDROGUI_Module* theModule )
: HYDROGUI_Operation( theModule )
{
- setName( tr( "IMPORT_LANDCOVERMAP" ) );
+ setName( tr( "IMPORT_LANDCOVER_MAP" ) );
}
HYDROGUI_ImportLandCoverMapOp::~HYDROGUI_ImportLandCoverMapOp()
{
erasePreview();
+ myImporter.Free();
+ myAttrValue.clear();
}
void HYDROGUI_ImportLandCoverMapOp::startOperation()
{
HYDROGUI_Operation::startOperation();
- if ( !getPreviewManager() ) {
- setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>(
- module()->getApp()->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
- }
-
- if ( !isApplyAndClose() ) {
- return;
- }
+ if ( !getPreviewManager() )
+ setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>( module()->getApp()->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
- HYDROGUI_ImportLandCoverMapDlg* aPanel =
- ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
- if ( !aPanel ) {
+ HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
+ if ( !aPanel )
return;
- }
aPanel->reset();
}
if ( aViewManager && !aCtx.IsNull() )
{
HYDROGUI_Shape* aShape = new HYDROGUI_Shape( aCtx, NULL, getPreviewZLayer() );
- //Green color for now..
- aShape->setFillingColor(QColor(0,255,0), false, false);
- aShape->setBorderColor(QColor(0,255,0), false, false);
+ //Preview color is grey
+ aShape->setFillingColor(QColor(50,50,50), false, false);
+ aShape->setBorderColor(QColor(50,50,50), false, false);
if( !aFace.IsNull() )
aShape->setShape( aFace);
- myPolygonName2PrsShape.insert( "polygon_" + QString::number(i), aShape);
+ myPolygonName2PrsShape.insert( tr( "DEF_POLYGON_NAME" ) + "_" + QString::number(i), aShape);
}
}
+
if ( !aCtx.IsNull() )
{
UpdateZLayersOfHilightPresentationsOfDisplayedObjects( aCtx, Graphic3d_ZLayerId_TopOSD );
}
//
QApplication::restoreOverrideCursor();
- commitDocOperation();
+ aPanel->setFirstPageState(true);
+ //
+ //Try to load DBF-database...
+ QString aDBFFileName;
+ aDBFFileName = myFileName.simplified().replace( myFileName.simplified().size() - 4, 4, ".dbf");
+ bool DBF_Stat = myImporter.DBF_OpenDBF(aDBFFileName);
+ // TODO:
+ // add MSG BOX if stat is bad
+ if (DBF_Stat)
+ {
+ myFieldList = myImporter.DBF_GetFieldList();
+ aPanel->setAttributeNames(myFieldList);
+ }
+ aPanel->setDbfState(DBF_Stat);
}
else
{
aPanel->setPolygonNames(QStringList());
aPanel->setObjectName("");
QApplication::restoreOverrideCursor();
- QString aMess = "Cannot import land cover;\n";
+ QString aMess = tr( "CANNT_IMPORT_LCM" ) + "\n";
if (aStat == -1)
- aMess += "Cannot open SHP file";
+ aMess += tr( "CANNT_OPEN_SHP" );
else if (aStat == -2)
- aMess += "Cannot open SHX file";
+ aMess += tr( "CANNT_OPEN_SHX" );
else
- aMess += "The shape type of file is " + myImporter.GetShapeTypeName(aShapeTypeOfFile);
- SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "IMPORT_LANDCOVER" ), aMess);
+ aMess += tr ("SHP_TYPEFORMAT_MESS") + myImporter.GetShapeTypeName(aShapeTypeOfFile);
+ SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "IMPORT_LANDCOVER_MAP" ), aMess);
commitDocOperation();
myImporter.Free();
//abort();
+ aPanel->setFirstPageState(false);
}
-
}
-
}
HYDROGUI_InputPanel* HYDROGUI_ImportLandCoverMapOp::createInputPanel() const
HYDROGUI_InputPanel* aPanel = new HYDROGUI_ImportLandCoverMapDlg( module(), getName() );
connect( aPanel, SIGNAL( FileSelected( const QString& ) ), SLOT( onFileSelected() ) );
-
connect( aPanel, SIGNAL( selectionChanged( const QStringList& ) ), this, SLOT( onSelectionChanged( const QStringList& ) ) );
-
connect( aPanel, SIGNAL( Next( const int ) ), SLOT( onNext( const int ) ) );
- connect( aPanel, SIGNAL( Back( const int ) ), SLOT( onBack( const int ) ) );
- connect( aPanel, SIGNAL( Finish( const int ) ), SLOT( onFinish( const int ) ) );
return aPanel;
}
bool HYDROGUI_ImportLandCoverMapOp::processApply( int& theUpdateFlags,
- QString& theErrorMsg,
- QStringList& theBrowseObjectsEntries )
+ QString& theErrorMsg,
+ QStringList& theBrowseObjectsEntries )
{
HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
- if ( !aPanel ) {
+ if ( !aPanel )
return false;
- }
-
- QStringList aSelectedtPolygons = aPanel->getSelectedPolygonNames();
- aPanel->removePolygonNames( aSelectedtPolygons );
-
- if (!aSelectedtPolygons.empty())
+
+ myLCM = Handle(HYDROData_LandCoverMap)::DownCast( doc()->CreateObject( KIND_LAND_COVER_MAP ) );
+ HYDROData_MapOfFaceToStricklerType aMapFace2ST;
+
+ QStringList aAttrV_T;
+ QStringList aSTL;
+ aPanel->getValAttr2StricklerTypeCorr(aAttrV_T, aSTL);
+
+ QVector<int> aSelIndices = aPanel->getSelectedPolygonIndices();
+ foreach ( int Ind, aSelIndices )
{
- 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() )
+ TopoDS_Shape aShape = myPolygonFaces(Ind + 1);
+ if ( aShape.IsNull() )
+ continue;
+ QString aST = "";
+ if (aPanel->getAttrCheckBoxState())
{
- 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 );
-
+ HYDROData_ShapeFile::DBF_AttrValue aDataVal = myAttrValue[Ind];
+ int aStricklerTypesInd = aAttrV_T.indexOf(QString(aDataVal.myStrVal));
+ aST = aSTL.at(aStricklerTypesInd);
}
+ // else => ST is empty
+ aMapFace2ST.Add( TopoDS::Face( aShape ), aST );
}
-
- module()->update( UF_Model | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init );
- if ( isApplyAndClose() )
- erasePreview();
+ //
+ myLCM->StoreLandCovers(aMapFace2ST);
+
+ QString anObjName;
+ if ( !aPanel->getFileName().isEmpty() )
+ anObjName = aPanel->getObjectName();
+
+ Handle_HYDROData_Document aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
+
+ //check if name of LCM is already exists
+ QSet<QString> aNameList;
+ if (aDoc)
+ {
+ HYDROData_Iterator anIt( aDoc, KIND_LAND_COVER_MAP );
+ for( ; anIt.More(); anIt.Next() )
+ aNameList << anIt.Current()->GetName();
+ }
+
+ QString aNewName = anObjName;
+ for ( int i = 1; i < MAX_LCM_NAME_INDEX && aNameList.contains(aNewName); i++)
+ aNewName = anObjName + "_" + QString::number(i);
+
+ if( !myLCM.IsNull() )
+ {
+ myLCM->SetName( aNewName );
+ myLCM->SetColor( Qt::gray );
+ myLCM->Show();
+ module()->setIsToUpdate( myLCM );
+ }
+
+ module()->update( UF_Model | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init );
+ erasePreview();
myImporter.Free();
+ myAttrValue.clear();
return true;
}
bool isSelected = theSelectedNames.contains( aName );
if ( ( isSelected && !aCtx->IsSelected( anObject) ) ||
- ( !isSelected && aCtx->IsSelected( anObject) ) ) {
+ ( !isSelected && aCtx->IsSelected( anObject) ) )
aCtx->AddOrRemoveSelected( anObject, Standard_False );
- }
- /*if (isSelected)
- {
- HYDROGUI_Shape* aHydroSh = myPolygonName2PrsShape.value( aName );
- aHydroSh->setFillingColor(QColor(0,255,255), true, true);
- }*/
}
aCtx->UpdateCurrentViewer();
}
void HYDROGUI_ImportLandCoverMapOp::onViewerSelectionChanged()
{
- // Get panel
HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
- if ( !aPanel ) {
+ if ( !aPanel )
return;
- }
OCCViewer_ViewManager* aViewManager = getPreviewManager();
Handle(AIS_InteractiveContext) aCtx = NULL;
QStringList aSelectedNames;
foreach ( QString aName, myPolygonName2PrsShape.keys() ) {
bool isSelected = aCtx->IsSelected( myPolygonName2PrsShape.value(aName)->getAISObject() );
- if ( isSelected ) {
+ if ( isSelected )
aSelectedNames << aName;
- }
}
aPanel->setSelectedPolygonNames( aSelectedNames );
}
void HYDROGUI_ImportLandCoverMapOp::erasePreview()
{
- foreach ( HYDROGUI_Shape* aShape, myPolygonName2PrsShape ) {
+ foreach ( HYDROGUI_Shape* aShape, myPolygonName2PrsShape )
delete aShape;
- }
myPolygonName2PrsShape.clear();
}
void HYDROGUI_ImportLandCoverMapOp::abortOperation()
{
LightApp_Application* anApp = module()->getApp();
- if ( anApp ) {
+ if ( anApp )
anApp->disconnect( this );
- }
erasePreview();
-
+ myImporter.Free();
HYDROGUI_Operation::abortOperation();
}
-
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> theAttrV;
- int Ind = myFieldList.indexOf(aPanel->getSelectedFieldName());
- if (Ind == -1)
- return; //TODO add acceptor here!!
- myImporter.DBF_GetAttributeList(Ind, theAttrV );
+ //std::vector<HYDROData_ShapeFile::DBF_AttrValue> myAttrValue;
+ int anIndOfSelField = myFieldList.indexOf(aPanel->getSelectedFieldName());
+ if (anIndOfSelField == -1)
+ return;
+ //aPanel->setSecondPageState(true);
+ myAttrValue.clear();
+ myImporter.DBF_GetAttributeList(anIndOfSelField, myAttrValue );
mySetOfAttrValues.clear();
- for (size_t i = 0; i < theAttrV.size(); i++)
+ for (size_t i = 0; i < myAttrValue.size(); i++)
{
- HYDROData_ShapeFile::DBF_AttrValue aV = theAttrV[i];
- mySetOfAttrValues << QString(aV.myStrVal); //take myStrVal by now..
+ HYDROData_ShapeFile::DBF_AttrValue aV = myAttrValue[i];
+ mySetOfAttrValues << QString(aV.myStrVal); //take myStrVal for now..
}
//Collect all strickler_types
QSet<QString> aSTSet;
Handle_HYDROData_Document aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
+ Handle(HYDROData_StricklerTable) DefStricklerTableObj;
if ( aDoc )
{
HYDROData_Iterator It( aDoc, KIND_STRICKLER_TABLE );
for( ; It.More(); It.Next() )
{
- Handle(HYDROData_StricklerTable) aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( It.Current() );
+ Handle(HYDROData_StricklerTable) aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( It.Current() );
+ if (DefStricklerTableObj.IsNull())
+ DefStricklerTableObj = aStricklerTableObj;
if ( !aStricklerTableObj.IsNull())
{
const QStringList& aStricklerList = aStricklerTableObj->GetTypes();
}
}
}
- aPanel->FillCorrTable(mySetOfAttrValues.toList(), aSTSet.toList());
+
+ QList<QString> aSetOfAttrValuesList = mySetOfAttrValues.toList();
+ QList<QString> aSTSetList = aSTSet.toList();
+ QVector<QColor> aSTColors;
+ aSTColors.reserve(aSTSetList.size());
+ foreach (QString str, aSTSetList)
+ {
+ QColor col = aDoc->GetAssociatedColor(str, NULL);
+ aSTColors.append (col);
+ }
+
+ //add an empty Strickler type
+ aSTSetList.prepend("");
+ aSTColors.prepend(QColor(Qt::gray));
+
+ QVector<int> aCurCBIndices(aSetOfAttrValuesList.size());
+ if (DefStricklerTableObj->GetAttrName() == aPanel->getSelectedFieldName())
+ for (int i = 0; i < aSetOfAttrValuesList.size(); i++)
+ {
+ QString aST = DefStricklerTableObj->GetType(aSetOfAttrValuesList[i]);
+ int anIndOfSt = aSTSetList.indexOf(aST);
+ aCurCBIndices[i] = anIndOfSt;
+ }
+ else
+ //TODO add warning ???
+ for (int i = 0; i < aSetOfAttrValuesList.size(); i++)
+ aCurCBIndices[i] = 0;
+
+ aPanel->FillCorrespondenceTable(aSetOfAttrValuesList, aSTSetList, aCurCBIndices, aSTColors);
}
}
-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 (...)
+ {
+ SUIT_MessageBox::critical( module()->getApp()->desktop(), tr( "LCM_IMPORT_ERROR" ), tr ("CANNT_IMPORT_POLYGONS_FROM_SHP"));
+ aResult = false;
+ }
+
+ QApplication::restoreOverrideCursor();
+ if ( aResult )
+ {
+ module()->update( anUpdateFlags );
+ commit();
+ browseObjects( aBrowseObjectsEntries );
+ }
+ else
+ {
+ Handle(HYDROData_Entity) LCM = Handle(HYDROData_Entity)::DownCast( myLCM );
+ LCM->Remove();
+ module()->setObjectRemoved( myLCM );
+ abort();
+ }
}
+
