Commit a03e79ed authored by Paul B Mahol's avatar Paul B Mahol

lavfi: fix broken logic in metadata handling

Signed-off-by: 's avatarPaul B Mahol <onemda@gmail.com>
parent 970c8df7
......@@ -62,7 +62,7 @@ static int query_formats(AVFilterContext *ctx)
#define SET_META(key, value) \
snprintf(buf, sizeof(buf), "%d", value); \
av_dict_set(metadata, #key, buf, 0); \
av_dict_set(metadata, key, buf, 0);
static int filter_frame(AVFilterLink *inlink, AVFrame *frame)
{
......
......@@ -114,7 +114,7 @@ static int config_input(AVFilterLink *inlink)
#define SET_META(key, value) \
snprintf(buf, sizeof(buf), "%d", value); \
av_dict_set(metadata, #key, buf, 0) \
av_dict_set(metadata, key, buf, 0)
static int filter_frame(AVFilterLink *inlink, AVFrame *frame)
{
......
......@@ -98,16 +98,24 @@ void compute_images_mse(const uint8_t *main_data[4], const int main_linesizes[4]
}
}
#define SET_META(key, comp, value) \
snprintf(buf, sizeof(buf), "%0.2f", value); \
av_dict_set(metadata, #key #comp, buf, 0); \
static void set_meta(AVDictionary **metadata, const char *key, char comp, float d)
{
char value[128];
snprintf(value, sizeof(value), "%0.2f", d);
if (comp) {
char key2[128];
snprintf(key2, sizeof(key2), "%s%c", key, comp);
av_dict_set(metadata, key2, value, 0);
} else {
av_dict_set(metadata, key, value, 0);
}
}
static AVFrame *do_psnr(AVFilterContext *ctx, AVFrame *main,
const AVFrame *ref)
{
PSNRContext *s = ctx->priv;
double comp_mse[4], mse = 0;
char buf[32];
int j, c;
AVDictionary **metadata = avpriv_frame_get_metadatap(main);
......@@ -127,10 +135,10 @@ static AVFrame *do_psnr(AVFilterContext *ctx, AVFrame *main,
for (j = 0; j < s->desc->nb_components; j++) {
c = s->is_rgb ? s->rgba_map[j] : j;
SET_META("lavfi.psnr.mse.", s->comps[j], comp_mse[c]);
SET_META("lavfi.psnr.mse_avg", "", mse);
SET_META("lavfi.psnr.s.", s->comps[j], get_psnr(comp_mse[c], 1, s->max[c]));
SET_META("lavfi.psnr.s_avg", "", get_psnr(mse, 1, s->average_max));
set_meta(metadata, "lavfi.psnr.mse.", s->comps[j], comp_mse[c]);
set_meta(metadata, "lavfi.psnr.mse_avg", 0, mse);
set_meta(metadata, "lavfi.psnr.s.", s->comps[j], get_psnr(comp_mse[c], 1, s->max[c]));
set_meta(metadata, "lavfi.psnr.s_avg", 0, get_psnr(mse, 1, s->average_max));
}
if (s->stats_file) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment