|| || || || || || || ||

---

BoostedGraph.h

Go to the documentation of this file.

#include "OglGui/Graph.h"

#include <string>
#include <map>
#include <vector>
#include <utility>                   // for std::pair
#include <algorithm>                 // for std::for_each

#include <boost/graph/graph_traits.hpp>
#include <boost/graph/adjacency_list.hpp>
#include <boost/graph/simple_point.hpp>
#include <boost/graph/circle_layout.hpp>
#include <boost/graph/fruchterman_reingold.hpp>

#include "Tag.h"

using namespace OglGui;
using namespace std;
using namespace boost;

class BoostedGraph : public OglGui::Graph, public OglGui::WindowListener {
public:
        typedef adjacency_list<vecS, vecS, undirectedS, property<vertex_name_t, string> > Graph;
        typedef graph_traits<Graph>::vertex_descriptor Vertex;

        BoostedGraph(Window* parent, int w, int h, int drawShape=0, strconst txt="") : OglGui::Graph(parent, w, h, drawShape, txt) {
                layoutDone = false;
        }

        ~BoostedGraph() {
                ILOG_DEBUG("Destructing BoostedGraph");
                initialLayoutDone = false;
                layoutDone = false;
        }

        void addNodes(map<string, vector<string> > labels) {
                map<string, vector<string> >::iterator iter;
                for(iter = labels.begin(); iter != labels.end(); iter++) {
                        string label = iter->first;
                        if(nodes.count(label) == 0) addNode(label);
                        for(uint i = 0; i < iter->second.size(); i++) {
                                string related = iter->second[i];
                                if(nodes.count(related) == 0) addNode(related);
                                ConnectNode(label, related);
                        }
                }

                if(labels.begin()->first.compare("obama") == 0) {
                        ConnectNode("election", "president"); ConnectNode("barack", "barackobama");
                        ConnectNode("vote", "election"); ConnectNode("president", "politics");
                        ConnectNode("vote", "president"); ConnectNode("president", "obama");
                        ConnectNode("change", "hope"); ConnectNode("campaign", "election");
                }
        }

        virtual void DisplayFunc() {
                //ILOG_DEBUG("DisplayFunc");
                if(!layoutDone) layoutDone = doLayout();
                OglGui::Graph::DisplayFunc();
        }

    virtual void OnLayoutChange()
    {
                layoutDone = false;
                OglGui::Graph::OnLayoutChange();
    }

    virtual void WindowMouseEvent(OglGui::Window *src, int msg, int but, int state, int x, int y, void *userData )
    {
        if (but==oglLeftButton && msg == oglMouseMove) dragging = true;

        if (dragging && but==oglLeftButton && msg == oglMouseUp) {
                OglGui::Graph* g = (OglGui::Graph*) userData;
                // update BoostGraph with move of node
                // TODO: This updating should be more elegant
                graph_traits<Graph>::vertex_iterator vi, vi_end;
                for (tie(vi, vi_end) = vertices(graph); vi != vi_end; ++vi) {
                        string name = get(vertex_name, graph, *vi);
                        if(name.compare(g->GetText()) != 0) continue;
                        int x, y, w, h;
                        g->GetDimensions(x, y, w, h);
                ILOG_DEBUG(g->GetText() + " moved");// to (" + MakeString(x) + "," + MakeString(y) + ")");
                        position[*vi].x = x-center.x;
                        position[*vi].y = y-center.y;
                layoutDone = false;
                break;
                }
                dragging = false;
        }
    }

private:
        void addNode(string label) {
                int width, height;
                this->GetDimensions(width, height);
                int x  = ABSRND(width-20);
                int y  = ABSRND(height-20);

                nodes[label].first = add_vertex(label, graph);
                nodes[label].second = new OglGui::Graph(this,x,y,75,20,1,label);
                nodes[label].second->ShapeBg(ARGB2COLOR(208,255,24,0));
                nodes[label].second->SetBorderType(0);
                nodes[label].second->ActOnModifier(0);
                nodes[label].second->SetAllowSizeDirections(AllDir, false);
        nodes[label].second->SetWindowListener(this, nodes[label].second);

                this->AddNode(nodes[label].second);
        }

        void ConnectNode(string from, string to) {
                add_edge(nodes[from].first, nodes[to].first, graph);
                OglGui::Graph::ConnectNode(nodes[from].second, nodes[to].second);
        }

        void doInitialLayout() {
                position_vec = PositionVec(num_vertices(graph));
                position = PositionMap(position_vec.begin(), get(vertex_index, graph));
                circle_graph_layout(graph, position, 10);
                fruchterman_reingold_force_directed_layout(graph, position, 2*center.x-10, 2*center.y-10,
                                force_pairs(all_force_pairs()));
                initialLayoutDone = true;
        }

        bool doLayout() {
                center.x = (W()-75)/2; center.y = (H()-20)/2;
                if(!initialLayoutDone) doInitialLayout();
                fruchterman_reingold_force_directed_layout(graph, position, 2*center.x-10, 2*center.y-10,
                                force_pairs(all_force_pairs()).
                                cooling(linear_cooling<double>(50)));

                graph_traits<Graph>::vertex_iterator vi, vi_end;
                for (tie(vi, vi_end) = vertices(graph); vi != vi_end; ++vi) {
                        string name = get(vertex_name, graph, *vi);
                        //ILOG_DEBUG(name + ": " + Impala::MakeString(position[*vi].x) + " " + Impala::MakeString(position[*vi].y));
                        int x, y, w, h;
                        nodes[name].second->GetDimensions(x, y, w, h);
                        nodes[name].second->SetDimensions(center.x+position[*vi].x, center.y+position[*vi].y, w, h);
                        nodes[name].second->ShapeBg(ARGB2COLOR(208,255,24,0));
                }
                return true;
        }

        typedef vector<simple_point<double> > PositionVec;
        typedef iterator_property_map<PositionVec::iterator, property_map<Graph, vertex_index_t>::type> PositionMap;

        PositionVec position_vec;
        PositionMap position;

        simple_point<double> center;

    map<string, pair<Vertex, OglGui::Graph*> > nodes;
    Graph graph;

        bool initialLayoutDone;
    bool layoutDone;
    bool dragging;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(BoostedGraph, Impala.Application.TagsLife);

---