yading@11: /*
yading@11:  * SBG (SBaGen) file format decoder
yading@11:  * Copyright (c) 2011 Nicolas George
yading@11:  *
yading@11:  * This file is part of FFmpeg.
yading@11:  *
yading@11:  * FFmpeg is free software; you can redistribute it and/or
yading@11:  * modify it under the terms of the GNU Lesser General Public
yading@11:  * License as published by the Free Software Foundation; either
yading@11:  * version 2.1 of the License, or (at your option) any later version.
yading@11:  *
yading@11:  * FFmpeg is distributed in the hope that it will be useful,
yading@11:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@11:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
yading@11:  * Lesser General Public License for more details.
yading@11:  *
yading@11:  * You should have received a copy of the GNU Lesser General Public
yading@11:  * License along with FFmpeg; if not, write to the Free Software
yading@11:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@11:  */
yading@11: 
yading@11: #include <stdio.h>
yading@11: #include <stdlib.h>
yading@11: #include <time.h>
yading@11: #include "libavutil/intreadwrite.h"
yading@11: #include "libavutil/log.h"
yading@11: #include "libavutil/opt.h"
yading@11: #include "avformat.h"
yading@11: #include "internal.h"
yading@11: 
yading@11: #define SBG_SCALE (1 << 16)
yading@11: #define DAY (24 * 60 * 60)
yading@11: #define DAY_TS ((int64_t)DAY * AV_TIME_BASE)
yading@11: 
yading@11: struct sbg_demuxer {
yading@11:     AVClass *class;
yading@11:     int sample_rate;
yading@11:     int frame_size;
yading@11:     int max_file_size;
yading@11: };
yading@11: 
yading@11: struct sbg_string {
yading@11:     char *s;
yading@11:     char *e;
yading@11: };
yading@11: 
yading@11: enum sbg_fade_type {
yading@11:     SBG_FADE_SILENCE = 0,
yading@11:     SBG_FADE_SAME    = 1,
yading@11:     SBG_FADE_ADAPT   = 3,
yading@11: };
yading@11: 
yading@11: struct sbg_fade {
yading@11:     int8_t in, out, slide;
yading@11: };
yading@11: 
yading@11: enum sbg_synth_type {
yading@11:     SBG_TYPE_NONE,
yading@11:     SBG_TYPE_SINE,
yading@11:     SBG_TYPE_NOISE,
yading@11:     SBG_TYPE_BELL,
yading@11:     SBG_TYPE_MIX,
yading@11:     SBG_TYPE_SPIN,
yading@11: };
yading@11: 
yading@11: /* bell: freq constant, ampl decreases exponentially, can be approx lin */
yading@11: 
yading@11: struct sbg_timestamp {
yading@11:     int64_t t;
yading@11:     char type; /* 0 for relative, 'N' for now, 'T' for absolute */
yading@11: };
yading@11: 
yading@11: struct sbg_script_definition {
yading@11:     char *name;
yading@11:     int name_len;
yading@11:     int elements, nb_elements;
yading@11:     char type; /* 'S' or 'B' */
yading@11: };
yading@11: 
yading@11: struct sbg_script_synth {
yading@11:     int carrier;
yading@11:     int beat;
yading@11:     int vol;
yading@11:     enum sbg_synth_type type;
yading@11:     struct {
yading@11:         int l, r;
yading@11:     } ref;
yading@11: };
yading@11: 
yading@11: struct sbg_script_tseq {
yading@11:     struct sbg_timestamp ts;
yading@11:     char *name;
yading@11:     int name_len;
yading@11:     int lock;
yading@11:     struct sbg_fade fade;
yading@11: };
yading@11: 
yading@11: struct sbg_script_event {
yading@11:     int64_t ts;
yading@11:     int64_t ts_int, ts_trans, ts_next;
yading@11:     int elements, nb_elements;
yading@11:     struct sbg_fade fade;
yading@11: };
yading@11: 
yading@11: struct sbg_script {
yading@11:     struct sbg_script_definition *def;
yading@11:     struct sbg_script_synth *synth;
yading@11:     struct sbg_script_tseq *tseq;
yading@11:     struct sbg_script_tseq *block_tseq;
yading@11:     struct sbg_script_event *events;
yading@11:     int nb_def;
yading@11:     int nb_tseq;
yading@11:     int nb_events;
yading@11:     int nb_synth;
yading@11:     int64_t start_ts;
yading@11:     int64_t end_ts;
yading@11:     int64_t opt_fade_time;
yading@11:     int64_t opt_duration;
yading@11:     char *opt_mix;
yading@11:     int sample_rate;
yading@11:     uint8_t opt_start_at_first;
yading@11:     uint8_t opt_end_at_last;
yading@11: };
yading@11: 
yading@11: struct sbg_parser {
yading@11:     void *log;
yading@11:     char *script, *end;
yading@11:     char *cursor;
yading@11:     struct sbg_script scs;
yading@11:     struct sbg_timestamp current_time;
yading@11:     int nb_block_tseq;
yading@11:     int nb_def_max, nb_synth_max, nb_tseq_max, nb_block_tseq_max;
yading@11:     int line_no;
yading@11:     char err_msg[128];
yading@11: };
yading@11: 
yading@11: enum ws_interval_type {
yading@11:     WS_SINE  = MKTAG('S','I','N','E'),
yading@11:     WS_NOISE = MKTAG('N','O','I','S'),
yading@11: };
yading@11: 
yading@11: struct ws_interval {
yading@11:     int64_t ts1, ts2;
yading@11:     enum ws_interval_type type;
yading@11:     uint32_t channels;
yading@11:     int32_t f1, f2;
yading@11:     int32_t a1, a2;
yading@11:     uint32_t phi;
yading@11: };
yading@11: 
yading@11: struct ws_intervals {
yading@11:     struct ws_interval *inter;
yading@11:     int nb_inter;
yading@11:     int max_inter;
yading@11: };
yading@11: 
yading@11: static void *alloc_array_elem(void **array, size_t elsize,
yading@11:                               int *size, int *max_size)
yading@11: {
yading@11:     void *ret;
yading@11: 
yading@11:     if (*size == *max_size) {
yading@11:         int m = FFMAX(32, FFMIN(*max_size, INT_MAX / 2) * 2);
yading@11:         if (*size >= m)
yading@11:             return NULL;
yading@11:         *array = av_realloc_f(*array, m, elsize);
yading@11:         if (!*array)
yading@11:             return NULL;
yading@11:         *max_size = m;
yading@11:     }
yading@11:     ret = (char *)*array + elsize * *size;
yading@11:     memset(ret, 0, elsize);
yading@11:     (*size)++;
yading@11:     return ret;
yading@11: }
yading@11: 
yading@11: static int str_to_time(const char *str, int64_t *rtime)
yading@11: {
yading@11:     const char *cur = str;
yading@11:     char *end;
yading@11:     int hours, minutes;
yading@11:     double seconds = 0;
yading@11: 
yading@11:     if (*cur < '0' || *cur > '9')
yading@11:         return 0;
yading@11:     hours = strtol(cur, &end, 10);
yading@11:     if (end == cur || *end != ':' || end[1] < '0' || end[1] > '9')
yading@11:         return 0;
yading@11:     cur = end + 1;
yading@11:     minutes = strtol(cur, &end, 10);
yading@11:     if (end == cur)
yading@11:         return 0;
yading@11:     cur = end;
yading@11:     if (*end == ':'){
yading@11:         seconds = strtod(cur + 1, &end);
yading@11:         if (end > cur + 1)
yading@11:             cur = end;
yading@11:     }
yading@11:     *rtime = (hours * 3600 + minutes * 60 + seconds) * AV_TIME_BASE;
yading@11:     return cur - str;
yading@11: }
yading@11: 
yading@11: static inline int is_space(char c)
yading@11: {
yading@11:     return c == ' '  || c == '\t' || c == '\r';
yading@11: }
yading@11: 
yading@11: static inline int scale_double(void *log, double d, double m, int *r)
yading@11: {
yading@11:     m *= d * SBG_SCALE;
yading@11:     if (m < INT_MIN || m >= INT_MAX) {
yading@11:         if (log)
yading@11:             av_log(log, AV_LOG_ERROR, "%g is too large\n", d);
yading@11:         return AVERROR(EDOM);
yading@11:     }
yading@11:     *r = m;
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int lex_space(struct sbg_parser *p)
yading@11: {
yading@11:     char *c = p->cursor;
yading@11: 
yading@11:     while (p->cursor < p->end && is_space(*p->cursor))
yading@11:         p->cursor++;
yading@11:     return p->cursor > c;
yading@11: }
yading@11: 
yading@11: static int lex_char(struct sbg_parser *p, char c)
yading@11: {
yading@11:     int r = p->cursor < p->end && *p->cursor == c;
yading@11: 
yading@11:     p->cursor += r;
yading@11:     return r;
yading@11: }
yading@11: 
yading@11: static int lex_double(struct sbg_parser *p, double *r)
yading@11: {
yading@11:     double d;
yading@11:     char *end;
yading@11: 
yading@11:     if (p->cursor == p->end || is_space(*p->cursor) || *p->cursor == '\n')
yading@11:         return 0;
yading@11:     d = strtod(p->cursor, &end);
yading@11:     if (end > p->cursor) {
yading@11:         *r = d;
yading@11:         p->cursor = end;
yading@11:         return 1;
yading@11:     }
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int lex_fixed(struct sbg_parser *p, const char *t, int l)
yading@11: {
yading@11:     if (p->end - p->cursor < l || memcmp(p->cursor, t, l))
yading@11:         return 0;
yading@11:     p->cursor += l;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int lex_line_end(struct sbg_parser *p)
yading@11: {
yading@11:     if (p->cursor < p->end && *p->cursor == '#') {
yading@11:         p->cursor++;
yading@11:         while (p->cursor < p->end && *p->cursor != '\n')
yading@11:             p->cursor++;
yading@11:     }
yading@11:     if (p->cursor == p->end)
yading@11:         /* simulate final LF for files lacking it */
yading@11:         return 1;
yading@11:     if (*p->cursor != '\n')
yading@11:         return 0;
yading@11:     p->cursor++;
yading@11:     p->line_no++;
yading@11:     lex_space(p);
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int lex_wsword(struct sbg_parser *p, struct sbg_string *rs)
yading@11: {
yading@11:     char *s = p->cursor, *c = s;
yading@11: 
yading@11:     if (s == p->end || *s == '\n')
yading@11:         return 0;
yading@11:     while (c < p->end && *c != '\n' && !is_space(*c))
yading@11:         c++;
yading@11:     rs->s = s;
yading@11:     rs->e = p->cursor = c;
yading@11:     lex_space(p);
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int lex_name(struct sbg_parser *p, struct sbg_string *rs)
yading@11: {
yading@11:     char *s = p->cursor, *c = s;
yading@11: 
yading@11:     while (c < p->end && ((*c >= 'a' && *c <= 'z') || (*c >= 'A' && *c <= 'Z')
yading@11:            || (*c >= '0' && *c <= '9') || *c == '_' || *c == '-'))
yading@11:         c++;
yading@11:     if (c == s)
yading@11:         return 0;
yading@11:     rs->s = s;
yading@11:     rs->e = p->cursor = c;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int lex_time(struct sbg_parser *p, int64_t *rt)
yading@11: {
yading@11:     int r = str_to_time(p->cursor, rt);
yading@11:     p->cursor += r;
yading@11:     return r > 0;
yading@11: }
yading@11: 
yading@11: #define FORWARD_ERROR(c) \
yading@11:     do { \
yading@11:         int errcode = c; \
yading@11:         if (errcode <= 0) \
yading@11:             return errcode ? errcode : AVERROR_INVALIDDATA; \
yading@11:     } while(0);
yading@11: 
yading@11: static int parse_immediate(struct sbg_parser *p)
yading@11: {
yading@11:     snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:              "immediate sequences not yet implemented");
yading@11:     return AVERROR_PATCHWELCOME;
yading@11: }
yading@11: 
yading@11: static int parse_preprogrammed(struct sbg_parser *p)
yading@11: {
yading@11:     snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:              "preprogrammed sequences not yet implemented");
yading@11:     return AVERROR_PATCHWELCOME;
yading@11: }
yading@11: 
yading@11: static int parse_optarg(struct sbg_parser *p, char o, struct sbg_string *r)
yading@11: {
yading@11:     if (!lex_wsword(p, r)) {
yading@11:         snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                  "option '%c' requires an argument", o);
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     }
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_options(struct sbg_parser *p)
yading@11: {
yading@11:     struct sbg_string ostr, oarg;
yading@11:     char mode = 0;
yading@11:     int r;
yading@11:     char *tptr;
yading@11:     double v;
yading@11: 
yading@11:     if (p->cursor == p->end || *p->cursor != '-')
yading@11:         return 0;
yading@11:     while (lex_char(p, '-') && lex_wsword(p, &ostr)) {
yading@11:         for (; ostr.s < ostr.e; ostr.s++) {
yading@11:             char opt = *ostr.s;
yading@11:             switch (opt) {
yading@11:                 case 'S':
yading@11:                     p->scs.opt_start_at_first = 1;
yading@11:                     break;
yading@11:                 case 'E':
yading@11:                     p->scs.opt_end_at_last = 1;
yading@11:                     break;
yading@11:                 case 'i':
yading@11:                     mode = 'i';
yading@11:                     break;
yading@11:                 case 'p':
yading@11:                     mode = 'p';
yading@11:                     break;
yading@11:                 case 'F':
yading@11:                     FORWARD_ERROR(parse_optarg(p, opt, &oarg));
yading@11:                     v = strtod(oarg.s, &tptr);
yading@11:                     if (oarg.e != tptr) {
yading@11:                         snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                                  "syntax error for option -F");
yading@11:                         return AVERROR_INVALIDDATA;
yading@11:                     }
yading@11:                     p->scs.opt_fade_time = v * AV_TIME_BASE / 1000;
yading@11:                     break;
yading@11:                 case 'L':
yading@11:                     FORWARD_ERROR(parse_optarg(p, opt, &oarg));
yading@11:                     r = str_to_time(oarg.s, &p->scs.opt_duration);
yading@11:                     if (oarg.e != oarg.s + r) {
yading@11:                         snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                                  "syntax error for option -L");
yading@11:                         return AVERROR_INVALIDDATA;
yading@11:                     }
yading@11:                     break;
yading@11:                 case 'T':
yading@11:                     FORWARD_ERROR(parse_optarg(p, opt, &oarg));
yading@11:                     r = str_to_time(oarg.s, &p->scs.start_ts);
yading@11:                     if (oarg.e != oarg.s + r) {
yading@11:                         snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                                  "syntax error for option -T");
yading@11:                         return AVERROR_INVALIDDATA;
yading@11:                     }
yading@11:                     break;
yading@11:                 case 'm':
yading@11:                     FORWARD_ERROR(parse_optarg(p, opt, &oarg));
yading@11:                     tptr = av_malloc(oarg.e - oarg.s + 1);
yading@11:                     if (!tptr)
yading@11:                         return AVERROR(ENOMEM);
yading@11:                     memcpy(tptr, oarg.s, oarg.e - oarg.s);
yading@11:                     tptr[oarg.e - oarg.s] = 0;
yading@11:                     av_free(p->scs.opt_mix);
yading@11:                     p->scs.opt_mix = tptr;
yading@11:                     break;
yading@11:                 case 'q':
yading@11:                     FORWARD_ERROR(parse_optarg(p, opt, &oarg));
yading@11:                     v = strtod(oarg.s, &tptr);
yading@11:                     if (oarg.e != tptr) {
yading@11:                         snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                                  "syntax error for option -q");
yading@11:                         return AVERROR_INVALIDDATA;
yading@11:                     }
yading@11:                     if (v != 1) {
yading@11:                         snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                                  "speed factor other than 1 not supported");
yading@11:                         return AVERROR_PATCHWELCOME;
yading@11:                     }
yading@11:                     break;
yading@11:                 case 'r':
yading@11:                     FORWARD_ERROR(parse_optarg(p, opt, &oarg));
yading@11:                     r = strtol(oarg.s, &tptr, 10);
yading@11:                     if (oarg.e != tptr) {
yading@11:                         snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                                  "syntax error for option -r");
yading@11:                         return AVERROR_INVALIDDATA;
yading@11:                     }
yading@11:                     if (r < 40) {
yading@11:                         snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                                  "invalid sample rate");
yading@11:                         return AVERROR_PATCHWELCOME;
yading@11:                     }
yading@11:                     p->scs.sample_rate = r;
yading@11:                     break;
yading@11:                 default:
yading@11:                     snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                              "unknown option: '%c'", *ostr.s);
yading@11:                     return AVERROR_INVALIDDATA;
yading@11:             }
yading@11:         }
yading@11:     }
yading@11:     switch (mode) {
yading@11:         case 'i':
yading@11:             return parse_immediate(p);
yading@11:         case 'p':
yading@11:             return parse_preprogrammed(p);
yading@11:         case 0:
yading@11:             if (!lex_line_end(p))
yading@11:                 return AVERROR_INVALIDDATA;
yading@11:             return 1;
yading@11:     }
yading@11:     return AVERROR_BUG;
yading@11: }
yading@11: 
yading@11: static int parse_timestamp(struct sbg_parser *p,
yading@11:                                struct sbg_timestamp *rts, int64_t *rrel)
yading@11: {
yading@11:     int64_t abs = 0, rel = 0, dt;
yading@11:     char type = 0;
yading@11:     int r;
yading@11: 
yading@11:     if (lex_fixed(p, "NOW", 3)) {
yading@11:         type = 'N';
yading@11:         r = 1;
yading@11:     } else {
yading@11:         r = lex_time(p, &abs);
yading@11:         if (r)
yading@11:             type = 'T';
yading@11:     }
yading@11:     while (lex_char(p, '+')) {
yading@11:         if (!lex_time(p, &dt))
yading@11:             return AVERROR_INVALIDDATA;
yading@11:         rel += dt;
yading@11:         r = 1;
yading@11:     }
yading@11:     if (r) {
yading@11:         if (!lex_space(p))
yading@11:             return AVERROR_INVALIDDATA;
yading@11:         rts->type = type;
yading@11:         rts->t    = abs;
yading@11:         *rrel     = rel;
yading@11:     }
yading@11:     return r;
yading@11: }
yading@11: 
yading@11: static int parse_fade(struct sbg_parser *p, struct sbg_fade *fr)
yading@11: {
yading@11:     struct sbg_fade f = {0};
yading@11: 
yading@11:     if (lex_char(p, '<'))
yading@11:         f.in = SBG_FADE_SILENCE;
yading@11:     else if (lex_char(p, '-'))
yading@11:         f.in = SBG_FADE_SAME;
yading@11:     else if (lex_char(p, '='))
yading@11:         f.in = SBG_FADE_ADAPT;
yading@11:     else
yading@11:         return 0;
yading@11:     if (lex_char(p, '>'))
yading@11:         f.out = SBG_FADE_SILENCE;
yading@11:     else if (lex_char(p, '-'))
yading@11:         f.out = SBG_FADE_SAME;
yading@11:     else if (lex_char(p, '='))
yading@11:         f.out = SBG_FADE_ADAPT;
yading@11:     else
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     *fr = f;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_time_sequence(struct sbg_parser *p, int inblock)
yading@11: {
yading@11:     struct sbg_timestamp ts;
yading@11:     int64_t rel_ts;
yading@11:     int r;
yading@11:     struct sbg_fade fade = { SBG_FADE_SAME, SBG_FADE_SAME, 0 };
yading@11:     struct sbg_string name;
yading@11:     struct sbg_script_tseq *tseq;
yading@11: 
yading@11:     r = parse_timestamp(p, &ts, &rel_ts);
yading@11:     if (!r)
yading@11:         return 0;
yading@11:     if (r < 0)
yading@11:         return r;
yading@11:     if (ts.type) {
yading@11:         if (inblock)
yading@11:             return AVERROR_INVALIDDATA;
yading@11:         p->current_time.type = ts.type;
yading@11:         p->current_time.t    = ts.t;
yading@11:     } else if(!inblock && !p->current_time.type) {
yading@11:         snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:                  "relative time without previous absolute time");
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     }
yading@11:     ts.type = p->current_time.type;
yading@11:     ts.t    = p->current_time.t + rel_ts;
yading@11:     r = parse_fade(p, &fade);
yading@11:     if (r < 0)
yading@11:         return r;
yading@11:     lex_space(p);
yading@11:     if (!lex_name(p, &name))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     lex_space(p);
yading@11:     if (lex_fixed(p, "->", 2)) {
yading@11:         fade.slide = SBG_FADE_ADAPT;
yading@11:         lex_space(p);
yading@11:     }
yading@11:     if (!lex_line_end(p))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     tseq = inblock ?
yading@11:            alloc_array_elem((void **)&p->scs.block_tseq, sizeof(*tseq),
yading@11:                             &p->nb_block_tseq, &p->nb_block_tseq_max) :
yading@11:            alloc_array_elem((void **)&p->scs.tseq, sizeof(*tseq),
yading@11:                             &p->scs.nb_tseq, &p->nb_tseq_max);
yading@11:     if (!tseq)
yading@11:         return AVERROR(ENOMEM);
yading@11:     tseq->ts       = ts;
yading@11:     tseq->name     = name.s;
yading@11:     tseq->name_len = name.e - name.s;
yading@11:     tseq->fade     = fade;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_wave_def(struct sbg_parser *p, int wavenum)
yading@11: {
yading@11:     snprintf(p->err_msg, sizeof(p->err_msg),
yading@11:              "waveform definitions not yet implemented");
yading@11:     return AVERROR_PATCHWELCOME;
yading@11: }
yading@11: 
yading@11: static int parse_block_def(struct sbg_parser *p,
yading@11:                            struct sbg_script_definition *def)
yading@11: {
yading@11:     int r, tseq;
yading@11: 
yading@11:     lex_space(p);
yading@11:     if (!lex_line_end(p))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     tseq = p->nb_block_tseq;
yading@11:     while (1) {
yading@11:         r = parse_time_sequence(p, 1);
yading@11:         if (r < 0)
yading@11:             return r;
yading@11:         if (!r)
yading@11:             break;
yading@11:     }
yading@11:     if (!lex_char(p, '}'))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     lex_space(p);
yading@11:     if (!lex_line_end(p))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     def->type        = 'B';
yading@11:     def->elements    = tseq;
yading@11:     def->nb_elements = p->nb_block_tseq - tseq;
yading@11:     if (!def->nb_elements)
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_volume(struct sbg_parser *p, int *vol)
yading@11: {
yading@11:     double v;
yading@11: 
yading@11:     if (!lex_char(p, '/'))
yading@11:         return 0;
yading@11:     if (!lex_double(p, &v))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     if (scale_double(p->log, v, 0.01, vol))
yading@11:         return AVERROR(ERANGE);
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_synth_channel_sine(struct sbg_parser *p,
yading@11:                                     struct sbg_script_synth *synth)
yading@11: {
yading@11:     double carrierf, beatf;
yading@11:     int carrier, beat, vol;
yading@11: 
yading@11:     if (!lex_double(p, &carrierf))
yading@11:         return 0;
yading@11:     if (!lex_double(p, &beatf))
yading@11:         beatf = 0;
yading@11:     FORWARD_ERROR(parse_volume(p, &vol));
yading@11:     if (scale_double(p->log, carrierf, 1, &carrier) < 0 ||
yading@11:         scale_double(p->log, beatf, 1, &beat) < 0)
yading@11:         return AVERROR(EDOM);
yading@11:     synth->type    = SBG_TYPE_SINE;
yading@11:     synth->carrier = carrier;
yading@11:     synth->beat    = beat;
yading@11:     synth->vol     = vol;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_synth_channel_pink(struct sbg_parser *p,
yading@11:                                     struct sbg_script_synth *synth)
yading@11: {
yading@11:     int vol;
yading@11: 
yading@11:     if (!lex_fixed(p, "pink", 4))
yading@11:         return 0;
yading@11:     FORWARD_ERROR(parse_volume(p, &vol));
yading@11:     synth->type    = SBG_TYPE_NOISE;
yading@11:     synth->vol     = vol;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_synth_channel_bell(struct sbg_parser *p,
yading@11:                                     struct sbg_script_synth *synth)
yading@11: {
yading@11:     double carrierf;
yading@11:     int carrier, vol;
yading@11: 
yading@11:     if (!lex_fixed(p, "bell", 4))
yading@11:         return 0;
yading@11:     if (!lex_double(p, &carrierf))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     FORWARD_ERROR(parse_volume(p, &vol));
yading@11:     if (scale_double(p->log, carrierf, 1, &carrier) < 0)
yading@11:         return AVERROR(EDOM);
yading@11:     synth->type    = SBG_TYPE_BELL;
yading@11:     synth->carrier = carrier;
yading@11:     synth->vol     = vol;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_synth_channel_mix(struct sbg_parser *p,
yading@11:                                    struct sbg_script_synth *synth)
yading@11: {
yading@11:     int vol;
yading@11: 
yading@11:     if (!lex_fixed(p, "mix", 3))
yading@11:         return 0;
yading@11:     FORWARD_ERROR(parse_volume(p, &vol));
yading@11:     synth->type    = SBG_TYPE_MIX;
yading@11:     synth->vol     = vol;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_synth_channel_spin(struct sbg_parser *p,
yading@11:                                     struct sbg_script_synth *synth)
yading@11: {
yading@11:     double carrierf, beatf;
yading@11:     int carrier, beat, vol;
yading@11: 
yading@11:     if (!lex_fixed(p, "spin:", 5))
yading@11:         return 0;
yading@11:     if (!lex_double(p, &carrierf))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     if (!lex_double(p, &beatf))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     FORWARD_ERROR(parse_volume(p, &vol));
yading@11:     if (scale_double(p->log, carrierf, 1, &carrier) < 0 ||
yading@11:         scale_double(p->log, beatf, 1, &beat) < 0)
yading@11:         return AVERROR(EDOM);
yading@11:     synth->type    = SBG_TYPE_SPIN;
yading@11:     synth->carrier = carrier;
yading@11:     synth->beat    = beat;
yading@11:     synth->vol     = vol;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_synth_channel(struct sbg_parser *p)
yading@11: {
yading@11:     int r;
yading@11:     struct sbg_script_synth *synth;
yading@11: 
yading@11:     synth = alloc_array_elem((void **)&p->scs.synth, sizeof(*synth),
yading@11:                              &p->scs.nb_synth, &p->nb_synth_max);
yading@11:     if (!synth)
yading@11:         return AVERROR(ENOMEM);
yading@11:     r = lex_char(p, '-');
yading@11:     if (!r)
yading@11:         r = parse_synth_channel_pink(p, synth);
yading@11:     if (!r)
yading@11:         r = parse_synth_channel_bell(p, synth);
yading@11:     if (!r)
yading@11:         r = parse_synth_channel_mix(p, synth);
yading@11:     if (!r)
yading@11:         r = parse_synth_channel_spin(p, synth);
yading@11:     /* Unimplemented: wave%d:%f%f/vol (carrier, beat) */
yading@11:     if (!r)
yading@11:         r = parse_synth_channel_sine(p, synth);
yading@11:     if (r <= 0)
yading@11:         p->scs.nb_synth--;
yading@11:     return r;
yading@11: }
yading@11: 
yading@11: static int parse_synth_def(struct sbg_parser *p,
yading@11:                            struct sbg_script_definition *def)
yading@11: {
yading@11:     int r, synth;
yading@11: 
yading@11:     synth = p->scs.nb_synth;
yading@11:     while (1) {
yading@11:         r = parse_synth_channel(p);
yading@11:         if (r < 0)
yading@11:             return r;
yading@11:         if (!r || !lex_space(p))
yading@11:             break;
yading@11:     }
yading@11:     lex_space(p);
yading@11:     if (synth == p->scs.nb_synth)
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     if (!lex_line_end(p))
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     def->type        = 'S';
yading@11:     def->elements    = synth;
yading@11:     def->nb_elements = p->scs.nb_synth - synth;
yading@11:     return 1;
yading@11: }
yading@11: 
yading@11: static int parse_named_def(struct sbg_parser *p)
yading@11: {
yading@11:     char *cursor_save = p->cursor;
yading@11:     struct sbg_string name;
yading@11:     struct sbg_script_definition *def;
yading@11: 
yading@11:     if (!lex_name(p, &name) || !lex_char(p, ':') || !lex_space(p)) {
yading@11:         p->cursor = cursor_save;
yading@11:         return 0;
yading@11:     }
yading@11:     if (name.e - name.s == 6 && !memcmp(name.s, "wave", 4) &&
yading@11:         name.s[4] >= '0' && name.s[4] <= '9' &&
yading@11:         name.s[5] >= '0' && name.s[5] <= '9') {
yading@11:         int wavenum = (name.s[4] - '0') * 10 + (name.s[5] - '0');
yading@11:         return parse_wave_def(p, wavenum);
yading@11:     }
yading@11:     def = alloc_array_elem((void **)&p->scs.def, sizeof(*def),
yading@11:                            &p->scs.nb_def, &p->nb_def_max);
yading@11:     if (!def)
yading@11:         return AVERROR(ENOMEM);
yading@11:     def->name     = name.s;
yading@11:     def->name_len = name.e - name.s;
yading@11:     if (lex_char(p, '{'))
yading@11:         return parse_block_def(p, def);
yading@11:     return parse_synth_def(p, def);
yading@11: }
yading@11: 
yading@11: static void free_script(struct sbg_script *s)
yading@11: {
yading@11:     av_freep(&s->def);
yading@11:     av_freep(&s->synth);
yading@11:     av_freep(&s->tseq);
yading@11:     av_freep(&s->block_tseq);
yading@11:     av_freep(&s->events);
yading@11:     av_freep(&s->opt_mix);
yading@11: }
yading@11: 
yading@11: static int parse_script(void *log, char *script, int script_len,
yading@11:                             struct sbg_script *rscript)
yading@11: {
yading@11:     struct sbg_parser sp = {
yading@11:         .log     = log,
yading@11:         .script  = script,
yading@11:         .end     = script + script_len,
yading@11:         .cursor  = script,
yading@11:         .line_no = 1,
yading@11:         .err_msg = "",
yading@11:         .scs = {
yading@11:             /* default values */
yading@11:             .start_ts      = AV_NOPTS_VALUE,
yading@11:             .sample_rate   = 44100,
yading@11:             .opt_fade_time = 60 * AV_TIME_BASE,
yading@11:         },
yading@11:     };
yading@11:     int r;
yading@11: 
yading@11:     lex_space(&sp);
yading@11:     while (sp.cursor < sp.end) {
yading@11:         r = parse_options(&sp);
yading@11:         if (r < 0)
yading@11:             goto fail;
yading@11:         if (!r && !lex_line_end(&sp))
yading@11:             break;
yading@11:     }
yading@11:     while (sp.cursor < sp.end) {
yading@11:         r = parse_named_def(&sp);
yading@11:         if (!r)
yading@11:             r = parse_time_sequence(&sp, 0);
yading@11:         if (!r)
yading@11:             r = lex_line_end(&sp) ? 1 : AVERROR_INVALIDDATA;
yading@11:         if (r < 0)
yading@11:             goto fail;
yading@11:     }
yading@11:     *rscript = sp.scs;
yading@11:     return 1;
yading@11: fail:
yading@11:     free_script(&sp.scs);
yading@11:     if (!*sp.err_msg)
yading@11:         if (r == AVERROR_INVALIDDATA)
yading@11:             snprintf(sp.err_msg, sizeof(sp.err_msg), "syntax error");
yading@11:     if (log && *sp.err_msg) {
yading@11:         const char *ctx = sp.cursor;
yading@11:         const char *ectx = av_x_if_null(memchr(ctx, '\n', sp.end - sp.cursor),
yading@11:                                         sp.end);
yading@11:         int lctx = ectx - ctx;
yading@11:         const char *quote = "\"";
yading@11:         if (lctx > 0 && ctx[lctx - 1] == '\r')
yading@11:             lctx--;
yading@11:         if (lctx == 0) {
yading@11:             ctx = "the end of line";
yading@11:             lctx = strlen(ctx);
yading@11:             quote = "";
yading@11:         }
yading@11:         av_log(log, AV_LOG_ERROR, "Error line %d: %s near %s%.*s%s.\n",
yading@11:                sp.line_no, sp.err_msg, quote, lctx, ctx, quote);
yading@11:     }
yading@11:     return r;
yading@11: }
yading@11: 
yading@11: static int read_whole_file(AVIOContext *io, int max_size, char **rbuf)
yading@11: {
yading@11:     char *buf = NULL;
yading@11:     int size = 0, bufsize = 0, r;
yading@11: 
yading@11:     while (1) {
yading@11:         if (bufsize - size < 1024) {
yading@11:             bufsize = FFMIN(FFMAX(2 * bufsize, 8192), max_size);
yading@11:             if (bufsize - size < 2) {
yading@11:                 size = AVERROR(EFBIG);
yading@11:                 goto fail;
yading@11:             }
yading@11:             buf = av_realloc_f(buf, bufsize, 1);
yading@11:             if (!buf) {
yading@11:                 size = AVERROR(ENOMEM);
yading@11:                 goto fail;
yading@11:             }
yading@11:         }
yading@11:         r = avio_read(io, buf, bufsize - size - 1);
yading@11:         if (r == AVERROR_EOF)
yading@11:             break;
yading@11:         if (r < 0)
yading@11:             goto fail;
yading@11:         size += r;
yading@11:     }
yading@11:     buf[size] = 0;
yading@11:     *rbuf = buf;
yading@11:     return size;
yading@11: fail:
yading@11:     av_free(buf);
yading@11:     return size;
yading@11: }
yading@11: 
yading@11: static void expand_timestamps(void *log, struct sbg_script *s)
yading@11: {
yading@11:     int i, nb_rel = 0;
yading@11:     int64_t now, cur_ts, delta = 0;
yading@11: 
yading@11:     for (i = 0; i < s->nb_tseq; i++)
yading@11:         nb_rel += s->tseq[i].ts.type == 'N';
yading@11:     if (nb_rel == s->nb_tseq) {
yading@11:         /* All ts are relative to NOW: consider NOW = 0 */
yading@11:         now = 0;
yading@11:         if (s->start_ts != AV_NOPTS_VALUE)
yading@11:             av_log(log, AV_LOG_WARNING,
yading@11:                    "Start time ignored in a purely relative script.\n");
yading@11:     } else if (nb_rel == 0 && s->start_ts != AV_NOPTS_VALUE ||
yading@11:                s->opt_start_at_first) {
yading@11:         /* All ts are absolute and start time is specified */
yading@11:         if (s->start_ts == AV_NOPTS_VALUE)
yading@11:             s->start_ts = s->tseq[0].ts.t;
yading@11:         now = s->start_ts;
yading@11:     } else {
yading@11:         /* Mixed relative/absolute ts: expand */
yading@11:         time_t now0;
yading@11:         struct tm *tm;
yading@11: 
yading@11:         av_log(log, AV_LOG_WARNING,
yading@11:                "Scripts with mixed absolute and relative timestamps can give "
yading@11:                "unexpected results (pause, seeking, time zone change).\n");
yading@11: #undef time
yading@11:         time(&now0);
yading@11:         tm = localtime(&now0);
yading@11:         now = tm ? tm->tm_hour * 3600 + tm->tm_min * 60 + tm->tm_sec :
yading@11:                    now0 % DAY;
yading@11:         av_log(log, AV_LOG_INFO, "Using %02d:%02d:%02d as NOW.\n",
yading@11:                (int)(now / 3600), (int)(now / 60) % 60, (int)now % 60);
yading@11:         now *= AV_TIME_BASE;
yading@11:         for (i = 0; i < s->nb_tseq; i++) {
yading@11:             if (s->tseq[i].ts.type == 'N') {
yading@11:                 s->tseq[i].ts.t += now;
yading@11:                 s->tseq[i].ts.type = 'T'; /* not necessary */
yading@11:             }
yading@11:         }
yading@11:     }
yading@11:     if (s->start_ts == AV_NOPTS_VALUE)
yading@11:         s->start_ts = s->opt_start_at_first ? s->tseq[0].ts.t : now;
yading@11:     s->end_ts = s->opt_duration ? s->start_ts + s->opt_duration :
yading@11:                 AV_NOPTS_VALUE; /* may be overridden later by -E option */
yading@11:     cur_ts = now;
yading@11:     for (i = 0; i < s->nb_tseq; i++) {
yading@11:         if (s->tseq[i].ts.t + delta < cur_ts)
yading@11:             delta += DAY_TS;
yading@11:         cur_ts = s->tseq[i].ts.t += delta;
yading@11:     }
yading@11: }
yading@11: 
yading@11: static int expand_tseq(void *log, struct sbg_script *s, int *nb_ev_max,
yading@11:                        int64_t t0, struct sbg_script_tseq *tseq)
yading@11: {
yading@11:     int i, r;
yading@11:     struct sbg_script_definition *def;
yading@11:     struct sbg_script_tseq *be;
yading@11:     struct sbg_script_event *ev;
yading@11: 
yading@11:     if (tseq->lock++) {
yading@11:         av_log(log, AV_LOG_ERROR, "Recursion loop on \"%.*s\"\n",
yading@11:                tseq->name_len, tseq->name);
yading@11:         return AVERROR(EINVAL);
yading@11:     }
yading@11:     t0 += tseq->ts.t;
yading@11:     for (i = 0; i < s->nb_def; i++) {
yading@11:         if (s->def[i].name_len == tseq->name_len &&
yading@11:             !memcmp(s->def[i].name, tseq->name, tseq->name_len))
yading@11:             break;
yading@11:     }
yading@11:     if (i >= s->nb_def) {
yading@11:         av_log(log, AV_LOG_ERROR, "Tone-set \"%.*s\" not defined\n",
yading@11:                tseq->name_len, tseq->name);
yading@11:         return AVERROR(EINVAL);
yading@11:     }
yading@11:     def = &s->def[i];
yading@11:     if (def->type == 'B') {
yading@11:         be = s->block_tseq + def->elements;
yading@11:         for (i = 0; i < def->nb_elements; i++) {
yading@11:             r = expand_tseq(log, s, nb_ev_max, t0, &be[i]);
yading@11:             if (r < 0)
yading@11:                 return r;
yading@11:         }
yading@11:     } else {
yading@11:         ev = alloc_array_elem((void **)&s->events, sizeof(*ev),
yading@11:                               &s->nb_events, nb_ev_max);
yading@11:         ev->ts          = tseq->ts.t;
yading@11:         ev->elements    = def->elements;
yading@11:         ev->nb_elements = def->nb_elements;
yading@11:         ev->fade        = tseq->fade;
yading@11:     }
yading@11:     tseq->lock--;
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int expand_script(void *log, struct sbg_script *s)
yading@11: {
yading@11:     int i, r, nb_events_max = 0;
yading@11: 
yading@11:     expand_timestamps(log, s);
yading@11:     for (i = 0; i < s->nb_tseq; i++) {
yading@11:         r = expand_tseq(log, s, &nb_events_max, 0, &s->tseq[i]);
yading@11:         if (r < 0)
yading@11:             return r;
yading@11:     }
yading@11:     if (!s->nb_events) {
yading@11:         av_log(log, AV_LOG_ERROR, "No events in script\n");
yading@11:         return AVERROR_INVALIDDATA;
yading@11:     }
yading@11:     if (s->opt_end_at_last)
yading@11:         s->end_ts = s->events[s->nb_events - 1].ts;
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int add_interval(struct ws_intervals *inter,
yading@11:                         enum ws_interval_type type, uint32_t channels, int ref,
yading@11:                         int64_t ts1, int32_t f1, int32_t a1,
yading@11:                         int64_t ts2, int32_t f2, int32_t a2)
yading@11: {
yading@11:     struct ws_interval *i, *ri;
yading@11: 
yading@11:     if (ref >= 0) {
yading@11:         ri = &inter->inter[ref];
yading@11:         /* ref and new intervals are constant, identical and adjacent */
yading@11:         if (ri->type == type && ri->channels == channels &&
yading@11:             ri->f1 == ri->f2 && ri->f2 == f1 && f1 == f2 &&
yading@11:             ri->a1 == ri->a2 && ri->a2 == a1 && a1 == a2 &&
yading@11:             ri->ts2 == ts1) {
yading@11:             ri->ts2 = ts2;
yading@11:             return ref;
yading@11:         }
yading@11:     }
yading@11:     i = alloc_array_elem((void **)&inter->inter, sizeof(*i),
yading@11:                          &inter->nb_inter, &inter->max_inter);
yading@11:     if (!i)
yading@11:         return AVERROR(ENOMEM);
yading@11:     i->ts1      = ts1;
yading@11:     i->ts2      = ts2;
yading@11:     i->type     = type;
yading@11:     i->channels = channels;
yading@11:     i->f1       = f1;
yading@11:     i->f2       = f2;
yading@11:     i->a1       = a1;
yading@11:     i->a2       = a2;
yading@11:     i->phi      = ref >= 0 ? ref | 0x80000000 : 0;
yading@11:     return i - inter->inter;
yading@11: }
yading@11: 
yading@11: static int add_bell(struct ws_intervals *inter, struct sbg_script *s,
yading@11:                     int64_t ts1, int64_t ts2, int32_t f, int32_t a)
yading@11: {
yading@11:     /* SBaGen uses an exponential decrease every 50ms.
yading@11:        We approximate it with piecewise affine segments. */
yading@11:     int32_t cpoints[][2] = {
yading@11:         {  2, a },
yading@11:         {  4, a - a / 4 },
yading@11:         {  8, a / 2 },
yading@11:         { 16, a / 4 },
yading@11:         { 25, a / 10 },
yading@11:         { 50, a / 80 },
yading@11:         { 75, 0 },
yading@11:     };
yading@11:     int i, r;
yading@11:     int64_t dt = s->sample_rate / 20, ts3 = ts1, ts4;
yading@11:     for (i = 0; i < FF_ARRAY_ELEMS(cpoints); i++) {
yading@11:         ts4 = FFMIN(ts2, ts1 + cpoints[i][0] * dt);
yading@11:         r = add_interval(inter, WS_SINE, 3, -1,
yading@11:                          ts3, f, a, ts4, f, cpoints[i][1]);
yading@11:         if (r < 0)
yading@11:             return r;
yading@11:         ts3 = ts4;
yading@11:         a = cpoints[i][1];
yading@11:     }
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int generate_interval(void *log, struct sbg_script *s,
yading@11:                              struct ws_intervals *inter,
yading@11:                              int64_t ts1, int64_t ts2,
yading@11:                              struct sbg_script_synth *s1,
yading@11:                              struct sbg_script_synth *s2,
yading@11:                              int transition)
yading@11: {
yading@11:     int r;
yading@11: 
yading@11:     if (ts2 <= ts1 || (s1->vol == 0 && s2->vol == 0))
yading@11:         return 0;
yading@11:     switch (s1->type) {
yading@11:         case SBG_TYPE_NONE:
yading@11:             break;
yading@11:         case SBG_TYPE_SINE:
yading@11:             if (s1->beat == 0 && s2->beat == 0) {
yading@11:                 r = add_interval(inter, WS_SINE, 3, s1->ref.l,
yading@11:                                  ts1, s1->carrier, s1->vol,
yading@11:                                  ts2, s2->carrier, s2->vol);
yading@11:                 if (r < 0)
yading@11:                     return r;
yading@11:                 s2->ref.l = s2->ref.r = r;
yading@11:             } else {
yading@11:                 r = add_interval(inter, WS_SINE, 1, s1->ref.l,
yading@11:                                  ts1, s1->carrier + s1->beat / 2, s1->vol,
yading@11:                                  ts2, s2->carrier + s2->beat / 2, s2->vol);
yading@11:                 if (r < 0)
yading@11:                     return r;
yading@11:                 s2->ref.l = r;
yading@11:                 r = add_interval(inter, WS_SINE, 2, s1->ref.r,
yading@11:                                  ts1, s1->carrier - s1->beat / 2, s1->vol,
yading@11:                                  ts2, s2->carrier - s2->beat / 2, s2->vol);
yading@11:                 if (r < 0)
yading@11:                     return r;
yading@11:                 s2->ref.r = r;
yading@11:             }
yading@11:             break;
yading@11: 
yading@11:         case SBG_TYPE_BELL:
yading@11:             if (transition == 2) {
yading@11:                 r = add_bell(inter, s, ts1, ts2, s1->carrier, s2->vol);
yading@11:                 if (r < 0)
yading@11:                     return r;
yading@11:             }
yading@11:             break;
yading@11: 
yading@11:         case SBG_TYPE_SPIN:
yading@11:             av_log(log, AV_LOG_WARNING, "Spinning noise not implemented, "
yading@11:                                         "using pink noise instead.\n");
yading@11:             /* fall through */
yading@11:         case SBG_TYPE_NOISE:
yading@11:             /* SBaGen's pink noise generator uses:
yading@11:                - 1 band of white noise, mean square: 1/3;
yading@11:                - 9 bands of subsampled white noise with linear
yading@11:                  interpolation, mean square: 2/3 each;
yading@11:                with 1/10 weight each: the total mean square is 7/300.
yading@11:                Our pink noise generator uses 8 bands of white noise with
yading@11:                rectangular subsampling: the total mean square is 1/24.
yading@11:                Therefore, to match SBaGen's volume, we must multiply vol by
yading@11:                sqrt((7/300) / (1/24)) = sqrt(14/25) =~ 0.748
yading@11:              */
yading@11:             r = add_interval(inter, WS_NOISE, 3, s1->ref.l,
yading@11:                              ts1, 0, s1->vol - s1->vol / 4,
yading@11:                              ts2, 0, s2->vol - s2->vol / 4);
yading@11:             if (r < 0)
yading@11:                 return r;
yading@11:             s2->ref.l = s2->ref.r = r;
yading@11:             break;
yading@11: 
yading@11:         case SBG_TYPE_MIX:
yading@11:             /* Unimplemented: silence; warning present elsewhere */
yading@11:         default:
yading@11:             av_log(log, AV_LOG_ERROR,
yading@11:                    "Type %d is not implemented\n", s1->type);
yading@11:             return AVERROR_PATCHWELCOME;
yading@11:     }
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int generate_plateau(void *log, struct sbg_script *s,
yading@11:                             struct ws_intervals *inter,
yading@11:                             struct sbg_script_event *ev1)
yading@11: {
yading@11:     int64_t ts1 = ev1->ts_int, ts2 = ev1->ts_trans;
yading@11:     int i, r;
yading@11:     struct sbg_script_synth *s1;
yading@11: 
yading@11:     for (i = 0; i < ev1->nb_elements; i++) {
yading@11:         s1 = &s->synth[ev1->elements + i];
yading@11:         r = generate_interval(log, s, inter, ts1, ts2, s1, s1, 0);
yading@11:         if (r < 0)
yading@11:             return r;
yading@11:     }
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: /*
yading@11: 
yading@11:    ts1             ts2         ts1    tsmid    ts2
yading@11:     |               |           |       |       |
yading@11:     v               v           v       |       v
yading@11: ____                        ____        v       ____
yading@11:     ''''....                    ''..        ..''
yading@11:             ''''....____            ''....''
yading@11: 
yading@11:   compatible transition      incompatible transition
yading@11:  */
yading@11: 
yading@11: static int generate_transition(void *log, struct sbg_script *s,
yading@11:                                struct ws_intervals *inter,
yading@11:                                struct sbg_script_event *ev1,
yading@11:                                struct sbg_script_event *ev2)
yading@11: {
yading@11:     int64_t ts1 = ev1->ts_trans, ts2 = ev1->ts_next;
yading@11:     /* (ts1 + ts2) / 2 without overflow */
yading@11:     int64_t tsmid = (ts1 >> 1) + (ts2 >> 1) + (ts1 & ts2 & 1);
yading@11:     enum sbg_fade_type type = ev1->fade.slide | (ev1->fade.out & ev2->fade.in);
yading@11:     int nb_elements = FFMAX(ev1->nb_elements, ev2->nb_elements);
yading@11:     struct sbg_script_synth *s1, *s2, s1mod, s2mod, smid;
yading@11:     int pass, i, r;
yading@11: 
yading@11:     for (pass = 0; pass < 2; pass++) {
yading@11:         /* pass = 0 -> compatible and first half of incompatible
yading@11:            pass = 1 -> second half of incompatible
yading@11:            Using two passes like that ensures that the intervals are generated
yading@11:            in increasing order according to their start timestamp.
yading@11:            Otherwise it would be necessary to sort them
yading@11:            while keeping the mutual references.
yading@11:          */
yading@11:         for (i = 0; i < nb_elements; i++) {
yading@11:             s1 = i < ev1->nb_elements ? &s->synth[ev1->elements + i] : &s1mod;
yading@11:             s2 = i < ev2->nb_elements ? &s->synth[ev2->elements + i] : &s2mod;
yading@11:             s1mod = s1 != &s1mod ? *s1 : (struct sbg_script_synth){ 0 };
yading@11:             s2mod = s2 != &s2mod ? *s2 : (struct sbg_script_synth){ 0 };
yading@11:             if (ev1->fade.slide) {
yading@11:                 /* for slides, and only for slides, silence ("-") is equivalent
yading@11:                    to anything with volume 0 */
yading@11:                 if (s1mod.type == SBG_TYPE_NONE) {
yading@11:                     s1mod = s2mod;
yading@11:                     s1mod.vol = 0;
yading@11:                 } else if (s2mod.type == SBG_TYPE_NONE) {
yading@11:                     s2mod = s1mod;
yading@11:                     s2mod.vol = 0;
yading@11:                 }
yading@11:             }
yading@11:             if (s1mod.type == s2mod.type &&
yading@11:                 s1mod.type != SBG_TYPE_BELL &&
yading@11:                 (type == SBG_FADE_ADAPT ||
yading@11:                  (s1mod.carrier == s2mod.carrier &&
yading@11:                   s1mod.beat == s2mod.beat))) {
yading@11:                 /* compatible: single transition */
yading@11:                 if (!pass) {
yading@11:                     r = generate_interval(log, s, inter,
yading@11:                                           ts1, ts2, &s1mod, &s2mod, 3);
yading@11:                     if (r < 0)
yading@11:                         return r;
yading@11:                     s2->ref = s2mod.ref;
yading@11:                 }
yading@11:             } else {
yading@11:                 /* incompatible: silence at midpoint */
yading@11:                 if (!pass) {
yading@11:                     smid = s1mod;
yading@11:                     smid.vol = 0;
yading@11:                     r = generate_interval(log, s, inter,
yading@11:                                           ts1, tsmid, &s1mod, &smid, 1);
yading@11:                     if (r < 0)
yading@11:                         return r;
yading@11:                 } else {
yading@11:                     smid = s2mod;
yading@11:                     smid.vol = 0;
yading@11:                     r = generate_interval(log, s, inter,
yading@11:                                           tsmid, ts2, &smid, &s2mod, 2);
yading@11:                     if (r < 0)
yading@11:                         return r;
yading@11:                     s2->ref = s2mod.ref;
yading@11:                 }
yading@11:             }
yading@11:         }
yading@11:     }
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: /*
yading@11:     ev1                  trats ev2  intts           endts ev3
yading@11:      |                     |    |     |               |    |
yading@11:      v                     v    v     v               v    v
yading@11:                                       ________________
yading@11: ....                              ....                ....
yading@11:     '''....________________....'''                        '''...._______________
yading@11: 
yading@11: \_________/\______________/\_________/\______________/\_________/\_____________/
yading@11:   tr x->1        int1        tr 1->2        int2        tr 2->3        int3
yading@11:  */
yading@11: 
yading@11: static int generate_intervals(void *log, struct sbg_script *s, int sample_rate,
yading@11:                               struct ws_intervals *inter)
yading@11: {
yading@11:     int64_t trans_time = s->opt_fade_time / 2;
yading@11:     struct sbg_script_event ev0, *ev1, *ev2;
yading@11:     int64_t period;
yading@11:     int i, r;
yading@11: 
yading@11:     /* SBaGen handles the time before and after the extremal events,
yading@11:        and the corresponding transitions, as if the sequence were cyclic
yading@11:        with a 24-hours period. */
yading@11:     period = s->events[s->nb_events - 1].ts - s->events[0].ts;
yading@11:     period = (period + (DAY_TS - 1)) / DAY_TS * DAY_TS;
yading@11:     period = FFMAX(period, DAY_TS);
yading@11: 
yading@11:     /* Prepare timestamps for transitions */
yading@11:     for (i = 0; i < s->nb_events; i++) {
yading@11:         ev1 = &s->events[i];
yading@11:         ev2 = &s->events[(i + 1) % s->nb_events];
yading@11:         ev1->ts_int   = ev1->ts;
yading@11:         ev1->ts_trans = ev1->fade.slide ? ev1->ts
yading@11:                                         : ev2->ts + (ev1 < ev2 ? 0 : period);
yading@11:     }
yading@11:     for (i = 0; i < s->nb_events; i++) {
yading@11:         ev1 = &s->events[i];
yading@11:         ev2 = &s->events[(i + 1) % s->nb_events];
yading@11:         if (!ev1->fade.slide) {
yading@11:             ev1->ts_trans = FFMAX(ev1->ts_int,   ev1->ts_trans - trans_time);
yading@11:             ev2->ts_int   = FFMIN(ev2->ts_trans, ev2->ts_int   + trans_time);
yading@11:         }
yading@11:         ev1->ts_next  = ev2->ts_int + (ev1 < ev2 ? 0 : period);
yading@11:     }
yading@11: 
yading@11:     /* Pseudo event before the first one */
yading@11:     ev0 = s->events[s->nb_events - 1];
yading@11:     ev0.ts_int   -= period;
yading@11:     ev0.ts_trans -= period;
yading@11:     ev0.ts_next  -= period;
yading@11: 
yading@11:     /* Convert timestamps */
yading@11:     for (i = -1; i < s->nb_events; i++) {
yading@11:         ev1 = i < 0 ? &ev0 : &s->events[i];
yading@11:         ev1->ts_int   = av_rescale(ev1->ts_int,   sample_rate, AV_TIME_BASE);
yading@11:         ev1->ts_trans = av_rescale(ev1->ts_trans, sample_rate, AV_TIME_BASE);
yading@11:         ev1->ts_next  = av_rescale(ev1->ts_next,  sample_rate, AV_TIME_BASE);
yading@11:     }
yading@11: 
yading@11:     /* Generate intervals */
yading@11:     for (i = 0; i < s->nb_synth; i++)
yading@11:         s->synth[i].ref.l = s->synth[i].ref.r = -1;
yading@11:     for (i = -1; i < s->nb_events; i++) {
yading@11:         ev1 = i < 0 ? &ev0 : &s->events[i];
yading@11:         ev2 = &s->events[(i + 1) % s->nb_events];
yading@11:         r = generate_plateau(log, s, inter, ev1);
yading@11:         if (r < 0)
yading@11:             return r;
yading@11:         r = generate_transition(log, s, inter, ev1, ev2);
yading@11:         if (r < 0)
yading@11:             return r;
yading@11:     }
yading@11:     if (!inter->nb_inter)
yading@11:         av_log(log, AV_LOG_WARNING, "Completely silent script.\n");
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int encode_intervals(struct sbg_script *s, AVCodecContext *avc,
yading@11:                             struct ws_intervals *inter)
yading@11: {
yading@11:     int i, edata_size = 4;
yading@11:     uint8_t *edata;
yading@11: 
yading@11:     for (i = 0; i < inter->nb_inter; i++) {
yading@11:         edata_size += inter->inter[i].type == WS_SINE  ? 44 :
yading@11:                       inter->inter[i].type == WS_NOISE ? 32 : 0;
yading@11:         if (edata_size < 0)
yading@11:             return AVERROR(ENOMEM);
yading@11:     }
yading@11:     edata = av_malloc(edata_size);
yading@11:     if (!edata)
yading@11:         return AVERROR(ENOMEM);
yading@11:     avc->extradata = edata;
yading@11:     avc->extradata_size = edata_size;
yading@11: 
yading@11: #define ADD_EDATA32(v) do { AV_WL32(edata, (v)); edata += 4; } while(0)
yading@11: #define ADD_EDATA64(v) do { AV_WL64(edata, (v)); edata += 8; } while(0)
yading@11:     ADD_EDATA32(inter->nb_inter);
yading@11:     for (i = 0; i < inter->nb_inter; i++) {
yading@11:         ADD_EDATA64(inter->inter[i].ts1);
yading@11:         ADD_EDATA64(inter->inter[i].ts2);
yading@11:         ADD_EDATA32(inter->inter[i].type);
yading@11:         ADD_EDATA32(inter->inter[i].channels);
yading@11:         switch (inter->inter[i].type) {
yading@11:             case WS_SINE:
yading@11:                 ADD_EDATA32(inter->inter[i].f1);
yading@11:                 ADD_EDATA32(inter->inter[i].f2);
yading@11:                 ADD_EDATA32(inter->inter[i].a1);
yading@11:                 ADD_EDATA32(inter->inter[i].a2);
yading@11:                 ADD_EDATA32(inter->inter[i].phi);
yading@11:                 break;
yading@11:             case WS_NOISE:
yading@11:                 ADD_EDATA32(inter->inter[i].a1);
yading@11:                 ADD_EDATA32(inter->inter[i].a2);
yading@11:                 break;
yading@11:         }
yading@11:     }
yading@11:     if (edata != avc->extradata + edata_size)
yading@11:         return AVERROR_BUG;
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static av_cold int sbg_read_probe(AVProbeData *p)
yading@11: {
yading@11:     int r, score;
yading@11:     struct sbg_script script = { 0 };
yading@11: 
yading@11:     r = parse_script(NULL, p->buf, p->buf_size, &script);
yading@11:     score = r < 0 || !script.nb_def || !script.nb_tseq ? 0 :
yading@11:             AVPROBE_SCORE_MAX / 3;
yading@11:     free_script(&script);
yading@11:     return score;
yading@11: }
yading@11: 
yading@11: static av_cold int sbg_read_header(AVFormatContext *avf)
yading@11: {
yading@11:     struct sbg_demuxer *sbg = avf->priv_data;
yading@11:     int r;
yading@11:     char *buf = NULL;
yading@11:     struct sbg_script script = { 0 };
yading@11:     AVStream *st;
yading@11:     struct ws_intervals inter = { 0 };
yading@11: 
yading@11:     r = read_whole_file(avf->pb, sbg->max_file_size, &buf);
yading@11:     if (r < 0)
yading@11:         goto fail;
yading@11:     r = parse_script(avf, buf, r, &script);
yading@11:     if (r < 0)
yading@11:         goto fail;
yading@11:     if (!sbg->sample_rate)
yading@11:         sbg->sample_rate = script.sample_rate;
yading@11:     else
yading@11:         script.sample_rate = sbg->sample_rate;
yading@11:     if (!sbg->frame_size)
yading@11:         sbg->frame_size = FFMAX(1, sbg->sample_rate / 10);
yading@11:     if (script.opt_mix)
yading@11:         av_log(avf, AV_LOG_WARNING, "Mix feature not implemented: "
yading@11:                "-m is ignored and mix channels will be silent.\n");
yading@11:     r = expand_script(avf, &script);
yading@11:     if (r < 0)
yading@11:         goto fail;
yading@11:     av_freep(&buf);
yading@11:     r = generate_intervals(avf, &script, sbg->sample_rate, &inter);
yading@11:     if (r < 0)
yading@11:         goto fail;
yading@11: 
yading@11:     st = avformat_new_stream(avf, NULL);
yading@11:     if (!st)
yading@11:         return AVERROR(ENOMEM);
yading@11:     st->codec->codec_type     = AVMEDIA_TYPE_AUDIO;
yading@11:     st->codec->codec_id       = AV_CODEC_ID_FFWAVESYNTH;
yading@11:     st->codec->channels       = 2;
yading@11:     st->codec->channel_layout = AV_CH_LAYOUT_STEREO;
yading@11:     st->codec->sample_rate    = sbg->sample_rate;
yading@11:     st->codec->frame_size     = sbg->frame_size;
yading@11:     avpriv_set_pts_info(st, 64, 1, st->codec->sample_rate);
yading@11:     st->probe_packets = 0;
yading@11:     st->start_time    = av_rescale(script.start_ts,
yading@11:                                    sbg->sample_rate, AV_TIME_BASE);
yading@11:     st->duration      = script.end_ts == AV_NOPTS_VALUE ? AV_NOPTS_VALUE :
yading@11:                         av_rescale(script.end_ts - script.start_ts,
yading@11:                                    sbg->sample_rate, AV_TIME_BASE);
yading@11:     st->cur_dts       = st->start_time;
yading@11:     r = encode_intervals(&script, st->codec, &inter);
yading@11:     if (r < 0)
yading@11:         goto fail;
yading@11: 
yading@11:     av_free(inter.inter);
yading@11:     free_script(&script);
yading@11:     return 0;
yading@11: 
yading@11: fail:
yading@11:     av_free(inter.inter);
yading@11:     free_script(&script);
yading@11:     av_free(buf);
yading@11:     return r;
yading@11: }
yading@11: 
yading@11: static int sbg_read_packet(AVFormatContext *avf, AVPacket *packet)
yading@11: {
yading@11:     int64_t ts, end_ts;
yading@11: 
yading@11:     ts = avf->streams[0]->cur_dts;
yading@11:     end_ts = ts + avf->streams[0]->codec->frame_size;
yading@11:     if (avf->streams[0]->duration != AV_NOPTS_VALUE)
yading@11:         end_ts = FFMIN(avf->streams[0]->start_time + avf->streams[0]->duration,
yading@11:                        end_ts);
yading@11:     if (end_ts <= ts)
yading@11:         return AVERROR_EOF;
yading@11:     if (av_new_packet(packet, 12) < 0)
yading@11:         return AVERROR(ENOMEM);
yading@11:     packet->dts = packet->pts = ts;
yading@11:     packet->duration = end_ts - ts;
yading@11:     AV_WL64(packet->data + 0, ts);
yading@11:     AV_WL32(packet->data + 8, packet->duration);
yading@11:     return packet->size;
yading@11: }
yading@11: 
yading@11: static int sbg_read_seek2(AVFormatContext *avf, int stream_index,
yading@11:                           int64_t min_ts, int64_t ts, int64_t max_ts, int flags)
yading@11: {
yading@11:     if (flags || stream_index > 0)
yading@11:         return AVERROR(EINVAL);
yading@11:     if (stream_index < 0)
yading@11:         ts = av_rescale_q(ts, AV_TIME_BASE_Q, avf->streams[0]->time_base);
yading@11:     avf->streams[0]->cur_dts = ts;
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: static int sbg_read_seek(AVFormatContext *avf, int stream_index,
yading@11:                          int64_t ts, int flags)
yading@11: {
yading@11:     return sbg_read_seek2(avf, stream_index, ts, ts, ts, 0);
yading@11: }
yading@11: 
yading@11: static const AVOption sbg_options[] = {
yading@11:     { "sample_rate", "", offsetof(struct sbg_demuxer, sample_rate),
yading@11:       AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX,
yading@11:       AV_OPT_FLAG_DECODING_PARAM },
yading@11:     { "frame_size", "", offsetof(struct sbg_demuxer, frame_size),
yading@11:       AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX,
yading@11:       AV_OPT_FLAG_DECODING_PARAM },
yading@11:     { "max_file_size", "", offsetof(struct sbg_demuxer, max_file_size),
yading@11:       AV_OPT_TYPE_INT, { .i64 = 5000000 }, 0, INT_MAX,
yading@11:       AV_OPT_FLAG_DECODING_PARAM },
yading@11:     { NULL },
yading@11: };
yading@11: 
yading@11: static const AVClass sbg_demuxer_class = {
yading@11:     .class_name = "sbg_demuxer",
yading@11:     .item_name  = av_default_item_name,
yading@11:     .option     = sbg_options,
yading@11:     .version    = LIBAVUTIL_VERSION_INT,
yading@11: };
yading@11: 
yading@11: AVInputFormat ff_sbg_demuxer = {
yading@11:     .name           = "sbg",
yading@11:     .long_name      = NULL_IF_CONFIG_SMALL("SBaGen binaural beats script"),
yading@11:     .priv_data_size = sizeof(struct sbg_demuxer),
yading@11:     .read_probe     = sbg_read_probe,
yading@11:     .read_header    = sbg_read_header,
yading@11:     .read_packet    = sbg_read_packet,
yading@11:     .read_seek      = sbg_read_seek,
yading@11:     .read_seek2     = sbg_read_seek2,
yading@11:     .extensions     = "sbg",
yading@11:     .priv_class     = &sbg_demuxer_class,
yading@11: };