[vlc-commits] demux: add mock module

Thomas Guillem git at videolan.org
Thu Oct 18 12:57:36 CEST 2018


vlc | branch: master | Thomas Guillem <thomas at gllm.fr> | Tue Oct  2 18:15:15 2018 +0200| [98da604cf6c223b65059324e4e0c0be0d250d937] | committer: Thomas Guillem

demux: add mock module

Changes:
 - Use var_LocationParse, with some X-Macros
 - Use check_LTLIBRARIES
 - Remove from coverage stats
 - Add more params (size, rate, channels, chroma, format...)

--

This new demuxer will be used for various VLC tests. It will be built for tests
(check_LTLIBRARIES). It will be mainly used to test the new vlc_player API.

Usage: ./vlc mock://length=10000000;audio_track_count=7

Why putting the parameters inside the URL ? In order to test input slaves for
example:

./vlc mock://length=10000000;video_track_count=1 \
    --input-slave mock://length=10000000;audio_track_count=1

> http://git.videolan.org/gitweb.cgi/vlc.git/?a=commit;h=98da604cf6c223b65059324e4e0c0be0d250d937
---

 Makefile.am               |   2 +-
 modules/demux/Makefile.am |   4 +
 modules/demux/mock.c      | 722 ++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 727 insertions(+), 1 deletion(-)

diff --git a/Makefile.am b/Makefile.am
index 87516e3a6a..d0ef08b829 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -187,7 +187,7 @@ lcov-raw.out:
 
 lcov.out: lcov-raw.out
 	lcov -r lcov-raw.out -o lcov.out \
-		'*test*' 'contrib/*' '/usr/include/*'
+		'*test*' 'contrib/*' '/usr/include/*' '*mock*'
 
 lcov: lcov.out
 	rm -Rf lcov lcov.tmp
diff --git a/modules/demux/Makefile.am b/modules/demux/Makefile.am
index 35dd0de5df..8fab97b525 100644
--- a/modules/demux/Makefile.am
+++ b/modules/demux/Makefile.am
@@ -476,3 +476,7 @@ demux_LTLIBRARIES += libadaptive_plugin.la
 
 libnoseek_plugin_la_SOURCES = demux/filter/noseek.c
 demux_LTLIBRARIES += libnoseek_plugin.la
