beacon from 2009
[aversive.git] / projects / microb2010 / ballboard / cs.c
1 /*
2  *  Copyright Droids Corporation
3  *  Olivier Matz <zer0@droids-corp.org>
4  *
5  *  This program is free software; you can redistribute it and/or modify
6  *  it under the terms of the GNU General Public License as published by
7  *  the Free Software Foundation; either version 2 of the License, or
8  *  (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU General Public License for more details.
14  *
15  *  You should have received a copy of the GNU General Public License
16  *  along with this program; if not, write to the Free Software
17  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18  *
19  *  Revision : $Id: cs.c,v 1.4 2009-05-27 20:04:07 zer0 Exp $
20  *
21  */
22
23 #include <stdio.h>
24 #include <string.h>
25
26 #include <aversive.h>
27 #include <aversive/error.h>
28
29 #include <ax12.h>
30 #include <spi.h>
31 #include <encoders_spi.h>
32 #include <pwm_ng.h>
33 #include <timer.h>
34 #include <scheduler.h>
35 #include <clock_time.h>
36 #include <adc.h>
37
38 #include <pid.h>
39 #include <quadramp.h>
40 #include <control_system_manager.h>
41 #include <blocking_detection_manager.h>
42
43 #include <parse.h>
44 #include <rdline.h>
45
46 #include "main.h"
47 #include "actuator.h"
48 #include "beacon.h"
49
50 int32_t encoders_spi_update_roller_speed(void *number)
51 {
52         static volatile int32_t roller_pos;
53         int32_t tmp, speed;
54         tmp = encoders_spi_get_value(number);
55         speed = tmp - roller_pos;
56         roller_pos = tmp;
57         return speed;
58 }
59
60 /* called every 5 ms */
61 static void do_cs(void *dummy)
62 {
63         /* read encoders */
64         if (ballboard.flags & DO_ENCODERS) {
65                 encoders_spi_manage(NULL);
66         }
67         /* control system */
68         if (ballboard.flags & DO_CS) {
69                 if (ballboard.roller.on)
70                         cs_manage(&ballboard.roller.cs);
71                 if (ballboard.forktrans.on)
72                         cs_manage(&ballboard.forktrans.cs);
73                 if (ballboard.forkrot.on)
74                         cs_manage(&ballboard.forkrot.cs);
75                 if (ballboard.beacon.on)
76                         cs_manage(&ballboard.beacon.cs);
77         }
78         if ((ballboard.flags & DO_BD) && (ballboard.flags & DO_POWER)) {
79                 bd_manage_from_cs(&ballboard.forktrans.bd, &ballboard.forktrans.cs);
80                 bd_manage_from_cs(&ballboard.forkrot.bd, &ballboard.forkrot.cs);
81                 bd_manage_from_speed_cmd(&ballboard.roller.bd,
82                                          cs_get_filtered_feedback(&ballboard.roller.cs),
83                                          cs_get_out(&ballboard.roller.cs));
84
85                 /* urgent case: stop power on blocking */
86                 if (ballboard.flags & DO_ERRBLOCKING) {
87                         if (bd_get(&ballboard.forktrans.bd) ||
88                             bd_get(&ballboard.forkrot.bd)) {
89                                 printf_P(PSTR("MOTOR BLOCKED STOP ALL\r\n"));
90                                 ballboard.flags &= ~(DO_POWER | DO_ERRBLOCKING);
91                         }
92                 }
93         }
94         if (ballboard.flags & DO_POWER)
95                 BRAKE_OFF();
96         else
97                 BRAKE_ON();
98 }
99
100 void dump_cs(const char *name, struct cs *cs)
101 {
102         printf_P(PSTR("%s cons=% .5ld fcons=% .5ld err=% .5ld "
103                       "in=% .5ld out=% .5ld\r\n"),
104                  name, cs_get_consign(cs), cs_get_filtered_consign(cs),
105                  cs_get_error(cs), cs_get_filtered_feedback(cs),
106                  cs_get_out(cs));
107 }
108
109 void dump_pid(const char *name, struct pid_filter *pid)
110 {
111         printf_P(PSTR("%s P=% .8ld I=% .8ld D=% .8ld out=% .8ld\r\n"),
112                  name,
113                  pid_get_value_in(pid) * pid_get_gain_P(pid),
114                  pid_get_value_I(pid) * pid_get_gain_I(pid),
115                  pid_get_value_D(pid) * pid_get_gain_D(pid),
116                  pid_get_value_out(pid));
117 }
118
119 void microb_cs_init(void)
120 {
121         /* ---- CS roller */
122         /* PID */
123         pid_init(&ballboard.roller.pid);
124         pid_set_gains(&ballboard.roller.pid, 80, 80, 250);
125         pid_set_maximums(&ballboard.roller.pid, 0, 10000, 4095);
126         pid_set_out_shift(&ballboard.roller.pid, 6);
127         pid_set_derivate_filter(&ballboard.roller.pid, 6);
128
129         /* QUADRAMP (used as a ramp filter) */
130         quadramp_init(&ballboard.roller.qr);
131         quadramp_set_1st_order_vars(&ballboard.roller.qr, 20, 20);
132         quadramp_set_2nd_order_vars(&ballboard.roller.qr, 0, 0);
133
134         /* CS */
135         cs_init(&ballboard.roller.cs);
136         cs_set_consign_filter(&ballboard.roller.cs, quadramp_do_filter, &ballboard.roller.qr);
137         cs_set_correct_filter(&ballboard.roller.cs, pid_do_filter, &ballboard.roller.pid);
138         cs_set_process_in(&ballboard.roller.cs, pwm_ng_set, ROLLER_PWM);
139         cs_set_process_out(&ballboard.roller.cs, encoders_spi_update_roller_speed, ROLLER_ENCODER);
140         cs_set_consign(&ballboard.roller.cs, 0);
141
142         /* Blocking detection */
143         bd_init(&ballboard.roller.bd);
144 #define ROLLER_SPEED_THRES (ROLLER_SPEED * 0.75)
145         bd_set_speed_threshold(&ballboard.roller.bd, ROLLER_SPEED_THRES);
146         bd_set_current_thresholds(&ballboard.roller.bd, 500, 1500, 1200000, 70);
147
148         /* ---- CS forktrans */
149         /* PID */
150         pid_init(&ballboard.forktrans.pid);
151         pid_set_gains(&ballboard.forktrans.pid, 30, 5, 0);
152         pid_set_maximums(&ballboard.forktrans.pid, 0, 10000, 2047);
153         pid_set_out_shift(&ballboard.forktrans.pid, 6);
154         pid_set_derivate_filter(&ballboard.forktrans.pid, 6);
155
156         /* QUADRAMP */
157         quadramp_init(&ballboard.forktrans.qr);
158         quadramp_set_1st_order_vars(&ballboard.forktrans.qr, 200, 200); /* set speed */
159         quadramp_set_2nd_order_vars(&ballboard.forktrans.qr, 20, 20); /* set accel */
160
161         /* CS */
162         cs_init(&ballboard.forktrans.cs);
163         cs_set_consign_filter(&ballboard.forktrans.cs, quadramp_do_filter, &ballboard.forktrans.qr);
164         cs_set_correct_filter(&ballboard.forktrans.cs, pid_do_filter, &ballboard.forktrans.pid);
165         cs_set_process_in(&ballboard.forktrans.cs, pwm_ng_set, FORKTRANS_PWM);
166         cs_set_process_out(&ballboard.forktrans.cs, encoders_spi_get_value, FORKTRANS_ENCODER);
167         cs_set_consign(&ballboard.forktrans.cs, 0);
168
169         /* Blocking detection */
170         bd_init(&ballboard.forktrans.bd);
171         bd_set_speed_threshold(&ballboard.forktrans.bd, 150);
172         bd_set_current_thresholds(&ballboard.forktrans.bd, 500, 8000, 1000000, 200);
173
174         /* ---- CS forkrot */
175         /* PID */
176         pid_init(&ballboard.forkrot.pid);
177         pid_set_gains(&ballboard.forkrot.pid, 30, 5, 0);
178         pid_set_maximums(&ballboard.forkrot.pid, 0, 10000, 2047);
179         pid_set_out_shift(&ballboard.forkrot.pid, 6);
180         pid_set_derivate_filter(&ballboard.forkrot.pid, 6);
181
182         /* QUADRAMP */
183         quadramp_init(&ballboard.forkrot.qr);
184         quadramp_set_1st_order_vars(&ballboard.forkrot.qr, 200, 800); /* set speed */
185         quadramp_set_2nd_order_vars(&ballboard.forkrot.qr, 20, 20); /* set accel */
186
187         /* CS */
188         cs_init(&ballboard.forkrot.cs);
189         cs_set_consign_filter(&ballboard.forkrot.cs, quadramp_do_filter, &ballboard.forkrot.qr);
190         cs_set_correct_filter(&ballboard.forkrot.cs, pid_do_filter, &ballboard.forkrot.pid);
191         cs_set_process_in(&ballboard.forkrot.cs, pwm_ng_set, FORKROT_PWM);
192         cs_set_process_out(&ballboard.forkrot.cs, encoders_spi_get_value, FORKROT_ENCODER);
193         cs_set_consign(&ballboard.forkrot.cs, 0);
194
195         /* Blocking detection */
196         bd_init(&ballboard.forkrot.bd);
197         bd_set_speed_threshold(&ballboard.forkrot.bd, 150);
198         bd_set_current_thresholds(&ballboard.forkrot.bd, 500, 8000, 1000000, 200);
199
200         /* BEACON */
201
202         /* PID */
203         pid_init(&ballboard.beacon.pid);
204         pid_set_gains(&ballboard.beacon.pid, 80, 80, 250);
205         pid_set_maximums(&ballboard.beacon.pid, 0, 10000, 1500);
206         pid_set_out_shift(&ballboard.beacon.pid, 6);
207         pid_set_derivate_filter(&ballboard.beacon.pid, 6);
208
209         /* CS */
210         cs_init(&ballboard.beacon.cs);
211         cs_set_correct_filter(&ballboard.beacon.cs, pid_do_filter, &ballboard.beacon.pid);
212         cs_set_process_in(&ballboard.beacon.cs, pwm_ng_set, BEACON_PWM);
213         cs_set_process_out(&ballboard.beacon.cs, encoders_spi_update_beacon_speed, BEACON_ENCODER);
214         cs_set_consign(&ballboard.beacon.cs, 0);
215
216         /* set them on !! */
217         ballboard.roller.on = 1;
218         ballboard.forktrans.on = 1;
219         ballboard.forkrot.on = 0;
220         ballboard.beacon.on = 0;
221
222
223         scheduler_add_periodical_event_priority(do_cs, NULL,
224                                                 5000L / SCHEDULER_UNIT,
225                                                 CS_PRIO);
226 }