ioport_lpf2.c

// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2020 The Pybricks Authors

// LEGO Power Functions 2 I/O port

#include <pbdrv/config.h>

#if PBDRV_CONFIG_IOPORT_LPF2

#include <assert.h>
#include <stdbool.h>
#include <stdio.h>

#include <contiki.h>
#include <lego_uart.h>

#include <pbdrv/gpio.h>
#include <pbio/error.h>
#include <pbio/iodev.h>
#include <pbio/uartdev.h>

#include "ioport_lpf2.h"

typedef enum _dev_id1_group_t {
    DEV_ID1_GROUP_GND,
    DEV_ID1_GROUP_VCC,
    DEV_ID1_GROUP_PULL_DOWN,
    DEV_ID1_GROUP_OPEN,
} dev_id1_group_t;

// Device connection manager state for each port
typedef struct _dcm_data_t {
    dev_id1_group_t dev_id1_group;
    pbio_iodev_type_id_t type_id;
    pbio_iodev_type_id_t prev_type_id;
    uint8_t gpio_value;
    uint8_t prev_gpio_value;
    uint8_t dev_id_match_count;
} dcm_data_t;

typedef struct {
    pbio_iodev_t *iodev;
    const pbdrv_ioport_lpf2_platform_port_t *pins;
    dcm_data_t dcm;
    struct pt pt;
    pbio_iodev_type_id_t connected_type_id;
    pbio_iodev_type_id_t prev_type_id;
} ioport_dev_t;

typedef struct {
    pbio_iodev_info_t info;
    pbio_iodev_mode_t mode;
} basic_info_t;

static const basic_info_t basic_infos[] = {
    [PBIO_IODEV_TYPE_ID_LPF2_MMOTOR] = {
        .info = {
            .type_id = PBIO_IODEV_TYPE_ID_LPF2_MMOTOR,
            .num_modes = 1,
        },
        .mode = {
            .name = "LPF2-MMOTOR",
            .num_values = 1,
            .flags = {
                .flags0 = LUMP_MODE_FLAGS0_MOTOR_POWER,
                .flags4 = LUMP_MODE_FLAGS4_USES_HBRIDGE,
            },
            .num_values = 1,
            .data_type = PBIO_IODEV_DATA_TYPE_INT8,
            .digits = 4,
            .decimals = 0,
            .raw_min = -100.0,
            .raw_max = 100.0,
            .pct_min = -100.0,
            .pct_max = 100.0,
            .si_min = -100.0,
            .si_max = 100.0,
            .uom = "",
            .input_flags = 0,
            .output_flags = LPF2_MAPPING_FLAG_ABSOLUTE,
        },
    },
    [PBIO_IODEV_TYPE_ID_LPF2_TRAIN] = {
        .info = {
            .type_id = PBIO_IODEV_TYPE_ID_LPF2_TRAIN,
            .num_modes = 1,
        },
        .mode = {
            .name = "LPF2-TRAIN",
            .num_values = 1,
            .flags = {
                .flags0 = LUMP_MODE_FLAGS0_MOTOR_POWER,
                .flags4 = LUMP_MODE_FLAGS4_USES_HBRIDGE,
            },
            .num_values = 1,
            .data_type = PBIO_IODEV_DATA_TYPE_INT8,
            .digits = 4,
            .decimals = 0,
            .raw_min = -100.0,
            .raw_max = 100.0,
            .pct_min = -100.0,
            .pct_max = 100.0,
            .si_min = -100.0,
            .si_max = 100.0,
            .uom = "",
            .input_flags = 0,
            .output_flags = LPF2_MAPPING_FLAG_RELATIVE | LPF2_MAPPING_FLAG_ABSOLUTE,
        },
    },
    [PBIO_IODEV_TYPE_ID_LPF2_LIGHT] = {
        .info = {
            .type_id = PBIO_IODEV_TYPE_ID_LPF2_LIGHT,
            .num_modes = 1,
        },
        .mode = {
            .name = "LPF2-LIGHT",
            .num_values = 1,
            .flags = {
                .flags4 = LUMP_MODE_FLAGS4_USES_HBRIDGE,
            },
            .num_values = 1,
            .data_type = PBIO_IODEV_DATA_TYPE_INT8,
            .digits = 1,
            .decimals = 0,
            .raw_min = 0.0,
            .raw_max = 1.0,
            .pct_min = 0.0,
            .pct_max = 100.0,
            .si_min = 0.0,
            .si_max = 1.0,
            .uom = "",
            .input_flags = 0,
            .output_flags = LPF2_MAPPING_FLAG_DISCRETE | LPF2_MAPPING_FLAG_ABSOLUTE,
        },
    },
    [PBIO_IODEV_TYPE_ID_LPF2_LIGHT1] = {
        .info = {
            .type_id = PBIO_IODEV_TYPE_ID_LPF2_LIGHT1,
            .num_modes = 1,
        },
        .mode = {
            .name = "LPF2-LIGHT1",
            .num_values = 1,
            .flags = {
                .flags4 = LUMP_MODE_FLAGS4_USES_HBRIDGE,
            },
            .num_values = 1,
            .data_type = PBIO_IODEV_DATA_TYPE_INT8,
            .digits = 1,
            .decimals = 0,
            .raw_min = -100.0,
            .raw_max = 100.0,
            .pct_min = -100.0,
            .pct_max = 100.0,
            .si_min = -100.0,
            .si_max = 100.0,
            .uom = "",
            .input_flags = 0,
            .output_flags = LPF2_MAPPING_FLAG_DISCRETE | LPF2_MAPPING_FLAG_ABSOLUTE,
        },
    },
    [PBIO_IODEV_TYPE_ID_LPF2_LIGHT2] = {
        .info = {
            .type_id = PBIO_IODEV_TYPE_ID_LPF2_LIGHT2,
            .num_modes = 1,
        },
        .mode = {
            .name = "LPF2-LIGHT2",
            .num_values = 1,
            .flags = {
                .flags4 = LUMP_MODE_FLAGS4_USES_HBRIDGE,
            },
            .num_values = 1,
            .data_type = PBIO_IODEV_DATA_TYPE_INT8,
            .digits = 1,
            .decimals = 0,
            .raw_min = 0.0,
            .raw_max = 1.0,
            .pct_min = 0.0,
            .pct_max = 100.0,
            .si_min = 0.0,
            .si_max = 1.0,
            .uom = "",
            .input_flags = 0,
            .output_flags = LPF2_MAPPING_FLAG_DISCRETE | LPF2_MAPPING_FLAG_ABSOLUTE,
        },
    },
};

