You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

234 lines
8.5 KiB

2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
2 years ago
  1. #include "ztmc5130.hpp"
  2. #include "sdk/os/zos.hpp"
  3. #ifdef HAL_SPI_MODULE_ENABLED
  4. using namespace iflytop;
  5. /**
  6. * @brief ̬ĬϵTMC5130ò,ʹʱֻ޸ԼҪIJ
  7. *
  8. * ע:
  9. * 1. ÷ڲʹõһ̬ÿεø÷ʱصĶͬһĵַ??
  10. * 2. ÷صֵҪͷ??
  11. * @param config
  12. */
  13. #define PRV_FIELD_WRITE(address, mask, shift, value) (writeInt(address, FIELD_SET(readInt(address), mask, shift, value)))
  14. #define PRV_FIELD_READ(address, mask, shift) FIELD_GET(readInt(address), mask, shift)
  15. #define SET_PIN(pin, val) \
  16. if (pin) { \
  17. pin->setState(val); \
  18. }
  19. TMC5130::TMC5130(/* args */) {}
  20. void TMC5130::initialize(cfg_t *cfg) {
  21. m_cfg = *cfg;
  22. // m_channel = channel;
  23. // m_config = config;
  24. // m_port = config->m_port;
  25. m_registerAccessTable = &tmc5130_defaultRegisterAccess[0];
  26. m_defaultRegisterResetState = &tmc5130_defaultRegisterResetState[0];
  27. if (cfg->csgpio != PinNull) {
  28. m_csnpin = new ZGPIO();
  29. m_csnpin->initAsOutput(cfg->csgpio, ZGPIO::kMode_nopull, false, true);
  30. ZASSERT(m_csnpin);
  31. }
  32. if (cfg->ennPin != PinNull) {
  33. m_ennpin = new ZGPIO();
  34. m_ennpin->initAsOutput(cfg->ennPin, ZGPIO::kMode_nopull, false, true);
  35. ZASSERT(m_ennpin);
  36. }
  37. if (cfg->spi_mode_select != PinNull) {
  38. m_spi_mode_select_gpio = new ZGPIO();
  39. m_spi_mode_select_gpio->initAsOutput(cfg->spi_mode_select, ZGPIO::kMode_nopull, false, false);
  40. }
  41. m_hspi = cfg->spi;
  42. enableIC(false);
  43. // tmc5130_init(&m_TMC5130, channel, &m_tmc_api_config, &tmc5130_defaultRegisterResetState[0]);
  44. // tmc5130_setCallback(&m_TMC5130, pri_tmc4361A_callback);
  45. reset();
  46. writeInt(TMC5130_PWMCONF, 0x000500C8);
  47. // writeInt( TMC5130_GCONF, 0x00000004);
  48. writeInt(TMC5130_CHOPCONF, 0x000100c3);
  49. writeInt(TMC5130_IHOLD_IRUN, 0x00051A00);
  50. writeInt(TMC5130_PWMCONF, 0x000401c8);
  51. writeInt(TMC5130_XTARGET, 0);
  52. writeInt(TMC5130_XACTUAL, 0x00000000);
  53. writeInt(TMC5130_VACTUAL, 0x00000000);
  54. writeInt(TMC5130_VSTART, 5);
  55. writeInt(TMC5130_A1, 1000);
  56. writeInt(TMC5130_V1, 0);
  57. writeInt(TMC5130_D1, 1000);
  58. writeInt(TMC5130_VSTOP, 10);
  59. writeInt(TMC5130_TZEROWAIT, 1000);
  60. setAcceleration(100);
  61. setDeceleration(100);
  62. setIHOLD_IRUN(2, 10, 1);
  63. enableIC(true);
  64. }
  65. void TMC5130::enableIC(bool enable) {
  66. // m_port->TMC5130Port_setENNPinState(m_channel, !enable);
  67. SET_PIN(m_ennpin, !enable);
  68. }
  69. uint8_t TMC5130::reset() {
  70. stop();
  71. // m_port->TMC5130Port_setResetNPinState(m_channel, false);
  72. SET_PIN(m_csnpin, false);
  73. // m_port->TMC5130Port_sleepus(1000);
  74. zchip_clock_early_delayus(1000);
  75. // m_port->TMC5130Port_setResetNPinState(m_channel, true);
  76. SET_PIN(m_csnpin, true);
  77. for (uint32_t add = 0; add < TMC5130_REGISTER_COUNT; add++) {
  78. if (!TMC_IS_RESETTABLE(m_registerAccessTable[add])) {
  79. continue;
  80. }
  81. writeInt(add, m_defaultRegisterResetState[add]);
  82. }
  83. return 0;
  84. }
  85. int32_t TMC5130::getXACTUAL() { return to_user_pos(readInt(TMC5130_XACTUAL)); }
  86. void TMC5130::setXACTUAL(int32_t value) { writeInt(TMC5130_XACTUAL, to_motor_pos(value)); }
  87. int32_t TMC5130::getVACTUAL() { return to_user_pos(readInt(TMC5130_VACTUAL)); }
  88. void TMC5130::setAcceleration(float accelerationpps2) { writeInt(TMC5130_AMAX, (int32_t)(to_motor_acc(accelerationpps2))); } // �����������ٶ�
  89. void TMC5130::setDeceleration(float accelerationpps2) { writeInt(TMC5130_DMAX, (int32_t)(to_motor_acc(accelerationpps2))); } // �����������ٶ�
  90. void TMC5130::setMotorShaft(bool reverse) { PRV_FIELD_WRITE(TMC5130_GCONF, TMC5130_SHAFT_MASK, TMC5130_SHAFT_SHIFT, reverse); }
  91. void TMC5130::setIHOLD_IRUN(uint8_t ihold, uint8_t irun, uint16_t iholddelay) { writeInt(TMC5130_IHOLD_IRUN, (iholddelay << TMC5130_IHOLDDELAY_SHIFT) | (irun << TMC5130_IRUN_SHIFT) | (ihold << TMC5130_IHOLD_SHIFT)); }
  92. uint32_t TMC5130::readICVersion() {
  93. uint32_t chipID = PRV_FIELD_READ(TMC5130_IOIN, TMC5130_VERSION_MASK, TMC5130_VERSION_SHIFT);
  94. return chipID;
  95. }
  96. uint32_t TMC5130::getTMC5130_RAMPSTAT() { return readInt(TMC5130_RAMPSTAT); }
  97. Tmc5130RampStat TMC5130::getTMC5130_RAMPSTAT2() {
  98. uint32_t value = getTMC5130_RAMPSTAT();
  99. return Tmc5130RampStat(value);
  100. }
  101. void TMC5130::stop() { rotate(0); }
  102. void TMC5130::rotate(int32_t velocity) {
  103. // velocity *= m_scale;
  104. velocity = to_motor_vel(velocity);
  105. writeInt(TMC5130_VMAX, abs(velocity));
  106. writeInt(TMC5130_RAMPMODE, (velocity >= 0) ? TMC5130_MODE_VELPOS : TMC5130_MODE_VELNEG);
  107. }
  108. void TMC5130::right(int32_t velocity) { rotate(velocity); }
  109. void TMC5130::left(int32_t velocity) { rotate(-velocity); }
  110. void TMC5130::moveTo(int32_t position, uint32_t velocityMax) {
  111. // position *= m_scale;
  112. // velocityMax *= m_scale;
  113. position = to_motor_pos(position);
  114. velocityMax = to_motor_vel(velocityMax);
  115. // ZLOGI("TMC5130", "moveTo %d %d", position, velocityMax);
  116. writeInt(TMC5130_RAMPMODE, TMC5130_MODE_POSITION);
  117. writeInt(TMC5130_VMAX, velocityMax);
  118. writeInt(TMC5130_XTARGET, position);
  119. }
  120. void TMC5130::moveBy(int32_t relativePosition, uint32_t velocityMax) { // determine actual position and add numbers of ticks to move
  121. relativePosition += readInt(TMC5130_XACTUAL);
  122. moveTo(relativePosition, velocityMax);
  123. }
  124. uint32_t TMC5130::readXTARGET() { return readInt(TMC5130_XTARGET); }
  125. uint32_t TMC5130::haspassedms(uint32_t now, uint32_t last) {
  126. if (now >= last) {
  127. return now - last;
  128. } else {
  129. return 0xFFFFFFFF - last + now;
  130. }
  131. }
  132. bool TMC5130::isReachTarget() {
  133. /**
  134. * @brief
  135. */
  136. int mode = readInt(TMC5130_RAMPMODE);
  137. if (mode == TMC5130_MODE_POSITION) {
  138. uint32_t state = getTMC5130_RAMPSTAT();
  139. Tmc5130RampStat event = Tmc5130RampStat(state);
  140. return event.isSetted(Tmc5130RampStat::ktmc5130_rs_posreached);
  141. } else {
  142. uint32_t state = getTMC5130_RAMPSTAT();
  143. Tmc5130RampStat event = Tmc5130RampStat(state);
  144. return event.isSetted(Tmc5130RampStat::ktmc5130_rs_vzero) && event.isSetted(Tmc5130RampStat::ktmc5130_rs_velreached);
  145. }
  146. }
  147. void TMC5130::setScale(int32_t scale) { m_scale = scale; }
  148. void TMC5130::setScaleDenominator(int32_t scale) { m_scale_deceleration = scale; }
  149. /*******************************************************************************
  150. * basic *
  151. *******************************************************************************/
  152. // void TMC5130::writeSubRegister(uint8_t address, uint32_t mask, uint32_t shift, uint32_t value) {
  153. // CriticalContext cc;
  154. // writeInt(address, readInt(address) & ~mask | value << shift);
  155. // }
  156. void TMC5130::readWriteArray(uint8_t *data, size_t length) {
  157. CriticalContext cc;
  158. // m_port->TMC5130Port_readWriteArray(m_channel, data, length);
  159. // m_csnpin
  160. SET_PIN(m_csnpin, false);
  161. HAL_SPI_TransmitReceive(m_hspi, data, data, length, 100);
  162. SET_PIN(m_csnpin, true);
  163. }
  164. void TMC5130::writeDatagram(uint8_t address, uint8_t x1, uint8_t x2, uint8_t x3, uint8_t x4) {
  165. CriticalContext cc;
  166. uint8_t data[5] = {static_cast<uint8_t>(address | static_cast<uint8_t>(TMC5130_WRITE_BIT)), x1, x2, x3, x4};
  167. readWriteArray(&data[0], 5);
  168. int32_t value = ((uint32_t)x1 << 24) | ((uint32_t)x2 << 16) | (x3 << 8) | x4;
  169. // Write to the shadow register and mark the register dirty
  170. address = TMC_ADDRESS(address);
  171. m_shadowRegister[address] = value;
  172. }
  173. void TMC5130::writeInt(uint8_t address, int32_t value) {
  174. CriticalContext cc;
  175. writeDatagram(address, BYTE(value, 3), BYTE(value, 2), BYTE(value, 1), BYTE(value, 0));
  176. }
  177. int32_t TMC5130::readInt(uint8_t address) {
  178. CriticalContext cc;
  179. address = TMC_ADDRESS(address);
  180. // register not readable -> shadow register copy
  181. if (!TMC_IS_READABLE(tmc5130_defaultRegisterAccess[address])) return m_shadowRegister[address];
  182. uint8_t data[5] = {0, 0, 0, 0, 0};
  183. data[0] = address;
  184. readWriteArray(&data[0], 5);
  185. data[0] = address;
  186. readWriteArray(&data[0], 5);
  187. return ((uint32_t)data[1] << 24) | ((uint32_t)data[2] << 16) | (data[3] << 8) | data[4];
  188. }
  189. int32_t TMC5130::to_motor_acc(int32_t acc) { //
  190. int32_t val = acc / 60.0 * 51200;
  191. return val;
  192. }
  193. int32_t TMC5130::to_motor_vel(int32_t vel) { //
  194. int32_t val = vel / 60.0 * 51200;
  195. return val;
  196. } // rpm
  197. int32_t TMC5130::to_motor_pos(int32_t pos) { //
  198. int32_t val = pos * 1.0 / m_scale * m_scale_deceleration * 51200.0;
  199. return val;
  200. } //
  201. int32_t TMC5130::to_user_pos(int32_t pos) { //
  202. int32_t val = pos / 51200.0 * m_scale / m_scale_deceleration;
  203. return val;
  204. } //
  205. int32_t TMC5130::to_user_vel(int32_t vel) { //
  206. int32_t val = vel * 60.0 / 51200.0;
  207. return val;
  208. }
  209. #endif