#include "strat_base.h"
#include "strat_corn.h"
#include "strat_utils.h"
+#include "strat_avoid.h"
#include "sensor.h"
#include "actuator.h"
#define COL_DISP_MARGIN 400 /* stop 40 cm in front of dispenser */
#define COL_SCAN_PRE_MARGIN 250
+static volatile uint8_t strat_running = 0;
+volatile uint8_t strat_want_pack = 0;
+volatile uint8_t strat_lpack60 = 0;
+volatile uint8_t strat_rpack60 = 0;
struct strat_conf strat_conf;
/*************************************************************/
mainboard.flags = DO_ENCODERS | DO_CS | DO_RS |
DO_POS | DO_BD | DO_POWER;
- //i2c_cobboard_mode_init();
+ strat_db_init();
strat_conf_dump(__FUNCTION__);
strat_db_dump(__FUNCTION__);
}
}
+void strat_event_enable(void)
+{
+ strat_running = 1;
+}
+
+void strat_event_disable(void)
+{
+ strat_running = 0;
+}
+
/* call it just before launching the strat */
void strat_init(void)
{
- /* XXX init rollers, .. */
-
- strat_db_init();
+#ifdef HOST_VERSION
+ position_set(&mainboard.pos, 298.16,
+ COLOR_Y(308.78), COLOR_A(70.00));
+#endif
/* we consider that the color is correctly set */
-
+ strat_running = 1;
strat_set_speed(SPEED_DIST_FAST, SPEED_ANGLE_FAST);
time_reset();
interrupt_traj_reset();
- i2c_cobboard_mode_harvest(I2C_LEFT_SIDE);
- i2c_cobboard_mode_harvest(I2C_RIGHT_SIDE);
+ i2c_cobboard_set_mode(I2C_COBBOARD_MODE_HARVEST);
i2c_ballboard_set_mode(I2C_BALLBOARD_MODE_HARVEST);
/* used in strat_base for END_TIMER */
DO_POS | DO_BD | DO_TIMER | DO_POWER;
}
-
/* call it after each strat */
void strat_exit(void)
{
uint8_t flags;
#endif
+ strat_running = 0;
mainboard.flags &= ~(DO_TIMER);
strat_hardstop();
time_reset();
#endif
}
-/* called periodically */
-void strat_event(void *dummy)
+/* mark tomato as not present */
+static void check_tomato(void)
{
-#if 0
- /* pack or deploy spickle */
- if (strat_infos.status.flags & STRAT_STATUS_LHARVEST) {
- if (sensor_get(S_LCOB_PRESENT)) {
- if (sensor_get(S_LCOB_WHITE))
- i2c_ballboard_set_mode();
+ int16_t x, y;
+ uint8_t i, j;
+
+ x = position_get_x_s16(&mainboard.pos);
+ y = position_get_y_s16(&mainboard.pos);
+
+ if (xycoord_to_ijcoord(&x, &y, &i, &j) < 0)
+ return;
+
+ if (strat_db.wp_table[i][j].type != WP_TYPE_TOMATO)
+ return;
+
+ strat_db.wp_table[i][j].present = 0;
+}
+
+/* mark corn as not present and give correct commands to the cobboard
+ * for spickles */
+static void check_corn(void)
+{
+ uint8_t flags;
+ int8_t lcob_near, rcob_near;
+ uint8_t lcob, rcob;
+ uint8_t lidx, ridx;
+
+ /* read sensors from ballboard */
+ IRQ_LOCK(flags);
+ lcob = ballboard.lcob;
+ ballboard.lcob = I2C_COB_NONE;
+ rcob = ballboard.rcob;
+ ballboard.rcob = I2C_COB_NONE;
+ IRQ_UNLOCK(flags);
+
+/* if (lcob != I2C_COB_NONE) */
+/* DEBUG(E_USER_STRAT, "XXX lcob %s", */
+/* lcob == I2C_COB_WHITE ? "white" : "black"); */
+/* if (rcob != I2C_COB_NONE) */
+/* DEBUG(E_USER_STRAT, "XXX rcob %s", */
+/* rcob == I2C_COB_WHITE ? "white" : "black"); */
+ /* XXX take opponent position into account */
+
+#ifdef HOST_VERSION
+ if (time_get_s() == 15)
+ cobboard.cob_count = 5;
+ if (time_get_s() == 16)
+ cobboard.cob_count = 0;
+#endif
+
+ /* detect cob on left side */
+ lcob_near = corn_is_near(&lidx, I2C_LEFT_SIDE);
+ if (lcob_near && lcob != I2C_COB_NONE) {
+ if (strat_db.corn_table[lidx]->corn.color == I2C_COB_UNKNOWN)
+ DEBUG(E_USER_STRAT, "lcob %s %d",
+ lcob == I2C_COB_WHITE ? "white" : "black", lidx);
+ corn_set_color(strat_db.corn_table[lidx], lcob);
+ }
+
+ /* detect cob on right side */
+ rcob_near = corn_is_near(&ridx, I2C_RIGHT_SIDE);
+ if (rcob_near && rcob != I2C_COB_NONE) {
+ if (strat_db.corn_table[ridx]->corn.color == I2C_COB_UNKNOWN)
+ DEBUG(E_USER_STRAT, "rcob %s %d",
+ rcob == I2C_COB_WHITE ? "white" : "black", ridx);
+ corn_set_color(strat_db.corn_table[ridx], rcob);
+ }
+
+ /* control the cobboard mode for left spickle */
+ if (lcob_near && strat_db.corn_table[lidx]->present) {
+ if (get_cob_count() >= 5 || strat_want_pack || strat_lpack60) {
+ /* nothing */
+ }
+ else {
+ /* deploy spickle and harvest white ones */
+ if (strat_db.corn_table[lidx]->corn.color == I2C_COB_WHITE) {
+ i2c_cobboard_autoharvest_nomove(I2C_LEFT_SIDE);
+ if (cobboard.status == I2C_COBBOARD_STATUS_LBUSY)
+ strat_db.corn_table[lidx]->present = 0;
+ }
else
- ;
+ i2c_cobboard_deploy_nomove(I2C_LEFT_SIDE);
}
}
-#endif
- /* limit speed when opponent is close */
- strat_limit_speed();
+ else {
+ /* no cob near us, we can pack or deploy freely */
+ if (get_cob_count() >= 5 || strat_want_pack || strat_lpack60)
+ i2c_cobboard_pack_weak(I2C_LEFT_SIDE);
+ else
+ i2c_cobboard_deploy(I2C_LEFT_SIDE);
+ }
+
+ /* control the cobboard mode for right spickle */
+ if (rcob_near && strat_db.corn_table[ridx]->present) {
+ if (get_cob_count() >= 5 || strat_want_pack || strat_rpack60) {
+ /* nothing */
+ }
+ else {
+ /* deploy spickle and harvest white ones */
+ if (strat_db.corn_table[ridx]->corn.color == I2C_COB_WHITE) {
+ i2c_cobboard_autoharvest_nomove(I2C_RIGHT_SIDE);
+ if (cobboard.status == I2C_COBBOARD_STATUS_RBUSY)
+ strat_db.corn_table[ridx]->present = 0;
+ }
+ else
+ i2c_cobboard_deploy_nomove(I2C_RIGHT_SIDE);
+ }
+ }
+ else {
+ /* no cob near us, we can pack or deploy freely */
+ if (get_cob_count() >= 5 || strat_want_pack || strat_rpack60)
+ i2c_cobboard_pack_weak(I2C_RIGHT_SIDE);
+ else
+ i2c_cobboard_deploy(I2C_RIGHT_SIDE);
+ }
}
-static uint8_t strat_beginning(void)
+/* called periodically (10ms) */
+void strat_event(void *dummy)
{
- uint8_t err;
+ /* ignore when strat is not running */
+ if (strat_running == 0)
+ return;
- strat_set_speed(250, SPEED_ANGLE_FAST);
+ check_tomato();
+ check_corn();
+ /* limit speed when opponent is near */
+ //strat_limit_speed();
+}
- l1:
- err = line2line(LINE_UP, 0, LINE_R_DOWN, 2);
- if (!TRAJ_SUCCESS(err)) {
- trajectory_hardstop(&mainboard.traj);
- time_wait_ms(2000);
- goto l1;
- }
- l2:
- err = line2line(LINE_R_DOWN, 2, LINE_R_UP, 2);
- if (!TRAJ_SUCCESS(err)) {
- trajectory_hardstop(&mainboard.traj);
- time_wait_ms(2000);
- goto l2;
- }
+/* must be called from a terminal line */
+static uint8_t strat_eject(void)
+{
+ uint8_t err;
- l3:
- err = line2line(LINE_R_UP, 2, LINE_UP, 5);
- if (!TRAJ_SUCCESS(err)) {
- trajectory_hardstop(&mainboard.traj);
- time_wait_ms(2000);
- goto l3;
+ /* XXX MUST be a on the line !! */
+ //XXX return vals
+
+ DEBUG(E_USER_STRAT, "%s()", __FUNCTION__);
+
+ /* go to eject point */
+ trajectory_goto_xy_abs(&mainboard.traj, 2625, COLOR_Y(1847));
+ err = WAIT_COND_OR_TRAJ_END(get_cob_count() >= 5,
+ TRAJ_FLAGS_NO_NEAR);
+ if (err != 0 && !TRAJ_SUCCESS(err))
+ return err;
+
+ strat_want_pack = 1;
+ if (err == 0) {
+ strat_set_speed(SPEED_CLITOID_FAST, SPEED_ANGLE_SLOW);
+ err = wait_traj_end(TRAJ_FLAGS_NO_NEAR);
}
- trajectory_hardstop(&mainboard.traj);
+ /* pack arms */
+ strat_event_disable();
+ i2c_cobboard_pack(I2C_LEFT_SIDE);
+ i2c_cobboard_pack(I2C_RIGHT_SIDE);
/* ball ejection */
- trajectory_a_abs(&mainboard.traj, COLOR_A(90));
i2c_ballboard_set_mode(I2C_BALLBOARD_MODE_EJECT);
+ trajectory_a_abs(&mainboard.traj, COLOR_A(70));
+ err = wait_traj_end(TRAJ_FLAGS_NO_NEAR);
+ if (!TRAJ_SUCCESS(err))
+ goto fail;
+
+ DEBUG(E_USER_STRAT, "%s():%d", __FUNCTION__, __LINE__);
+ strat_hardstop();
+#ifdef HOST_VERSION
time_wait_ms(2000);
+#else
+ WAIT_COND_OR_TIMEOUT(ballboard.status == I2C_BALLBOARD_STATUS_F_BUSY,
+ 2000);
+ WAIT_COND_OR_TIMEOUT(ballboard.status == I2C_BALLBOARD_STATUS_F_READY,
+ 2000);
+#endif
/* half turn */
- trajectory_goto_xy_abs(&mainboard.traj, 2625, COLOR_Y(1847));
- err = wait_traj_end(END_INTR|END_TRAJ);
- i2c_cobboard_mode_pack(I2C_LEFT_SIDE);
- i2c_cobboard_mode_pack(I2C_RIGHT_SIDE);
trajectory_a_rel(&mainboard.traj, COLOR_A(180));
err = wait_traj_end(END_INTR|END_TRAJ);
+ if (!TRAJ_SUCCESS(err))
+ goto fail;
/* cob ejection */
- trajectory_d_rel(&mainboard.traj, -100);
+ trajectory_d_rel(&mainboard.traj, -70);
err = wait_traj_end(END_INTR|END_TRAJ);
- i2c_cobboard_mode_eject();
+ if (!TRAJ_SUCCESS(err))
+ goto fail;
+
+ i2c_cobboard_set_mode(I2C_COBBOARD_MODE_EJECT);
+ strat_db_dump(__FUNCTION__);
time_wait_ms(2000);
- trajectory_hardstop(&mainboard.traj);
+ err = END_TRAJ;
+
+ fail:
+ strat_event_enable();
+ strat_want_pack = 0;
+ return err;
+}
+
+static uint8_t strat_beginning(void)
+{
+ uint8_t err;
+
+ strat_set_acc(ACC_DIST, ACC_ANGLE);
+ strat_set_speed(600, 60); /* OK */
+
+ trajectory_d_a_rel(&mainboard.traj, 1000, COLOR_A(20));
+ err = WAIT_COND_OR_TRAJ_END(trajectory_angle_finished(&mainboard.traj),
+ TRAJ_FLAGS_STD);
+
+ strat_set_acc(ACC_DIST, ACC_ANGLE);
+ strat_set_speed(SPEED_CLITOID_SLOW, SPEED_ANGLE_SLOW);
+
+ err = line2line(0, LINE_UP, 2, LINE_R_DOWN, TRAJ_FLAGS_NO_NEAR);
+ if (!TRAJ_SUCCESS(err))
+ return err;
+
+ err = line2line(2, LINE_R_DOWN, 2, LINE_R_UP, TRAJ_FLAGS_NO_NEAR);
+ if (!TRAJ_SUCCESS(err)) {
+ return err;
+ }
+
return END_TRAJ;
}
} while (0)
+#if 0
+/* return true if we need to grab some more elements */
+static uint8_t need_more_elements(void)
+{
+ if (time_get_s() <= 75) {
+ /* we have enough time left */
+ if (get_ball_count() >= 4)
+ return 0;
+ if (get_cob_count() >= 4)
+ return 0;
+ if ((get_ball_count() >= 2) &&
+ (get_cob_count() >= 2))
+ return 0;
+ return 1;
+ }
+ else {
+ /* not much time remaining */
+ if ((get_ball_count() >= 1) &&
+ (get_cob_count() >= 1))
+ return 0;
+ return 1;
+ }
+}
+#endif
+
uint8_t strat_main(void)
{
uint8_t err;
- /* */
+ /* harvest the first cobs + balls */
err = strat_beginning();
+ if (!TRAJ_SUCCESS(err))
+ strat_unblock();
+ else
+ err = strat_eject();
+
+ while (1) {
+
+ /**********************/
+ /* harvest on circuit */
+ /**********************/
+
+ DEBUG(E_USER_STRAT, "start main loop");
+
+ err = strat_harvest_circuit();
+ if (err == END_TIMER) {
+ DEBUG(E_USER_STRAT, "End of time");
+ strat_exit();
+ break;
+ }
+ if (!TRAJ_SUCCESS(err)) {
+ strat_unblock();
+ continue;
+ }
+
+ /***********************/
+ /* eject game elements */
+ /***********************/
+
+ err = strat_eject();
+ /* end of time exit ! */
+ if (err == END_TIMER) {
+ DEBUG(E_USER_STRAT, "End of time");
+ strat_exit();
+ break;
+ }
+ if (!TRAJ_SUCCESS(err)) {
+ strat_unblock();
+ continue;
+ }
+ }
return err;
}