graph_scene.h

#ifndef GUI_GRAPH_SCENE_H_
#define GUI_GRAPH_SCENE_H_

// Qt-imports
#include <QGraphicsScene>
#include <QDebug>

// General imports
#include <queue>

// Local imports
#include "grapher.h"
#include "node_widget.h"
#include "edge_widget.h"
#include "graph_model.h"
#include "graph_model_item.h"


class GraphScene : public QGraphicsScene {
  Q_OBJECT

    public:
        GraphScene(GraphModel* graph, QObject* parent = 0);

    virtual ~GraphScene();

        void set_model(GraphModel* model);

    GraphModel* model(); 

    void set_node_label_column(int column_idx);

    void set_xy_columns(int x_idx, int y_idx);

    void set_color_column(int color); 

        void populateScene();

    NodeWidget* get_node_widget(std::string item_name);

    NodeWidget* get_node_widget(QString item_name);

        NodeWidget* get_node_widget(QModelIndex item_index);

        NodeWidget* add_node_widget(QModelIndex item_index);

        void del_node_widget(QModelIndex item_index);

    
    int nodes();

    std::vector<std::string> get_names_of_all_nodes();


  private:

    GraphModel* graph_;
    
    std::map<QString, NodeWidget*> node_widgets_;
    
    int num_nodes_;

    int label_column_;
    
    int x_column_;
    
    int y_column_;
    
    int color_column_;
};

#endif