1 /*****************************************************************************
2  * psychedelic.c : Psychedelic video effect plugin for vlc
3  *****************************************************************************
4  * Copyright (C) 2000-2006 VLC authors and VideoLAN
5  * $Id: 06b8b2d2636cdbf9a4367e961599125ca5e741cb $
6  *
7  * Authors: Samuel Hocevar <sam@zoy.org>
8  *          Antoine Cellerier <dionoea -at- videolan -dot- org>
9  *
10  * This program is free software; you can redistribute it and/or modify it
11  * under the terms of the GNU Lesser General Public License as published by
12  * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public License
21  * along with this program; if not, write to the Free Software Foundation,
22  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23  *****************************************************************************/
24 
25 /*****************************************************************************
26  * Preamble
27  *****************************************************************************/
28 
29 #ifdef HAVE_CONFIG_H
30 # include "config.h"
31 #endif
32 
33 #include <math.h>                                            /* sin(), cos() */
34 
35 #include <vlc_common.h>
36 #include <vlc_plugin.h>
37 #include <vlc_filter.h>
38 #include <vlc_picture.h>
39 #include <vlc_image.h>
40 #include "filter_picture.h"
41 
42 /*****************************************************************************
43  * Local prototypes
44  *****************************************************************************/
45 static int  Create    ( vlc_object_t * );
46 static void Destroy   ( vlc_object_t * );
47 
48 static picture_t *Filter( filter_t *, picture_t * );
49 
50 /*****************************************************************************
51  * Module descriptor
52  *****************************************************************************/
53 vlc_module_begin ()
54     set_description( N_("Psychedelic video filter") )
55     set_shortname( N_( "Psychedelic" ))
56     set_capability( "video filter", 0 )
57     set_category( CAT_VIDEO )
58     set_subcategory( SUBCAT_VIDEO_VFILTER )
59 
60     add_shortcut( "psychedelic" )
61     set_callbacks( Create, Destroy )
62 vlc_module_end ()
63 
64 /*****************************************************************************
65  * filter_sys_t: Distort video output method descriptor
66  *****************************************************************************
67  * This structure is part of the video output thread descriptor.
68  * It describes the Distort specific properties of an output thread.
69  *****************************************************************************/
70 struct filter_sys_t
71 {
72     image_handler_t *p_image;
73     unsigned int x, y, scale;
74     int xinc, yinc, scaleinc;
75     uint8_t u,v;
76 };
77 
78 /*****************************************************************************
79  * Create: allocates Distort video thread output method
80  *****************************************************************************
81  * This function allocates and initializes a Distort vout method.
82  *****************************************************************************/
Create(vlc_object_t * p_this)83 static int Create( vlc_object_t *p_this )
84 {
85     filter_t *p_filter = (filter_t *)p_this;
86 
87     const vlc_fourcc_t fourcc = p_filter->fmt_in.video.i_chroma;
88     const vlc_chroma_description_t *p_chroma = vlc_fourcc_GetChromaDescription( fourcc );
89     if( !p_chroma || p_chroma->plane_count != 3 || p_chroma->pixel_size != 1 ) {
90         msg_Err( p_filter, "Unsupported chroma (%4.4s)", (char*)&fourcc );
91         return VLC_EGENERIC;
92     }
93 
94     /* Allocate structure */
95     p_filter->p_sys = malloc( sizeof( filter_sys_t ) );
96     if( p_filter->p_sys == NULL )
97         return VLC_ENOMEM;
98 
99     p_filter->pf_video_filter = Filter;
100 
101     p_filter->p_sys->x = 10;
102     p_filter->p_sys->y = 10;
103     p_filter->p_sys->scale = 1;
104     p_filter->p_sys->xinc = 1;
105     p_filter->p_sys->yinc = 1;
106     p_filter->p_sys->scaleinc = 1;
107     p_filter->p_sys->u = 0;
108     p_filter->p_sys->v = 0;
109     p_filter->p_sys->p_image = NULL;
110 
111     return VLC_SUCCESS;
112 }
113 
114 /*****************************************************************************
115  * Destroy: destroy Distort video thread output method
116  *****************************************************************************
117  * Terminate an output method created by DistortCreateOutputMethod
118  *****************************************************************************/
Destroy(vlc_object_t * p_this)119 static void Destroy( vlc_object_t *p_this )
120 {
121     filter_t *p_filter = (filter_t *)p_this;
122 
123     if( p_filter->p_sys->p_image )
124         image_HandlerDelete( p_filter->p_sys->p_image );
125     p_filter->p_sys->p_image = NULL;
126 
127     free( p_filter->p_sys );
128 }
129 
130 /*****************************************************************************
131  * Render: displays previously rendered output
132  *****************************************************************************
133  * This function send the currently rendered image to Distort image, waits
134  * until it is displayed and switch the two rendering buffers, preparing next
135  * frame.
136  *****************************************************************************/
Filter(filter_t * p_filter,picture_t * p_pic)137 static picture_t *Filter( filter_t *p_filter, picture_t *p_pic )
138 {
139     picture_t *p_outpic;
140 
141     unsigned int w, h;
142     uint8_t u,v;
143 
144     picture_t *p_converted;
145     video_format_t fmt_out;
146 
147     if( !p_pic ) return NULL;
148 
149     p_outpic = filter_NewPicture( p_filter );
150     if( !p_outpic )
151     {
152         picture_Release( p_pic );
153         return NULL;
154     }
155 
156     if( !p_filter->p_sys->p_image )
157         p_filter->p_sys->p_image = image_HandlerCreate( p_filter );
158 
159     /* chrominance */
160     u = p_filter->p_sys->u;
161     v = p_filter->p_sys->v;
162     for( int y = 0; y < p_outpic->p[U_PLANE].i_lines; y++ )
163     {
164         memset(
165                 p_outpic->p[U_PLANE].p_pixels+y*p_outpic->p[U_PLANE].i_pitch,
166                 u, p_outpic->p[U_PLANE].i_pitch );
167         memset(
168                 p_outpic->p[V_PLANE].p_pixels+y*p_outpic->p[V_PLANE].i_pitch,
169                 v, p_outpic->p[V_PLANE].i_pitch );
170         if( v == 0 && u != 0 )
171             u --;
172         else if( u == 0xff )
173             v --;
174         else if( v == 0xff )
175             u ++;
176         else if( u == 0 )
177             v ++;
178     }
179 
180     /* luminance */
181     plane_CopyPixels( &p_outpic->p[Y_PLANE], &p_pic->p[Y_PLANE] );
182 
183     /* image visualization */
184     fmt_out = p_filter->fmt_out.video;
185     fmt_out.i_width = p_filter->fmt_out.video.i_width*p_filter->p_sys->scale/150;
186     fmt_out.i_height = p_filter->fmt_out.video.i_height*p_filter->p_sys->scale/150;
187     fmt_out.i_visible_width = fmt_out.i_width;
188     fmt_out.i_visible_height = fmt_out.i_height;
189     p_converted = image_Convert( p_filter->p_sys->p_image, p_pic,
190                                  &(p_pic->format), &fmt_out );
191 
192     if( p_converted )
193     {
194 #define copyimage( plane, b ) \
195         for( int y = 0; y<p_converted->p[plane].i_visible_lines; y++ ) { \
196         for( int x = 0; x<p_converted->p[plane].i_visible_pitch; x++ ) { \
197             int nx, ny; \
198             if( p_filter->p_sys->yinc == 1 ) \
199                 ny= y; \
200             else \
201                 ny = p_converted->p[plane].i_visible_lines-y; \
202             if( p_filter->p_sys->xinc == 1 ) \
203                 nx = x; \
204             else \
205                 nx = p_converted->p[plane].i_visible_pitch-x; \
206             p_outpic->p[plane].p_pixels[(p_filter->p_sys->x*b+nx)+(ny+p_filter->p_sys->y*b)*p_outpic->p[plane].i_pitch ] = p_converted->p[plane].p_pixels[y*p_converted->p[plane].i_pitch+x]; \
207         } }
208         copyimage( Y_PLANE, 2 );
209         copyimage( U_PLANE, 1 );
210         copyimage( V_PLANE, 1 );
211 #undef copyimage
212 
213         picture_Release( p_converted );
214     }
215     else
216     {
217         msg_Err( p_filter, "Image scaling failed miserably." );
218     }
219 
220     p_filter->p_sys->x += p_filter->p_sys->xinc;
221     p_filter->p_sys->y += p_filter->p_sys->yinc;
222 
223     p_filter->p_sys->scale += p_filter->p_sys->scaleinc;
224     if( p_filter->p_sys->scale >= 50 ) p_filter->p_sys->scaleinc = -1;
225     if( p_filter->p_sys->scale <= 1 ) p_filter->p_sys->scaleinc = 1;
226 
227     w = p_filter->fmt_out.video.i_width*p_filter->p_sys->scale/150;
228     h = p_filter->fmt_out.video.i_height*p_filter->p_sys->scale/150;
229     if( p_filter->p_sys->x*2 + w >= p_filter->fmt_out.video.i_width )
230         p_filter->p_sys->xinc = -1;
231     if( p_filter->p_sys->x <= 0 )
232         p_filter->p_sys->xinc = 1;
233 
234     if( p_filter->p_sys->x*2 + w >= p_filter->fmt_out.video.i_width )
235         p_filter->p_sys->x = (p_filter->fmt_out.video.i_width-w)/2;
236     if( p_filter->p_sys->y*2 + h >= p_filter->fmt_out.video.i_height )
237         p_filter->p_sys->y = (p_filter->fmt_out.video.i_height-h)/2;
238 
239     if( p_filter->p_sys->y*2 + h >= p_filter->fmt_out.video.i_height )
240         p_filter->p_sys->yinc = -1;
241     if( p_filter->p_sys->y <= 0 )
242         p_filter->p_sys->yinc = 1;
243 
244     for( int y = 0; y < 16; y++ )
245     {
246         if( p_filter->p_sys->v == 0 && p_filter->p_sys->u != 0 )
247             p_filter->p_sys->u -= 1;
248         else if( p_filter->p_sys->u == 0xff )
249             p_filter->p_sys->v -= 1;
250         else if( p_filter->p_sys->v == 0xff )
251             p_filter->p_sys->u += 1;
252         else if( p_filter->p_sys->u == 0 )
253             p_filter->p_sys->v += 1;
254     }
255 
256     return CopyInfoAndRelease( p_outpic, p_pic );
257 }
258