main.c 17 KB
Newer Older
1
#include <stdio.h>
Damien's avatar
Damien committed
2
3
#include <stm32f4xx.h>
#include <stm32f4xx_rcc.h>
4
5
#include <stm32f4xx_gpio.h>
#include <stm_misc.h>
Damien's avatar
Damien committed
6
7
#include "std.h"

8
#include "misc.h"
Damien's avatar
Damien committed
9
10
#include "mpyconfig.h"
#include "gc.h"
11
#include "systick.h"
12
#include "led.h"
13
#include "lcd.h"
14
#include "storage.h"
15
#include "mma.h"
Damien's avatar
Damien committed
16
#include "usb.h"
17
#include "ff.h"
Damien's avatar
Damien committed
18

Damien's avatar
Damien committed
19
20
extern uint32_t _heap_start;

21
static void impl02_c_version() {
Damien's avatar
Damien committed
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
    int x = 0;
    while (x < 400) {
        int y = 0;
        while (y < 400) {
            volatile int z = 0;
            while (z < 400) {
                z = z + 1;
            }
            y = y + 1;
        }
        x = x + 1;
    }
}

#define PYB_USRSW_PORT (GPIOA)
37
#define PYB_USRSW_PIN (GPIO_Pin_13)
Damien's avatar
Damien committed
38
39
40

void sw_init() {
    // make it an input with pull-up
41
42
43
44
45
    GPIO_InitTypeDef GPIO_InitStructure;
    GPIO_InitStructure.GPIO_Pin = PYB_USRSW_PIN;
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_IN;
    GPIO_InitStructure.GPIO_PuPd = GPIO_PuPd_UP;
    GPIO_Init(PYB_USRSW_PORT, &GPIO_InitStructure);
Damien's avatar
Damien committed
46
47
48
}

int sw_get() {
49
    if (PYB_USRSW_PORT->IDR & PYB_USRSW_PIN) {
Damien's avatar
Damien committed
50
51
52
53
54
55
56
57
58
59
60
61
62
        // pulled high, so switch is not pressed
        return 0;
    } else {
        // pulled low, so switch is pressed
        return 1;
    }
}

void __fatal_error(const char *msg) {
    lcd_print_strn("\nFATAL ERROR:\n", 14);
    lcd_print_strn(msg, strlen(msg));

    for (;;) {
63
64
        led_state(PYB_LED_R1, 1);
        led_state(PYB_LED_R2, 0);
65
        sys_tick_delay_ms(150);
66
67
        led_state(PYB_LED_R1, 0);
        led_state(PYB_LED_R2, 1);
68
        sys_tick_delay_ms(150);
Damien's avatar
Damien committed
69
70
71
    }
}

72
#include "nlr.h"
Damien's avatar
Damien committed
73
74
#include "misc.h"
#include "lexer.h"
75
#include "lexerstm.h"
Damien's avatar
Damien committed
76
77
78
79
80
81
#include "mpyconfig.h"
#include "parse.h"
#include "compile.h"
#include "runtime.h"

py_obj_t pyb_delay(py_obj_t count) {
82
    sys_tick_delay_ms(rt_get_int(count));
Damien's avatar
Damien committed
83
84
85
86
    return py_const_none;
}

py_obj_t pyb_led(py_obj_t state) {
87
    led_state(PYB_LED_G1, rt_is_true(state));
Damien's avatar
Damien committed
88
89
90
91
92
93
94
95
96
97
98
99
100
    return state;
}

py_obj_t pyb_sw() {
    if (sw_get()) {
        return py_const_true;
    } else {
        return py_const_false;
    }
}

FATFS fatfs0;

101
102

/*
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
void g(uint i) {
    printf("g:%d\n", i);
    if (i & 1) {
        nlr_jump((void*)(42 + i));
    }
}
void f() {
    nlr_buf_t nlr;
    int i;
    for (i = 0; i < 4; i++) {
        printf("f:loop:%d:%p\n", i, &nlr);
        if (nlr_push(&nlr) == 0) {
            // normal
            //printf("a:%p:%p %p %p %u\n", &nlr, nlr.ip, nlr.sp, nlr.prev, nlr.ret_val);
            g(i);
            printf("f:lp:%d:nrm\n", i);
            nlr_pop();
        } else {
            // nlr
            //printf("b:%p:%p %p %p %u\n", &nlr, nlr.ip, nlr.sp, nlr.prev, nlr.ret_val);
            printf("f:lp:%d:nlr:%d\n", i, (int)nlr.ret_val);
        }
    }
}
void nlr_test() {
    f(1);
}
130
131
*/

