mirror of
https://github.com/betaflight/betaflight.git
synced 2025-07-13 03:20:00 +03:00
aae8ef4c3d
git-svn-id: https://afrodevices.googlecode.com/svn/trunk/baseflight@104 7c89a4a9-59b9-e629-4cfe-3a2d53b20e61
315 lines
10 KiB
C
Executable file
315 lines
10 KiB
C
Executable file
#include "board.h"
|
|
|
|
#define PULSE_1MS (1000) // 1ms pulse width
|
|
#define PULSE_PERIOD (2500) // pulse period (400Hz)
|
|
#define PULSE_PERIOD_SERVO_DIGITAL (5000) // pulse period for digital servo (200Hz)
|
|
#define PULSE_PERIOD_SERVO_ANALOG (20000) // pulse period for analog servo (50Hz)
|
|
|
|
// Forward declaration
|
|
static void pwmIRQHandler(TIM_TypeDef *tim);
|
|
static void ppmIRQHandler(TIM_TypeDef *tim);
|
|
|
|
// local vars
|
|
static struct TIM_Channel {
|
|
TIM_TypeDef *tim;
|
|
uint16_t channel;
|
|
uint16_t cc;
|
|
} Channels[] = {
|
|
{ TIM2, TIM_Channel_1, TIM_IT_CC1 },
|
|
{ TIM2, TIM_Channel_2, TIM_IT_CC2 },
|
|
{ TIM2, TIM_Channel_3, TIM_IT_CC3 },
|
|
{ TIM2, TIM_Channel_4, TIM_IT_CC4 },
|
|
{ TIM3, TIM_Channel_1, TIM_IT_CC1 },
|
|
{ TIM3, TIM_Channel_2, TIM_IT_CC2 },
|
|
{ TIM3, TIM_Channel_3, TIM_IT_CC3 },
|
|
{ TIM3, TIM_Channel_4, TIM_IT_CC4 },
|
|
};
|
|
|
|
static volatile uint16_t *OutputChannels[] = {
|
|
&(TIM1->CCR1),
|
|
&(TIM1->CCR4),
|
|
&(TIM4->CCR1),
|
|
&(TIM4->CCR2),
|
|
&(TIM4->CCR3),
|
|
&(TIM4->CCR4),
|
|
};
|
|
|
|
static struct PWM_State {
|
|
uint8_t state;
|
|
uint16_t rise;
|
|
uint16_t fall;
|
|
uint16_t capture;
|
|
} Inputs[8] = { { 0, } };
|
|
|
|
static TIM_ICInitTypeDef TIM_ICInitStructure = { 0, };
|
|
static bool usePPMFlag = false;
|
|
|
|
void TIM2_IRQHandler(void)
|
|
{
|
|
if (usePPMFlag)
|
|
ppmIRQHandler(TIM2);
|
|
else
|
|
pwmIRQHandler(TIM2);
|
|
}
|
|
|
|
void TIM3_IRQHandler(void)
|
|
{
|
|
pwmIRQHandler(TIM3);
|
|
}
|
|
|
|
static void ppmIRQHandler(TIM_TypeDef *tim)
|
|
{
|
|
uint16_t diff;
|
|
static uint16_t now;
|
|
static uint16_t last = 0;
|
|
static uint8_t chan = 0;
|
|
|
|
if (TIM_GetITStatus(TIM2, TIM_IT_CC1) == SET) {
|
|
last = now;
|
|
now = TIM_GetCapture1(TIM2);
|
|
}
|
|
|
|
TIM_ClearITPendingBit(TIM2, TIM_IT_CC1);
|
|
|
|
if (now > last) {
|
|
diff = (now - last);
|
|
} else {
|
|
diff = ((0xFFFF - last) + now);
|
|
}
|
|
|
|
if (diff > 4000) {
|
|
chan = 0;
|
|
} else {
|
|
if (diff > 750 && diff < 2250 && chan < 8) { // 750 to 2250 ms is our 'valid' channel range
|
|
Inputs[chan].capture = diff;
|
|
}
|
|
chan++;
|
|
}
|
|
}
|
|
|
|
static void pwmIRQHandler(TIM_TypeDef *tim)
|
|
{
|
|
uint8_t i;
|
|
uint16_t val = 0;
|
|
|
|
for (i = 0; i < 8; i++) {
|
|
struct TIM_Channel channel = Channels[i];
|
|
struct PWM_State *state = &Inputs[i];
|
|
|
|
if (channel.tim == tim && (TIM_GetITStatus(tim, channel.cc) == SET)) {
|
|
TIM_ClearITPendingBit(channel.tim, channel.cc);
|
|
|
|
switch (channel.channel) {
|
|
case TIM_Channel_1:
|
|
val = TIM_GetCapture1(channel.tim);
|
|
break;
|
|
case TIM_Channel_2:
|
|
val = TIM_GetCapture2(channel.tim);
|
|
break;
|
|
case TIM_Channel_3:
|
|
val = TIM_GetCapture3(channel.tim);
|
|
break;
|
|
case TIM_Channel_4:
|
|
val = TIM_GetCapture4(channel.tim);
|
|
break;
|
|
}
|
|
|
|
if (state->state == 0)
|
|
state->rise = val;
|
|
else
|
|
state->fall = val;
|
|
|
|
if (state->state == 0) {
|
|
// switch states
|
|
state->state = 1;
|
|
|
|
TIM_ICInitStructure.TIM_ICPolarity = TIM_ICPolarity_Falling;
|
|
TIM_ICInitStructure.TIM_Channel = channel.channel;
|
|
TIM_ICInit(channel.tim, &TIM_ICInitStructure);
|
|
} else {
|
|
// compute capture
|
|
if (state->fall > state->rise)
|
|
state->capture = (state->fall - state->rise);
|
|
else
|
|
state->capture = ((0xffff - state->rise) + state->fall);
|
|
|
|
// switch state
|
|
state->state = 0;
|
|
|
|
TIM_ICInitStructure.TIM_ICPolarity = TIM_ICPolarity_Rising;
|
|
TIM_ICInitStructure.TIM_Channel = channel.channel;
|
|
TIM_ICInit(channel.tim, &TIM_ICInitStructure);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void pwmInit(bool usePPM, bool useServos, bool useDigitalServos)
|
|
{
|
|
uint8_t i;
|
|
GPIO_InitTypeDef GPIO_InitStructure = { 0, };
|
|
TIM_TimeBaseInitTypeDef TIM_TimeBaseStructure = { 0, };
|
|
TIM_OCInitTypeDef TIM_OCInitStructure = { 0, };
|
|
NVIC_InitTypeDef NVIC_InitStructure = { 0, };
|
|
|
|
// Inputs
|
|
|
|
// RX1 TIM2_CH1 PA0 [also PPM]
|
|
// RX2 TIM2_CH2 PA1
|
|
// RX3 TIM2_CH3 PA2
|
|
// RX4 TIM2_CH4 PA3
|
|
// RX5 TIM3_CH1 PA6
|
|
// RX6 TIM3_CH2 PA7
|
|
// RX7 TIM3_CH3 PB0
|
|
// RX8 TIM3_CH4 PB1
|
|
|
|
// Outputs
|
|
// PWM1 TIM1_CH1 PA8
|
|
// PWM2 TIM1_CH4 PA11
|
|
// PWM3 TIM4_CH1 PB6
|
|
// PWM4 TIM4_CH2 PB7
|
|
// PWM5 TIM4_CH3 PB8
|
|
// PWM6 TIM4_CH4 PB9
|
|
|
|
// use PPM or PWM input
|
|
usePPMFlag = usePPM;
|
|
|
|
// preset channels to center
|
|
for (i = 0; i < 8; i++)
|
|
Inputs[i].capture = 1500;
|
|
|
|
// Timers run at 1mhz.
|
|
// TODO: clean this shit up. Make it all dynamic etc.
|
|
|
|
// Input pins
|
|
if (usePPM) {
|
|
// Configure TIM2_CH1 for PPM input
|
|
GPIO_InitStructure.GPIO_Pin = GPIO_Pin_0;
|
|
GPIO_InitStructure.GPIO_Mode = GPIO_Mode_IN_FLOATING;
|
|
GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
|
|
GPIO_Init(GPIOA, &GPIO_InitStructure);
|
|
|
|
// Input timer on TIM2 only for PPM
|
|
NVIC_InitStructure.NVIC_IRQChannel = TIM2_IRQn;
|
|
NVIC_InitStructure.NVIC_IRQChannelPreemptionPriority = 0;
|
|
NVIC_InitStructure.NVIC_IRQChannelSubPriority = 1;
|
|
NVIC_InitStructure.NVIC_IRQChannelCmd = ENABLE;
|
|
NVIC_Init(&NVIC_InitStructure);
|
|
|
|
// TIM2 timebase
|
|
TIM_TimeBaseStructInit(&TIM_TimeBaseStructure);
|
|
TIM_TimeBaseStructure.TIM_Prescaler = (72 - 1);
|
|
TIM_TimeBaseStructure.TIM_Period = 0xffff;
|
|
TIM_TimeBaseStructure.TIM_CounterMode = TIM_CounterMode_Up;
|
|
TIM_TimeBaseInit(TIM2, &TIM_TimeBaseStructure);
|
|
|
|
// Input capture on TIM2_CH1 for PPM
|
|
TIM_ICInitStructure.TIM_ICPolarity = TIM_ICPolarity_Rising;
|
|
TIM_ICInitStructure.TIM_ICSelection = TIM_ICSelection_DirectTI;
|
|
TIM_ICInitStructure.TIM_ICPrescaler = TIM_ICPSC_DIV1;
|
|
TIM_ICInitStructure.TIM_ICFilter = 0x0;
|
|
TIM_ICInitStructure.TIM_Channel = TIM_Channel_1;
|
|
TIM_ICInit(TIM2, &TIM_ICInitStructure);
|
|
|
|
// TIM2_CH1 capture compare interrupt enable
|
|
TIM_ITConfig(TIM2, TIM_IT_CC1, ENABLE);
|
|
TIM_Cmd(TIM2, ENABLE);
|
|
|
|
} else {
|
|
// Configure TIM2, TIM3 all 4 channels
|
|
GPIO_InitStructure.GPIO_Pin = GPIO_Pin_0 | GPIO_Pin_1 | GPIO_Pin_2 | GPIO_Pin_3 | GPIO_Pin_6 | GPIO_Pin_7;
|
|
GPIO_InitStructure.GPIO_Mode = GPIO_Mode_IPD;
|
|
GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
|
|
GPIO_Init(GPIOA, &GPIO_InitStructure);
|
|
GPIO_InitStructure.GPIO_Pin = GPIO_Pin_0 | GPIO_Pin_1;
|
|
GPIO_Init(GPIOB, &GPIO_InitStructure);
|
|
|
|
// Input timers on TIM2 and TIM3 for PWM
|
|
NVIC_InitStructure.NVIC_IRQChannel = TIM2_IRQn;
|
|
NVIC_InitStructure.NVIC_IRQChannelPreemptionPriority = 0;
|
|
NVIC_InitStructure.NVIC_IRQChannelSubPriority = 1;
|
|
NVIC_InitStructure.NVIC_IRQChannelCmd = ENABLE;
|
|
NVIC_Init(&NVIC_InitStructure);
|
|
NVIC_InitStructure.NVIC_IRQChannel = TIM3_IRQn;
|
|
NVIC_Init(&NVIC_InitStructure);
|
|
|
|
// TIM2 and TIM3 timebase
|
|
TIM_TimeBaseStructInit(&TIM_TimeBaseStructure);
|
|
TIM_TimeBaseStructure.TIM_Prescaler = (72 - 1);
|
|
TIM_TimeBaseStructure.TIM_Period = 0xffff;
|
|
TIM_TimeBaseStructure.TIM_CounterMode = TIM_CounterMode_Up;
|
|
TIM_TimeBaseInit(TIM2, &TIM_TimeBaseStructure);
|
|
TIM_TimeBaseInit(TIM3, &TIM_TimeBaseStructure);
|
|
|
|
// PWM Input capture
|
|
TIM_ICInitStructure.TIM_ICPolarity = TIM_ICPolarity_Rising;
|
|
TIM_ICInitStructure.TIM_ICSelection = TIM_ICSelection_DirectTI;
|
|
TIM_ICInitStructure.TIM_ICPrescaler = TIM_ICPSC_DIV1;
|
|
TIM_ICInitStructure.TIM_ICFilter = 0x0;
|
|
|
|
for (i = 0; i < 8; i++) {
|
|
TIM_ICInitStructure.TIM_Channel = Channels[i].channel;
|
|
TIM_ICInit(Channels[i].tim, &TIM_ICInitStructure);
|
|
}
|
|
|
|
TIM_ITConfig(TIM2, TIM_IT_CC1 | TIM_IT_CC2 | TIM_IT_CC3 | TIM_IT_CC4, ENABLE);
|
|
TIM_ITConfig(TIM3, TIM_IT_CC1 | TIM_IT_CC2 | TIM_IT_CC3 | TIM_IT_CC4, ENABLE);
|
|
TIM_Cmd(TIM2, ENABLE);
|
|
TIM_Cmd(TIM3, ENABLE);
|
|
}
|
|
|
|
// Output pins
|
|
GPIO_InitStructure.GPIO_Pin = GPIO_Pin_8 | GPIO_Pin_11;
|
|
GPIO_InitStructure.GPIO_Mode = GPIO_Mode_AF_PP;
|
|
GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
|
|
GPIO_Init(GPIOA, &GPIO_InitStructure);
|
|
GPIO_InitStructure.GPIO_Pin = GPIO_Pin_6 | GPIO_Pin_7 | GPIO_Pin_8 | GPIO_Pin_9;
|
|
GPIO_Init(GPIOB, &GPIO_InitStructure);
|
|
|
|
// Output timers
|
|
if (useServos) {
|
|
// 50Hz/200Hz period on ch1, 2 for servo
|
|
if (useDigitalServos)
|
|
TIM_TimeBaseStructure.TIM_Period = PULSE_PERIOD_SERVO_DIGITAL - 1;
|
|
else
|
|
TIM_TimeBaseStructure.TIM_Period = PULSE_PERIOD_SERVO_ANALOG - 1;
|
|
TIM_TimeBaseInit(TIM1, &TIM_TimeBaseStructure);
|
|
TIM_TimeBaseStructure.TIM_Period = PULSE_PERIOD - 1;
|
|
TIM_TimeBaseInit(TIM4, &TIM_TimeBaseStructure);
|
|
} else {
|
|
TIM_TimeBaseStructure.TIM_Period = PULSE_PERIOD - 1;
|
|
TIM_TimeBaseInit(TIM1, &TIM_TimeBaseStructure);
|
|
TIM_TimeBaseInit(TIM4, &TIM_TimeBaseStructure);
|
|
}
|
|
|
|
TIM_OCInitStructure.TIM_OCMode = TIM_OCMode_PWM2;
|
|
TIM_OCInitStructure.TIM_OutputState = TIM_OutputState_Enable;
|
|
TIM_OCInitStructure.TIM_OutputNState = TIM_OutputNState_Disable;
|
|
TIM_OCInitStructure.TIM_Pulse = PULSE_1MS;
|
|
TIM_OCInitStructure.TIM_OCPolarity = TIM_OCPolarity_Low;
|
|
TIM_OCInitStructure.TIM_OCIdleState = TIM_OCIdleState_Set;
|
|
|
|
// PWM1,2
|
|
TIM_OC1Init(TIM1, &TIM_OCInitStructure);
|
|
TIM_OC4Init(TIM1, &TIM_OCInitStructure);
|
|
// PWM3,4,5,6
|
|
TIM_OC1Init(TIM4, &TIM_OCInitStructure);
|
|
TIM_OC2Init(TIM4, &TIM_OCInitStructure);
|
|
TIM_OC3Init(TIM4, &TIM_OCInitStructure);
|
|
TIM_OC4Init(TIM4, &TIM_OCInitStructure);
|
|
|
|
TIM_Cmd(TIM1, ENABLE);
|
|
TIM_Cmd(TIM4, ENABLE);
|
|
TIM_CtrlPWMOutputs(TIM1, ENABLE);
|
|
TIM_CtrlPWMOutputs(TIM4, ENABLE);
|
|
}
|
|
|
|
void pwmWrite(uint8_t channel, uint16_t value)
|
|
{
|
|
*OutputChannels[channel] = value;
|
|
}
|
|
|
|
uint16_t pwmRead(uint8_t channel)
|
|
{
|
|
return Inputs[channel].capture;
|
|
}
|