2 * Copyright (c) 2011 Anton Khirnov <anton@khirnov.net>
4 * This file is part of FFmpeg.
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.
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.
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
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>
36 #include "libavutil/log.h"
37 #include "libavutil/mem.h"
38 #include "libavutil/opt.h"
40 #include "libavformat/avformat.h"
41 #include "libavformat/internal.h"
43 typedef struct CDIOContext
{
46 cdrom_paranoia_t
*paranoia
;
54 static av_cold
int read_header(AVFormatContext
*ctx
)
56 CDIOContext
*s
= ctx
->priv_data
;
61 if (!(st
= avformat_new_stream(ctx
, NULL
)))
62 return AVERROR(ENOMEM
);
63 s
->drive
= cdio_cddap_identify(ctx
->filename
, CDDA_MESSAGE_LOGIT
, &err
);
65 av_log(ctx
, AV_LOG_ERROR
, "Could not open drive %s.\n", ctx
->filename
);
66 return AVERROR(EINVAL
);
69 av_log(ctx
, AV_LOG_VERBOSE
, "%s\n", err
);
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
);
77 cdio_cddap_verbose_set(s
->drive
, CDDA_MESSAGE_LOGIT
, CDDA_MESSAGE_LOGIT
);
79 cdio_cddap_speed_set(s
->drive
, s
->speed
);
81 s
->paranoia
= cdio_paranoia_init(s
->drive
);
83 av_log(ctx
, AV_LOG_ERROR
, "Could not init paranoia.\n");
84 return AVERROR(EINVAL
);
86 cdio_paranoia_modeset(s
->paranoia
, s
->paranoia_mode
);
88 st
->codec
->codec_type
= AVMEDIA_TYPE_AUDIO
;
89 if (s
->drive
->bigendianp
)
90 st
->codec
->codec_id
= AV_CODEC_ID_PCM_S16BE
;
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
);
102 for (i
= 0; i
< s
->drive
->tracks
; i
++) {
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
);
109 s
->last_sector
= cdio_cddap_disc_lastsector(s
->drive
);
114 static int read_packet(AVFormatContext
*ctx
, AVPacket
*pkt
)
116 CDIOContext
*s
= ctx
->priv_data
;
121 if (ctx
->streams
[0]->cur_dts
> s
->last_sector
)
124 buf
= cdio_paranoia_read(s
->paranoia
, NULL
);
128 if (err
= cdio_cddap_errors(s
->drive
)) {
129 av_log(ctx
, AV_LOG_ERROR
, "%s\n", err
);
133 if (err
= cdio_cddap_messages(s
->drive
)) {
134 av_log(ctx
, AV_LOG_VERBOSE
, "%s\n", err
);
139 if ((ret
= av_new_packet(pkt
, CDIO_CD_FRAMESIZE_RAW
)) < 0)
141 memcpy(pkt
->data
, buf
, CDIO_CD_FRAMESIZE_RAW
);
145 static av_cold
int read_close(AVFormatContext
*ctx
)
147 CDIOContext
*s
= ctx
->priv_data
;
148 cdio_paranoia_free(s
->paranoia
);
149 cdio_cddap_close(s
->drive
);
153 static int read_seek(AVFormatContext
*ctx
, int stream_index
, int64_t timestamp
,
156 CDIOContext
*s
= ctx
->priv_data
;
157 AVStream
*st
= ctx
->streams
[0];
159 cdio_paranoia_seek(s
->paranoia
, timestamp
, SEEK_SET
);
160 st
->cur_dts
= timestamp
;
164 #define OFFSET(x) offsetof(CDIOContext, x)
165 #define DEC AV_OPT_FLAG_DECODING_PARAM
166 static 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" },
175 static const AVClass libcdio_class
= {
176 .class_name
= "libcdio indev",
177 .item_name
= av_default_item_name
,
179 .version
= LIBAVUTIL_VERSION_INT
,
180 .category
= AV_CLASS_CATEGORY_DEVICE_AUDIO_INPUT
,
183 AVInputFormat ff_libcdio_demuxer
= {
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
,