SFMLGraphVisualizer.cpp

//
// Created by hartings on 23.05.18.
//

#include <sstream>
#include <math.h>
#include "SFMLGraphVisualizer.h"
#include <iostream>
#include "Node.h"

#define NODERADIUS 10
using namespace std;


SFMLGraphVisualizer::SFMLGraphVisualizer() {
    this->font.loadFromFile("../FreeMono.ttf");
}

void SFMLGraphVisualizer::visualize(DiGraph &graph) {

    window.create(sf::VideoMode(1024, 1024), "Graph Window");
    List<Node> allNodes = graph.getAllNodes();

    while (window.isOpen()) {
        sf::Event event;

        while (window.pollEvent(event)) {
            if (event.type == sf::Event::Closed)
                window.close();
        }

        window.clear(sf::Color::White);

        for (int i = 0; i < allNodes.getListSize(); i++) {
            Node *node = allNodes.getValueAtPos(i);
            sf::Color color(255, 0, 0);
            drawNode(*node, color);
            List<Edge> allEdges = graph.getAllEdgesOfNode(node->getKey());
            for (unsigned int j = 0; j < allEdges.getListSize(); j++) {
                drawEdge(*(allEdges.getValueAtPos(j)), sf::Color::Black, allEdges.getValueAtPos(j)->getWeight(), 1);
            }
        }
        window.display();
    }
}

void SFMLGraphVisualizer::drawNode(Node &node, sf::Color nodeColor) {
    sf::CircleShape Circle(NODERADIUS);
    Circle.setPosition(node.getPosX() - NODERADIUS, node.getPosY() - NODERADIUS);
    Circle.setFillColor(sf::Color::White);
    Circle.setOutlineColor(nodeColor);
    Circle.setOutlineThickness(5);

    window.draw(Circle);

    sf::Text Label(node.getKey(), font, 32);
    Label.setPosition(node.getPosX() - NODERADIUS / 2 + 5, node.getPosY() - NODERADIUS / 2 - 5);
    Label.setColor(sf::Color::Blue);

    window.draw(Label);
}

void SFMLGraphVisualizer::drawEdge(Edge &edge, sf::Color color, double weight, int thickness, int arrowMagnitude) {
    sf::Vector2f p(edge.getStartNode()->getPosX(), edge.getStartNode()->getPosY());
    sf::Vector2f q(edge.getEndNode()->getPosX(), edge.getEndNode()->getPosY());

    // WINKEL BERECHNUNG
    const double PI = 3.141592653;
    double angle = atan2((double) p.y - q.y, (double) p.x - q.x);

    //  Berechne  verkuerzten  Pfeil
    q.x = (int) (q.x + NODERADIUS * cos(angle));
    q.y = (int) (q.y + NODERADIUS * sin(angle));
    p.x = (int) (p.x - NODERADIUS * cos(angle));
    p.y = (int) (p.y - NODERADIUS * sin(angle));

    sf::Vertex line[2] =
            {
                    sf::Vertex(sf::Vector2f(p.x, p.y), color),
                    sf::Vertex(sf::Vector2f(q.x, q.y), color)
            };

    //  thickness
    window.draw(line, 2, sf::Lines);

    std::stringstream weightstring;
    weightstring << weight;
    sf::Text Labelweight(weightstring.str(), font, 32);
    int size = sqrt(pow(p.x - q.x, 2) + pow(p.y - q.y, 2));
    Labelweight.setPosition(p.x - (size / 2) * cos(angle) + 10 * sin(angle),
                            p.y - (size / 2) * sin(angle) + 10 * cos(angle));
    Labelweight.setColor(sf::Color::Blue);
    window.draw(Labelweight);

// Erstes  Segment
    p.x = (int) (q.x + arrowMagnitude * cos(angle + PI / 8));
    p.y = (int) (q.y + arrowMagnitude * sin(angle + PI / 8));
    sf::Vertex first[2] =
            {
                    sf::Vertex(sf::Vector2f(p.x, p.y), color),
                    sf::Vertex(sf::Vector2f(q.x, q.y), color)
            };
    window.draw(first, 2, sf::Lines);

// Zweites  Segment
    p.x = (int) (q.x + arrowMagnitude * cos(angle - PI / 8));
    p.y = (int) (q.y + arrowMagnitude * sin(angle - PI / 8));

    sf::Vertex second[2] =
            {
                    sf::Vertex(sf::Vector2f(p.x, p.y), color),
                    sf::Vertex(sf::Vector2f(q.x, q.y), color)
            };
    window.draw(second, 2, sf::Lines);

}