123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540 |
- /*
- * main.c
- *
- * Copyright 2012 Thomas Buck <xythobuz@me.com>
- *
- * This file is part of LED-Cube.
- *
- * LED-Cube 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
- * (at your option) any later version.
- *
- * LED-Cube 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 LED-Cube. If not, see <http://www.gnu.org/licenses/>.
- */
-
- #ifndef F_CPU
- #define F_CPU 16000000L
- #endif
-
- #define OK 0x42
- #define ERROR 0x23
-
- #include <avr/io.h>
- #include <util/delay.h>
- #include <avr/interrupt.h>
- #include <avr/pgmspace.h>
- #include <stdint.h>
- #include <stdlib.h>
- #include <avr/wdt.h>
-
- #include "serial.h"
- #include "cube.h"
- #include "time.h"
- #include "audio.h"
- #include "mem.h"
- #include "memLayer.h"
- #include "twi.h"
- #include "strings.h"
- #include "visualizer.h"
- #include "animations.h"
- #include "transmit.h"
- #include "font.h"
- #include "buffhelp.h"
-
- // Length of an idle animation frame, 24 -> 1 second
- #define IDLELENGTH 48
-
- void init(void);
- uint8_t selfTest(void);
- void serialHandler(char c);
-
- uint8_t defaultImageCube[64] = { 0x81, 0x42, 0x24, 0x18, 0x18, 0x24, 0x42, 0x81,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x81, 0x42, 0x24, 0x18, 0x18, 0x24, 0x42, 0x81,
- 0x81, 0x42, 0x24, 0x18, 0x18, 0x24, 0x42, 0x81,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x81, 0x42, 0x24, 0x18, 0x18, 0x24, 0x42, 0x81 };
-
- #define NOERROR 0
- // Audio does not answer
- #define AUDIOERROR 1
- // Memory does not answer
- #define MEMORYERROR 2
- // Memory not writeable
- #define MEMORYWRITEERROR 4
- // x = errorcode, e = error definition, not NOERROR
- #define ISERROR(x, e) ((x) & (e))
-
- uint8_t shouldRestart = 0;
- uint8_t refreshAnimationCount = 0;
- uint8_t lastButtonState = 0;
- uint8_t maxButtonState = 1; // No audio, get's checked later
- uint8_t disableAudioData = 0;
- uint8_t disableMemory = 0;
- uint8_t disableAnim = 0;
-
- /* Not so powerful Debouncing Example
- * No Interrupt needed
- * Author: Peter Dannegger
- */
- #define debounce(port, pin) ({ \
- static uint8_t flag = 0; /* new variable on every macro usage */ \
- uint8_t i = 0; \
- if (flag) { /* check for key release: */ \
- for(;;) { /* loop ... */ \
- if(!(port & 1 << pin)) { /* ... until key pressed or ... */ \
- i = 0; \
- break; \
- } \
- _delay_us(98); /* * 256 = 25ms */ \
- if(--i == 0) { /* ... until key >25ms released */ \
- flag = 0; /* clear press flag */ \
- i = 0; /* 0 = key release debounced */ \
- break; \
- } \
- } \
- } else { /* else check for key press: */ \
- for(;;) { /* loop ... */ \
- if ((port & 1<<pin)) { /* ... until key released or ... */ \
- i = 0; \
- break; \
- } \
- _delay_us( 98 ); /* * 256 = 25ms */ \
- if (--i == 0) { /* ... until key >25ms pressed */ \
- flag = 1; /* set press flag */ \
- i = 1; /* 1 = key press debounced */ \
- break; \
- } \
- } \
- } \
- i; /* return value of Macro */ \
- })
-
- int main(void) {
- /*
- Quick Summary of the jobs main has to do:
- - Initialize Cube
- - Regularly check the buttonstate
- - Check for incoming serial transmission and run serialHandler
- - Check if animations are stored
- --> If yes, display them
- --> If no, display our idle animations
- */
-
- uint8_t i;
- uint16_t imageIndex = 0, imageCount = 0;
- uint8_t idleIndex = 0, idleCount = 0;
- uint8_t *imageData = NULL, *audioData = NULL;
- uint8_t duration = 0;
-
- // Initialization:
- MCUCSR = 0; // Reset Watchdog
- wdt_disable();
- init(); // Initialize all subsystems, set port directions, enable interrupts
- wdt_enable(WDTO_1S); // Watchdog reset after 1 second
- i = selfTest(); // Run selftest, print errors
- // Disable subsystems if they are unavailable
- if (ISERROR(i, AUDIOERROR))
- disableAudioData = 1;
- if (ISERROR(i, MEMORYERROR) || ISERROR(i, MEMORYWRITEERROR))
- disableMemory = 1;
- serialWriteString(getString(0)); // Print Version
-
- if (disableMemory == 0) {
- imageCount = getAnimationCount(); // Retrieve image count from memory
- lastButtonState = 0;
- } else {
- lastButtonState = 1;
- }
- if (disableAudioData == 0) {
- maxButtonState = numberOfVisualizations() + 1; // Number of toggle steps for button
- } else {
- lastButtonState = 0;
- maxButtonState = 1;
- // If both memory and audio are not working, we have nothing to do anyways...
- }
-
- idleCount = numOfAnimations();
-
- serialWriteString(getString(2));
-
- while(1) { // Our Mainloop
- if (!shouldRestart) { // A flag to trigger a watchdog reset
- wdt_reset();
- }
-
- if (refreshAnimationCount) {
- refreshAnimationCount = 0;
- if (disableMemory == 0)
- imageCount = getAnimationCount();
- }
-
- if (serialHasChar()) { // Run serialHandler
- serialHandler((char)(serialGet()));
- }
-
- if (debounce(PINB, PB0)) {
- if (lastButtonState < (maxButtonState - 1)) {
- lastButtonState++;
- } else {
- lastButtonState = 0;
- }
- }
-
- if (lastButtonState == 0) {
- // Display animations, stored or built-in
- if (disableAnim == 0) {
- if ((imageCount > 0) && (disableMemory == 0)) {
- // Memory frames
- if (isFinished() > duration) {
- // Last frame was displayed long enough
- imageIndex = (imageIndex < (imageCount - 1)) ? (imageIndex + 1) : 0;
- imageData = getFrame(imageIndex);
- if (imageData == NULL) {
- duration = 24;
- setImage(defaultImageCube);
- } else {
- duration = imageData[64];
- setImage(imageData);
- free(imageData);
- }
- }
- } else {
- // Built-In Anims
- if (isFinished() > duration) {
- idleIndex = (idleIndex < (idleCount - 1)) ? (idleIndex + 1) : 0;
- duration = executeAnimation(idleIndex);
- }
- }
- }
- } else {
- // An audiomode is selected
- if (disableAudioData == 0) {
- if (isFinished()) {
- audioData = getAudioData();
- if (audioData != NULL) {
- runVisualization(audioData, (lastButtonState - 1));
- } else {
- lastButtonState = 0;
- duration = 24;
- setImage(defaultImageCube); // Quasi Error Screen
- }
- }
- } else {
- lastButtonState = 0;
- }
- }
- }
-
- return 0; // We never reach this point, of course!
- }
-
- void init(void) {
- DDRA = 0xFF; // Latch Data Bus as Output
- DDRD = 0xFC; DDRB = 24; // Mosfets as Output
- DDRC = 0xFC; DDRB |= 6; // Latch Enable as Output
- DDRB &= ~(1 << PB0); // Pushbutton as Input
-
- initCube();
- serialInit(25, 8, NONE, 1);
- i2c_init();
- initSystemTimer();
- sei(); // Enable Interrupts
-
- setImage(defaultImageCube); // Display something
- }
-
- uint8_t selfTest(void) {
- uint8_t result = NOERROR;
-
- uint8_t *data = getAudioData();
- if (data == NULL) {
- result |= AUDIOERROR;
- } else {
- }
-
- data = memGetBytes(0, 1);
- if (data == NULL) {
- result |= MEMORYERROR;
- } else {
- free(data);
- }
-
- setGeneralPurposeByte(0, 0x23);
- if (getGeneralPurposeByte(0) != 0x23) {
- result |= MEMORYWRITEERROR;
- }
-
- if (result) { // Error in Selftest
- serialWriteString(getString(1));
- serialWriteString(itoa(result, buffer, 2));
- serialWrite('\n');
- if (ISERROR(result, AUDIOERROR)) {
- serialWriteString(getString(3));
- }
- if (ISERROR(result, MEMORYERROR)) {
- serialWriteString(getString(4));
- }
- if (ISERROR(result, MEMORYWRITEERROR)) {
- serialWriteString(getString(5));
- }
- }
-
- return result;
- }
-
- void randomAnimation(void) {
- uint8_t *b = (uint8_t *)malloc(64);
- uint8_t x, y, z;
- if (b == NULL) {
- serialWriteString(getString(24));
- return;
- }
- for (x = 0; x < 64; x++) {
- b[x] = 0;
- }
- while(1) {
- wdt_reset();
- setImage(b);
- while(isFinished() == 0);
- x = rand() / 4096;
- y = rand() / 4096;
- z = rand() / 4096;
- b[x + (8 * y)] ^= (1 << z);
-
- if (serialHasChar()) {
- serialWriteString(getString(25));
- free(b);
- serialGet();
- return;
- }
- }
- free(b);
- }
-
- void serialHandler(char c) {
- // Used letters:
- // a, b, c, d, e, f, g, h, i, m, n, o, p, q, r, s, t, u, v, w, x, y, 0, 1, 2, 3, #
- uint8_t i, y, z;
- uint8_t *tmp;
-
- switch(c) {
- case OK:
- serialWrite(OK);
- break;
-
- case 'h': case 'H':
- serialWriteString(getString(6));
- serialWriteString(getString(7));
- serialWriteString(getString(8));
- serialWriteString(getString(9));
- serialWriteString(getString(10));
- serialWriteString(getString(26));
- serialWriteString(getString(34));
- serialWriteString(getString(17));
-
- serialWriteString(getString(11));
- serialWriteString(getString(12));
- serialWriteString(getString(13));
- break;
-
- case 'd': case 'D':
- setAnimationCount(0);
- serialWrite(OK);
- refreshAnimationCount = 1;
- break;
-
- case 'f': case 'F':
- serialWriteString(getString(32));
- clearMem();
- serialWriteString(getString(33));
- refreshAnimationCount = 1;
- break;
-
- case 'g': case 'G':
- transmitAnimations();
- break;
-
- case 's': case 'S':
- recieveAnimations();
- refreshAnimationCount = 1;
- break;
-
- case 'v': case 'V': case '?':
- serialWriteString(getString(0));
- break;
-
- case 'm': case 'M':
- if (lastButtonState < (maxButtonState - 1)) {
- lastButtonState++;
- } else {
- lastButtonState = 0;
- }
- if (lastButtonState) {
- writeNumber(lastButtonState, 10);
- serialWrite(' ');
- serialWriteString(getString(41));
- } else {
- serialWriteString(getString(40));
- }
- break;
-
- case 'q': case 'Q':
- shouldRestart = 1;
- serialWriteString(getString(30));
- break;
-
- case 'r': case 'R':
- randomAnimation();
- break;
-
- case 't': case 'T':
- printTime();
- break;
-
- case 'a': case 'A':
- sendAudioData();
- break;
-
- case 'c': case 'C':
- serialWriteString(itoa(getAnimationCount(), buffer, 10));
- serialWriteString(" (");
- serialWriteString(itoa(refreshAnimationCount, buffer, 10));
- serialWriteString(")");
- serialWriteString(getString(15));
- break;
-
- case 'u': case 'U':
- serialWriteString(getString(31));
- tmp = getFrame(readNumber(10));
- dumpFrame(tmp);
- free(tmp);
- break;
-
- case 'o': case 'O':
- serialWriteString(getString(36)); // Start:
- i = readNumber(10);
- serialWriteString(getString(39)); // Ende:
- y = readNumber(10);
- serialWriteString(getString(35)); // Duration:
- z = readNumber(10);
- for (c = i; c <= y; c++) {
- setDuration(c, z);
- }
- serialWriteString(getString(33)); // Done
- break;
-
- case 'w': case 'W':
- serialWriteString(getString(42)); // Text:
- textRenderInput();
- break;
-
- case 'x': case 'X':
- // Get byte, store as animation count
- serialWriteString(getString(16)); // "New animation count: "
- setAnimationCount(readNumber(10));
- refreshAnimationCount = 1;
- break;
-
- case 'p': case 'P':
- simpleAnimationInput();
- break;
-
- case 'y': case 'Y':
- serialWriteString(getString(18)); // "Frame to change: "
- i = readNumber(10);
- tmp = readAFrame();
- setFrame(i, tmp);
- free(tmp);
- break;
-
- case 'e': case 'E':
- selfTest();
- break;
-
- case 'n': case 'N':
- // snake();
- break;
-
- case '0':
- fillBuffer(0);
- disableAnim = 1;
- break;
-
- case '1':
- fillBuffer(0xFF);
- disableAnim = 1;
- break;
-
- case '2':
- buffClearAllPixels(defaultImageCube);
- while(1) {
- for (i = 0; i < 8; i++) {
- for (y = 0; y < 8; y++) {
- defaultImageCube[y + (i * 8)] = 0;
- for (z = 0; z < 8; z++) {
- defaultImageCube[y + (i * 8)] |= (1 << z);
- setImage(defaultImageCube);
- while (isFinished() == 0) {
- wdt_reset();
- if (serialHasChar()) {
- goto killMeForIt; // Yes I know...
- // But I need to break out of 2 while Loops...
- }
- }
- }
- // defaultImageCube[y + (i * 8)] = 0;
- }
- }
- }
- break;
- killMeForIt:
- serialGet(); // Killed because we got a serial char. Remove it from buffer.
- serialWriteString(getString(25));
- break;
-
- case '3':
- buffClearAllPixels(defaultImageCube);
- y = 0x20;
- while(1) {
- tmp = getFont(y);
- if (y < 0x7E) {
- y++;
- } else {
- y = 0x20;
- }
- for (i = 0; i < 8; i++) {
- defaultImageCube[i] = tmp[i];
- }
- setImage(defaultImageCube);
- while (isFinished() < 12) {
- wdt_reset();
- if (serialHasChar()) {
- goto killMeForIt; // Yes, yes. But i already need it for '2'
- // So why not do it again... Please, no raptors :)
- // http://xkcd.com/292/
- }
- }
- }
- break;
-
- case 'I': case 'i':
- serialWriteString(ltoa(getTriggerCount(), buffer, 10));
- serialWrite('\n');
- break;
-
- default:
- serialWrite(ERROR);
- break;
- }
- // c was used as temp var and does not contain the char anymore...!
- }
|