lua-lqtk/example/example03a.lua at master · osch/lua-lqtk · 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
local lqtk = require("lqtk")

lqtk.import("Q.*", "newClass")

--local BaseClass = QOpenGLWidget
local BaseClass = QWidget

local BallsExample = newClass(BaseClass)
do
    function BallsExample:new(parent)
        BaseClass.new(self, parent)
        self:setWindowTitle("example03a.lua")
        local initialWidth, initialHeight = 640, 480
        self:resize(initialWidth, initialHeight)
        self.timer = QTimer()
        self.timer:setTimerType(Qt.PreciseTimer)
        self.timer:connect("timeout()", self, "update")
        self.timer:start(20)
        self.lastTime = QDateTime.currentMSecsSinceEpoch()
        local colors = {
            QColor("#e0ff0000"),
            QColor("#e000ff00"),
            QColor("#e00000ff")
        }
        self.gradients = {}
        for i = 1, 3 do
            local gradient = QLinearGradient(-1, -1, 1, 1)
            self.gradients[i] = gradient
            gradient:setColorAt(0, QColor("#e0ffffff"))
            gradient:setColorAt(1, colors[i])
        end

        self.objects = {}
        for i = 1, 500 do
            local obj = {}
            self.objects[i] = obj
            obj.r  = math.random(10, 30)
            obj.x  = math.random(obj.r, initialWidth  - 2 * obj.r)
            obj.y  = math.random(obj.r, initialHeight - 2 * obj.r)
            obj.g  = math.random(1, 3)
            obj.dx = math.random() / 20
            obj.dy = math.random() / 20
        end

        self.path = QPainterPath()
        self.path:addEllipse(-1, -1, 2, 2)
    end

    function BallsExample:update(...)
        local now = QDateTime.currentMSecsSinceEpoch()
        local dt = now - self.lastTime
        local w, h = self:width(), self:height()
        for i = 1, #self.objects do
            local obj = self.objects[i]
            local x, y = obj.x, obj.y
            x = x + obj.dx * dt
            if x + obj.r > w then
                x = w - obj.r
                obj.dx = -obj.dx
            elseif x - obj.r < 0 then
                x = obj.r
                obj.dx = -obj.dx
            end
            y = y + obj.dy * dt
            if y + obj.r > h then
                y = h - obj.r
                obj.dy = -obj.dy
            elseif y - obj.r < 0 then
                y = obj.r
                obj.dy = -obj.dy
            end
            obj.x, obj.y = x, y

            obj.dx = obj.dx * (1 - dt * 0.0001)
            obj.dy = obj.dy * (1 - dt * 0.0001)
        end
        self.lastTime = now
        BaseClass.update(self, ...)
    end

    function BallsExample:mousePressEvent(event)
        local pos = event:position()
        local mx, my = pos:x(), pos:y()
        for i = 1, #self.objects do
            local obj = self.objects[i]
            local x, y = obj.x, obj.y
            local bdx, bdy = x - mx, y - my
            local bd = math.sqrt(bdx^2 + bdy^2)
            local a = (bd + 0.01)^-1.5
            obj.dx = obj.dx + a * bdx
            obj.dy = obj.dy + a * bdy
        end
        self:update()
    end

    function BallsExample:paintEvent(event)
        local w, h = self:width(), self:height()
        local painter = QPainter(self)
        painter:setRenderHint(QPainter.Antialiasing)
        painter:setPen(Qt.NoPen)
        painter:fillRect(0, 0, w, h, QColor("#ffffff"))
        for _, obj in ipairs(self.objects) do
            painter:save()
            painter:translate(obj.x, obj.y)
            painter:scale(obj.r, obj.r)
            local gradient = self.gradients[obj.g]
            painter:setBrush(gradient)
            painter:drawPath(self.path)
            painter:restore()
        end
        painter:finish()
    end

end

QApplication(arg)
local example = BallsExample()
example:show()
QApplication.exec()