static pbio_iodev_t basic_devs[PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS];

static const pbio_iodev_type_id_t ioport_type_id_lookup[3][3] = {
    [DEV_ID1_GROUP_GND] = {
        [0] = PBIO_IODEV_TYPE_ID_LPF2_POWER,
        [1] = PBIO_IODEV_TYPE_ID_LPF2_TURN,
        [2] = PBIO_IODEV_TYPE_ID_LPF2_LIGHT2,
    },
    [DEV_ID1_GROUP_VCC] = {
        [0] = PBIO_IODEV_TYPE_ID_LPF2_TRAIN,
        [1] = PBIO_IODEV_TYPE_ID_LPF2_LMOTOR,
        [2] = PBIO_IODEV_TYPE_ID_LPF2_LIGHT1,
    },
    [DEV_ID1_GROUP_PULL_DOWN] = {
        [0] = PBIO_IODEV_TYPE_ID_LPF2_MMOTOR,
        [1] = PBIO_IODEV_TYPE_ID_LPF2_XMOTOR,
        [2] = PBIO_IODEV_TYPE_ID_LPF2_LIGHT,
    },
};

PROCESS(pbdrv_ioport_lpf2_process, "I/O port");

static ioport_dev_t ioport_devs[PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS] = {
    #if PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS > 0
    [0] = {
        .pins = &pbdrv_ioport_lpf2_platform_port_0,
    },
    #endif
    #if PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS > 1
    [1] = {
        .pins = &pbdrv_ioport_lpf2_platform_port_1,
    },
    #endif
    #if PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS > 2
    [2] = {
        .pins = &pbdrv_ioport_lpf2_platform_port_2,
    },
    #endif
    #if PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS > 3
    [3] = {
        .pins = &pbdrv_ioport_lpf2_platform_port_3,
    },
    #endif
    #if PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS > 4
    [4] = {
        .pins = &pbdrv_ioport_lpf2_platform_port_4,
    },
    #endif
    #if PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS > 5
    [5] = {
        .pins = &pbdrv_ioport_lpf2_platform_port_5,
    },
    #endif
};

static void ioport_enable_uart(ioport_dev_t *ioport) {
    const pbdrv_ioport_lpf2_platform_port_t *pins = ioport->pins;

    pbdrv_gpio_alt(&pins->uart_rx, pins->alt);
    pbdrv_gpio_alt(&pins->uart_tx, pins->alt);
    pbdrv_gpio_out_low(&pins->uart_buf);
}

static const pbio_iodev_ops_t basic_dev_ops = {
};

static void init_one(uint8_t ioport) {
    const pbdrv_ioport_lpf2_platform_port_t *pins = ioport_devs[ioport].pins;

    PT_INIT(&ioport_devs[ioport].pt);

    pbdrv_gpio_input(&pins->id1);
    pbdrv_gpio_input(&pins->id2);
    pbdrv_gpio_input(&pins->uart_buf);
    pbdrv_gpio_input(&pins->uart_tx);
    pbdrv_gpio_input(&pins->uart_rx);

    basic_devs[ioport].port = PBDRV_CONFIG_IOPORT_LPF2_FIRST_PORT + ioport;
    basic_devs[ioport].ops = &basic_dev_ops;
}