132
133
134
135
136
137
138
void fatality() {
    led_state(PYB_LED_R1, 1);
    led_state(PYB_LED_G1, 1);
    led_state(PYB_LED_R2, 1);
    led_state(PYB_LED_G2, 1);
}

139
static const char fresh_boot_py[] =
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
"# boot.py -- run on boot-up\n"
"# can run arbitrary Python, but best to keep it minimal\n"
"\n"
"pyb.source_dir('/src')\n"
"pyb.main('main.py')\n"
"#pyb.usb_usr('VCP')\n"
"#pyb.usb_msd(True, 'dual partition')\n"
"#pyb.flush_cache(False)\n"
"#pyb.error_log('error.txt')\n"
;

// get lots of info about the board
static void board_info() {
    // get and print clock speeds
    // SYSCLK=168MHz, HCLK=168MHz, PCLK1=42MHz, PCLK2=84MHz
    {
        RCC_ClocksTypeDef rcc_clocks;
        RCC_GetClocksFreq(&rcc_clocks);
        printf("S=%lu\nH=%lu\nP1=%lu\nP2=%lu\n", rcc_clocks.SYSCLK_Frequency, rcc_clocks.HCLK_Frequency, rcc_clocks.PCLK1_Frequency, rcc_clocks.PCLK2_Frequency);
    }

    // to print info about memory
    {
        extern void *_sidata;
        extern void *_sdata;
        extern void *_edata;
        extern void *_sbss;
        extern void *_ebss;
        extern void *_estack;
        extern void *_etext;
        printf("_sidata=%p\n", &_sidata);
        printf("_sdata=%p\n", &_sdata);
        printf("_edata=%p\n", &_edata);
        printf("_sbss=%p\n", &_sbss);
        printf("_ebss=%p\n", &_ebss);
        printf("_estack=%p\n", &_estack);
        printf("_etext=%p\n", &_etext);
        printf("_heap_start=%p\n", &_heap_start);
    }

    // free space on flash
    {
        DWORD nclst;
        FATFS *fatfs;
        f_getfree("0:", &nclst, &fatfs);
        printf("free=%u\n", (uint)(nclst * fatfs->csize * 512));
    }
}

189
char *readline(const char *prompt) {
190
191
    vstr_t vstr;
    vstr_init(&vstr);
192
193
    usb_vcp_send_str(prompt);
    for (;;) {
194
195
196
197
198
199
        //extern int rx_buf_in;
        //extern int rx_buf_out;
        while (usb_vcp_rx_any() == 0) {
            //printf("nope %x %x\n", rx_buf_in, rx_buf_out);
            sys_tick_delay_ms(10);
        }
200
        char c = usb_vcp_rx_get();
201
202
203
204
205
206
207
208
209
210
211
212
213
214
        if (c == 4 && vstr_len(&vstr) == 0) {
            return NULL;
        } else if (c == '\r') {
            usb_vcp_send_str("\r\n");
            return vstr_str(&vstr);
        } else if (c == 127) {
            if (vstr_len(&vstr) > 0) {
                vstr_cut_tail(&vstr, 1);
                usb_vcp_send_str("\b \b");
            }
        } else if (32 <= c && c <= 126) {
            vstr_add_char(&vstr, c);
            usb_vcp_send_strn(&c, 1);
        }
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
        sys_tick_delay_ms(100);
    }
    return NULL;
}

void do_repl() {
    usb_vcp_send_str("Micro Python\r\n");

    for (;;) {
        char *line = readline(">>> ");
        if (line == NULL) {
            // EOF
            return;
        }
        /*
        if (is_compound_stmt(line)) {
            for (;;) {
                char *line2 = readline("... ");
                if (line2 == NULL || strlen(line2) == 0) {
                    break;
                }
                char *line3 = str_join(line, '\n', line2);
                m_free(line);
                m_free(line2);
                line = line3;
            }
        }
        */

        py_lexer_str_buf_t sb;
        py_lexer_t *lex = py_lexer_new_from_str_len("<stdin>", line, strlen(line), false, &sb);
        py_parse_node_t pn = py_parse(lex, PY_PARSE_SINGLE_INPUT);
        py_lexer_free(lex);

        if (pn != PY_PARSE_NODE_NULL) {
            bool comp_ok = py_compile(pn, true);
            if (comp_ok) {
                py_obj_t module_fun = rt_make_function_from_id(1);
                if (module_fun != py_const_none) {
                    nlr_buf_t nlr;
                    if (nlr_push(&nlr) == 0) {
                        rt_call_function_0(module_fun);
                        nlr_pop();
                    } else {
                        // uncaught exception
                        py_obj_print((py_obj_t)nlr.ret_val);
                        printf("\n");
                    }
                }
            }
        }
    }
}

