src/main/drivers/dshot_bitbang.c

   1 /*
   2  * This file is part of Cleanflight and Betaflight.
   3  *
   4  * Cleanflight and Betaflight are free software. You can redistribute
   5  * this software and/or modify this software under the terms of the
   6  * GNU General Public License as published by the Free Software
   7  * Foundation, either version 3 of the License, or (at your option)
   8  * any later version.
   9  *
  10  * Cleanflight and Betaflight are distributed in the hope that they
  11  * will be useful, but WITHOUT ANY WARRANTY; without even the implied
  12  * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13  * See the GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this software.
  17  *
  18  * If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #include <stdint.h>
  22 #include <math.h>
  23 #include <string.h>
  24
  25 #include "platform.h"
  26
  27 #ifdef USE_DSHOT_BITBANG
  28
  29 #include "build/debug.h"
  30 #include "build/debug_pin.h"
  31
  32 #include "drivers/io.h"
  33 #include "drivers/io_impl.h"
  34 #include "drivers/dma.h"
  35 #include "drivers/dma_reqmap.h"
  36 #include "drivers/dshot.h"
  37 #include "drivers/dshot_bitbang.h"
  38 #include "drivers/dshot_bitbang_impl.h"
  39 #include "drivers/dshot_command.h"
  40 #include "drivers/motor.h"
  41 #include "drivers/nvic.h"
  42 #include "drivers/pwm_output.h" // XXX for pwmOutputPort_t motors[]; should go away with refactoring
  43 #include "drivers/dshot_dpwm.h" // XXX for motorDmaOutput_t *getMotorDmaOutput(uint8_t index); should go away with refactoring
  44 #include "drivers/dshot_bitbang_decode.h"
  45 #include "drivers/time.h"
  46 #include "drivers/timer.h"
  47
  48 #include "pg/motor.h"
  49 #include "pg/pinio.h"
  50
  51 // DEBUG_DSHOT_TELEMETRY_COUNTS
  52 // 0 - Count of telemetry packets read
  53 // 1 - Count of missing edge
  54 // 2 - Count of reception not complete in time
  55 // 3 - Number of high bits before telemetry start
  56
  57 // Maximum time to wait for telemetry reception to complete
  58 #define DSHOT_TELEMETRY_TIMEOUT 2000
  59
  60 FAST_DATA_ZERO_INIT bbPacer_t bbPacers[MAX_MOTOR_PACERS];  // TIM1 or TIM8
  61 FAST_DATA_ZERO_INIT int usedMotorPacers = 0;
  62
  63 FAST_DATA_ZERO_INIT bbPort_t bbPorts[MAX_SUPPORTED_MOTOR_PORTS];
  64 FAST_DATA_ZERO_INIT int usedMotorPorts;
  65
  66 FAST_DATA_ZERO_INIT bbMotor_t bbMotors[MAX_SUPPORTED_MOTORS];
  67
  68 static FAST_DATA_ZERO_INIT int motorCount;
  69 dshotBitbangStatus_e bbStatus;
  70
  71 // For MCUs that use MPU to control DMA coherency, there might be a performance hit
  72 // on manipulating input buffer content especially if it is read multiple times,
  73 // as the buffer region is attributed as not cachable.
  74 // If this is not desirable, we should use manual cache invalidation.
  75 #ifdef USE_DSHOT_CACHE_MGMT
  76 #define BB_OUTPUT_BUFFER_ATTRIBUTE DMA_RW_AXI __attribute__((aligned(32)))
  77 #define BB_INPUT_BUFFER_ATTRIBUTE  DMA_RW_AXI __attribute__((aligned(32)))
  78 #else
  79 #if defined(STM32F4)
  80 #define BB_OUTPUT_BUFFER_ATTRIBUTE
  81 #define BB_INPUT_BUFFER_ATTRIBUTE
  82 #elif defined(STM32F7)
  83 #define BB_OUTPUT_BUFFER_ATTRIBUTE FAST_DATA_ZERO_INIT
  84 #define BB_INPUT_BUFFER_ATTRIBUTE  FAST_DATA_ZERO_INIT
  85 #elif defined(STM32H7)
  86 #define BB_OUTPUT_BUFFER_ATTRIBUTE DMA_RAM
  87 #define BB_INPUT_BUFFER_ATTRIBUTE  DMA_RAM
  88 #elif defined(STM32G4)
  89 #define BB_OUTPUT_BUFFER_ATTRIBUTE FAST_DATA_ZERO_INIT
  90 #define BB_INPUT_BUFFER_ATTRIBUTE  FAST_DATA_ZERO_INIT
  91 #endif
  92 #endif // USE_DSHOT_CACHE_MGMT
  93
  94 BB_OUTPUT_BUFFER_ATTRIBUTE uint32_t bbOutputBuffer[MOTOR_DSHOT_BUF_CACHE_ALIGN_LENGTH * MAX_SUPPORTED_MOTOR_PORTS];
  95 BB_INPUT_BUFFER_ATTRIBUTE uint16_t bbInputBuffer[DSHOT_BB_PORT_IP_BUF_CACHE_ALIGN_LENGTH * MAX_SUPPORTED_MOTOR_PORTS];
  96
  97 uint8_t bbPuPdMode;
  98 FAST_DATA_ZERO_INIT timeUs_t dshotFrameUs;
  99
 100
 101 const timerHardware_t bbTimerHardware[] = {
 102 #if defined(STM32F4) || defined(STM32F7)
 103 #if !defined(STM32F411xE)
 104     DEF_TIM(TIM8,  CH1, NONE,  TIM_USE_NONE, 0, 1),
 105     DEF_TIM(TIM8,  CH2, NONE,  TIM_USE_NONE, 0, 1),
 106     DEF_TIM(TIM8,  CH3, NONE,  TIM_USE_NONE, 0, 1),
 107     DEF_TIM(TIM8,  CH4, NONE,  TIM_USE_NONE, 0, 0),
 108 #endif
 109     DEF_TIM(TIM1,  CH1, NONE,  TIM_USE_NONE, 0, 1),
 110     DEF_TIM(TIM1,  CH1, NONE,  TIM_USE_NONE, 0, 2),
 111     DEF_TIM(TIM1,  CH2, NONE,  TIM_USE_NONE, 0, 1),
 112     DEF_TIM(TIM1,  CH3, NONE,  TIM_USE_NONE, 0, 1),
 113     DEF_TIM(TIM1,  CH4, NONE,  TIM_USE_NONE, 0, 0),
 114
 115 #elif defined(STM32G4) || defined(STM32H7)
 116     // XXX TODO: STM32G4 and STM32H7 can use any timer for pacing
 117
 118     // DMA request numbers are duplicated for TIM1 and TIM8:
 119     //   - Any pacer can serve a GPIO port.
 120     //   - For quads (or less), 4 pacers can cover the worst case scenario of
 121     //     4 motors scattered across 4 different GPIO ports.
 122     //   - For hexas (and larger), more channels may become necessary,
 123     //     in which case the DMA request numbers should be modified.
 124     DEF_TIM(TIM8,  CH1, NONE,  TIM_USE_NONE, 0, 0, 0),
 125     DEF_TIM(TIM8,  CH2, NONE,  TIM_USE_NONE, 0, 1, 0),
 126     DEF_TIM(TIM8,  CH3, NONE,  TIM_USE_NONE, 0, 2, 0),
 127     DEF_TIM(TIM8,  CH4, NONE,  TIM_USE_NONE, 0, 3, 0),
 128     DEF_TIM(TIM1,  CH1, NONE,  TIM_USE_NONE, 0, 0, 0),
 129     DEF_TIM(TIM1,  CH2, NONE,  TIM_USE_NONE, 0, 1, 0),
 130     DEF_TIM(TIM1,  CH3, NONE,  TIM_USE_NONE, 0, 2, 0),
 131     DEF_TIM(TIM1,  CH4, NONE,  TIM_USE_NONE, 0, 3, 0),
 132
 133 #else
 134 #error MCU dependent code required
 135 #endif
 136 };
 137
 138 static FAST_DATA_ZERO_INIT motorDevice_t bbDevice;
 139 static FAST_DATA_ZERO_INIT timeUs_t lastSendUs;
 140
 141 static motorPwmProtocolTypes_e motorPwmProtocol;
 142
 143 // DMA GPIO output buffer formatting
 144
 145 static void bbOutputDataInit(uint32_t *buffer, uint16_t portMask, bool inverted)
 146 {
 147     uint32_t resetMask;
 148     uint32_t setMask;
 149
 150     if (inverted) {
 151         resetMask = portMask;
 152         setMask = (portMask << 16);
 153     } else {
 154         resetMask = (portMask << 16);
 155         setMask = portMask;
 156     }
 157
 158     int symbol_index;
 159
 160     for (symbol_index = 0; symbol_index < MOTOR_DSHOT_FRAME_BITS; symbol_index++) {
 161         buffer[symbol_index * MOTOR_DSHOT_STATE_PER_SYMBOL + 0] |= setMask ; // Always set all ports
 162         buffer[symbol_index * MOTOR_DSHOT_STATE_PER_SYMBOL + 1] = 0;          // Reset bits are port dependent
 163         buffer[symbol_index * MOTOR_DSHOT_STATE_PER_SYMBOL + 2] |= resetMask; // Always reset all ports
 164     }
 165
 166     //
 167     // output one more 'bit' that keeps the line level at idle to allow the ESC to sample the last bit
 168     //
 169     // Avoid CRC errors in the case of bi-directional d-shot.  CRC errors can occur if the output is
 170     // transitioned to an input before the signal has been sampled by the ESC as the sampled voltage
 171     // may be somewhere between logic-high and logic-low depending on how the motor output line is
 172     // driven or floating.  On some MCUs it's observed that the voltage momentarily drops low on transition
 173     // to input.
 174
 175     int hold_bit_index = MOTOR_DSHOT_FRAME_BITS * MOTOR_DSHOT_STATE_PER_SYMBOL;
 176     buffer[hold_bit_index + 0] |= resetMask; // Always reset all ports
 177     buffer[hold_bit_index + 1] = 0;          // Never any change
 178     buffer[hold_bit_index + 2] = 0;          // Never any change
 179 }
 180
 181 static void bbOutputDataSet(uint32_t *buffer, int pinNumber, uint16_t value, bool inverted)
 182 {
 183     uint32_t middleBit;
 184
 185     if (inverted) {
 186         middleBit = (1 << (pinNumber + 0));
 187     } else {
 188         middleBit = (1 << (pinNumber + 16));
 189     }
 190
 191     for (int pos = 0; pos < 16; pos++) {
 192         if (!(value & 0x8000)) {
 193             buffer[pos * 3 + 1] |= middleBit;
 194         }
 195         value <<= 1;
 196     }
 197 }
 198
 199 static void bbOutputDataClear(uint32_t *buffer)
 200 {
 201     // Middle position to no change
 202     for (int bitpos = 0; bitpos < 16; bitpos++) {
 203         buffer[bitpos * 3 + 1] = 0;
 204     }
 205 }
 206
 207 // bbPacer management
 208
 209 static bbPacer_t *bbFindMotorPacer(TIM_TypeDef *tim)
 210 {
 211     for (int i = 0; i < MAX_MOTOR_PACERS; i++) {
 212
 213         bbPacer_t *bbPacer = &bbPacers[i];
 214
 215         if (bbPacer->tim == NULL) {
 216             bbPacer->tim = tim;
 217             ++usedMotorPacers;
 218             return bbPacer;
 219         }
 220
 221         if (bbPacer->tim == tim) {
 222             return bbPacer;
 223         }
 224     }
 225
 226     return NULL;
 227 }
 228
 229 // bbPort management
 230
 231 static bbPort_t *bbFindMotorPort(int portIndex)
 232 {
 233     for (int i = 0; i < usedMotorPorts; i++) {
 234         if (bbPorts[i].portIndex == portIndex) {
 235             return &bbPorts[i];
 236         }
 237     }
 238     return NULL;
 239 }
 240
 241 static bbPort_t *bbAllocateMotorPort(int portIndex)
 242 {
 243     if (usedMotorPorts >= MAX_SUPPORTED_MOTOR_PORTS) {
 244         bbStatus = DSHOT_BITBANG_STATUS_TOO_MANY_PORTS;
 245         return NULL;
 246     }
 247
 248     bbPort_t *bbPort = &bbPorts[usedMotorPorts];
 249
 250     if (!bbPort->timhw) {
 251         // No more pacer channel available
 252         bbStatus = DSHOT_BITBANG_STATUS_NO_PACER;
 253         return NULL;
 254     }
 255
 256     bbPort->portIndex = portIndex;
 257     bbPort->owner.owner = OWNER_DSHOT_BITBANG;
 258     bbPort->owner.resourceIndex = RESOURCE_INDEX(portIndex);
 259
 260     ++usedMotorPorts;
 261
 262     return bbPort;
 263 }
 264
 265 const timerHardware_t *dshotBitbangTimerGetAllocatedByNumberAndChannel(int8_t timerNumber, uint16_t timerChannel)
 266 {
 267     for (int index = 0; index < usedMotorPorts; index++) {
 268         const timerHardware_t *bitbangTimer = bbPorts[index].timhw;
 269         if (bitbangTimer && timerGetTIMNumber(bitbangTimer->tim) == timerNumber && bitbangTimer->channel == timerChannel && bbPorts[index].owner.owner) {
 270             return bitbangTimer;
 271         }
 272     }
 273
 274     return NULL;
 275 }
 276
 277 const resourceOwner_t *dshotBitbangTimerGetOwner(const timerHardware_t *timer)
 278 {
 279     for (int index = 0; index < usedMotorPorts; index++) {
 280         const timerHardware_t *bitbangTimer = bbPorts[index].timhw;
 281         if (bitbangTimer && bitbangTimer == timer) {
 282             return &bbPorts[index].owner;
 283         }
 284     }
 285
 286     return &freeOwner;
 287 }
 288
 289 // Return frequency of smallest change [state/sec]
 290
 291 static uint32_t getDshotBaseFrequency(motorPwmProtocolTypes_e pwmProtocolType)
 292 {
 293     switch (pwmProtocolType) {
 294     case(PWM_TYPE_DSHOT600):
 295         return MOTOR_DSHOT600_SYMBOL_RATE * MOTOR_DSHOT_STATE_PER_SYMBOL;
 296     case(PWM_TYPE_DSHOT300):
 297         return MOTOR_DSHOT300_SYMBOL_RATE * MOTOR_DSHOT_STATE_PER_SYMBOL;
 298     default:
 299     case(PWM_TYPE_DSHOT150):
 300         return MOTOR_DSHOT150_SYMBOL_RATE * MOTOR_DSHOT_STATE_PER_SYMBOL;
 301     }
 302 }
 303
 304 static void bbSetupDma(bbPort_t *bbPort)
 305 {
 306     const dmaIdentifier_e dmaIdentifier = dmaGetIdentifier(bbPort->dmaResource);
 307     dmaEnable(dmaIdentifier);
 308     bbPort->dmaSource = timerDmaSource(bbPort->timhw->channel);
 309
 310     bbPacer_t *bbPacer = bbFindMotorPacer(bbPort->timhw->tim);
 311     bbPacer->dmaSources |= bbPort->dmaSource;
 312
 313     dmaSetHandler(dmaIdentifier, bbDMAIrqHandler, NVIC_BUILD_PRIORITY(2, 1), (uint32_t)bbPort);
 314
 315     bbDMA_ITConfig(bbPort);
 316 }
 317
 318 FAST_IRQ_HANDLER void bbDMAIrqHandler(dmaChannelDescriptor_t *descriptor)
 319 {
 320     dbgPinHi(0);
 321
 322     bbPort_t *bbPort = (bbPort_t *)descriptor->userParam;
 323
 324     bbDMA_Cmd(bbPort, DISABLE);
 325
 326     bbTIM_DMACmd(bbPort->timhw->tim, bbPort->dmaSource, DISABLE);
 327
 328     if (DMA_GET_FLAG_STATUS(descriptor, DMA_IT_TEIF)) {
 329         while (1) {};
 330     }
 331
 332     DMA_CLEAR_FLAG(descriptor, DMA_IT_TCIF);
 333
 334 #ifdef USE_DSHOT_TELEMETRY
 335     if (useDshotTelemetry) {
 336         if (bbPort->direction == DSHOT_BITBANG_DIRECTION_INPUT) {
 337             bbPort->telemetryPending = false;
 338 #ifdef DEBUG_COUNT_INTERRUPT
 339             bbPort->inputIrq++;
 340 #endif
 341             // Disable DMA as telemetry reception is complete
 342             bbDMA_Cmd(bbPort, DISABLE);
 343         } else {
 344 #ifdef DEBUG_COUNT_INTERRUPT
 345             bbPort->outputIrq++;
 346 #endif
 347
 348             // Switch to input
 349
 350             bbSwitchToInput(bbPort);
 351             bbPort->telemetryPending = true;
 352
 353             bbTIM_DMACmd(bbPort->timhw->tim, bbPort->dmaSource, ENABLE);
 354         }
 355     }
 356 #endif
 357     dbgPinLo(0);
 358 }
 359
 360 // Setup bbPorts array elements so that they each have a TIM1 or TIM8 channel
 361 // in timerHardware array for BB-DShot.
 362
 363 static void bbFindPacerTimer(void)
 364 {
 365     for (int bbPortIndex = 0; bbPortIndex < MAX_SUPPORTED_MOTOR_PORTS; bbPortIndex++) {
 366         for (unsigned timerIndex = 0; timerIndex < ARRAYLEN(bbTimerHardware); timerIndex++) {
 367             const timerHardware_t *timer = &bbTimerHardware[timerIndex];
 368             int timNumber = timerGetTIMNumber(timer->tim);
 369             if ((motorConfig()->dev.useDshotBitbangedTimer == DSHOT_BITBANGED_TIMER_TIM1 && timNumber != 1)
 370                 || (motorConfig()->dev.useDshotBitbangedTimer == DSHOT_BITBANGED_TIMER_TIM8 && timNumber != 8)) {
 371                 continue;
 372             }
 373             bool timerConflict = false;
 374             for (int channel = 0; channel < CC_CHANNELS_PER_TIMER; channel++) {
 375                 const timerHardware_t *timer = timerGetAllocatedByNumberAndChannel(timNumber, CC_CHANNEL_FROM_INDEX(channel));
 376                 const resourceOwner_e timerOwner = timerGetOwner(timer)->owner;
 377                 if (timerOwner != OWNER_FREE && timerOwner != OWNER_DSHOT_BITBANG) {
 378                     timerConflict = true;
 379                     break;
 380                 }
 381             }
 382
 383             for (int index = 0; index < bbPortIndex; index++) {
 384                 const timerHardware_t* t = bbPorts[index].timhw;
 385                 if (timerGetTIMNumber(t->tim) == timNumber && timer->channel == t->channel) {
 386                     timerConflict = true;
 387                     break;
 388                 }
 389             }
 390
 391             if (timerConflict) {
 392                 continue;
 393             }
 394
 395 #ifdef USE_DMA_SPEC
 396             dmaoptValue_t dmaopt = dmaGetOptionByTimer(timer);
 397             const dmaChannelSpec_t *dmaChannelSpec = dmaGetChannelSpecByTimerValue(timer->tim, timer->channel, dmaopt);
 398             dmaResource_t *dma = dmaChannelSpec->ref;
 399 #else
 400             dmaResource_t *dma = timer->dmaRef;
 401 #endif
 402             dmaIdentifier_e dmaIdentifier = dmaGetIdentifier(dma);
 403             if (dmaGetOwner(dmaIdentifier)->owner == OWNER_FREE) {
 404                 bbPorts[bbPortIndex].timhw = timer;
 405
 406                 break;
 407             }
 408         }
 409     }
 410 }
 411
 412 static void bbTimebaseSetup(bbPort_t *bbPort, motorPwmProtocolTypes_e dshotProtocolType)
 413 {
 414     uint32_t timerclock = timerClock(bbPort->timhw->tim);
 415
 416     uint32_t outputFreq = getDshotBaseFrequency(dshotProtocolType);
 417     dshotFrameUs = 1000000 * 17 * 3 / outputFreq;
 418     bbPort->outputARR = timerclock / outputFreq - 1;
 419
 420     // XXX Explain this formula
 421     uint32_t inputFreq = outputFreq * 5 * 2 * DSHOT_BITBANG_TELEMETRY_OVER_SAMPLE / 24;
 422     bbPort->inputARR = timerclock / inputFreq - 1;
 423 }
 424
 425 //
 426 // bb only use pin info associated with timerHardware entry designated as TIM_USE_MOTOR;
 427 // it does not use the timer channel associated with the pin.
 428 //
 429
 430 static bool bbMotorConfig(IO_t io, uint8_t motorIndex, motorPwmProtocolTypes_e pwmProtocolType, uint8_t output)
 431 {
 432     int pinIndex = IO_GPIOPinIdx(io);
 433     int portIndex = IO_GPIOPortIdx(io);
 434
 435     bbPort_t *bbPort = bbFindMotorPort(portIndex);
 436
 437     if (!bbPort) {
 438
 439         // New port group
 440
 441         bbPort = bbAllocateMotorPort(portIndex);
 442
 443         if (bbPort) {
 444             const timerHardware_t *timhw = bbPort->timhw;
 445
 446 #ifdef USE_DMA_SPEC
 447             const dmaChannelSpec_t *dmaChannelSpec = dmaGetChannelSpecByTimerValue(timhw->tim, timhw->channel, dmaGetOptionByTimer(timhw));
 448             bbPort->dmaResource = dmaChannelSpec->ref;
 449             bbPort->dmaChannel = dmaChannelSpec->channel;
 450 #else
 451             bbPort->dmaResource = timhw->dmaRef;
 452             bbPort->dmaChannel = timhw->dmaChannel;
 453 #endif
 454         }
 455
 456         if (!bbPort || !dmaAllocate(dmaGetIdentifier(bbPort->dmaResource), bbPort->owner.owner, bbPort->owner.resourceIndex)) {
 457             bbDevice.vTable.write = motorWriteNull;
 458             bbDevice.vTable.decodeTelemetry = motorDecodeTelemetryNull;
 459             bbDevice.vTable.updateComplete = motorUpdateCompleteNull;
 460
 461             return false;
 462         }
 463
 464         bbPort->gpio = IO_GPIO(io);
 465
 466         bbPort->portOutputCount = MOTOR_DSHOT_BUF_LENGTH;
 467         bbPort->portOutputBuffer = &bbOutputBuffer[(bbPort - bbPorts) * MOTOR_DSHOT_BUF_CACHE_ALIGN_LENGTH];
 468
 469         bbPort->portInputCount = DSHOT_BB_PORT_IP_BUF_LENGTH;
 470         bbPort->portInputBuffer = &bbInputBuffer[(bbPort - bbPorts) * DSHOT_BB_PORT_IP_BUF_CACHE_ALIGN_LENGTH];
 471
 472         bbTimebaseSetup(bbPort, pwmProtocolType);
 473         bbTIM_TimeBaseInit(bbPort, bbPort->outputARR);
 474         bbTimerChannelInit(bbPort);
 475
 476         bbSetupDma(bbPort);
 477         bbDMAPreconfigure(bbPort, DSHOT_BITBANG_DIRECTION_OUTPUT);
 478         bbDMAPreconfigure(bbPort, DSHOT_BITBANG_DIRECTION_INPUT);
 479
 480         bbDMA_ITConfig(bbPort);
 481     }
 482
 483     bbMotors[motorIndex].pinIndex = pinIndex;
 484     bbMotors[motorIndex].io = io;
 485     bbMotors[motorIndex].output = output;
 486     bbMotors[motorIndex].bbPort = bbPort;
 487
 488     IOInit(io, OWNER_MOTOR, RESOURCE_INDEX(motorIndex));
 489
 490     // Setup GPIO_MODER and GPIO_ODR register manipulation values
 491
 492     bbGpioSetup(&bbMotors[motorIndex]);
 493
 494 #ifdef USE_DSHOT_TELEMETRY
 495     if (useDshotTelemetry) {
 496         bbOutputDataInit(bbPort->portOutputBuffer, (1 << pinIndex), DSHOT_BITBANG_INVERTED);
 497     } else
 498 #endif
 499     {
 500         bbOutputDataInit(bbPort->portOutputBuffer, (1 << pinIndex), DSHOT_BITBANG_NONINVERTED);
 501     }
 502
 503     bbSwitchToOutput(bbPort);
 504
 505     bbMotors[motorIndex].configured = true;
 506
 507     return true;
 508 }
 509
 510 static bool bbTelemetryWait(void)
 511 {
 512     // Wait for telemetry reception to complete
 513     bool telemetryPending;
 514     bool telemetryWait = false;
 515     const timeUs_t startTimeUs = micros();
 516
 517     do {
 518         telemetryPending = false;
 519         for (int i = 0; i < usedMotorPorts; i++) {
 520             telemetryPending |= bbPorts[i].telemetryPending;
 521         }
 522
 523         telemetryWait |= telemetryPending;
 524
 525         if (cmpTimeUs(micros(), startTimeUs) > DSHOT_TELEMETRY_TIMEOUT) {
 526             break;
 527         }
 528     } while (telemetryPending);
 529
 530     if (telemetryWait) {
 531         DEBUG_SET(DEBUG_DSHOT_TELEMETRY_COUNTS, 2, debug[2] + 1);
 532     }
 533
 534     return telemetryWait;
 535 }
 536
 537 static void bbUpdateInit(void)
 538 {
 539     for (int i = 0; i < usedMotorPorts; i++) {
 540         bbOutputDataClear(bbPorts[i].portOutputBuffer);
 541     }
 542 }
 543
 544 static bool bbDecodeTelemetry(void)
 545 {
 546 #ifdef USE_DSHOT_TELEMETRY
 547     if (useDshotTelemetry) {
 548 #ifdef USE_DSHOT_TELEMETRY_STATS
 549         const timeMs_t currentTimeMs = millis();
 550 #endif
 551
 552 #ifdef USE_DSHOT_CACHE_MGMT
 553         for (int i = 0; i < usedMotorPorts; i++) {
 554             bbPort_t *bbPort = &bbPorts[i];
 555             SCB_InvalidateDCache_by_Addr((uint32_t *)bbPort->portInputBuffer, DSHOT_BB_PORT_IP_BUF_CACHE_ALIGN_BYTES);
 556         }
 557 #endif
 558         for (int motorIndex = 0; motorIndex < MAX_SUPPORTED_MOTORS && motorIndex < motorCount; motorIndex++) {
 559 #ifdef STM32F4
 560             uint32_t rawValue = decode_bb_bitband(
 561                 bbMotors[motorIndex].bbPort->portInputBuffer,
 562                 bbMotors[motorIndex].bbPort->portInputCount,
 563                 bbMotors[motorIndex].pinIndex);
 564 #else
 565             uint32_t rawValue = decode_bb(
 566                 bbMotors[motorIndex].bbPort->portInputBuffer,
 567                 bbMotors[motorIndex].bbPort->portInputCount,
 568                 bbMotors[motorIndex].pinIndex);
 569 #endif
 570             if (rawValue == DSHOT_TELEMETRY_NOEDGE) {
 571                 DEBUG_SET(DEBUG_DSHOT_TELEMETRY_COUNTS, 1, debug[1] + 1);
 572                 continue;
 573             }
 574             DEBUG_SET(DEBUG_DSHOT_TELEMETRY_COUNTS, 0, debug[0] + 1);
 575             dshotTelemetryState.readCount++;
 576
 577             if (rawValue != DSHOT_TELEMETRY_INVALID) {
 578                 // Check EDT enable or store raw value
 579                 if ((rawValue == 0x0E00) && (dshotCommandGetCurrent(motorIndex) == DSHOT_CMD_EXTENDED_TELEMETRY_ENABLE)) {
 580                     dshotTelemetryState.motorState[motorIndex].telemetryTypes = 1 << DSHOT_TELEMETRY_TYPE_STATE_EVENTS;
 581                 } else {
 582                     dshotTelemetryState.motorState[motorIndex].rawValue = rawValue;
 583                 }
 584             } else {
 585                 dshotTelemetryState.invalidPacketCount++;
 586             }
 587 #ifdef USE_DSHOT_TELEMETRY_STATS
 588             updateDshotTelemetryQuality(&dshotTelemetryQuality[motorIndex], rawValue != DSHOT_TELEMETRY_INVALID, currentTimeMs);
 589 #endif
 590         }
 591
 592         dshotTelemetryState.rawValueState = DSHOT_RAW_VALUE_STATE_NOT_PROCESSED;
 593     }
 594 #endif
 595
 596     return true;
 597 }
 598
 599 static void bbWriteInt(uint8_t motorIndex, uint16_t value)
 600 {
 601     bbMotor_t *const bbmotor = &bbMotors[motorIndex];
 602
 603     if (!bbmotor->configured) {
 604         return;
 605     }
 606
 607     // fetch requestTelemetry from motors. Needs to be refactored.
 608     motorDmaOutput_t * const motor = getMotorDmaOutput(motorIndex);
 609     bbmotor->protocolControl.requestTelemetry = motor->protocolControl.requestTelemetry;
 610     motor->protocolControl.requestTelemetry = false;
 611
 612     // If there is a command ready to go overwrite the value and send that instead
 613     if (dshotCommandIsProcessing()) {
 614         value = dshotCommandGetCurrent(motorIndex);
 615         if (value) {
 616             bbmotor->protocolControl.requestTelemetry = true;
 617         }
 618     }
 619
 620     bbmotor->protocolControl.value = value;
 621
 622     uint16_t packet = prepareDshotPacket(&bbmotor->protocolControl);
 623
 624     bbPort_t *bbPort = bbmotor->bbPort;
 625
 626 #ifdef USE_DSHOT_TELEMETRY
 627     if (useDshotTelemetry) {
 628         bbOutputDataSet(bbPort->portOutputBuffer, bbmotor->pinIndex, packet, DSHOT_BITBANG_INVERTED);
 629     } else
 630 #endif
 631     {
 632         bbOutputDataSet(bbPort->portOutputBuffer, bbmotor->pinIndex, packet, DSHOT_BITBANG_NONINVERTED);
 633     }
 634 }
 635
 636 static void bbWrite(uint8_t motorIndex, float value)
 637 {
 638     bbWriteInt(motorIndex, lrintf(value));
 639 }
 640
 641 static void bbUpdateComplete(void)
 642 {
 643     // If there is a dshot command loaded up, time it correctly with motor update
 644
 645     if (!dshotCommandQueueEmpty()) {
 646         if (!dshotCommandOutputIsEnabled(bbDevice.count)) {
 647             return;
 648         }
 649     }
 650
 651     for (int i = 0; i < usedMotorPorts; i++) {
 652         bbPort_t *bbPort = &bbPorts[i];
 653 #ifdef USE_DSHOT_CACHE_MGMT
 654         SCB_CleanDCache_by_Addr(bbPort->portOutputBuffer, MOTOR_DSHOT_BUF_CACHE_ALIGN_BYTES);
 655 #endif
 656
 657 #ifdef USE_DSHOT_TELEMETRY
 658         if (useDshotTelemetry) {
 659             if (bbPort->direction == DSHOT_BITBANG_DIRECTION_INPUT) {
 660                 bbPort->inputActive = false;
 661                 bbSwitchToOutput(bbPort);
 662             }
 663         } else
 664 #endif
 665         {
 666 #if defined(STM32G4)
 667             // Using circular mode resets the counter one short, so explicitly reload
 668             bbSwitchToOutput(bbPort);
 669 #endif
 670         }
 671
 672         bbDMA_Cmd(bbPort, ENABLE);
 673     }
 674
 675     lastSendUs = micros();
 676     for (int i = 0; i < usedMotorPacers; i++) {
 677         bbPacer_t *bbPacer = &bbPacers[i];
 678         bbTIM_DMACmd(bbPacer->tim, bbPacer->dmaSources, ENABLE);
 679     }
 680 }
 681
 682 static bool bbEnableMotors(void)
 683 {
 684     for (int i = 0; i < motorCount; i++) {
 685         if (bbMotors[i].configured) {
 686             IOConfigGPIO(bbMotors[i].io, bbMotors[i].iocfg);
 687         }
 688     }
 689     return true;
 690 }
 691
 692 static void bbDisableMotors(void)
 693 {
 694     return;
 695 }
 696
 697 static void bbShutdown(void)
 698 {
 699     return;
 700 }
 701
 702 static bool bbIsMotorEnabled(uint8_t index)
 703 {
 704     return bbMotors[index].enabled;
 705 }
 706
 707 static void bbPostInit(void)
 708 {
 709     bbFindPacerTimer();
 710
 711     for (int motorIndex = 0; motorIndex < MAX_SUPPORTED_MOTORS && motorIndex < motorCount; motorIndex++) {
 712
 713         if (!bbMotorConfig(bbMotors[motorIndex].io, motorIndex, motorPwmProtocol, bbMotors[motorIndex].output)) {
 714             return;
 715         }
 716
 717
 718         bbMotors[motorIndex].enabled = true;
 719
 720         // Fill in motors structure for 4way access (XXX Should be refactored)
 721
 722         motors[motorIndex].enabled = true;
 723     }
 724 }
 725
 726 static motorVTable_t bbVTable = {
 727     .postInit = bbPostInit,
 728     .enable = bbEnableMotors,
 729     .disable = bbDisableMotors,
 730     .isMotorEnabled = bbIsMotorEnabled,
 731     .telemetryWait = bbTelemetryWait,
 732     .decodeTelemetry = bbDecodeTelemetry,
 733     .updateInit = bbUpdateInit,
 734     .write = bbWrite,
 735     .writeInt = bbWriteInt,
 736     .updateComplete = bbUpdateComplete,
 737     .convertExternalToMotor = dshotConvertFromExternal,
 738     .convertMotorToExternal = dshotConvertToExternal,
 739     .shutdown = bbShutdown,
 740 };
 741
 742 dshotBitbangStatus_e dshotBitbangGetStatus(void)
 743 {
 744     return bbStatus;
 745 }
 746
 747 motorDevice_t *dshotBitbangDevInit(const motorDevConfig_t *motorConfig, uint8_t count)
 748 {
 749     dbgPinLo(0);
 750     dbgPinLo(1);
 751
 752     motorPwmProtocol = motorConfig->motorPwmProtocol;
 753     bbDevice.vTable = bbVTable;
 754     motorCount = count;
 755     bbStatus = DSHOT_BITBANG_STATUS_OK;
 756
 757 #ifdef USE_DSHOT_TELEMETRY
 758     useDshotTelemetry = motorConfig->useDshotTelemetry;
 759 #endif
 760
 761     memset(bbOutputBuffer, 0, sizeof(bbOutputBuffer));
 762
 763     for (int motorIndex = 0; motorIndex < MAX_SUPPORTED_MOTORS && motorIndex < motorCount; motorIndex++) {
 764         const unsigned reorderedMotorIndex = motorConfig->motorOutputReordering[motorIndex];
 765         const timerHardware_t *timerHardware = timerGetConfiguredByTag(motorConfig->ioTags[reorderedMotorIndex]);
 766         const IO_t io = IOGetByTag(motorConfig->ioTags[reorderedMotorIndex]);
 767
 768         uint8_t output = motorConfig->motorPwmInversion ?  timerHardware->output ^ TIMER_OUTPUT_INVERTED : timerHardware->output;
 769         bbPuPdMode = (output & TIMER_OUTPUT_INVERTED) ? BB_GPIO_PULLDOWN : BB_GPIO_PULLUP;
 770
 771 #ifdef USE_DSHOT_TELEMETRY
 772         if (useDshotTelemetry) {
 773             output ^= TIMER_OUTPUT_INVERTED;
 774         }
 775 #endif
 776
 777         if (!IOIsFreeOrPreinit(io)) {
 778             /* not enough motors initialised for the mixer or a break in the motors */
 779             bbDevice.vTable.write = motorWriteNull;
 780             bbDevice.vTable.decodeTelemetry = motorDecodeTelemetryNull;
 781             bbDevice.vTable.updateComplete = motorUpdateCompleteNull;
 782             bbStatus = DSHOT_BITBANG_STATUS_MOTOR_PIN_CONFLICT;
 783             return NULL;
 784         }
 785
 786         int pinIndex = IO_GPIOPinIdx(io);
 787
 788         bbMotors[motorIndex].pinIndex = pinIndex;
 789         bbMotors[motorIndex].io = io;
 790         bbMotors[motorIndex].output = output;
 791 #if defined(STM32F4)
 792         bbMotors[motorIndex].iocfg = IO_CONFIG(GPIO_Mode_OUT, GPIO_Speed_50MHz, GPIO_OType_PP, bbPuPdMode);
 793 #elif defined(STM32F7) || defined(STM32G4) || defined(STM32H7)
 794         bbMotors[motorIndex].iocfg = IO_CONFIG(GPIO_MODE_OUTPUT_PP, GPIO_SPEED_FREQ_LOW, bbPuPdMode);
 795 #endif
 796
 797         IOInit(io, OWNER_MOTOR, RESOURCE_INDEX(motorIndex));
 798         IOConfigGPIO(io, bbMotors[motorIndex].iocfg);
 799         if (output & TIMER_OUTPUT_INVERTED) {
 800             IOLo(io);
 801         } else {
 802             IOHi(io);
 803         }
 804
 805         // Fill in motors structure for 4way access (XXX Should be refactored)
 806         motors[motorIndex].io = bbMotors[motorIndex].io;
 807     }
 808
 809     return &bbDevice;
 810 }
 811
 812 #endif // USE_DSHOT_BB