|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#include <inttypes.h> |
|
#include <stdint.h> |
|
#include <string.h> |
|
|
|
#include "avformat.h" |
|
#include "internal.h" |
|
#include "lrc.h" |
|
#include "metadata.h" |
|
#include "mux.h" |
|
#include "version.h" |
|
#include "libavutil/dict.h" |
|
#include "libavutil/log.h" |
|
#include "libavutil/macros.h" |
|
|
|
static int lrc_write_header(AVFormatContext *s) |
|
{ |
|
const AVDictionaryEntry *metadata_item; |
|
|
|
if(s->nb_streams != 1 || |
|
s->streams[0]->codecpar->codec_type != AVMEDIA_TYPE_SUBTITLE) { |
|
av_log(s, AV_LOG_ERROR, |
|
"LRC supports only a single subtitle stream.\n"); |
|
return AVERROR(EINVAL); |
|
} |
|
if(s->streams[0]->codecpar->codec_id != AV_CODEC_ID_SUBRIP && |
|
s->streams[0]->codecpar->codec_id != AV_CODEC_ID_TEXT) { |
|
av_log(s, AV_LOG_ERROR, "Unsupported subtitle codec: %s\n", |
|
avcodec_get_name(s->streams[0]->codecpar->codec_id)); |
|
return AVERROR(EINVAL); |
|
} |
|
avpriv_set_pts_info(s->streams[0], 64, 1, 100); |
|
|
|
ff_standardize_creation_time(s); |
|
ff_metadata_conv_ctx(s, ff_lrc_metadata_conv, NULL); |
|
if(!(s->flags & AVFMT_FLAG_BITEXACT)) { |
|
|
|
|
|
|
|
|
|
av_dict_set(&s->metadata, "ve", AV_STRINGIFY(LIBAVFORMAT_VERSION), 0); |
|
} else { |
|
av_dict_set(&s->metadata, "ve", NULL, 0); |
|
} |
|
for(metadata_item = NULL; |
|
(metadata_item = av_dict_iterate(s->metadata, metadata_item));) { |
|
char *delim; |
|
if(!metadata_item->value[0]) { |
|
continue; |
|
} |
|
while((delim = strchr(metadata_item->value, '\n'))) { |
|
*delim = ' '; |
|
} |
|
while((delim = strchr(metadata_item->value, '\r'))) { |
|
*delim = ' '; |
|
} |
|
avio_printf(s->pb, "[%s:%s]\n", |
|
metadata_item->key, metadata_item->value); |
|
} |
|
avio_w8(s->pb, '\n'); |
|
return 0; |
|
} |
|
|
|
static int lrc_write_packet(AVFormatContext *s, AVPacket *pkt) |
|
{ |
|
if(pkt->pts != AV_NOPTS_VALUE) { |
|
const uint8_t *line = pkt->data; |
|
const uint8_t *end = pkt->data + pkt->size; |
|
|
|
while (end > line && (end[-1] == '\n' || end[-1] == '\r')) |
|
end--; |
|
if (line != end) { |
|
while (line[0] == '\n' || line[0] == '\r') |
|
line++; |
|
} |
|
|
|
while(line) { |
|
const uint8_t *next_line = memchr(line, '\n', end - line); |
|
size_t size = end - line; |
|
|
|
if (next_line) { |
|
size = next_line - line; |
|
if (next_line > line && next_line[-1] == '\r') |
|
size--; |
|
next_line++; |
|
} |
|
if (size && line[0] == '[') { |
|
av_log(s, AV_LOG_WARNING, |
|
"Subtitle starts with '[', may cause problems with LRC format.\n"); |
|
} |
|
|
|
|
|
|
|
avio_write(s->pb, "[-", 1 + (pkt->pts < 0)); |
|
avio_printf(s->pb, "%02"PRIu64":%02"PRIu64".%02"PRIu64"]", |
|
(FFABS64U(pkt->pts) / 6000), |
|
((FFABS64U(pkt->pts) / 100) % 60), |
|
(FFABS64U(pkt->pts) % 100)); |
|
|
|
avio_write(s->pb, line, size); |
|
avio_w8(s->pb, '\n'); |
|
line = next_line; |
|
} |
|
} |
|
return 0; |
|
} |
|
|
|
const FFOutputFormat ff_lrc_muxer = { |
|
.p.name = "lrc", |
|
.p.long_name = NULL_IF_CONFIG_SMALL("LRC lyrics"), |
|
.p.extensions = "lrc", |
|
.p.flags = AVFMT_VARIABLE_FPS | AVFMT_GLOBALHEADER | |
|
AVFMT_TS_NEGATIVE | AVFMT_TS_NONSTRICT, |
|
.p.subtitle_codec = AV_CODEC_ID_SUBRIP, |
|
.priv_data_size = 0, |
|
.write_header = lrc_write_header, |
|
.write_packet = lrc_write_packet, |
|
}; |
|
|