Merge branch 'master' of git://git.gag.com/fw/altos
[fw/altos] / src / stm-demo / ao_demo.c
1 /*
2  * Copyright © 2011 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; version 2 of the License.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License along
14  * with this program; if not, write to the Free Software Foundation, Inc.,
15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
16  */
17
18 #include "ao.h"
19 #include <ao_exti.h>
20 #include <ao_event.h>
21 #include <ao_quadrature.h>
22 #include <ao_button.h>
23
24 struct ao_task demo_task;
25
26 static inline int min(int a, int b) { return a < b ? a : b; }
27
28 void
29 ao_demo(void)
30 {
31         char    message[] = "Hello, Mike & Bdale --- ";
32         char    part[7];
33         int     i = 0;
34         int     len = sizeof(message) - 1;
35         int     first, second;
36
37         part[6] = '\0';
38         for (;;) {
39                 ao_delay(AO_MS_TO_TICKS(150));
40                 first = min(6, len - i);
41                 second = 6 - first;
42                 memcpy(part, message + i, first);
43                 memcpy(part + first, message, second);
44                 ao_lcd_font_string(part);
45                 if (++i >= len)
46                         i = 0;
47         }
48 }
49
50 void _close() { }
51 void _sbrk() { }
52 void _isatty() { }
53 void _lseek() { }
54 void _exit () { }
55 void _read () { }
56 void _fstat() { }
57
58 #define AO_DMA_TEST_INDEX       STM_DMA_INDEX(4)
59
60 static void
61 ao_dma_test(void) {
62         static char     src[20] = "hello, world";
63         static char     dst[20];
64         
65         dst[0] = '\0';
66         ao_dma_set_transfer(AO_DMA_TEST_INDEX, dst, src, 13,
67                             (1 << STM_DMA_CCR_MEM2MEM) |
68                             (STM_DMA_CCR_PL_LOW << STM_DMA_CCR_PL) |
69                             (STM_DMA_CCR_MSIZE_8 << STM_DMA_CCR_MSIZE) |
70                             (STM_DMA_CCR_PSIZE_8 << STM_DMA_CCR_PSIZE) |
71                             (1 << STM_DMA_CCR_MINC) |
72                             (1 << STM_DMA_CCR_PINC) |
73                             (0 << STM_DMA_CCR_CIRC) |
74                             (STM_DMA_CCR_DIR_MEM_TO_PER << STM_DMA_CCR_DIR));
75         ao_dma_start(AO_DMA_TEST_INDEX);
76         ao_arch_critical(
77                 while (!ao_dma_done[AO_DMA_TEST_INDEX])
78                         ao_sleep(&ao_dma_done[AO_DMA_TEST_INDEX]);
79                 );
80         ao_dma_done_transfer(AO_DMA_TEST_INDEX);
81         printf ("copied %s\n", dst);
82 }
83
84 static void
85 ao_spi_write(void) {
86         unsigned char   data[] = { 0x55, 0xaa, 0xff, 0x00 };
87         int i;
88
89         for (i = 0; i < 10; i++) {
90                 ao_spi_get(0, AO_SPI_SPEED_FAST);
91                 stm_gpio_set(&stm_gpioc, 12, 0);
92                 ao_spi_send(data, 4, 0);
93                 stm_gpio_set(&stm_gpioc, 12, 1);
94                 ao_spi_put(0);
95                 printf(".");
96                 flush();
97                 ao_delay(100);
98         }
99 }
100
101 static void
102 ao_spi_read(void) {
103         unsigned char   data[4];
104         int i;
105
106         for (i = 0; i < 10; i++) {
107                 ao_spi_get(0, AO_SPI_SPEED_FAST);
108                 stm_gpio_set(&stm_gpioc, 12, 0);
109                 ao_spi_recv(data, 4, 0);
110                 stm_gpio_set(&stm_gpioc, 12, 1);
111                 ao_spi_put(0);
112                 printf(".");
113                 flush();
114                 ao_delay(100);
115         }
116 }
117
118 static void
119 ao_i2c_write(void) {
120         unsigned char   data[] = { 0x55, 0xaa, 0xff, 0x00 };
121         int i;
122
123         for (i = 0; i < 10; i++) {
124                 ao_i2c_get(0);
125                 if (ao_i2c_start(0, 0x55))
126                         ao_i2c_send(data, 4, 0, TRUE);
127                 else {
128                         printf ("i2c start failed\n");
129                         ao_i2c_put(0);
130                         break;
131                 }
132                 ao_i2c_put(0);
133                 printf(".");
134                 flush();
135                 ao_delay(100);
136         }
137 }
138
139 static void
140 ao_temp (void)
141 {
142         struct ao_data  packet;
143         int temp;
144
145         ao_data_get(&packet);
146
147         /*
148          * r = (110 - 25) / (ts_cal_hot - ts_cal_cold)
149          * 25 + (110 - 25) * (temp - ts_cal_cold) / (ts_cal_hot - ts_cal_cold)
150          */
151         temp = 25 + (110 - 25) * (packet.adc.temp - stm_temp_cal.ts_cal_cold) / (stm_temp_cal.ts_cal_hot - stm_temp_cal.ts_cal_cold);
152         printf ("temp: %d\n", temp);
153 }
154
155 static void
156 ao_event(void)
157 {
158         struct ao_event event;
159
160         for (;;) {
161                 flush();
162                 ao_event_get(&event);
163                 printf ("type %1d unit %1d tick %5u value %ld\n",
164                         event.type, event.unit, event.tick, event.value);
165                 if (event.value == 100)
166                         break;
167         }
168
169 }
170
171 __code struct ao_cmds ao_demo_cmds[] = {
172         { ao_dma_test,  "D\0DMA test" },
173         { ao_spi_write, "W\0SPI write" },
174         { ao_spi_read, "R\0SPI read" },
175         { ao_i2c_write, "i\0I2C write" },
176         { ao_temp, "t\0Show temp" },
177         { ao_event, "e\0Monitor event queue" },
178         { 0, NULL }
179 };
180
181 int
182 main(void)
183 {
184         ao_clock_init();
185         
186         ao_serial_init();
187         ao_timer_init();
188         ao_dma_init();
189         ao_cmd_init();
190         ao_lcd_stm_init();
191 //      ao_lcd_font_init();
192         ao_spi_init();
193         ao_i2c_init();
194         ao_exti_init();
195         ao_quadrature_init();
196         ao_button_init();
197
198         ao_timer_set_adc_interval(100);
199
200         ao_adc_init();
201         ao_usb_init();
202
203         ao_cmd_register(&ao_demo_cmds[0]);
204         
205         ao_start_scheduler();
206         return 0;
207 }