LibrePilot/flight/pios/stm32f4xx/pios_servo.c

/**
 ******************************************************************************
 * @addtogroup PIOS PIOS Core hardware abstraction layer
 * @{
 * @addtogroup   PIOS_SERVO RC Servo Functions
 * @brief Code to do set RC servo output
 * @{
 *
 * @file       pios_servo.c
 * @author     The OpenPilot Team, http://www.openpilot.org Copyright (C) 2012.
 * @brief      RC Servo routines (STM32 dependent)
 * @see        The GNU Public License (GPL) Version 3
 *
 *****************************************************************************/
/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
 * for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#include "pios.h"

#ifdef PIOS_INCLUDE_SERVO

#include "pios_servo_priv.h"
#include "pios_tim_priv.h"

/* Private Function Prototypes */

static const struct pios_servo_cfg *servo_cfg;

// determine if the related timer will work in synchronous (or OneShot/OneShot125) One Pulse mode.
static uint8_t pios_servo_bank_mode[PIOS_SERVO_BANKS] = { 0 };

// timer associated to each bank
static TIM_TypeDef *pios_servo_bank_timer[PIOS_SERVO_BANKS] = { 0 };

// index of bank used for each pin
static uint8_t *pios_servo_pin_bank;


/**
 * Initialise Servos
 */
int32_t PIOS_Servo_Init(const struct pios_servo_cfg *cfg)
{
    uint32_t tim_id;

    if (PIOS_TIM_InitChannels(&tim_id, cfg->channels, cfg->num_channels, NULL, 0)) {
        return -1;
    }

    /* Store away the requested configuration */
    servo_cfg = cfg;
    pios_servo_pin_bank = pios_malloc(sizeof(uint8_t) * cfg->num_channels);

    uint8_t bank = 0;
    for (uint8_t i = 0; (i < servo_cfg->num_channels); i++) {
        const struct pios_tim_channel *chan = &servo_cfg->channels[i];
        bool new = true;
        /* See if any previous channels use that same timer */
        for (uint8_t j = 0; (j < i) && new; j++) {
            new &= chan->timer != servo_cfg->channels[j].timer;
        }

        if (new) {
            PIOS_Assert(bank < PIOS_SERVO_BANKS);
            for (uint8_t j = i; j < servo_cfg->num_channels; j++) {
                if (servo_cfg->channels[j].timer == chan->timer) {
                    pios_servo_pin_bank[j] = bank;
                }
            }
            pios_servo_bank_timer[i] = chan->timer;

            TIM_ARRPreloadConfig(chan->timer, ENABLE);
            TIM_CtrlPWMOutputs(chan->timer, ENABLE);
            TIM_Cmd(chan->timer, DISABLE);

            bank++;
        }

        /* Set up for output compare function */
        switch (chan->timer_chan) {
        case TIM_Channel_1:
            TIM_OC1Init(chan->timer, &servo_cfg->tim_oc_init);
            TIM_OC1PreloadConfig(chan->timer, TIM_OCPreload_Enable);
            TIM_OC1PolarityConfig(chan->timer, TIM_OCPolarity_Low);
            break;
        case TIM_Channel_2:
            TIM_OC2Init(chan->timer, &servo_cfg->tim_oc_init);
            TIM_OC2PreloadConfig(chan->timer, TIM_OCPreload_Enable);
            TIM_OC2PolarityConfig(chan->timer, TIM_OCPolarity_Low);
            break;
        case TIM_Channel_3:
            TIM_OC3Init(chan->timer, &servo_cfg->tim_oc_init);
            TIM_OC3PreloadConfig(chan->timer, TIM_OCPreload_Enable);
            TIM_OC3PolarityConfig(chan->timer, TIM_OCPolarity_Low);
            break;
        case TIM_Channel_4:
            TIM_OC4Init(chan->timer, &servo_cfg->tim_oc_init);
            TIM_OC4PreloadConfig(chan->timer, TIM_OCPreload_Enable);
            TIM_OC4PolarityConfig(chan->timer, TIM_OCPolarity_Low);
            break;
        }
    }

    return 0;
}

