StationTreeView Class Reference

Detailed Description

A view for the StationTreeModel.

See also

StationTreeModel, ModelTreeItem

Definition at line 16 of file StationTreeView.h.

#include <StationTreeView.h>

Inheritance diagram for StationTreeView:

Collaboration diagram for StationTreeView:

Signals
void	enableSaveButton (bool)
void	enableRemoveButton (bool)
void	geoItemSelected (const vtkPolyDataAlgorithm *, int)
void	geometryMappingRequested (const std::string &)
void	removeGeoItemSelection ()
void	itemSelectionChanged (const QItemSelection &selected, const QItemSelection &deselected)
void	openStationListFile (int)
void	propertiesDialogRequested (std::string name)
void	requestNameChangeDialog (const std::string &, std::size_t)
void	stationListExportRequested (std::string listName, std::string fileName)
void	stationListRemoved (std::string name)
void	stationListSaved (QString listName, QString fileName)
void	diagramRequested (QModelIndex &)

Public Member Functions
	StationTreeView (QWidget *parent=nullptr)
	Constructor.
void	updateView ()
	Update the view to visualise changes made to the underlying data.

Protected Slots
void	selectionChanged (const QItemSelection &selected, const QItemSelection &deselected) override
	Instructions if the selection of items in the view has changed.
void	selectionChangedFromOutside (const QItemSelection &selected, const QItemSelection &deselected)
	Instructions if the selection of items in the view has changed by events outside the view (i.e. by actions made in the visualisation).

Private Slots
void	addStationList ()
void	displayStratigraphy ()
void	exportList ()
void	exportStation ()
void	mapStations ()
void	removeStationList ()
void	setNameForElement ()
	Calls a SetNameDialog.
void	writeToFile ()
void	showDiagramPrefsDialog ()

Private Member Functions
void	contextMenuEvent (QContextMenuEvent *e) override
	Actions to be taken after a right mouse click is performed in the station view.
void	writeStratigraphiesAsImages (QString listName)
	Create image files from all stratigraphies in a borehole vector.

Constructor & Destructor Documentation

StationTreeView()

StationTreeView::StationTreeView ( QWidget * parent = nullptr )

explicit

Constructor.

Definition at line 19 of file StationTreeView.cpp.

                                                : QTreeView(parent)
{
    //    setContextMenuPolicy(Qt::CustomContextMenu);
    //    connect(this, SIGNAL(customContextMenuRequested(const QPoint &)),
    //    this, SLOT(showContextMenu(const QPoint &)));
}

Member Function Documentation

addStationList

void StationTreeView::addStationList ( )

privateslot

Definition at line 178 of file StationTreeView.cpp.

{
    emit openStationListFile(ImportFileType::OGS_STN);
}

References ImportFileType::OGS_STN, and openStationListFile().

contextMenuEvent()

void StationTreeView::contextMenuEvent ( QContextMenuEvent * e )

overrideprivate

Actions to be taken after a right mouse click is performed in the station view.

Definition at line 81 of file StationTreeView.cpp.

{
    QModelIndex index = this->selectionModel()->currentIndex();
    auto* item = static_cast<ModelTreeItem*>(index.internalPointer());
 
    if (!item)
    {  // Otherwise sometimes it crashes when (unmotivated ;-) ) clicking in a
       // treeview
        return;
    }
 
    // The current index refers to a parent item (e.g. a listname)
    if (item->childCount() > 0)
    {
        QMenu menu;
        QAction* mapAction = menu.addAction("Map stations...");
        QAction* exportAction = menu.addAction("Export to GMS...");
        menu.addSeparator();
 
        connect(mapAction, SIGNAL(triggered()), this, SLOT(mapStations()));
        connect(exportAction, SIGNAL(triggered()), this, SLOT(exportList()));
        menu.exec(event->globalPos());
    }
    // The current index refers to a station object
    else
    {
        QString temp_name;
        QMenu menu;
 
        QAction* setNameAction = menu.addAction("Set name...");
        connect(setNameAction, SIGNAL(triggered()), this,
                SLOT(setNameForElement()));
        if (dynamic_cast<GeoLib::StationBorehole*>(
                static_cast<StationTreeModel*>(model())->stationFromIndex(
                    index, temp_name)))
        {
            QAction* stratAction = menu.addAction("Display Stratigraphy...");
            QAction* exportAction = menu.addAction("Export to GMS...");
            connect(stratAction, SIGNAL(triggered()), this,
                    SLOT(displayStratigraphy()));
            connect(exportAction, SIGNAL(triggered()), this,
                    SLOT(exportStation()));
            menu.exec(event->globalPos());
        }
        else
        {
            QAction* showDiagramAction = menu.addAction("View Diagram...");
            connect(showDiagramAction, SIGNAL(triggered()), this,
                    SLOT(showDiagramPrefsDialog()));
            menu.exec(event->globalPos());
        }
    }
}

