Liyou Zhou / Mbed 2 deprecated light_painting_strip

A neopixel light painting strip

Dependencies:   NeoStrip PinDetect mbed

main.cpp@0:a26aca3e4760, 2015-11-10 (annotated)

Committer:
lz307
Date:
Tue Nov 10 21:28:51 2015 +0000
Revision:
0:a26aca3e4760
Initial commit

Who changed what in which revision?

UserRevisionLine numberNew contents of line
lz307 0:a26aca3e4760 1 #include "mbed.h"
lz307 0:a26aca3e4760 2 #include "BitmapFile.h"
lz307 0:a26aca3e4760 3 #include <string>
lz307 0:a26aca3e4760 4 #include "PinDetect.h"
lz307 0:a26aca3e4760 5 #include "NeoStrip.h"
lz307 0:a26aca3e4760 6
lz307 0:a26aca3e4760 7 # define MAX_FILE 30
lz307 0:a26aca3e4760 8
lz307 0:a26aca3e4760 9 #define N 32
lz307 0:a26aca3e4760 10 NeoStrip strip(p5, N);
lz307 0:a26aca3e4760 11 float bright = 0.2; // 20% is plenty for indoor use
lz307 0:a26aca3e4760 12
lz307 0:a26aca3e4760 13 PinDetect b1(p25, PullUp);
lz307 0:a26aca3e4760 14 PinDetect b2(p29, PullUp);
lz307 0:a26aca3e4760 15 PinDetect b3(p22, PullUp);
lz307 0:a26aca3e4760 16 PinDetect b4(p27, PullUp);
lz307 0:a26aca3e4760 17
lz307 0:a26aca3e4760 18 LocalFileSystem local("local"); // Create the local filesystem under the name "local"
lz307 0:a26aca3e4760 19 Serial pc(USBTX, USBRX); // tx, rx
lz307 0:a26aca3e4760 20 BitmapFile *MyBitmap;
lz307 0:a26aca3e4760 21
lz307 0:a26aca3e4760 22 char *f_list[MAX_FILE];
lz307 0:a26aca3e4760 23 int file_index = 0; // get the index of file being displayed
lz307 0:a26aca3e4760 24 unsigned max_file = 0; // register how many files there are
lz307 0:a26aca3e4760 25
lz307 0:a26aca3e4760 26 char * get_file_name() {
lz307 0:a26aca3e4760 27 char *fn = NULL;
lz307 0:a26aca3e4760 28 unsigned i = 0; // index in the file list;
lz307 0:a26aca3e4760 29 DIR *d = opendir("/local"); // Opens the root directory of the local file system
lz307 0:a26aca3e4760 30 struct dirent *p;
lz307 0:a26aca3e4760 31 while((p = readdir(d)) != NULL) { // Print the names of the files in the local file system
lz307 0:a26aca3e4760 32 int l = strlen(p->d_name);
lz307 0:a26aca3e4760 33 if (strcmp(&(p->d_name[l-3]), "BMP") == 0) {
lz307 0:a26aca3e4760 34 i++;
lz307 0:a26aca3e4760 35 if (file_index == i) {
lz307 0:a26aca3e4760 36 fn = (char *) malloc(50);
lz307 0:a26aca3e4760 37 strcpy(fn, p->d_name);
lz307 0:a26aca3e4760 38 break;
lz307 0:a26aca3e4760 39 }
lz307 0:a26aca3e4760 40 }
lz307 0:a26aca3e4760 41 }
lz307 0:a26aca3e4760 42 closedir(d);
lz307 0:a26aca3e4760 43
lz307 0:a26aca3e4760 44 return fn;
lz307 0:a26aca3e4760 45 }
lz307 0:a26aca3e4760 46
lz307 0:a26aca3e4760 47 void strip_off() {
lz307 0:a26aca3e4760 48 for (int i=0; i<N; i++) {
lz307 0:a26aca3e4760 49 strip.setPixel(i, 0);
lz307 0:a26aca3e4760 50 }
lz307 0:a26aca3e4760 51 strip.write();
lz307 0:a26aca3e4760 52 }
lz307 0:a26aca3e4760 53
lz307 0:a26aca3e4760 54 void display_number(float f) {
lz307 0:a26aca3e4760 55 for (int i=0; i<N; i++) {
lz307 0:a26aca3e4760 56 if (i<(f*N))
lz307 0:a26aca3e4760 57 strip.setPixel(i, 0xffffff);
lz307 0:a26aca3e4760 58 }
lz307 0:a26aca3e4760 59 strip.write();
lz307 0:a26aca3e4760 60 wait(0.2);
lz307 0:a26aca3e4760 61 strip_off();
lz307 0:a26aca3e4760 62 }
lz307 0:a26aca3e4760 63
lz307 0:a26aca3e4760 64 void brightnessUp(void) {
lz307 0:a26aca3e4760 65 if (bright < 1)
lz307 0:a26aca3e4760 66 {
lz307 0:a26aca3e4760 67 bright += 0.01;
lz307 0:a26aca3e4760 68 if (bright > 1)
lz307 0:a26aca3e4760 69 bright = 1;
lz307 0:a26aca3e4760 70 printf("increase brightness\r\n");
lz307 0:a26aca3e4760 71 strip.setBrightness(bright);
lz307 0:a26aca3e4760 72 display_number(bright);
lz307 0:a26aca3e4760 73 }
lz307 0:a26aca3e4760 74 }
lz307 0:a26aca3e4760 75
lz307 0:a26aca3e4760 76 void brightnessDown(void) {
lz307 0:a26aca3e4760 77 if (bright > 0)
lz307 0:a26aca3e4760 78 {
lz307 0:a26aca3e4760 79 bright -= 0.01;
lz307 0:a26aca3e4760 80 if (bright < 0)
lz307 0:a26aca3e4760 81 bright = 0;
lz307 0:a26aca3e4760 82 printf("decrease brightness\r\n");
lz307 0:a26aca3e4760 83 strip.setBrightness(bright);
lz307 0:a26aca3e4760 84 display_number(bright);
lz307 0:a26aca3e4760 85 }
lz307 0:a26aca3e4760 86 }
lz307 0:a26aca3e4760 87
lz307 0:a26aca3e4760 88 void fileDown(void) {
lz307 0:a26aca3e4760 89 file_index--;
lz307 0:a26aca3e4760 90 if (file_index < 0)
lz307 0:a26aca3e4760 91 file_index = max_file;
lz307 0:a26aca3e4760 92 printf("fileDown to %i:%s\r\n", file_index, f_list[file_index]);
lz307 0:a26aca3e4760 93 display_number(file_index/float(N));
lz307 0:a26aca3e4760 94 }
lz307 0:a26aca3e4760 95
lz307 0:a26aca3e4760 96 void fileUp(void) {
lz307 0:a26aca3e4760 97 file_index++;
lz307 0:a26aca3e4760 98 if (file_index > max_file)
lz307 0:a26aca3e4760 99 file_index = 0;
lz307 0:a26aca3e4760 100 printf("fileUp to %i:%s\r\n", file_index, f_list[file_index]);
lz307 0:a26aca3e4760 101 display_number(file_index/float(N));
lz307 0:a26aca3e4760 102 }
lz307 0:a26aca3e4760 103
lz307 0:a26aca3e4760 104
lz307 0:a26aca3e4760 105 // Converts HSV to RGB with the given hue, assuming
lz307 0:a26aca3e4760 106 // maximum saturation and value
lz307 0:a26aca3e4760 107 int hueToRGB(float h)
lz307 0:a26aca3e4760 108 {
lz307 0:a26aca3e4760 109 // lots of floating point magic from the internet and scratching my head
lz307 0:a26aca3e4760 110 float r, g, b;
lz307 0:a26aca3e4760 111 if (h > 360)
lz307 0:a26aca3e4760 112 h -= 360;
lz307 0:a26aca3e4760 113 if (h < 0)
lz307 0:a26aca3e4760 114 h += 360;
lz307 0:a26aca3e4760 115 int i = (int)(h / 60.0);
lz307 0:a26aca3e4760 116 float f = (h / 60.0) - i;
lz307 0:a26aca3e4760 117 float q = 1 - f;
lz307 0:a26aca3e4760 118
lz307 0:a26aca3e4760 119 switch (i % 6)
lz307 0:a26aca3e4760 120 {
lz307 0:a26aca3e4760 121 case 0: r = 1; g = f; b = 0; break;
lz307 0:a26aca3e4760 122 case 1: r = q; g = 1; b = 0; break;
lz307 0:a26aca3e4760 123 case 2: r = 0; g = 1; b = f; break;
lz307 0:a26aca3e4760 124 case 3: r = 0; g = q; b = 1; break;
lz307 0:a26aca3e4760 125 case 4: r = f; g = 0; b = 1; break;
lz307 0:a26aca3e4760 126 case 5: r = 1; g = 0; b = q; break;
lz307 0:a26aca3e4760 127 default: r = 0; g = 0; b = 0; break;
lz307 0:a26aca3e4760 128 }
lz307 0:a26aca3e4760 129
lz307 0:a26aca3e4760 130 // scale to integers and return the packed value
lz307 0:a26aca3e4760 131 uint8_t R = (uint8_t)(r * 255);
lz307 0:a26aca3e4760 132 uint8_t G = (uint8_t)(g * 255);
lz307 0:a26aca3e4760 133 uint8_t B = (uint8_t)(b * 255);
lz307 0:a26aca3e4760 134
lz307 0:a26aca3e4760 135 return (R << 16) | (G << 8) | B;
lz307 0:a26aca3e4760 136 }
lz307 0:a26aca3e4760 137
lz307 0:a26aca3e4760 138
lz307 0:a26aca3e4760 139 void pattern1()
lz307 0:a26aca3e4760 140 {
lz307 0:a26aca3e4760 141 static float dh = 360.0 / N;
lz307 0:a26aca3e4760 142 static float x = 0;
lz307 0:a26aca3e4760 143 printf("%f\r\n",x);
lz307 0:a26aca3e4760 144
lz307 0:a26aca3e4760 145 for (int i = 0; i < N; i++) {
lz307 0:a26aca3e4760 146 int c = hueToRGB((dh * i) - x);
lz307 0:a26aca3e4760 147 //printf("R %i, G %i, B %i\r\n", (c>>16)&0xff, (c>>8)&0xff, c&0xff);
lz307 0:a26aca3e4760 148 strip.setPixel(i, c);
lz307 0:a26aca3e4760 149 }
lz307 0:a26aca3e4760 150
lz307 0:a26aca3e4760 151 x += 1;
lz307 0:a26aca3e4760 152 if (x > 360)
lz307 0:a26aca3e4760 153 x = 0;
lz307 0:a26aca3e4760 154 }
lz307 0:a26aca3e4760 155
lz307 0:a26aca3e4760 156
lz307 0:a26aca3e4760 157 void patternStart(void) {
lz307 0:a26aca3e4760 158 printf("file_index: %i\r\n", file_index);
lz307 0:a26aca3e4760 159 if (file_index == 0) {
lz307 0:a26aca3e4760 160 for ( int j=0; j<100; j++ ){
lz307 0:a26aca3e4760 161 pattern1();
lz307 0:a26aca3e4760 162 strip.write();
lz307 0:a26aca3e4760 163 wait_ms(20);
lz307 0:a26aca3e4760 164 }
lz307 0:a26aca3e4760 165 return;
lz307 0:a26aca3e4760 166 }
lz307 0:a26aca3e4760 167
lz307 0:a26aca3e4760 168 printf("fn: %s\r\n", f_list[file_index]);
lz307 0:a26aca3e4760 169
lz307 0:a26aca3e4760 170 char *fn = get_file_name();
lz307 0:a26aca3e4760 171
lz307 0:a26aca3e4760 172 unsigned l = strlen(fn);
lz307 0:a26aca3e4760 173 char *path = (char *) malloc(l+7);
lz307 0:a26aca3e4760 174 path[0] = 0;
lz307 0:a26aca3e4760 175 strcat(path, "/local/");
lz307 0:a26aca3e4760 176 strcat(path, fn);
lz307 0:a26aca3e4760 177 printf("path: %s\r\n", path);
lz307 0:a26aca3e4760 178
lz307 0:a26aca3e4760 179 MyBitmap = new BitmapFile(path);
lz307 0:a26aca3e4760 180 for(int row = 0; row < MyBitmap->getHeight(); row++)
lz307 0:a26aca3e4760 181 {
lz307 0:a26aca3e4760 182 int *row_color = MyBitmap->getRow(row, false);
lz307 0:a26aca3e4760 183 for(int col = 0; col < MyBitmap->getWidth(); col++)
lz307 0:a26aca3e4760 184 {
lz307 0:a26aca3e4760 185 unsigned c = row_color[col] & 0xffffff;
lz307 0:a26aca3e4760 186 strip.setPixel(col, c);
lz307 0:a26aca3e4760 187 if ( c > 0xfff )
lz307 0:a26aca3e4760 188 printf(" ");
lz307 0:a26aca3e4760 189 else
lz307 0:a26aca3e4760 190 printf("*");
lz307 0:a26aca3e4760 191 }
lz307 0:a26aca3e4760 192 strip.write();
lz307 0:a26aca3e4760 193 wait_ms(20);
lz307 0:a26aca3e4760 194 printf("\r\n");
lz307 0:a26aca3e4760 195 delete [] row_color;
lz307 0:a26aca3e4760 196 }
lz307 0:a26aca3e4760 197 printf("closing\r\n");
lz307 0:a26aca3e4760 198 MyBitmap->close();
lz307 0:a26aca3e4760 199 printf("closed\r\n");
lz307 0:a26aca3e4760 200 free(fn);
lz307 0:a26aca3e4760 201 free(path);
lz307 0:a26aca3e4760 202 strip_off();
lz307 0:a26aca3e4760 203 }
lz307 0:a26aca3e4760 204
lz307 0:a26aca3e4760 205
lz307 0:a26aca3e4760 206
lz307 0:a26aca3e4760 207
lz307 0:a26aca3e4760 208 int main() {
lz307 0:a26aca3e4760 209 b1.setAssertValue( 0 );
lz307 0:a26aca3e4760 210 b2.setAssertValue( 0 );
lz307 0:a26aca3e4760 211 b3.setAssertValue( 0 );
lz307 0:a26aca3e4760 212 b4.setAssertValue( 0 );
lz307 0:a26aca3e4760 213
lz307 0:a26aca3e4760 214 b1.attach_asserted( &brightnessDown );
lz307 0:a26aca3e4760 215 b2.attach_asserted( &patternStart );
lz307 0:a26aca3e4760 216 b3.attach_asserted( &brightnessUp );
lz307 0:a26aca3e4760 217
lz307 0:a26aca3e4760 218 b1.attach_asserted_held( &fileDown );
lz307 0:a26aca3e4760 219 b3.attach_asserted_held( &fileUp );
lz307 0:a26aca3e4760 220
lz307 0:a26aca3e4760 221 strip.setBrightness(bright); // set default brightness
lz307 0:a26aca3e4760 222
lz307 0:a26aca3e4760 223
lz307 0:a26aca3e4760 224 // build a list of files
lz307 0:a26aca3e4760 225 for (unsigned i=0; i<MAX_FILE; i++)
lz307 0:a26aca3e4760 226 f_list[i] = NULL;
lz307 0:a26aca3e4760 227
lz307 0:a26aca3e4760 228 unsigned i = 0; // index in the file list;
lz307 0:a26aca3e4760 229 DIR *d = opendir("/local"); // Opens the root directory of the local file system
lz307 0:a26aca3e4760 230 struct dirent *p;
lz307 0:a26aca3e4760 231 while((p = readdir(d)) != NULL) { // Print the names of the files in the local file system
lz307 0:a26aca3e4760 232 int l = strlen(p->d_name);
lz307 0:a26aca3e4760 233 if (strcmp(&(p->d_name[l-3]), "BMP") == 0) {
lz307 0:a26aca3e4760 234 char *tmp_str = (char *) malloc(l+1);
lz307 0:a26aca3e4760 235 strcpy(tmp_str, p->d_name);
lz307 0:a26aca3e4760 236 i++;
lz307 0:a26aca3e4760 237 f_list[i] = tmp_str;
lz307 0:a26aca3e4760 238 printf("f_list: %i:%s\r\n", i, tmp_str);
lz307 0:a26aca3e4760 239 }
lz307 0:a26aca3e4760 240 }
lz307 0:a26aca3e4760 241 closedir(d);
lz307 0:a26aca3e4760 242
lz307 0:a26aca3e4760 243 max_file = i;
lz307 0:a26aca3e4760 244 b1.setSampleFrequency();
lz307 0:a26aca3e4760 245 b2.setSampleFrequency();
lz307 0:a26aca3e4760 246 b3.setSampleFrequency();
lz307 0:a26aca3e4760 247 b4.setSampleFrequency();
lz307 0:a26aca3e4760 248
lz307 0:a26aca3e4760 249 while(1)
lz307 0:a26aca3e4760 250 __WFI();
lz307 0:a26aca3e4760 251 }