// TODO: This should be moved to a common ioport_core.c file or removed entirely
pbio_error_t pbdrv_ioport_get_iodev(pbio_port_t port, pbio_iodev_t **iodev) {
    if (port < PBDRV_CONFIG_IOPORT_LPF2_FIRST_PORT || port > PBDRV_CONFIG_IOPORT_LPF2_LAST_PORT) {
        return PBIO_ERROR_INVALID_PORT;
    }

    *iodev = ioport_devs[port - PBDRV_CONFIG_IOPORT_LPF2_FIRST_PORT].iodev;
    if (*iodev == NULL) {
        return PBIO_ERROR_NO_DEV;
    }

    return PBIO_SUCCESS;
}

// This is the device connection manager (dcm). It monitors the ID1 and ID2 pins
// on the port to see when devices are connected or disconnected.
// It is expected for there to be a 2ms delay between calls to this function.
static PT_THREAD(poll_dcm(ioport_dev_t * ioport)) {
    struct pt *pt = &ioport->pt;
    dcm_data_t *data = &ioport->dcm;
    const pbdrv_ioport_lpf2_platform_port_t pins = *ioport->pins;

    PT_BEGIN(pt);

    data->type_id = PBIO_IODEV_TYPE_ID_NONE;
    data->dev_id1_group = DEV_ID1_GROUP_OPEN;

    // set ID1 high
    pbdrv_gpio_out_high(&pins.uart_tx);
    pbdrv_gpio_out_low(&pins.uart_buf);

    // set ID2 as input
    pbdrv_gpio_input(&pins.id2);

    PT_YIELD(pt);

    // save current ID2 value
    data->prev_gpio_value = pbdrv_gpio_input(&pins.id2);

    // set ID1 low
    pbdrv_gpio_out_low(&pins.uart_tx);

    PT_YIELD(pt);

    // read ID2
    data->gpio_value = pbdrv_gpio_input(&pins.id2);

    // if ID2 changed from high to low
    if (data->prev_gpio_value == 1 && data->gpio_value == 0) {
        // we have touch sensor
        data->type_id = PBIO_IODEV_TYPE_ID_LPF2_TOUCH;

        // set ID1 as input
        pbdrv_gpio_out_high(&pins.uart_buf);
        pbdrv_gpio_input(&pins.uart_tx);

        PT_YIELD(pt);

        // ID1 is inverse of touch sensor value
        // TODO: save this value to sensor data
        // sensor_data = !pbdrv_gpio_input(&pins.id1);
    }
    // if ID2 changed from low to high
    else if (data->prev_gpio_value == 0 && data->gpio_value == 1) {
        data->type_id = PBIO_IODEV_TYPE_ID_LPF2_TPOINT;
    } else {
        // read ID1
        data->prev_gpio_value = pbdrv_gpio_input(&pins.id1);

        // set ID1 high
        pbdrv_gpio_out_high(&pins.uart_tx);

        PT_YIELD(pt);

        // read ID1
        data->gpio_value = pbdrv_gpio_input(&pins.id1);

        // if ID1 did not change and is high
        if (data->prev_gpio_value == 1 && data->gpio_value == 1) {
            // we have ID1 == VCC
            data->dev_id1_group = DEV_ID1_GROUP_VCC;
        }
        // if ID1 did not change and is low
        else if (data->prev_gpio_value == 0 && data->gpio_value == 0) {
            // we have ID1 == GND
            data->dev_id1_group = DEV_ID1_GROUP_GND;
        } else {
            // set ID1 as input
            pbdrv_gpio_out_high(&pins.uart_buf);
            pbdrv_gpio_input(&pins.uart_tx);

            PT_YIELD(pt);

            // read ID1
            if (pbdrv_gpio_input(&pins.id1) == 1) {
                // we have ID1 == open
                data->dev_id1_group = DEV_ID1_GROUP_OPEN;
            } else {
                // we have ID1 == pull down
                data->dev_id1_group = DEV_ID1_GROUP_PULL_DOWN;
            }
        }

        PT_YIELD(pt);

        // set ID1 as input
        pbdrv_gpio_out_high(&pins.uart_buf);
        pbdrv_gpio_input(&pins.uart_tx);

        // set ID2 high
        pbdrv_gpio_out_high(&pins.id2);

        PT_YIELD(pt);

        // read ID1
        data->prev_gpio_value = pbdrv_gpio_input(&pins.id1);

        // set ID2 low
        pbdrv_gpio_out_low(&pins.id2);

        PT_YIELD(pt);

        // read ID1
        data->gpio_value = pbdrv_gpio_input(&pins.id1);

        // if ID1 changed from high to low
        if (data->prev_gpio_value == 1 && data->gpio_value == 0) {
            // if we have ID1 = open
            if (data->dev_id1_group == DEV_ID1_GROUP_OPEN) {
                // then we have this
                data->type_id = PBIO_IODEV_TYPE_ID_LPF2_3_PART;
            }
        }
        // if ID1 changed from low to high
        else if (data->prev_gpio_value == 0 && data->gpio_value == 1) {
            // something might explode
            data->type_id = PBIO_IODEV_TYPE_ID_LPF2_EXPLOD;
        } else {
            // set ID1 high
            pbdrv_gpio_out_high(&pins.uart_tx);
            pbdrv_gpio_out_low(&pins.uart_buf);

            // set ID2 high
            pbdrv_gpio_out_high(&pins.id2);

            PT_YIELD(pt);

            // if ID2 is high
            if (pbdrv_gpio_input(&pins.uart_rx) == 1) {
                // set ID2 low
                pbdrv_gpio_out_low(&pins.id2);

                PT_YIELD(pt);

                // if ID2 is low
                if (pbdrv_gpio_input(&pins.uart_rx) == 0) {
                    if (data->dev_id1_group < 3) {
                        data->type_id = ioport_type_id_lookup[data->dev_id1_group][2];
                    }
                } else {
                    if (data->dev_id1_group < 3) {
                        data->type_id = ioport_type_id_lookup[data->dev_id1_group][1];
                    }
                }
            } else {
                // we know the device now
                if (data->dev_id1_group < 3) {
                    data->type_id = ioport_type_id_lookup[data->dev_id1_group][0];
                } else {
                    data->type_id = PBIO_IODEV_TYPE_ID_LPF2_UNKNOWN_UART;
                }
            }
        }
    }

    PT_YIELD(pt);

    // set ID2 as input
    pbdrv_gpio_input(&pins.id2);

    // set ID1 high
    pbdrv_gpio_out_high(&pins.uart_tx);
    pbdrv_gpio_out_low(&pins.uart_buf);

    if (data->type_id == data->prev_type_id) {
        if (++data->dev_id_match_count >= 20) {

            if (data->type_id != ioport->connected_type_id) {
                ioport->connected_type_id = data->type_id;
            }

            // don't want to wrap around and re-trigger
            data->dev_id_match_count--;
        }
    }

    data->prev_type_id = data->type_id;

    PT_END(pt);
}

