388ce699b75d1c92940d8630c2e0bf8b84cecde7
[vlc.git] / modules / video_filter / scale.c
1 /*****************************************************************************
2  * resize.c: video scaling module for YUVP/A pictures
3  *  Uses the low quality "nearest neighbour" algorithm.
4  *****************************************************************************
5  * Copyright (C) 2003 the VideoLAN team
6  * $Id$
7  *
8  * Author: Gildas Bazin <gbazin@videolan.org>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23  *****************************************************************************/
24
25 /*****************************************************************************
26  * Preamble
27  *****************************************************************************/
28 #include <vlc/vlc.h>
29 #include <vlc_vout.h>
30 #include "vlc_filter.h"
31
32 /*****************************************************************************
33  * filter_sys_t : filter descriptor
34  *****************************************************************************/
35 struct filter_sys_t
36 {
37     es_format_t fmt_in;
38     es_format_t fmt_out;
39 };
40
41 /****************************************************************************
42  * Local prototypes
43  ****************************************************************************/
44 static int  OpenFilter ( vlc_object_t * );
45 static void CloseFilter( vlc_object_t * );
46
47 static picture_t *Filter( filter_t *, picture_t * );
48
49 /*****************************************************************************
50  * Module descriptor
51  *****************************************************************************/
52 vlc_module_begin();
53     set_description( _("Video scaling filter") );
54     set_capability( "video filter2", 10000 );
55 //    set_category( CAT_VIDEO );
56 //    set_subcategory( SUBCAT_VIDEO_VFILTER2 );
57     set_callbacks( OpenFilter, CloseFilter );
58 vlc_module_end();
59
60 /*****************************************************************************
61  * OpenFilter: probe the filter and return score
62  *****************************************************************************/
63 static int OpenFilter( vlc_object_t *p_this )
64 {
65     filter_t *p_filter = (filter_t*)p_this;
66     filter_sys_t *p_sys;
67
68     if( ( p_filter->fmt_in.video.i_chroma != VLC_FOURCC('Y','U','V','P') &&
69           p_filter->fmt_in.video.i_chroma != VLC_FOURCC('Y','U','V','A') &&
70           p_filter->fmt_in.video.i_chroma != VLC_FOURCC('R','G','B','A') ) ||
71         p_filter->fmt_in.video.i_chroma != p_filter->fmt_out.video.i_chroma )
72     {
73         return VLC_EGENERIC;
74     }
75
76     /* Allocate the memory needed to store the decoder's structure */
77     if( ( p_filter->p_sys = p_sys =
78           (filter_sys_t *)malloc(sizeof(filter_sys_t)) ) == NULL )
79     {
80         msg_Err( p_filter, "out of memory" );
81         return VLC_EGENERIC;
82     }
83
84     p_filter->pf_video_filter = Filter;
85
86     msg_Dbg( p_filter, "%ix%i -> %ix%i", p_filter->fmt_in.video.i_width,
87              p_filter->fmt_in.video.i_height, p_filter->fmt_out.video.i_width,
88              p_filter->fmt_out.video.i_height );
89
90     return VLC_SUCCESS;
91 }
92
93 /*****************************************************************************
94  * CloseFilter: clean up the filter
95  *****************************************************************************/
96 static void CloseFilter( vlc_object_t *p_this )
97 {
98     filter_t *p_filter = (filter_t*)p_this;
99     filter_sys_t *p_sys = p_filter->p_sys;
100
101     free( p_sys );
102 }
103
104 /****************************************************************************
105  * Filter: the whole thing
106  ****************************************************************************/
107 static picture_t *Filter( filter_t *p_filter, picture_t *p_pic )
108 {
109     picture_t *p_pic_dst;
110     int i_plane, i, j, k, l;
111
112     if( !p_pic ) return NULL;
113
114     /* Request output picture */
115     p_pic_dst = p_filter->pf_vout_buffer_new( p_filter );
116     if( !p_pic_dst )
117     {
118         msg_Warn( p_filter, "can't get output picture" );
119         if( p_pic->pf_release )
120             p_pic->pf_release( p_pic );
121         return NULL;
122     }
123
124     if( p_filter->fmt_in.video.i_chroma == VLC_FOURCC('Y','U','V','P') ||
125         p_filter->fmt_in.video.i_chroma == VLC_FOURCC('Y','U','V','A') )
126     {
127         for( i_plane = 0; i_plane < p_pic_dst->i_planes; i_plane++ )
128         {
129             uint8_t *p_src = p_pic->p[i_plane].p_pixels;
130             uint8_t *p_dst = p_pic_dst->p[i_plane].p_pixels;
131             int i_src_pitch = p_pic->p[i_plane].i_pitch;
132             int i_dst_pitch = p_pic_dst->p[i_plane].i_pitch;
133
134             for( i = 0; i < p_pic_dst->p[i_plane].i_visible_lines; i++ )
135             {
136                 l = ( p_filter->fmt_in.video.i_height * i +
137                       p_filter->fmt_out.video.i_height / 2 ) /
138                     p_filter->fmt_out.video.i_height;
139
140                 l = __MIN( (int)p_filter->fmt_in.video.i_height - 1, l );
141
142                 for( j = 0; j < p_pic_dst->p[i_plane].i_visible_pitch; j++ )
143                 {
144                     k = ( p_filter->fmt_in.video.i_width * j +
145                           p_filter->fmt_out.video.i_width / 2 ) /
146                         p_filter->fmt_out.video.i_width;
147
148                     k = __MIN( (int)p_filter->fmt_in.video.i_width - 1, k );
149
150                     p_dst[i * i_dst_pitch + j] = p_src[l * i_src_pitch + k];
151                 }
152             }
153         }
154     }
155     else /* RGBA */
156     {
157         uint8_t *p_src = p_pic->p->p_pixels;
158         uint8_t *p_dst = p_pic_dst->p->p_pixels;
159         int i_src_pitch = p_pic->p->i_pitch;
160         int i_dst_pitch = p_pic_dst->p->i_pitch;
161         for( i = 0; i < p_pic_dst->p->i_visible_lines; i++ )
162         {
163             l = ( p_filter->fmt_in.video.i_height * i +
164                   p_filter->fmt_out.video.i_height / 2 ) /
165                 p_filter->fmt_out.video.i_height;
166
167             l = __MIN( (int)p_filter->fmt_in.video.i_height - 1, l );
168
169             for( j = 0; j < p_pic_dst->p->i_visible_pitch/4; j++ )
170             {
171                 k = ( p_filter->fmt_in.video.i_width * j +
172                       p_filter->fmt_out.video.i_width / 2 ) /
173                     p_filter->fmt_out.video.i_width;
174
175                 k = __MIN( (int)p_filter->fmt_in.video.i_width - 1, k );
176
177                 *(uint32_t*)(&p_dst[i * i_dst_pitch + 4*j]) =
178                     *(uint32_t*)(&p_src[l * i_src_pitch + 4*k]);
179             }
180         }
181     }
182
183     p_pic_dst->date = p_pic->date;
184     p_pic_dst->b_force = p_pic->b_force;
185     p_pic_dst->i_nb_fields = p_pic->i_nb_fields;
186     p_pic_dst->b_progressive = p_pic->b_progressive;
187     p_pic_dst->b_top_field_first = p_pic->b_top_field_first;
188
189     p_pic->pf_release( p_pic );
190     return p_pic_dst;
191 }