--- plugin_psnr.c 2003/03/15 14:38:34 1.1 +++ plugin_psnr.c 2003/03/15 14:38:34 1.1.2.1 @@ -0,0 +1,60 @@ +#include "../xvid.h" +#include "../image/image.h" + + +int xvid_plugin_psnr(void * handle, int opt, void * param1, void * param2) +{ + switch(opt) + { + case XVID_PLG_INFO : + { + xvid_plg_info_t * info = (xvid_plg_info_t*)param1; + info->flags = XVID_REQORIGINAL; + return 0; + } + + case XVID_PLG_CREATE : + case XVID_PLG_DESTROY : + case XVID_PLG_BEFORE : + return 0; + + case XVID_PLG_AFTER : + { + xvid_plg_data_t * data = (xvid_plg_data_t*)param1; + + long sse_y = plane_sse(data->original.plane[0], data->current.plane[0], + data->current.stride[0], data->width, data->height); + + long sse_u = plane_sse(data->original.plane[1], data->current.plane[1], + data->current.stride[1], data->width/2, data->height/2); + + long sse_v = plane_sse(data->original.plane[2], data->current.plane[2], + data->current.stride[2], data->width/2, data->height/2); + + printf("y_psnr=%2.2f u_psnr=%2.2f v_psnr=%2.2f\n", + sse_to_PSNR(sse_y, data->width*data->height), + sse_to_PSNR(sse_u, data->width*data->height/4), + sse_to_PSNR(sse_v, data->width*data->height/4)); + + { + IMAGE img; + char tmp[100]; + img.y = data->original.plane[0]; + img.u = data->original.plane[1]; + img.v = data->original.plane[2]; + sprintf(tmp, "ori-%03i.pgm", data->frame_num); + image_dump_yuvpgm(&img, data->original.stride[0], data->width, data->height, tmp); + + img.y = data->current.plane[0]; + img.u = data->current.plane[1]; + img.v = data->current.plane[2]; + sprintf(tmp, "enc-%03i.pgm", data->frame_num); + image_dump_yuvpgm(&img, data->reference.stride[0], data->width, data->height, tmp); + } + + return 0; + } + } + + return XVID_ERR_FAIL; +}