Statistics
| Branch: | Revision:

ffmpeg / libavfilter / libmpcodecs / vf_rgbtest.c @ e4852fb3

History | View | Annotate | Download (5.02 KB)

1
/*
2
 * This file is part of MPlayer.
3
 *
4
 * MPlayer is free software; you can redistribute it and/or modify
5
 * it under the terms of the GNU General Public License as published by
6
 * the Free Software Foundation; either version 2 of the License, or
7
 * (at your option) any later version.
8
 *
9
 * MPlayer is distributed in the hope that it will be useful,
10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 * GNU General Public License for more details.
13
 *
14
 * You should have received a copy of the GNU General Public License along
15
 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
16
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
17
 */
18

    
19
#include <stdio.h>
20
#include <stdlib.h>
21
#include <string.h>
22
#include <inttypes.h>
23

    
24
#include "config.h"
25
#include "mp_msg.h"
26

    
27
#include "img_format.h"
28
#include "mp_image.h"
29
#include "vf.h"
30

    
31
//===========================================================================//
32

    
33
struct vf_priv_s {
34
    unsigned int fmt;
35
    int w, h;
36
};
37

    
38
static unsigned int getfmt(unsigned int outfmt){
39
    switch(outfmt){
40
    case IMGFMT_RGB12:
41
    case IMGFMT_RGB15:
42
    case IMGFMT_RGB16:
43
    case IMGFMT_RGB24:
44
    case IMGFMT_RGBA:
45
    case IMGFMT_ARGB:
46
    case IMGFMT_BGR12:
47
    case IMGFMT_BGR15:
48
    case IMGFMT_BGR16:
49
    case IMGFMT_BGR24:
50
    case IMGFMT_BGRA:
51
    case IMGFMT_ABGR:
52
        return outfmt;
53
    }
54
    return 0;
55
}
56

    
57
static void put_pixel(uint8_t *buf, int x, int y, int stride, int r, int g, int b, int fmt){
58
    switch(fmt){
59
    case IMGFMT_BGR12: ((uint16_t*)(buf + y*stride))[x]=
60
                           ((r >> 4) << 8) | ((g >> 4) << 4) | (b >> 4);
61
    break;
62
    case IMGFMT_RGB12: ((uint16_t*)(buf + y*stride))[x]=
63
                           ((b >> 4) << 8) | ((g >> 4) << 4) | (r >> 4);
64
    break;
65
    case IMGFMT_BGR15: ((uint16_t*)(buf + y*stride))[x]= ((r>>3)<<10) | ((g>>3)<<5) | (b>>3);
66
    break;
67
    case IMGFMT_RGB15: ((uint16_t*)(buf + y*stride))[x]= ((b>>3)<<10) | ((g>>3)<<5) | (r>>3);
68
    break;
69
    case IMGFMT_BGR16: ((uint16_t*)(buf + y*stride))[x]= ((r>>3)<<11) | ((g>>2)<<5) | (b>>3);
70
    break;
71
    case IMGFMT_RGB16: ((uint16_t*)(buf + y*stride))[x]= ((b>>3)<<11) | ((g>>2)<<5) | (r>>3);
72
    break;
73
    case IMGFMT_RGB24:
74
        buf[3*x + y*stride + 0]= r;
75
        buf[3*x + y*stride + 1]= g;
76
        buf[3*x + y*stride + 2]= b;
77
    break;
78
    case IMGFMT_BGR24:
79
        buf[3*x + y*stride + 0]= b;
80
        buf[3*x + y*stride + 1]= g;
81
        buf[3*x + y*stride + 2]= r;
82
    break;
83
    case IMGFMT_RGBA:
84
        buf[4*x + y*stride + 0]= r;
85
        buf[4*x + y*stride + 1]= g;
86
        buf[4*x + y*stride + 2]= b;
87
    break;
88
    case IMGFMT_BGRA:
89
        buf[4*x + y*stride + 0]= b;
90
        buf[4*x + y*stride + 1]= g;
91
        buf[4*x + y*stride + 2]= r;
92
    break;
93
    case IMGFMT_ARGB:
94
        buf[4*x + y*stride + 1]= r;
95
        buf[4*x + y*stride + 2]= g;
96
        buf[4*x + y*stride + 3]= b;
97
    break;
98
    case IMGFMT_ABGR:
99
        buf[4*x + y*stride + 1]= b;
100
        buf[4*x + y*stride + 2]= g;
101
        buf[4*x + y*stride + 3]= r;
102
    break;
103
    }
104
}
105

    
106
static int config(struct vf_instance *vf,
107
        int width, int height, int d_width, int d_height,
108
        unsigned int flags, unsigned int outfmt){
109
    if (vf->priv->w > 0) { d_width  = width  = vf->priv->w; }
110
    if (vf->priv->h > 0) { d_height = height = vf->priv->h; }
111
    vf->priv->fmt=getfmt(outfmt);
112
    mp_msg(MSGT_VFILTER,MSGL_V,"rgb test format:%s\n", vo_format_name(outfmt));
113
    return vf_next_config(vf,width,height,d_width,d_height,flags,vf->priv->fmt);
114
}
115

    
116
static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts){
117
    mp_image_t *dmpi;
118
    int x, y;
119
    int w = vf->priv->w > 0 ? vf->priv->w : mpi->w;
120
    int h = vf->priv->h > 0 ? vf->priv->h : mpi->h;
121

    
122
    // hope we'll get DR buffer:
123
    dmpi=vf_get_image(vf->next,vf->priv->fmt,
124
        MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
125
        w, h);
126

    
127
     for(y=0; y<h; y++){
128
         for(x=0; x<w; x++){
129
             int c= 256*x/w;
130
             int r=0,g=0,b=0;
131

    
132
             if(3*y<h)        r=c;
133
             else if(3*y<2*h) g=c;
134
             else                  b=c;
135

    
136
             put_pixel(dmpi->planes[0], x, y, dmpi->stride[0], r, g, b, vf->priv->fmt);
137
         }
138
     }
139

    
140
    return vf_next_put_image(vf,dmpi, pts);
141
}
142

    
143
//===========================================================================//
144

    
145
static int query_format(struct vf_instance *vf, unsigned int outfmt){
146
    unsigned int fmt=getfmt(outfmt);
147
    if(!fmt) return 0;
148
    return vf_next_query_format(vf,fmt) & (~VFCAP_CSP_SUPPORTED_BY_HW);
149
}
150

    
151
static int vf_open(vf_instance_t *vf, char *args){
152
    vf->config=config;
153
    vf->put_image=put_image;
154
    vf->query_format=query_format;
155
    vf->priv=malloc(sizeof(struct vf_priv_s));
156
    vf->priv->w = vf->priv->h = 0;
157
    if (args)
158
        sscanf(args, "%d:%d", &vf->priv->w, &vf->priv->h);
159
    return 1;
160
}
161

    
162
const vf_info_t vf_info_rgbtest = {
163
    "rgbtest",
164
    "rgbtest",
165
    "Michael Niedermayer",
166
    "",
167
    vf_open,
168
    NULL
169
};
170

    
171
//===========================================================================//