TreesDrawer/functions.cpp at main · veryCoolBebries/TreesDrawer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
//
// Created by Максим Лейхнер on 27.11.2022.
//
#include <SFML/Graphics.hpp>
#include <list>
#include <cmath>
#include "trees.h"
#include "functions.h"

using namespace sf;
using namespace std;

Texture createBackgroundTexture(Uint16 windowWidth, Uint16 windowHeight, Color color){
    //Создаем текстуру фона
    Texture backgroundTexture;
    backgroundTexture.create(windowWidth, windowHeight);
    //Массив пикселей
    Uint8 * backgroundPixels = new Uint8 [windowHeight * windowWidth * 4];
    //Раскладываем цвет на составляющие
    Uint8 myColor[4] = {color.r, color.g, color.b, color.a};
    //Записываем цвет в каждый пиксель
    for(int pixel = 0; pixel < windowWidth * windowHeight * 4; pixel++){
        backgroundPixels[pixel] = myColor[pixel % 4];
    }
    //Обновляем текстуру новыми пикселями
    backgroundTexture.update(backgroundPixels);
    //Возвращяем текстуру
    return backgroundTexture;
}

void renderNodes(list <Drawable *> * renderQueue, Node * node, Font *font, int x, int y, int xOffset, int yOffset, int horizontalDistribution, int verticalDistribution, Node * toHighlight, int layer){

    //Рисуем ветки
    if(node->left) {
        VertexArray * line = new VertexArray(sf::Lines);
        Vertex * pointOne = new Vertex();
        Vertex * pointTwo = new Vertex();
        pointOne->color = Color::Black;
        pointOne->position = Vector2f(x + xOffset, y + yOffset);
        pointTwo->color = Color::Black;
        pointTwo->position = Vector2f(x - horizontalDistribution / pow(2, layer - 1) + xOffset, y + verticalDistribution + yOffset);
        line->resize(2);
        line->append(*pointOne);
        line->append(*pointTwo);
        renderQueue->push_back(line);
    }
    if(node->right) {
        VertexArray * line = new VertexArray(sf::Lines);
        Vertex * pointOne = new Vertex();
        Vertex * pointTwo = new Vertex();
        pointOne->color = Color::Black;
        pointOne->position = Vector2f(x + xOffset, y + yOffset);
        pointTwo->color = Color::Black;
        pointTwo->position = Vector2f(x + horizontalDistribution / pow(2, layer - 1) + xOffset, y + verticalDistribution + yOffset);
        line->resize(2);
        line->append(*pointOne);
        line->append(*pointTwo);
        renderQueue->push_back(line);
    }

    bool desiredNode = toHighlight == node;

    //Рисуем узел
    CircleShape * newCircle = new CircleShape(50);
    newCircle->setOutlineThickness(desiredNode ? 10 : 5);
    newCircle->setOutlineColor(desiredNode ? Color(177, 36, 36) : Color(217, 155, 102));
    newCircle->setOrigin(Vector2f(50, 50));
    newCircle->setPosition(x + xOffset, y + yOffset);
    renderQueue->push_back(newCircle);

    //Рисуем надпись
    Text * number = new Text;
    number->setString(to_string(node->data));
    number->setFont(*font);
    number->setCharacterSize(50);
    number->setFillColor(Color( 191, 90, 90));
    number->setOrigin(number->getLocalBounds().width/2, number->getLocalBounds().height/2);
    number->setPosition(x + xOffset, y + yOffset - 15);
    renderQueue->push_back(number);

    //Рекурсивно вызываем функцию для левого и правого ребёнка
    if(node->left)
        renderNodes(renderQueue, node->left, font, x - horizontalDistribution / pow(2, layer - 1), y + verticalDistribution, xOffset, yOffset, horizontalDistribution, verticalDistribution, toHighlight, layer + 1);
    if(node->right)
        renderNodes(renderQueue, node->right, font, x + horizontalDistribution / pow(2, layer - 1), y + verticalDistribution, xOffset, yOffset, horizontalDistribution, verticalDistribution, toHighlight, layer + 1);

}

void renderTree(list <Drawable *> * renderQueue, Tree * tree, Font *font, int x, int y, int xOffset, int yOffset, int horizontalDistribution, int verticalDistribution, Node * toHighlight, int layer){
    if(tree->root)
    renderNodes(renderQueue, tree->root, font, x, y, xOffset, yOffset, horizontalDistribution, verticalDistribution, toHighlight, layer);
    else{
        CircleShape * newCircle = new CircleShape(50);
        newCircle->setOutlineThickness(7);
        newCircle->setFillColor(Color::Transparent);
        newCircle->setOutlineColor(Color(107, 107, 107));
        newCircle->setOrigin(Vector2f(50, 50));
        newCircle->setPosition(x + xOffset, y + yOffset);
        renderQueue->push_back(newCircle);
    }
}

void clearRenderQueue(list <Drawable *> * renderQueue){
    while(!renderQueue->empty()){
        delete *renderQueue->begin();
        renderQueue->pop_front();
    }
}

String generateTreeDescription(int treeSize, int treeHeight, int treeSum) {
    string description;
    description = to_string(treeSize) + " элементов\nВысота – " + to_string(treeHeight) + "\nСумма путей до четных – " +
            to_string(treeSum);
    return String::fromUtf8(description.begin(), description.end());
}