ffmpeg: set the video filters timebase to the demuxers.
[ffmpeg.git] / libavfilter / libmpcodecs / vf_rectangle.c
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 };