I want to die

Author: AhmedBakrXI

parser_gui/Widgets/src/TreeVisualiser.cpp

@@ -1,26 +1,119 @@
 #include "TreeVisualiser.h"
+#include <QPainter>
+#include <QWheelEvent>
 
 using Tiny::Widgets::TreeVisualiser;
 
+void TreeVisualiser::paintEvent(QPaintEvent *event) {
+    Q_UNUSED(event);
+    if (!root) return;
+
+    QPainter painter(this);
+    painter.setRenderHint(QPainter::Antialiasing);
+
+    // Clear background
+    painter.fillRect(rect(), QColor(240, 240, 240));
+    painter.scale(zoomFactor, zoomFactor);
+
+    // Calculate the total width needed for the tree
+    int treeWidth = calculateTreeWidth(root);
+
+    // Draw the entire tree
+    drawTree(&painter, root, width() / 2, 50, treeWidth, 1);
+}
+
 TreeVisualiser::TreeVisualiser(QWidget *parent) : QWidget(parent) {
-    // set the layout
-    QVBoxLayout* layout = new QVBoxLayout(this);
-    layout->setContentsMargins(0, 0, 0, 0);
-    setLayout(layout);
+    Node *nroot = new Node(Node::NodeType::Expression, "+");
+    // nroot->setLevel(1);
+    Node *child1 = new Node(Node::NodeType::Number, "3");
+    // child1->setLevel(1);
+    Node *child2 = new Node(Node::NodeType::Number, "4");
+    // child2->setLevel(2);
+    Node *child3 = new Node(Node::NodeType::Number, "5");
+    // child3->setLevel(3);
+
+
+    this->root = nroot;
+    this->root->addChild(child1);
+    this->root->addChild(child2);
+    child2->addChild(child3);
+}
+
+void TreeVisualiser::drawTree(QPainter *painter, Node *node, int x, int y, int availableWidth, int currentLevel) {
+    if (!node) return;
+
+    // Draw the current node
+    painter->setPen(QPen(Qt::black, 2));
+    painter->setBrush(QColor(70, 130, 180)); // Steel Blue
+
+    if (isOval(node->getType())) {
+        painter->drawEllipse(x - 20, y - 20, 40, 40);
+    } else {
+        painter->drawRect(x - 40, y - 20, 80, 40);
+    }
 
-    // set the style
-    setStyleSheet("background-color: #2e2e2e ; color: #c0c0c0; border: 1px solid #2a2a2a;");
+    // Draw node value
+    painter->setPen(Qt::white);
+    QString nodeText;
+    if (!hasValue(node->getType())) {
+        nodeText = node->getNodeTypeString().toString() + "\n (" + node->getValue() + ")";
+    } else {
+        nodeText = node->getValue();
+    }
+    painter->drawText(QRect(x - 40, y - 20, 80, 40),
+                      Qt::AlignCenter | Qt::TextWordWrap, nodeText);
 
-    // add the label
-    QLabel* label = new QLabel("Tree Visualiser");
-    label->setAlignment(Qt::AlignCenter);
-    label->setStyleSheet("font-size: 20px; color: #c0c0c0;");
+    if (node->getChildren().empty()) return;
 
-    // add the label to the layout
-    layout->addWidget(label);
+    // Calculate spacing for children
+    int childCount = node->getChildren().size();
+    int totalChildWidth = 0;
+    std::vector<int> childWidths;
+
+    // Calculate width for each child's subtree
+    for (Node* child : node->getChildren()) {
+        int childWidth = calculateTreeWidth(child);
+        childWidths.push_back(childWidth);
+        totalChildWidth += childWidth;
+    }
+
+    // Determine vertical spacing based on level
+    int verticalSpacing = (currentLevel == 1) ? 0 : 100;
+
+    // Starting x position for first child
+    int startX = x - totalChildWidth / 2;
+
+    // Draw children and their subtrees
+    for (size_t i = 0; i < node->getChildren().size(); ++i) {
+        Node* child = node->getChildren()[i];
+        int childWidth = childWidths[i];
+
+        // Draw connection line
+        painter->setPen(QPen(Qt::black, 2));
+        painter->drawLine(x, y + 20, startX + childWidth/2, y + verticalSpacing);
+
+        // Recursive call with updated level
+        drawTree(painter, child, startX + childWidth/2, y + verticalSpacing, childWidth, 1);
+
+        startX += childWidth;
+    }
 }
 
-void TreeVisualiser::setRoot(Node *root)
+void TreeVisualiser::wheelEvent(QWheelEvent *event)
 {
-    qDebug() << "Drawing tree";
+    const qreal zoomStep = 0.1;   // Amount to zoom in/out
+    const qreal minZoom = 0.5;   // Minimum zoom level
+    const qreal maxZoom = 3.0;   // Maximum zoom level
+
+    if (event->angleDelta().y() > 0) {
+        // Zoom in
+        zoomFactor = qMin(maxZoom, zoomFactor + zoomStep);
+    } else {
+        // Zoom out
+        zoomFactor = qMax(minZoom, zoomFactor - zoomStep);
+    }
+
+    update(); // Redraw the widget
 }
+
+