+
+libdemux_mock_plugin_la_SOURCES = demux/mock.c
+libdemux_mock_plugin_la_LDFLAGS = $(AM_LDFLAGS) -rpath '$(demuxdir)'
+check_LTLIBRARIES += libdemux_mock_plugin.la
diff --git a/modules/demux/mock.c b/modules/demux/mock.c
new file mode 100644
index 0000000000..f4facd5fb7
--- /dev/null
+++ b/modules/demux/mock.c
@@ -0,0 +1,722 @@
+/*****************************************************************************
+ * mock.c : mock demux module for vlc
+ *****************************************************************************
+ * Copyright (C) 2018 VLC authors and VideoLAN
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation; either version 2.1 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
+ *****************************************************************************/
+
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
+#include <ctype.h>
+#include <limits.h>
+
+#include <vlc_common.h>
+#include <vlc_plugin.h>
+#include <vlc_aout.h>
+#include <vlc_picture.h>
+#include <vlc_demux.h>
+#include <vlc_input.h>
+#include <vlc_vector.h>
+
+
+struct mock_track
+{
+    es_format_t fmt;
+    es_out_id_t *id;
+};
+typedef struct VLC_VECTOR(struct mock_track *) mock_track_vector;
+
+static ssize_t
+var_InheritSsize(vlc_object_t *obj, const char *name)
+{
+    int64_t value = var_InheritInteger(obj, name);
+    return value >= 0 ? value : -1;
+}
+
+static unsigned
+var_InheritUnsigned(vlc_object_t *obj, const char *name)
+{
+    int64_t value = var_InheritInteger(obj, name);
+    return value >= 0 && value < UINT_MAX ? value : UINT_MAX;
+}
+
+static vlc_fourcc_t
+var_InheritFourcc(vlc_object_t *obj, const char *name)
+{
+    char *var_value = var_InheritString(obj, name);
+    if (!var_value)
+        return 0;
+
+    size_t var_len = strlen(var_value);
+    if (var_len > 4)
+    {
+        free(var_value);
+        return 0;
+    }
+
+    /* Pad with spaces if the string len is less than 4 */
+    char value[] = "    ";
+    strcpy(value, var_value);
+    if (var_len != 4)
+        value[var_len] = ' ';
+    free(var_value);
+
+    vlc_fourcc_t fourcc;
+    memcpy(&fourcc, value, 4);
+    return fourcc;
+}
+
+/* var_name, type, module_header_type, getter, default_value */
+#define LIST_OPTIONS \
+    X(length, vlc_tick_t, add_integer, var_InheritInteger, VLC_TICK_FROM_MS(5000)) \
+    X(audio_track_count, ssize_t, add_integer, var_InheritSsize, 0) \
+    X(audio_channels, unsigned, add_integer, var_InheritUnsigned, 2) \
+    X(audio_format, vlc_fourcc_t, add_string, var_InheritFourcc, "u8") \
+    X(audio_rate, unsigned, add_integer, var_InheritUnsigned, 44100) \
+    X(audio_packetized, bool, add_bool, var_InheritBool, true) \
+    X(video_track_count, ssize_t, add_integer, var_InheritSsize, 0) \
+    X(video_chroma, vlc_fourcc_t, add_string, var_InheritFourcc, "I420") \
+    X(video_width, unsigned, add_integer, var_InheritUnsigned, 640) \
+    X(video_height, unsigned, add_integer, var_InheritUnsigned, 480) \
+    X(video_frame_rate, unsigned, add_integer, var_InheritUnsigned, 25) \
+    X(video_frame_rate_base, unsigned, add_integer, var_InheritUnsigned, 1) \
+    X(video_packetized, bool, add_bool, var_InheritBool, true) \
+    X(sub_track_count, ssize_t, add_integer, var_InheritSsize, 0) \
+    X(sub_packetized, bool, add_bool, var_InheritBool, true) \
+    X(title_count, ssize_t, add_integer, var_InheritSsize, 0 ) \
+    X(chapter_count, ssize_t, add_integer, var_InheritSsize, 0) \
+    X(null_names, bool, add_bool, var_InheritBool, false) \
+    X(program_count, ssize_t, add_integer, var_InheritSsize, 0) \
+    X(can_seek, bool, add_bool, var_InheritBool, true) \
+    X(can_pause, bool, add_bool, var_InheritBool, true) \
+    X(can_control_pace, bool, add_bool, var_InheritBool, true) \
+    X(can_control_rate, bool, add_bool, var_InheritBool, true) \
+    X(can_record, bool, add_bool, var_InheritBool, true) \
+    X(error, bool, add_bool, var_InheritBool, false) \
+
+struct demux_sys
+{
+    mock_track_vector tracks;
+    vlc_tick_t pts;
+    vlc_tick_t step_length;
+
+    int current_title;
+    size_t chapter_gap;
+
+#define X(var_name, type, module_header_type, getter, default_value) \
+    type var_name;
+    LIST_OPTIONS
+#undef X
+};
+
+static input_title_t *
+CreateTitle(demux_t *demux, size_t idx)
+{
+    struct demux_sys *sys = demux->p_sys;
+
+    input_title_t *t = vlc_input_title_New();
+    if (!t)
+        return NULL;
+
+    t->i_length = sys->length;
+    if (!sys->null_names
+     && asprintf(&t->psz_name, "Mock Title %zu", idx) == -1)
+    {
+        t->psz_name = NULL;
+        vlc_input_title_Delete(t);
+        return NULL;
+    }
+    t->seekpoint = vlc_alloc(sys->chapter_count, sizeof(*t->seekpoint));
+    if (!t->seekpoint)
+    {
+        vlc_input_title_Delete(t);
+        return NULL;
+    }
+
+    for (ssize_t i = 0; i < sys->chapter_count; ++i)
+    {
+        t->seekpoint[i] = vlc_seekpoint_New();
+        if (!t->seekpoint[i])
+        {
+            vlc_input_title_Delete(t);
+            return NULL;
+        }
+        t->i_seekpoint++;
+        if (!sys->null_names
+         && asprintf(&t->seekpoint[i]->psz_name, "Mock Chapter %zu-%zu", idx, i)
+            == -1)
+        {
+            vlc_input_title_Delete(t);
+            return NULL;
+        }
+        t->seekpoint[i]->i_time_offset = i * sys->chapter_gap;
+    }
+    return t;
+}
+
+static int
+Control(demux_t *demux, int query, va_list args)
+{
+    struct demux_sys *sys = demux->p_sys;
+
+    switch (query)
+    {
+        case DEMUX_CAN_SEEK:
+            *va_arg(args, bool *) = sys->can_seek;
+            return VLC_SUCCESS;
+        case DEMUX_CAN_PAUSE:
+            *va_arg(args, bool *) = sys->can_pause;
+            return VLC_SUCCESS;
+        case DEMUX_CAN_CONTROL_PACE:
+            *va_arg(args, bool *) = sys->can_control_pace;
+            return VLC_SUCCESS;
+        case DEMUX_GET_PTS_DELAY:
+            *va_arg(args, vlc_tick_t *) = 0;
+            return VLC_SUCCESS;
+        case DEMUX_GET_META:
+            return VLC_EGENERIC;
+        case DEMUX_GET_SIGNAL:
+            return VLC_EGENERIC;
+        case DEMUX_SET_PAUSE_STATE:
+            return sys->can_pause ? VLC_SUCCESS : VLC_EGENERIC;
+        case DEMUX_SET_TITLE:
+            if (sys->title_count > 0)
+            {
+                int new_title = va_arg(args, int);
+                if (new_title >= sys->title_count)
+                    return VLC_EGENERIC;
+                sys->current_title = new_title;
+                sys->pts = VLC_TICK_0;
+                return VLC_SUCCESS;
+            }
+            return VLC_EGENERIC;
+        case DEMUX_SET_SEEKPOINT:
+            if (sys->chapter_gap > 0)
+            {
+                const int seekpoint_idx = va_arg(args, int);
+                if (seekpoint_idx < sys->chapter_count)
+                {
+                    sys->pts = seekpoint_idx * sys->chapter_gap;
+                    return VLC_SUCCESS;
+                }
+            }
+            return VLC_EGENERIC;
+        case DEMUX_TEST_AND_CLEAR_FLAGS:
+            return VLC_EGENERIC;
+        case DEMUX_GET_TITLE:
+            if (sys->title_count > 0)
+            {
+                *va_arg(args, int *) = sys->current_title;
+                return VLC_SUCCESS;
+            }
+            return VLC_EGENERIC;
+        case DEMUX_GET_SEEKPOINT:
+            if (sys->chapter_gap > 0)
+            {
+                *va_arg(args, int *) = sys->pts / sys->chapter_gap;
+                return VLC_SUCCESS;
+            }
+            return VLC_EGENERIC;
+        case DEMUX_GET_POSITION:
+            *va_arg(args, double *) = sys->pts / (double) sys->length;
+            return VLC_SUCCESS;
+        case DEMUX_SET_POSITION:
+            if (!sys->can_seek)
+                return VLC_EGENERIC;
+            sys->pts = va_arg(args, double) * sys->length;
+            return VLC_SUCCESS;
+        case DEMUX_GET_LENGTH:
+            *va_arg(args, vlc_tick_t *) = sys->length;
+            return VLC_SUCCESS;
+        case DEMUX_GET_TIME:
+            *va_arg(args, vlc_tick_t *) = sys->pts;
+            return VLC_SUCCESS;
+        case DEMUX_SET_TIME:
+            if (!sys->can_seek)
+                return VLC_EGENERIC;
+            sys->pts = va_arg(args, vlc_tick_t);
+            return VLC_SUCCESS;
+        case DEMUX_GET_TITLE_INFO:
+            if (sys->title_count > 0)
+            {
+                input_title_t ***titles = va_arg(args, input_title_t ***);
+                *titles = vlc_alloc(sys->title_count, sizeof(*titles));
+                if (!*titles)
+                    return VLC_ENOMEM;
+                for (ssize_t i = 0; i < sys->title_count; ++i)
+                {
+                    (*titles)[i] = CreateTitle(demux, i);
+                    if (!(*titles)[i])
+                    {
+                        while (i--)
+                            vlc_input_title_Delete((*titles)[i - 1]);
+                        free(*titles);
+                        *titles = NULL;
+                        return VLC_ENOMEM;
+                    }
+                }
+                *va_arg(args, int *) = sys->title_count;
+                *va_arg(args, int *) = 0;
+                *va_arg(args, int *) = 0;
+                return VLC_SUCCESS;
+            }
+            return VLC_EGENERIC;
+        case DEMUX_SET_GROUP_DEFAULT:
+            return VLC_EGENERIC;
+        case DEMUX_SET_GROUP_ALL:
+            return VLC_EGENERIC;
+        case DEMUX_SET_GROUP_LIST:
+            return VLC_EGENERIC;
+        case DEMUX_SET_ES:
+            return VLC_EGENERIC;
+        case DEMUX_SET_NEXT_DEMUX_TIME:
+            return VLC_EGENERIC;
+        case DEMUX_GET_FPS:
+            return VLC_EGENERIC;
+        case DEMUX_HAS_UNSUPPORTED_META:
+            return VLC_EGENERIC;
+        case DEMUX_GET_ATTACHMENTS:
+            return VLC_EGENERIC;
+        case DEMUX_CAN_RECORD:
+            *va_arg(args, bool *) = sys->can_record;
+            return VLC_SUCCESS;
+        case DEMUX_SET_RECORD_STATE:
+            return sys->can_record ? VLC_SUCCESS : VLC_EGENERIC;
+        case DEMUX_CAN_CONTROL_RATE:
+            *va_arg(args, bool *) = sys->can_control_rate;
+            return VLC_SUCCESS;
+        case DEMUX_SET_RATE:
+            return sys->can_control_rate ? VLC_SUCCESS : VLC_EGENERIC;
+        case DEMUX_IS_PLAYLIST:
+            return VLC_EGENERIC;
+        case DEMUX_NAV_ACTIVATE:
+            return VLC_EGENERIC;
+        case DEMUX_NAV_UP:
+            return VLC_EGENERIC;
+        case DEMUX_NAV_DOWN:
+            return VLC_EGENERIC;
+        case DEMUX_NAV_LEFT:
+            return VLC_EGENERIC;
+        case DEMUX_NAV_RIGHT:
+            return VLC_EGENERIC;
+        case DEMUX_NAV_POPUP:
+            return VLC_EGENERIC;
+        case DEMUX_NAV_MENU:
+            return VLC_EGENERIC;
+        default:
+            return VLC_EGENERIC;
+    }
+}
+
+static block_t *
+CreateAudioBlock(demux_t *demux, struct mock_track *track)
+{
+    struct demux_sys *sys = demux->p_sys;
+    const int64_t samples =
+        samples_from_vlc_tick(sys->step_length, track->fmt.audio.i_rate);
+    const int64_t bytes = samples / track->fmt.audio.i_frame_length
+                        * track->fmt.audio.i_bytes_per_frame;
+    block_t *b = block_Alloc(bytes);
+    if (!b)
+        return NULL;
+    memset(b->p_buffer, 0, b->i_buffer);
+    return b;
+}
+
+struct video_block
+{
+    block_t b;
+    picture_t *pic;
+};
+
+static void
+video_block_free_cb(block_t *b)
+{
+    struct video_block *video = container_of(b, struct video_block, b);
+    picture_Release(video->pic);
+    free(video);
+}
+
+static block_t *
+CreateVideoBlock(demux_t *demux, struct mock_track *track)
+{
+    struct demux_sys *sys = demux->p_sys;
+    picture_t *pic = picture_NewFromFormat(&track->fmt.video);
+    if (!pic)
+        return NULL;
+
+    struct video_block *video = malloc(sizeof(*video));
+    if (!video)
+    {
+        picture_Release(pic);
+        return NULL;
+    }
+    video->pic = pic;
+
+    static const struct vlc_block_callbacks cbs =
+    {
+        .free = video_block_free_cb
+    };
+
+    size_t block_len = 0;
+    for (int i = 0; i < pic->i_planes; ++i)
+        block_len += pic->p[i].i_lines * pic->p[i].i_pitch;
+    memset(pic->p[0].p_pixels, (sys->pts / VLC_TICK_FROM_MS(10)) % 255,
+           block_len);
+    return block_Init(&video->b, &cbs, pic->p[0].p_pixels, block_len);
+    (void) demux;
+}
+
+static block_t *
+CreateSubBlock(demux_t *demux, struct mock_track *track)
+{
+    struct demux_sys *sys = demux->p_sys;
+    char *text;
+    if (asprintf(&text, "subtitle @ %"PRId64, sys->pts) == -1)
+        return NULL;
+    size_t len = strlen(text) + 1;
+
+    block_t *b = block_Alloc(len);
+    if (!b)
+    {
+        free(text);
+        return NULL;
+    }
+
+    memcpy(b->p_buffer, text, len);
+    b->i_buffer = len;
+
+    free(text);
+    return b;
+    (void) track;
+}
+
+static int
+AppendMockTrack(demux_t *demux, const es_format_t *fmt, int group,
+                bool packetized)
+{
+    struct demux_sys *sys = demux->p_sys;
+    struct mock_track *mock_track = malloc(sizeof(*mock_track));
+    if (!mock_track)
+        return VLC_EGENERIC;
+    mock_track->fmt = *fmt;
+    mock_track->fmt.i_group = group;
+    mock_track->fmt.b_packetized = packetized;
+    mock_track->id = es_out_Add(demux->out, & mock_track->fmt);
+    if (!mock_track->id)
+    {
+        free(mock_track);
+        return VLC_ENOMEM;
+    }
+    bool success = vlc_vector_push(&sys->tracks, mock_track);
+    assert(success); (void) success; /* checked by reserve() */
+    return VLC_SUCCESS;
+}
+
+static int
+InitVideoTracks(demux_t *demux, int group, size_t count)
+{
+    struct demux_sys *sys = demux->p_sys;
+
+    if (count == 0)
+        return VLC_SUCCESS;
+
+    const vlc_chroma_description_t *desc =
+        vlc_fourcc_GetChromaDescription(sys->video_chroma);
+    if (!desc || desc->plane_count == 0)
+        sys->video_chroma = 0;
+
+    const bool frame_rate_ok =
+        sys->video_frame_rate != 0 && sys->video_frame_rate != UINT_MAX &&
+        sys->video_frame_rate_base != 0 && sys->video_frame_rate_base != UINT_MAX;
+    const bool chroma_ok = sys->video_chroma != 0;
+    const bool size_ok = sys->video_width != UINT_MAX &&
+                         sys->video_height != UINT_MAX;
+
+    if (sys->video_frame_rate == 0 || sys->video_frame_rate_base == 0
+     || sys->video_chroma == 0)
+    if (!frame_rate_ok || !chroma_ok || !size_ok)
+    {
+        if (!frame_rate_ok)
+            msg_Err(demux, "Invalid video frame rate");
+        if (!chroma_ok)
+            msg_Err(demux, "Invalid video chroma");
+        if (!size_ok)
+            msg_Err(demux, "Invalid video size");
+        return VLC_EGENERIC;
+    }
+
+    for (size_t i = 0; i < count; ++i)
+    {
+        es_format_t fmt;
+        es_format_Init(&fmt, VIDEO_ES, sys->video_chroma);
+        fmt.video.i_chroma = fmt.i_codec;
+        fmt.video.i_width = fmt.video.i_visible_width = sys->video_width;
+        fmt.video.i_height = fmt.video.i_visible_height = sys->video_height;
+        fmt.video.i_frame_rate = sys->video_frame_rate;
+        fmt.video.i_frame_rate_base = sys->video_frame_rate_base;
+
+        if (AppendMockTrack(demux, &fmt, group, sys->video_packetized))
+            return VLC_ENOMEM;
+    }
+    return VLC_SUCCESS;
+}
+
+static int
+InitAudioTracks(demux_t *demux, int group, size_t count)
+{
+    struct demux_sys *sys = demux->p_sys;
+
+    if (count == 0)
+        return VLC_SUCCESS;
+
+    const bool rate_ok = sys->audio_rate > 0 && sys->audio_rate != UINT_MAX;
+    const bool format_ok = aout_BitsPerSample(sys->audio_format) != 0;
+    const bool channels_ok = sys->audio_channels > 0 &&
+                             sys->audio_channels <= AOUT_CHAN_MAX;
+
+    if (!rate_ok || !format_ok || !channels_ok)
+    {
+        if (!rate_ok)
+            msg_Err(demux, "Invalid audio rate");
+        if (!format_ok)
+            msg_Err(demux, "Invalid audio format");
+        if (!channels_ok)
+            msg_Err(demux, "Invalid audio channels");
+        return VLC_EGENERIC;
+    }
+
+    uint16_t physical_channels = 0;
+    switch (sys->audio_channels)
+    {
+        case 1: physical_channels = AOUT_CHAN_CENTER; break;
+        case 2: physical_channels = AOUT_CHANS_2_0; break;
+        case 3: physical_channels = AOUT_CHANS_2_1; break;
+        case 4: physical_channels = AOUT_CHANS_4_0; break;
+        case 5: physical_channels = AOUT_CHANS_4_1; break;
+        case 6: physical_channels = AOUT_CHANS_6_0; break;
+        case 7: physical_channels = AOUT_CHANS_7_0; break;
+        case 8: physical_channels = AOUT_CHANS_7_1; break;
+        case 9: physical_channels = AOUT_CHANS_8_1; break;
+        default: vlc_assert_unreachable();
+    }
+
+    for (size_t i = 0; i < count; ++i)
+    {
+        es_format_t fmt;
+        es_format_Init(&fmt, AUDIO_ES, sys->audio_format);
+        fmt.audio.i_format = fmt.i_codec;
+        fmt.audio.i_rate = sys->audio_rate;
+        fmt.audio.i_physical_channels = physical_channels;
+        aout_FormatPrepare(&fmt.audio);
+
+        if (AppendMockTrack(demux, &fmt, group, sys->audio_packetized))
+            return VLC_ENOMEM;
+    }
+
+    return VLC_SUCCESS;
+}
+
+static int
+InitSubTracks(demux_t *demux, int group, size_t count)
+{
+    struct demux_sys *sys = demux->p_sys;
+
+    if (count == 0)
+        return VLC_SUCCESS;
+
+    for (size_t i = 0; i < count; ++i)
+    {
+        es_format_t fmt;
+        es_format_Init(&fmt, SPU_ES, VLC_CODEC_SUBT);
+
+        if (AppendMockTrack(demux, &fmt, group, sys->sub_packetized))
+            return VLC_ENOMEM;
+    }
+
+    return VLC_SUCCESS;
+}
+
+static int
+Demux(demux_t *demux)
+{
+    struct demux_sys *sys = demux->p_sys;
+
+    if (sys->error)
+        return VLC_DEMUXER_EGENERIC;
+
+    struct mock_track *track;
+    vlc_vector_foreach(track, &sys->tracks)
+    {
+        block_t *block;
+        switch (track->fmt.i_cat)
+        {
+            case AUDIO_ES:
+                block = CreateAudioBlock(demux, track);
+                break;
+            case VIDEO_ES:
+                block = CreateVideoBlock(demux, track);
+                break;
+            case SPU_ES:
+                block = CreateSubBlock(demux, track);
+                break;
+            default:
+                vlc_assert_unreachable();
+        }
+        if (!block)
+            return VLC_DEMUXER_EGENERIC;
+        block->i_length = sys->step_length;
+        block->i_pts = block->i_dts = sys->pts;
+        int ret = es_out_Send(demux->out, track->id, block);
+        if (ret != VLC_SUCCESS)
+            return VLC_DEMUXER_EGENERIC;
+    }
+    es_out_SetPCR(demux->out, sys->pts);
+    sys->pts += sys->step_length;
+    if (sys->pts > sys->length)
+        sys->pts = sys->length;
+    return sys->pts == sys->length ? VLC_DEMUXER_EOF : VLC_DEMUXER_SUCCESS;
+}
+
+static void
+Close(vlc_object_t *obj)
+{
+    demux_t *demux = (demux_t*)obj;
+    struct demux_sys *sys = demux->p_sys;
+
+    struct mock_track *track;
+    vlc_vector_foreach(track, &sys->tracks)
+    {
+        es_out_Del(demux->out, track->id);
+        free(track);
+    }
+    vlc_vector_clear(&sys->tracks);
+}
+
+static int
+Open(vlc_object_t *obj)
+{
+    demux_t *demux = (demux_t*)obj;
+
+    if (demux->out == NULL)
+        return VLC_EGENERIC;
+    struct demux_sys *sys = vlc_obj_malloc(obj, sizeof(*sys));
+    if (!sys)
+        return VLC_ENOMEM;
+
+    demux->p_sys = sys;
+
+    if (var_LocationParse(obj, demux->psz_location, "mock-") != VLC_SUCCESS)
+        return VLC_ENOMEM;
+
+#define X(var_name, type, module_header_type, getter, default_value) \
+    sys->var_name = getter(obj, "mock-"#var_name);
+    LIST_OPTIONS
+#undef X
+
+    if (sys->chapter_count > 0 && sys->title_count == 0)
+        sys->title_count++;
+
+    const bool length_ok = sys->length >= 0;
+    const bool tracks_ok = sys->video_track_count >= 0 &&
+                           sys->audio_track_count >= 0 &&
+                           sys->sub_track_count >= 0;
+    const bool titles_ok = sys->title_count >= 0;
+    const bool chapters_ok = sys->chapter_count >= 0;
+    const bool programs_ok = sys->program_count >= 0;
+
+    if (!length_ok || !tracks_ok || !titles_ok || !chapters_ok || !programs_ok)
+    {
+        if (!length_ok)
+            msg_Err(demux, "Invalid length");
+        if (!tracks_ok)
+            msg_Err(demux, "Invalid track count");
+        if (!titles_ok)
+            msg_Err(demux, "Invalid title count");
+        if (!chapters_ok)
+            msg_Err(demux, "Invalid chapter count");
+        if (!programs_ok)
+            msg_Err(demux, "Invalid program count");
+        return VLC_EGENERIC;
+    }
+
+    if (sys->program_count == 0)
+        sys->program_count = 1;
+    size_t track_count = (sys->video_track_count + sys->audio_track_count +
+                          sys->sub_track_count) * sys->program_count;
+    vlc_vector_init(&sys->tracks);
+    if (track_count > 0)
+    {
+        bool success = vlc_vector_reserve(&sys->tracks, track_count);
+        if (!success)
+            return VLC_ENOMEM;
+    }
+
+    int ret;
+    for (ssize_t i = 0; i < sys->program_count; ++i)
+    {
+        ret = InitVideoTracks(demux, i, sys->video_track_count);
+        if (ret != VLC_SUCCESS)
+            goto error;
+
+        ret = InitAudioTracks(demux, i, sys->audio_track_count);
+        if (ret != VLC_SUCCESS)
+            goto error;
+
+        ret = InitSubTracks(demux, i, sys->sub_track_count);
+        if (ret != VLC_SUCCESS)
+            goto error;
+    }
+
+    if (sys->video_track_count > 0)
+        sys->step_length = VLC_TICK_FROM_SEC(1) * sys->video_frame_rate_base
+                         / sys->video_frame_rate;
+    else
+        sys->step_length = VLC_TICK_FROM_MS(100);
+
+    sys->pts = VLC_TICK_0;
+    sys->current_title = 0;
+    sys->chapter_gap = sys->chapter_count > 0 ?
+                       (sys->length / sys->chapter_count) : 0;
+
+    demux->pf_control = Control;
+    demux->pf_demux = Demux;
+
+    return VLC_SUCCESS;
+error:
+    Close(obj);
+    demux->p_sys = NULL;
+    return ret;
+}
+
+#define X(var_name, type, module_header_type, getter, default_value) \
+    module_header_type("mock-"#var_name, default_value, NULL, NULL, true) \
+    change_volatile() \
+    change_safe()
+
+vlc_module_begin()
+    set_description("mock access demux")
+    set_capability("access", 0)
+    set_category(CAT_INPUT)
+    set_subcategory(SUBCAT_INPUT_ACCESS)
+    set_callbacks(Open, Close)
+    LIST_OPTIONS
+    add_shortcut("mock")
+vlc_module_end()
+
+#undef X



More information about the vlc-commits mailing list