123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706 |
- /*
- * main.c
- *
- * Copyright 2011 Thomas Buck <xythobuz@me.com>
- * Copyright 2011 Max Nuding <max.nuding@gmail.com>
- * Copyright 2011 Felix Bäder <baeder.felix@gmail.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 "serial.h"
- #include "cube.h"
- #include "time.h"
- #include "audio.h"
- #include "mem.h"
- #include "memLayer.h"
- #include "twi.h"
- #include "strings.h"
-
- #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))
-
- // Length of an idle animation frame, 24 -> 1 second
- #define IDLELENGTH 24
-
- void serialHandler(char c);
- void sendAudioData(void);
- void recieveAnimations(void);
- void transmitAnimations(void);
- uint8_t audioModeSelected(void);
- void setPixelBuffer(uint8_t x, uint8_t y, uint8_t z, uint8_t *buf);
- void setRow(uint8_t x, uint8_t z, uint8_t height, uint8_t *buf);
- void visualizeAudioData(uint8_t *audioData, uint8_t *imageData);
- #ifdef DEBUG
- void printErrors(uint8_t e);
- uint8_t selfTest(void);
- void printTime(void);
-
- #include "snake.c"
- #endif
-
- uint8_t refreshAnimationCount = 1;
- uint8_t lastButtonState = 0;
- char buffer[11];
-
- uint8_t defaultImage[64] = { 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
- 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
- 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
- 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
- 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
- 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
- 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
- 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF };
-
- uint8_t DebugDone = 0; // Bit 0: 10s int. count, Bit 1: switch idle display
- // Bit 2: state changed, disable idle
-
- int main(void) {
- uint8_t *audioData = NULL;
- uint8_t *imageData = NULL;
- uint8_t i, length = 0, lastMode;
- uint16_t count;
- uint64_t lastChecked;
- uint32_t temp;
-
- initCube();
- serialInit(25, 8, NONE, 1);
- i2c_init();
- initSystemTimer();
- sei(); // Enable Interrupts
-
- DDRD = 0xFC; // Mosfets as Output
- DDRB = 0xFE;
- DDRC = 0xFF; // Latch Enable
- DDRA = 0xFF; // Latch Data
-
- setImage(defaultImage); // Display something
-
- #ifdef DEBUG
- // Kill animation counter in debug mode
- // => Don't preserve animations while power down
- setAnimationCount(0);
-
- i = selfTest();
- if (i) {
- serialWriteString(getString(1));
- serialWriteString(itoa(i, buffer, 2));
- serialWrite('\n');
- printErrors(i);
- }
- #endif
-
- #ifdef DEBUG
- serialWriteString(getString(2));
- serialWriteString(getString(0));
- serialWriteString("Took ");
- serialWriteString(itoa(getSystemTime(), buffer, 10));
- serialWriteString(" ms!\n");
- #endif
-
- lastMode = audioModeSelected();
- lastChecked = getSystemTime();
-
- i = 0;
- count = getAnimationCount();
- while (1) {
- if(lastMode) {
- // Get Audio Data and visualize it
- if (isFinished()) {
- audioData = getAudioData();
- if (audioData != NULL) {
- imageData = (uint8_t *)malloc(64);
- if (imageData == NULL) {
- #ifdef DEBUG
- serialWriteString(getString(24));
- #endif
- while(1);
- }
- visualizeAudioData(audioData, imageData);
- setImage(imageData);
- free(imageData);
- }
- }
- } else {
- if (refreshAnimationCount) {
- // Get animation count stored in FRAM via TWI, if needed
- count = getAnimationCount();
- refreshAnimationCount = 0;
- i = 0;
- }
-
- if (count > 0) {
- if (isFinished() > length) {
- // Load next image
- if (i < (count - 1)) {
- i++;
- } else {
- i = 0;
- }
-
- imageData = getFrame(i);
- length = imageData[64];
- setImage(imageData);
- free(imageData);
- }
- } else {
- if (!(DebugDone & 4)) {
- if (isFinished() >= IDLELENGTH) {
- // Should happen every half second
- if (DebugDone & 2) {
- fillBuffer(0);
- DebugDone &= ~(2);
- } else {
- fillBuffer(0xFF);
- DebugDone |= 2;
- }
- }
- }
- }
- }
-
- if (serialHasChar()) {
- serialHandler((char)(serialGet()));
- }
-
- if ((getSystemTime() >= 1000) && ((DebugDone & 1) == 0)) {
- temp = getTriggerCount();
- serialWriteString(ltoa(temp, buffer, 10));
- serialWriteString(getString(27));
- serialWriteString(ltoa((temp / 8), buffer, 10));
- serialWriteString(getString(28));
- DebugDone |= 1;
- }
-
- if ((getSystemTime() - lastChecked) > 150) {
- lastMode = audioModeSelected();
- lastChecked = getSystemTime();
- }
- }
-
- close();
- return 0;
- }
-
- #ifdef DEBUG
- uint8_t selfTest(void) {
- uint8_t result = NOERROR;
-
- uint8_t *data = getAudioData();
- if (data == NULL) {
- result |= AUDIOERROR;
- } else {
- free(data);
- }
-
- data = memGetBytes(0, 1);
- if (data == NULL) {
- result |= MEMORYERROR;
- } else {
- free(data);
- }
-
- setGeneralPurposeByte(0, 0x42);
- if (getGeneralPurposeByte(0) != 0x42) {
- result |= MEMORYWRITEERROR;
- }
-
- return result;
- }
-
- void printErrors(uint8_t e) {
- if (ISERROR(e, AUDIOERROR)) {
- serialWriteString(getString(3));
- }
- if (ISERROR(e, MEMORYERROR)) {
- serialWriteString(getString(4));
- }
- if (ISERROR(e, MEMORYWRITEERROR)) {
- serialWriteString(getString(5));
- }
- }
- #endif
-
- void serialHandler(char c) {
- // Used letters:
- // a, c, d, e, g, n, s, t, v, x, 0, 1, 2
- uint8_t i, y, z;
- #ifdef DEBUG
- serialWrite(c);
- serialWriteString(": ");
- #endif
-
- switch(c) {
- case OK:
- serialWrite(OK);
- break;
-
- case 'h': case 'H': case '?':
- serialWriteString(getString(6));
- #ifdef DEBUG
- serialWriteString(getString(7));
- serialWriteString(getString(8));
- serialWriteString(getString(9));
- serialWriteString(getString(10));
- serialWriteString(getString(11));
- serialWriteString(getString(12));
- serialWriteString(getString(13));
- serialWriteString(getString(26));
- #endif
- break;
-
- case 'd': case 'D':
- clearMem();
- #ifndef DEBUG
- serialWrite(OK);
- #endif
- #ifdef DEBUG
- serialWriteString(getString(29));
- #endif
- break;
-
- #ifndef DEBUG
- case 'g': case 'G':
- transmitAnimations();
- break;
-
- case 's': case 'S':
- recieveAnimations();
- break;
- #endif
-
- case 'v': case 'V':
- serialWriteString(getString(0));
- break;
-
- #ifdef DEBUG
- case 't': case 'T':
- printTime();
- break;
-
- case 'a': case 'A':
- sendAudioData();
- break;
-
- case 'c': case 'C':
- serialWriteString(itoa(getAnimationCount(), buffer, 10));
- serialWriteString(getString(15));
- break;
-
- case 'x': case 'X':
- // Get byte, store as animation count
- serialWriteString(getString(16));
- while (!serialHasChar());
- c = serialGet();
- setAnimationCount(c);
- serialWriteString(itoa(c, buffer, 10));
- serialWriteString(getString(17));
- break;
-
- case 'y': case 'Y':
- setAnimationCount(0x2201);
- serialWriteString(getString(18));
- break;
-
- case 'e': case 'E':
- c = selfTest();
- serialWriteString(getString(19));
- serialWriteString(itoa(c, buffer, 2));
- serialWrite('\n');
- printErrors(c);
- break;
-
- case 'n': case 'N':
- snake();
- break;
-
- case '0':
- fillBuffer(0);
- setAnimationCount(0);
- refreshAnimationCount = 1;
- serialWriteString(getString(20));
- DebugDone |= 4;
- break;
-
- case '1':
- fillBuffer(0xFF);
- setAnimationCount(0);
- refreshAnimationCount = 1;
- serialWriteString(getString(20));
- DebugDone |= 4;
- break;
-
- case '2':
- DebugDone |= 4;
- fillBuffer(0);
- for (i = 0; i < 64; i++) {
- defaultImage[i] = 0;
- }
- while(1) {
- for (i = 0; i < 8; i++) {
- for (y = 0; y < 8; y++) {
- defaultImage[y + (i * 8)] = 0;
- for (z = 0; z < 8; z++) {
- defaultImage[y + (i * 8)] |= (1 << z);
- setImage(defaultImage);
- while (isFinished() == 0) {
- if (serialHasChar()) {
- goto killMeForIt; // Yes I know...
- // But I need to break out of 2 while Loops...
- }
- }
- }
- defaultImage[y + (i * 8)] = 0;
- }
- }
- }
- break;
- killMeForIt:
- serialGet();
- serialWriteString(getString(25));
- break;
-
- case 'I': case 'i':
- serialWriteString(ltoa(getTriggerCount(), buffer, 10));
- serialWrite('\n');
- break;
- #endif
-
- default:
- serialWrite(ERROR);
- break;
- }
- // c was used as temp var and does not contain the char anymore...!
- }
-
- #ifdef DEBUG
- void printTime(void) {
- serialWriteString(getString(14));
- serialWriteString(ltoa(getSystemTime(), buffer, 10));
- serialWriteString("ms");
- if (getSystemTime() > 1000) {
- serialWriteString(" (");
- serialWriteString(itoa(getSystemTime() / 1000, buffer, 10));
- itoa(getSystemTime() % 1000, buffer, 10);
- if (buffer[0] != '\0')
- serialWrite('.');
- if (buffer[2] == '\0')
- serialWrite('0');
- if (buffer[1] == '\0')
- serialWrite('0');
- if (buffer[0] != '\0')
- serialWriteString(buffer);
- serialWriteString("s)\n");
- } else {
- serialWrite('\n');
- }
- }
-
- void sendAudioData(void) {
- uint8_t i;
- uint8_t *audioData = getAudioData();
- if (audioData == NULL) {
- serialWriteString(getString(21));
- } else {
- serialWriteString(getString(22));
- for (i = 0; i < 7; i++) {
- serialWrite(i + '0');
- serialWriteString(": ");
- itoa(audioData[i], buffer, 10);
- serialWriteString(buffer);
- serialWrite('\n');
- }
- }
- }
- #endif
-
- void recieveAnimations() {
- uint8_t animCount, a, frameCount, f, i;
- uint16_t completeCount = 0, character;
- uint8_t frame[65];
-
- serialWrite(OK); // We are ready...
-
- character = serialGet();
- while (character & 0xFF00) { // Wait for answer
- character = serialGet();
- }
-
- animCount = (uint8_t)(character & 0x00FF); // Got animation count
- serialWrite(OK);
-
- for (a = 0; a < animCount; a++) {
- character = serialGet();
- while (character & 0xFF00) { // Wait for answer
- character = serialGet();
- }
-
- frameCount = (uint8_t)(character & 0x00FF); // Got frame count
- serialWrite(OK);
-
- for (f = 0; f < frameCount; f++) {
- character = serialGet();
- while (character & 0xFF00) { // Wait for answer
- character = serialGet();
- }
-
- frame[64] = (uint8_t)(character & 0x00FF); // Got duration
- serialWrite(OK);
-
- for (i = 0; i < 64; i++) {
- character = serialGet();
- while (character & 0xFF00) { // Wait for answer
- character = serialGet();
- }
-
- frame[i] = (uint8_t)(character & 0x00FF); // Got data byte
- }
- serialWrite(OK);
-
- setFrame(completeCount++, frame);
- }
- }
-
- character = serialGet();
- while (character & 0xFF00) { // Wait for answer
- character = serialGet();
- }
- character = serialGet();
- while (character & 0xFF00) { // Wait for answer
- character = serialGet();
- }
- character = serialGet();
- while (character & 0xFF00) { // Wait for answer
- character = serialGet();
- }
- character = serialGet();
- while (character & 0xFF00) { // Wait for answer
- character = serialGet();
- }
- serialWrite(OK);
- setAnimationCount(completeCount);
- refreshAnimationCount = 1;
- }
-
- void transmitAnimations() {
- // We store no animation information in here
- // So we have to place all frames in one or more
- // animations... We need 8 animations max...
- uint8_t animationsToGo;
- uint16_t framesToGo = getAnimationCount();
- uint16_t character;
- uint8_t a;
- uint8_t f, fMax, i;
- uint8_t *frame;
-
- if ((framesToGo % 255) == 0) {
- animationsToGo = framesToGo / 255;
- } else {
- animationsToGo = (framesToGo / 255) + 1;
- }
-
- serialWrite(OK);
- serialWrite(animationsToGo);
- while ((character = serialGet()) & 0xFF00); // Wait for answer
- if ((character & 0x00FF) != OK) { // Error code recieved
- return;
- }
-
- for (a = 0; a < animationsToGo; a++) {
- if (framesToGo > 255) {
- fMax = 255;
- } else {
- fMax = framesToGo;
- }
-
- serialWrite(fMax); // Number of Frames in current animation
- while ((character = serialGet()) & 0xFF00); // Wait for answer
- if ((character & 0x00FF) != OK) { // Error code recieved
- return;
- }
-
- for (f = 0; f < fMax; f++) {
- frame = getFrame(f + (255 * a));
-
- serialWrite(frame[64]); // frame duration
- while ((character = serialGet()) & 0xFF00); // Wait for answer
- if ((character & 0x00FF) != OK) { // Error code recieved
- free(frame);
- return;
- }
-
- for (i = 0; i < 64; i++) {
- serialWrite(frame[i]);
- }
- while ((character = serialGet()) & 0xFF00); // Wait for answer
- if ((character & 0x00FF) != OK) { // Error code recieved
- free(frame);
- return;
- }
-
- free(frame);
- }
- framesToGo -= fMax;
- }
-
- serialWrite(OK);
- serialWrite(OK);
- serialWrite(OK);
- serialWrite(OK);
-
- while ((character = serialGet()) & 0xFF00); // Wait for answer
- // Error code ignored...
- }
-
- uint8_t audioModeSelected(void) {
- // Pushbutton: PB0, Low active
-
- if (!(PINB & (1 << PB0))) {
- // Button pushed
- if (lastButtonState == 0) {
- lastButtonState = 1;
- } else {
- lastButtonState = 0;
- }
-
- #ifdef DEBUG
- serialWriteString("New State (");
- serialWriteString(itoa(lastButtonState, buffer, 10));
- serialWriteString(")\n");
- #endif
-
- }
- return lastButtonState;
- }
-
- void setRow(uint8_t x, uint8_t z, uint8_t height, uint8_t *buf) {
- uint8_t i = 0;
- for (; i < height; i++) {
- setPixelBuffer(x, i, z, buf);
- }
- }
-
- void setPixelBuffer(uint8_t x, uint8_t y, uint8_t z, uint8_t *buf) {
- buf[(8 * z) + y] |= (1 << x);
- }
-
- void visualizeAudioData(uint8_t *audioData, uint8_t *imageData) {
- uint8_t i;
- for (i = 0; i < 64; i++) {
- imageData[i] = 0;
- }
-
- // 8 LEDs, Max Val 255:
- // 256 / 8 = 32 => Divide by 31 (FACTOR) to get num of leds to light
- // 255 / FACTOR = 8,...
- // 127 / FACTOR = 4,...
-
- #define FACTOR 31
-
- // Could not figure out a way to represent this easily in a loop
- // without using a shitload of 'if's...
- setRow(0, 0, (audioData[0] / FACTOR), imageData);
- setRow(0, 1, (audioData[0] / FACTOR), imageData);
- setRow(1, 0, (audioData[0] / FACTOR), imageData);
-
- setRow(0, 2, (audioData[1] / FACTOR), imageData);
- setRow(0, 3, (audioData[1] / FACTOR), imageData);
- setRow(1, 1, (audioData[1] / FACTOR), imageData);
- setRow(1, 2, (audioData[1] / FACTOR), imageData);
- setRow(2, 0, (audioData[1] / FACTOR), imageData);
- setRow(2, 1, (audioData[1] / FACTOR), imageData);
-
- setRow(0, 4, (audioData[2] / FACTOR), imageData);
- setRow(0, 5, (audioData[2] / FACTOR), imageData);
- setRow(1, 3, (audioData[2] / FACTOR), imageData);
- setRow(1, 4, (audioData[2] / FACTOR), imageData);
- setRow(2, 2, (audioData[2] / FACTOR), imageData);
- setRow(2, 3, (audioData[2] / FACTOR), imageData);
- setRow(3, 0, (audioData[2] / FACTOR), imageData);
- setRow(3, 1, (audioData[2] / FACTOR), imageData);
- setRow(3, 2, (audioData[2] / FACTOR), imageData);
- setRow(4, 0, (audioData[2] / FACTOR), imageData);
- setRow(4, 1, (audioData[2] / FACTOR), imageData);
-
- setRow(0, 6, (audioData[3] / FACTOR), imageData);
- setRow(0, 7, (audioData[3] / FACTOR), imageData);
- setRow(1, 5, (audioData[3] / FACTOR), imageData);
- setRow(1, 6, (audioData[3] / FACTOR), imageData);
- setRow(2, 4, (audioData[3] / FACTOR), imageData);
- setRow(2, 5, (audioData[3] / FACTOR), imageData);
- setRow(3, 3, (audioData[3] / FACTOR), imageData);
- setRow(3, 4, (audioData[3] / FACTOR), imageData);
- setRow(4, 2, (audioData[3] / FACTOR), imageData);
- setRow(4, 3, (audioData[3] / FACTOR), imageData);
- setRow(5, 0, (audioData[3] / FACTOR), imageData);
- setRow(5, 1, (audioData[3] / FACTOR), imageData);
- setRow(5, 2, (audioData[3] / FACTOR), imageData);
- setRow(6, 0, (audioData[3] / FACTOR), imageData);
- setRow(6, 1, (audioData[3] / FACTOR), imageData);
-
- setRow(1, 7, (audioData[4] / FACTOR), imageData);
- setRow(2, 6, (audioData[4] / FACTOR), imageData);
- setRow(2, 7, (audioData[4] / FACTOR), imageData);
- setRow(3, 5, (audioData[4] / FACTOR), imageData);
- setRow(3, 6, (audioData[4] / FACTOR), imageData);
- setRow(4, 4, (audioData[4] / FACTOR), imageData);
- setRow(4, 5, (audioData[4] / FACTOR), imageData);
- setRow(5, 3, (audioData[4] / FACTOR), imageData);
- setRow(5, 4, (audioData[4] / FACTOR), imageData);
- setRow(6, 2, (audioData[4] / FACTOR), imageData);
- setRow(6, 3, (audioData[4] / FACTOR), imageData);
- setRow(7, 0, (audioData[4] / FACTOR), imageData);
- setRow(7, 1, (audioData[4] / FACTOR), imageData);
-
- setRow(3, 7, (audioData[5] / FACTOR), imageData);
- setRow(4, 6, (audioData[5] / FACTOR), imageData);
- setRow(4, 7, (audioData[5] / FACTOR), imageData);
- setRow(5, 5, (audioData[5] / FACTOR), imageData);
- setRow(5, 6, (audioData[5] / FACTOR), imageData);
- setRow(6, 4, (audioData[5] / FACTOR), imageData);
- setRow(6, 5, (audioData[5] / FACTOR), imageData);
- setRow(7, 2, (audioData[5] / FACTOR), imageData);
- setRow(7, 3, (audioData[5] / FACTOR), imageData);
- setRow(7, 4, (audioData[5] / FACTOR), imageData);
-
- setRow(5, 7, (audioData[6] / FACTOR), imageData);
- setRow(6, 6, (audioData[6] / FACTOR), imageData);
- setRow(6, 7, (audioData[6] / FACTOR), imageData);
- setRow(7, 5, (audioData[6] / FACTOR), imageData);
- setRow(7, 6, (audioData[6] / FACTOR), imageData);
- setRow(7, 7, (audioData[6] / FACTOR), imageData);
- }
|