diff options
Diffstat (limited to 'usb-rainbow/usb-rainbow.cpp')
-rw-r--r-- | usb-rainbow/usb-rainbow.cpp | 105 |
1 files changed, 105 insertions, 0 deletions
diff --git a/usb-rainbow/usb-rainbow.cpp b/usb-rainbow/usb-rainbow.cpp new file mode 100644 index 0000000..6465e18 --- /dev/null +++ b/usb-rainbow/usb-rainbow.cpp @@ -0,0 +1,105 @@ +/* + * spreadspace avr utils + * + * + * Copyright (C) 2013-2016 Christian Pointner <equinox@spreadspace.org> + * + * This file is part of spreadspace avr utils. + * + * spreadspace avr utils is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * any later version. + * + * spreadspace avr utils is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with spreadspace avr utils. If not, see <http://www.gnu.org/licenses/>. + */ + +#include <avr/io.h> +#include <avr/wdt.h> +#include <avr/interrupt.h> +#include <avr/power.h> +#include <stdio.h> + +#include "util.h" +#include "led.h" +#include "usbio.h" + +#include "Arduino.h" +#include "FastLED.h" + + + + +#define NUM_LEDS 8 +#define DATA_PIN 1 +CRGB leds[NUM_LEDS]; + +void fastled_init() +{ + arduino_init(); + FastLED.addLeds<WS2812B, DATA_PIN, GRB>(leds, NUM_LEDS); + FastLED.setBrightness(10); +} + +void rainbow_init() +{ + fill_solid(leds, NUM_LEDS, CRGB::Black); + FastLED.show(); +} + +void rainbow_task() +{ + static uint8_t idx= 0; + for(int i = 0; i < NUM_LEDS; i++) { + uint8_t hue = ((80 / NUM_LEDS ) * i) + idx; + leds[i] = CHSV(hue, 255, 255); + } + + FastLED.show(); + delay(5); + idx++; +} + +void handle_cmd(uint8_t cmd) +{ + switch(cmd) { + case '0': led_off(); break; + case '1': led_on(); break; + case '!': reset2bootloader(); break; + default: printf("error\r\n"); return; + } + printf("ok\r\n"); +} + +int main(void) +{ + MCUSR &= ~(1 << WDRF); + wdt_disable(); + + cpu_init(); + led_init(); + usbio_init(); + fastled_init(); + sei(); + + rainbow_init(); + for(;;) { + int16_t BytesReceived = usbio_bytes_received(); + while(BytesReceived > 0) { + int ReceivedByte = fgetc(stdin); + if(ReceivedByte != EOF) { + handle_cmd(ReceivedByte); + } + BytesReceived--; + } + + usbio_task(); + rainbow_task(); + } +} |