-
Notifications
You must be signed in to change notification settings - Fork 2
/
balls.cpp
79 lines (61 loc) · 1.64 KB
/
balls.cpp
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
#include <Arduino.h>
#include "LPD8806.h"
#include "SPI.h"
#include "display.h"
#include "balls.h"
extern DisplayWrapper display;
namespace widget { namespace balls {
#define GRID_SIZE 6
struct local_color_t {
double r, g, b;
};
const local_color_t black = { 0.0, 0.0, 0.0 };
struct ball_t {
double x, y;
double dx, dy;
local_color_t color;
};
ball_t balls[] = {
{ 0.0, 0.0, 0.15, 0.10, { 1.0, 0.0, 0.0 } },
{ 2.0, 5.0, 0.18, -0.08, { 0.0, 1.0, 0.0 } },
{ 5.0, 2.0, -0.08, 0.18, { 0.0, 0.0, 1.0 } }
};
#define NUM_BALLS (sizeof(balls) / sizeof(balls[0]))
void move_through_dimension(double & xy, double & dxy) {
xy += dxy;
if (xy < 0) {
xy = -xy;
dxy = -dxy;
}
if (xy >= GRID_SIZE) {
xy = GRID_SIZE - (xy - GRID_SIZE);
dxy = -dxy;
}
}
void move_balls() {
for (unsigned int i = 0; i < NUM_BALLS; ++i) {
ball_t & b = balls[i];
move_through_dimension(b.x, b.dx);
move_through_dimension(b.y, b.dy);
}
}
void show_balls() {
for (int x = 0; x < GRID_SIZE; ++x) {
for (int y = 0; y < GRID_SIZE; ++y) {
local_color_t c = black;
for (unsigned int i = 0; i < NUM_BALLS; ++i) {
ball_t & b = balls[i];
double distance = pow(pow(b.x - x, 2) + pow(b.y - y, 2), 0.5);
double f = 1.0 / (pow(distance, 2) + 1.0);
c.r += b.color.r * f;
c.g += b.color.g * f;
c.b += b.color.b * f;
}
display.setPixelColor(x, y, int(c.r / NUM_BALLS * 255) + (int(c.g / NUM_BALLS * 255) << 8) + (long(c.b / NUM_BALLS * 255) << 16));
}
}
display.show();
}
void Balls::update() { move_balls(); show_balls(); }
void Balls::delay() { ::delay(BALL_DELAY); }
} }