mirror of
https://gitlab1.ptb.de/waltem01/Matrix
synced 2024-09-17 10:09:43 +00:00
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
from ..samplebase import SampleBase
|
|
import math
|
|
|
|
|
|
def scale_col(val, lo, hi):
|
|
if val < lo:
|
|
return 0
|
|
if val > hi:
|
|
return 255
|
|
return 255 * (val - lo) / (hi - lo)
|
|
|
|
|
|
def rotate(x, y, sin, cos):
|
|
return x * cos - y * sin, x * sin + y * cos
|
|
|
|
|
|
class RotatingBlockGenerator(SampleBase):
|
|
def __init__(self, *args, **kwargs):
|
|
super(RotatingBlockGenerator, self).__init__(*args, **kwargs)
|
|
|
|
def run(self):
|
|
cent_x = self.matrix.width / 2
|
|
cent_y = self.matrix.height / 2
|
|
|
|
rotate_square = min(self.matrix.width, self.matrix.height) * 1.41
|
|
min_rotate = cent_x - rotate_square / 2
|
|
max_rotate = cent_x + rotate_square / 2
|
|
|
|
display_square = min(self.matrix.width, self.matrix.height) * 0.7
|
|
min_display = cent_x - display_square / 2
|
|
max_display = cent_x + display_square / 2
|
|
|
|
deg_to_rad = 2 * 3.14159265 / 360
|
|
rotation = 0
|
|
|
|
# Pre calculate colors
|
|
col_table = []
|
|
for x in range(int(min_rotate), int(max_rotate)):
|
|
col_table.insert(x, scale_col(x, min_display, max_display))
|
|
|
|
offset_canvas = self.matrix.CreateFrameCanvas()
|
|
|
|
while True:
|
|
rotation += 1
|
|
rotation %= 360
|
|
|
|
# calculate sin and cos once for each frame
|
|
angle = rotation * deg_to_rad
|
|
sin = math.sin(angle)
|
|
cos = math.cos(angle)
|
|
|
|
for x in range(int(min_rotate), int(max_rotate)):
|
|
for y in range(int(min_rotate), int(max_rotate)):
|
|
# Our rotate center is always offset by cent_x
|
|
rot_x, rot_y = rotate(x - cent_x, y - cent_x, sin, cos)
|
|
|
|
if x >= min_display and x < max_display and y >= min_display and y < max_display:
|
|
x_col = col_table[x]
|
|
y_col = col_table[y]
|
|
offset_canvas.SetPixel(rot_x + cent_x, rot_y + cent_y, x_col, 255 - y_col, y_col)
|
|
else:
|
|
offset_canvas.SetPixel(rot_x + cent_x, rot_y + cent_y, 0, 0, 0)
|
|
|
|
offset_canvas = self.matrix.SwapOnVSync(offset_canvas)
|
|
|
|
|
|
# Main function
|
|
if __name__ == "__main__":
|
|
rotating_block_generator = RotatingBlockGenerator()
|
|
if (not rotating_block_generator.process()):
|
|
rotating_block_generator.print_help()
|