void PIOS_Servo_SetBankMode(uint8_t bank, uint8_t mode)
{
    PIOS_Assert(bank < PIOS_SERVO_BANKS);
    pios_servo_bank_mode[bank] = mode;

    for (uint8_t i = 0; (i < servo_cfg->num_channels); i++) {
        const struct pios_tim_channel *chan = &servo_cfg->channels[i];
        switch (mode) {
        case PIOS_SERVO_BANK_MODE_SINGLE_PULSE:
            /* Set up for output compare function */
            switch (chan->timer_chan) {
            case TIM_Channel_1:
                TIM_OC1PolarityConfig(chan->timer, TIM_OCPolarity_Low);
                break;
            case TIM_Channel_2:
                TIM_OC2PolarityConfig(chan->timer, TIM_OCPolarity_Low);
                break;
            case TIM_Channel_3:
                TIM_OC3PolarityConfig(chan->timer, TIM_OCPolarity_Low);
                break;
            case TIM_Channel_4:
                TIM_OC4PolarityConfig(chan->timer, TIM_OCPolarity_Low);
                break;
            }
            break;
        case PIOS_SERVO_BANK_MODE_PWM:
            /* Set up for output compare function */
            switch (chan->timer_chan) {
            case TIM_Channel_1:
                TIM_OC1PolarityConfig(chan->timer, TIM_OCPolarity_Low);
                break;
            case TIM_Channel_2:
                TIM_OC2PolarityConfig(chan->timer, TIM_OCPolarity_Low);
                break;
            case TIM_Channel_3:
                TIM_OC3PolarityConfig(chan->timer, TIM_OCPolarity_Low);
                break;
            case TIM_Channel_4:
                TIM_OC4PolarityConfig(chan->timer, TIM_OCPolarity_Low);
                break;
            }
            break;
        default:
            PIOS_Assert(false);
        }
    }
    // Setup the timer accordingly
    switch (mode) {
    case PIOS_SERVO_BANK_MODE_SINGLE_PULSE:
        TIM_SelectOnePulseMode(pios_servo_bank_timer[bank], TIM_OPMode_Single);
        TIM_CounterModeConfig(pios_servo_bank_timer[bank], TIM_CounterMode_Up);
        break;
    case PIOS_SERVO_BANK_MODE_PWM:
        TIM_SelectOnePulseMode(pios_servo_bank_timer[bank], TIM_OPMode_Repetitive);
        TIM_Cmd(pios_servo_bank_timer[bank], ENABLE);
        break;
    default:
        PIOS_Assert(false);
    }
}

void PIOS_Servo_Update()
{
    for (uint8_t i = 0; (i < PIOS_SERVO_BANKS); i++) {
        const TIM_TypeDef *timer = pios_servo_bank_timer[i];
        if (timer) {
            TIM_Cmd((TIM_TypeDef *)timer, ENABLE);
        }
    }
}

/**
 * Set the servo update rate (Max 500Hz)
 * \param[in] array of rates in Hz
 * \param[in] maximum number of banks
 */
void PIOS_Servo_SetHz(const uint16_t *speeds, uint8_t banks)
{
    PIOS_Assert(banks <= PIOS_SERVO_BANKS);
    if (!servo_cfg) {
        return;
    }

    TIM_TimeBaseInitTypeDef TIM_TimeBaseStructure = servo_cfg->tim_base_init;
    TIM_TimeBaseStructure.TIM_ClockDivision = TIM_CKD_DIV1;
    TIM_TimeBaseStructure.TIM_CounterMode   = TIM_CounterMode_Up;

    for (uint8_t i = 0; (i < PIOS_SERVO_BANKS); i++) {
        const TIM_TypeDef *timer = pios_servo_bank_timer[i];
        if (timer) {
            // Choose the correct prescaler value for the APB the timer is attached
            if (timer == TIM1 || timer == TIM8 || timer == TIM9 || timer == TIM10 || timer == TIM11) {
                TIM_TimeBaseStructure.TIM_Prescaler = (PIOS_PERIPHERAL_APB2_CLOCK / 1000000) - 1;
            } else {
                TIM_TimeBaseStructure.TIM_Prescaler = (PIOS_PERIPHERAL_APB1_CLOCK / 1000000) - 1;
            }

            TIM_TimeBaseStructure.TIM_Period = ((1000000 / speeds[i]) - 1);
            TIM_TimeBaseInit((TIM_TypeDef *)timer, &TIM_TimeBaseStructure);
        }
    }
}

/**
 * Set servo position
 * \param[in] Servo Servo number (0-7)
 * \param[in] Position Servo position in microseconds
 */
void PIOS_Servo_Set(uint8_t servo, uint16_t position)
{
    /* Make sure servo exists */
    if (!servo_cfg || servo >= servo_cfg->num_channels) {
        return;
    }

    /* Update the position */
    const struct pios_tim_channel *chan = &servo_cfg->channels[servo];

    uint8_t bank = pios_servo_pin_bank[servo];
    uint8_t mode = pios_servo_bank_mode[bank];
    uint16_t val;
    switch (mode) {
    case PIOS_SERVO_BANK_MODE_PWM:
        val = position;
        break;
    case PIOS_SERVO_BANK_MODE_SINGLE_PULSE:
        val = chan->timer->ARR - position;
        break;
    default:
        PIOS_Assert(false);
    }

    switch (chan->timer_chan) {
    case TIM_Channel_1:
        TIM_SetCompare1(chan->timer, val);
        break;
    case TIM_Channel_2:
        TIM_SetCompare2(chan->timer, val);
        break;
    case TIM_Channel_3:
        TIM_SetCompare3(chan->timer, val);
        break;
    case TIM_Channel_4:
        TIM_SetCompare4(chan->timer, val);
        break;
    }
}

uint8_t PIOS_Servo_GetPinBank(uint8_t pin)
{
    if (pin < servo_cfg->num_channels) {
        return pios_servo_pin_bank[pin];
    } else {
        return 0;
    }
}

#endif /* PIOS_INCLUDE_SERVO */