output_i2s.c 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730
  1. /*
  2. * Squeezelite for esp32
  3. *
  4. * (c) Sebastien 2019
  5. * Philippe G. 2019, philippe_44@outlook.com
  6. *
  7. * This software is released under the MIT License.
  8. * https://opensource.org/licenses/MIT
  9. *
  10. */
  11. /*
  12. Synchronisation is a bit of a hack with i2s. The esp32 driver is always
  13. full when it starts, so there is a delay of the total length of buffers.
  14. In other words, i2s_write blocks at first call, until at least one buffer
  15. has been written (it uses a queue with produce / consume).
  16. The first hack is to consume that length at the beginning of tracks when
  17. synchronization is active. It's about ~180ms @ 44.1kHz
  18. The second hack is that we never know exactly the number of frames in the
  19. DMA buffers when we update the output.frames_played_dmp. We assume that
  20. after i2s_write, these buffers are always full so by measuring the gap
  21. between time after i2s_write and update of frames_played_dmp, we have a
  22. good idea of the error.
  23. The third hack is when sample rate changes, buffers are reset and we also
  24. do the change too early, but can't do that exaclty at the right time. So
  25. there might be a pop and a de-sync when sampling rate change happens. Not
  26. sure that using rate_delay would fix that
  27. */
  28. #include "squeezelite.h"
  29. #include "slimproto.h"
  30. #include "esp_pthread.h"
  31. #include "driver/i2s.h"
  32. #include "driver/i2c.h"
  33. #include "driver/gpio.h"
  34. #include "perf_trace.h"
  35. #include <signal.h>
  36. #include "adac.h"
  37. #include "time.h"
  38. #include "led.h"
  39. #include "monitor.h"
  40. #include "platform_config.h"
  41. #include "accessors.h"
  42. #include "equalizer.h"
  43. #include "globdefs.h"
  44. #define LOCK mutex_lock(outputbuf->mutex)
  45. #define UNLOCK mutex_unlock(outputbuf->mutex)
  46. #define FRAME_BLOCK MAX_SILENCE_FRAMES
  47. #define SPDIF_BLOCK 256
  48. // must have an integer ratio with FRAME_BLOCK (see spdif comment)
  49. #define DMA_BUF_LEN 512
  50. #define DMA_BUF_COUNT 12
  51. #define DECLARE_ALL_MIN_MAX \
  52. DECLARE_MIN_MAX(o); \
  53. DECLARE_MIN_MAX(s); \
  54. DECLARE_MIN_MAX(rec); \
  55. DECLARE_MIN_MAX(i2s_time); \
  56. DECLARE_MIN_MAX(buffering);
  57. #define RESET_ALL_MIN_MAX \
  58. RESET_MIN_MAX(o); \
  59. RESET_MIN_MAX(s); \
  60. RESET_MIN_MAX(rec); \
  61. RESET_MIN_MAX(i2s_time); \
  62. RESET_MIN_MAX(buffering);
  63. #define STATS_PERIOD_MS 5000
  64. #define STAT_STACK_SIZE (3*1024)
  65. extern struct outputstate output;
  66. extern struct buffer *streambuf;
  67. extern struct buffer *outputbuf;
  68. extern u8_t *silencebuf;
  69. const struct adac_s *dac_set[] = { &dac_tas57xx, &dac_tas5713, &dac_ac101, &dac_wm8978, NULL };
  70. const struct adac_s *adac = &dac_external;
  71. static log_level loglevel;
  72. static bool (*slimp_handler_chain)(u8_t *data, int len);
  73. static bool jack_mutes_amp;
  74. static bool running, isI2SStarted, ended;
  75. static i2s_config_t i2s_config;
  76. static u8_t *obuf;
  77. static frames_t oframes;
  78. static struct {
  79. bool enabled;
  80. u8_t *buf;
  81. size_t count;
  82. } spdif;
  83. static size_t dma_buf_frames;
  84. static TaskHandle_t stats_task, output_i2s_task;
  85. static bool stats;
  86. static struct {
  87. int gpio, active;
  88. } amp_control = { -1, 1 },
  89. mute_control = { CONFIG_MUTE_GPIO, CONFIG_MUTE_GPIO_LEVEL };
  90. DECLARE_ALL_MIN_MAX;
  91. static int _i2s_write_frames(frames_t out_frames, bool silence, s32_t gainL, s32_t gainR, u8_t flags,
  92. s32_t cross_gain_in, s32_t cross_gain_out, ISAMPLE_T **cross_ptr);
  93. static void output_thread_i2s(void *arg);
  94. static void output_thread_i2s_stats(void *arg);
  95. static void spdif_convert(ISAMPLE_T *src, size_t frames, u32_t *dst, size_t *count);
  96. static void (*jack_handler_chain)(bool inserted);
  97. #define I2C_PORT 0
  98. /****************************************************************************************
  99. * AUDO packet handler
  100. */
  101. static bool handler(u8_t *data, int len){
  102. bool res = true;
  103. if (!strncmp((char*) data, "audo", 4)) {
  104. struct audo_packet *pkt = (struct audo_packet*) data;
  105. // 0 = headphone (internal speakers off), 1 = sub out,
  106. // 2 = always on (internal speakers on), 3 = always off
  107. if (jack_mutes_amp != (pkt->config == 0)) {
  108. jack_mutes_amp = pkt->config == 0;
  109. config_set_value(NVS_TYPE_STR, "jack_mutes_amp", jack_mutes_amp ? "y" : "n");
  110. if (jack_mutes_amp && jack_inserted_svc()) adac->speaker(false);
  111. else adac->speaker(true);
  112. }
  113. LOG_INFO("got AUDO %02x", pkt->config);
  114. } else {
  115. res = false;
  116. }
  117. // chain protocol handlers (bitwise or is fine)
  118. if (*slimp_handler_chain) res |= (*slimp_handler_chain)(data, len);
  119. return res;
  120. }
  121. /****************************************************************************************
  122. * jack insertion handler
  123. */
  124. static void jack_handler(bool inserted) {
  125. // jack detection bounces a bit but that seems fine
  126. if (jack_mutes_amp) {
  127. LOG_INFO("switching amplifier %s", inserted ? "OFF" : "ON");
  128. if (inserted) adac->speaker(false);
  129. else adac->speaker(true);
  130. }
  131. // activate headset
  132. if (inserted) adac->headset(true);
  133. else adac->headset(false);
  134. // and chain if any
  135. if (jack_handler_chain) (jack_handler_chain)(inserted);
  136. }
  137. /****************************************************************************************
  138. * amp GPIO
  139. */
  140. static void set_amp_gpio(int gpio, char *value) {
  141. char *p;
  142. if (strcasestr(value, "amp")) {
  143. amp_control.gpio = gpio;
  144. if ((p = strchr(value, ':')) != NULL) amp_control.active = atoi(p + 1);
  145. gpio_pad_select_gpio(amp_control.gpio);
  146. gpio_set_direction(amp_control.gpio, GPIO_MODE_OUTPUT);
  147. gpio_set_level(amp_control.gpio, !amp_control.active);
  148. LOG_INFO("setting amplifier GPIO %d (active:%d)", amp_control.gpio, amp_control.active);
  149. }
  150. }
  151. /****************************************************************************************
  152. * Set pin from config string
  153. */
  154. static void set_i2s_pin(char *config, i2s_pin_config_t *pin_config) {
  155. char *p;
  156. pin_config->bck_io_num = pin_config->ws_io_num = pin_config->data_out_num = pin_config->data_in_num = -1;
  157. if ((p = strcasestr(config, "bck")) != NULL) pin_config->bck_io_num = atoi(strchr(p, '=') + 1);
  158. if ((p = strcasestr(config, "ws")) != NULL) pin_config->ws_io_num = atoi(strchr(p, '=') + 1);
  159. if ((p = strcasestr(config, "do")) != NULL) pin_config->data_out_num = atoi(strchr(p, '=') + 1);
  160. }
  161. /****************************************************************************************
  162. * Initialize the DAC output
  163. */
  164. void output_init_i2s(log_level level, char *device, unsigned output_buf_size, char *params, unsigned rates[], unsigned rate_delay, unsigned idle) {
  165. loglevel = level;
  166. int silent_do = -1;
  167. char *p;
  168. esp_err_t res;
  169. // chain SLIMP handlers
  170. slimp_handler_chain = slimp_handler;
  171. slimp_handler = handler;
  172. p = config_alloc_get_default(NVS_TYPE_STR, "jack_mutes_amp", "n", 0);
  173. jack_mutes_amp = (strcmp(p,"1") == 0 ||strcasecmp(p,"y") == 0);
  174. free(p);
  175. #if BYTES_PER_FRAME == 8
  176. output.format = S32_LE;
  177. #else
  178. output.format = S16_LE;
  179. #endif
  180. output.write_cb = &_i2s_write_frames;
  181. obuf = malloc(FRAME_BLOCK * BYTES_PER_FRAME);
  182. if (!obuf) {
  183. LOG_ERROR("Cannot allocate i2s buffer");
  184. return;
  185. }
  186. running = true;
  187. // get SPDIF configuration from NVS or compile
  188. char *spdif_config = config_alloc_get_str("spdif_config", CONFIG_SPDIF_CONFIG, "bck=" STR(CONFIG_SPDIF_BCK_IO)
  189. ",ws=" STR(CONFIG_SPDIF_WS_IO) ",do=" STR(CONFIG_SPDIF_DO_IO));
  190. char *dac_config = config_alloc_get_str("dac_config", CONFIG_DAC_CONFIG, "model=i2s,bck=" STR(CONFIG_I2S_BCK_IO)
  191. ",ws=" STR(CONFIG_I2S_WS_IO) ",do=" STR(CONFIG_I2S_DO_IO)
  192. ",sda=" STR(CONFIG_I2C_SDA) ",scl=" STR(CONFIG_I2C_SCL)
  193. ",mute=" STR(CONFIG_MUTE_GPIO));
  194. i2s_pin_config_t i2s_dac_pin, i2s_spdif_pin;
  195. set_i2s_pin(spdif_config, &i2s_spdif_pin);
  196. set_i2s_pin(dac_config, &i2s_dac_pin);
  197. /* BEWARE: i2s.c must be patched otherwise L/R are swapped in 32 bits mode */
  198. // common I2S initialization
  199. i2s_config.mode = I2S_MODE_MASTER | I2S_MODE_TX;
  200. i2s_config.channel_format = I2S_CHANNEL_FMT_RIGHT_LEFT;
  201. i2s_config.communication_format = I2S_COMM_FORMAT_I2S| I2S_COMM_FORMAT_I2S_MSB;
  202. // in case of overflow, do not replay old buffer
  203. i2s_config.tx_desc_auto_clear = true;
  204. i2s_config.use_apll = true;
  205. i2s_config.intr_alloc_flags = ESP_INTR_FLAG_LEVEL1; //Interrupt level 1
  206. if (strcasestr(device, "spdif")) {
  207. spdif.enabled = true;
  208. if ((spdif.buf = heap_caps_malloc(SPDIF_BLOCK * 16, MALLOC_CAP_INTERNAL)) == NULL) {
  209. LOG_ERROR("Cannot allocate SPDIF buffer");
  210. }
  211. if (i2s_spdif_pin.bck_io_num == -1 || i2s_spdif_pin.ws_io_num == -1 || i2s_spdif_pin.data_out_num == -1) {
  212. LOG_WARN("Cannot initialize I2S for SPDIF bck:%d ws:%d do:%d", i2s_spdif_pin.bck_io_num,
  213. i2s_spdif_pin.ws_io_num,
  214. i2s_spdif_pin.data_out_num);
  215. }
  216. i2s_config.sample_rate = output.current_sample_rate * 2;
  217. i2s_config.bits_per_sample = 32;
  218. // Normally counted in frames, but 16 sample are transformed into 32 bits in spdif
  219. i2s_config.dma_buf_len = DMA_BUF_LEN / 2;
  220. i2s_config.dma_buf_count = DMA_BUF_COUNT * 2;
  221. /*
  222. In DMA, we have room for (LEN * COUNT) frames of 32 bits samples that
  223. we push at sample_rate * 2. Each of these peuso-frames is a single true
  224. audio frame. So the real depth is true frames is (LEN * COUNT / 2)
  225. */
  226. dma_buf_frames = DMA_BUF_COUNT * DMA_BUF_LEN / 2;
  227. // silence DAC output if sharing the same ws/bck
  228. if (i2s_dac_pin.ws_io_num == i2s_spdif_pin.ws_io_num && i2s_dac_pin.bck_io_num == i2s_spdif_pin.bck_io_num) silent_do = i2s_dac_pin.data_out_num;
  229. res = i2s_driver_install(CONFIG_I2S_NUM, &i2s_config, 0, NULL);
  230. res |= i2s_set_pin(CONFIG_I2S_NUM, &i2s_spdif_pin);
  231. LOG_INFO("SPDIF using I2S bck:%u, ws:%u, do:%u", i2s_spdif_pin.bck_io_num, i2s_spdif_pin.ws_io_num, i2s_spdif_pin.data_out_num);
  232. } else {
  233. i2s_config.sample_rate = output.current_sample_rate;
  234. i2s_config.bits_per_sample = BYTES_PER_FRAME * 8 / 2;
  235. // Counted in frames (but i2s allocates a buffer <= 4092 bytes)
  236. i2s_config.dma_buf_len = DMA_BUF_LEN;
  237. i2s_config.dma_buf_count = DMA_BUF_COUNT;
  238. dma_buf_frames = DMA_BUF_COUNT * DMA_BUF_LEN;
  239. // silence SPDIF output
  240. silent_do = i2s_spdif_pin.data_out_num;
  241. char model[32] = "i2s";
  242. if ((p = strcasestr(dac_config, "model")) != NULL) sscanf(p, "%*[^=]=%31[^,]", model);
  243. if ((p = strcasestr(dac_config, "mute")) != NULL) {
  244. char mute[8] = "";
  245. sscanf(p, "%*[^=]=%7[^,]", mute);
  246. mute_control.gpio = atoi(mute);
  247. if ((p = strchr(mute, ':')) != NULL) mute_control.active = atoi(p + 1);
  248. }
  249. for (int i = 0; adac == &dac_external && dac_set[i]; i++) if (strcasestr(dac_set[i]->model, model)) adac = dac_set[i];
  250. res = adac->init(dac_config, I2C_PORT, &i2s_config) ? ESP_OK : ESP_FAIL;
  251. res |= i2s_driver_install(CONFIG_I2S_NUM, &i2s_config, 0, NULL);
  252. res |= i2s_set_pin(CONFIG_I2S_NUM, &i2s_dac_pin);
  253. if (res == ESP_OK && mute_control.gpio >= 0) {
  254. gpio_pad_select_gpio(mute_control.gpio);
  255. gpio_set_direction(mute_control.gpio, GPIO_MODE_OUTPUT);
  256. gpio_set_level(mute_control.gpio, mute_control.active);
  257. }
  258. LOG_INFO("%s DAC using I2S bck:%d, ws:%d, do:%d, mute:%d:%d (res:%d)", model, i2s_dac_pin.bck_io_num, i2s_dac_pin.ws_io_num,
  259. i2s_dac_pin.data_out_num, mute_control.gpio, mute_control.active, res);
  260. }
  261. free(dac_config);
  262. free(spdif_config);
  263. if (res != ESP_OK) {
  264. LOG_WARN("no DAC configured");
  265. return;
  266. }
  267. // turn off GPIO than is not used (SPDIF of DAC DO when shared)
  268. if (silent_do >= 0) {
  269. gpio_pad_select_gpio(silent_do);
  270. gpio_set_direction(silent_do, GPIO_MODE_OUTPUT);
  271. gpio_set_level(silent_do, 0);
  272. }
  273. LOG_INFO("Initializing I2S mode %s with rate: %d, bits per sample: %d, buffer frames: %d, number of buffers: %d ",
  274. spdif.enabled ? "S/PDIF" : "normal",
  275. i2s_config.sample_rate, i2s_config.bits_per_sample, i2s_config.dma_buf_len, i2s_config.dma_buf_count);
  276. i2s_stop(CONFIG_I2S_NUM);
  277. i2s_zero_dma_buffer(CONFIG_I2S_NUM);
  278. isI2SStarted=false;
  279. adac->power(ADAC_STANDBY);
  280. jack_handler_chain = jack_handler_svc;
  281. jack_handler_svc = jack_handler;
  282. if (jack_mutes_amp && jack_inserted_svc()) adac->speaker(false);
  283. else adac->speaker(true);
  284. adac->headset(jack_inserted_svc());
  285. parse_set_GPIO(set_amp_gpio);
  286. // create task as a FreeRTOS task but uses stack in internal RAM
  287. {
  288. static DRAM_ATTR StaticTask_t xTaskBuffer __attribute__ ((aligned (4)));
  289. static DRAM_ATTR StackType_t xStack[OUTPUT_THREAD_STACK_SIZE] __attribute__ ((aligned (4)));
  290. output_i2s_task = xTaskCreateStaticPinnedToCore( (TaskFunction_t) output_thread_i2s, "output_i2s", OUTPUT_THREAD_STACK_SIZE,
  291. NULL, CONFIG_ESP32_PTHREAD_TASK_PRIO_DEFAULT + 1, xStack, &xTaskBuffer, 0 );
  292. }
  293. // do we want stats
  294. p = config_alloc_get_default(NVS_TYPE_STR, "stats", "n", 0);
  295. stats = p && (*p == '1' || *p == 'Y' || *p == 'y');
  296. free(p);
  297. // memory still used but at least task is not created
  298. if (stats) {
  299. static DRAM_ATTR StaticTask_t xTaskBuffer __attribute__ ((aligned (4)));
  300. static EXT_RAM_ATTR StackType_t xStack[STAT_STACK_SIZE] __attribute__ ((aligned (4)));
  301. stats_task = xTaskCreateStatic( (TaskFunction_t) output_thread_i2s_stats, "output_i2s_sts", STAT_STACK_SIZE,
  302. NULL, ESP_TASK_PRIO_MIN, xStack, &xTaskBuffer);
  303. }
  304. }
  305. /****************************************************************************************
  306. * Terminate DAC output
  307. */
  308. void output_close_i2s(void) {
  309. LOCK;
  310. running = false;
  311. UNLOCK;
  312. while (!ended) vTaskDelay(20 / portTICK_PERIOD_MS);
  313. if (stats) vTaskDelete(stats_task);
  314. i2s_driver_uninstall(CONFIG_I2S_NUM);
  315. free(obuf);
  316. equalizer_close();
  317. adac->deinit();
  318. }
  319. /****************************************************************************************
  320. * change volume
  321. */
  322. bool output_volume_i2s(unsigned left, unsigned right) {
  323. if (mute_control.gpio >= 0) gpio_set_level(mute_control.gpio, (left | right) ? !mute_control.active : mute_control.active);
  324. return adac->volume(left, right);
  325. }
  326. /****************************************************************************************
  327. * Write frames to the output buffer
  328. */
  329. static int _i2s_write_frames(frames_t out_frames, bool silence, s32_t gainL, s32_t gainR, u8_t flags,
  330. s32_t cross_gain_in, s32_t cross_gain_out, ISAMPLE_T **cross_ptr) {
  331. if (!silence) {
  332. if (output.fade == FADE_ACTIVE && output.fade_dir == FADE_CROSS && *cross_ptr) {
  333. _apply_cross(outputbuf, out_frames, cross_gain_in, cross_gain_out, cross_ptr);
  334. }
  335. _apply_gain(outputbuf, out_frames, gainL, gainR, flags);
  336. memcpy(obuf + oframes * BYTES_PER_FRAME, outputbuf->readp, out_frames * BYTES_PER_FRAME);
  337. } else {
  338. memcpy(obuf + oframes * BYTES_PER_FRAME, silencebuf, out_frames * BYTES_PER_FRAME);
  339. }
  340. // don't update visu if we don't have enough data in buffer
  341. if (silence || output.external || _buf_used(outputbuf) > outputbuf->size >> 2 ) {
  342. output_visu_export(obuf + oframes * BYTES_PER_FRAME, out_frames, output.current_sample_rate, silence, (gainL + gainR) / 2);
  343. }
  344. oframes += out_frames;
  345. return out_frames;
  346. }
  347. /****************************************************************************************
  348. * Main output thread
  349. */
  350. static void output_thread_i2s(void *arg) {
  351. size_t bytes;
  352. frames_t iframes = FRAME_BLOCK;
  353. uint32_t timer_start = 0;
  354. int discard = 0;
  355. uint32_t fullness = gettime_ms();
  356. bool synced;
  357. output_state state = OUTPUT_OFF - 1;
  358. while (running) {
  359. TIME_MEASUREMENT_START(timer_start);
  360. LOCK;
  361. // manage led display & analogue
  362. if (state != output.state) {
  363. LOG_INFO("Output state is %d", output.state);
  364. if (output.state == OUTPUT_OFF) {
  365. led_blink(LED_GREEN, 100, 2500);
  366. if (amp_control.gpio != -1) gpio_set_level(amp_control.gpio, !amp_control.active);
  367. LOG_INFO("switching off amp GPIO %d", amp_control.gpio);
  368. } else if (output.state == OUTPUT_STOPPED) {
  369. adac->speaker(false);
  370. led_blink(LED_GREEN, 200, 1000);
  371. } else if (output.state == OUTPUT_RUNNING) {
  372. if (!jack_mutes_amp || !jack_inserted_svc()) adac->speaker(true);
  373. led_on(LED_GREEN);
  374. }
  375. }
  376. state = output.state;
  377. if (output.state == OUTPUT_OFF) {
  378. UNLOCK;
  379. if (isI2SStarted) {
  380. isI2SStarted = false;
  381. i2s_stop(CONFIG_I2S_NUM);
  382. adac->power(ADAC_STANDBY);
  383. spdif.count = 0;
  384. }
  385. usleep(100000);
  386. continue;
  387. } else if (output.state == OUTPUT_STOPPED) {
  388. synced = false;
  389. }
  390. oframes = 0;
  391. output.updated = gettime_ms();
  392. output.frames_played_dmp = output.frames_played;
  393. // try to estimate how much we have consumed from the DMA buffer (calculation is incorrect at the very beginning ...)
  394. output.device_frames = dma_buf_frames - ((output.updated - fullness) * output.current_sample_rate) / 1000;
  395. _output_frames( iframes );
  396. // oframes must be a global updated by the write callback
  397. output.frames_in_process = oframes;
  398. SET_MIN_MAX_SIZED(oframes,rec,iframes);
  399. SET_MIN_MAX_SIZED(_buf_used(outputbuf),o,outputbuf->size);
  400. SET_MIN_MAX_SIZED(_buf_used(streambuf),s,streambuf->size);
  401. SET_MIN_MAX( TIME_MEASUREMENT_GET(timer_start),buffering);
  402. /* must skip first whatever is in the pipe (but not when resuming).
  403. This test is incorrect when we pause a track that has just started,
  404. but this is higly unlikely and I don't have a better one for now */
  405. if (output.state == OUTPUT_START_AT) {
  406. discard = output.frames_played_dmp ? 0 : output.device_frames;
  407. synced = true;
  408. } else if (discard) {
  409. discard -= oframes;
  410. iframes = discard ? min(FRAME_BLOCK, discard) : FRAME_BLOCK;
  411. UNLOCK;
  412. continue;
  413. }
  414. UNLOCK;
  415. // now send all the data
  416. TIME_MEASUREMENT_START(timer_start);
  417. if (!isI2SStarted ) {
  418. isI2SStarted = true;
  419. LOG_INFO("Restarting I2S.");
  420. i2s_zero_dma_buffer(CONFIG_I2S_NUM);
  421. i2s_start(CONFIG_I2S_NUM);
  422. adac->power(ADAC_ON);
  423. if (amp_control.gpio != -1) gpio_set_level(amp_control.gpio, amp_control.active);
  424. }
  425. // this does not work well as set_sample_rates resets the fifos (and it's too early)
  426. if (i2s_config.sample_rate != output.current_sample_rate) {
  427. LOG_INFO("changing sampling rate %u to %u", i2s_config.sample_rate, output.current_sample_rate);
  428. if (synced) {
  429. /*
  430. // can sleep for a buffer_queue - 1 and then eat a buffer (discard) if we are synced
  431. usleep(((DMA_BUF_COUNT - 1) * DMA_BUF_LEN * BYTES_PER_FRAME * 1000) / 44100 * 1000);
  432. discard = DMA_BUF_COUNT * DMA_BUF_LEN * BYTES_PER_FRAME;
  433. */
  434. }
  435. i2s_config.sample_rate = output.current_sample_rate;
  436. i2s_set_sample_rates(CONFIG_I2S_NUM, spdif.enabled ? i2s_config.sample_rate * 2 : i2s_config.sample_rate);
  437. i2s_zero_dma_buffer(CONFIG_I2S_NUM);
  438. #if BYTES_PER_FRAME == 4
  439. equalizer_close();
  440. equalizer_open(output.current_sample_rate);
  441. #endif
  442. }
  443. #if BYTES_PER_FRAME == 4
  444. // run equalizer
  445. equalizer_process(obuf, oframes * BYTES_PER_FRAME, output.current_sample_rate);
  446. #endif
  447. // we assume that here we have been able to entirely fill the DMA buffers
  448. if (spdif.enabled) {
  449. size_t obytes, count = 0;
  450. bytes = 0;
  451. // need IRAM for speed but can't allocate a FRAME_BLOCK * 16, so process by smaller chunks
  452. while (count < oframes) {
  453. size_t chunk = min(SPDIF_BLOCK, oframes - count);
  454. spdif_convert((ISAMPLE_T*) obuf + count * 2, chunk, (u32_t*) spdif.buf, &spdif.count);
  455. i2s_write(CONFIG_I2S_NUM, spdif.buf, chunk * 16, &obytes, portMAX_DELAY);
  456. bytes += obytes / (16 / BYTES_PER_FRAME);
  457. count += chunk;
  458. }
  459. #if BYTES_PER_FRAME == 4
  460. } else if (i2s_config.bits_per_sample == 32) {
  461. i2s_write_expand(CONFIG_I2S_NUM, obuf, oframes * BYTES_PER_FRAME, 16, 32, &bytes, portMAX_DELAY);
  462. #endif
  463. } else {
  464. i2s_write(CONFIG_I2S_NUM, obuf, oframes * BYTES_PER_FRAME, &bytes, portMAX_DELAY);
  465. }
  466. fullness = gettime_ms();
  467. if (bytes != oframes * BYTES_PER_FRAME) {
  468. LOG_WARN("I2S DMA Overflow! available bytes: %d, I2S wrote %d bytes", oframes * BYTES_PER_FRAME, bytes);
  469. }
  470. SET_MIN_MAX( TIME_MEASUREMENT_GET(timer_start),i2s_time);
  471. }
  472. if (spdif.enabled) free(spdif.buf);
  473. ended = true;
  474. vTaskDelete(NULL);
  475. }
  476. /****************************************************************************************
  477. * Stats output thread
  478. */
  479. static void output_thread_i2s_stats(void *arg) {
  480. while (1) {
  481. // no need to lock
  482. output_state state = output.state;
  483. if(stats && state>OUTPUT_STOPPED){
  484. LOG_INFO( "Output State: %d, current sample rate: %d, bytes per frame: %d",state,output.current_sample_rate, BYTES_PER_FRAME);
  485. LOG_INFO( LINE_MIN_MAX_FORMAT_HEAD1);
  486. LOG_INFO( LINE_MIN_MAX_FORMAT_HEAD2);
  487. LOG_INFO( LINE_MIN_MAX_FORMAT_HEAD3);
  488. LOG_INFO( LINE_MIN_MAX_FORMAT_HEAD4);
  489. LOG_INFO(LINE_MIN_MAX_FORMAT_STREAM, LINE_MIN_MAX_STREAM("stream",s));
  490. LOG_INFO(LINE_MIN_MAX_FORMAT,LINE_MIN_MAX("output",o));
  491. LOG_INFO(LINE_MIN_MAX_FORMAT_FOOTER);
  492. LOG_INFO(LINE_MIN_MAX_FORMAT,LINE_MIN_MAX("received",rec));
  493. LOG_INFO(LINE_MIN_MAX_FORMAT_FOOTER);
  494. LOG_INFO("");
  495. LOG_INFO(" ----------+----------+-----------+-----------+ ");
  496. LOG_INFO(" max (us) | min (us) | avg(us) | count | ");
  497. LOG_INFO(" ----------+----------+-----------+-----------+ ");
  498. LOG_INFO(LINE_MIN_MAX_DURATION_FORMAT,LINE_MIN_MAX_DURATION("Buffering(us)",buffering));
  499. LOG_INFO(LINE_MIN_MAX_DURATION_FORMAT,LINE_MIN_MAX_DURATION("i2s tfr(us)",i2s_time));
  500. LOG_INFO(" ----------+----------+-----------+-----------+");
  501. RESET_ALL_MIN_MAX;
  502. }
  503. vTaskDelay( pdMS_TO_TICKS( STATS_PERIOD_MS ) );
  504. }
  505. }
  506. /****************************************************************************************
  507. * SPDIF support
  508. */
  509. #define PREAMBLE_B (0xE8) //11101000
  510. #define PREAMBLE_M (0xE2) //11100010
  511. #define PREAMBLE_W (0xE4) //11100100
  512. #define VUCP ((0xCC) << 24)
  513. #define VUCP_MUTE ((0xD4) << 24) // To mute PCM, set VUCP = invalid.
  514. static const u16_t spdif_bmclookup[256] = { //biphase mark encoded values (least significant bit first)
  515. 0xcccc, 0x4ccc, 0x2ccc, 0xaccc, 0x34cc, 0xb4cc, 0xd4cc, 0x54cc,
  516. 0x32cc, 0xb2cc, 0xd2cc, 0x52cc, 0xcacc, 0x4acc, 0x2acc, 0xaacc,
  517. 0x334c, 0xb34c, 0xd34c, 0x534c, 0xcb4c, 0x4b4c, 0x2b4c, 0xab4c,
  518. 0xcd4c, 0x4d4c, 0x2d4c, 0xad4c, 0x354c, 0xb54c, 0xd54c, 0x554c,
  519. 0x332c, 0xb32c, 0xd32c, 0x532c, 0xcb2c, 0x4b2c, 0x2b2c, 0xab2c,
  520. 0xcd2c, 0x4d2c, 0x2d2c, 0xad2c, 0x352c, 0xb52c, 0xd52c, 0x552c,
  521. 0xccac, 0x4cac, 0x2cac, 0xacac, 0x34ac, 0xb4ac, 0xd4ac, 0x54ac,
  522. 0x32ac, 0xb2ac, 0xd2ac, 0x52ac, 0xcaac, 0x4aac, 0x2aac, 0xaaac,
  523. 0x3334, 0xb334, 0xd334, 0x5334, 0xcb34, 0x4b34, 0x2b34, 0xab34,
  524. 0xcd34, 0x4d34, 0x2d34, 0xad34, 0x3534, 0xb534, 0xd534, 0x5534,
  525. 0xccb4, 0x4cb4, 0x2cb4, 0xacb4, 0x34b4, 0xb4b4, 0xd4b4, 0x54b4,
  526. 0x32b4, 0xb2b4, 0xd2b4, 0x52b4, 0xcab4, 0x4ab4, 0x2ab4, 0xaab4,
  527. 0xccd4, 0x4cd4, 0x2cd4, 0xacd4, 0x34d4, 0xb4d4, 0xd4d4, 0x54d4,
  528. 0x32d4, 0xb2d4, 0xd2d4, 0x52d4, 0xcad4, 0x4ad4, 0x2ad4, 0xaad4,
  529. 0x3354, 0xb354, 0xd354, 0x5354, 0xcb54, 0x4b54, 0x2b54, 0xab54,
  530. 0xcd54, 0x4d54, 0x2d54, 0xad54, 0x3554, 0xb554, 0xd554, 0x5554,
  531. 0x3332, 0xb332, 0xd332, 0x5332, 0xcb32, 0x4b32, 0x2b32, 0xab32,
  532. 0xcd32, 0x4d32, 0x2d32, 0xad32, 0x3532, 0xb532, 0xd532, 0x5532,
  533. 0xccb2, 0x4cb2, 0x2cb2, 0xacb2, 0x34b2, 0xb4b2, 0xd4b2, 0x54b2,
  534. 0x32b2, 0xb2b2, 0xd2b2, 0x52b2, 0xcab2, 0x4ab2, 0x2ab2, 0xaab2,
  535. 0xccd2, 0x4cd2, 0x2cd2, 0xacd2, 0x34d2, 0xb4d2, 0xd4d2, 0x54d2,
  536. 0x32d2, 0xb2d2, 0xd2d2, 0x52d2, 0xcad2, 0x4ad2, 0x2ad2, 0xaad2,
  537. 0x3352, 0xb352, 0xd352, 0x5352, 0xcb52, 0x4b52, 0x2b52, 0xab52,
  538. 0xcd52, 0x4d52, 0x2d52, 0xad52, 0x3552, 0xb552, 0xd552, 0x5552,
  539. 0xccca, 0x4cca, 0x2cca, 0xacca, 0x34ca, 0xb4ca, 0xd4ca, 0x54ca,
  540. 0x32ca, 0xb2ca, 0xd2ca, 0x52ca, 0xcaca, 0x4aca, 0x2aca, 0xaaca,
  541. 0x334a, 0xb34a, 0xd34a, 0x534a, 0xcb4a, 0x4b4a, 0x2b4a, 0xab4a,
  542. 0xcd4a, 0x4d4a, 0x2d4a, 0xad4a, 0x354a, 0xb54a, 0xd54a, 0x554a,
  543. 0x332a, 0xb32a, 0xd32a, 0x532a, 0xcb2a, 0x4b2a, 0x2b2a, 0xab2a,
  544. 0xcd2a, 0x4d2a, 0x2d2a, 0xad2a, 0x352a, 0xb52a, 0xd52a, 0x552a,
  545. 0xccaa, 0x4caa, 0x2caa, 0xacaa, 0x34aa, 0xb4aa, 0xd4aa, 0x54aa,
  546. 0x32aa, 0xb2aa, 0xd2aa, 0x52aa, 0xcaaa, 0x4aaa, 0x2aaa, 0xaaaa
  547. };
  548. /*
  549. SPDIF is supposed to be (before BMC encoding, from LSB to MSB)
  550. PPPP AAAA SSSS SSSS SSSS SSSS SSSS VUCP
  551. after BMC encoding, each bits becomes 2 hence this becomes a 64 bits word. The
  552. the trick is to start not with a PPPP sequence but with an VUCP sequence to that
  553. the 16 bits samples are aligned with a BMC word boundary. Note that the LSB of the
  554. audio is transmitted first (not the MSB) and that ESP32 libray sends R then L,
  555. contrary to what seems to be usually done, so (dst) order had to be changed
  556. */
  557. void spdif_convert(ISAMPLE_T *src, size_t frames, u32_t *dst, size_t *count) {
  558. register u16_t hi, lo, aux;
  559. size_t cnt = *count;
  560. while (frames--) {
  561. // start with left channel
  562. #if BYTES_PER_FRAME == 4
  563. hi = spdif_bmclookup[(u8_t)(*src >> 8)];
  564. lo = spdif_bmclookup[(u8_t) *src++];
  565. // invert if last preceeding bit is 1
  566. lo ^= ~((s16_t)hi) >> 16;
  567. // first 16 bits
  568. *dst++ = ((u32_t)lo << 16) | hi;
  569. aux = 0xb333 ^ (((u32_t)((s16_t)lo)) >> 17);
  570. #else
  571. hi = spdif_bmclookup[(u8_t)(*src >> 24)];
  572. lo = spdif_bmclookup[(u8_t)(*src >> 16)];
  573. // invert if last preceeding bit is 1
  574. lo ^= ~((s16_t)hi) >> 16;
  575. // first 16 bits
  576. *dst++ = ((u32_t)lo << 16) | hi;
  577. // we use 20 bits samples as we need to force parity
  578. aux = spdif_bmclookup[(u8_t)(*src++ >> 12)];
  579. aux = (u8_t) (aux ^ (~((s16_t)lo) >> 16));
  580. aux |= (0xb3 ^ (((u16_t)((s8_t)aux)) >> 9)) << 8;
  581. #endif
  582. // VUCP-Bits: Valid, Subcode, Channelstatus, Parity = 0
  583. // As parity is always 0, we can use fixed preambles
  584. if (++cnt > 191) {
  585. *dst++ = VUCP | (PREAMBLE_B << 16 ) | aux; //special preamble for one of 192 frames
  586. cnt = 0;
  587. } else {
  588. *dst++ = VUCP | (PREAMBLE_M << 16) | aux;
  589. }
  590. // then do right channel, no need to check PREAMBLE_B
  591. #if BYTES_PER_FRAME == 4
  592. hi = spdif_bmclookup[(u8_t)(*src >> 8)];
  593. lo = spdif_bmclookup[(u8_t) *src++];
  594. lo ^= ~((s16_t)hi) >> 16;
  595. *dst++ = ((u32_t)lo << 16) | hi;
  596. aux = 0xb333 ^ (((u32_t)((s16_t)lo)) >> 17);
  597. #else
  598. hi = spdif_bmclookup[(u8_t)(*src >> 24)];
  599. lo = spdif_bmclookup[(u8_t)(*src >> 16)];
  600. lo ^= ~((s16_t)hi) >> 16;
  601. *dst++ = ((u32_t)lo << 16) | hi;
  602. aux = spdif_bmclookup[(u8_t)(*src++ >> 12)];
  603. aux = (u8_t) (aux ^ (~((s16_t)lo) >> 16));
  604. aux |= (0xb3 ^ (((u16_t)((s8_t)aux)) >> 9)) << 8;
  605. #endif
  606. *dst++ = VUCP | (PREAMBLE_W << 16) | aux;
  607. }
  608. *count = cnt;
  609. }