yading@10: /* yading@10: * Copyright (c) 2013 Clément Bœsch yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: #include "libavutil/lfg.h" yading@10: #include "libavutil/opt.h" yading@10: #include "libavutil/random_seed.h" yading@10: #include "audio.h" yading@10: #include "video.h" yading@10: yading@10: enum mode { yading@10: MODE_NONE, yading@10: MODE_RO, yading@10: MODE_RW, yading@10: MODE_TOGGLE, yading@10: MODE_RANDOM, yading@10: NB_MODES yading@10: }; yading@10: yading@10: typedef struct { yading@10: const AVClass *class; yading@10: AVLFG lfg; yading@10: int64_t random_seed; yading@10: enum mode mode; yading@10: } PermsContext; yading@10: yading@10: #define OFFSET(x) offsetof(PermsContext, x) yading@10: #define FLAGS AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_VIDEO_PARAM yading@10: yading@10: static const AVOption options[] = { yading@10: { "mode", "select permissions mode", OFFSET(mode), AV_OPT_TYPE_INT, {.i64 = MODE_NONE}, MODE_NONE, NB_MODES-1, FLAGS, "mode" }, yading@10: { "none", "do nothing", 0, AV_OPT_TYPE_CONST, {.i64 = MODE_NONE}, INT_MIN, INT_MAX, FLAGS, "mode" }, yading@10: { "ro", "set all output frames read-only", 0, AV_OPT_TYPE_CONST, {.i64 = MODE_RO}, INT_MIN, INT_MAX, FLAGS, "mode" }, yading@10: { "rw", "set all output frames writable", 0, AV_OPT_TYPE_CONST, {.i64 = MODE_RW}, INT_MIN, INT_MAX, FLAGS, "mode" }, yading@10: { "toggle", "switch permissions", 0, AV_OPT_TYPE_CONST, {.i64 = MODE_TOGGLE}, INT_MIN, INT_MAX, FLAGS, "mode" }, yading@10: { "random", "set permissions randomly", 0, AV_OPT_TYPE_CONST, {.i64 = MODE_RANDOM}, INT_MIN, INT_MAX, FLAGS, "mode" }, yading@10: { "seed", "set the seed for the random mode", OFFSET(random_seed), AV_OPT_TYPE_INT64, {.i64 = -1}, -1, UINT32_MAX, FLAGS }, yading@10: { NULL } yading@10: }; yading@10: yading@10: static av_cold int init(AVFilterContext *ctx) yading@10: { yading@10: PermsContext *perms = ctx->priv; yading@10: yading@10: if (perms->mode == MODE_RANDOM) { yading@10: uint32_t seed; yading@10: yading@10: if (perms->random_seed == -1) yading@10: perms->random_seed = av_get_random_seed(); yading@10: seed = perms->random_seed; yading@10: av_log(ctx, AV_LOG_INFO, "random seed: 0x%08x\n", seed); yading@10: av_lfg_init(&perms->lfg, seed); yading@10: } yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: enum perm { RO, RW }; yading@10: static const char *perm_str[2] = { "RO", "RW" }; yading@10: yading@10: static int filter_frame(AVFilterLink *inlink, AVFrame *frame) yading@10: { yading@10: int ret; yading@10: AVFilterContext *ctx = inlink->dst; yading@10: PermsContext *perms = ctx->priv; yading@10: AVFrame *out = frame; yading@10: enum perm in_perm = av_frame_is_writable(frame) ? RW : RO; yading@10: enum perm out_perm; yading@10: yading@10: switch (perms->mode) { yading@10: case MODE_TOGGLE: out_perm = in_perm == RO ? RW : RO; break; yading@10: case MODE_RANDOM: out_perm = av_lfg_get(&perms->lfg) & 1 ? RW : RO; break; yading@10: case MODE_RO: out_perm = RO; break; yading@10: case MODE_RW: out_perm = RW; break; yading@10: default: out_perm = in_perm; break; yading@10: } yading@10: yading@10: av_log(ctx, AV_LOG_VERBOSE, "%s -> %s%s\n", yading@10: perm_str[in_perm], perm_str[out_perm], yading@10: in_perm == out_perm ? " (no-op)" : ""); yading@10: yading@10: if (in_perm == RO && out_perm == RW) { yading@10: if ((ret = av_frame_make_writable(frame)) < 0) yading@10: return ret; yading@10: } else if (in_perm == RW && out_perm == RO) { yading@10: out = av_frame_clone(frame); yading@10: if (!out) yading@10: return AVERROR(ENOMEM); yading@10: } yading@10: yading@10: ret = ff_filter_frame(ctx->outputs[0], out); yading@10: yading@10: if (in_perm == RW && out_perm == RO) yading@10: av_frame_free(&frame); yading@10: return ret; yading@10: } yading@10: yading@10: #if CONFIG_APERMS_FILTER yading@10: yading@10: #define aperms_options options yading@10: AVFILTER_DEFINE_CLASS(aperms); yading@10: yading@10: static const AVFilterPad aperms_inputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_AUDIO, yading@10: .filter_frame = filter_frame, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: static const AVFilterPad aperms_outputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_AUDIO, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: AVFilter avfilter_af_aperms = { yading@10: .name = "aperms", yading@10: .description = NULL_IF_CONFIG_SMALL("Set permissions for the output audio frame."), yading@10: .init = init, yading@10: .priv_size = sizeof(PermsContext), yading@10: .inputs = aperms_inputs, yading@10: .outputs = aperms_outputs, yading@10: .priv_class = &aperms_class, yading@10: }; yading@10: #endif /* CONFIG_APERMS_FILTER */ yading@10: yading@10: #if CONFIG_PERMS_FILTER yading@10: yading@10: #define perms_options options yading@10: AVFILTER_DEFINE_CLASS(perms); yading@10: yading@10: static const AVFilterPad perms_inputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .filter_frame = filter_frame, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: static const AVFilterPad perms_outputs[] = { yading@10: { yading@10: .name = "default", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: }, yading@10: { NULL } yading@10: }; yading@10: yading@10: AVFilter avfilter_vf_perms = { yading@10: .name = "perms", yading@10: .description = NULL_IF_CONFIG_SMALL("Set permissions for the output video frame."), yading@10: .init = init, yading@10: .priv_size = sizeof(PermsContext), yading@10: .inputs = perms_inputs, yading@10: .outputs = perms_outputs, yading@10: .priv_class = &perms_class, yading@10: }; yading@10: #endif /* CONFIG_PERMS_FILTER */