rmt.c 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775
  1. // Copyright 2015-2016 Espressif Systems (Shanghai) PTE LTD
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. // http://www.apache.org/licenses/LICENSE-2.0
  7. //
  8. // Unless required by applicable law or agreed to in writing, software
  9. // distributed under the License is distributed on an "AS IS" BASIS,
  10. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  11. // See the License for the specific language governing permissions and
  12. // limitations under the License.
  13. #include <esp_types.h>
  14. #include <string.h>
  15. #include <stdlib.h>
  16. #include "freertos/FreeRTOS.h"
  17. #include "freertos/semphr.h"
  18. #include "freertos/xtensa_api.h"
  19. #include "freertos/ringbuf.h"
  20. #include "esp_intr.h"
  21. #include "esp_log.h"
  22. #include "esp_err.h"
  23. #include "esp_intr_alloc.h"
  24. #include "soc/gpio_sig_map.h"
  25. #include "soc/rmt_struct.h"
  26. #include "driver/periph_ctrl.h"
  27. #include "driver/rmt.h"
  28. #include <sys/lock.h>
  29. #define RMT_SOUCCE_CLK_APB (APB_CLK_FREQ) /*!< RMT source clock is APB_CLK */
  30. #define RMT_SOURCE_CLK_REF (1 * 1000000) /*!< not used yet */
  31. #define RMT_SOURCE_CLK(select) ((select == RMT_BASECLK_REF) ? (RMT_SOURCE_CLK_REF) : (RMT_SOUCCE_CLK_APB)) /*! RMT source clock frequency */
  32. #define RMT_CHANNEL_ERROR_STR "RMT CHANNEL ERR"
  33. #define RMT_ADDR_ERROR_STR "RMT ADDRESS ERR"
  34. #define RMT_MEM_CNT_ERROR_STR "RMT MEM BLOCK NUM ERR"
  35. #define RMT_CARRIER_ERROR_STR "RMT CARRIER LEVEL ERR"
  36. #define RMT_MEM_OWNER_ERROR_STR "RMT MEM OWNER_ERR"
  37. #define RMT_BASECLK_ERROR_STR "RMT BASECLK ERR"
  38. #define RMT_WR_MEM_OVF_ERROR_STR "RMT WR MEM OVERFLOW"
  39. #define RMT_GPIO_ERROR_STR "RMT GPIO ERROR"
  40. #define RMT_MODE_ERROR_STR "RMT MODE ERROR"
  41. #define RMT_CLK_DIV_ERROR_STR "RMT CLK DIV ERR"
  42. #define RMT_DRIVER_ERROR_STR "RMT DRIVER ERR"
  43. #define RMT_DRIVER_LENGTH_ERROR_STR "RMT PARAM LEN ERROR"
  44. static const char* RMT_TAG = "rmt";
  45. static uint8_t s_rmt_driver_channels; // Bitmask (bits 0-7) of installed drivers' channels
  46. static rmt_isr_handle_t s_rmt_driver_intr_handle;
  47. #define RMT_CHECK(a, str, ret_val) \
  48. if (!(a)) { \
  49. ESP_LOGE(RMT_TAG,"%s(%d): %s", __FUNCTION__, __LINE__, str); \
  50. return (ret_val); \
  51. }
  52. // Spinlock for protecting concurrent register-level access only
  53. static portMUX_TYPE rmt_spinlock = portMUX_INITIALIZER_UNLOCKED;
  54. // Mutex lock for protecting concurrent register/unregister of RMT channels' ISR
  55. static _lock_t rmt_driver_isr_lock;
  56. typedef struct {
  57. int tx_offset;
  58. int tx_len_rem;
  59. int tx_sub_len;
  60. rmt_channel_t channel;
  61. const rmt_item32_t* tx_data;
  62. xSemaphoreHandle tx_sem;
  63. RingbufHandle_t tx_buf;
  64. RingbufHandle_t rx_buf;
  65. } rmt_obj_t;
  66. rmt_obj_t* p_rmt_obj[RMT_CHANNEL_MAX] = {0};
  67. static void rmt_set_tx_wrap_en(rmt_channel_t channel, bool en)
  68. {
  69. portENTER_CRITICAL(&rmt_spinlock);
  70. RMT.apb_conf.mem_tx_wrap_en = en;
  71. portEXIT_CRITICAL(&rmt_spinlock);
  72. }
  73. static void rmt_set_data_mode(rmt_data_mode_t data_mode)
  74. {
  75. portENTER_CRITICAL(&rmt_spinlock);
  76. RMT.apb_conf.fifo_mask = data_mode;
  77. portEXIT_CRITICAL(&rmt_spinlock);
  78. }
  79. esp_err_t rmt_set_clk_div(rmt_channel_t channel, uint8_t div_cnt)
  80. {
  81. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  82. RMT.conf_ch[channel].conf0.div_cnt = div_cnt;
  83. return ESP_OK;
  84. }
  85. esp_err_t rmt_get_clk_div(rmt_channel_t channel, uint8_t* div_cnt)
  86. {
  87. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  88. RMT_CHECK(div_cnt != NULL, RMT_ADDR_ERROR_STR, ESP_ERR_INVALID_ARG);
  89. *div_cnt = RMT.conf_ch[channel].conf0.div_cnt;
  90. return ESP_OK;
  91. }
  92. esp_err_t rmt_set_rx_idle_thresh(rmt_channel_t channel, uint16_t thresh)
  93. {
  94. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  95. RMT.conf_ch[channel].conf0.idle_thres = thresh;
  96. return ESP_OK;
  97. }
  98. esp_err_t rmt_get_rx_idle_thresh(rmt_channel_t channel, uint16_t *thresh)
  99. {
  100. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  101. RMT_CHECK(thresh != NULL, RMT_ADDR_ERROR_STR, ESP_ERR_INVALID_ARG);
  102. *thresh = RMT.conf_ch[channel].conf0.idle_thres;
  103. return ESP_OK;
  104. }
  105. esp_err_t rmt_set_mem_block_num(rmt_channel_t channel, uint8_t rmt_mem_num)
  106. {
  107. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  108. RMT_CHECK(rmt_mem_num < 16, RMT_MEM_CNT_ERROR_STR, ESP_ERR_INVALID_ARG);
  109. RMT.conf_ch[channel].conf0.mem_size = rmt_mem_num;
  110. return ESP_OK;
  111. }
  112. esp_err_t rmt_get_mem_block_num(rmt_channel_t channel, uint8_t* rmt_mem_num)
  113. {
  114. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  115. RMT_CHECK(rmt_mem_num != NULL, RMT_ADDR_ERROR_STR, ESP_ERR_INVALID_ARG);
  116. *rmt_mem_num = RMT.conf_ch[channel].conf0.mem_size;
  117. return ESP_OK;
  118. }
  119. esp_err_t rmt_set_tx_carrier(rmt_channel_t channel, bool carrier_en, uint16_t high_level, uint16_t low_level,
  120. rmt_carrier_level_t carrier_level)
  121. {
  122. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  123. RMT_CHECK(carrier_level < RMT_CARRIER_LEVEL_MAX, RMT_CARRIER_ERROR_STR, ESP_ERR_INVALID_ARG);
  124. RMT.carrier_duty_ch[channel].high = high_level;
  125. RMT.carrier_duty_ch[channel].low = low_level;
  126. RMT.conf_ch[channel].conf0.carrier_out_lv = carrier_level;
  127. RMT.conf_ch[channel].conf0.carrier_en = carrier_en;
  128. return ESP_OK;
  129. }
  130. esp_err_t rmt_set_mem_pd(rmt_channel_t channel, bool pd_en)
  131. {
  132. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  133. RMT.conf_ch[channel].conf0.mem_pd = pd_en;
  134. return ESP_OK;
  135. }
  136. esp_err_t rmt_get_mem_pd(rmt_channel_t channel, bool* pd_en)
  137. {
  138. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  139. *pd_en = (bool) RMT.conf_ch[channel].conf0.mem_pd;
  140. return ESP_OK;
  141. }
  142. esp_err_t rmt_tx_start(rmt_channel_t channel, bool tx_idx_rst)
  143. {
  144. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  145. portENTER_CRITICAL(&rmt_spinlock);
  146. if(tx_idx_rst) {
  147. RMT.conf_ch[channel].conf1.mem_rd_rst = 1;
  148. }
  149. RMT.conf_ch[channel].conf1.mem_owner = RMT_MEM_OWNER_TX;
  150. RMT.conf_ch[channel].conf1.tx_start = 1;
  151. portEXIT_CRITICAL(&rmt_spinlock);
  152. return ESP_OK;
  153. }
  154. esp_err_t rmt_tx_stop(rmt_channel_t channel)
  155. {
  156. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  157. portENTER_CRITICAL(&rmt_spinlock);
  158. RMT.conf_ch[channel].conf1.tx_start = 0;
  159. portEXIT_CRITICAL(&rmt_spinlock);
  160. return ESP_OK;
  161. }
  162. esp_err_t rmt_rx_start(rmt_channel_t channel, bool rx_idx_rst)
  163. {
  164. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  165. portENTER_CRITICAL(&rmt_spinlock);
  166. if(rx_idx_rst) {
  167. RMT.conf_ch[channel].conf1.mem_wr_rst = 1;
  168. }
  169. RMT.conf_ch[channel].conf1.rx_en = 0;
  170. RMT.conf_ch[channel].conf1.mem_owner = RMT_MEM_OWNER_RX;
  171. RMT.conf_ch[channel].conf1.rx_en = 1;
  172. portEXIT_CRITICAL(&rmt_spinlock);
  173. return ESP_OK;
  174. }
  175. esp_err_t rmt_rx_stop(rmt_channel_t channel)
  176. {
  177. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  178. portENTER_CRITICAL(&rmt_spinlock);
  179. RMT.conf_ch[channel].conf1.rx_en = 0;
  180. portEXIT_CRITICAL(&rmt_spinlock);
  181. return ESP_OK;
  182. }
  183. esp_err_t rmt_memory_rw_rst(rmt_channel_t channel)
  184. {
  185. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  186. portENTER_CRITICAL(&rmt_spinlock);
  187. RMT.conf_ch[channel].conf1.mem_rd_rst = 1;
  188. RMT.conf_ch[channel].conf1.mem_wr_rst = 1;
  189. portEXIT_CRITICAL(&rmt_spinlock);
  190. return ESP_OK;
  191. }
  192. esp_err_t rmt_set_memory_owner(rmt_channel_t channel, rmt_mem_owner_t owner)
  193. {
  194. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  195. RMT_CHECK(owner < RMT_MEM_OWNER_MAX, RMT_MEM_OWNER_ERROR_STR, ESP_ERR_INVALID_ARG);
  196. portENTER_CRITICAL(&rmt_spinlock);
  197. RMT.conf_ch[channel].conf1.mem_owner = owner;
  198. portEXIT_CRITICAL(&rmt_spinlock);
  199. return ESP_OK;
  200. }
  201. esp_err_t rmt_get_memory_owner(rmt_channel_t channel, rmt_mem_owner_t* owner)
  202. {
  203. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  204. RMT_CHECK(owner != NULL, RMT_MEM_OWNER_ERROR_STR, ESP_ERR_INVALID_ARG);
  205. *owner = (rmt_mem_owner_t) RMT.conf_ch[channel].conf1.mem_owner;
  206. return ESP_OK;
  207. }
  208. esp_err_t rmt_set_tx_loop_mode(rmt_channel_t channel, bool loop_en)
  209. {
  210. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  211. portENTER_CRITICAL(&rmt_spinlock);
  212. RMT.conf_ch[channel].conf1.tx_conti_mode = loop_en;
  213. portEXIT_CRITICAL(&rmt_spinlock);
  214. return ESP_OK;
  215. }
  216. esp_err_t rmt_get_tx_loop_mode(rmt_channel_t channel, bool* loop_en)
  217. {
  218. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  219. *loop_en = (bool) RMT.conf_ch[channel].conf1.tx_conti_mode;
  220. return ESP_OK;
  221. }
  222. esp_err_t rmt_set_rx_filter(rmt_channel_t channel, bool rx_filter_en, uint8_t thresh)
  223. {
  224. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  225. portENTER_CRITICAL(&rmt_spinlock);
  226. RMT.conf_ch[channel].conf1.rx_filter_en = rx_filter_en;
  227. RMT.conf_ch[channel].conf1.rx_filter_thres = thresh;
  228. portEXIT_CRITICAL(&rmt_spinlock);
  229. return ESP_OK;
  230. }
  231. esp_err_t rmt_set_source_clk(rmt_channel_t channel, rmt_source_clk_t base_clk)
  232. {
  233. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  234. RMT_CHECK(base_clk < RMT_BASECLK_MAX, RMT_BASECLK_ERROR_STR, ESP_ERR_INVALID_ARG);
  235. portENTER_CRITICAL(&rmt_spinlock);
  236. RMT.conf_ch[channel].conf1.ref_always_on = base_clk;
  237. portEXIT_CRITICAL(&rmt_spinlock);
  238. return ESP_OK;
  239. }
  240. esp_err_t rmt_get_source_clk(rmt_channel_t channel, rmt_source_clk_t* src_clk)
  241. {
  242. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  243. *src_clk = (rmt_source_clk_t) (RMT.conf_ch[channel].conf1.ref_always_on);
  244. return ESP_OK;
  245. }
  246. esp_err_t rmt_set_idle_level(rmt_channel_t channel, bool idle_out_en, rmt_idle_level_t level)
  247. {
  248. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  249. RMT_CHECK(level < RMT_IDLE_LEVEL_MAX, "RMT IDLE LEVEL ERR", ESP_ERR_INVALID_ARG);
  250. portENTER_CRITICAL(&rmt_spinlock);
  251. RMT.conf_ch[channel].conf1.idle_out_en = idle_out_en;
  252. RMT.conf_ch[channel].conf1.idle_out_lv = level;
  253. portEXIT_CRITICAL(&rmt_spinlock);
  254. return ESP_OK;
  255. }
  256. esp_err_t rmt_get_status(rmt_channel_t channel, uint32_t* status)
  257. {
  258. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  259. *status = RMT.status_ch[channel];
  260. return ESP_OK;
  261. }
  262. rmt_data_mode_t rmt_get_data_mode()
  263. {
  264. return (rmt_data_mode_t) (RMT.apb_conf.fifo_mask);
  265. }
  266. void rmt_set_intr_enable_mask(uint32_t mask)
  267. {
  268. portENTER_CRITICAL(&rmt_spinlock);
  269. RMT.int_ena.val |= mask;
  270. portEXIT_CRITICAL(&rmt_spinlock);
  271. }
  272. void rmt_clr_intr_enable_mask(uint32_t mask)
  273. {
  274. portENTER_CRITICAL(&rmt_spinlock);
  275. RMT.int_ena.val &= (~mask);
  276. portEXIT_CRITICAL(&rmt_spinlock);
  277. }
  278. esp_err_t rmt_set_rx_intr_en(rmt_channel_t channel, bool en)
  279. {
  280. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  281. if(en) {
  282. rmt_set_intr_enable_mask(BIT(channel * 3 + 1));
  283. } else {
  284. rmt_clr_intr_enable_mask(BIT(channel * 3 + 1));
  285. }
  286. return ESP_OK;
  287. }
  288. esp_err_t rmt_set_err_intr_en(rmt_channel_t channel, bool en)
  289. {
  290. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  291. if(en) {
  292. rmt_set_intr_enable_mask(BIT(channel * 3 + 2));
  293. } else {
  294. rmt_clr_intr_enable_mask(BIT(channel * 3 + 2));
  295. }
  296. return ESP_OK;
  297. }
  298. esp_err_t rmt_set_tx_intr_en(rmt_channel_t channel, bool en)
  299. {
  300. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  301. if(en) {
  302. rmt_set_intr_enable_mask(BIT(channel * 3));
  303. } else {
  304. rmt_clr_intr_enable_mask(BIT(channel * 3));
  305. }
  306. return ESP_OK;
  307. }
  308. esp_err_t rmt_set_tx_thr_intr_en(rmt_channel_t channel, bool en, uint16_t evt_thresh)
  309. {
  310. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  311. if(en) {
  312. RMT_CHECK(evt_thresh <= 256, "RMT EVT THRESH ERR", ESP_ERR_INVALID_ARG);
  313. portENTER_CRITICAL(&rmt_spinlock);
  314. RMT.tx_lim_ch[channel].limit = evt_thresh;
  315. portEXIT_CRITICAL(&rmt_spinlock);
  316. rmt_set_tx_wrap_en(channel, true);
  317. rmt_set_intr_enable_mask(BIT(channel + 24));
  318. } else {
  319. rmt_clr_intr_enable_mask(BIT(channel + 24));
  320. }
  321. return ESP_OK;
  322. }
  323. esp_err_t rmt_set_pin(rmt_channel_t channel, rmt_mode_t mode, gpio_num_t gpio_num)
  324. {
  325. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  326. RMT_CHECK(mode < RMT_MODE_MAX, RMT_MODE_ERROR_STR, ESP_ERR_INVALID_ARG);
  327. RMT_CHECK(((GPIO_IS_VALID_GPIO(gpio_num) && (mode == RMT_MODE_RX)) || (GPIO_IS_VALID_OUTPUT_GPIO(gpio_num) && (mode == RMT_MODE_TX))),
  328. RMT_GPIO_ERROR_STR, ESP_ERR_INVALID_ARG);
  329. PIN_FUNC_SELECT(GPIO_PIN_MUX_REG[gpio_num], 2);
  330. if(mode == RMT_MODE_TX) {
  331. gpio_set_direction(gpio_num, GPIO_MODE_OUTPUT);
  332. gpio_matrix_out(gpio_num, RMT_SIG_OUT0_IDX + channel, 0, 0);
  333. } else {
  334. gpio_set_direction(gpio_num, GPIO_MODE_INPUT);
  335. gpio_matrix_in(gpio_num, RMT_SIG_IN0_IDX + channel, 0);
  336. }
  337. return ESP_OK;
  338. }
  339. esp_err_t rmt_config(const rmt_config_t* rmt_param)
  340. {
  341. uint8_t mode = rmt_param->rmt_mode;
  342. uint8_t channel = rmt_param->channel;
  343. uint8_t gpio_num = rmt_param->gpio_num;
  344. uint8_t mem_cnt = rmt_param->mem_block_num;
  345. int clk_div = rmt_param->clk_div;
  346. uint32_t carrier_freq_hz = rmt_param->tx_config.carrier_freq_hz;
  347. bool carrier_en = rmt_param->tx_config.carrier_en;
  348. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  349. RMT_CHECK(GPIO_IS_VALID_GPIO(gpio_num), RMT_GPIO_ERROR_STR, ESP_ERR_INVALID_ARG);
  350. RMT_CHECK((mem_cnt + channel <= 8 && mem_cnt > 0), RMT_MEM_CNT_ERROR_STR, ESP_ERR_INVALID_ARG);
  351. RMT_CHECK((clk_div > 0), RMT_CLK_DIV_ERROR_STR, ESP_ERR_INVALID_ARG);
  352. if (mode == RMT_MODE_TX) {
  353. RMT_CHECK((!carrier_en || carrier_freq_hz > 0), "RMT carrier frequency can't be zero", ESP_ERR_INVALID_ARG);
  354. }
  355. periph_module_enable(PERIPH_RMT_MODULE);
  356. RMT.conf_ch[channel].conf0.div_cnt = clk_div;
  357. /*Visit data use memory not FIFO*/
  358. rmt_set_data_mode(RMT_DATA_MODE_MEM);
  359. /*Reset tx/rx memory index */
  360. portENTER_CRITICAL(&rmt_spinlock);
  361. RMT.conf_ch[channel].conf1.mem_rd_rst = 1;
  362. RMT.conf_ch[channel].conf1.mem_wr_rst = 1;
  363. portEXIT_CRITICAL(&rmt_spinlock);
  364. if(mode == RMT_MODE_TX) {
  365. uint32_t rmt_source_clk_hz = 0;
  366. uint16_t carrier_duty_percent = rmt_param->tx_config.carrier_duty_percent;
  367. uint8_t carrier_level = rmt_param->tx_config.carrier_level;
  368. uint8_t idle_level = rmt_param->tx_config.idle_level;
  369. portENTER_CRITICAL(&rmt_spinlock);
  370. RMT.conf_ch[channel].conf1.tx_conti_mode = rmt_param->tx_config.loop_en;
  371. /*Memory set block number*/
  372. RMT.conf_ch[channel].conf0.mem_size = mem_cnt;
  373. RMT.conf_ch[channel].conf1.mem_owner = RMT_MEM_OWNER_TX;
  374. /*We use APB clock in this version, which is 80Mhz, later we will release system reference clock*/
  375. RMT.conf_ch[channel].conf1.ref_always_on = RMT_BASECLK_APB;
  376. rmt_source_clk_hz = RMT_SOURCE_CLK(RMT_BASECLK_APB);
  377. /*Set idle level */
  378. RMT.conf_ch[channel].conf1.idle_out_en = rmt_param->tx_config.idle_output_en;
  379. RMT.conf_ch[channel].conf1.idle_out_lv = idle_level;
  380. /*Set carrier*/
  381. RMT.conf_ch[channel].conf0.carrier_en = carrier_en;
  382. if (carrier_en) {
  383. uint32_t duty_div, duty_h, duty_l;
  384. duty_div = rmt_source_clk_hz / carrier_freq_hz;
  385. duty_h = duty_div * carrier_duty_percent / 100;
  386. duty_l = duty_div - duty_h;
  387. RMT.conf_ch[channel].conf0.carrier_out_lv = carrier_level;
  388. RMT.carrier_duty_ch[channel].high = duty_h;
  389. RMT.carrier_duty_ch[channel].low = duty_l;
  390. } else {
  391. RMT.conf_ch[channel].conf0.carrier_out_lv = 0;
  392. RMT.carrier_duty_ch[channel].high = 0;
  393. RMT.carrier_duty_ch[channel].low = 0;
  394. }
  395. portEXIT_CRITICAL(&rmt_spinlock);
  396. ESP_LOGD(RMT_TAG, "Rmt Tx Channel %u|Gpio %u|Sclk_Hz %u|Div %u|Carrier_Hz %u|Duty %u",
  397. channel, gpio_num, rmt_source_clk_hz, clk_div, carrier_freq_hz, carrier_duty_percent);
  398. }
  399. else if(RMT_MODE_RX == mode) {
  400. uint8_t filter_cnt = rmt_param->rx_config.filter_ticks_thresh;
  401. uint16_t threshold = rmt_param->rx_config.idle_threshold;
  402. portENTER_CRITICAL(&rmt_spinlock);
  403. /*clock init*/
  404. RMT.conf_ch[channel].conf1.ref_always_on = RMT_BASECLK_APB;
  405. uint32_t rmt_source_clk_hz = RMT_SOURCE_CLK(RMT_BASECLK_APB);
  406. /*memory set block number and owner*/
  407. RMT.conf_ch[channel].conf0.mem_size = mem_cnt;
  408. RMT.conf_ch[channel].conf1.mem_owner = RMT_MEM_OWNER_RX;
  409. /*Set idle threshold*/
  410. RMT.conf_ch[channel].conf0.idle_thres = threshold;
  411. /* Set RX filter */
  412. RMT.conf_ch[channel].conf1.rx_filter_thres = filter_cnt;
  413. RMT.conf_ch[channel].conf1.rx_filter_en = rmt_param->rx_config.filter_en;
  414. portEXIT_CRITICAL(&rmt_spinlock);
  415. ESP_LOGD(RMT_TAG, "Rmt Rx Channel %u|Gpio %u|Sclk_Hz %u|Div %u|Thresold %u|Filter %u",
  416. channel, gpio_num, rmt_source_clk_hz, clk_div, threshold, filter_cnt);
  417. }
  418. rmt_set_pin(channel, mode, gpio_num);
  419. return ESP_OK;
  420. }
  421. static void IRAM_ATTR rmt_fill_memory(rmt_channel_t channel, const rmt_item32_t* item, uint16_t item_num, uint16_t mem_offset)
  422. {
  423. portENTER_CRITICAL(&rmt_spinlock);
  424. RMT.apb_conf.fifo_mask = RMT_DATA_MODE_MEM;
  425. portEXIT_CRITICAL(&rmt_spinlock);
  426. int i;
  427. for(i = 0; i < item_num; i++) {
  428. RMTMEM.chan[channel].data32[i + mem_offset].val = item[i].val;
  429. }
  430. }
  431. esp_err_t rmt_fill_tx_items(rmt_channel_t channel, const rmt_item32_t* item, uint16_t item_num, uint16_t mem_offset)
  432. {
  433. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, (0));
  434. RMT_CHECK((item != NULL), RMT_ADDR_ERROR_STR, ESP_ERR_INVALID_ARG);
  435. RMT_CHECK((item_num > 0), RMT_DRIVER_LENGTH_ERROR_STR, ESP_ERR_INVALID_ARG);
  436. /*Each block has 64 x 32 bits of data*/
  437. uint8_t mem_cnt = RMT.conf_ch[channel].conf0.mem_size;
  438. RMT_CHECK((mem_cnt * RMT_MEM_ITEM_NUM >= item_num), RMT_WR_MEM_OVF_ERROR_STR, ESP_ERR_INVALID_ARG);
  439. rmt_fill_memory(channel, item, item_num, mem_offset);
  440. return ESP_OK;
  441. }
  442. esp_err_t rmt_isr_register(void (*fn)(void*), void * arg, int intr_alloc_flags, rmt_isr_handle_t *handle)
  443. {
  444. RMT_CHECK((fn != NULL), RMT_ADDR_ERROR_STR, ESP_ERR_INVALID_ARG);
  445. RMT_CHECK(s_rmt_driver_channels == 0, "RMT driver installed, can not install generic ISR handler", ESP_FAIL);
  446. return esp_intr_alloc(ETS_RMT_INTR_SOURCE, intr_alloc_flags, fn, arg, handle);
  447. }
  448. esp_err_t rmt_isr_deregister(rmt_isr_handle_t handle)
  449. {
  450. return esp_intr_free(handle);
  451. }
  452. static int IRAM_ATTR rmt_get_mem_len(rmt_channel_t channel)
  453. {
  454. int block_num = RMT.conf_ch[channel].conf0.mem_size;
  455. int item_block_len = block_num * RMT_MEM_ITEM_NUM;
  456. volatile rmt_item32_t* data = RMTMEM.chan[channel].data32;
  457. int idx;
  458. for(idx = 0; idx < item_block_len; idx++) {
  459. if(data[idx].duration0 == 0) {
  460. return idx;
  461. } else if(data[idx].duration1 == 0) {
  462. return idx + 1;
  463. }
  464. }
  465. return idx;
  466. }
  467. static void IRAM_ATTR rmt_driver_isr_default(void* arg)
  468. {
  469. uint32_t intr_st = RMT.int_st.val;
  470. uint32_t i = 0;
  471. uint8_t channel;
  472. portBASE_TYPE HPTaskAwoken = 0;
  473. for(i = 0; i < 32; i++) {
  474. if(i < 24) {
  475. if(intr_st & BIT(i)) {
  476. channel = i / 3;
  477. rmt_obj_t* p_rmt = p_rmt_obj[channel];
  478. switch(i % 3) {
  479. //TX END
  480. case 0:
  481. ESP_EARLY_LOGD(RMT_TAG, "RMT INTR : TX END");
  482. xSemaphoreGiveFromISR(p_rmt->tx_sem, &HPTaskAwoken);
  483. if(HPTaskAwoken == pdTRUE) {
  484. portYIELD_FROM_ISR();
  485. }
  486. p_rmt->tx_data = NULL;
  487. p_rmt->tx_len_rem = 0;
  488. p_rmt->tx_offset = 0;
  489. p_rmt->tx_sub_len = 0;
  490. break;
  491. //RX_END
  492. case 1:
  493. ESP_EARLY_LOGD(RMT_TAG, "RMT INTR : RX END");
  494. RMT.conf_ch[channel].conf1.rx_en = 0;
  495. int item_len = rmt_get_mem_len(channel);
  496. //change memory owner to protect data.
  497. RMT.conf_ch[channel].conf1.mem_owner = RMT_MEM_OWNER_TX;
  498. if(p_rmt->rx_buf) {
  499. BaseType_t res = xRingbufferSendFromISR(p_rmt->rx_buf, (void*) RMTMEM.chan[channel].data32, item_len * 4, &HPTaskAwoken);
  500. if(res == pdFALSE) {
  501. ESP_EARLY_LOGE(RMT_TAG, "RMT RX BUFFER FULL");
  502. } else {
  503. }
  504. if(HPTaskAwoken == pdTRUE) {
  505. portYIELD_FROM_ISR();
  506. }
  507. } else {
  508. ESP_EARLY_LOGE(RMT_TAG, "RMT RX BUFFER ERROR\n");
  509. }
  510. RMT.conf_ch[channel].conf1.mem_wr_rst = 1;
  511. RMT.conf_ch[channel].conf1.mem_owner = RMT_MEM_OWNER_RX;
  512. RMT.conf_ch[channel].conf1.rx_en = 1;
  513. break;
  514. //ERR
  515. case 2:
  516. ESP_EARLY_LOGE(RMT_TAG, "RMT[%d] ERR", channel);
  517. ESP_EARLY_LOGE(RMT_TAG, "status: 0x%08x", RMT.status_ch[channel]);
  518. RMT.int_ena.val &= (~(BIT(i)));
  519. break;
  520. default:
  521. break;
  522. }
  523. RMT.int_clr.val = BIT(i);
  524. }
  525. } else {
  526. if(intr_st & (BIT(i))) {
  527. channel = i - 24;
  528. rmt_obj_t* p_rmt = p_rmt_obj[channel];
  529. RMT.int_clr.val = BIT(i);
  530. ESP_EARLY_LOGD(RMT_TAG, "RMT CH[%d]: EVT INTR", channel);
  531. if(p_rmt->tx_data == NULL) {
  532. //skip
  533. } else {
  534. const rmt_item32_t* pdata = p_rmt->tx_data;
  535. int len_rem = p_rmt->tx_len_rem;
  536. if(len_rem >= p_rmt->tx_sub_len) {
  537. rmt_fill_memory(channel, pdata, p_rmt->tx_sub_len, p_rmt->tx_offset);
  538. p_rmt->tx_data += p_rmt->tx_sub_len;
  539. p_rmt->tx_len_rem -= p_rmt->tx_sub_len;
  540. } else if(len_rem == 0) {
  541. RMTMEM.chan[channel].data32[p_rmt->tx_offset].val = 0;
  542. } else {
  543. rmt_fill_memory(channel, pdata, len_rem, p_rmt->tx_offset);
  544. RMTMEM.chan[channel].data32[p_rmt->tx_offset + len_rem].val = 0;
  545. p_rmt->tx_data += len_rem;
  546. p_rmt->tx_len_rem -= len_rem;
  547. }
  548. if(p_rmt->tx_offset == 0) {
  549. p_rmt->tx_offset = p_rmt->tx_sub_len;
  550. } else {
  551. p_rmt->tx_offset = 0;
  552. }
  553. }
  554. }
  555. }
  556. }
  557. }
  558. esp_err_t rmt_driver_uninstall(rmt_channel_t channel)
  559. {
  560. esp_err_t err = ESP_OK;
  561. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  562. RMT_CHECK((s_rmt_driver_channels & BIT(channel)) != 0, "No RMT driver for this channel", ESP_ERR_INVALID_STATE);
  563. if(p_rmt_obj[channel] == NULL) {
  564. return ESP_OK;
  565. }
  566. xSemaphoreTake(p_rmt_obj[channel]->tx_sem, portMAX_DELAY);
  567. rmt_set_rx_intr_en(channel, 0);
  568. rmt_set_err_intr_en(channel, 0);
  569. rmt_set_tx_intr_en(channel, 0);
  570. rmt_set_tx_thr_intr_en(channel, 0, 0xffff);
  571. _lock_acquire_recursive(&rmt_driver_isr_lock);
  572. s_rmt_driver_channels &= ~BIT(channel);
  573. if (s_rmt_driver_channels == 0) { // all channels have driver disabled
  574. err = rmt_isr_deregister(s_rmt_driver_intr_handle);
  575. s_rmt_driver_intr_handle = NULL;
  576. }
  577. _lock_release_recursive(&rmt_driver_isr_lock);
  578. if (err != ESP_OK) {
  579. return err;
  580. }
  581. if(p_rmt_obj[channel]->tx_sem) {
  582. vSemaphoreDelete(p_rmt_obj[channel]->tx_sem);
  583. p_rmt_obj[channel]->tx_sem = NULL;
  584. }
  585. if(p_rmt_obj[channel]->rx_buf) {
  586. vRingbufferDelete(p_rmt_obj[channel]->rx_buf);
  587. p_rmt_obj[channel]->rx_buf = NULL;
  588. }
  589. free(p_rmt_obj[channel]);
  590. p_rmt_obj[channel] = NULL;
  591. return ESP_OK;
  592. }
  593. esp_err_t rmt_driver_install(rmt_channel_t channel, size_t rx_buf_size, int intr_alloc_flags)
  594. {
  595. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  596. RMT_CHECK((s_rmt_driver_channels & BIT(channel)) == 0, "RMT driver already installed for channel", ESP_ERR_INVALID_STATE);
  597. esp_err_t err = ESP_OK;
  598. if(p_rmt_obj[channel] != NULL) {
  599. ESP_LOGD(RMT_TAG, "RMT driver already installed");
  600. return ESP_ERR_INVALID_STATE;
  601. }
  602. p_rmt_obj[channel] = (rmt_obj_t*) malloc(sizeof(rmt_obj_t));
  603. if(p_rmt_obj[channel] == NULL) {
  604. ESP_LOGE(RMT_TAG, "RMT driver malloc error");
  605. return ESP_ERR_NO_MEM;
  606. }
  607. memset(p_rmt_obj[channel], 0, sizeof(rmt_obj_t));
  608. p_rmt_obj[channel]->tx_len_rem = 0;
  609. p_rmt_obj[channel]->tx_data = NULL;
  610. p_rmt_obj[channel]->channel = channel;
  611. p_rmt_obj[channel]->tx_offset = 0;
  612. p_rmt_obj[channel]->tx_sub_len = 0;
  613. if(p_rmt_obj[channel]->tx_sem == NULL) {
  614. p_rmt_obj[channel]->tx_sem = xSemaphoreCreateBinary();
  615. xSemaphoreGive(p_rmt_obj[channel]->tx_sem);
  616. }
  617. if(p_rmt_obj[channel]->rx_buf == NULL && rx_buf_size > 0) {
  618. p_rmt_obj[channel]->rx_buf = xRingbufferCreate(rx_buf_size, RINGBUF_TYPE_NOSPLIT);
  619. rmt_set_rx_intr_en(channel, 1);
  620. rmt_set_err_intr_en(channel, 1);
  621. }
  622. _lock_acquire_recursive(&rmt_driver_isr_lock);
  623. if(s_rmt_driver_channels == 0) { // first RMT channel using driver
  624. err = rmt_isr_register(rmt_driver_isr_default, NULL, intr_alloc_flags, &s_rmt_driver_intr_handle);
  625. }
  626. if (err == ESP_OK) {
  627. s_rmt_driver_channels |= BIT(channel);
  628. rmt_set_tx_intr_en(channel, 1);
  629. }
  630. _lock_release_recursive(&rmt_driver_isr_lock);
  631. return err;
  632. }
  633. esp_err_t rmt_write_items(rmt_channel_t channel, const rmt_item32_t* rmt_item, int item_num, bool wait_tx_done)
  634. {
  635. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  636. RMT_CHECK(p_rmt_obj[channel] != NULL, RMT_DRIVER_ERROR_STR, ESP_FAIL);
  637. RMT_CHECK(rmt_item != NULL, RMT_ADDR_ERROR_STR, ESP_FAIL);
  638. RMT_CHECK(item_num > 0, RMT_DRIVER_LENGTH_ERROR_STR, ESP_ERR_INVALID_ARG);
  639. rmt_obj_t* p_rmt = p_rmt_obj[channel];
  640. int block_num = RMT.conf_ch[channel].conf0.mem_size;
  641. int item_block_len = block_num * RMT_MEM_ITEM_NUM;
  642. int item_sub_len = block_num * RMT_MEM_ITEM_NUM / 2;
  643. int len_rem = item_num;
  644. xSemaphoreTake(p_rmt->tx_sem, portMAX_DELAY);
  645. // fill the memory block first
  646. if(item_num >= item_block_len) {
  647. rmt_fill_memory(channel, rmt_item, item_block_len, 0);
  648. RMT.tx_lim_ch[channel].limit = item_sub_len;
  649. RMT.apb_conf.mem_tx_wrap_en = 1;
  650. len_rem -= item_block_len;
  651. RMT.conf_ch[channel].conf1.tx_conti_mode = 0;
  652. rmt_set_tx_thr_intr_en(channel, 1, item_sub_len);
  653. p_rmt->tx_data = rmt_item + item_block_len;
  654. p_rmt->tx_len_rem = len_rem;
  655. p_rmt->tx_offset = 0;
  656. p_rmt->tx_sub_len = item_sub_len;
  657. } else {
  658. rmt_fill_memory(channel, rmt_item, len_rem, 0);
  659. RMTMEM.chan[channel].data32[len_rem].val = 0;
  660. len_rem = 0;
  661. }
  662. rmt_tx_start(channel, true);
  663. if(wait_tx_done) {
  664. xSemaphoreTake(p_rmt->tx_sem, portMAX_DELAY);
  665. xSemaphoreGive(p_rmt->tx_sem);
  666. }
  667. return ESP_OK;
  668. }
  669. esp_err_t rmt_wait_tx_done(rmt_channel_t channel, TickType_t wait_time)
  670. {
  671. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  672. RMT_CHECK(p_rmt_obj[channel] != NULL, RMT_DRIVER_ERROR_STR, ESP_FAIL);
  673. if(xSemaphoreTake(p_rmt_obj[channel]->tx_sem, wait_time) == pdTRUE) {
  674. xSemaphoreGive(p_rmt_obj[channel]->tx_sem);
  675. return ESP_OK;
  676. }
  677. else {
  678. ESP_LOGE(RMT_TAG, "Timeout on wait_tx_done");
  679. return ESP_ERR_TIMEOUT;
  680. }
  681. }
  682. esp_err_t rmt_get_ringbuf_handle(rmt_channel_t channel, RingbufHandle_t* buf_handle)
  683. {
  684. RMT_CHECK(channel < RMT_CHANNEL_MAX, RMT_CHANNEL_ERROR_STR, ESP_ERR_INVALID_ARG);
  685. RMT_CHECK(p_rmt_obj[channel] != NULL, RMT_DRIVER_ERROR_STR, ESP_FAIL);
  686. RMT_CHECK(buf_handle != NULL, RMT_ADDR_ERROR_STR, ESP_ERR_INVALID_ARG);
  687. *buf_handle = p_rmt_obj[channel]->rx_buf;
  688. return ESP_OK;
  689. }