summaryrefslogtreecommitdiff
path: root/libavformat/libopenmpt.c (plain)
blob: a7e385959a6f18a5d6ed8eb56c3737f1aeccc0bc
1/*
2 * Tracked MOD demuxer (libopenmpt)
3 * Copyright (c) 2016 Josh de Kock
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 <libopenmpt/libopenmpt.h>
23#include <libopenmpt/libopenmpt_stream_callbacks_file.h>
24
25#include "libavutil/avstring.h"
26#include "libavutil/opt.h"
27#include "avformat.h"
28#include "internal.h"
29
30typedef struct OpenMPTContext {
31 const AVClass *class;
32 openmpt_module *module;
33
34 int channels;
35 double duration;
36 /* options */
37 int sample_rate;
38 int64_t layout;
39 int subsong;
40} OpenMPTContext;
41
42#define OFFSET(x) offsetof(OpenMPTContext, x)
43#define A AV_OPT_FLAG_AUDIO_PARAM
44#define D AV_OPT_FLAG_DECODING_PARAM
45static const AVOption options[] = {
46 { "sample_rate", "set sample rate", OFFSET(sample_rate), AV_OPT_TYPE_INT, { .i64 = 48000 }, 1000, INT_MAX, A | D },
47 { "layout", "set channel layout", OFFSET(layout), AV_OPT_TYPE_CHANNEL_LAYOUT, { .i64 = AV_CH_LAYOUT_STEREO }, 0, INT64_MAX, A | D },
48 { "subsong", "set subsong", OFFSET(subsong), AV_OPT_TYPE_INT, { .i64 = -2 }, -2, INT_MAX, A | D, "subsong"},
49 { "all", "all", 0, AV_OPT_TYPE_CONST, { .i64 = -1}, 0, 0, A | D, "subsong" },
50 { "auto", "auto", 0, AV_OPT_TYPE_CONST, { .i64 = -2}, 0, 0, A | D, "subsong" },
51 { NULL }
52};
53
54static void openmpt_logfunc(const char *message, void *userdata)
55{
56 int level = AV_LOG_INFO;
57 if (strstr(message, "ERROR") != NULL) {
58 level = AV_LOG_ERROR;
59 }
60 av_log(userdata, level, "%s\n", message);
61}
62
63#define add_meta(s, name, meta) \
64do { \
65 const char *value = meta; \
66 if (value && value[0]) \
67 av_dict_set(&s->metadata, name, value, 0); \
68 openmpt_free_string(value); \
69} while(0)
70
71static int read_header_openmpt(AVFormatContext *s)
72{
73 AVStream *st;
74 OpenMPTContext *openmpt = s->priv_data;
75 int64_t size = avio_size(s->pb);
76 if (size <= 0)
77 return AVERROR_INVALIDDATA;
78 char *buf = av_malloc(size);
79 int ret;
80
81
82 if (!buf)
83 return AVERROR(ENOMEM);
84 size = avio_read(s->pb, buf, size);
85 if (size < 0) {
86 av_log(s, AV_LOG_ERROR, "Reading input buffer failed.\n");
87 av_freep(&buf);
88 return size;
89 }
90
91 openmpt->module = openmpt_module_create_from_memory(buf, size, openmpt_logfunc, s, NULL);
92 av_freep(&buf);
93 if (!openmpt->module)
94 return AVERROR_INVALIDDATA;
95
96 openmpt->channels = av_get_channel_layout_nb_channels(openmpt->layout);
97 openmpt->duration = openmpt_module_get_duration_seconds(openmpt->module);
98
99 add_meta(s, "artist", openmpt_module_get_metadata(openmpt->module, "artist"));
100 add_meta(s, "title", openmpt_module_get_metadata(openmpt->module, "title"));
101 add_meta(s, "encoder", openmpt_module_get_metadata(openmpt->module, "tracker"));
102 add_meta(s, "comment", openmpt_module_get_metadata(openmpt->module, "message"));
103 add_meta(s, "date", openmpt_module_get_metadata(openmpt->module, "date"));
104
105 if (openmpt->subsong >= openmpt_module_get_num_subsongs(openmpt->module)) {
106 openmpt_module_destroy(openmpt->module);
107 av_log(s, AV_LOG_ERROR, "Invalid subsong index: %d\n", openmpt->subsong);
108 return AVERROR(EINVAL);
109 }
110
111 if (openmpt->subsong != -2) {
112 if (openmpt->subsong >= 0) {
113 av_dict_set_int(&s->metadata, "track", openmpt->subsong + 1, 0);
114 }
115 ret = openmpt_module_select_subsong(openmpt->module, openmpt->subsong);
116 if (!ret){
117 openmpt_module_destroy(openmpt->module);
118 av_log(s, AV_LOG_ERROR, "Could not select requested subsong: %d", openmpt->subsong);
119 return AVERROR(EINVAL);
120 }
121 }
122
123 st = avformat_new_stream(s, NULL);
124 if (!st) {
125 openmpt_module_destroy(openmpt->module);
126 openmpt->module = NULL;
127 return AVERROR(ENOMEM);
128 }
129 avpriv_set_pts_info(st, 64, 1, AV_TIME_BASE);
130 st->duration = llrint(openmpt->duration*AV_TIME_BASE);
131
132 st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
133 st->codecpar->codec_id = AV_NE(AV_CODEC_ID_PCM_F32BE, AV_CODEC_ID_PCM_F32LE);
134 st->codecpar->channels = openmpt->channels;
135 st->codecpar->sample_rate = openmpt->sample_rate;
136
137 return 0;
138}
139
140#define AUDIO_PKT_SIZE 2048
141
142static int read_packet_openmpt(AVFormatContext *s, AVPacket *pkt)
143{
144 OpenMPTContext *openmpt = s->priv_data;
145 int n_samples = AUDIO_PKT_SIZE / (openmpt->channels ? openmpt->channels*4 : 4);
146 int ret;
147
148 if ((ret = av_new_packet(pkt, AUDIO_PKT_SIZE)) < 0)
149 return ret;
150
151 switch (openmpt->channels) {
152 case 1:
153 ret = openmpt_module_read_float_mono(openmpt->module, openmpt->sample_rate,
154 n_samples, (float *)pkt->data);
155 break;
156 case 2:
157 ret = openmpt_module_read_interleaved_float_stereo(openmpt->module, openmpt->sample_rate,
158 n_samples, (float *)pkt->data);
159 break;
160 case 4:
161 ret = openmpt_module_read_interleaved_float_quad(openmpt->module, openmpt->sample_rate,
162 n_samples, (float *)pkt->data);
163 break;
164 default:
165 av_log(s, AV_LOG_ERROR, "Unsupported number of channels: %d", openmpt->channels);
166 return AVERROR(EINVAL);
167 }
168
169 if (ret < 1) {
170 pkt->size = 0;
171 return AVERROR_EOF;
172 }
173
174 pkt->size = ret * (openmpt->channels * 4);
175
176 return 0;
177}
178
179static int read_close_openmpt(AVFormatContext *s)
180{
181 OpenMPTContext *openmpt = s->priv_data;
182 openmpt_module_destroy(openmpt->module);
183 openmpt->module = NULL;
184 return 0;
185}
186
187static int read_seek_openmpt(AVFormatContext *s, int stream_idx, int64_t ts, int flags)
188{
189 OpenMPTContext *openmpt = s->priv_data;
190 openmpt_module_set_position_seconds(openmpt->module, (double)ts/AV_TIME_BASE);
191 return 0;
192}
193
194static const AVClass class_openmpt = {
195 .class_name = "libopenmpt",
196 .item_name = av_default_item_name,
197 .option = options,
198 .version = LIBAVUTIL_VERSION_INT,
199};
200
201AVInputFormat ff_libopenmpt_demuxer = {
202 .name = "libopenmpt",
203 .long_name = NULL_IF_CONFIG_SMALL("Tracker formats (libopenmpt)"),
204 .priv_data_size = sizeof(OpenMPTContext),
205 .read_header = read_header_openmpt,
206 .read_packet = read_packet_openmpt,
207 .read_close = read_close_openmpt,
208 .read_seek = read_seek_openmpt,
209 .priv_class = &class_openmpt,
210 .extensions = "669,amf,ams,dbm,digi,dmf,dsm,far,gdm,imf,it,j2b,m15,mdl,med,mmcmp,mms,mo3,mod,mptm,mt2,mtm,nst,okt,plm,ppm,psm,pt36,ptm,s3m,sfx,sfx2,stk,stm,ult,umx,wow,xm,xpk",
211};
212
generated by cgit at 2024-05-17 05:01:39 (GMT)