References displayStratigraphy(), exportList(), exportStation(), mapStations(), setNameForElement(), and showDiagramPrefsDialog().

diagramRequested

void StationTreeView::diagramRequested ( QModelIndex & )

signal

Referenced by showDiagramPrefsDialog().

displayStratigraphy

void StationTreeView::displayStratigraphy ( )

privateslot

Definition at line 154 of file StationTreeView.cpp.

{
    QModelIndex index = this->selectionModel()->currentIndex();
 
    QString temp_name;
    // get list name
    static_cast<StationTreeModel*>(model())->stationFromIndex(
        this->selectionModel()->currentIndex(), temp_name);
    // get color table (horrible way to do it but there you go ...)
    std::map<std::string, DataHolderLib::Color> colorLookupTable =
        static_cast<VtkStationSource*>(
            static_cast<StationTreeModel*>(model())->vtkSource(
                temp_name.toStdString()))
            ->getColorLookupTable();
    auto* stratView = new StratWindow(
        static_cast<GeoLib::StationBorehole*>(
            static_cast<StationTreeModel*>(model())->stationFromIndex(
                index, temp_name)),
        &colorLookupTable);
    stratView->setAttribute(
        Qt::WA_DeleteOnClose);  // this fixes the memory leak shown by cppcheck
    stratView->show();
}

References VtkStationSource::getColorLookupTable(), and StationTreeModel::vtkSource().

Referenced by contextMenuEvent().

enableRemoveButton

void StationTreeView::enableRemoveButton ( bool )

signal

Referenced by removeStationList(), and selectionChanged().

enableSaveButton

void StationTreeView::enableSaveButton ( bool )

signal

Referenced by removeStationList(), and selectionChanged().

exportList

void StationTreeView::exportList ( )

privateslot

Definition at line 206 of file StationTreeView.cpp.

{
    // only a test for the stratigraphy screenshot tool and writer!!
    // QString Name =
    // static_cast<StationTreeModel*>(model())->getItem(this->selectionModel()->currentIndex())->data(0).toString();
    // writeStratigraphiesAsImages(Name);
 
    TreeItem* item = static_cast<StationTreeModel*>(model())->getItem(
        this->selectionModel()->currentIndex());
    QString listName = item->data(0).toString();
    QString fileName = QFileDialog::getSaveFileName(
        this, "Export Boreholes to GMS-Format",
        LastSavedFileDirectory::getDir() + listName, "*.txt");
    if (!fileName.isEmpty())
    {
        LastSavedFileDirectory::setDir(fileName);
        emit stationListExportRequested(listName.toStdString(),
                                        fileName.toStdString());
    }
}

References TreeItem::data(), LastSavedFileDirectory::getDir(), LastSavedFileDirectory::setDir(), and stationListExportRequested().

Referenced by contextMenuEvent().

exportStation

void StationTreeView::exportStation ( )

privateslot

Definition at line 227 of file StationTreeView.cpp.

{
    QModelIndex index = this->selectionModel()->currentIndex();
    QString fileName =
        QFileDialog::getSaveFileName(this, "Export Borehole to GMS-Format",
                                     LastSavedFileDirectory::getDir(), "*.txt");
    if (!fileName.isEmpty())
    {
        QString temp_name;
        std::vector<GeoLib::Point*> stations;
        stations.push_back(static_cast<GeoLib::StationBorehole*>(
            static_cast<StationTreeModel*>(model())->stationFromIndex(
                index, temp_name)));
        FileIO::GMSInterface::writeBoreholesToGMS(&stations,
                                                  fileName.toStdString());
        LastSavedFileDirectory::setDir(fileName);
    }
}

