jack.c 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354
  1. /*
  2. * JACK Audio Connection Kit input device
  3. * Copyright (c) 2009 Samalyse
  4. * Author: Olivier Guilyardi <olivier samalyse com>
  5. *
  6. * This file is part of FFmpeg.
  7. *
  8. * FFmpeg is free software; you can redistribute it and/or
  9. * modify it under the terms of the GNU Lesser General Public
  10. * License as published by the Free Software Foundation; either
  11. * version 2.1 of the License, or (at your option) any later version.
  12. *
  13. * FFmpeg is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16. * Lesser General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU Lesser General Public
  19. * License along with FFmpeg; if not, write to the Free Software
  20. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21. */
  22. #include "config.h"
  23. #include <semaphore.h>
  24. #include <jack/jack.h>
  25. #include "libavutil/internal.h"
  26. #include "libavutil/log.h"
  27. #include "libavutil/fifo.h"
  28. #include "libavutil/opt.h"
  29. #include "libavutil/time.h"
  30. #include "libavcodec/avcodec.h"
  31. #include "libavformat/avformat.h"
  32. #include "libavformat/internal.h"
  33. #include "timefilter.h"
  34. #include "avdevice.h"
  35. /**
  36. * Size of the internal FIFO buffers as a number of audio packets
  37. */
  38. #define FIFO_PACKETS_NUM 16
  39. typedef struct JackData {
  40. AVClass *class;
  41. jack_client_t * client;
  42. int activated;
  43. sem_t packet_count;
  44. jack_nframes_t sample_rate;
  45. jack_nframes_t buffer_size;
  46. jack_port_t ** ports;
  47. int nports;
  48. TimeFilter * timefilter;
  49. AVFifoBuffer * new_pkts;
  50. AVFifoBuffer * filled_pkts;
  51. int pkt_xrun;
  52. int jack_xrun;
  53. } JackData;
  54. static int process_callback(jack_nframes_t nframes, void *arg)
  55. {
  56. /* Warning: this function runs in realtime. One mustn't allocate memory here
  57. * or do any other thing that could block. */
  58. int i, j;
  59. JackData *self = arg;
  60. float * buffer;
  61. jack_nframes_t latency, cycle_delay;
  62. AVPacket pkt;
  63. float *pkt_data;
  64. double cycle_time;
  65. if (!self->client)
  66. return 0;
  67. /* The approximate delay since the hardware interrupt as a number of frames */
  68. cycle_delay = jack_frames_since_cycle_start(self->client);
  69. /* Retrieve filtered cycle time */
  70. cycle_time = ff_timefilter_update(self->timefilter,
  71. av_gettime() / 1000000.0 - (double) cycle_delay / self->sample_rate,
  72. self->buffer_size);
  73. /* Check if an empty packet is available, and if there's enough space to send it back once filled */
  74. if ((av_fifo_size(self->new_pkts) < sizeof(pkt)) || (av_fifo_space(self->filled_pkts) < sizeof(pkt))) {
  75. self->pkt_xrun = 1;
  76. return 0;
  77. }
  78. /* Retrieve empty (but allocated) packet */
  79. av_fifo_generic_read(self->new_pkts, &pkt, sizeof(pkt), NULL);
  80. pkt_data = (float *) pkt.data;
  81. latency = 0;
  82. /* Copy and interleave audio data from the JACK buffer into the packet */
  83. for (i = 0; i < self->nports; i++) {
  84. jack_latency_range_t range;
  85. jack_port_get_latency_range(self->ports[i], JackCaptureLatency, &range);
  86. latency += range.max;
  87. buffer = jack_port_get_buffer(self->ports[i], self->buffer_size);
  88. for (j = 0; j < self->buffer_size; j++)
  89. pkt_data[j * self->nports + i] = buffer[j];
  90. }
  91. /* Timestamp the packet with the cycle start time minus the average latency */
  92. pkt.pts = (cycle_time - (double) latency / (self->nports * self->sample_rate)) * 1000000.0;
  93. /* Send the now filled packet back, and increase packet counter */
  94. av_fifo_generic_write(self->filled_pkts, &pkt, sizeof(pkt), NULL);
  95. sem_post(&self->packet_count);
  96. return 0;
  97. }
  98. static void shutdown_callback(void *arg)
  99. {
  100. JackData *self = arg;
  101. self->client = NULL;
  102. }
  103. static int xrun_callback(void *arg)
  104. {
  105. JackData *self = arg;
  106. self->jack_xrun = 1;
  107. ff_timefilter_reset(self->timefilter);
  108. return 0;
  109. }
  110. static int supply_new_packets(JackData *self, AVFormatContext *context)
  111. {
  112. AVPacket pkt;
  113. int test, pkt_size = self->buffer_size * self->nports * sizeof(float);
  114. /* Supply the process callback with new empty packets, by filling the new
  115. * packets FIFO buffer with as many packets as possible. process_callback()
  116. * can't do this by itself, because it can't allocate memory in realtime. */
  117. while (av_fifo_space(self->new_pkts) >= sizeof(pkt)) {
  118. if ((test = av_new_packet(&pkt, pkt_size)) < 0) {
  119. av_log(context, AV_LOG_ERROR, "Could not create packet of size %d\n", pkt_size);
  120. return test;
  121. }
  122. av_fifo_generic_write(self->new_pkts, &pkt, sizeof(pkt), NULL);
  123. }
  124. return 0;
  125. }
  126. static int start_jack(AVFormatContext *context)
  127. {
  128. JackData *self = context->priv_data;
  129. jack_status_t status;
  130. int i, test;
  131. /* Register as a JACK client, using the context url as client name. */
  132. self->client = jack_client_open(context->url, JackNullOption, &status);
  133. if (!self->client) {
  134. av_log(context, AV_LOG_ERROR, "Unable to register as a JACK client\n");
  135. return AVERROR(EIO);
  136. }
  137. sem_init(&self->packet_count, 0, 0);
  138. self->sample_rate = jack_get_sample_rate(self->client);
  139. self->ports = av_malloc_array(self->nports, sizeof(*self->ports));
  140. if (!self->ports)
  141. return AVERROR(ENOMEM);
  142. self->buffer_size = jack_get_buffer_size(self->client);
  143. /* Register JACK ports */
  144. for (i = 0; i < self->nports; i++) {
  145. char str[16];
  146. snprintf(str, sizeof(str), "input_%d", i + 1);
  147. self->ports[i] = jack_port_register(self->client, str,
  148. JACK_DEFAULT_AUDIO_TYPE,
  149. JackPortIsInput, 0);
  150. if (!self->ports[i]) {
  151. av_log(context, AV_LOG_ERROR, "Unable to register port %s:%s\n",
  152. context->url, str);
  153. jack_client_close(self->client);
  154. return AVERROR(EIO);
  155. }
  156. }
  157. /* Register JACK callbacks */
  158. jack_set_process_callback(self->client, process_callback, self);
  159. jack_on_shutdown(self->client, shutdown_callback, self);
  160. jack_set_xrun_callback(self->client, xrun_callback, self);
  161. /* Create time filter */
  162. self->timefilter = ff_timefilter_new (1.0 / self->sample_rate, self->buffer_size, 1.5);
  163. if (!self->timefilter) {
  164. jack_client_close(self->client);
  165. return AVERROR(ENOMEM);
  166. }
  167. /* Create FIFO buffers */
  168. self->filled_pkts = av_fifo_alloc_array(FIFO_PACKETS_NUM, sizeof(AVPacket));
  169. /* New packets FIFO with one extra packet for safety against underruns */
  170. self->new_pkts = av_fifo_alloc_array((FIFO_PACKETS_NUM + 1), sizeof(AVPacket));
  171. if (!self->new_pkts) {
  172. jack_client_close(self->client);
  173. return AVERROR(ENOMEM);
  174. }
  175. if ((test = supply_new_packets(self, context))) {
  176. jack_client_close(self->client);
  177. return test;
  178. }
  179. return 0;
  180. }
  181. static void free_pkt_fifo(AVFifoBuffer **fifo)
  182. {
  183. AVPacket pkt;
  184. while (av_fifo_size(*fifo)) {
  185. av_fifo_generic_read(*fifo, &pkt, sizeof(pkt), NULL);
  186. av_packet_unref(&pkt);
  187. }
  188. av_fifo_freep(fifo);
  189. }
  190. static void stop_jack(JackData *self)
  191. {
  192. if (self->client) {
  193. if (self->activated)
  194. jack_deactivate(self->client);
  195. jack_client_close(self->client);
  196. }
  197. sem_destroy(&self->packet_count);
  198. free_pkt_fifo(&self->new_pkts);
  199. free_pkt_fifo(&self->filled_pkts);
  200. av_freep(&self->ports);
  201. ff_timefilter_destroy(self->timefilter);
  202. }
  203. static int audio_read_header(AVFormatContext *context)
  204. {
  205. JackData *self = context->priv_data;
  206. AVStream *stream;
  207. int test;
  208. if ((test = start_jack(context)))
  209. return test;
  210. stream = avformat_new_stream(context, NULL);
  211. if (!stream) {
  212. stop_jack(self);
  213. return AVERROR(ENOMEM);
  214. }
  215. stream->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
  216. #if HAVE_BIGENDIAN
  217. stream->codecpar->codec_id = AV_CODEC_ID_PCM_F32BE;
  218. #else
  219. stream->codecpar->codec_id = AV_CODEC_ID_PCM_F32LE;
  220. #endif
  221. stream->codecpar->sample_rate = self->sample_rate;
  222. stream->codecpar->channels = self->nports;
  223. avpriv_set_pts_info(stream, 64, 1, 1000000); /* 64 bits pts in us */
  224. return 0;
  225. }
  226. static int audio_read_packet(AVFormatContext *context, AVPacket *pkt)
  227. {
  228. JackData *self = context->priv_data;
  229. struct timespec timeout = {0, 0};
  230. int test;
  231. /* Activate the JACK client on first packet read. Activating the JACK client
  232. * means that process_callback() starts to get called at regular interval.
  233. * If we activate it in audio_read_header(), we're actually reading audio data
  234. * from the device before instructed to, and that may result in an overrun. */
  235. if (!self->activated) {
  236. if (!jack_activate(self->client)) {
  237. self->activated = 1;
  238. av_log(context, AV_LOG_INFO,
  239. "JACK client registered and activated (rate=%dHz, buffer_size=%d frames)\n",
  240. self->sample_rate, self->buffer_size);
  241. } else {
  242. av_log(context, AV_LOG_ERROR, "Unable to activate JACK client\n");
  243. return AVERROR(EIO);
  244. }
  245. }
  246. /* Wait for a packet coming back from process_callback(), if one isn't available yet */
  247. timeout.tv_sec = av_gettime() / 1000000 + 2;
  248. if (sem_timedwait(&self->packet_count, &timeout)) {
  249. if (errno == ETIMEDOUT) {
  250. av_log(context, AV_LOG_ERROR,
  251. "Input error: timed out when waiting for JACK process callback output\n");
  252. } else {
  253. char errbuf[128];
  254. int ret = AVERROR(errno);
  255. av_strerror(ret, errbuf, sizeof(errbuf));
  256. av_log(context, AV_LOG_ERROR, "Error while waiting for audio packet: %s\n",
  257. errbuf);
  258. }
  259. if (!self->client)
  260. av_log(context, AV_LOG_ERROR, "Input error: JACK server is gone\n");
  261. return AVERROR(EIO);
  262. }
  263. if (self->pkt_xrun) {
  264. av_log(context, AV_LOG_WARNING, "Audio packet xrun\n");
  265. self->pkt_xrun = 0;
  266. }
  267. if (self->jack_xrun) {
  268. av_log(context, AV_LOG_WARNING, "JACK xrun\n");
  269. self->jack_xrun = 0;
  270. }
  271. /* Retrieve the packet filled with audio data by process_callback() */
  272. av_fifo_generic_read(self->filled_pkts, pkt, sizeof(*pkt), NULL);
  273. if ((test = supply_new_packets(self, context)))
  274. return test;
  275. return 0;
  276. }
  277. static int audio_read_close(AVFormatContext *context)
  278. {
  279. JackData *self = context->priv_data;
  280. stop_jack(self);
  281. return 0;
  282. }
  283. #define OFFSET(x) offsetof(JackData, x)
  284. static const AVOption options[] = {
  285. { "channels", "Number of audio channels.", OFFSET(nports), AV_OPT_TYPE_INT, { .i64 = 2 }, 1, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
  286. { NULL },
  287. };
  288. static const AVClass jack_indev_class = {
  289. .class_name = "JACK indev",
  290. .item_name = av_default_item_name,
  291. .option = options,
  292. .version = LIBAVUTIL_VERSION_INT,
  293. .category = AV_CLASS_CATEGORY_DEVICE_AUDIO_INPUT,
  294. };
  295. AVInputFormat ff_jack_demuxer = {
  296. .name = "jack",
  297. .long_name = NULL_IF_CONFIG_SMALL("JACK Audio Connection Kit"),
  298. .priv_data_size = sizeof(JackData),
  299. .read_header = audio_read_header,
  300. .read_packet = audio_read_packet,
  301. .read_close = audio_read_close,
  302. .flags = AVFMT_NOFILE,
  303. .priv_class = &jack_indev_class,
  304. };