Damien's avatar
Damien committed
269
270
271
272
273
274
275
276
void gc_collect() {
    gc_collect_start();
    gc_collect_root((void**)0x20000000, (((uint32_t)&_heap_start) - 0x20000000) / 4);
    gc_collect_root((void**)(0x20000000 + 0x18000), (0x20000 - 0x18000) / 4);
    // TODO registers
    gc_collect_end();
}

Damien's avatar
Damien committed
277
int main() {
278
    // TODO disable JTAG
Damien's avatar
Damien committed
279

280
281
282
283
284
285
    // set interrupt priority config to use all 4 bits for pre-empting
    NVIC_PriorityGroupConfig(NVIC_PriorityGroup_4);

    // enable the CCM RAM and the GPIO's
    RCC->AHB1ENR |= RCC_AHB1ENR_CCMDATARAMEN | RCC_AHB1ENR_GPIOAEN | RCC_AHB1ENR_GPIOBEN | RCC_AHB1ENR_GPIOCEN;

286
287
    // basic sub-system init
    sys_tick_init();
Damien's avatar
Damien committed
288
    led_init();
289
290
291
292
293

    // turn on LED to indicate bootup
    led_state(PYB_LED_G1, 1);

    // more sub-system init
Damien's avatar
Damien committed
294
295
    sw_init();
    lcd_init();
296
    storage_init();
Damien's avatar
Damien committed
297

Damien's avatar
Damien committed
298
299
300
301
    // GC init
    gc_init(&_heap_start, (void*)(0x20000000 + 0x18000));
    sys_tick_delay_ms(2000);

302
    // Python init
303
304
    qstr_init();
    rt_init();
305

306
307
308
309
310
    // add some functions to the python namespace
    rt_store_name(qstr_from_str_static("pyb_delay"), rt_make_function_1(pyb_delay));
    rt_store_name(qstr_from_str_static("pyb_led"), rt_make_function_1(pyb_led));
    rt_store_name(qstr_from_str_static("pyb_sw"), rt_make_function_0(pyb_sw));

Damien's avatar
Damien committed
311
312
313
    // print a message
    printf(" micro py board\n");

314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
    // local filesystem init
    {
        // try to mount the flash
        FRESULT res = f_mount(&fatfs0, "0:", 1);
        if (res == FR_OK) {
            // mount sucessful
        } else if (res == FR_NO_FILESYSTEM) {
            // no filesystem, so create a fresh one

            // LED on to indicate creation of LFS
            led_state(PYB_LED_R2, 1);
            uint32_t stc = sys_tick_counter;

            res = f_mkfs("0:", 0, 0);
            if (res == FR_OK) {
                // success creating fresh LFS
            } else {
                __fatal_error("could not create LFS");
            }

334
335
            // keep LED on for at least 200ms
            sys_tick_wait_at_least(stc, 200);
336
337
338
339
            led_state(PYB_LED_R2, 0);
        } else {
            __fatal_error("could not access LFS");
        }
Damien's avatar
Damien committed
340
341
    }

342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
    // make sure we have a /boot.py
    {
        FILINFO fno;
        FRESULT res = f_stat("0:/boot.py", &fno);
        if (res == FR_OK) {
            if (fno.fattrib & AM_DIR) {
                // exists as a directory
                // TODO handle this case
                // see http://elm-chan.org/fsw/ff/img/app2.c for a "rm -rf" implementation
            } else {
                // exists as a file, good!
            }
        } else {
            // doesn't exist, create fresh file

            // LED on to indicate creation of boot.py
            led_state(PYB_LED_R2, 1);
            uint32_t stc = sys_tick_counter;

            FIL fp;
            f_open(&fp, "0:/boot.py", FA_WRITE | FA_CREATE_ALWAYS);
            UINT n;
            f_write(&fp, fresh_boot_py, sizeof(fresh_boot_py), &n);
            // TODO check we could write n bytes
            f_close(&fp);

368
369
            // keep LED on for at least 200ms
            sys_tick_wait_at_least(stc, 200);
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
            led_state(PYB_LED_R2, 0);
        }
    }

    // run /boot.py
    if (0) {
        FIL fp;
        f_open(&fp, "0:/boot.py", FA_READ);
        UINT n;
        char buf[20];
        f_read(&fp, buf, 18, &n);
        buf[n + 1] = 0;
        printf("read %d\n%s", n, buf);
        f_close(&fp);
    }

    // turn boot-up LED off
387
    led_state(PYB_LED_G1, 0);
Damien's avatar
Damien committed
388

389
    // USB
Damien's avatar
Damien committed
390
    if (1) {
391
392
393
        usb_init();
    }

Damien's avatar
Damien committed
394
    //printf("init;al=%u\n", m_get_total_bytes_allocated()); // 1600, due to qstr_init
395
    //sys_tick_delay_ms(1000);
Damien's avatar
Damien committed
396
397

    // Python!
398
    if (0) {
Damien's avatar
Damien committed
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
        //const char *pysrc = "def f():\n  x=x+1\nprint(42)\n";
        const char *pysrc =
            // impl01.py
            /*
            "x = 0\n"
            "while x < 400:\n"
            "    y = 0\n"
            "    while y < 400:\n"
            "        z = 0\n"
            "        while z < 400:\n"
            "            z = z + 1\n"
            "        y = y + 1\n"
            "    x = x + 1\n";
            */
            // impl02.py
414
            /*
Damien's avatar
Damien committed
415
416
417
418
419
420
421
422
423
424
425
426
            "#@micropython.native\n"
            "def f():\n"
            "    x = 0\n"
            "    while x < 400:\n"
            "        y = 0\n"
            "        while y < 400:\n"
            "            z = 0\n"
            "            while z < 400:\n"
            "                z = z + 1\n"
            "            y = y + 1\n"
            "        x = x + 1\n"
            "f()\n";
427
            */
Damien's avatar
Damien committed
428
429
430
431
432
433
434
            "print('in python!')\n"
            "x = 0\n"
            "while x < 4:\n"
            "    pyb_led(True)\n"
            "    pyb_delay(201)\n"
            "    pyb_led(False)\n"
            "    pyb_delay(201)\n"
435
            "    x += 1\n"
Damien's avatar
Damien committed
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
            "print('press me!')\n"
            "while True:\n"
            "    pyb_led(pyb_sw())\n";
            /*
            // impl16.py
            "@micropython.asm_thumb\n"
            "def delay(r0):\n"
            "    b(loop_entry)\n"
            "    label(loop1)\n"
            "    movw(r1, 55999)\n"
            "    label(loop2)\n"
            "    subs(r1, r1, 1)\n"
            "    cmp(r1, 0)\n"
            "    bgt(loop2)\n"
            "    subs(r0, r0, 1)\n"
            "    label(loop_entry)\n"
            "    cmp(r0, 0)\n"
            "    bgt(loop1)\n"
            "print('in python!')\n"
            "@micropython.native\n"
            "def flash(n):\n"
            "    x = 0\n"
            "    while x < n:\n"
            "        pyb_led(True)\n"
            "        delay(249)\n"
            "        pyb_led(False)\n"
            "        delay(249)\n"
            "        x = x + 1\n"
            "flash(20)\n";
            */
466
467
468
469
470
471
472
473
474
            // impl18.py
            /*
            "# basic exceptions\n"
            "x = 1\n"
            "try:\n"
            "    x.a()\n"
            "except:\n"
            "    print(x)\n";
            */
475
            /*
476
477
478
479
480
481
482
483
            // impl19.py
            "# for loop\n"
            "def f():\n"
            "    for x in range(400):\n"
            "        for y in range(400):\n"
            "            for z in range(400):\n"
            "                pass\n"
            "f()\n";
484
            */
Damien's avatar
Damien committed
485

486
487
        py_lexer_str_buf_t py_lexer_str_buf;
        py_lexer_t *lex = py_lexer_new_from_str_len("<stdin>", pysrc, strlen(pysrc), false, &py_lexer_str_buf);
Damien's avatar
Damien committed
488

489
490
491
492
493
494
        // nalloc=1740;6340;6836 -> 140;4600;496 bytes for lexer, parser, compiler
        printf("lex; al=%u\n", m_get_total_bytes_allocated());
        sys_tick_delay_ms(1000);
        py_parse_node_t pn = py_parse(lex, PY_PARSE_FILE_INPUT);
        py_lexer_free(lex);
        if (pn != PY_PARSE_NODE_NULL) {
Damien's avatar
Damien committed
495
            printf("pars;al=%u\n", m_get_total_bytes_allocated());
496
            sys_tick_delay_ms(1000);
Damien's avatar
Damien committed
497
            //parse_node_show(pn, 0);
498
            bool comp_ok = py_compile(pn, false);
Damien's avatar
Damien committed
499
            printf("comp;al=%u\n", m_get_total_bytes_allocated());
500
            sys_tick_delay_ms(1000);
Damien's avatar
Damien committed
501

502
503
504
            if (!comp_ok) {
                printf("compile error\n");
            } else {
Damien's avatar
Damien committed
505
506
507
508
                // execute it!

                py_obj_t module_fun = rt_make_function_from_id(1);

509
                // flash once
510
                led_state(PYB_LED_G1, 1);
511
                sys_tick_delay_ms(100);
512
                led_state(PYB_LED_G1, 0);
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528

                nlr_buf_t nlr;
                if (nlr_push(&nlr) == 0) {
                    py_obj_t ret = rt_call_function_0(module_fun);
                    printf("done! got: ");
                    py_obj_print(ret);
                    printf("\n");
                    nlr_pop();
                } else {
                    // uncaught exception
                    printf("exception: ");
                    py_obj_print((py_obj_t)nlr.ret_val);
                    printf("\n");
                }

                // flash once
529
                led_state(PYB_LED_G1, 1);
530
                sys_tick_delay_ms(100);
531
                led_state(PYB_LED_G1, 0);
Damien's avatar
Damien committed
532

533
                sys_tick_delay_ms(1000);
Damien's avatar
Damien committed
534
                printf("nalloc=%u\n", m_get_total_bytes_allocated());
535
                sys_tick_delay_ms(1000);
Damien's avatar
Damien committed
536
537
538
539
            }
        }
    }

540
541
    do_repl();

Damien's avatar
Damien committed
542
543
    // benchmark C version of impl02.py
    if (0) {
544
        led_state(PYB_LED_G1, 1);
545
        sys_tick_delay_ms(100);
546
        led_state(PYB_LED_G1, 0);
Damien's avatar
Damien committed
547
        impl02_c_version();
548
        led_state(PYB_LED_G1, 1);
549
        sys_tick_delay_ms(100);
550
        led_state(PYB_LED_G1, 0);
Damien's avatar
Damien committed
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
    }

    // MMA testing
    if (0) {
        printf("1");
        mma_init();
        printf("2");
        mma_start(0x4c, 1);
        printf("3");
        mma_send_byte(0);
        printf("4");
        mma_stop();
        printf("5");
        mma_start(0x4c, 1);
        printf("6");
        mma_send_byte(0);
        printf("7");
        mma_restart(0x4c, 0);
        for (int i = 0; i <= 0xa; i++) {
            int data;
            if (i == 0xa) {
                data = mma_read_nack();
            } else {
                data = mma_read_ack();
            }
            printf(" %02x", data);
        }
        printf("\n");

        mma_start(0x4c, 1);
        mma_send_byte(7); // mode
        mma_send_byte(1); // active mode
        mma_stop();

        for (;;) {
586
            sys_tick_delay_ms(500);
Damien's avatar
Damien committed
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613

            mma_start(0x4c, 1);
            mma_send_byte(0);
            mma_restart(0x4c, 0);
            for (int i = 0; i <= 3; i++) {
                int data;
                if (i == 3) {
                    data = mma_read_nack();
                    printf(" %02x\n", data);
                } else {
                    data = mma_read_ack() & 0x3f;
                    if (data & 0x20) {
                        data |= 0xc0;
                    }
                    printf(" % 2d", data);
                }
            }
        }
    }

    // SD card testing
    if (0) {
        //sdio_init();
    }

    int i = 0;
    int n = 0;
614
    uint32_t stc = sys_tick_counter;
Damien's avatar
Damien committed
615
616

    for (;;) {
617
        sys_tick_delay_ms(10);
Damien's avatar
Damien committed
618
        if (sw_get()) {
619
            led_state(PYB_LED_G1, 1);
Damien's avatar
Damien committed
620
621
622
623
624
625
626
627
628
629
            i = 1 - i;
            if (i) {
                printf(" angel %05x.\n", n);
                //usb_vcp_send("hello!\r\n", 8);
            } else {
                printf(" mishka %4u.\n", n);
                //usb_vcp_send("angel!\r\n", 8);
            }
            n += 1;
        } else {
630
631
632
            led_state(PYB_LED_G1, 0);
        }
        if (sys_tick_has_passed(stc, 500)) {
633
            stc += 500;
634
            led_toggle(PYB_LED_G2);
Damien's avatar
Damien committed
635
636
637
638
639
        }
    }

    return 0;
}