Imported Debian version 2.5.0~trusty1.1
[deb_ffmpeg.git] / ffmpeg / libavdevice / libcdio.c
CommitLineData
2ba45a60
DM
1/*
2 * Copyright (c) 2011 Anton Khirnov <anton@khirnov.net>
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21/**
22 * @file
23 * libcdio CD grabbing
24 */
25
26#include "config.h"
27
28#if HAVE_CDIO_PARANOIA_H
29#include <cdio/cdda.h>
30#include <cdio/paranoia.h>
31#elif HAVE_CDIO_PARANOIA_PARANOIA_H
32#include <cdio/paranoia/cdda.h>
33#include <cdio/paranoia/paranoia.h>
34#endif
35
36#include "libavutil/log.h"
37#include "libavutil/mem.h"
38#include "libavutil/opt.h"
39
40#include "libavformat/avformat.h"
41#include "libavformat/internal.h"
42
43typedef struct CDIOContext {
44 const AVClass *class;
45 cdrom_drive_t *drive;
46 cdrom_paranoia_t *paranoia;
47 int32_t last_sector;
48
49 /* private options */
50 int speed;
51 int paranoia_mode;
52} CDIOContext;
53
54static av_cold int read_header(AVFormatContext *ctx)
55{
56 CDIOContext *s = ctx->priv_data;
57 AVStream *st;
58 int ret, i;
59 char *err = NULL;
60
61 if (!(st = avformat_new_stream(ctx, NULL)))
62 return AVERROR(ENOMEM);
63 s->drive = cdio_cddap_identify(ctx->filename, CDDA_MESSAGE_LOGIT, &err);
64 if (!s->drive) {
65 av_log(ctx, AV_LOG_ERROR, "Could not open drive %s.\n", ctx->filename);
66 return AVERROR(EINVAL);
67 }
68 if (err) {
69 av_log(ctx, AV_LOG_VERBOSE, "%s\n", err);
70 free(err);
71 }
72 if ((ret = cdio_cddap_open(s->drive)) < 0 || !s->drive->opened) {
73 av_log(ctx, AV_LOG_ERROR, "Could not open disk in drive %s.\n", ctx->filename);
74 return AVERROR(EINVAL);
75 }
76
77 cdio_cddap_verbose_set(s->drive, CDDA_MESSAGE_LOGIT, CDDA_MESSAGE_LOGIT);
78 if (s->speed)
79 cdio_cddap_speed_set(s->drive, s->speed);
80
81 s->paranoia = cdio_paranoia_init(s->drive);
82 if (!s->paranoia) {
83 av_log(ctx, AV_LOG_ERROR, "Could not init paranoia.\n");
84 return AVERROR(EINVAL);
85 }
86 cdio_paranoia_modeset(s->paranoia, s->paranoia_mode);
87
88 st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
89 if (s->drive->bigendianp)
90 st->codec->codec_id = AV_CODEC_ID_PCM_S16BE;
91 else
92 st->codec->codec_id = AV_CODEC_ID_PCM_S16LE;
93 st->codec->sample_rate = 44100;
94 st->codec->channels = 2;
95 if (s->drive->audio_last_sector != CDIO_INVALID_LSN &&
96 s->drive->audio_first_sector != CDIO_INVALID_LSN)
97 st->duration = s->drive->audio_last_sector - s->drive->audio_first_sector;
98 else if (s->drive->tracks)
99 st->duration = s->drive->disc_toc[s->drive->tracks].dwStartSector;
100 avpriv_set_pts_info(st, 64, CDIO_CD_FRAMESIZE_RAW, 2*st->codec->channels*st->codec->sample_rate);
101
102 for (i = 0; i < s->drive->tracks; i++) {
103 char title[16];
104 snprintf(title, sizeof(title), "track %02d", s->drive->disc_toc[i].bTrack);
105 avpriv_new_chapter(ctx, i, st->time_base, s->drive->disc_toc[i].dwStartSector,
106 s->drive->disc_toc[i+1].dwStartSector, title);
107 }
108
109 s->last_sector = cdio_cddap_disc_lastsector(s->drive);
110
111 return 0;
112}
113
114static int read_packet(AVFormatContext *ctx, AVPacket *pkt)
115{
116 CDIOContext *s = ctx->priv_data;
117 int ret;
118 uint16_t *buf;
119 char *err = NULL;
120
121 if (ctx->streams[0]->cur_dts > s->last_sector)
122 return AVERROR_EOF;
123
124 buf = cdio_paranoia_read(s->paranoia, NULL);
125 if (!buf)
126 return AVERROR_EOF;
127
128 if (err = cdio_cddap_errors(s->drive)) {
129 av_log(ctx, AV_LOG_ERROR, "%s\n", err);
130 free(err);
131 err = NULL;
132 }
133 if (err = cdio_cddap_messages(s->drive)) {
134 av_log(ctx, AV_LOG_VERBOSE, "%s\n", err);
135 free(err);
136 err = NULL;
137 }
138
139 if ((ret = av_new_packet(pkt, CDIO_CD_FRAMESIZE_RAW)) < 0)
140 return ret;
141 memcpy(pkt->data, buf, CDIO_CD_FRAMESIZE_RAW);
142 return 0;
143}
144
145static av_cold int read_close(AVFormatContext *ctx)
146{
147 CDIOContext *s = ctx->priv_data;
148 cdio_paranoia_free(s->paranoia);
149 cdio_cddap_close(s->drive);
150 return 0;
151}
152
153static int read_seek(AVFormatContext *ctx, int stream_index, int64_t timestamp,
154 int flags)
155{
156 CDIOContext *s = ctx->priv_data;
157 AVStream *st = ctx->streams[0];
158
159 cdio_paranoia_seek(s->paranoia, timestamp, SEEK_SET);
160 st->cur_dts = timestamp;
161 return 0;
162}
163
164#define OFFSET(x) offsetof(CDIOContext, x)
165#define DEC AV_OPT_FLAG_DECODING_PARAM
166static const AVOption options[] = {
167 { "speed", "Drive reading speed.", OFFSET(speed), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, DEC },
168 { "paranoia_mode", "Error recovery mode.", OFFSET(paranoia_mode), AV_OPT_TYPE_FLAGS, { .i64 = 0 }, INT_MIN, INT_MAX, DEC, "paranoia_mode" },
169 { "verify", "Verify data integrity in overlap area", 0, AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_VERIFY }, 0, 0, DEC, "paranoia_mode" },
170 { "overlap", "Perform overlapped reads.", 0, AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_OVERLAP }, 0, 0, DEC, "paranoia_mode" },
171 { "neverskip", "Do not skip failed reads.", 0, AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_NEVERSKIP }, 0, 0, DEC, "paranoia_mode" },
172 { NULL },
173};
174
175static const AVClass libcdio_class = {
176 .class_name = "libcdio indev",
177 .item_name = av_default_item_name,
178 .option = options,
179 .version = LIBAVUTIL_VERSION_INT,
180 .category = AV_CLASS_CATEGORY_DEVICE_AUDIO_INPUT,
181};
182
183AVInputFormat ff_libcdio_demuxer = {
184 .name = "libcdio",
185 .read_header = read_header,
186 .read_packet = read_packet,
187 .read_close = read_close,
188 .read_seek = read_seek,
189 .priv_data_size = sizeof(CDIOContext),
190 .flags = AVFMT_NOFILE,
191 .priv_class = &libcdio_class,
192};