nodeeditor2/examples/calculator/DivisionModel.hpp at 87e219b51e5fd6ccec00e1ffd2d1c5c0535667b9 · fabns-nano/nodeeditor2 · 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
#pragma once

#include "DecimalData.hpp"
#include "MathOperationDataModel.hpp"

#include <QtNodes/NodeDelegateModel>

#include <QtCore/QObject>
#include <QtWidgets/QLabel>

/// The model dictates the number of inputs and outputs for the Node.
/// In this example it has no logic.
class DivisionModel : public MathOperationDataModel
{
public:
    virtual ~DivisionModel() {}

public:
    QString caption() const override { return QStringLiteral("Division"); }

    bool portCaptionVisible(PortType portType, PortIndex portIndex) const override
    {
        Q_UNUSED(portType);
        Q_UNUSED(portIndex);
        return true;
    }

    QString portCaption(PortType portType, PortIndex portIndex) const override
    {
        switch (portType) {
        case PortType::In:
            if (portIndex == 0)
                return QStringLiteral("Dividend");
            else if (portIndex == 1)
                return QStringLiteral("Divisor");

            break;

        case PortType::Out:
            return QStringLiteral("Result");

        default:
            break;
        }
        return QString();
    }

    QString name() const override { return QStringLiteral("Division"); }

private:
    void compute() override
    {
        PortIndex const outPortIndex = 0;

        auto n1 = _number1.lock();
        auto n2 = _number2.lock();

        QtNodes::NodeValidationState state;
        if (n2 && (n2->number() == 0.0)) {
            state._state = QtNodes::NodeValidationState::State::Error;
            state._stateMessage = QStringLiteral("Division by zero error");
            setValidationState(state);
            _result.reset();
        } else if ( n2 && (n2->number() < 1e-5)) {
            state._state = QtNodes::NodeValidationState::State::Warning;
            state._stateMessage = QStringLiteral("Very small divident. Result might overflow");
            setValidationState(state);
            if (n1) {
                _result = std::make_shared<DecimalData>(n1->number() / n2->number());
            } else {
                _result.reset();
            }
        } else if (n1 && n2) {
            setValidationState(state);
            _result = std::make_shared<DecimalData>(n1->number() / n2->number());
        } else {
            QtNodes::NodeValidationState state;
            setValidationState(state);
            _result.reset();
        }

        Q_EMIT dataUpdated(outPortIndex);
    }
};