libavdevice/sndio_enc.c

   1 /*
   2  * sndio play and grab interface
   3  * Copyright (c) 2010 Jacob Meuser
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 #include <stdint.h>
  23 #include <sndio.h>
  24
  25 #include "libavutil/internal.h"
  26
  27 #include "libavformat/mux.h"
  28
  29 #include "libavdevice/avdevice.h"
  30 #include "libavdevice/sndio.h"
  31
  32 static av_cold int audio_write_header(AVFormatContext *s1)
  33 {
  34     SndioData *s = s1->priv_data;
  35     AVStream *st;
  36     int ret;
  37
  38     st             = s1->streams[0];
  39     s->sample_rate = st->codecpar->sample_rate;
  40     s->channels    = st->codecpar->ch_layout.nb_channels;
  41
  42     ret = ff_sndio_open(s1, 1, s1->url);
  43
  44     return ret;
  45 }
  46
  47 static int audio_write_packet(AVFormatContext *s1, AVPacket *pkt)
  48 {
  49     SndioData *s = s1->priv_data;
  50     const uint8_t *buf = pkt->data;
  51     int size = pkt->size;
  52     int len, ret;
  53
  54     while (size > 0) {
  55         len = FFMIN(s->buffer_size - s->buffer_offset, size);
  56         memcpy(s->buffer + s->buffer_offset, buf, len);
  57         buf  += len;
  58         size -= len;
  59         s->buffer_offset += len;
  60         if (s->buffer_offset >= s->buffer_size) {
  61             ret = sio_write(s->hdl, s->buffer, s->buffer_size);
  62             if (ret == 0 || sio_eof(s->hdl))
  63                 return AVERROR(EIO);
  64             s->softpos      += ret;
  65             s->buffer_offset = 0;
  66         }
  67     }
  68
  69     return 0;
  70 }
  71
  72 static int audio_write_trailer(AVFormatContext *s1)
  73 {
  74     SndioData *s = s1->priv_data;
  75
  76     sio_write(s->hdl, s->buffer, s->buffer_offset);
  77
  78     ff_sndio_close(s);
  79
  80     return 0;
  81 }
  82
  83 static const AVClass sndio_muxer_class = {
  84     .class_name     = "sndio outdev",
  85     .item_name      = av_default_item_name,
  86     .version        = LIBAVUTIL_VERSION_INT,
  87     .category       = AV_CLASS_CATEGORY_DEVICE_AUDIO_OUTPUT,
  88 };
  89
  90 const FFOutputFormat ff_sndio_muxer = {
  91     .p.name         = "sndio",
  92     .p.long_name    = NULL_IF_CONFIG_SMALL("sndio audio playback"),
  93     .priv_data_size = sizeof(SndioData),
  94     /* XXX: we make the assumption that the soundcard accepts this format */
  95     /* XXX: find better solution with "preinit" method, needed also in
  96        other formats */
  97     .p.audio_codec  = AV_NE(AV_CODEC_ID_PCM_S16BE, AV_CODEC_ID_PCM_S16LE),
  98     .p.video_codec  = AV_CODEC_ID_NONE,
  99     .write_header   = audio_write_header,
 100     .write_packet   = audio_write_packet,
 101     .write_trailer  = audio_write_trailer,
 102     .p.flags        = AVFMT_NOFILE,
 103     .p.priv_class   = &sndio_muxer_class,
 104 };