References LastSavedFileDirectory::getDir(), LastSavedFileDirectory::setDir(), and FileIO::GMSInterface::writeBoreholesToGMS().

Referenced by contextMenuEvent().

geoItemSelected

void StationTreeView::geoItemSelected ( const vtkPolyDataAlgorithm * , int  )

signal

Referenced by selectionChanged().

geometryMappingRequested

void StationTreeView::geometryMappingRequested ( const std::string & )

signal

Referenced by mapStations().

itemSelectionChanged

void StationTreeView::itemSelectionChanged ( const QItemSelection & selected, const QItemSelection & deselected )

signal

mapStations

void StationTreeView::mapStations ( )

privateslot

Definition at line 145 of file StationTreeView.cpp.

{
    TreeItem const* const item =
        static_cast<StationTreeModel*>(model())->getItem(
            this->selectionModel()->currentIndex());
    std::string const& geo_name(item->data(0).toString().toStdString());
    emit geometryMappingRequested(geo_name);
}

References TreeItem::data(), and geometryMappingRequested().

Referenced by contextMenuEvent().

openStationListFile

void StationTreeView::openStationListFile ( int )

signal

Referenced by addStationList().

propertiesDialogRequested

void StationTreeView::propertiesDialogRequested ( std::string name )

signal

removeGeoItemSelection

void StationTreeView::removeGeoItemSelection ( )

signal

Referenced by selectionChanged().

removeStationList

void StationTreeView::removeStationList ( )

privateslot

Definition at line 246 of file StationTreeView.cpp.

{
    QModelIndex index(this->selectionModel()->currentIndex());
    if (!index.isValid())
    {
        OGSError::box("No station list selected.");
    }
    else
    {
        TreeItem* item =
            static_cast<StationTreeModel*>(model())->getItem(index);
        emit stationListRemoved((item->data(0).toString()).toStdString());
 
        if (this->selectionModel()->selectedIndexes().count() == 0)
        {
            emit enableSaveButton(false);
            emit enableRemoveButton(false);
        }
    }
}

References OGSError::box(), TreeItem::data(), enableRemoveButton(), enableSaveButton(), and stationListRemoved().

requestNameChangeDialog

void StationTreeView::requestNameChangeDialog ( const std::string & , std::size_t  )

signal

Referenced by setNameForElement().

selectionChanged

void StationTreeView::selectionChanged ( const QItemSelection & selected, const QItemSelection & deselected )

overrideprotectedslot

Instructions if the selection of items in the view has changed.

Definition at line 35 of file StationTreeView.cpp.

{
    Q_UNUSED(deselected);
    if (!selected.isEmpty())
    {
        const QModelIndex idx = *(selected.indexes().begin());
        const TreeItem* tree_item =
            static_cast<TreeModel*>(this->model())->getItem(idx);
 
        const auto* list_item =
            dynamic_cast<const ModelTreeItem*>(tree_item->parentItem());
        if (list_item->getItem())
        {
            if (list_item)
            {
                emit geoItemSelected(list_item->getItem()->vtkSource(),
                                     tree_item->row());
            }
            emit enableRemoveButton(false);
            emit enableSaveButton(false);
        }
        else
        {
            emit removeGeoItemSelection();
            emit enableSaveButton(true);
            emit enableRemoveButton(true);
        }
    }
    // emit itemSelectionChanged(selected, deselected);
    // return QTreeView::selectionChanged(selected, deselected);
}

References enableRemoveButton(), enableSaveButton(), geoItemSelected(), TreeItem::parentItem(), removeGeoItemSelection(), and TreeItem::row().

selectionChangedFromOutside

void StationTreeView::selectionChangedFromOutside ( const QItemSelection & selected, const QItemSelection & deselected )

protectedslot

Instructions if the selection of items in the view has changed by events outside the view (i.e. by actions made in the visualisation).

Definition at line 68 of file StationTreeView.cpp.

{
    QItemSelectionModel* selModel = this->selectionModel();
 
    selModel->blockSignals(true);
    selModel->select(deselected, QItemSelectionModel::Deselect);
    selModel->select(selected, QItemSelectionModel::Select);
    selModel->blockSignals(false);
 
    return QTreeView::selectionChanged(selected, deselected);
}

