Statistics
| Branch: | Revision:

ffmpeg / libavfilter / libmpcodecs / vf_rectangle.c @ e4852fb3

History | View | Annotate | Download (5.56 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 "mp_image.h"
23
#include "mp_msg.h"
24
#include "vf.h"
25

    
26
#include "libvo/fastmemcpy.h"
27
#include "libavutil/common.h"
28

    
29
struct vf_priv_s {
30
    int x, y, w, h;
31
};
32

    
33
static int
34
config(struct vf_instance *vf,
35
       int width, int height, int d_width, int d_height,
36
       unsigned int flags, unsigned int outfmt)
37
{
38
    if (vf->priv->w < 0 || width < vf->priv->w)
39
        vf->priv->w = width;
40
    if (vf->priv->h < 0 || height < vf->priv->h)
41
        vf->priv->h = height;
42
    if (vf->priv->x < 0)
43
        vf->priv->x = (width - vf->priv->w) / 2;
44
    if (vf->priv->y < 0)
45
        vf->priv->y = (height - vf->priv->h) / 2;
46
    if (vf->priv->w + vf->priv->x > width
47
        || vf->priv->h + vf->priv->y > height) {
48
        mp_msg(MSGT_VFILTER,MSGL_WARN,"rectangle: bad position/width/height - rectangle area is out of the original!\n");
49
        return 0;
50
    }
51
    return vf_next_config(vf, width, height, d_width, d_height, flags, outfmt);
52
}
53

    
54
static int
55
control(struct vf_instance *vf, int request, void *data)
56
{
57
    const int *const tmp = data;
58
    switch(request){
59
    case VFCTRL_CHANGE_RECTANGLE:
60
        switch (tmp[0]){
61
        case 0:
62
            vf->priv->w += tmp[1];
63
            return 1;
64
            break;
65
        case 1:
66
            vf->priv->h += tmp[1];
67
            return 1;
68
            break;
69
        case 2:
70
            vf->priv->x += tmp[1];
71
            return 1;
72
            break;
73
        case 3:
74
            vf->priv->y += tmp[1];
75
            return 1;
76
            break;
77
        default:
78
            mp_msg(MSGT_VFILTER,MSGL_FATAL,"Unknown param %d \n", tmp[0]);
79
            return 0;
80
        }
81
    }
82
    return vf_next_control(vf, request, data);
83
    return 0;
84
}
85
static int
86
put_image(struct vf_instance *vf, mp_image_t* mpi, double pts){
87
    mp_image_t* dmpi;
88
    unsigned int bpp = mpi->bpp / 8;
89
    int x, y, w, h;
90
    dmpi = vf_get_image(vf->next, mpi->imgfmt, MP_IMGTYPE_TEMP,
91
                        MP_IMGFLAG_ACCEPT_STRIDE | MP_IMGFLAG_PREFER_ALIGNED_STRIDE,
92
                        mpi->w, mpi->h);
93

    
94
    memcpy_pic(dmpi->planes[0],mpi->planes[0],mpi->w*bpp, mpi->h,
95
               dmpi->stride[0],mpi->stride[0]);
96
    if(mpi->flags&MP_IMGFLAG_PLANAR && mpi->flags&MP_IMGFLAG_YUV){
97
        memcpy_pic(dmpi->planes[1],mpi->planes[1],
98
                   mpi->w>>mpi->chroma_x_shift, mpi->h>>mpi->chroma_y_shift,
99
                   dmpi->stride[1],mpi->stride[1]);
100
        memcpy_pic(dmpi->planes[2],mpi->planes[2],
101
                   mpi->w>>mpi->chroma_x_shift, mpi->h>>mpi->chroma_y_shift,
102
                   dmpi->stride[2],mpi->stride[2]);
103
    }
104

    
105
    /* Draw the rectangle */
106

    
107
    mp_msg(MSGT_VFILTER,MSGL_INFO, "rectangle: -vf rectangle=%d:%d:%d:%d \n", vf->priv->w, vf->priv->h, vf->priv->x, vf->priv->y);
108

    
109
    x = FFMIN(vf->priv->x, dmpi->width);
110
    x = FFMAX(x, 0);
111

    
112
    w = vf->priv->x + vf->priv->w - 1 - x;
113
    w = FFMIN(w, dmpi->width - x);
114
    w = FFMAX(w, 0);
115

    
116
    y = FFMIN(vf->priv->y, dmpi->height);
117
    y = FFMAX(y, 0);
118

    
119
    h = vf->priv->y + vf->priv->h - 1 - y;
120
    h = FFMIN(h, dmpi->height - y);
121
    h = FFMAX(h, 0);
122

    
123
    if (0 <= vf->priv->y && vf->priv->y <= dmpi->height) {
124
        unsigned char *p = dmpi->planes[0] + y * dmpi->stride[0] + x * bpp;
125
        unsigned int count = w * bpp;
126
        while (count--)
127
            p[count] = 0xff - p[count];
128
    }
129
    if (h != 1 && vf->priv->y + vf->priv->h - 1 <= mpi->height) {
130
        unsigned char *p = dmpi->planes[0] + (vf->priv->y + vf->priv->h - 1) * dmpi->stride[0] + x * bpp;
131
        unsigned int count = w * bpp;
132
        while (count--)
133
            p[count] = 0xff - p[count];
134
    }
135
    if (0 <= vf->priv->x  && vf->priv->x <= dmpi->width) {
136
        unsigned char *p = dmpi->planes[0] + y * dmpi->stride[0] + x * bpp;
137
        unsigned int count = h;
138
        while (count--) {
139
            unsigned int i = bpp;
140
            while (i--)
141
                p[i] = 0xff - p[i];
142
            p += dmpi->stride[0];
143
        }
144
    }
145
    if (w != 1 && vf->priv->x + vf->priv->w - 1 <= mpi->width) {
146
        unsigned char *p = dmpi->planes[0] + y * dmpi->stride[0] + (vf->priv->x + vf->priv->w - 1) * bpp;
147
        unsigned int count = h;
148
        while (count--) {
149
            unsigned int i = bpp;
150
            while (i--)
151
                p[i] = 0xff - p[i];
152
            p += dmpi->stride[0];
153
        }
154
    }
155
    return vf_next_put_image(vf, dmpi, pts);
156
}
157

    
158
static int
159
vf_open(vf_instance_t *vf, char *args) {
160
    vf->config = config;
161
    vf->control = control;
162
    vf->put_image = put_image;
163
    vf->priv = malloc(sizeof(struct vf_priv_s));
164
    vf->priv->x = -1;
165
    vf->priv->y = -1;
166
    vf->priv->w = -1;
167
    vf->priv->h = -1;
168
    if (args)
169
        sscanf(args, "%d:%d:%d:%d",
170
               &vf->priv->w, &vf->priv->h, &vf->priv->x, &vf->priv->y);
171
    return 1;
172
}
173

    
174
const vf_info_t vf_info_rectangle = {
175
    "draw rectangle",
176
    "rectangle",
177
    "Kim Minh Kaplan",
178
    "",
179
    vf_open,
180
    NULL
181
};