yading@10: /* yading@10: * Copyright (c) 2011 Anton Khirnov yading@10: * yading@10: * This file is part of Libav. yading@10: * yading@10: * Libav is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * Libav is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with Libav; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: /** yading@10: * @file yading@10: * libcdio CD grabbing yading@10: */ yading@10: yading@10: #include "config.h" yading@10: yading@10: #if HAVE_CDIO_PARANOIA_H yading@10: #include yading@10: #include yading@10: #elif HAVE_CDIO_PARANOIA_PARANOIA_H yading@10: #include yading@10: #include yading@10: #endif yading@10: yading@10: #include "libavutil/log.h" yading@10: #include "libavutil/mem.h" yading@10: #include "libavutil/opt.h" yading@10: yading@10: #include "libavformat/avformat.h" yading@10: #include "libavformat/internal.h" yading@10: yading@10: typedef struct CDIOContext { yading@10: const AVClass *class; yading@10: cdrom_drive_t *drive; yading@10: cdrom_paranoia_t *paranoia; yading@10: int32_t last_sector; yading@10: yading@10: /* private options */ yading@10: int speed; yading@10: int paranoia_mode; yading@10: } CDIOContext; yading@10: yading@10: static av_cold int read_header(AVFormatContext *ctx) yading@10: { yading@10: CDIOContext *s = ctx->priv_data; yading@10: AVStream *st; yading@10: int ret, i; yading@10: char *err = NULL; yading@10: yading@10: if (!(st = avformat_new_stream(ctx, NULL))) yading@10: return AVERROR(ENOMEM); yading@10: s->drive = cdio_cddap_identify(ctx->filename, CDDA_MESSAGE_LOGIT, &err); yading@10: if (!s->drive) { yading@10: av_log(ctx, AV_LOG_ERROR, "Could not open drive %s.\n", ctx->filename); yading@10: return AVERROR(EINVAL); yading@10: } yading@10: if (err) { yading@10: av_log(ctx, AV_LOG_VERBOSE, "%s\n", err); yading@10: free(err); yading@10: } yading@10: if ((ret = cdio_cddap_open(s->drive)) < 0 || !s->drive->opened) { yading@10: av_log(ctx, AV_LOG_ERROR, "Could not open disk in drive %s.\n", ctx->filename); yading@10: return AVERROR(EINVAL); yading@10: } yading@10: yading@10: cdio_cddap_verbose_set(s->drive, CDDA_MESSAGE_LOGIT, CDDA_MESSAGE_LOGIT); yading@10: if (s->speed) yading@10: cdio_cddap_speed_set(s->drive, s->speed); yading@10: yading@10: s->paranoia = cdio_paranoia_init(s->drive); yading@10: if (!s->paranoia) { yading@10: av_log(ctx, AV_LOG_ERROR, "Could not init paranoia.\n"); yading@10: return AVERROR(EINVAL); yading@10: } yading@10: cdio_paranoia_modeset(s->paranoia, s->paranoia_mode); yading@10: yading@10: st->codec->codec_type = AVMEDIA_TYPE_AUDIO; yading@10: if (s->drive->bigendianp) yading@10: st->codec->codec_id = AV_CODEC_ID_PCM_S16BE; yading@10: else yading@10: st->codec->codec_id = AV_CODEC_ID_PCM_S16LE; yading@10: st->codec->sample_rate = 44100; yading@10: st->codec->channels = 2; yading@10: if (s->drive->audio_last_sector != CDIO_INVALID_LSN && yading@10: s->drive->audio_first_sector != CDIO_INVALID_LSN) yading@10: st->duration = s->drive->audio_last_sector - s->drive->audio_first_sector; yading@10: else if (s->drive->tracks) yading@10: st->duration = s->drive->disc_toc[s->drive->tracks].dwStartSector; yading@10: avpriv_set_pts_info(st, 64, CDIO_CD_FRAMESIZE_RAW, 2*st->codec->channels*st->codec->sample_rate); yading@10: yading@10: for (i = 0; i < s->drive->tracks; i++) { yading@10: char title[16]; yading@10: snprintf(title, sizeof(title), "track %02d", s->drive->disc_toc[i].bTrack); yading@10: avpriv_new_chapter(ctx, i, st->time_base, s->drive->disc_toc[i].dwStartSector, yading@10: s->drive->disc_toc[i+1].dwStartSector, title); yading@10: } yading@10: yading@10: s->last_sector = cdio_cddap_disc_lastsector(s->drive); yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static int read_packet(AVFormatContext *ctx, AVPacket *pkt) yading@10: { yading@10: CDIOContext *s = ctx->priv_data; yading@10: int ret; yading@10: uint16_t *buf; yading@10: char *err = NULL; yading@10: yading@10: if (ctx->streams[0]->cur_dts > s->last_sector) yading@10: return AVERROR_EOF; yading@10: yading@10: buf = cdio_paranoia_read(s->paranoia, NULL); yading@10: if (!buf) yading@10: return AVERROR_EOF; yading@10: yading@10: if (err = cdio_cddap_errors(s->drive)) { yading@10: av_log(ctx, AV_LOG_ERROR, "%s\n", err); yading@10: free(err); yading@10: err = NULL; yading@10: } yading@10: if (err = cdio_cddap_messages(s->drive)) { yading@10: av_log(ctx, AV_LOG_VERBOSE, "%s\n", err); yading@10: free(err); yading@10: err = NULL; yading@10: } yading@10: yading@10: if ((ret = av_new_packet(pkt, CDIO_CD_FRAMESIZE_RAW)) < 0) yading@10: return ret; yading@10: memcpy(pkt->data, buf, CDIO_CD_FRAMESIZE_RAW); yading@10: return 0; yading@10: } yading@10: yading@10: static av_cold int read_close(AVFormatContext *ctx) yading@10: { yading@10: CDIOContext *s = ctx->priv_data; yading@10: cdio_paranoia_free(s->paranoia); yading@10: cdio_cddap_close(s->drive); yading@10: return 0; yading@10: } yading@10: yading@10: static int read_seek(AVFormatContext *ctx, int stream_index, int64_t timestamp, yading@10: int flags) yading@10: { yading@10: CDIOContext *s = ctx->priv_data; yading@10: AVStream *st = ctx->streams[0]; yading@10: yading@10: cdio_paranoia_seek(s->paranoia, timestamp, SEEK_SET); yading@10: st->cur_dts = timestamp; yading@10: return 0; yading@10: } yading@10: yading@10: #define OFFSET(x) offsetof(CDIOContext, x) yading@10: #define DEC AV_OPT_FLAG_DECODING_PARAM yading@10: static const AVOption options[] = { yading@10: { "speed", "Drive reading speed.", OFFSET(speed), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, DEC }, yading@10: { "paranoia_mode", "Error recovery mode.", OFFSET(paranoia_mode), AV_OPT_TYPE_FLAGS, { .i64 = 0 }, INT_MIN, INT_MAX, DEC, "paranoia_mode" }, yading@10: { "verify", "Verify data integrity in overlap area", 0, AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_VERIFY }, 0, 0, DEC, "paranoia_mode" }, yading@10: { "overlap", "Perform overlapped reads.", 0, AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_OVERLAP }, 0, 0, DEC, "paranoia_mode" }, yading@10: { "neverskip", "Do not skip failed reads.", 0, AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_NEVERSKIP }, 0, 0, DEC, "paranoia_mode" }, yading@10: { NULL }, yading@10: }; yading@10: yading@10: static const AVClass libcdio_class = { yading@10: .class_name = "libcdio indev", yading@10: .item_name = av_default_item_name, yading@10: .option = options, yading@10: .version = LIBAVUTIL_VERSION_INT, yading@10: }; yading@10: yading@10: AVInputFormat ff_libcdio_demuxer = { yading@10: .name = "libcdio", yading@10: .read_header = read_header, yading@10: .read_packet = read_packet, yading@10: .read_close = read_close, yading@10: .read_seek = read_seek, yading@10: .priv_data_size = sizeof(CDIOContext), yading@10: .flags = AVFMT_NOFILE, yading@10: .priv_class = &libcdio_class, yading@10: };