setNameForElement

void StationTreeView::setNameForElement ( )

privateslot

Calls a SetNameDialog.

Definition at line 135 of file StationTreeView.cpp.

{
    TreeItem const* const item =
        static_cast<StationTreeModel*>(model())->getItem(
            this->selectionModel()->currentIndex());
    std::string const stn_vec_name =
        item->parentItem()->data(0).toString().toStdString();
    emit requestNameChangeDialog(stn_vec_name, item->row());
}

References TreeItem::data(), TreeItem::parentItem(), requestNameChangeDialog(), and TreeItem::row().

Referenced by contextMenuEvent().

showDiagramPrefsDialog

void StationTreeView::showDiagramPrefsDialog ( )

privateslot

Definition at line 267 of file StationTreeView.cpp.

{
    QModelIndex index = this->selectionModel()->currentIndex();
    emit diagramRequested(index);
}

References diagramRequested().

Referenced by contextMenuEvent().

stationListExportRequested

void StationTreeView::stationListExportRequested ( std::string listName, std::string fileName )

signal

Referenced by exportList().

stationListRemoved

void StationTreeView::stationListRemoved ( std::string name )

signal

Referenced by removeStationList().

stationListSaved

void StationTreeView::stationListSaved ( QString listName, QString fileName )

signal

Referenced by writeToFile().

updateView()

void StationTreeView::updateView

(

)

Update the view to visualise changes made to the underlying data.

Definition at line 26 of file StationTreeView.cpp.

{
    setAlternatingRowColors(true);
    setColumnWidth(0, 150);
    setColumnWidth(1, 75);
    setColumnWidth(2, 75);
    setColumnWidth(3, 75);
}

writeStratigraphiesAsImages()

void StationTreeView::writeStratigraphiesAsImages ( QString listName )

private

Create image files from all stratigraphies in a borehole vector.

Definition at line 273 of file StationTreeView.cpp.

{
    std::map<std::string, DataHolderLib::Color> colorLookupTable =
        static_cast<VtkStationSource*>(
            static_cast<StationTreeModel*>(model())->vtkSource(
                listName.toStdString()))
            ->getColorLookupTable();
    std::vector<ModelTreeItem*> lists =
        static_cast<StationTreeModel*>(model())->getLists();
    std::size_t nLists = lists.size();
    for (std::size_t i = 0; i < nLists; i++)
    {
        if (listName.compare(lists[i]->data(0).toString()) != 0)
        {
            continue;
        }
 
        std::vector<GeoLib::Point*> const& stations =
            *dynamic_cast<BaseItem*>(lists[i]->getItem())->getStations();
 
        for (auto station : stations)
        {
            auto* stratView =
                new StratWindow(static_cast<GeoLib::StationBorehole*>(station),
                                &colorLookupTable);
            stratView->setAttribute(Qt::WA_DeleteOnClose);
            stratView->show();
            stratView->stationView->saveAsImage(
                QString::fromStdString(
                    static_cast<GeoLib::StationBorehole*>(station)->getName()) +
                ".jpg");
            stratView->close();
        }
    }
}

References VtkStationSource::getColorLookupTable(), getName(), BaseItem::getStations(), and StationTreeModel::vtkSource().

writeToFile

void StationTreeView::writeToFile ( )

privateslot

Definition at line 183 of file StationTreeView.cpp.

{
    QModelIndex index(this->selectionModel()->currentIndex());
    if (!index.isValid())
    {
        OGSError::box("No station list selected.");
    }
    else
    {
        TreeItem* item =
            static_cast<StationTreeModel*>(model())->getItem(index);
        QString listName = item->data(0).toString();
        QString fileName = QFileDialog::getSaveFileName(
            this, "Save station list",
            LastSavedFileDirectory::getDir() + listName, "*.stn");
        if (!fileName.isEmpty())
        {
            LastSavedFileDirectory::setDir(fileName);
            emit stationListSaved(listName, fileName);
        }
    }
}

References OGSError::box(), TreeItem::data(), LastSavedFileDirectory::getDir(), LastSavedFileDirectory::setDir(), and stationListSaved().

---

The documentation for this class was generated from the following files:

• StationTreeView.h
• StationTreeView.cpp