root / branches / colonetmk2 / code / projects / libdragonfly / bom.c @ 1456
History | View | Annotate | Download (10.2 KB)
1 | 241 | bcoltin | /**
|
---|---|---|---|
2 | * Copyright (c) 2007 Colony Project
|
||
3 | *
|
||
4 | * Permission is hereby granted, free of charge, to any person
|
||
5 | * obtaining a copy of this software and associated documentation
|
||
6 | * files (the "Software"), to deal in the Software without
|
||
7 | * restriction, including without limitation the rights to use,
|
||
8 | * copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
9 | * copies of the Software, and to permit persons to whom the
|
||
10 | * Software is furnished to do so, subject to the following
|
||
11 | * conditions:
|
||
12 | *
|
||
13 | * The above copyright notice and this permission notice shall be
|
||
14 | * included in all copies or substantial portions of the Software.
|
||
15 | *
|
||
16 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||
17 | * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
|
||
18 | * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||
19 | * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
|
||
20 | * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
||
21 | * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||
22 | * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
||
23 | * OTHER DEALINGS IN THE SOFTWARE.
|
||
24 | **/
|
||
25 | |||
26 | |||
27 | /**
|
||
28 | * @file bom.c
|
||
29 | * @brief Implementation for using the BOM
|
||
30 | *
|
||
31 | * Contains functions for using the Bearing and Orientation Module (BOM)
|
||
32 | *
|
||
33 | * @author Colony Project, CMU Robotics Club
|
||
34 | **/
|
||
35 | |||
36 | 8 | bcoltin | #include "bom.h" |
37 | #include "dio.h" |
||
38 | 945 | abuchan | #include "serial.h" |
39 | 8 | bcoltin | #include "analog.h" |
40 | |||
41 | 945 | abuchan | //On the original BOM1.0, the emmitter angular order does not match the analog mux order
|
42 | //so you need to iterate through the mux index in the following order if you want to get
|
||
43 | //the detector readings in order:
|
||
44 | 448 | cmar | static const char lookup[16] = {7,6,5,0xe,1,4,3,2,0xf,0,0xd,8,0xc,0xb,9,0xa}; |
45 | 8 | bcoltin | |
46 | // internal function prototypes
|
||
47 | 448 | cmar | static void bom_select(char which); |
48 | 8 | bcoltin | |
49 | /*
|
||
50 | Bk R Y (Analog)
|
||
51 | ---------
|
||
52 | Green
|
||
53 | Blue
|
||
54 | White
|
||
55 | ---------
|
||
56 | Blue
|
||
57 | White
|
||
58 | */
|
||
59 | |||
60 | |||
61 | /*
|
||
62 | the analog pin definitions from dio.h DO NOT work here,
|
||
63 | so we must use PF0 from avrgcc (as opposed to _PIN_F0).
|
||
64 | BUT the dio pin definitions from dio.h must be used (no PE...).
|
||
65 | |||
66 | also, _PIN_E2 is initialized to high for some reason,
|
||
67 | which turns the BOM on when the robot is turned on.
|
||
68 | WORK-AROUND: call digital_output(_PIN_E2,0) at some point.
|
||
69 | |||
70 | */
|
||
71 | |||
72 | #define MONKI PF0 //analog (yellow) |
||
73 | //------------------------//
|
||
74 | #define MONKL _PIN_E2 //green |
||
75 | #define MONK1 _PIN_E3 //blue |
||
76 | #define MONK0 _PIN_E4 //white |
||
77 | //------------------------//
|
||
78 | #define MONK3 _PIN_E6 //blue |
||
79 | #define MONK2 _PIN_E7 //white |
||
80 | |||
81 | 1424 | jsexton | #define BOM_VALUE_THRESHOLD 150 //200 |
82 | 448 | cmar | #define NUM_BOM_LEDS 16 |
83 | 8 | bcoltin | |
84 | 945 | abuchan | /*
|
85 | *The following pin definitions are for the BOM v1.5
|
||
86 | */
|
||
87 | 8 | bcoltin | |
88 | 945 | abuchan | #define BOM_MODE _PIN_E2 //dio0 |
89 | #define BOM_STROBE _PIN_E3 //dio1 |
||
90 | |||
91 | #define BOM_DATA _PIN_A0 //servo0 |
||
92 | #define BOM_CLOCK _PIN_A1 //servo1 |
||
93 | |||
94 | #define BOM_S0 _PIN_E5 //dio3 |
||
95 | #define BOM_S1 _PIN_E4 //dio2 |
||
96 | 1079 | deffi | #define BOM_S2 _PIN_E7 //dio4 |
97 | #define BOM_S3 _PIN_E6 //dio5 |
||
98 | 945 | abuchan | #define BOM_OUT PF0 //analog(yellow) |
99 | |||
100 | 8 | bcoltin | /**
|
101 | * @defgroup bom BOM (Bearing and Orientation Module)
|
||
102 | * @brief Functions for dealing with the BOM.
|
||
103 | *
|
||
104 | * The Bearing and Orientation Module / Barrel of Monkeys / BOM
|
||
105 | * is a custom sensor designed and built by the Colony Project.
|
||
106 | * It consists of a ring of 16 IR emitters and 16 IR detectors.
|
||
107 | * The BOM is most often use to determine the direction of other
|
||
108 | * robots. This module contains functions for controlling the BOM.
|
||
109 | *
|
||
110 | * Include bom.h to access these functions.
|
||
111 | *
|
||
112 | * @{
|
||
113 | 945 | abuchan | **/
|
114 | |||
115 | static unsigned int bom_val[NUM_BOM_LEDS]; |
||
116 | 1456 | rcahoon | static volatile char bom_type = BOM10; |
117 | 945 | abuchan | static int select_pins[4]; |
118 | static int analog_pin; |
||
119 | |||
120 | /**
|
||
121 | * Initializes the BOM.
|
||
122 | * Call bom_init before reading bom values or turning bom leds.
|
||
123 | *
|
||
124 | * @bugs INCOMPLETE - No utilization of BOM1.5 RSSI capability. Probably leave this out
|
||
125 | * until Cornell and Pras return
|
||
126 | *
|
||
127 | * @see bom_refresh, bom_leds_on, bom_leds_off
|
||
128 | **/
|
||
129 | void bom_init(char type) { |
||
130 | bom_type = type; |
||
131 | |||
132 | switch(bom_type) {
|
||
133 | case BOM10:
|
||
134 | select_pins[0] = MONK0;
|
||
135 | select_pins[1] = MONK1;
|
||
136 | select_pins[2] = MONK2;
|
||
137 | select_pins[3] = MONK3;
|
||
138 | analog_pin = MONKI; |
||
139 | break;
|
||
140 | case BOM15:
|
||
141 | //Sets BOM1.5 to normal [BOM] mode
|
||
142 | digital_output(BOM_MODE, 0);
|
||
143 | select_pins[0] = BOM_S0;
|
||
144 | select_pins[1] = BOM_S1;
|
||
145 | select_pins[2] = BOM_S2;
|
||
146 | select_pins[3] = BOM_S3;
|
||
147 | bom_set_leds(BOM_ALL); |
||
148 | analog_pin = BOM_OUT; |
||
149 | break;
|
||
150 | case RBOM:
|
||
151 | break;
|
||
152 | //default:
|
||
153 | } |
||
154 | } |
||
155 | |||
156 | /**
|
||
157 | * Iterates through each bit in the bit_field. For each set bit, sets the corresponding bom select bits
|
||
158 | * and updates the corresponding bom value with an analog_get8 reading. analog_init and bom_init
|
||
159 | 1417 | emullini | * must be called for this to work. Must call this before reading BOM values!
|
160 | 945 | abuchan | *
|
161 | *
|
||
162 | * @param bit_field specifies which elements in bom_val[] should be updated. Use BOM_ALL to refresh all values.
|
||
163 | * Ex. if 0x0003 is passed, bom_val[0] and bom_val[1] will be updated.
|
||
164 | *
|
||
165 | * @see bom_get
|
||
166 | **/
|
||
167 | void bom_refresh(int bit_field) { |
||
168 | int i;
|
||
169 | 1406 | jykong | int loop_was_running = 0; |
170 | 945 | abuchan | |
171 | 1406 | jykong | //Check analog loop status
|
172 | if(analog_loop_status() == ADC_LOOP_RUNNING) {
|
||
173 | loop_was_running = 1;
|
||
174 | analog_stop_loop(); |
||
175 | } |
||
176 | 945 | abuchan | |
177 | 1406 | jykong | //Read BOM values
|
178 | 945 | abuchan | for(i = 0; i < NUM_BOM_LEDS; i++) { |
179 | if(bit_field & 0x1) { |
||
180 | bom_select(i); |
||
181 | bom_val[i] = analog_get8(analog_pin); |
||
182 | } |
||
183 | bit_field = bit_field >> 1;
|
||
184 | } |
||
185 | |||
186 | 1406 | jykong | //Restore analog loop status
|
187 | if(loop_was_running)
|
||
188 | analog_start_loop(); |
||
189 | 945 | abuchan | } |
190 | |||
191 | /**
|
||
192 | * Gets the bom reading from bom_val[which]. Call bom_refresh beforehand to read new bom values.
|
||
193 | *
|
||
194 | 1417 | emullini | * @pre must call bom refresh first
|
195 | *
|
||
196 | 945 | abuchan | * @param which which bom value to return
|
197 | *
|
||
198 | * @return the bom value
|
||
199 | *
|
||
200 | * see bom_refresh
|
||
201 | **/
|
||
202 | int bom_get(int which) { |
||
203 | return bom_val[which];
|
||
204 | } |
||
205 | |||
206 | /**
|
||
207 | * Compares all the values in bom_val[] and returns the index to the lowest (max) value element.
|
||
208 | *
|
||
209 | 1417 | emullini | * @pre must call bom refresh
|
210 | 945 | abuchan | * @return index to the lowest (max) bom value element. -1 if no value is lower than
|
211 | * BOM_VALUE_THRESHOLD
|
||
212 | **/
|
||
213 | int bom_get_max(void) { |
||
214 | int i, lowest_val, lowest_i;
|
||
215 | lowest_i = -1;
|
||
216 | 448 | cmar | lowest_val = 255;
|
217 | for(i = 0; i < NUM_BOM_LEDS; i++) { |
||
218 | 945 | abuchan | if(bom_val[i] < lowest_val) {
|
219 | lowest_val = bom_val[i]; |
||
220 | lowest_i = i; |
||
221 | } |
||
222 | 448 | cmar | } |
223 | 945 | abuchan | |
224 | if(lowest_val < BOM_VALUE_THRESHOLD)
|
||
225 | return lowest_i;
|
||
226 | else
|
||
227 | return -1; |
||
228 | 1443 | rcahoon | } |
229 | |||
230 | /**
|
||
231 | * Computes the weighted average of all the bom readings to estimate the position (and distance) of another robot.
|
||
232 | *
|
||
233 | * @pre must call bom refresh
|
||
234 | * @param dist pointer to int in which to return the estimated distance to the other robot
|
||
235 | * @return estimated position of the max bom value element as a fixed point value analogous to 10 times the
|
||
236 | * index of the max bom value. -1 if no value is lower than BOM_VALUE_THRESHOLD.
|
||
237 | **/
|
||
238 | int bom_get_max10(int *dist) { |
||
239 | int i, max;
|
||
240 | 1456 | rcahoon | long long mean = 0, sum = 0; |
241 | 1443 | rcahoon | |
242 | max = bom_get_max(); |
||
243 | if (max < 0) |
||
244 | { |
||
245 | if (dist)
|
||
246 | { |
||
247 | *dist = -1;
|
||
248 | } |
||
249 | return -1; |
||
250 | } |
||
251 | /* Record values into an array */
|
||
252 | for (i = 0; i < NUM_BOM_LEDS; i++) { |
||
253 | int idx = ((i + (NUM_BOM_LEDS/2 - max) + NUM_BOM_LEDS) % NUM_BOM_LEDS) - (NUM_BOM_LEDS/2 - max); |
||
254 | int val = 255 - bom_val[i]; |
||
255 | mean += idx * val; |
||
256 | sum += val; |
||
257 | } |
||
258 | mean = (mean * 10) / sum;
|
||
259 | mean = (mean + NUM_BOM_LEDS*10) % (NUM_BOM_LEDS*10); |
||
260 | |||
261 | if (dist)
|
||
262 | { |
||
263 | *dist = 50 - sum/48; |
||
264 | } |
||
265 | |||
266 | return mean;
|
||
267 | 945 | abuchan | } |
268 | |||
269 | /**
|
||
270 | * Iterates through each bit in the bit_field. If the bit is set, the corresponding emitter will
|
||
271 | * be enabled to turn on when bom_on() is called.
|
||
272 | 1051 | abuchan | * bom_init must be called for this to work. Does nothing if a BOM1.0 is installed
|
273 | 945 | abuchan | *
|
274 | 1051 | abuchan | * @param bit_field specifies which leds should be turned on when bom_on is called. Use BOM_ALL to turn on all bom leds.
|
275 | 945 | abuchan | * Ex. if 0x0005 is passed, leds 0 and 2 will be turned on.
|
276 | **/
|
||
277 | void bom_set_leds(int bit_field) { |
||
278 | int i;
|
||
279 | unsigned int mask = 1<<(NUM_BOM_LEDS-1); |
||
280 | switch(bom_type) {
|
||
281 | 1051 | abuchan | case BOM10:
|
282 | //TODO: put an assert here to alert the user that this should not be called
|
||
283 | 945 | abuchan | break;
|
284 | |||
285 | case BOM15:
|
||
286 | for(i=NUM_BOM_LEDS; i>0; i--) |
||
287 | { |
||
288 | //set the current bit, sending MSB first
|
||
289 | digital_output(BOM_DATA, bit_field&mask); |
||
290 | //then pulse the clock
|
||
291 | digital_output(BOM_CLOCK, 1);
|
||
292 | digital_output(BOM_CLOCK, 0);
|
||
293 | mask = mask>>1;
|
||
294 | } |
||
295 | break;
|
||
296 | |||
297 | case RBOM:
|
||
298 | //add rbom code here
|
||
299 | break;
|
||
300 | 448 | cmar | } |
301 | 945 | abuchan | } |
302 | 8 | bcoltin | |
303 | 448 | cmar | |
304 | 8 | bcoltin | /**
|
305 | 945 | abuchan | * (DEPRECATED) Returns the direction of the maximum BOM reading,
|
306 | * as an integer in the range 0-15. 0 indicates to the
|
||
307 | * robot's right, while the rest of the sensors are
|
||
308 | * numbered counterclockwise. This is useful for determining
|
||
309 | * the direction of a robot flashing its BOM, of only one
|
||
310 | * robot is currently doing so. analog_init must be called
|
||
311 | * before this function can be used.
|
||
312 | 8 | bcoltin | *
|
313 | * @return the direction of the maximum BOM reading
|
||
314 | *
|
||
315 | 945 | abuchan | * @see analog_init
|
316 | 8 | bcoltin | **/
|
317 | int get_max_bom(void) { |
||
318 | 448 | cmar | bom_refresh(BOM_ALL); |
319 | return bom_get_max();
|
||
320 | 8 | bcoltin | } |
321 | |||
322 | /**
|
||
323 | 945 | abuchan | * Flashes the BOM. If using a BOM1.5, only the emitters that have been enabled using
|
324 | * bom_set_leds will turn on.
|
||
325 | 8 | bcoltin | *
|
326 | 945 | abuchan | * @see bom_off, bom_set_leds
|
327 | 8 | bcoltin | **/
|
328 | void bom_on(void) |
||
329 | { |
||
330 | 945 | abuchan | switch(bom_type) {
|
331 | case BOM10:
|
||
332 | digital_output(MONKL, 1);
|
||
333 | break;
|
||
334 | case BOM15:
|
||
335 | digital_output(BOM_STROBE, 1);
|
||
336 | break;
|
||
337 | case RBOM:
|
||
338 | break;
|
||
339 | } |
||
340 | 8 | bcoltin | } |
341 | |||
342 | /**
|
||
343 | 945 | abuchan | * Turns off all bom leds.
|
344 | 8 | bcoltin | *
|
345 | 448 | cmar | * @see bom_on
|
346 | 8 | bcoltin | **/
|
347 | void bom_off(void) |
||
348 | { |
||
349 | 945 | abuchan | switch(bom_type) {
|
350 | 1051 | abuchan | case BOM10:
|
351 | 945 | abuchan | digital_output(MONKL, 0);
|
352 | break;
|
||
353 | case BOM15:
|
||
354 | digital_output(BOM_STROBE, 0);
|
||
355 | break;
|
||
356 | case RBOM:
|
||
357 | break;
|
||
358 | } |
||
359 | 8 | bcoltin | } |
360 | |||
361 | /** @} **/ //end group |
||
362 | |||
363 | 945 | abuchan | //select a detector to read
|
364 | static void bom_select(char which) { |
||
365 | 1051 | abuchan | if(bom_type == BOM10)
|
366 | 945 | abuchan | which = lookup[(int)which];
|
367 | |||
368 | if (which&8) |
||
369 | digital_output(select_pins[3], 1); |
||
370 | else
|
||
371 | digital_output(select_pins[3], 0); |
||
372 | 8 | bcoltin | |
373 | 945 | abuchan | if (which&4) |
374 | digital_output(select_pins[2], 1); |
||
375 | else
|
||
376 | digital_output(select_pins[2], 0); |
||
377 | 8 | bcoltin | |
378 | 945 | abuchan | if (which&2) |
379 | digital_output(select_pins[1], 1); |
||
380 | else
|
||
381 | digital_output(select_pins[1], 0); |
||
382 | |||
383 | if (which&1) |
||
384 | digital_output(select_pins[0], 1); |
||
385 | else
|
||
386 | digital_output(select_pins[0], 0); |
||
387 | |||
388 | } |