/
potentiometer.py
59 lines (46 loc) · 1.77 KB
/
potentiometer.py
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
from math import pi, degrees
import pygame as pg
import prepare
from angles import get_distance, get_angle
class Potentiometer(object):
def __init__(self, pos):
self.pos = pos
self.grabbed = False
self.hovered = False
self.base_image = self.image = prepare.GFX["knob"]
self.rect = self.image.get_rect(center=self.pos)
self.angle = self.last_angle = 0
self.radius = self.rect.w // 2
self.min_angle, self.max_angle = pi, 0
self.angle = .5 * pi
self.adjust_output()
def get_event(self, event):
if event.type == pg.MOUSEBUTTONDOWN:
if self.hovered:
self.grab(event.pos)
elif event.type == pg.MOUSEBUTTONUP:
self.grabbed = False
def grab(self, pos):
self.grabbed = True
def rotate(self):
self.image = pg.transform.rotate(self.base_image, degrees(self.angle))
self.rect = self.image.get_rect(center=self.pos)
def adjust_output(self):
self.output = 1.0 - (self.angle / float(self.min_angle))
def update(self, mouse_pos):
self.hovered = get_distance(self.pos, mouse_pos) <= self.radius
if self.grabbed:
angle = get_angle(self.pos, mouse_pos)
if self.angle < .5 * pi:
if angle > self.min_angle:
angle = self.max_angle
else:
if angle > self.min_angle:
angle = self.min_angle
self.angle = angle
if self.angle != self.last_angle:
self.rotate()
self.adjust_output()
self.last_angle = self.angle
def draw(self, surface):
surface.blit(self.image, self.rect)