mirror of https://github.com/mpv-player/mpv
333 lines
7.2 KiB
C
333 lines
7.2 KiB
C
/*
|
|
Copyright (C) 2002 Michael Niedermayer <michaelni@gmx.at>
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <inttypes.h>
|
|
|
|
#include "config.h"
|
|
#include "mp_msg.h"
|
|
|
|
#include "img_format.h"
|
|
#include "mp_image.h"
|
|
#include "vf.h"
|
|
|
|
#include "libvo/fastmemcpy.h"
|
|
|
|
//===========================================================================//
|
|
|
|
#include <inttypes.h>
|
|
#include <math.h>
|
|
|
|
#define MAX(a,b) ((a) > (b) ? (a) : (b))
|
|
#define MIN(a,b) ((a) < (b) ? (a) : (b))
|
|
#define ABS(a,b) ((a) > 0 ? (a) : -(a))
|
|
|
|
#define WIDTH 512
|
|
#define HEIGHT 512
|
|
|
|
struct vf_priv_s {
|
|
int frame_num;
|
|
};
|
|
|
|
static int config(struct vf_instance_s* vf,
|
|
int width, int height, int d_width, int d_height,
|
|
unsigned int flags, unsigned int outfmt){
|
|
|
|
if(vf_next_query_format(vf,IMGFMT_YV12)<=0){
|
|
printf("yv12 not supported by next filter/vo :(\n");
|
|
return 0;
|
|
}
|
|
|
|
//hmm whats the meaning of these ... ;)
|
|
d_width= width= WIDTH;
|
|
d_height= height= HEIGHT;
|
|
|
|
return vf_next_config(vf,width,height,d_width,d_height,flags,IMGFMT_YV12);
|
|
}
|
|
|
|
static double c[64];
|
|
|
|
static void initIdct(void)
|
|
{
|
|
int i;
|
|
|
|
for (i=0; i<8; i++)
|
|
{
|
|
double s= i==0 ? sqrt(0.125) : 0.5;
|
|
int j;
|
|
|
|
for(j=0; j<8; j++)
|
|
c[i*8+j]= s*cos((3.141592654/8.0)*i*(j+0.5));
|
|
}
|
|
}
|
|
|
|
|
|
static void idct(uint8_t *dst, int dstStride, int src[64])
|
|
{
|
|
int i, j, k;
|
|
double tmp[64];
|
|
|
|
for(i=0; i<8; i++)
|
|
{
|
|
for(j=0; j<8; j++)
|
|
{
|
|
double sum= 0.0;
|
|
|
|
for(k=0; k<8; k++)
|
|
sum+= c[k*8+j]*src[8*i+k];
|
|
|
|
tmp[8*i+j]= sum;
|
|
}
|
|
}
|
|
|
|
for(j=0; j<8; j++)
|
|
{
|
|
for(i=0; i<8; i++)
|
|
{
|
|
int v;
|
|
double sum= 0.0;
|
|
|
|
for(k=0; k<8; k++)
|
|
sum+= c[k*8+i]*tmp[8*k+j];
|
|
|
|
v= (int)floor(sum+0.5);
|
|
if(v<0) v=0;
|
|
else if(v>255) v=255;
|
|
|
|
dst[dstStride*i + j] = v;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void drawDc(uint8_t *dst, int stride, int color, int w, int h)
|
|
{
|
|
int y;
|
|
for(y=0; y<h; y++)
|
|
{
|
|
int x;
|
|
for(x=0; x<w; x++)
|
|
{
|
|
dst[x + y*stride]= color;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void drawBasis(uint8_t *dst, int stride, int amp, int freq, int dc)
|
|
{
|
|
int src[64];
|
|
|
|
memset(src, 0, 64*sizeof(int));
|
|
src[0]= dc;
|
|
if(amp) src[freq]= amp;
|
|
idct(dst, stride, src);
|
|
}
|
|
|
|
static void drawCbp(uint8_t *dst[3], int stride[3], int cbp, int amp, int dc)
|
|
{
|
|
if(cbp&1) drawBasis(dst[0] , stride[0], amp, 1, dc);
|
|
if(cbp&2) drawBasis(dst[0]+8 , stride[0], amp, 1, dc);
|
|
if(cbp&4) drawBasis(dst[0]+ 8*stride[0], stride[0], amp, 1, dc);
|
|
if(cbp&8) drawBasis(dst[0]+8+8*stride[0], stride[0], amp, 1, dc);
|
|
if(cbp&16)drawBasis(dst[1] , stride[1], amp, 1, dc);
|
|
if(cbp&32)drawBasis(dst[2] , stride[2], amp, 1, dc);
|
|
}
|
|
|
|
static void dc1Test(uint8_t *dst, int stride, int w, int h, int off)
|
|
{
|
|
const int step= MAX(256/(w*h/256), 1);
|
|
int y;
|
|
int color=off;
|
|
for(y=0; y<h; y+=16)
|
|
{
|
|
int x;
|
|
for(x=0; x<w; x+=16)
|
|
{
|
|
drawDc(dst + x + y*stride, stride, color, 8, 8);
|
|
color+=step;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void freq1Test(uint8_t *dst, int stride, int off)
|
|
{
|
|
int y;
|
|
int freq=0;
|
|
for(y=0; y<8*16; y+=16)
|
|
{
|
|
int x;
|
|
for(x=0; x<8*16; x+=16)
|
|
{
|
|
drawBasis(dst + x + y*stride, stride, 4*(96+off), freq, 128*8);
|
|
freq++;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void amp1Test(uint8_t *dst, int stride, int off)
|
|
{
|
|
int y;
|
|
int amp=off;
|
|
for(y=0; y<16*16; y+=16)
|
|
{
|
|
int x;
|
|
for(x=0; x<16*16; x+=16)
|
|
{
|
|
drawBasis(dst + x + y*stride, stride, 4*(amp), 1, 128*8);
|
|
amp++;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void cbp1Test(uint8_t *dst[3], int stride[3], int off)
|
|
{
|
|
int y;
|
|
int cbp=0;
|
|
for(y=0; y<16*8; y+=16)
|
|
{
|
|
int x;
|
|
for(x=0; x<16*8; x+=16)
|
|
{
|
|
uint8_t *dst1[3];
|
|
dst1[0]= dst[0] + x*2 + y*2*stride[0];
|
|
dst1[1]= dst[1] + x + y*stride[1];
|
|
dst1[2]= dst[2] + x + y*stride[2];
|
|
|
|
drawCbp(dst1, stride, cbp, (64+off)*4, 128*8);
|
|
cbp++;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void mv1Test(uint8_t *dst, int stride, int off)
|
|
{
|
|
int y;
|
|
for(y=0; y<16*16; y++)
|
|
{
|
|
int x;
|
|
if(y&16) continue;
|
|
for(x=0; x<16*16; x++)
|
|
{
|
|
dst[x + y*stride]= x + off*8/(y/32+1);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void ring1Test(uint8_t *dst, int stride, int off)
|
|
{
|
|
int y;
|
|
int color=0;
|
|
for(y=off; y<16*16; y+=16)
|
|
{
|
|
int x;
|
|
for(x=off; x<16*16; x+=16)
|
|
{
|
|
drawDc(dst + x + y*stride, stride, ((x+y)&16) ? color : -color, 16, 16);
|
|
// dst[x + y*stride]= 255 + (off&1);
|
|
color++;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void ring2Test(uint8_t *dst, int stride, int off)
|
|
{
|
|
int y;
|
|
for(y=0; y<16*16; y++)
|
|
{
|
|
int x;
|
|
for(x=0; x<16*16; x++)
|
|
{
|
|
double d= sqrt((x-8*16)*(x-8*16) + (y-8*16)*(y-8*16));
|
|
double r= d/20 - (int)(d/20);
|
|
if(r<off/30.0)
|
|
{
|
|
dst[x + y*stride]= 255;
|
|
dst[x + y*stride+256]= 0;
|
|
}
|
|
else{
|
|
dst[x + y*stride]= x;
|
|
dst[x + y*stride+256]= x;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static int put_image(struct vf_instance_s* vf, mp_image_t *mpi){
|
|
mp_image_t *dmpi;
|
|
int frame= vf->priv->frame_num;
|
|
|
|
// hope we'll get DR buffer:
|
|
dmpi=vf_get_image(vf->next,IMGFMT_YV12,
|
|
MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
|
|
WIDTH, HEIGHT);
|
|
|
|
// clean
|
|
memset(dmpi->planes[0], 0, dmpi->stride[0]*dmpi->h);
|
|
memset(dmpi->planes[1], 128, dmpi->stride[1]*dmpi->h>>dmpi->chroma_y_shift);
|
|
memset(dmpi->planes[2], 128, dmpi->stride[2]*dmpi->h>>dmpi->chroma_y_shift);
|
|
|
|
if(frame%30)
|
|
{
|
|
switch(frame/30)
|
|
{
|
|
case 0: dc1Test(dmpi->planes[0], dmpi->stride[0], 256, 256, frame%30); break;
|
|
case 1: dc1Test(dmpi->planes[1], dmpi->stride[1], 256, 256, frame%30); break;
|
|
case 2: freq1Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
|
|
case 3: freq1Test(dmpi->planes[1], dmpi->stride[1], frame%30); break;
|
|
case 4: amp1Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
|
|
case 5: amp1Test(dmpi->planes[1], dmpi->stride[1], frame%30); break;
|
|
case 6: cbp1Test(dmpi->planes , dmpi->stride , frame%30); break;
|
|
case 7: mv1Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
|
|
case 8: ring1Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
|
|
case 9: ring2Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
|
|
}
|
|
}
|
|
|
|
frame++;
|
|
vf->priv->frame_num= frame;
|
|
return vf_next_put_image(vf,dmpi);
|
|
}
|
|
|
|
//===========================================================================//
|
|
|
|
static int query_format(struct vf_instance_s* vf, unsigned int fmt){
|
|
return vf_next_query_format(vf,IMGFMT_YV12) & (~VFCAP_CSP_SUPPORTED_BY_HW);
|
|
}
|
|
|
|
static int open(vf_instance_t *vf, char* args){
|
|
vf->config=config;
|
|
vf->put_image=put_image;
|
|
vf->query_format=query_format;
|
|
vf->priv=malloc(sizeof(struct vf_priv_s));
|
|
vf->priv->frame_num= args ? atoi(args) : 0;
|
|
initIdct();
|
|
return 1;
|
|
}
|
|
|
|
vf_info_t vf_info_test = {
|
|
"test pattern generator",
|
|
"test",
|
|
"Michael Niedermayer",
|
|
"",
|
|
open,
|
|
NULL
|
|
};
|
|
|
|
//===========================================================================//
|