StationTreeModel Class Reference

Detailed Description

A model for the StationTreeView implementing a tree as a double-linked list.

A model for the StationTreeView implementing a tree as a double-linked list. In addition to a simple TreeModel each item also contains a 2D / 3D GraphicsItem for visualization.

See also

TreeModel, StationTreeView, TreeItem, ModelTreeItem

Definition at line 31 of file StationTreeModel.h.

#include <StationTreeModel.h>

Inheritance diagram for StationTreeModel:

Collaboration diagram for StationTreeModel:

Public Member Functions
	StationTreeModel (QObject *parent=nullptr)
	~StationTreeModel () override
void	addStationList (QString listName, const std::vector< GeoLib::Point * > *stations)
const std::vector< ModelTreeItem * > &	getLists ()
QModelIndex	index (int row, int column, const QModelIndex &parent=QModelIndex()) const override
void	removeStationList (QModelIndex index)
void	removeStationList (const std::string &name)
void	setNameForItem (const std::string &stn_vec_name, std::size_t id, std::string const &item_name)
GeoLib::Station *	stationFromIndex (const QModelIndex &index, QString &listName) const
vtkPolyDataAlgorithm *	vtkSource (const std::string &name) const
Public Member Functions inherited from TreeModel
	TreeModel (QObject *parent=nullptr)
	~TreeModel () override
QVariant	data (const QModelIndex &index, int role) const override
bool	setData (const QModelIndex &index, const QVariant &value, int role) override
Qt::ItemFlags	flags (const QModelIndex &index) const override
TreeItem *	getItem (const QModelIndex &index) const
QVariant	headerData (int section, Qt::Orientation orientation, int role=Qt::DisplayRole) const override
QModelIndex	index (int row, int column, const QModelIndex &parent=QModelIndex()) const override
QModelIndex	parent (const QModelIndex &index) const override
bool	removeRows (int position, int count, const QModelIndex &parent) override
int	rowCount (const QModelIndex &parent=QModelIndex()) const override
int	columnCount (const QModelIndex &parent=QModelIndex()) const override
TreeItem *	rootItem () const

Private Attributes
std::vector< ModelTreeItem * >	_lists

Additional Inherited Members
Public Slots inherited from TreeModel
void	updateData ()
Protected Attributes inherited from TreeModel
TreeItem *	_rootItem

Constructor & Destructor Documentation

StationTreeModel()

StationTreeModel::StationTreeModel ( QObject * parent = nullptr )

explicit

Constructor.

Definition at line 15 of file StationTreeModel.cpp.

                                                  : TreeModel(parent)
{
    QList<QVariant> rootData;
    delete _rootItem;
    rootData << "Station Name"
             << "x"
             << "y"
             << "z";
    _rootItem = new ModelTreeItem(rootData, nullptr, nullptr);
}

References TreeModel::TreeModel(), TreeModel::_rootItem, and TreeModel::parent().

~StationTreeModel()

StationTreeModel::~StationTreeModel ( )

overridedefault

Member Function Documentation

addStationList()

void StationTreeModel::addStationList	(	QString	listName,
		const std::vector< GeoLib::Point * > *	stations )

Inserts a subtree under _rootItem.

Parameters

listName	Name of the new subtree. If no name is given a default name is assigned.
stations	The list with stations to be added as children of that subtree

Definition at line 128 of file StationTreeModel.cpp.

{
    beginResetModel();
 
    QList<QVariant> grpName;
    if (listName.compare("") ==
        0)  // if no name is given a default name is assigned
    {
        listName = "List";
        listName.append(QString::number(rowCount() + 1));
    }
    grpName << listName << ""
            << ""
            << "";
    auto* group =
        new ModelTreeItem(grpName, _rootItem, new BaseItem(listName, stations));
    _lists.push_back(group);
    _rootItem->appendChild(group);
    int vectorSize = stations->size();
 
    for (int i = 0; i < vectorSize; i++)
    {
        QList<QVariant> stn;
        stn << QString::fromStdString(
                   static_cast<GeoLib::Station*>((*stations)[i])->getName())
            << QString::number((*(*stations)[i])[0], 'f')
            << QString::number((*(*stations)[i])[1], 'f')
            << QString::number((*(*stations)[i])[2], 'f');
 
        auto* child = new ModelTreeItem(stn, group);
        child->setStation(static_cast<GeoLib::Station*>((*stations)[i]));
        group->appendChild(child);
    }
 
    qDebug() << "List" << listName << "loaded, " << stations->size()
             << "items added.";
 
    endResetModel();
}

References _lists, TreeModel::_rootItem, getName(), and TreeModel::rowCount().

getLists()

const std::vector< ModelTreeItem * > & StationTreeModel::getLists ( )

inline

Definition at line 40 of file StationTreeModel.h.

{ return _lists; }

References _lists.

index()

QModelIndex StationTreeModel::index ( int row, int column, const QModelIndex & parent = QModelIndex() ) const

override

Returns the model index of an item in the tree.

Parameters

