/* * This file 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 file 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, see . * * Code by Andrew Tridgell and Siddharth Bharat Purohit * Bi-directional dshot based on Betaflight, code by Andy Piper and Siddharth Bharat Purohit */ #include "RCOutput.h" #include #include #include #include "GPIO.h" #include "hwdef/common/stm32_util.h" #include "hwdef/common/watchdog.h" #include #include #ifndef HAL_NO_UARTDRIVER #include #endif #if HAL_USE_PWM == TRUE using namespace ChibiOS; extern const AP_HAL::HAL& hal; #if HAL_WITH_IO_MCU #include extern AP_IOMCU iomcu; #endif #define RCOU_SERIAL_TIMING_DEBUG 0 #define TELEM_IC_SAMPLE 16 struct RCOutput::pwm_group RCOutput::pwm_group_list[] = { HAL_PWM_GROUPS }; struct RCOutput::irq_state RCOutput::irq; const uint8_t RCOutput::NUM_GROUPS = ARRAY_SIZE(RCOutput::pwm_group_list); // event mask for triggering a PWM send static const eventmask_t EVT_PWM_SEND = EVENT_MASK(11); static const eventmask_t EVT_PWM_START = EVENT_MASK(12); static const eventmask_t EVT_PWM_SYNTHETIC_SEND = EVENT_MASK(13); static const eventmask_t EVT_PWM_SEND_NEXT = EVENT_MASK(14); // #pragma GCC optimize("Og") /* initialise RC output driver */ void RCOutput::init() { uint8_t pwm_count = AP_BoardConfig::get_pwm_count(); for (auto &group : pwm_group_list) { const uint8_t i = &group - pwm_group_list; //Start Pwm groups group.current_mode = MODE_PWM_NORMAL; group.dshot_event_mask = EVENT_MASK(i); for (uint8_t j = 0; j < 4; j++ ) { uint8_t chan = group.chan[j]; if (chan >= pwm_count) { group.chan[j] = CHAN_DISABLED; } if (group.chan[j] != CHAN_DISABLED) { num_fmu_channels = MAX(num_fmu_channels, group.chan[j]+1); group.ch_mask |= (1U<pinMode(54, 1); hal.gpio->pinMode(55, 1); hal.gpio->pinMode(56, 1); hal.gpio->pinMode(57, 1); #endif hal.scheduler->register_timer_process(FUNCTOR_BIND(this, &RCOutput::safety_update, void)); _initialised = true; } /* thread for handling RCOutput send */ void RCOutput::rcout_thread() { uint32_t last_thread_run_us = 0; // last time we did a 1kHz run of rcout uint32_t last_cycle_run_us = 0; rcout_thread_ctx = chThdGetSelfX(); // don't start outputting until fully configured while (!hal.scheduler->is_system_initialized()) { hal.scheduler->delay_microseconds(1000); } // dshot is quite sensitive to timing, it's important to output pulses as // regularly as possible at the correct bitrate while (true) { chEvtWaitOne(EVT_PWM_SEND | EVT_PWM_SYNTHETIC_SEND); // start the clock last_thread_run_us = AP_HAL::micros(); // this is when the cycle is supposed to start if (_dshot_cycle == 0) { last_cycle_run_us = AP_HAL::micros(); // register a timer for the next tick if push() will not be providing it if (_dshot_rate != 1) { chVTSet(&_dshot_rate_timer, chTimeUS2I(_dshot_period_us), dshot_update_tick, this); } } // if DMA sharing is in effect there can be quite a delay between the request to begin the cycle and // actually sending out data - thus we need to work out how much time we have left to collect the locks uint32_t time_out_us = (_dshot_cycle + 1) * _dshot_period_us + last_cycle_run_us; if (!_dshot_rate) { time_out_us = last_thread_run_us + _dshot_period_us; } // main thread requested a new dshot send or we timed out - if we are not running // as a multiple of loop rate then ignore EVT_PWM_SEND events to preserve periodicity if (!serial_group) { dshot_send_groups(time_out_us); // now unlock everything dshot_collect_dma_locks(time_out_us); if (_dshot_rate > 0) { _dshot_cycle = (_dshot_cycle + 1) % _dshot_rate; } } // process any pending RC output requests timer_tick(time_out_us); } } void RCOutput::dshot_update_tick(void* p) { chSysLockFromISR(); RCOutput* rcout = (RCOutput*)p; if (rcout->_dshot_cycle + 1 < rcout->_dshot_rate) { chVTSetI(&rcout->_dshot_rate_timer, chTimeUS2I(rcout->_dshot_period_us), dshot_update_tick, p); } chEvtSignalI(rcout->rcout_thread_ctx, EVT_PWM_SYNTHETIC_SEND); chSysUnlockFromISR(); } #ifndef HAL_NO_SHARED_DMA // release locks on the groups that are pending in reverse order void RCOutput::dshot_collect_dma_locks(uint32_t time_out_us) { if (NUM_GROUPS == 0) { return; } for (int8_t i = NUM_GROUPS - 1; i >= 0; i--) { pwm_group &group = pwm_group_list[i]; if (group.dma_handle != nullptr && group.dma_handle->is_locked()) { // calculate how long we have left uint32_t now = AP_HAL::micros(); // if we have time left wait for the event eventmask_t mask = 0; const uint32_t pulse_elapsed_us = now - group.last_dmar_send_us; if (now < time_out_us) { mask = chEvtWaitOneTimeout(group.dshot_event_mask, chTimeUS2I(MAX(time_out_us - now, group.dshot_pulse_send_time_us - pulse_elapsed_us))); } else if (pulse_elapsed_us < group.dshot_pulse_send_time_us) { // better to let the burst write in progress complete rather than cancelling mid way through mask = chEvtWaitOneTimeout(group.dshot_event_mask, chTimeUS2I(group.dshot_pulse_send_time_us - pulse_elapsed_us)); } // no time left cancel and restart if (!mask) { dma_cancel(group); } group.dshot_waiter = nullptr; #ifdef HAL_WITH_BIDIR_DSHOT // if using input capture DMA then clean up if (group.bdshot.enabled) { // the channel index only moves on with success const uint8_t chan = mask ? group.bdshot.prev_telem_chan : group.bdshot.curr_telem_chan; // only unlock if not shared if (group.bdshot.ic_dma_handle[chan] != nullptr && group.bdshot.ic_dma_handle[chan] != group.dma_handle) { group.bdshot.ic_dma_handle[chan]->unlock(); } } #endif group.dma_handle->unlock(); } } } #endif // HAL_NO_SHARED_DMA /* setup the output frequency for a group and start pwm output */ void RCOutput::set_freq_group(pwm_group &group) { if (mode_requires_dma(group.current_mode)) { // speed setup in DMA handler return; } uint16_t freq_set = group.rc_frequency; uint32_t old_clock = group.pwm_cfg.frequency; uint32_t old_period = group.pwm_cfg.period; if (freq_set > 400 || group.current_mode == MODE_PWM_ONESHOT125) { // use a 8MHz clock for higher frequencies or for // oneshot125. Using 8MHz for oneshot125 results in the full // 1000 steps for smooth output group.pwm_cfg.frequency = 8000000; } else if (freq_set <= 400) { // use a 1MHz clock group.pwm_cfg.frequency = 1000000; } // check if the frequency is possible, and keep halving // down to 1MHz until it is OK with the hardware timer we // are using. If we don't do this we'll hit an assert in // the ChibiOS PWM driver on some timers PWMDriver *pwmp = group.pwm_drv; uint32_t psc = (pwmp->clock / pwmp->config->frequency) - 1; while ((psc > 0xFFFF || ((psc + 1) * pwmp->config->frequency) != pwmp->clock) && group.pwm_cfg.frequency > 1000000) { group.pwm_cfg.frequency /= 2; psc = (pwmp->clock / pwmp->config->frequency) - 1; } if (group.current_mode == MODE_PWM_ONESHOT || group.current_mode == MODE_PWM_ONESHOT125) { // force a period of 0, meaning no pulses till we trigger group.pwm_cfg.period = 0; } else { group.pwm_cfg.period = group.pwm_cfg.frequency/freq_set; } bool force_reconfig = false; for (uint8_t j=0; j<4; j++) { if (group.pwm_cfg.channels[j].mode == PWM_OUTPUT_ACTIVE_LOW) { group.pwm_cfg.channels[j].mode = PWM_OUTPUT_ACTIVE_HIGH; force_reconfig = true; } if (group.pwm_cfg.channels[j].mode == PWM_COMPLEMENTARY_OUTPUT_ACTIVE_LOW) { group.pwm_cfg.channels[j].mode = PWM_COMPLEMENTARY_OUTPUT_ACTIVE_HIGH; force_reconfig = true; } } if (old_clock != group.pwm_cfg.frequency || old_period != group.pwm_cfg.period || !group.pwm_started || force_reconfig) { // we need to stop and start to setup the new clock if (group.pwm_started) { pwmStop(group.pwm_drv); } pwmStart(group.pwm_drv, &group.pwm_cfg); group.pwm_started = true; } pwmChangePeriod(group.pwm_drv, group.pwm_cfg.period); } /* set output frequency in HZ for a set of channels given by a mask */ void RCOutput::set_freq(uint32_t chmask, uint16_t freq_hz) { #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { // change frequency on IOMCU uint16_t io_chmask = chmask & 0xFF; if (io_chmask) { // disallow changing frequency of this group if it is greater than the default for (uint8_t i=0; i 50) { io_fast_channel_mask |= mask; } else { io_fast_channel_mask &= ~mask; } } } iomcu.set_freq(io_fast_channel_mask, freq_hz); } } #endif // convert to a local (FMU) channel mask chmask >>= chan_offset; if (chmask == 0) { return; } /* we enable the new frequency on all groups that have one of the requested channels. This means we may enable high speed on some channels that aren't requested, but that is needed in order to fly a vehicle such as a hex multicopter properly */ for (auto &group : pwm_group_list) { // greater than 400 doesn't give enough room at higher periods for // the down pulse. This still allows for high rate with oneshot and dshot. uint16_t group_freq = freq_hz; if (group_freq > 400 && group.current_mode != MODE_PWM_BRUSHED) { group_freq = 400; } if ((group.ch_mask & chmask) != 0) { group.rc_frequency = group_freq; set_freq_group(group); // disallow changing frequency of this group if it is greater than the default if (group_freq > 50) { fast_channel_mask |= group.ch_mask; } } } } /* set default output rate */ void RCOutput::set_default_rate(uint16_t freq_hz) { #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { iomcu.set_default_rate(freq_hz); } #endif for (auto &group : pwm_group_list) { if ((group.ch_mask & fast_channel_mask) || group.ch_mask == 0) { // don't change fast channels continue; } group.pwm_cfg.period = group.pwm_cfg.frequency/freq_hz; if (group.pwm_started) { pwmChangePeriod(group.pwm_drv, group.pwm_cfg.period); } } } /* Set the dshot rate as a multiple of the loop rate. This is called late after init_ardupilot() so groups will have been setup */ void RCOutput::set_dshot_rate(uint8_t dshot_rate, uint16_t loop_rate_hz) { // for low loop rates simply output at 1Khz on a timer if (loop_rate_hz <= 100 || dshot_rate == 0) { _dshot_period_us = 1000UL; _dshot_rate = 0; return; } // if there are non-dshot channels then do likewise for (auto &group : pwm_group_list) { if (group.current_mode == MODE_PWM_ONESHOT || group.current_mode == MODE_PWM_ONESHOT125 || group.current_mode == MODE_PWM_BRUSHED) { _dshot_period_us = 1000UL; _dshot_rate = 0; return; } } uint16_t drate = dshot_rate * loop_rate_hz; _dshot_rate = dshot_rate; // BLHeli32 uses a 16 bit counter for input calibration which at 48Mhz will wrap // at 732Hz so never allow rates below 800hz while (drate < 800) { _dshot_rate++; drate = _dshot_rate * loop_rate_hz; } // prevent stupidly high rates, ideally should also prevent high rates // with slower dshot variants if (drate > 4000) { _dshot_rate = 4000 / loop_rate_hz; drate = _dshot_rate * loop_rate_hz; } _dshot_period_us = 1000000UL / drate; } /* find pwm_group and index in group given a channel number */ RCOutput::pwm_group *RCOutput::find_chan(uint8_t chan, uint8_t &group_idx) { if (chan >= max_channels) { return nullptr; } if (chan < chan_offset) { return nullptr; } chan -= chan_offset; for (auto &group : pwm_group_list) { for (uint8_t j = 0; j < 4; j++) { if (group.chan[j] == chan) { group_idx = j; return &group; } } } return nullptr; } uint16_t RCOutput::get_freq(uint8_t chan) { #if HAL_WITH_IO_MCU if (chan < chan_offset) { return iomcu.get_freq(chan); } #endif uint8_t i; pwm_group *grp = find_chan(chan, i); if (grp) { return grp->pwm_drv->config->frequency / grp->pwm_drv->period; } // assume 50Hz default return 50; } void RCOutput::enable_ch(uint8_t chan) { uint8_t i; pwm_group *grp = find_chan(chan, i); if (grp) { en_mask |= 1U << (chan - chan_offset); grp->en_mask |= 1U << (chan - chan_offset); } } void RCOutput::disable_ch(uint8_t chan) { uint8_t i; pwm_group *grp = find_chan(chan, i); if (grp) { pwmDisableChannel(grp->pwm_drv, i); en_mask &= ~(1U<<(chan - chan_offset)); grp->en_mask &= ~(1U << (chan - chan_offset)); } } void RCOutput::write(uint8_t chan, uint16_t period_us) { if (chan >= max_channels) { return; } last_sent[chan] = period_us; #if HAL_WITH_IO_MCU // handle IO MCU channels if (AP_BoardConfig::io_enabled()) { uint16_t io_period_us = period_us; if ((iomcu_mode == MODE_PWM_ONESHOT125) && ((1U<safety_switch_state() == AP_HAL::Util::SAFETY_DISARMED; for (auto &group : pwm_group_list) { if (serial_group) { continue; } if (!group.pwm_started) { continue; } for (uint8_t j = 0; j < 4; j++) { uint8_t chan = group.chan[j]; if (!group.is_chan_enabled(j)) { continue; } if (outmask & (1UL<= _esc_pwm_max) { period_us = PWM_FRACTION_TO_WIDTH(group.pwm_drv, 1, 1); } else { period_us = PWM_FRACTION_TO_WIDTH(group.pwm_drv,\ (_esc_pwm_max - _esc_pwm_min), (period_us - _esc_pwm_min)); } pwmEnableChannel(group.pwm_drv, j, period_us); } else if (group.current_mode == MODE_PWM_ONESHOT125) { // this gives us a width in 125 ns increments, giving 1000 steps over the 125 to 250 range uint32_t width = ((group.pwm_cfg.frequency/1000000U) * period_us) / 8U; pwmEnableChannel(group.pwm_drv, j, width); // scale the period down so we don't delay for longer than we need to period_us /= 8; } else if (group.current_mode < MODE_PWM_DSHOT150) { uint32_t width = (group.pwm_cfg.frequency/1000000U) * period_us; pwmEnableChannel(group.pwm_drv, j, width); } #ifndef DISABLE_DSHOT else if (is_dshot_protocol(group.current_mode) || group.current_mode == MODE_NEOPIXEL || group.current_mode == MODE_PROFILED) { // set period_us to time for pulse output, to enable very fast rates period_us = group.dshot_pulse_time_us; } #endif //#ifndef DISABLE_DSHOT if (period_us > widest_pulse) { widest_pulse = period_us; } if (group.current_mode == MODE_PWM_ONESHOT || group.current_mode == MODE_PWM_ONESHOT125 || group.current_mode == MODE_NEOPIXEL || group.current_mode == MODE_PROFILED || is_dshot_protocol(group.current_mode)) { const uint8_t i = &group - pwm_group_list; need_trigger |= (1U< 2300) { widest_pulse = 2300; } trigger_widest_pulse = widest_pulse; trigger_groupmask = need_trigger; if (trigger_groupmask) { trigger_groups(); } } uint16_t RCOutput::read(uint8_t chan) { if (chan >= max_channels) { return 0; } #if HAL_WITH_IO_MCU if (chan < chan_offset) { return iomcu.read_channel(chan); } #endif chan -= chan_offset; return period[chan]; } void RCOutput::read(uint16_t* period_us, uint8_t len) { if (len > max_channels) { len = max_channels; } #if HAL_WITH_IO_MCU for (uint8_t i=0; i= max_channels) { return 0; } return last_sent[chan]; } void RCOutput::read_last_sent(uint16_t* period_us, uint8_t len) { if (len > max_channels) { len = max_channels; } for (uint8_t i=0; ilock(); if (!group.dma_buffer || buffer_length != group.dma_buffer_len) { if (group.dma_buffer) { hal.util->free_type(group.dma_buffer, group.dma_buffer_len, AP_HAL::Util::MEM_DMA_SAFE); group.dma_buffer_len = 0; } group.dma_buffer = (uint32_t *)hal.util->malloc_type(buffer_length, AP_HAL::Util::MEM_DMA_SAFE); if (!group.dma_buffer) { group.dma_handle->unlock(); print_group_setup_error(group, "failed to allocate DMA buffer"); return false; } group.dma_buffer_len = buffer_length; } // reset the pulse time inside the lock group.dshot_pulse_time_us = group.dshot_pulse_send_time_us = pulse_time_us; #ifdef HAL_WITH_BIDIR_DSHOT // configure input capture DMA if required if (is_bidir_dshot_enabled()) { if (!bdshot_setup_group_ic_DMA(group)) { group.dma_handle->unlock(); return false; } } #endif // configure timer driver for DMAR at requested rate if (group.pwm_started) { pwmStop(group.pwm_drv); group.pwm_started = false; } // original prescaler calculation from betaflight. bi-dir dshot is incredibly sensitive to the bitrate const uint32_t target_frequency = bitrate * bit_width; uint32_t prescaler = uint32_t(lrintf((float) group.pwm_drv->clock / (bitrate * bit_width) + 0.01f) - 1); uint32_t freq = group.pwm_drv->clock / prescaler; if (freq > target_frequency && !choose_high) { prescaler++; } else if (freq < target_frequency && choose_high) { prescaler--; } if (prescaler > 0x8000) { group.dma_handle->unlock(); print_group_setup_error(group, "failed to match clock speed"); return false; } freq = group.pwm_drv->clock / prescaler; group.pwm_cfg.frequency = freq; group.pwm_cfg.period = bit_width; group.pwm_cfg.dier = TIM_DIER_UDE; group.pwm_cfg.cr2 = 0; group.bit_width_mul = (freq + (target_frequency/2)) / target_frequency; //hal.console->printf("CLOCK=%u BW=%u FREQ=%u BR=%u MUL=%u PRE=%u\n", unsigned(group.pwm_drv->clock), unsigned(bit_width), unsigned(group.pwm_cfg.frequency), // unsigned(bitrate), unsigned(group.bit_width_mul), unsigned(prescaler)); for (uint8_t j=0; j<4; j++) { pwmmode_t mode = group.pwm_cfg.channels[j].mode; if (mode != PWM_OUTPUT_DISABLED) { if(mode == PWM_COMPLEMENTARY_OUTPUT_ACTIVE_LOW || mode == PWM_COMPLEMENTARY_OUTPUT_ACTIVE_HIGH) { group.pwm_cfg.channels[j].mode = active_high ? PWM_COMPLEMENTARY_OUTPUT_ACTIVE_HIGH : PWM_COMPLEMENTARY_OUTPUT_ACTIVE_LOW; } else { group.pwm_cfg.channels[j].mode = active_high ? PWM_OUTPUT_ACTIVE_HIGH : PWM_OUTPUT_ACTIVE_LOW; } } } pwmStart(group.pwm_drv, &group.pwm_cfg); group.pwm_started = true; for (uint8_t j=0; j<4; j++) { if (group.is_chan_enabled(j)) { pwmEnableChannel(group.pwm_drv, j, 0); } } group.dma_handle->unlock(); return true; #else return false; #endif //#ifndef DISABLE_DSHOT } /* setup output mode for a group, using group.current_mode. Used to restore output after serial operations */ void RCOutput::set_group_mode(pwm_group &group) { if (group.pwm_started) { pwmStop(group.pwm_drv); group.pwm_started = false; } memset(group.bdshot.erpm, 0, 4*sizeof(uint16_t)); switch (group.current_mode) { case MODE_PWM_BRUSHED: // force zero output initially for (uint8_t i=0; i<4; i++) { if (group.chan[i] == CHAN_DISABLED) { continue; } uint8_t chan = chan_offset + group.chan[i]; write(chan, 0); } break; case MODE_NEOPIXEL: case MODE_PROFILED: { uint8_t bits_per_pixel = 24; bool active_high = true; if (group.current_mode == MODE_PROFILED) { bits_per_pixel = 25; active_high = false; } const uint32_t rate = protocol_bitrate(group.current_mode); const uint32_t bit_period = 20; // configure timer driver for DMAR at requested rate const uint8_t pad_end_bits = 8; const uint8_t pad_start_bits = 1; const uint8_t channels_per_group = 4; const uint16_t bit_length = bits_per_pixel * group.serial_nleds + pad_start_bits + pad_end_bits; const uint16_t buffer_length = bit_length * sizeof(uint32_t) * channels_per_group; // calculate min time between pulses taking into account the DMAR parallelism const uint32_t pulse_time_us = 1000000UL * bit_length / rate; if (!setup_group_DMA(group, rate, bit_period, active_high, buffer_length, true, pulse_time_us)) { group.current_mode = MODE_PWM_NONE; break; } break; } case MODE_PWM_DSHOT150 ... MODE_PWM_DSHOT1200: { const uint32_t rate = protocol_bitrate(group.current_mode); const uint32_t bit_period = 20; bool active_high = is_bidir_dshot_enabled() ? false : true; // calculate min time between pulses const uint32_t pulse_send_time_us = 1000000UL * dshot_bit_length / rate; // configure timer driver for DMAR at requested rate if (!setup_group_DMA(group, rate, bit_period, active_high, // choosing the low frequency for DSHOT150 is based on experimentation with BLHeli32 and bi-directional dshot MAX(DSHOT_BUFFER_LENGTH, GCR_TELEMETRY_BUFFER_LEN), group.current_mode != MODE_PWM_DSHOT150, pulse_send_time_us)) { group.current_mode = MODE_PWM_NORMAL; break; } if (is_bidir_dshot_enabled()) { group.dshot_pulse_send_time_us = pulse_send_time_us; // to all intents and purposes the pulse time of send and receive are the same group.dshot_pulse_time_us = pulse_send_time_us + pulse_send_time_us + 30; } break; } case MODE_PWM_ONESHOT: case MODE_PWM_ONESHOT125: // for oneshot we set a period of 0, which results in no pulses till we trigger group.pwm_cfg.period = 0; group.rc_frequency = 1; if (group.pwm_started) { pwmChangePeriod(group.pwm_drv, group.pwm_cfg.period); } break; case MODE_PWM_NORMAL: case MODE_PWM_NONE: // nothing needed break; } set_freq_group(group); if (group.current_mode != MODE_PWM_NONE && !group.pwm_started) { pwmStart(group.pwm_drv, &group.pwm_cfg); group.pwm_started = true; for (uint8_t j=0; j<4; j++) { if (group.is_chan_enabled(j)) { pwmEnableChannel(group.pwm_drv, j, 0); } } } } /* setup output mode */ void RCOutput::set_output_mode(uint16_t mask, const enum output_mode mode) { for (auto &group : pwm_group_list) { enum output_mode thismode = mode; if (((group.ch_mask << chan_offset) & mask) == 0) { // this group is not affected continue; } if (mode_requires_dma(thismode) && !group.have_up_dma) { print_group_setup_error(group, "failed, no DMA"); thismode = MODE_PWM_NORMAL; } if (mode > MODE_PWM_NORMAL) { fast_channel_mask |= group.ch_mask; } // setup of the group mode also sets up DMA which might have changed, so always // redo it if using DMA if (group.current_mode != thismode) { group.current_mode = thismode; set_group_mode(group); } } #if HAL_WITH_IO_MCU if ((mode == MODE_PWM_ONESHOT || mode == MODE_PWM_ONESHOT125) && (mask & ((1U< 0) && (iomcu_mode != MODE_PWM_NONE); } #endif // fill in ch_mode array for FMU channels for (auto &group : pwm_group_list) { if (group.current_mode != MODE_PWM_NONE) { for (uint8_t j = 0; j < ARRAY_SIZE(group.chan); j++) { if (group.chan[j] != CHAN_DISABLED) { const uint8_t chan_num = group.chan[j] + chan_offset; if (chan_num < ARRAY_SIZE(ch_mode)) { ch_mode[chan_num] = group.current_mode; have_nonzero_modes = true; } } } } } // handle simple case if (!have_nonzero_modes) { hal.util->snprintf(banner_msg, banner_msg_len, "RCOut: None"); return true; } // write banner to banner_msg hal.util->snprintf(banner_msg, banner_msg_len, "RCOut:"); uint8_t curr_mode_lowest_ch = 0; for (uint8_t k = 1; k < ARRAY_SIZE(ch_mode); k++) { if (ch_mode[k-1] != ch_mode[k]) { if (ch_mode[k-1] != MODE_PWM_NONE) { append_to_banner(banner_msg, banner_msg_len, ch_mode[k-1], curr_mode_lowest_ch + 1, k); } curr_mode_lowest_ch = k; } } // add final few channel's mode to banner (won't have been done by above loop) const uint8_t final_index = ARRAY_SIZE(ch_mode)-1; if (ch_mode[final_index] != MODE_PWM_NONE) { append_to_banner(banner_msg, banner_msg_len, ch_mode[final_index], curr_mode_lowest_ch + 1, final_index + 1); } return true; } /* start corking output */ void RCOutput::cork(void) { corked = true; #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { iomcu.cork(); } #endif } /* stop corking output */ void RCOutput::push(void) { corked = false; push_local(); #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { iomcu.push(); } #endif } /* enable sbus output */ bool RCOutput::enable_px4io_sbus_out(uint16_t rate_hz) { #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { return iomcu.enable_sbus_out(rate_hz); } #endif return false; } /* trigger output groups for oneshot or dshot modes */ void RCOutput::trigger_groups(void) { if (!chMtxTryLock(&trigger_mutex)) { return; } uint64_t now = AP_HAL::micros64(); if (now < min_pulse_trigger_us) { // guarantee minimum pulse separation hal.scheduler->delay_microseconds(min_pulse_trigger_us - now); } osalSysLock(); for (auto &group : pwm_group_list) { if (irq.waiter) { // doing serial output, don't send pulses continue; } if (group.current_mode == MODE_PWM_ONESHOT || group.current_mode == MODE_PWM_ONESHOT125) { const uint8_t i = &group - pwm_group_list; if (trigger_groupmask & (1U<tim->EGR = STM32_TIM_EGR_UG; } } } osalSysUnlock(); #ifndef HAL_NO_RCOUT_THREAD // trigger a PWM send if (!serial_group && hal.scheduler->in_main_thread()) { chEvtSignal(rcout_thread_ctx, EVT_PWM_SEND); } #endif /* calculate time that we are allowed to trigger next pulse to guarantee at least a 50us gap between pulses */ min_pulse_trigger_us = AP_HAL::micros64() + trigger_widest_pulse + 50; chMtxUnlock(&trigger_mutex); } /* periodic timer. This is used for oneshot and dshot modes, plus for safety switch update. Runs every 1000us. */ void RCOutput::timer_tick(uint32_t time_out_us) { if (serial_group) { return; } // if we have enough time left send out LED data if (serial_led_pending && (time_out_us > (AP_HAL::micros() + (_dshot_period_us >> 1)))) { serial_led_pending = false; for (auto &group : pwm_group_list) { serial_led_pending |= !serial_led_send(group); } // release locks on the groups that are pending in reverse order dshot_collect_dma_locks(time_out_us); } if (min_pulse_trigger_us == 0) { return; } uint32_t now = AP_HAL::micros(); if (now > min_pulse_trigger_us && now - min_pulse_trigger_us > 4000) { // trigger at a minimum of 250Hz trigger_groups(); } } // send dshot for all groups that support it void RCOutput::dshot_send_groups(uint32_t time_out_us) { #ifndef DISABLE_DSHOT if (serial_group) { return; } bool command_sent = false; // queue up a command if there is one if (!hal.util->get_soft_armed() && _dshot_current_command.cycle == 0 && _dshot_command_queue.pop(_dshot_current_command)) { // got a new command } for (auto &group : pwm_group_list) { // send a dshot command if (!hal.util->get_soft_armed() && is_dshot_protocol(group.current_mode) && dshot_command_is_active(group)) { command_sent = dshot_send_command(group, _dshot_current_command.command, _dshot_current_command.chan); // actually do a dshot send } else if (group.can_send_dshot_pulse()) { dshot_send(group, time_out_us); } } if (command_sent) { _dshot_current_command.cycle--; } #endif //#ifndef DISABLE_DSHOT } void RCOutput::dshot_send_next_group(void* p) { chSysLockFromISR(); RCOutput* rcout = (RCOutput*)p; chEvtSignalI(rcout->rcout_thread_ctx, EVT_PWM_SEND_NEXT); chSysUnlockFromISR(); } /* allocate DMA channel */ void RCOutput::dma_allocate(Shared_DMA *ctx) { for (auto &group : pwm_group_list) { if (group.dma_handle == ctx && group.dma == nullptr) { chSysLock(); group.dma = dmaStreamAllocI(group.dma_up_stream_id, 10, dma_up_irq_callback, &group); chSysUnlock(); #if STM32_DMA_SUPPORTS_DMAMUX if (group.dma) { dmaSetRequestSource(group.dma, group.dma_up_channel); } #endif } } } /* deallocate DMA channel */ void RCOutput::dma_deallocate(Shared_DMA *ctx) { for (auto &group : pwm_group_list) { if (group.dma_handle == ctx && group.dma != nullptr) { chSysLock(); dmaStreamFreeI(group.dma); group.dma = nullptr; chSysUnlock(); } } } /* create a DSHOT 16 bit packet. Based on prepareDshotPacket from betaflight */ uint16_t RCOutput::create_dshot_packet(const uint16_t value, bool telem_request, bool bidir_telem) { uint16_t packet = (value << 1); if (telem_request) { packet |= 1; } // compute checksum uint16_t csum = 0; uint16_t csum_data = packet; for (uint8_t i = 0; i < 3; i++) { csum ^= csum_data; csum_data >>= 4; } // trigger bi-dir dshot telemetry if (bidir_telem) { csum = ~csum; } // append checksum csum &= 0xf; packet = (packet << 4) | csum; return packet; } /* fill in a DMA buffer for dshot */ void RCOutput::fill_DMA_buffer_dshot(uint32_t *buffer, uint8_t stride, uint16_t packet, uint16_t clockmul) { const uint32_t DSHOT_MOTOR_BIT_0 = 7 * clockmul; const uint32_t DSHOT_MOTOR_BIT_1 = 14 * clockmul; uint16_t i = 0; for (; i < dshot_pre; i++) { buffer[i * stride] = 0; } for (; i < 16 + dshot_pre; i++) { buffer[i * stride] = (packet & 0x8000) ? DSHOT_MOTOR_BIT_1 : DSHOT_MOTOR_BIT_0; packet <<= 1; } for (; iis_locked(), "DMA handle is already locked"); group.dma_handle->lock(); // if we are sharing UP channels then it might have taken a long time to get here, // if there's not enough time to actually send a pulse then cancel if (AP_HAL::micros() + group.dshot_pulse_time_us > time_out_us) { group.dma_handle->unlock(); return; } // only the timer thread releases the locks group.dshot_waiter = rcout_thread_ctx; #ifdef HAL_WITH_BIDIR_DSHOT // assume that we won't be able to get the input capture lock group.bdshot.enabled = false; // now grab the input capture lock if we are able, we can only enable bi-dir on a group basis if (((_bdshot.mask & group.ch_mask) == group.ch_mask) && group.has_ic()) { if (group.has_shared_ic_up_dma()) { // no locking required group.bdshot.enabled = true; } else { osalDbgAssert(!group.bdshot.ic_dma_handle[group.bdshot.curr_telem_chan]->is_locked(), "DMA handle is already locked"); group.bdshot.ic_dma_handle[group.bdshot.curr_telem_chan]->lock(); group.bdshot.enabled = true; } } // if the last transaction returned telemetry, decode it if (group.dshot_state == DshotState::RECV_COMPLETE) { uint8_t chan = group.chan[group.bdshot.prev_telem_chan]; uint32_t now = AP_HAL::millis(); if (bdshot_decode_dshot_telemetry(group, group.bdshot.prev_telem_chan)) { _bdshot.erpm_clean_frames[chan]++; _active_escs_mask |= (1< 5000) { _bdshot.erpm_clean_frames[chan] = 0; _bdshot.erpm_errors[chan] = 0; _bdshot.erpm_last_stats_ms[chan] = now; } } if (group.bdshot.enabled) { if (group.pwm_started) { pwmStop(group.pwm_drv); } pwmStart(group.pwm_drv, &group.pwm_cfg); group.pwm_started = true; // we can be more precise for capture timer group.bdshot.telempsc = (uint16_t)(lrintf(((float)group.pwm_drv->clock / bdshot_get_output_rate_hz(group.current_mode) + 0.01f)/TELEM_IC_SAMPLE) - 1); } #endif bool safety_on = hal.util->safety_switch_state() == AP_HAL::Util::SAFETY_DISARMED; bool armed = hal.util->get_soft_armed(); memset((uint8_t *)group.dma_buffer, 0, DSHOT_BUFFER_LENGTH); for (uint8_t i=0; i<4; i++) { uint8_t chan = group.chan[i]; if (group.is_chan_enabled(i)) { #ifdef HAL_WITH_BIDIR_DSHOT // retrieve the last erpm values const uint16_t erpm = group.bdshot.erpm[i]; #if HAL_WITH_ESC_TELEM // update the ESC telemetry data if (erpm < 0xFFFF && group.bdshot.enabled) { update_rpm(chan, erpm * 200 / _bdshot.motor_poles, get_erpm_error_rate(chan)); } #endif _bdshot.erpm[chan] = erpm; #endif uint16_t pwm = period[chan]; if (safety_on && !(safety_mask & (1U<<(chan+chan_offset)))) { // safety is on, overwride pwm pwm = safe_pwm[chan+chan_offset]; } const uint16_t chan_mask = (1U<>chan_offset)) { // this is a DShot-3D output, map so that 1500 PWM is zero throttle reversed if (value < 1000) { value = 2000 - value; } else if (value > 1000) { value = value - 1000; } else { // mid-throttle is off value = 0; } } if (value != 0) { // dshot values are from 48 to 2047. Zero means off. value += 47; } if (!armed) { // when disarmed we always send a zero value value = 0; } // according to sskaug requesting telemetry while trying to arm may interfere with the good frame calc bool request_telemetry = telem_request_mask & chan_mask; uint16_t packet = create_dshot_packet(value, request_telemetry, group.bdshot.enabled); if (request_telemetry) { telem_request_mask &= ~chan_mask; } fill_DMA_buffer_dshot(group.dma_buffer + i, 4, packet, group.bit_width_mul); } } chEvtGetAndClearEvents(group.dshot_event_mask); // start sending the pulses out send_pulses_DMAR(group, DSHOT_BUFFER_LENGTH); #endif //#ifndef DISABLE_DSHOT } /* send a set of Serial LED packets for a channel group return true if send was successful */ bool RCOutput::serial_led_send(pwm_group &group) { if (!group.serial_led_pending || (group.current_mode != MODE_NEOPIXEL && group.current_mode != MODE_PROFILED)) { return true; } #ifndef DISABLE_DSHOT if (irq.waiter || !group.dma_handle->lock_nonblock()) { // doing serial output, don't send Serial LED pulses return false; } { WITH_SEMAPHORE(group.serial_led_mutex); group.serial_led_pending = false; group.prepared_send = false; // fill the DMA buffer while we have the lock fill_DMA_buffer_serial_led(group); } group.dshot_waiter = rcout_thread_ctx; chEvtGetAndClearEvents(group.dshot_event_mask); // start sending the pulses out send_pulses_DMAR(group, group.dma_buffer_len); #endif //#ifndef DISABLE_DSHOT return true; } /* send a series of pulses for a group using DMAR. Pulses must have been encoded into the group dma_buffer with interleaving for the 4 channels in the group */ void RCOutput::send_pulses_DMAR(pwm_group &group, uint32_t buffer_length) { #ifndef DISABLE_DSHOT osalDbgAssert(group.dma && group.dma_buffer, "DMA structures are corrupt"); /* The DMA approach we are using is based on the DMAR method from betaflight. We use the TIMn_UP DMA channel for the timer, and setup an interleaved set of pulse durations, with a stride of 4 (for the 4 channels). We use the DMAR register to point the DMA engine at the 4 CCR registers of the timer, so it fills in the pulse widths for each timer in turn. This means we only use a single DMA channel for groups of 4 timer channels. See the "DMA address for full transfer TIMx_DMAR" section of the datasheet. Many thanks to the betaflight developers for coming up with this great method. */ TOGGLE_PIN_DEBUG(54); dmaStreamSetPeripheral(group.dma, &(group.pwm_drv->tim->DMAR)); stm32_cacheBufferFlush(group.dma_buffer, buffer_length); dmaStreamSetMemory0(group.dma, group.dma_buffer); dmaStreamSetTransactionSize(group.dma, buffer_length/sizeof(uint32_t)); #ifdef STM32_DMA_FCR_FTH_FULL dmaStreamSetFIFO(group.dma, STM32_DMA_FCR_DMDIS | STM32_DMA_FCR_FTH_FULL); #endif dmaStreamSetMode(group.dma, STM32_DMA_CR_CHSEL(group.dma_up_channel) | STM32_DMA_CR_DIR_M2P | STM32_DMA_CR_PSIZE_WORD | STM32_DMA_CR_MSIZE_WORD | STM32_DMA_CR_MINC | STM32_DMA_CR_PL(3) | STM32_DMA_CR_TEIE | STM32_DMA_CR_TCIE); // setup for burst strided transfers into the timers 4 CCR registers const uint8_t ccr_ofs = offsetof(stm32_tim_t, CCR)/4; group.pwm_drv->tim->DCR = STM32_TIM_DCR_DBA(ccr_ofs) | STM32_TIM_DCR_DBL(3); group.dshot_state = DshotState::SEND_START; TOGGLE_PIN_DEBUG(54); dmaStreamEnable(group.dma); // record when the transaction was started group.last_dmar_send_us = AP_HAL::micros64(); #endif //#ifndef DISABLE_DSHOT } /* unlock DMA channel after a dshot send completes and no return value is expected */ void RCOutput::dma_unlock(void *p) { chSysLockFromISR(); pwm_group *group = (pwm_group *)p; group->dshot_state = DshotState::IDLE; if (group->dshot_waiter != nullptr) { // tell the waiting process we've done the DMA. Note that // dshot_waiter can be null if we have cancelled the send chEvtSignalI(group->dshot_waiter, group->dshot_event_mask); } chSysUnlockFromISR(); } #ifndef HAL_WITH_BIDIR_DSHOT /* DMA interrupt handler. Used to mark DMA completed for DShot */ void RCOutput::dma_up_irq_callback(void *p, uint32_t flags) { pwm_group *group = (pwm_group *)p; chSysLockFromISR(); dmaStreamDisable(group->dma); if (group->in_serial_dma && irq.waiter) { // tell the waiting process we've done the DMA chEvtSignalI(irq.waiter, serial_event_mask); } else { // this prevents us ever having two dshot pulses too close together // dshot mandates a minimum pulse separation of 40us, WS2812 mandates 50us so we // pick the higher value chVTSetI(&group->dma_timeout, chTimeUS2I(50), dma_unlock, p); } chSysUnlockFromISR(); } #endif /* Cancel a DMA transaction in progress */ void RCOutput::dma_cancel(pwm_group& group) { chSysLock(); dmaStreamDisable(group.dma); #ifdef HAL_WITH_BIDIR_DSHOT if (group.ic_dma_enabled()) { dmaStreamDisable(group.bdshot.ic_dma[group.bdshot.curr_telem_chan]); } #endif // normally the CCR registers are reset by the final 0 in the DMA buffer // since we are cancelling early they need to be reset to avoid infinite pulses for (uint8_t i = 0; i < 4; i++) { if (group.chan[i] != CHAN_DISABLED) { group.pwm_drv->tim->CCR[i] = 0; } } chVTResetI(&group.dma_timeout); chEvtGetAndClearEventsI(group.dshot_event_mask); group.dshot_state = DshotState::IDLE; chSysUnlock(); } /* setup for serial output to an ESC using the given baudrate. Assumes 1 start bit, 1 stop bit, LSB first and 8 databits. This is used for passthrough ESC configuration and firmware flashing While serial output is active normal output to the channel group is suspended. */ bool RCOutput::serial_setup_output(uint8_t chan, uint32_t baudrate, uint16_t chanmask) { // account for IOMCU channels chan -= chan_offset; chanmask >>= chan_offset; pwm_group *new_serial_group = nullptr; // find the channel group for (auto &group : pwm_group_list) { if (group.current_mode == MODE_PWM_BRUSHED) { // can't do serial output with brushed motors continue; } if (group.ch_mask & (1U<realprio; chThdSetPriority(HIGHPRIO); // remember the bit period for serial_read_byte() serial_group->serial.bit_time_us = 1000000UL / baudrate; // remember the thread that set things up. This is also used to // mark the group as doing serial output, so normal output is // suspended irq.waiter = chThdGetSelfX(); return true; } /* fill in a DMA buffer for a serial byte, assuming 1 start bit and 1 stop bit */ void RCOutput::fill_DMA_buffer_byte(uint32_t *buffer, uint8_t stride, uint8_t b, uint32_t bitval) { const uint32_t BIT_0 = bitval; const uint32_t BIT_1 = 0; // start bit buffer[0] = BIT_0; // stop bit buffer[9*stride] = BIT_1; // 8 data bits for (uint8_t i = 0; i < 8; i++) { buffer[(1 + i) * stride] = (b & 1) ? BIT_1 : BIT_0; b >>= 1; } } /* send one serial byte, blocking call, should be called with the DMA lock held */ bool RCOutput::serial_write_byte(uint8_t b) { chEvtGetAndClearEvents(serial_event_mask); fill_DMA_buffer_byte(serial_group->dma_buffer+serial_group->serial.chan, 4, b, serial_group->bit_width_mul*10); serial_group->in_serial_dma = true; // start sending the pulses out send_pulses_DMAR(*serial_group, 10*4*sizeof(uint32_t)); // wait for the event eventmask_t mask = chEvtWaitAnyTimeout(serial_event_mask, chTimeMS2I(2)); serial_group->in_serial_dma = false; return (mask & serial_event_mask) != 0; } /* send a set of serial bytes, blocking call */ bool RCOutput::serial_write_bytes(const uint8_t *bytes, uint16_t len) { #ifndef DISABLE_DSHOT if (!serial_group) { return false; } serial_group->dma_handle->lock(); memset(serial_group->dma_buffer, 0, DSHOT_BUFFER_LENGTH); while (len--) { if (!serial_write_byte(*bytes++)) { serial_group->dma_handle->unlock(); return false; } } // add a small delay for last word of output to have completely // finished hal.scheduler->delay_microseconds(25); serial_group->dma_handle->unlock(); return true; #else return false; #endif // DISABLE_DSHOT } /* irq handler for bit transition in serial_read_byte() This implements a one byte soft serial reader */ void RCOutput::serial_bit_irq(void) { uint32_t now = AP_HAL::micros(); uint8_t bit = palReadLine(irq.line); bool send_signal = false; #if RCOU_SERIAL_TIMING_DEBUG palWriteLine(HAL_GPIO_LINE_GPIO55, bit); #endif if (irq.nbits == 0 || bit == irq.last_bit) { // start of byte, should be low if (bit != 0) { irq.byteval = 0x200; send_signal = true; } else { irq.nbits = 1; irq.byte_start_tick = now; irq.bitmask = 0; } } else { systime_t dt = now - irq.byte_start_tick; uint8_t bitnum = (dt+(irq.bit_time_tick/2)) / irq.bit_time_tick; if (bitnum > 10) { bitnum = 10; } if (!bit) { // set the bits that we've processed irq.bitmask |= ((1U<>1); return true; } /* read a byte from a port, using serial parameters from serial_setup_output() */ uint16_t RCOutput::serial_read_bytes(uint8_t *buf, uint16_t len) { #ifndef DISABLE_SERIAL_ESC_COMM if (serial_group == nullptr) { return 0; } pwm_group &group = *serial_group; const ioline_t line = group.pal_lines[group.serial.chan]; // keep speed low to avoid noise when switching between input and output uint32_t gpio_mode = PAL_STM32_MODE_INPUT | PAL_STM32_OTYPE_PUSHPULL | PAL_STM32_PUPDR_PULLUP | PAL_STM32_OSPEED_LOWEST; // restore the line to what it was before iomode_t restore_mode = palReadLineMode(line); uint16_t i = 0; #if RCOU_SERIAL_TIMING_DEBUG hal.gpio->pinMode(54, 1); hal.gpio->pinMode(55, 1); #endif // assume GPIO mappings for PWM outputs start at 50 palSetLineMode(line, gpio_mode); chVTObjectInit(&irq.serial_timeout); chEvtGetAndClearEvents(serial_event_mask); irq.line = group.pal_lines[group.serial.chan]; irq.nbits = 0; irq.bitmask = 0; irq.byteval = 0; irq.bit_time_tick = serial_group->serial.bit_time_us; irq.last_bit = 0; irq.waiter = chThdGetSelfX(); #if RCOU_SERIAL_TIMING_DEBUG palWriteLine(HAL_GPIO_LINE_GPIO54, 1); #endif if (!((GPIO *)hal.gpio)->_attach_interrupt(line, serial_bit_irq, AP_HAL::GPIO::INTERRUPT_BOTH)) { #if RCOU_SERIAL_TIMING_DEBUG palWriteLine(HAL_GPIO_LINE_GPIO54, 0); #endif return false; } for (i=0; i_attach_interrupt(line, nullptr, 0); irq.waiter = nullptr; palSetLineMode(line, restore_mode); #if RCOU_SERIAL_TIMING_DEBUG palWriteLine(HAL_GPIO_LINE_GPIO54, 0); #endif return i; #else return false; #endif //#ifndef DISABLE_SERIAL_ESC_COMM } /* end serial output */ void RCOutput::serial_end(void) { #ifndef DISABLE_SERIAL_ESC_COMM if (serial_group) { if (serial_thread == chThdGetSelfX()) { chThdSetPriority(serial_priority); serial_thread = nullptr; } irq.waiter = nullptr; for (uint8_t i = 0; i < NUM_GROUPS; i++ ) { pwm_group &group = pwm_group_list[i]; // restore normal output if (group.pwm_started) { pwmStop(group.pwm_drv); group.pwm_started = false; } set_group_mode(group); set_freq_group(group); } } serial_group = nullptr; #endif //#ifndef DISABLE_SERIAL_ESC_COMM } /* get safety switch state for Util.cpp */ AP_HAL::Util::safety_state RCOutput::_safety_switch_state(void) { #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { safety_state = iomcu.get_safety_switch_state(); } #endif if (!hal.util->was_watchdog_reset()) { hal.util->persistent_data.safety_state = safety_state; } return safety_state; } /* force the safety switch on, disabling PWM output from the IO board */ bool RCOutput::force_safety_on(void) { #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { return iomcu.force_safety_on(); } return false; #else safety_state = AP_HAL::Util::SAFETY_DISARMED; return true; #endif } /* force the safety switch off, enabling PWM output from the IO board */ void RCOutput::force_safety_off(void) { #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { iomcu.force_safety_off(); } #else safety_state = AP_HAL::Util::SAFETY_ARMED; #endif } /* set PWM to send to a set of channels when the safety switch is in the safe state */ void RCOutput::set_safety_pwm(uint32_t chmask, uint16_t period_us) { #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { iomcu.set_safety_pwm(chmask, period_us); } #endif for (uint8_t i=0; i<16; i++) { if (chmask & (1U<get_safety_mask(); } #ifdef HAL_GPIO_PIN_SAFETY_IN // handle safety button bool safety_pressed = palReadLine(HAL_GPIO_PIN_SAFETY_IN); if (safety_pressed) { AP_BoardConfig *brdconfig = AP_BoardConfig::get_singleton(); if (safety_press_count < 255) { safety_press_count++; } if (brdconfig && brdconfig->safety_button_handle_pressed(safety_press_count)) { if (safety_state ==AP_HAL::Util::SAFETY_ARMED) { safety_state = AP_HAL::Util::SAFETY_DISARMED; } else { safety_state = AP_HAL::Util::SAFETY_ARMED; } } } else { safety_press_count = 0; } #elif HAL_WITH_IO_MCU safety_state = _safety_switch_state(); iomcu.set_safety_mask(safety_mask); #endif #ifdef HAL_GPIO_PIN_LED_SAFETY led_counter = (led_counter+1) % 16; const uint16_t led_pattern = safety_state==AP_HAL::Util::SAFETY_DISARMED?0x5500:0xFFFF; palWriteLine(HAL_GPIO_PIN_LED_SAFETY, (led_pattern & (1U << led_counter))?0:1); #endif } /* set PWM to send to a set of channels if the FMU firmware dies */ void RCOutput::set_failsafe_pwm(uint32_t chmask, uint16_t period_us) { #if HAL_WITH_IO_MCU if (AP_BoardConfig::io_enabled()) { iomcu.set_failsafe_pwm(chmask, period_us); } #endif } /* true when the output mode is of type dshot */ bool RCOutput::is_dshot_protocol(const enum output_mode mode) { switch (mode) { case MODE_PWM_DSHOT150: case MODE_PWM_DSHOT300: case MODE_PWM_DSHOT600: case MODE_PWM_DSHOT1200: return true; default: return false; } } /* returns the bitrate in Hz of the given output_mode */ uint32_t RCOutput::protocol_bitrate(const enum output_mode mode) { switch (mode) { case MODE_PWM_DSHOT150: return 150000; case MODE_PWM_DSHOT300: return 300000; case MODE_PWM_DSHOT600: return 600000; case MODE_PWM_DSHOT1200: return 1200000; case MODE_NEOPIXEL: return 800000; case MODE_PROFILED: return 1500000; // experiment winding this up 3000000 max from data sheet default: // use 1 to prevent a possible divide-by-zero return 1; } } /* setup serial led output for a given channel number, with the given max number of LEDs in the chain. */ bool RCOutput::set_serial_led_num_LEDs(const uint16_t chan, uint8_t num_leds, output_mode mode, uint16_t clock_mask) { if (!_initialised || num_leds == 0) { return false; } uint8_t i = 0; pwm_group *grp = find_chan(chan, i); if (!grp) { return false; } WITH_SEMAPHORE(grp->serial_led_mutex); // group is already setup correctly if ((grp->serial_nleds >= num_leds) && (mode == grp->current_mode)) { return true; } // we cant add more or change the type after the first setup if (grp->current_mode == MODE_NEOPIXEL || grp->current_mode == MODE_PROFILED) { return false; } switch (mode) { case MODE_NEOPIXEL: { grp->serial_nleds = MAX(num_leds, grp->serial_nleds); grp->led_mode = MODE_NEOPIXEL; return true; } case MODE_PROFILED: { // ProfiLED requires two dummy LED's to mark end of transmission grp->serial_nleds = MAX(num_leds + 2, grp->serial_nleds); grp->led_mode = MODE_PROFILED; // Enable any clock channels in the same group grp->clock_mask = 0; for (uint8_t j = 0; j < 4; j++) { if ((clock_mask & (1U<<(grp->chan[j] + chan_offset))) != 0) { grp->clock_mask |= 1U<serial_nleds = 0; grp->led_mode = MODE_PWM_NONE; return false; } } #pragma GCC push_options #pragma GCC optimize("O2") // Fill the group DMA buffer with data to be output void RCOutput::fill_DMA_buffer_serial_led(pwm_group& group) { memset(group.dma_buffer, 0, group.dma_buffer_len); for (uint8_t j = 0; j < 4; j++) { if (group.serial_led_data[j] == nullptr) { // something very bad has happended continue; } if (group.current_mode == MODE_PROFILED && (group.clock_mask & 1U<dma_buffer + (led * neopixel_bit_length + pad_start_bits) * stride + idx; uint32_t bits = (green<<16) | (red<<8) | blue; const uint32_t BIT_0 = 7 * grp->bit_width_mul; const uint32_t BIT_1 = 14 * grp->bit_width_mul; for (uint16_t b=0; b < 24; b++) { buf[b * stride] = (bits & 0x800000) ? BIT_1 : BIT_0; bits <<= 1; } } /* ProfiLED frame for a given output channel channel is active high and bits inverted to get clock rising edge away from data rising edge */ void RCOutput::_set_profiled_rgb_data(pwm_group *grp, uint8_t idx, uint8_t led, uint8_t red, uint8_t green, uint8_t blue) { const uint8_t pad_start_bits = 1; const uint8_t bit_length = 25; const uint8_t stride = 4; uint32_t *buf = grp->dma_buffer + (led * bit_length + pad_start_bits) * stride + idx; uint32_t bits = 0x1000000 | (blue<<16) | (red<<8) | green; const uint32_t BIT_1 = 14 * grp->bit_width_mul; for (uint16_t b=0; b < bit_length; b++) { buf[b * stride] = (bits & 0x1000000) ? 0 : BIT_1; bits <<= 1; } } /* ProfiLED blank frame for a given output channel channel is active high and bits inverted to get clock rising edge away from data rising edge */ void RCOutput::_set_profiled_blank_frame(pwm_group *grp, uint8_t idx, uint8_t led) { const uint8_t pad_start_bits = 1; const uint8_t bit_length = 25; const uint8_t stride = 4; uint32_t *buf = grp->dma_buffer + (led * bit_length + pad_start_bits) * stride + idx; const uint32_t BIT_1 = 14 * grp->bit_width_mul; for (uint16_t b=0; b < bit_length; b++) { buf[b * stride] = BIT_1; } } /* setup ProfiLED clock frame for a given output channel */ void RCOutput::_set_profiled_clock(pwm_group *grp, uint8_t idx, uint8_t led) { const uint8_t pad_start_bits = 1; const uint8_t bit_length = 25; const uint8_t stride = 4; uint32_t *buf = grp->dma_buffer + (led * bit_length + pad_start_bits) * stride + idx; const uint32_t BIT_1 = 7 * grp->bit_width_mul; for (uint16_t b=0; b < bit_length; b++) { buf[b * stride] = BIT_1; } } #pragma GCC pop_options /* setup serial LED output data for a given output channel and a LED number. LED -1 is all LEDs */ void RCOutput::set_serial_led_rgb_data(const uint16_t chan, int8_t led, uint8_t red, uint8_t green, uint8_t blue) { if (!_initialised) { return; } uint8_t i = 0; pwm_group *grp = find_chan(chan, i); if (!grp) { return; } if (grp->serial_led_pending) { // dont allow setting new data if a send is pending // would result in a fight over the mutex return; }; WITH_SEMAPHORE(grp->serial_led_mutex); if (grp->serial_nleds == 0 || led >= grp->serial_nleds) { return; } if ((grp->current_mode != grp->led_mode) && ((grp->led_mode == MODE_NEOPIXEL) || (grp->led_mode == MODE_PROFILED))) { // Arrays have not yet been setup, do it now for (uint8_t j = 0; j < 4; j++) { delete[] grp->serial_led_data[j]; grp->serial_led_data[j] = nullptr; grp->serial_led_data[j] = new SerialLed[grp->serial_nleds]; if (grp->serial_led_data[j] == nullptr) { // if allocation failed clear all memory for (uint8_t k = 0; k < 4; k++) { delete[] grp->serial_led_data[k]; grp->serial_led_data[k] = nullptr; } grp->led_mode = MODE_PWM_NONE; grp->serial_nleds = 0; return; } } // at this point the group led data is all setup but the dma buffer still needs to be resized set_output_mode(1U<led_mode); if (grp->current_mode != grp->led_mode) { // Failed to set output mode grp->led_mode = MODE_PWM_NONE; grp->serial_nleds = 0; return; } } else if ((grp->current_mode != MODE_NEOPIXEL) && (grp->current_mode != MODE_PROFILED)) { return; } if (led == -1) { grp->prepared_send = true; for (uint8_t n=0; nserial_nleds; n++) { serial_led_set_single_rgb_data(*grp, i, n, red, green, blue); } return; } // if not ouput clock and trailing frames, run through all LED's to do it now if (!grp->prepared_send) { grp->prepared_send = true; for (uint8_t n=0; nserial_nleds; n++) { serial_led_set_single_rgb_data(*grp, i, n, 0, 0, 0); } } serial_led_set_single_rgb_data(*grp, i, uint8_t(led), red, green, blue); } /* setup serial LED output data for a given output channel and a LED number. LED -1 is all LEDs */ void RCOutput::serial_led_set_single_rgb_data(pwm_group& group, uint8_t idx, uint8_t led, uint8_t red, uint8_t green, uint8_t blue) { switch (group.current_mode) { case MODE_PROFILED: case MODE_NEOPIXEL: group.serial_led_data[idx][led].red = red; group.serial_led_data[idx][led].green = green; group.serial_led_data[idx][led].blue = blue; break; default: break; } } /* trigger send of serial led data for one group */ void RCOutput::serial_led_send(const uint16_t chan) { if (!_initialised) { return; } uint8_t i; pwm_group *grp = find_chan(chan, i); if (!grp) { return; } WITH_SEMAPHORE(grp->serial_led_mutex); if (grp->serial_nleds == 0 || (grp->current_mode != MODE_NEOPIXEL && grp->current_mode != MODE_PROFILED)) { return; } if (grp->prepared_send) { grp->serial_led_pending = true; serial_led_pending = true; } } #endif // HAL_USE_PWM