SHPImportDialog.cpp

Go to the documentation of this file.

 
#include "SHPImportDialog.h"
 
#include <QDialogButtonBox>
#include <QFileInfo>
#include <QLabel>
#include <QLineEdit>
#include <QRadioButton>
#include <QVBoxLayout>
 
#include "Applications/FileIO/SHPInterface.h"
#include "Base/OGSError.h"
#include "GeoLib/GEOObjects.h"
 
SHPImportDialog::SHPImportDialog(std::string filename,
                                 GeoLib::GEOObjects& geo_objects,
                                 std::string const& gmsh_path,
                                 QDialog* parent)
    : QDialog(parent),
      _buttonBox(nullptr),
      _layout(nullptr),
      _shpContentLabel(nullptr),
      _nameLabel(nullptr),
      _listName(new QLineEdit()),
      _choice1(nullptr),
      _choice2(nullptr),
      _filename(std::move(filename)),
      _fileType(0),
      _shpInterface(new FileIO::SHPInterface(geo_objects)),
      _gmsh_path(gmsh_path)
{
    setupDialog();
    show();
}
 
SHPImportDialog::~SHPImportDialog()
{
    delete _shpInterface;
    delete _buttonBox;
    delete _listName;
    delete _choice1;
    delete _choice2;
    delete _shpContentLabel;
    delete _nameLabel;
    delete _layout;
}
 
void SHPImportDialog::setupDialog()
{
    _layout = new QGridLayout(this);
    int shpType = 0;
    int numberOfEntities = 0;
    QString type = "";
 
    setWindowTitle("Import SHP File");
 
    if (_shpInterface->readSHPInfo(_filename, shpType, numberOfEntities))
    {
        if ((shpType - 1) % 10 == 0)
        {
            type = "points";
        }
        if ((shpType - 3) % 10 == 0)
        {
            type = "polylines";
        }
        if ((shpType - 5) % 10 == 0)
        {
            type = "polygons";
        }
        if ((shpType - 8) % 10 == 0)
        {
            type = "multipoints";
        }
        if (shpType == 31)
        {
            type = "TIN elements";
        }
 
        _shpContentLabel =
            new QLabel("The selected file contains " +
                           QString::number(numberOfEntities) + " " + type,
                       this);
        _nameLabel = new QLabel("List Name: ", this);
 
        QFileInfo fi(QString::fromStdString(_filename));
        _listName->setText(fi.baseName());
 
        if ((shpType - 1) % 10 == 0 && shpType != 31)  // Points
        {
            _choice1 = new QRadioButton("Read as Geometry Points");
            _choice2 = new QRadioButton("Read as Station Points");
            _choice1->toggle();  // default choice
            _layout->addWidget(_shpContentLabel);
            _layout->addWidget(_choice1);
            _layout->addWidget(_choice2);
            _layout->addWidget(_nameLabel);
            _layout->addWidget(_listName);
            _fileType = 1;
        }
        else if ((shpType - 3) % 10 == 0 ||
                 (shpType - 5) % 10 == 0)  // Polylines
        {
            _choice1 = new QRadioButton("Read Polylines only");
            _choice2 = new QRadioButton("Read Polylines/Surfaces");
            if ((shpType - 3) % 10 == 0)
            {
                _choice2->setDisabled(true);  // disable polygon-choice if file
                                              // contains only polylines
            }
            _choice1->toggle();  // default choice
            _layout->addWidget(_shpContentLabel);
            _layout->addWidget(_choice1);
            _layout->addWidget(_choice2);
            _layout->addWidget(_nameLabel);
            _layout->addWidget(_listName);
            _fileType = 2;
        }
        else
        {
            _nameLabel->setText(
                "This element type is currently not supported.");
            _layout->addWidget(_shpContentLabel);
            _layout->addWidget(_nameLabel);
        }
 
        _buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok |
                                          QDialogButtonBox::Cancel);
        connect(_buttonBox, SIGNAL(accepted()), this, SLOT(accept()));
        connect(_buttonBox, SIGNAL(rejected()), this, SLOT(reject()));
        _layout->addWidget(_buttonBox);
 
        setLayout(_layout);
    }
    else
    {
        OGSError::box("Error reading shapefile!");
    }
}
 
void SHPImportDialog::accept()
{
    QString list_name(_listName->text());
    if (list_name.compare("") == 0)
    {
        OGSError::box("Please insert a name for the data in this file.");
        return;
    }
 
    if (_fileType == 1 && _choice1->isChecked())
    {
        _shpInterface->readSHPFile(_filename,
                                   FileIO::SHPInterface::OGSType::POINT,
                                   list_name.toStdString(), _gmsh_path);
    }
    if (_fileType == 1 && _choice2->isChecked())
    {
        _shpInterface->readSHPFile(_filename,
                                   FileIO::SHPInterface::OGSType::STATION,
                                   list_name.toStdString(), _gmsh_path);
    }
    if (_fileType == 2 && _choice1->isChecked())
    {
        _shpInterface->readSHPFile(_filename,
                                   FileIO::SHPInterface::OGSType::POLYLINE,
                                   list_name.toStdString(), _gmsh_path);
    }
    if (_fileType == 2 && _choice2->isChecked())
    {
        _shpInterface->readSHPFile(_filename,
                                   FileIO::SHPInterface::OGSType::POLYGON,
                                   list_name.toStdString(), _gmsh_path);
    }
    emit shpLoaded(list_name);
 
    this->done(QDialog::Accepted);
}
 
void SHPImportDialog::reject()
{
    this->done(QDialog::Rejected);
}