src/HYDROGUI/HYDROGUI_ImportPolylineOp.cxx

   1 // Copyright (C) 2014-2015  EDF-R&D
   2 // This library is free software; you can redistribute it and/or
   3 // modify it under the terms of the GNU Lesser General Public
   4 // License as published by the Free Software Foundation; either
   5 // version 2.1 of the License, or (at your option) any later version.
   6 //
   7 // This library is distributed in the hope that it will be useful,
   8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
   9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  10 // Lesser General Public License for more details.
  11 //
  12 // You should have received a copy of the GNU Lesser General Public
  13 // License along with this library; if not, write to the Free Software
  14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  15 //
  16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  17 //
  18
  19 #include "HYDROGUI_ImportPolylineOp.h"
  20
  21 #include "HYDROGUI_DataModel.h"
  22 #include "HYDROGUI_Module.h"
  23 #include "HYDROGUI_UpdateFlags.h"
  24 #include "HYDROGUI_Tool2.h"
  25 #include <HYDROData_PolylineXY.h>
  26 #include <HYDROData_Polyline3D.h>
  27 #include <HYDROGUI_DataObject.h>
  28 #include <HYDROData_Bathymetry.h>
  29 #include <HYDROData_Iterator.h>
  30 #include <HYDROData_ShapeFile.h>
  31
  32 #include <HYDROData_Profile.h>
  33
  34 #include <SUIT_Desktop.h>
  35 #include <SUIT_FileDlg.h>
  36 #include <LightApp_Application.h>
  37
  38 #include <QApplication>
  39 #include <QFile>
  40 #include <QFileInfo>
  41 #include <SUIT_MessageBox.h>
  42
  43
  44 HYDROGUI_ImportPolylineOp::HYDROGUI_ImportPolylineOp( HYDROGUI_Module* theModule )
  45 : HYDROGUI_Operation( theModule )
  46 {
  47   setName( tr( "IMPORT_POLYLINE" ) );
  48 }
  49
  50 HYDROGUI_ImportPolylineOp::~HYDROGUI_ImportPolylineOp()
  51 {
  52 }
  53
  54 void HYDROGUI_ImportPolylineOp::startOperation()
  55 {
  56   HYDROGUI_Operation::startOperation();
  57
  58   myFileDlg = new SUIT_FileDlg( module()->getApp()->desktop(), true );
  59   myFileDlg->setWindowTitle( getName() );
  60   myFileDlg->setFileMode( SUIT_FileDlg::ExistingFiles );
  61   myFileDlg->setFilter( tr("POLYLINE_FILTER") );
  62
  63   connect( myFileDlg, SIGNAL( accepted() ), this, SLOT( onApply() ) );
  64   connect( myFileDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
  65
  66   myFileDlg->exec();
  67 }
  68
  69 void HYDROGUI_ImportPolylineOp::onApply()
  70 {
  71   if ( !myFileDlg )
  72   {
  73     abort();
  74     return;
  75   }
  76
  77   QStringList aFileNames = myFileDlg->selectedFiles();
  78
  79   QApplication::setOverrideCursor( Qt::WaitCursor );
  80   startDocOperation();
  81
  82   foreach (QString aFileName, aFileNames)
  83   {
  84     if ( aFileName.isEmpty() )
  85       continue;
  86
  87     QString anExt = aFileName.split('.', QString::SkipEmptyParts).back();
  88
  89     if (anExt == "shp")
  90     {
  91       HYDROData_ShapeFile anImporter;
  92       NCollection_Sequence<Handle_HYDROData_Entity> theEntities;
  93       int aShapeTypeOfFile = -1;
  94       int aStat = anImporter.ImportPolylines(doc(), aFileName, theEntities, aShapeTypeOfFile );
  95       if (aStat == 1)
  96         UpdateView(theEntities);
  97       else if (aStat == 2)
  98       {
  99         UpdateView(theEntities);
 100         SUIT_MessageBox::information(module()->getApp()->desktop(),
 101           tr( "IMPORT_POLYLINE" ), "Contour of the polygon(s) have been imported as polyline(s)");
 102       }
 103       else
 104       {
 105         QString aMess = "Cannot import content of this file as polyline;\n";
 106         if (aStat == -1)
 107           aMess += "Cannot open SHP file";
 108         else if (aStat == -2)
 109           aMess += "Cannot open SHX file";
 110         else
 111           aMess += "The shape type of file is " + anImporter.GetShapeTypeName(aShapeTypeOfFile);
 112         SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "IMPORT_POLYLINE" ), aMess);
 113       }
 114     }
 115   }
 116   if (!aFileNames.empty())
 117   {
 118     commitDocOperation();
 119     commit();
 120     module()->update( UF_Model | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init );
 121   }
 122   else
 123     abort();
 124
 125   QApplication::restoreOverrideCursor();
 126 }
 127
 128 void HYDROGUI_ImportPolylineOp::UpdateView( NCollection_Sequence<Handle_HYDROData_Entity>& anEntities)
 129 {
 130   size_t anActiveViewId = HYDROGUI_Tool::GetActiveGraphicsViewId( module() );
 131   if ( anActiveViewId == 0 )
 132     anActiveViewId = HYDROGUI_Tool::GetActiveOCCViewId( module() );
 133
 134   for (int i = 1; i <= anEntities.Size() ; i++)
 135   {
 136     anEntities(i)->Update();
 137     module()->setObjectVisible( anActiveViewId, anEntities(i), true );
 138     module()->setIsToUpdate( anEntities(i) );
 139   }
 140 }