#include "HYDROGUI_DataModel.h"
#include "HYDROGUI_Module.h"
#include "HYDROGUI_UpdateFlags.h"
-#include "HYDROGUI_Tool.h"
+#include "HYDROGUI_Tool2.h"
#include "HYDROGUI_ImportLandCoverMapDlg.h"
#include "HYDROGUI_Shape.h"
#include <HYDROGUI_ZLayers.h>
#define MAX_LCM_NAME_INDEX 1000
-//TODO add definitions into TS file
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();
- myAttrV.clear();
+ myAttrValue.clear();
}
void HYDROGUI_ImportLandCoverMapOp::startOperation()
{
HYDROGUI_Operation::startOperation();
- if ( !getPreviewManager() ) {
- setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>(
- module()->getApp()->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
- }
+ 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();
}
return;
}
- QString anExt = myFileName.split('.', QString::SkipEmptyParts).back();
-
- if (anExt == "shp")
- {
- startDocOperation();
- QApplication::setOverrideCursor(Qt::WaitCursor);
-
- QStringList aPolygonsList;
- myPolygonFaces.Clear();
- myImporter.Free();
+ startDocOperation();
+ QApplication::setOverrideCursor(Qt::WaitCursor);
+
+ QStringList aPolygonsList;
+ myPolygonFaces.Clear();
+ myImporter.Free();
- SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( module()->getApp()->activeStudy() );
- if ( !aStudy )
- return;
+ SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( module()->getApp()->activeStudy() );
+ if ( !aStudy )
+ return;
- erasePreview();
+ erasePreview();
- Handle(AIS_InteractiveContext) aCtx = NULL;
- int aShapeTypeOfFile = -1;
+ Handle(AIS_InteractiveContext) aCtx = NULL;
+ int aShapeTypeOfFile = -1;
- //Import polygons from SHP file as faces
- //This faces should be added to the new LCM object
+ //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);
+ int aStat = myImporter.ImportPolygons(doc(), myFileName, aPolygonsList, myPolygonFaces, aShapeTypeOfFile);
+ if (aStat == 1)
+ {
+ aPanel->setPolygonNames(aPolygonsList);
- LightApp_Application* anApp = module()->getApp();
- if ( !getPreviewManager() )
- setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>( anApp->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
- OCCViewer_ViewManager* aViewManager = getPreviewManager();
+ LightApp_Application* anApp = module()->getApp();
+ if ( !getPreviewManager() )
+ setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>( anApp->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
+ OCCViewer_ViewManager* aViewManager = getPreviewManager();
- if ( aViewManager )
+ if ( aViewManager )
+ {
+ if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
{
- if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
- {
- aCtx = aViewer->getAISContext();
- connect( aViewer, SIGNAL( selectionChanged() ), this, SLOT( onViewerSelectionChanged() ) );
- }
+ aCtx = aViewer->getAISContext();
+ connect( aViewer, SIGNAL( selectionChanged() ), this, SLOT( onViewerSelectionChanged() ) );
}
+ }
- for ( int i = 1; i <= myPolygonFaces.Length(); i++ )
+ for ( int i = 1; i <= myPolygonFaces.Length(); i++ )
+ {
+ TopoDS_Face aFace = TopoDS::Face(myPolygonFaces.Value( i ));
+ if ( aViewManager && !aCtx.IsNull() )
{
- TopoDS_Face aFace = TopoDS::Face(myPolygonFaces.Value( i ));
- 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);
- if( !aFace.IsNull() )
- aShape->setShape( aFace);
-
- myPolygonName2PrsShape.insert( "polygon_" + QString::number(i), aShape);
- }
+ HYDROGUI_Shape* aShape = new HYDROGUI_Shape( aCtx, NULL, getPreviewZLayer() );
+ //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( tr( "DEF_POLYGON_NAME" ) + "_" + QString::number(i), aShape);
}
+ }
- if ( !aCtx.IsNull() )
- {
- UpdateZLayersOfHilightPresentationsOfDisplayedObjects( aCtx, Graphic3d_ZLayerId_TopOSD );
- aCtx->UpdateCurrentViewer();
- }
- //
- QApplication::restoreOverrideCursor();
- aPanel->setFirstPageState(true);
- //
- //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
- if (DBF_Stat)
- {
- myFieldList = myImporter.DBF_GetFieldList();
- aPanel->setAttributeNames(myFieldList);
- }
- aPanel->setDbfState(DBF_Stat);
+ if ( !aCtx.IsNull() )
+ {
+ UpdateZLayersOfHilightPresentationsOfDisplayedObjects( aCtx, Graphic3d_ZLayerId_TopOSD );
+ aCtx->UpdateCurrentViewer();
}
- else
+ //
+ QApplication::restoreOverrideCursor();
+ 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)
{
- erasePreview();
- aPanel->setPolygonNames(QStringList());
- aPanel->setObjectName("");
- QApplication::restoreOverrideCursor();
- QString aMess = "Cannot import land cover;\n";
- if (aStat == -1)
- aMess += "Cannot open SHP file";
- else if (aStat == -2)
- aMess += "Cannot open SHX file";
- else
- 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);
+ myFieldList = myImporter.DBF_GetFieldList();
+ aPanel->setAttributeNames(myFieldList);
}
-
+ aPanel->setDbfState(DBF_Stat);
}
-
+ else
+ {
+ erasePreview();
+ aPanel->setPolygonNames(QStringList());
+ aPanel->setObjectName("");
+ QApplication::restoreOverrideCursor();
+ QString aMess = tr( "CANNT_IMPORT_LCM" ) + "\n";
+ if (aStat == -1)
+ aMess += tr( "CANNT_OPEN_SHP" );
+ else if (aStat == -2)
+ aMess += tr( "CANNT_OPEN_SHX" );
+ else
+ 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
QStringList aAttrV_T;
QStringList aSTL;
- aPanel->GetAttribute2StricklerCorr(aAttrV_T, aSTL);
+ aPanel->getValAttr2StricklerTypeCorr(aAttrV_T, aSTL);
QVector<int> aSelIndices = aPanel->getSelectedPolygonIndices();
- foreach ( int Ind, aSelIndices ) //check i-base.
+ foreach ( int Ind, aSelIndices )
{
TopoDS_Shape aShape = myPolygonFaces(Ind + 1);
if ( aShape.IsNull() )
QString aST = "";
if (aPanel->getAttrCheckBoxState())
{
- HYDROData_ShapeFile::DBF_AttrValue aV = myAttrV[Ind];
- int StricklerTypesInd = aAttrV_T.indexOf(QString(aV.myStrVal));
- aST = aSTL.at(StricklerTypesInd);
+ 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 );
//
myLCM->StoreLandCovers(aMapFace2ST);
- QString ObjName;
+ QString anObjName;
if ( !aPanel->getFileName().isEmpty() )
- ObjName = aPanel->getObjectName();
+ anObjName = aPanel->getObjectName();
- Handle_HYDROData_Document aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
+ 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 It( aDoc, KIND_LAND_COVER_MAP );
- for( ; It.More(); It.Next() )
- aNameList << It.Current()->GetName();
+ HYDROData_Iterator anIt( aDoc, KIND_LAND_COVER_MAP );
+ for( ; anIt.More(); anIt.Next() )
+ aNameList << anIt.Current()->GetName();
}
- QString NewName = ObjName;
- for ( int i = 1; i < MAX_LCM_NAME_INDEX && aNameList.contains(NewName); i++)
- NewName = ObjName + "_" + QString::number(i);
+ 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( NewName );
+ myLCM->SetName( aNewName );
myLCM->SetColor( Qt::gray );
myLCM->Show();
module()->setIsToUpdate( myLCM );
erasePreview();
myImporter.Free();
- myAttrV.clear();
+ myAttrValue.clear();
return true;
}
}
if ( !aCtx.IsNull() ) {
- foreach ( QString aName, myPolygonName2PrsShape.keys() ) {
+ bool bSelChanged = false;
+ foreach ( QString aName, myPolygonName2PrsShape.keys() )
+ {
Handle(AIS_InteractiveObject) anObject =
myPolygonName2PrsShape.value(aName)->getAISObject();
bool isSelected = theSelectedNames.contains( aName );
- if ( ( isSelected && !aCtx->IsSelected( anObject) ) ||
- ( !isSelected && aCtx->IsSelected( anObject) ) )
+ if ( ( isSelected && !aCtx->IsSelected( anObject) ) || ( !isSelected && aCtx->IsSelected( anObject) ) )
+ {
aCtx->AddOrRemoveSelected( anObject, Standard_False );
+ bSelChanged = true;
+ }
}
- aCtx->UpdateCurrentViewer();
+
+ HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
+ if ( !aPanel )
+ return;
+
+ if (bSelChanged && aPanel->getViewerState())
+ aCtx->UpdateCurrentViewer();
}
}
void HYDROGUI_ImportLandCoverMapOp::onViewerSelectionChanged()
{
HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
- if ( !aPanel ) {
+ if ( !aPanel )
+ return;
+
+ int aCurIndex = -1;
+ aCurIndex = aPanel->getCurrentWizardIndex();
+
+ if (aCurIndex != 0)
return;
- }
OCCViewer_ViewManager* aViewManager = getPreviewManager();
Handle(AIS_InteractiveContext) aCtx = NULL;
- if ( aViewManager ) {
- if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() ) {
+ if ( aViewManager )
+ if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
aCtx = aViewer->getAISContext();
- }
- }
if ( !aCtx.IsNull() )
{
if ( isSelected )
aSelectedNames << aName;
}
+ aPanel->setViewerState(false);
aPanel->setSelectedPolygonNames( aSelectedNames );
+ aPanel->setViewerState(true);
}
}
void HYDROGUI_ImportLandCoverMapOp::erasePreview()
{
- foreach ( HYDROGUI_Shape* aShape, myPolygonName2PrsShape ) {
+ foreach ( HYDROGUI_Shape* aShape, myPolygonName2PrsShape )
delete aShape;
- }
myPolygonName2PrsShape.clear();
}
if (theIndex == 2)
{
- //std::vector<HYDROData_ShapeFile::DBF_AttrValue> myAttrV;
- int Ind = myFieldList.indexOf(aPanel->getSelectedFieldName());
- if (Ind == -1)
+ //std::vector<HYDROData_ShapeFile::DBF_AttrValue> myAttrValue;
+ int anIndOfSelField = myFieldList.indexOf(aPanel->getSelectedFieldName());
+ if (anIndOfSelField == -1)
return;
//aPanel->setSecondPageState(true);
- myAttrV.clear();
- myImporter.DBF_GetAttributeList(Ind, myAttrV );
+ myAttrValue.clear();
+ myImporter.DBF_GetAttributeList(anIndOfSelField, myAttrValue );
mySetOfAttrValues.clear();
- for (size_t i = 0; i < myAttrV.size(); i++)
+ for (size_t i = 0; i < myAttrValue.size(); i++)
{
- HYDROData_ShapeFile::DBF_AttrValue aV = myAttrV[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_Document) aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
Handle(HYDROData_StricklerTable) DefStricklerTableObj;
if ( aDoc )
{
}
}
}
- QList<QString> SetOfAttrValuesList = mySetOfAttrValues.toList();
+
+ QList<QString> aSetOfAttrValuesList = mySetOfAttrValues.toList();
QList<QString> aSTSetList = aSTSet.toList();
- QVector<QColor> STColors;
- STColors.reserve(aSTSetList.size());
+ QVector<QColor> aSTColors;
+ aSTColors.reserve(aSTSetList.size());
foreach (QString str, aSTSetList)
{
QColor col = aDoc->GetAssociatedColor(str, NULL);
- STColors.append (col);
+ aSTColors.append (col);
}
//add an empty Strickler type
aSTSetList.prepend("");
- STColors.prepend(QColor(Qt::gray));
+ aSTColors.prepend(QColor(Qt::gray));
- QVector<int> aCurCBIndices(SetOfAttrValuesList.size());
- if (DefStricklerTableObj->GetAttrName() == aPanel->getSelectedFieldName())
- {
- for (int i = 0; i < SetOfAttrValuesList.size(); i++)
+ QVector<int> aCurCBIndices(aSetOfAttrValuesList.size());
+ if (DefStricklerTableObj->GetAttrName().trimmed() == aPanel->getSelectedFieldName().trimmed())
+ for (int i = 0; i < aSetOfAttrValuesList.size(); i++)
{
- QString ST = DefStricklerTableObj->GetType(SetOfAttrValuesList[i]);
- int IndofSt = aSTSetList.indexOf(ST);
- aCurCBIndices[i] = IndofSt;
+ QString aST = DefStricklerTableObj->GetType(aSetOfAttrValuesList[i]);
+ int anIndOfSt = aSTSetList.indexOf(aST);
+ aCurCBIndices[i] = anIndOfSt;
}
- }
else
- {
//TODO add warning ???
- for (int i = 0; i < SetOfAttrValuesList.size(); i++)
+ for (int i = 0; i < aSetOfAttrValuesList.size(); i++)
aCurCBIndices[i] = 0;
- }
- aPanel->FillCorrTable(SetOfAttrValuesList, aSTSetList, aCurCBIndices, STColors);
+ aPanel->FillCorrespondenceTable(aSetOfAttrValuesList, aSTSetList, aCurCBIndices, aSTColors);
}
}
void HYDROGUI_ImportLandCoverMapOp::onApply()
{
+ HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
+ if ( !aPanel )
+ return;
+
+ if (!aPanel->CheckFirstPageFilling())
+ return;
+
QApplication::setOverrideCursor( Qt::WaitCursor );
int anUpdateFlags = 0;
QString anErrorMsg;
}
catch (...)
{
- SUIT_MessageBox::critical( module()->getApp()->desktop(), tr( "LCM_IMPORT_ERROR" ), "Cant import choosed polygons");
+ SUIT_MessageBox::critical( module()->getApp()->desktop(), tr( "LCM_IMPORT_ERROR" ), tr ("CANNT_IMPORT_POLYGONS_FROM_SHP"));
aResult = false;
}
}
else
{
- Handle(HYDROData_Entity) LCM = Handle(HYDROData_Entity)::DownCast( myLCM );
- LCM->Remove();
+ myLCM->Remove();
module()->setObjectRemoved( myLCM );
abort();
}