treeview.h

#ifndef GUI_TREEVIEW
#define GUI_TREEVIEW


// Necessary Qt-imports
#include <QWidget>
#include <QObject>
#include <QTreeView>
#include <QTabWidget>
#include <QAction>
#include <QMenu>
#include <QKeyEvent>
#include <QStandardItem>
#include <QSortFilterProxyModel>

// The UI-component itself
#include "ui_treeview.h"

// The graph model used for the tree view
#include "graph_model.h"

class TreeView : public QWidget, private Ui::TreeView {
  Q_OBJECT

  public:

    TreeView(QWidget* parent = 0);

    ~TreeView();

    void setModel(
      QSortFilterProxyModel* new_model_dependents,
      QSortFilterProxyModel* new_model_depenencies,
      QSortFilterProxyModel* new_model_both,
      std::string root_name);

    QSortFilterProxyModel* model_dependencies();

    QSortFilterProxyModel* model_dependents();

    QSortFilterProxyModel* model_both();

    void select_row_in_treeview(std::string item, bool last);

    void deselect_all_rows_in_treeview();

    std::vector<QTreeView*> tree_views();

    int column_index(QString str);

    void resize_columns_to_fit();

  private:

    std::vector<QTreeView*> treeViews;

    QAbstractItemView::SelectionMode selection_mode_;

    QList<QModelIndex> expanded_items_;


  public slots:

    void expand_row(QModelIndex item);

    void treeview_column_action(QAction* action);

    void onCustomContextMenu(const QPoint &point);

    void item_action(QAction* action);

    void keyPressEvent(QKeyEvent* event);

    void keyReleaseEvent(QKeyEvent* event);

    void mouseEvent(QModelIndex item);


  signals:

    void expand_item_in_graph(std::string item_name, bool last);

    void contract_item_in_graph(std::string item_name, bool last);


    void item_selected_in_treeview(std::string item_name, bool first, bool multiple);

    void focus_item_in_graph(std::string item_name);

    void exclude_item_in_graph(std::string item_name, bool last);

};

#endif