RS012-data-mining-toolbox/SceneNode.cpp at master · MATF-RS20/RS012-data-mining-toolbox · 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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
#include "SceneNode.hpp"

SceneNode::SceneNode(QString ID)
    : nodeID(ID)
{
    setFlag(ItemIsMovable);
    setFlag(ItemIsSelectable);
    setFlag(ItemIsFocusable);
}

QString SceneNode::getID()
{
    return nodeID;
}

SceneNode::Stanje SceneNode::GetNodeState()
{
    return s;
}

void SceneNode::ClearNodeState()
{
    s = Neoznacen;
}


QRectF SceneNode::boundingRect() const
{
    return QRectF(10,10,50,50);
}

void SceneNode::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget)
{
    Q_UNUSED(option)

    //fja za iscrtavanje cvora na sceni

    QPen olovka(Qt::black);
    olovka.setWidth(2);

    //ukoliko je cvor selektovan iscrtacamo i njegove granice
    if(isSelected()){

        QPen olovkaObod;
        olovkaObod.setStyle(Qt::DashLine);
        olovkaObod.setWidth(2);

        painter->setPen(olovkaObod);
        QRectF obod = boundingRect();
        painter->drawRect(obod);

        olovka.setColor(Qt::yellow);
    }


    QBrush cetkica(Qt::cyan);

    //bojimo cvor razlicitim bojama u zavisnosti od grup algoritama kojoj cvor pripada
    if(nodeID[0] == 'P')
        cetkica.setColor(Qt::darkCyan); //preprocesiranj
    else if(nodeID[0] == 'K')
        cetkica.setColor(Qt::darkGreen); //klasifikacija
    else if(nodeID[0] == 'C')
        cetkica.setColor(Qt::darkBlue); //klasterovanje
    else if(nodeID[0] == 'R')
        cetkica.setColor(Qt::darkRed); //regresija
    else if(nodeID[0] == 'U')
        cetkica.setColor(Qt::magenta); //ulazni podaci


    painter->setPen(olovka);
    painter->setBrush(cetkica);

    //crtamo cvor
    painter->drawEllipse(10,10,50,50);


    //ispisujemo naziv algoritma koji cvor predstavlja
    QCharRef s1 = nodeID[2];
    QCharRef s2 = nodeID[3];

    olovka.setWidth(10);
    painter->setPen(olovka);

    if(s2 == '_'){

        QString nazivAlg;
        nazivAlg.resize(1);
        nazivAlg[0] = s1;
        painter->drawText(40,25,nazivAlg);

    }else{

        QString nazivAlg;
        nazivAlg.resize(2);
        nazivAlg[0] = s1;
        nazivAlg[1] = s2;
        painter->drawText(31,25,nazivAlg);

    }

    //ispisujemo stanje cvora
    olovka.setColor(Qt::black);
    painter->setPen(olovka);

    if(s == Oznacen_1)
        painter->drawText(10,10, "1");
    if(s == Oznacen_2)
        painter->drawText(10,10, "2");
}


void SceneNode::keyPressEvent(QKeyEvent *event)
{
    //fja registruje pritiske na tastere i u zavisnoti od njih menja stanje cvora

    if(event->key() == Qt::Key_1){ //pritisnut taster 1
        s = Oznacen_1;
        update();
    }

    if(event->key() == Qt::Key_2){ //pritisnut taster 2
        s = Oznacen_2;
        update();
    }

    if(event->key() == Qt::Key_3){ //pritisnut taster 3
        s = Neoznacen;
        update();
    }
}