PROCESS_THREAD(pbdrv_ioport_lpf2_process, ev, data) {
    static struct etimer timer;

    PROCESS_BEGIN();

    etimer_set(&timer, clock_from_msec(2));

    for (int i = 0; i < PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS; i++) {
        init_one(i);
    }

    while (true) {
        PROCESS_WAIT_EVENT_UNTIL(ev == PROCESS_EVENT_TIMER && etimer_expired(&timer));
        etimer_reset(&timer);

        for (int i = 0; i < PBDRV_CONFIG_IOPORT_LPF2_NUM_PORTS; i++) {
            ioport_dev_t *ioport = &ioport_devs[i];

            if (ioport->connected_type_id != PBIO_IODEV_TYPE_ID_LPF2_UNKNOWN_UART) {
                poll_dcm(ioport);
            }

            if (ioport->connected_type_id != ioport->prev_type_id) {
                ioport->prev_type_id = ioport->connected_type_id;
                if (ioport->connected_type_id == PBIO_IODEV_TYPE_ID_LPF2_UNKNOWN_UART) {
                    ioport_enable_uart(ioport);
                    pbio_uartdev_get(i, &ioport->iodev);
                } else if (ioport->connected_type_id == PBIO_IODEV_TYPE_ID_NONE) {
                    ioport->iodev = NULL;
                } else {
                    assert(ioport->connected_type_id < PBIO_IODEV_TYPE_ID_LPF2_UNKNOWN_UART);
                    pbio_iodev_t *iodev = &basic_devs[i];
                    const pbio_iodev_info_t *info = &basic_infos[ioport->connected_type_id].info;
                    iodev->info = info;

                    const lump_mode_flags_t *flags = &info->mode_info[0].flags;
                    iodev->motor_flags = PBIO_IODEV_MOTOR_FLAG_NONE;
                    if (flags->flags0 & LUMP_MODE_FLAGS0_MOTOR_POWER) {
                        iodev->motor_flags |= PBIO_IODEV_MOTOR_FLAG_IS_MOTOR;
                    }
                    ioport->iodev = iodev;
                }
            }
        }
    }

    PROCESS_END();
}

#endif // PBDRV_CONFIG_IOPORT_LPF2