row	The row where the item is located
column	The column where the item is located
parent	The parent of the item

Returns

The model index of the item

Definition at line 35 of file StationTreeModel.cpp.

{
    if (!hasIndex(row, column, parent))
    {
        return QModelIndex();
    }
 
    ModelTreeItem* parentItem;
 
    if (!parent.isValid())
    {
        parentItem = static_cast<ModelTreeItem*>(_rootItem);
    }
    else
    {
        parentItem = static_cast<ModelTreeItem*>(parent.internalPointer());
    }
 
    auto* childItem = static_cast<ModelTreeItem*>(parentItem->child(row));
    if (childItem)
    {
        QModelIndex newIndex = createIndex(row, column, childItem);
        // assign ModelIndex to BaseItem so it can communicate with the model
        BaseItem* item = childItem->getItem();
        if (item != nullptr)
        {
            item->setModelIndex(newIndex);
        }
        return newIndex;
    }
 
    return QModelIndex();
}

References TreeModel::_rootItem, TreeItem::child(), TreeModel::parent(), and BaseItem::setModelIndex().

Referenced by removeStationList(), and stationFromIndex().

removeStationList() [1/2]

void StationTreeModel::removeStationList

(

const std::string &

name

)

Removes the TreeItem with the given name including all its children

Definition at line 195 of file StationTreeModel.cpp.

{
    for (auto& list : _lists)
    {
        if (name == list->data(0).toString().toStdString())
        {
            removeStationList(createIndex(list->row(), 0, list));
        }
    }
}

References _lists, and removeStationList().

removeStationList() [2/2]

void StationTreeModel::removeStationList

(

QModelIndex

index

)

Removes the TreeItem with the given Index including all its children

Definition at line 172 of file StationTreeModel.cpp.

{
    if (index.isValid())  //
    {
        auto* item = static_cast<ModelTreeItem*>(getItem(index));
 
        // also delete the lists entry in the list directory of the model
        for (std::size_t i = 0; i < _lists.size(); i++)
        {
            if (item == _lists[i])
            {
                _lists.erase(_lists.begin() + i);
            }
        }
 
        removeRows(0, item->childCount(), index);
        removeRows(item->row(), 1, parent(index));
    }
}

References _lists, TreeModel::getItem(), index(), TreeModel::parent(), and TreeModel::removeRows().

Referenced by removeStationList().

setNameForItem()

void StationTreeModel::setNameForItem	(	const std::string &	stn_vec_name,
		std::size_t	id,
		std::string const &	item_name )

Definition at line 103 of file StationTreeModel.cpp.

{
    auto const stn_list = find_if(
        _lists.begin(),
        _lists.end(),
        [&stn_vec_name](ModelTreeItem* item)
        { return (stn_vec_name == item->data(0).toString().toStdString()); });
 
    if (stn_list == _lists.end() ||
        id >= static_cast<std::size_t>((*stn_list)->childCount()))
    {
        return;
    }
    TreeItem* const item = (*stn_list)->child(id);
    item->setData(0, QString::fromStdString(item_name));
}

References _lists, TreeItem::child(), and TreeItem::setData().

Referenced by MainWindow::showStationNameDialog().

stationFromIndex()

GeoLib::Station * StationTreeModel::stationFromIndex	(	const QModelIndex &	index,
		QString &	listName ) const

Returns the Station-Object of the ModelTreeItem with the given index and the name of the list this station belongs to.

Parameters

index	Index of the requested item
listName	Here, the method will put the name of the list this station belongs to.

Returns

The station object associated with the tree item

Definition at line 76 of file StationTreeModel.cpp.

{
    if (index.isValid())
    {
        auto* treeItem = static_cast<ModelTreeItem*>(index.internalPointer());
        TreeItem* parentItem = treeItem->parentItem();
        listName = parentItem->data(0).toString();
        return treeItem->getStation();
    }
 
    return nullptr;
}

References TreeItem::data(), index(), and TreeItem::parentItem().

vtkSource()

vtkPolyDataAlgorithm * StationTreeModel::vtkSource

(

const std::string &

name

)

const

Definition at line 90 of file StationTreeModel.cpp.

{
    std::size_t nLists = _lists.size();
    for (std::size_t i = 0; i < nLists; i++)
    {
        if (name == _lists[i]->data(0).toString().toStdString())
        {
            return dynamic_cast<BaseItem*>(_lists[i]->getItem())->vtkSource();
        }
    }
    return nullptr;
}

References _lists, TreeModel::data(), and BaseItem::vtkSource().

Referenced by VtkVisPipeline::addPipelineItem(), StationTreeView::displayStratigraphy(), VtkVisPipeline::removeSourceItem(), and StationTreeView::writeStratigraphiesAsImages().

Member Data Documentation

_lists

std::vector<ModelTreeItem*> StationTreeModel::_lists

private

Definition at line 52 of file StationTreeModel.h.

Referenced by addStationList(), getLists(), removeStationList(), removeStationList(), setNameForItem(), and vtkSource().

---

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

• StationTreeModel.h
• StationTreeModel.cpp