Balazs Racz / Mbed 2 deprecated LedStrip_test

Important changes to repositories hosted on mbed.com

Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.

To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.

It is also possible to export all your personal repositories from the account settings page.

Dependencies:   mbed

main.cpp@8:77fd54b4864c, 2013-09-21 (annotated)

Committer:
juliogerchman
Date:
Sat Sep 21 20:49:33 2013 +0000
Revision:
8:77fd54b4864c
Parent:
6:62c9a5483a84
Parent:
7:e3b6f5741d9d
Child:
9:c3d4c7059979
Child:
10:01eb82c2a01b
Fade Out in the new framework

Who changed what in which revision?

UserRevisionLine numberNew contents of line
bracz 0:109a7a5e3e6e 1 #include <queue>
juliogerchman 3:42efa00ffef4 2 #include <algorithm>
bracz 0:109a7a5e3e6e 3
bracz 0:109a7a5e3e6e 4 #include "mbed.h"
bracz 0:109a7a5e3e6e 5
bracz 0:109a7a5e3e6e 6 DigitalOut myled(LED1);
bracz 4:5a2af711d510 7 DigitalOut led2(LED2);
bracz 0:109a7a5e3e6e 8
bracz 0:109a7a5e3e6e 9 Serial pc(USBTX, USBRX);
bracz 0:109a7a5e3e6e 10
bracz 0:109a7a5e3e6e 11 SPI spi(p5, p6, p20); // mosi, miso, sclk
bracz 0:109a7a5e3e6e 12 DigitalOut latch(p16);
bracz 0:109a7a5e3e6e 13 DigitalOut strobe(p17);
bracz 0:109a7a5e3e6e 14
juliogerchman 2:590e99f4a313 15 Ticker g_ticker;
juliogerchman 2:590e99f4a313 16
juliogerchman 2:590e99f4a313 17
bracz 1:1a0d5a780e57 18 #define LENGTH 160
bracz 1:1a0d5a780e57 19
bracz 0:109a7a5e3e6e 20 uint8_t strip[160];
bracz 0:109a7a5e3e6e 21
bracz 0:109a7a5e3e6e 22 #define BLACK 0x80
bracz 0:109a7a5e3e6e 23 #define BLUE (BLACK | 0x10)
bracz 0:109a7a5e3e6e 24 #define RED (BLACK | 0x04)
bracz 0:109a7a5e3e6e 25 #define GREEN (BLACK | 0x01)
bracz 0:109a7a5e3e6e 26 #define YELLOW (RED | GREEN)
bracz 0:109a7a5e3e6e 27 #define CYAN (GREEN | BLUE)
bracz 0:109a7a5e3e6e 28 #define MAGENTA (RED | BLUE)
bracz 0:109a7a5e3e6e 29 #define WHITE (RED | GREEN | BLUE)
bracz 0:109a7a5e3e6e 30
juliogerchman 2:590e99f4a313 31 #define FAST 0x40
juliogerchman 2:590e99f4a313 32
bracz 0:109a7a5e3e6e 33 uint8_t fade_result[] =
bracz 4:5a2af711d510 34 {0, 2, 3, 1};
bracz 4:5a2af711d510 35
bracz 0:109a7a5e3e6e 36
bracz 4:5a2af711d510 37 uint8_t google_colors[] = {BLUE, RED, YELLOW, BLUE, GREEN, RED};
bracz 4:5a2af711d510 38
bracz 4:5a2af711d510 39 uint8_t getbit(uint8_t from, uint8_t to)
bracz 4:5a2af711d510 40 {
bracz 0:109a7a5e3e6e 41 return fade_result[((from&1) << 1) | (to & 1)];
bracz 0:109a7a5e3e6e 42 }
bracz 0:109a7a5e3e6e 43
bracz 4:5a2af711d510 44 uint8_t getcolor(uint8_t from, uint8_t to)
bracz 4:5a2af711d510 45 {
bracz 0:109a7a5e3e6e 46 uint8_t result = 0x80;
bracz 0:109a7a5e3e6e 47 result |= getbit(from >> 0, to >> 0) << 0;
bracz 0:109a7a5e3e6e 48 result |= getbit(from >> 2, to >> 2) << 2;
bracz 0:109a7a5e3e6e 49 result |= getbit(from >> 4, to >> 4) << 4;
bracz 0:109a7a5e3e6e 50 return result;
bracz 0:109a7a5e3e6e 51 }
bracz 0:109a7a5e3e6e 52
bracz 0:109a7a5e3e6e 53
bracz 4:5a2af711d510 54 void write_strip(uint8_t* data, int len)
bracz 4:5a2af711d510 55 {
bracz 0:109a7a5e3e6e 56 latch = 0;
bracz 0:109a7a5e3e6e 57 for (int i = len - 1; i >= 0; i--) {
bracz 0:109a7a5e3e6e 58 spi.write(data[i]);
bracz 0:109a7a5e3e6e 59 }
bracz 0:109a7a5e3e6e 60 latch = 1;
bracz 0:109a7a5e3e6e 61 wait_us(2);
bracz 0:109a7a5e3e6e 62 latch = 0;
bracz 0:109a7a5e3e6e 63 }
bracz 0:109a7a5e3e6e 64
bracz 4:5a2af711d510 65 class Schedulable
bracz 4:5a2af711d510 66 {
bracz 0:109a7a5e3e6e 67 public:
bracz 0:109a7a5e3e6e 68 int time_;
bracz 0:109a7a5e3e6e 69 virtual void Run() = 0;
bracz 4:5a2af711d510 70
bracz 0:109a7a5e3e6e 71 bool operator<(const Schedulable& o) const {
bracz 0:109a7a5e3e6e 72 return time_ < o.time_;
bracz 0:109a7a5e3e6e 73 }
bracz 0:109a7a5e3e6e 74 };
bracz 0:109a7a5e3e6e 75
bracz 0:109a7a5e3e6e 76
bracz 0:109a7a5e3e6e 77 struct comp {
bracz 0:109a7a5e3e6e 78 bool operator()(const Schedulable* a, const Schedulable* b) {
bracz 0:109a7a5e3e6e 79 return *b < *a;
bracz 0:109a7a5e3e6e 80 }
bracz 0:109a7a5e3e6e 81 };
bracz 0:109a7a5e3e6e 82
bracz 0:109a7a5e3e6e 83 priority_queue<Schedulable*, vector<Schedulable*>, comp> task_list;
bracz 0:109a7a5e3e6e 84
bracz 4:5a2af711d510 85 void Schedule(Schedulable* action)
bracz 4:5a2af711d510 86 {
bracz 0:109a7a5e3e6e 87 task_list.push(action);
bracz 0:109a7a5e3e6e 88 }
bracz 0:109a7a5e3e6e 89
bracz 0:109a7a5e3e6e 90 int global_tick = 0;
bracz 0:109a7a5e3e6e 91 bool strip_changed;
bracz 0:109a7a5e3e6e 92
bracz 4:5a2af711d510 93 void tick_cb()
bracz 4:5a2af711d510 94 {
bracz 0:109a7a5e3e6e 95 ++global_tick;
bracz 0:109a7a5e3e6e 96 strobe = !strobe;
bracz 0:109a7a5e3e6e 97 }
bracz 0:109a7a5e3e6e 98
bracz 4:5a2af711d510 99 class RepeatedFadeInOut : public Schedulable
bracz 4:5a2af711d510 100 {
bracz 0:109a7a5e3e6e 101 public:
bracz 0:109a7a5e3e6e 102 RepeatedFadeInOut(int start_time, int led, uint8_t a, uint8_t b, bool fast)
bracz 0:109a7a5e3e6e 103 : led_(led), a_(a), b_(b), fast_(fast) {
bracz 0:109a7a5e3e6e 104 time_ = start_time;
bracz 0:109a7a5e3e6e 105 Schedule(this);
bracz 4:5a2af711d510 106 }
bracz 0:109a7a5e3e6e 107
bracz 0:109a7a5e3e6e 108 virtual void Run() {
bracz 4:5a2af711d510 109 strip[led_] = getcolor(a_, b_);
bracz 0:109a7a5e3e6e 110 if (fast_) {
juliogerchman 2:590e99f4a313 111 strip[led_] |= FAST;
bracz 0:109a7a5e3e6e 112 time_ += 128;
bracz 0:109a7a5e3e6e 113 } else {
bracz 0:109a7a5e3e6e 114 time_ += 256;
bracz 0:109a7a5e3e6e 115 }
bracz 0:109a7a5e3e6e 116 strip_changed = true;
bracz 0:109a7a5e3e6e 117 swap(a_,b_);
bracz 0:109a7a5e3e6e 118 Schedule(this);
bracz 0:109a7a5e3e6e 119 }
bracz 4:5a2af711d510 120
bracz 0:109a7a5e3e6e 121 private:
bracz 0:109a7a5e3e6e 122 int led_;
bracz 0:109a7a5e3e6e 123 uint8_t a_,b_;
bracz 0:109a7a5e3e6e 124 bool fast_;
bracz 0:109a7a5e3e6e 125 };
bracz 0:109a7a5e3e6e 126
bracz 4:5a2af711d510 127 class WalkingFadeInOut : public Schedulable
bracz 4:5a2af711d510 128 {
bracz 1:1a0d5a780e57 129 public:
bracz 1:1a0d5a780e57 130 WalkingFadeInOut(int start_time, int led, int stride, uint8_t a, uint8_t b, bool fast)
bracz 1:1a0d5a780e57 131 : led_(led - stride), stride_(stride), a_(a), b_(b), fast_(fast), step_(true) {
bracz 1:1a0d5a780e57 132 time_ = start_time;
bracz 1:1a0d5a780e57 133 Schedule(this);
bracz 4:5a2af711d510 134 }
bracz 1:1a0d5a780e57 135
bracz 1:1a0d5a780e57 136 virtual void Run() {
bracz 1:1a0d5a780e57 137 if (step_) {
bracz 1:1a0d5a780e57 138 step_ = false;
bracz 1:1a0d5a780e57 139 if (led_ >= 0) strip[led_] = a_;
bracz 1:1a0d5a780e57 140 led_ += stride_;
bracz 1:1a0d5a780e57 141 led_ %= LENGTH;
bracz 1:1a0d5a780e57 142 } else {
bracz 1:1a0d5a780e57 143 step_ = true;
bracz 1:1a0d5a780e57 144 }
bracz 4:5a2af711d510 145 strip[led_] = getcolor(a_, b_);
bracz 1:1a0d5a780e57 146 if (fast_) {
juliogerchman 2:590e99f4a313 147 strip[led_] |= FAST;
bracz 1:1a0d5a780e57 148 time_ += 128;
bracz 1:1a0d5a780e57 149 } else {
bracz 1:1a0d5a780e57 150 time_ += 256;
bracz 1:1a0d5a780e57 151 }
bracz 1:1a0d5a780e57 152 strip_changed = true;
bracz 1:1a0d5a780e57 153 swap(a_,b_);
bracz 1:1a0d5a780e57 154 Schedule(this);
bracz 1:1a0d5a780e57 155 }
bracz 4:5a2af711d510 156
bracz 1:1a0d5a780e57 157 private:
bracz 1:1a0d5a780e57 158 int led_, stride_;
bracz 1:1a0d5a780e57 159 uint8_t a_,b_;
bracz 1:1a0d5a780e57 160 bool fast_, step_;
bracz 4:5a2af711d510 161
bracz 1:1a0d5a780e57 162 };
bracz 1:1a0d5a780e57 163
bracz 4:5a2af711d510 164 class RegionWalkingFadeInOut : public Schedulable
bracz 4:5a2af711d510 165 {
juliogerchman 2:590e99f4a313 166 public:
juliogerchman 2:590e99f4a313 167 RegionWalkingFadeInOut(int start_time, int led, int stride, int start_led, int length, uint8_t a, uint8_t b, bool fast, bool repeat, bool drop, Schedulable* next)
juliogerchman 2:590e99f4a313 168 : led_(led - stride), stride_(stride), start_led_(start_led), length_(length), a_(a), b_(b), fast_(fast), step_(true), repeat_(repeat), drop_(drop), next_(next) {
juliogerchman 2:590e99f4a313 169 time_ = start_time;
juliogerchman 2:590e99f4a313 170 Schedule(this);
bracz 4:5a2af711d510 171 }
bracz 1:1a0d5a780e57 172
juliogerchman 2:590e99f4a313 173 virtual void Run() {
juliogerchman 2:590e99f4a313 174 if (step_) {
juliogerchman 2:590e99f4a313 175 step_ = false;
juliogerchman 2:590e99f4a313 176 if (led_ >= 0) strip[led_ + start_led_] = a_;
juliogerchman 2:590e99f4a313 177 led_ += stride_;
juliogerchman 2:590e99f4a313 178 if (repeat_) {
juliogerchman 2:590e99f4a313 179 led_ %= length_;
juliogerchman 2:590e99f4a313 180 } else {
juliogerchman 3:42efa00ffef4 181 if (led_ >= length_) {
juliogerchman 3:42efa00ffef4 182 /*if (next_ && led_ == (length_ + stride_)) {
juliogerchman 2:590e99f4a313 183 next_->time_ = global_tick + 1;
juliogerchman 2:590e99f4a313 184 Schedule(next_);
juliogerchman 3:42efa00ffef4 185 }*/
juliogerchman 2:590e99f4a313 186 delete this;
juliogerchman 2:590e99f4a313 187 return;
juliogerchman 2:590e99f4a313 188 }
juliogerchman 2:590e99f4a313 189 }
juliogerchman 2:590e99f4a313 190 } else {
juliogerchman 3:42efa00ffef4 191 if (led_ == (length_ - 1) && drop_ && !repeat_) {
juliogerchman 2:590e99f4a313 192 if (next_) {
juliogerchman 3:42efa00ffef4 193 next_->time_ = global_tick + 257;
juliogerchman 2:590e99f4a313 194 Schedule(next_);
juliogerchman 2:590e99f4a313 195 }
juliogerchman 2:590e99f4a313 196 delete this;
juliogerchman 2:590e99f4a313 197 return;
juliogerchman 2:590e99f4a313 198 }
bracz 4:5a2af711d510 199 step_ = true;
juliogerchman 2:590e99f4a313 200 }
bracz 4:5a2af711d510 201 strip[led_ + start_led_] = getcolor(a_, b_);
juliogerchman 2:590e99f4a313 202 if (fast_) {
juliogerchman 2:590e99f4a313 203 strip[led_ + start_led_] |= FAST;
juliogerchman 2:590e99f4a313 204 time_ += 128;
juliogerchman 2:590e99f4a313 205 } else {
juliogerchman 2:590e99f4a313 206 time_ += 256;
juliogerchman 2:590e99f4a313 207 }
juliogerchman 2:590e99f4a313 208 strip_changed = true;
juliogerchman 2:590e99f4a313 209 swap(a_,b_);
juliogerchman 2:590e99f4a313 210 Schedule(this);
juliogerchman 2:590e99f4a313 211 }
bracz 4:5a2af711d510 212
juliogerchman 2:590e99f4a313 213 private:
juliogerchman 2:590e99f4a313 214 int led_, stride_;
juliogerchman 2:590e99f4a313 215 int start_led_, length_;
juliogerchman 2:590e99f4a313 216 uint8_t a_,b_;
juliogerchman 2:590e99f4a313 217 bool fast_, step_;
juliogerchman 2:590e99f4a313 218 bool repeat_, drop_;
juliogerchman 2:590e99f4a313 219 Schedulable* next_;
juliogerchman 2:590e99f4a313 220 };
juliogerchman 2:590e99f4a313 221
juliogerchman 8:77fd54b4864c 222 class WaitAndSetDone : public Schedulable {
juliogerchman 8:77fd54b4864c 223 public:
juliogerchman 8:77fd54b4864c 224 WaitAndSetDone(int start_time, bool* done)
juliogerchman 8:77fd54b4864c 225 : done_(done) {
juliogerchman 8:77fd54b4864c 226 time_ = start_time;
juliogerchman 8:77fd54b4864c 227 Schedule(this);
juliogerchman 8:77fd54b4864c 228 }
juliogerchman 8:77fd54b4864c 229
juliogerchman 8:77fd54b4864c 230 virtual void Run() {
juliogerchman 8:77fd54b4864c 231 strip_changed = true;
juliogerchman 8:77fd54b4864c 232 if (done_) *done_ = true;
juliogerchman 8:77fd54b4864c 233 delete this;
juliogerchman 8:77fd54b4864c 234 }
juliogerchman 8:77fd54b4864c 235
juliogerchman 8:77fd54b4864c 236 private:
juliogerchman 8:77fd54b4864c 237 bool* done_;
juliogerchman 8:77fd54b4864c 238 };
juliogerchman 8:77fd54b4864c 239
juliogerchman 7:e3b6f5741d9d 240 class FadeFillRegion : public Schedulable {
juliogerchman 7:e3b6f5741d9d 241 public:
juliogerchman 8:77fd54b4864c 242 FadeFillRegion(int start_time, int start_led, int length, uint8_t from_color, uint8_t to_color, bool fast, bool* done)
juliogerchman 8:77fd54b4864c 243 : start_led_(start_led), length_(length), from_color_(from_color), to_color_(to_color), fast_(fast), done_(done) {
juliogerchman 7:e3b6f5741d9d 244 time_ = start_time;
juliogerchman 7:e3b6f5741d9d 245 Schedule(this);
juliogerchman 7:e3b6f5741d9d 246 }
juliogerchman 7:e3b6f5741d9d 247
juliogerchman 7:e3b6f5741d9d 248 virtual void Run() {
juliogerchman 7:e3b6f5741d9d 249 for (int i = start_led_; i < start_led_ + length_; ++i) {
juliogerchman 7:e3b6f5741d9d 250 strip[i] = getcolor(from_color_, to_color_) | (fast_ ? FAST : 0);
juliogerchman 7:e3b6f5741d9d 251 }
juliogerchman 7:e3b6f5741d9d 252 strip_changed = true;
juliogerchman 8:77fd54b4864c 253 new WaitAndSetDone(global_tick + (fast_ ? 256 : 512), done_);
juliogerchman 7:e3b6f5741d9d 254 delete this;
juliogerchman 7:e3b6f5741d9d 255 }
juliogerchman 8:77fd54b4864c 256
juliogerchman 7:e3b6f5741d9d 257 private:
juliogerchman 7:e3b6f5741d9d 258 int start_led_, length_;
juliogerchman 7:e3b6f5741d9d 259 uint8_t from_color_, to_color_;
juliogerchman 7:e3b6f5741d9d 260 bool fast_;
juliogerchman 8:77fd54b4864c 261 bool* done_;
juliogerchman 7:e3b6f5741d9d 262 };
juliogerchman 7:e3b6f5741d9d 263
juliogerchman 2:590e99f4a313 264 /* Keep dropping water drops in a bucket, until it fills up. */
bracz 4:5a2af711d510 265 class DropBucketFill : public Schedulable
bracz 4:5a2af711d510 266 {
juliogerchman 2:590e99f4a313 267 public:
juliogerchman 8:77fd54b4864c 268 DropBucketFill(int start_time, int start_led, int length, int drop_size, uint8_t from_color, uint8_t to_color, int fade_out_pause, uint8_t fade_out_color, bool* done)
juliogerchman 8:77fd54b4864c 269 : start_led_(start_led), length_(length), original_length_(length), drop_size_(drop_size), from_color_(from_color), to_color_(to_color), fade_out_pause_(fade_out_pause), fade_out_color_(fade_out_color), done_(done) {
juliogerchman 2:590e99f4a313 270 time_ = start_time;
juliogerchman 2:590e99f4a313 271 Schedule(this);
juliogerchman 2:590e99f4a313 272 }
bracz 4:5a2af711d510 273
juliogerchman 2:590e99f4a313 274 virtual void Run() {
juliogerchman 2:590e99f4a313 275 // The bucket starts with a drop at its end.
juliogerchman 2:590e99f4a313 276 //strip[start_led_ + length_] = to_color_;
juliogerchman 2:590e99f4a313 277 //strip[start_led_ + length_] = RED;
juliogerchman 2:590e99f4a313 278 if (length_ > 0) {
juliogerchman 2:590e99f4a313 279 // There's still space in the bucket. Drop a new drop.
juliogerchman 3:42efa00ffef4 280 for (int i = 0; i < min(drop_size_, length_); ++i) {
juliogerchman 7:e3b6f5741d9d 281 Schedulable* next_drop = this;
juliogerchman 3:42efa00ffef4 282 new RegionWalkingFadeInOut(time_ + (256 * i / drop_size_), i, drop_size_, start_led_, length_, from_color_, to_color_, true, false, true, next_drop);
juliogerchman 2:590e99f4a313 283 }
juliogerchman 2:590e99f4a313 284 length_--;
bracz 4:5a2af711d510 285 } else if (length_ == 0) {
juliogerchman 2:590e99f4a313 286 // There's no more space in the bucket. Bail out.
juliogerchman 8:77fd54b4864c 287 new FadeFillRegion(global_tick + fade_out_pause_, start_led_, original_length_, to_color_, fade_out_color_, false, done_);
juliogerchman 2:590e99f4a313 288 delete this;
juliogerchman 2:590e99f4a313 289 return;
juliogerchman 2:590e99f4a313 290 }
juliogerchman 2:590e99f4a313 291 }
bracz 4:5a2af711d510 292
juliogerchman 2:590e99f4a313 293 private:
juliogerchman 7:e3b6f5741d9d 294 int start_led_, length_, original_length_;
juliogerchman 2:590e99f4a313 295 int drop_size_;
juliogerchman 2:590e99f4a313 296 uint8_t from_color_, to_color_;
juliogerchman 7:e3b6f5741d9d 297 int fade_out_pause_;
juliogerchman 7:e3b6f5741d9d 298 uint8_t fade_out_color_;
bracz 4:5a2af711d510 299 bool* done_;
juliogerchman 2:590e99f4a313 300 };
juliogerchman 2:590e99f4a313 301
bracz 4:5a2af711d510 302 Schedulable* g_watchdog;
bracz 4:5a2af711d510 303
bracz 4:5a2af711d510 304 class ProgramSupervisor
bracz 4:5a2af711d510 305 {
bracz 4:5a2af711d510 306 public:
juliogerchman 8:77fd54b4864c 307 ProgramSupervisor() {
juliogerchman 8:77fd54b4864c 308 next_program_ = 0;
juliogerchman 8:77fd54b4864c 309 }
bracz 4:5a2af711d510 310 // The program should be alive forever. Programs do not delete themselves upon completion. Every scheduling of a program happens with global time set to zero.
bracz 4:5a2af711d510 311 void RegisterProgram(Schedulable* program) {
bracz 4:5a2af711d510 312 programs_.push_back(program);
bracz 4:5a2af711d510 313 }
bracz 4:5a2af711d510 314
bracz 4:5a2af711d510 315 // This should be called by the program executor when it is completed.
bracz 4:5a2af711d510 316 // It is desired to leave all LEDs black after program completion.
bracz 4:5a2af711d510 317 void CurrentProgramDone() {
bracz 4:5a2af711d510 318 next_program_++;
bracz 4:5a2af711d510 319 next_program_ %= programs_.size();
bracz 4:5a2af711d510 320 ScheduleProgram();
bracz 4:5a2af711d510 321 }
bracz 4:5a2af711d510 322
bracz 4:5a2af711d510 323 void ScheduleProgram() {
bracz 4:5a2af711d510 324 global_tick = 0;
bracz 4:5a2af711d510 325 while (!task_list.empty()) task_list.pop();
bracz 4:5a2af711d510 326 Schedule(g_watchdog);
bracz 6:62c9a5483a84 327 memset(strip, 0x80, sizeof(strip));
bracz 6:62c9a5483a84 328 write_strip(strip, sizeof(strip));
bracz 4:5a2af711d510 329 if (programs_.empty()) return;
juliogerchman 8:77fd54b4864c 330 programs_[next_program_]->time_ = 0;
juliogerchman 8:77fd54b4864c 331 Schedule(programs_[next_program_]);
bracz 4:5a2af711d510 332 }
bracz 4:5a2af711d510 333
bracz 4:5a2af711d510 334
bracz 4:5a2af711d510 335 private:
bracz 4:5a2af711d510 336 vector<Schedulable*> programs_;
bracz 4:5a2af711d510 337 int next_program_; // indexes the programs_ array.
bracz 4:5a2af711d510 338 } supervisor;
bracz 4:5a2af711d510 339
bracz 4:5a2af711d510 340
bracz 4:5a2af711d510 341 class ProgramWatchdog : public Schedulable
bracz 4:5a2af711d510 342 {
bracz 4:5a2af711d510 343 public:
bracz 4:5a2af711d510 344 ProgramWatchdog() {
bracz 4:5a2af711d510 345 time_ = 5 * 60 * 1000; // 5 minutes deadline
bracz 4:5a2af711d510 346 }
bracz 4:5a2af711d510 347
bracz 4:5a2af711d510 348 virtual void Run() {
bracz 4:5a2af711d510 349 supervisor.CurrentProgramDone();
bracz 4:5a2af711d510 350 led2 = 1;
bracz 4:5a2af711d510 351 }
bracz 4:5a2af711d510 352
bracz 4:5a2af711d510 353 } g_watchdog_impl;
bracz 4:5a2af711d510 354
bracz 4:5a2af711d510 355
bracz 4:5a2af711d510 356 class MultiDropBucketFillProgram : public Schedulable
bracz 4:5a2af711d510 357 {
bracz 4:5a2af711d510 358 public:
bracz 5:6f0f4e6c1e5f 359 MultiDropBucketFillProgram() {
bracz 5:6f0f4e6c1e5f 360 supervisor.RegisterProgram(this);
bracz 5:6f0f4e6c1e5f 361 }
bracz 5:6f0f4e6c1e5f 362
bracz 4:5a2af711d510 363 virtual void Run() {
bracz 4:5a2af711d510 364 int time = global_tick;
bracz 4:5a2af711d510 365 const int kLength = sizeof(google_colors);
bracz 4:5a2af711d510 366 memset(done_, 0, sizeof(done_));
bracz 4:5a2af711d510 367 for (int i = 0; i < kLength; i++) {
juliogerchman 8:77fd54b4864c 368 new DropBucketFill(time, 10 + i * 20, 20, 4, BLACK, google_colors[i], 2560, BLACK, done_ + i);
bracz 4:5a2af711d510 369 }
bracz 4:5a2af711d510 370 new EndWatcher(done_, kLength);
bracz 4:5a2af711d510 371 }
bracz 4:5a2af711d510 372
bracz 4:5a2af711d510 373 private:
bracz 4:5a2af711d510 374 class EndWatcher : public Schedulable {
bracz 4:5a2af711d510 375 public:
bracz 4:5a2af711d510 376 EndWatcher(bool* done_array, int len) : done_(done_array), len_(len) {
bracz 4:5a2af711d510 377 time_ = 0;
bracz 4:5a2af711d510 378 Schedule(this);
bracz 4:5a2af711d510 379 }
bracz 4:5a2af711d510 380
bracz 4:5a2af711d510 381 virtual void Run() {
bracz 4:5a2af711d510 382 int i;
bracz 4:5a2af711d510 383 for (i = 0; i < len_ && done_[i]; i++);
bracz 4:5a2af711d510 384 if (i < len_) {
bracz 4:5a2af711d510 385 // not done yet.
bracz 4:5a2af711d510 386 time_ = global_tick + 2;
bracz 4:5a2af711d510 387 Schedule(this);
bracz 4:5a2af711d510 388 } else {
bracz 4:5a2af711d510 389 supervisor.CurrentProgramDone();
bracz 4:5a2af711d510 390 delete this;
bracz 4:5a2af711d510 391 }
bracz 4:5a2af711d510 392 }
bracz 4:5a2af711d510 393
bracz 4:5a2af711d510 394 private:
bracz 4:5a2af711d510 395 bool* done_;
bracz 4:5a2af711d510 396 int len_;
bracz 4:5a2af711d510 397 };
bracz 4:5a2af711d510 398
bracz 4:5a2af711d510 399
bracz 4:5a2af711d510 400 bool done_[6];
bracz 4:5a2af711d510 401 };
bracz 4:5a2af711d510 402
bracz 4:5a2af711d510 403
bracz 4:5a2af711d510 404 void init_board()
bracz 4:5a2af711d510 405 {
bracz 0:109a7a5e3e6e 406 pc.baud(115200);
bracz 0:109a7a5e3e6e 407
bracz 0:109a7a5e3e6e 408 myled = 0;
bracz 0:109a7a5e3e6e 409 latch = 0;
bracz 4:5a2af711d510 410
bracz 0:109a7a5e3e6e 411 spi.format(8, 0);
bracz 0:109a7a5e3e6e 412 spi.frequency(300000);
bracz 0:109a7a5e3e6e 413 wait_ms(500);
bracz 0:109a7a5e3e6e 414 myled = 1;
juliogerchman 2:590e99f4a313 415 memset(strip, BLACK, sizeof(strip));
bracz 0:109a7a5e3e6e 416 write_strip(strip, sizeof(strip));
juliogerchman 2:590e99f4a313 417 g_ticker.attach(&tick_cb, 1.0/1000);
bracz 4:5a2af711d510 418
bracz 0:109a7a5e3e6e 419 memset(strip, 0x0, sizeof(strip));
juliogerchman 2:590e99f4a313 420 }
juliogerchman 2:590e99f4a313 421
bracz 4:5a2af711d510 422 void run_loop()
bracz 4:5a2af711d510 423 {
bracz 0:109a7a5e3e6e 424 while(1) {
bracz 0:109a7a5e3e6e 425 while (task_list.empty() || global_tick < task_list.top()->time_) {
bracz 0:109a7a5e3e6e 426 if (strip_changed) {
bracz 0:109a7a5e3e6e 427 write_strip(strip, sizeof(strip));
bracz 0:109a7a5e3e6e 428 strip_changed = false;
bracz 0:109a7a5e3e6e 429 memset(strip, 0x0, sizeof(strip));
bracz 0:109a7a5e3e6e 430 }
bracz 0:109a7a5e3e6e 431 }
bracz 0:109a7a5e3e6e 432 Schedulable* action = task_list.top();
bracz 0:109a7a5e3e6e 433 task_list.pop();
bracz 0:109a7a5e3e6e 434 action->Run();
bracz 0:109a7a5e3e6e 435 }
juliogerchman 2:590e99f4a313 436 }
bracz 0:109a7a5e3e6e 437
bracz 4:5a2af711d510 438 int main()
bracz 4:5a2af711d510 439 {
juliogerchman 2:590e99f4a313 440 init_board();
bracz 4:5a2af711d510 441 g_watchdog = &g_watchdog_impl;
bracz 5:6f0f4e6c1e5f 442
bracz 5:6f0f4e6c1e5f 443 MultiDropBucketFillProgram multi_drop;
bracz 5:6f0f4e6c1e5f 444
bracz 4:5a2af711d510 445 supervisor.ScheduleProgram();
juliogerchman 3:42efa00ffef4 446
juliogerchman 2:590e99f4a313 447 run_loop();
juliogerchman 2:590e99f4a313 448 }