git.sesse.net Git - vlc/blob - modules/video_filter/transform.c

   1 /*****************************************************************************
   2  * transform.c : transform image plugin for vlc
   3  *****************************************************************************
   4  * Copyright (C) 2000, 2001, 2002, 2003 VideoLAN
   5  * $Id: transform.c,v 1.7 2003/01/17 16:18:03 sam Exp $
   6  *
   7  * Authors: Samuel Hocevar <sam@zoy.org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #include <stdlib.h>                                      /* malloc(), free() */
  28 #include <string.h>
  29
  30 #include <vlc/vlc.h>
  31 #include <vlc/vout.h>
  32
  33 #include "filter_common.h"
  34
  35 #define TRANSFORM_MODE_HFLIP   1
  36 #define TRANSFORM_MODE_VFLIP   2
  37 #define TRANSFORM_MODE_90      3
  38 #define TRANSFORM_MODE_180     4
  39 #define TRANSFORM_MODE_270     5
  40
  41 /*****************************************************************************
  42  * Local prototypes
  43  *****************************************************************************/
  44 static int  Create    ( vlc_object_t * );
  45 static void Destroy   ( vlc_object_t * );
  46
  47 static int  Init      ( vout_thread_t * );
  48 static void End       ( vout_thread_t * );
  49 static void Render    ( vout_thread_t *, picture_t * );
  50
  51 static int  SendEvents( vlc_object_t *, char const *,
  52                         vlc_value_t, vlc_value_t, void * );
  53
  54 /*****************************************************************************
  55  * Module descriptor
  56  *****************************************************************************/
  57 #define TYPE_TEXT N_("transform type")
  58 #define TYPE_LONGTEXT N_("One of '90', '180', '270', 'hflip' and 'vflip'")
  59
  60 static char *type_list[] = { "90", "180", "270", "hflip", "vflip", NULL };
  61
  62 vlc_module_begin();
  63     add_category_hint( N_("Miscellaneous"), NULL );
  64     add_string_from_list( "transform-type", "90", type_list, NULL,
  65                           TYPE_TEXT, TYPE_LONGTEXT);
  66     set_description( _("image transformation module") );
  67     set_capability( "video filter", 0 );
  68     add_shortcut( "transform" );
  69     set_callbacks( Create, Destroy );
  70 vlc_module_end();
  71
  72 /*****************************************************************************
  73  * vout_sys_t: Transform video output method descriptor
  74  *****************************************************************************
  75  * This structure is part of the video output thread descriptor.
  76  * It describes the Transform specific properties of an output thread.
  77  *****************************************************************************/
  78 struct vout_sys_t
  79 {
  80     int i_mode;
  81     vlc_bool_t b_rotation;
  82     vout_thread_t *p_vout;
  83 };
  84
  85 /*****************************************************************************
  86  * Create: allocates Transform video thread output method
  87  *****************************************************************************
  88  * This function allocates and initializes a Transform vout method.
  89  *****************************************************************************/
  90 static int Create( vlc_object_t *p_this )
  91 {
  92     vout_thread_t *p_vout = (vout_thread_t *)p_this;
  93     char *psz_method;
  94
  95     /* Allocate structure */
  96     p_vout->p_sys = malloc( sizeof( vout_sys_t ) );
  97     if( p_vout->p_sys == NULL )
  98     {
  99         msg_Err( p_vout, "out of memory" );
 100         return VLC_ENOMEM;
 101     }
 102
 103     p_vout->pf_init = Init;
 104     p_vout->pf_end = End;
 105     p_vout->pf_manage = NULL;
 106     p_vout->pf_render = Render;
 107     p_vout->pf_display = NULL;
 108
 109     /* Look what method was requested */
 110     psz_method = config_GetPsz( p_vout, "transform-type" );
 111
 112     if( psz_method == NULL )
 113     {
 114         msg_Err( p_vout, "configuration variable %s empty", "transform-type" );
 115         msg_Err( p_vout, "no valid transform mode provided, using '90'" );
 116         p_vout->p_sys->i_mode = TRANSFORM_MODE_90;
 117         p_vout->p_sys->b_rotation = 1;
 118     }
 119     else
 120     {
 121         if( !strcmp( psz_method, "hflip" ) )
 122         {
 123             p_vout->p_sys->i_mode = TRANSFORM_MODE_HFLIP;
 124             p_vout->p_sys->b_rotation = 0;
 125         }
 126         else if( !strcmp( psz_method, "vflip" ) )
 127         {
 128             p_vout->p_sys->i_mode = TRANSFORM_MODE_VFLIP;
 129             p_vout->p_sys->b_rotation = 0;
 130         }
 131         else if( !strcmp( psz_method, "90" ) )
 132         {
 133             p_vout->p_sys->i_mode = TRANSFORM_MODE_90;
 134             p_vout->p_sys->b_rotation = 1;
 135         }
 136         else if( !strcmp( psz_method, "180" ) )
 137         {
 138             p_vout->p_sys->i_mode = TRANSFORM_MODE_180;
 139             p_vout->p_sys->b_rotation = 0;
 140         }
 141         else if( !strcmp( psz_method, "270" ) )
 142         {
 143             p_vout->p_sys->i_mode = TRANSFORM_MODE_270;
 144             p_vout->p_sys->b_rotation = 1;
 145         }
 146         else
 147         {
 148             msg_Err( p_vout, "no valid transform mode provided, using '90'" );
 149             p_vout->p_sys->i_mode = TRANSFORM_MODE_90;
 150             p_vout->p_sys->b_rotation = 1;
 151         }
 152
 153         free( psz_method );
 154     }
 155
 156     return VLC_EGENERIC;
 157 }
 158
 159 /*****************************************************************************
 160  * Init: initialize Transform video thread output method
 161  *****************************************************************************/
 162 static int Init( vout_thread_t *p_vout )
 163 {
 164     int i_index;
 165     picture_t *p_pic;
 166
 167     I_OUTPUTPICTURES = 0;
 168
 169     /* Initialize the output structure */
 170     p_vout->output.i_chroma = p_vout->render.i_chroma;
 171     p_vout->output.i_width  = p_vout->render.i_width;
 172     p_vout->output.i_height = p_vout->render.i_height;
 173     p_vout->output.i_aspect = p_vout->render.i_aspect;
 174
 175     /* Try to open the real video output */
 176     msg_Dbg( p_vout, "spawning the real video output" );
 177
 178     if( p_vout->p_sys->b_rotation )
 179     {
 180         p_vout->p_sys->p_vout = vout_Create( p_vout,
 181                            p_vout->render.i_height, p_vout->render.i_width,
 182                            p_vout->render.i_chroma,
 183                            (uint64_t)VOUT_ASPECT_FACTOR
 184                             * (uint64_t)VOUT_ASPECT_FACTOR
 185                             / (uint64_t)p_vout->render.i_aspect );
 186     }
 187     else
 188     {
 189         p_vout->p_sys->p_vout = vout_Create( p_vout,
 190                            p_vout->render.i_width, p_vout->render.i_height,
 191                            p_vout->render.i_chroma, p_vout->render.i_aspect );
 192     }
 193
 194     /* Everything failed */
 195     if( p_vout->p_sys->p_vout == NULL )
 196     {
 197         msg_Err( p_vout, "cannot open vout, aborting" );
 198         return VLC_EGENERIC;
 199     }
 200
 201     ALLOCATE_DIRECTBUFFERS( VOUT_MAX_PICTURES );
 202
 203     ADD_CALLBACKS( p_vout->p_sys->p_vout, SendEvents );
 204
 205     return VLC_SUCCESS;
 206 }
 207
 208 /*****************************************************************************
 209  * End: terminate Transform video thread output method
 210  *****************************************************************************/
 211 static void End( vout_thread_t *p_vout )
 212 {
 213     int i_index;
 214
 215     /* Free the fake output buffers we allocated */
 216     for( i_index = I_OUTPUTPICTURES ; i_index ; )
 217     {
 218         i_index--;
 219         free( PP_OUTPUTPICTURE[ i_index ]->p_data_orig );
 220     }
 221 }
 222
 223 /*****************************************************************************
 224  * Destroy: destroy Transform video thread output method
 225  *****************************************************************************
 226  * Terminate an output method created by TransformCreateOutputMethod
 227  *****************************************************************************/
 228 static void Destroy( vlc_object_t *p_this )
 229 {
 230     vout_thread_t *p_vout = (vout_thread_t *)p_this;
 231
 232     DEL_CALLBACKS( p_vout->p_sys->p_vout, SendEvents );
 233     vout_Destroy( p_vout->p_sys->p_vout );
 234
 235     free( p_vout->p_sys );
 236 }
 237
 238 /*****************************************************************************
 239  * Render: displays previously rendered output
 240  *****************************************************************************
 241  * This function send the currently rendered image to Transform image, waits
 242  * until it is displayed and switch the two rendering buffers, preparing next
 243  * frame.
 244  *****************************************************************************/
 245 static void Render( vout_thread_t *p_vout, picture_t *p_pic )
 246 {
 247     picture_t *p_outpic;
 248     int i_index;
 249
 250     /* This is a new frame. Get a structure from the video_output. */
 251     while( ( p_outpic = vout_CreatePicture( p_vout->p_sys->p_vout, 0, 0, 0 ) )
 252               == NULL )
 253     {
 254         if( p_vout->b_die || p_vout->b_error )
 255         {
 256             return;
 257         }
 258         msleep( VOUT_OUTMEM_SLEEP );
 259     }
 260
 261     vout_DatePicture( p_vout->p_sys->p_vout, p_outpic, p_pic->date );
 262     vout_LinkPicture( p_vout->p_sys->p_vout, p_outpic );
 263
 264     switch( p_vout->p_sys->i_mode )
 265     {
 266         case TRANSFORM_MODE_90:
 267             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 268             {
 269                 int i_pitch = p_pic->p[i_index].i_pitch;
 270
 271                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 272
 273                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 274                 uint8_t *p_out_end = p_out + p_outpic->p[i_index].i_lines
 275                                               * p_outpic->p[i_index].i_pitch;
 276
 277                 for( ; p_out < p_out_end ; )
 278                 {
 279                     uint8_t *p_line_end;
 280
 281                     p_out_end -= p_outpic->p[i_index].i_pitch
 282                                   - p_outpic->p[i_index].i_visible_pitch;
 283                     p_line_end = p_in + p_pic->p[i_index].i_lines * i_pitch;
 284
 285                     for( ; p_in < p_line_end ; )
 286                     {
 287                         p_line_end -= i_pitch;
 288                         *(--p_out_end) = *p_line_end;
 289                     }
 290
 291                     p_in++;
 292                 }
 293             }
 294             break;
 295
 296         case TRANSFORM_MODE_180:
 297             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 298             {
 299                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 300                 uint8_t *p_in_end = p_in + p_pic->p[i_index].i_lines
 301                                             * p_pic->p[i_index].i_pitch;
 302
 303                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 304
 305                 for( ; p_in < p_in_end ; )
 306                 {
 307                     uint8_t *p_line_start = p_in_end
 308                                              - p_pic->p[i_index].i_pitch;
 309                     p_in_end -= p_pic->p[i_index].i_pitch
 310                                  - p_pic->p[i_index].i_visible_pitch;
 311
 312                     for( ; p_line_start < p_in_end ; )
 313                     {
 314                         *p_out++ = *(--p_in_end);
 315                     }
 316
 317                     p_out += p_outpic->p[i_index].i_pitch
 318                               - p_outpic->p[i_index].i_visible_pitch;
 319                 }
 320             }
 321             break;
 322
 323         case TRANSFORM_MODE_270:
 324             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 325             {
 326                 int i_pitch = p_pic->p[i_index].i_pitch;
 327
 328                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 329
 330                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 331                 uint8_t *p_out_end = p_out + p_outpic->p[i_index].i_lines
 332                                               * p_outpic->p[i_index].i_pitch;
 333
 334                 for( ; p_out < p_out_end ; )
 335                 {
 336                     uint8_t *p_in_end;
 337
 338                     p_in_end = p_in + p_pic->p[i_index].i_lines * i_pitch;
 339
 340                     for( ; p_in < p_in_end ; )
 341                     {
 342                         p_in_end -= i_pitch;
 343                         *p_out++ = *p_in_end;
 344                     }
 345
 346                     p_out += p_outpic->p[i_index].i_pitch
 347                               - p_outpic->p[i_index].i_visible_pitch;
 348                     p_in++;
 349                 }
 350             }
 351             break;
 352
 353         case TRANSFORM_MODE_VFLIP:
 354             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 355             {
 356                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 357                 uint8_t *p_in_end = p_in + p_pic->p[i_index].i_lines
 358                                             * p_pic->p[i_index].i_pitch;
 359
 360                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 361
 362                 for( ; p_in < p_in_end ; )
 363                 {
 364                     p_in_end -= p_pic->p[i_index].i_pitch;
 365                     p_vout->p_vlc->pf_memcpy( p_out, p_in_end,
 366                                            p_pic->p[i_index].i_visible_pitch );
 367                     p_out += p_pic->p[i_index].i_pitch;
 368                 }
 369             }
 370             break;
 371
 372         case TRANSFORM_MODE_HFLIP:
 373             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 374             {
 375                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 376                 uint8_t *p_in_end = p_in + p_pic->p[i_index].i_lines
 377                                             * p_pic->p[i_index].i_pitch;
 378
 379                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 380
 381                 for( ; p_in < p_in_end ; )
 382                 {
 383                     uint8_t *p_line_end = p_in
 384                                            + p_pic->p[i_index].i_visible_pitch;
 385
 386                     for( ; p_in < p_line_end ; )
 387                     {
 388                         *p_out++ = *(--p_line_end);
 389                     }
 390
 391                     p_in += p_pic->p[i_index].i_pitch;
 392                 }
 393             }
 394             break;
 395
 396         default:
 397             break;
 398     }
 399
 400     vout_UnlinkPicture( p_vout->p_sys->p_vout, p_outpic );
 401
 402     vout_DisplayPicture( p_vout->p_sys->p_vout, p_outpic );
 403 }
 404
 405 /*****************************************************************************
 406  * SendEvents: forward mouse and keyboard events to the parent p_vout
 407  *****************************************************************************/
 408 static int SendEvents( vlc_object_t *p_this, char const *psz_var,
 409                        vlc_value_t oldval, vlc_value_t newval, void *_p_vout )
 410 {
 411     vout_thread_t *p_vout = (vout_thread_t *)_p_vout;
 412     vlc_value_t sentval = newval;
 413
 414     /* Translate the mouse coordinates */
 415     if( !strcmp( psz_var, "mouse-x" ) )
 416     {
 417         switch( p_vout->p_sys->i_mode )
 418         {
 419         case TRANSFORM_MODE_270:
 420             sentval.i_int = p_vout->p_sys->p_vout->output.i_width
 421                              - sentval.i_int;
 422         case TRANSFORM_MODE_90:
 423             var_Set( p_vout, "mouse-y", sentval );
 424             return VLC_SUCCESS;
 425
 426         case TRANSFORM_MODE_180:
 427         case TRANSFORM_MODE_HFLIP:
 428             sentval.i_int = p_vout->p_sys->p_vout->output.i_width
 429                              - sentval.i_int;
 430             break;
 431
 432         case TRANSFORM_MODE_VFLIP:
 433         default:
 434             break;
 435         }
 436     }
 437     else if( !strcmp( psz_var, "mouse-y" ) )
 438     {
 439         switch( p_vout->p_sys->i_mode )
 440         {
 441         case TRANSFORM_MODE_90:
 442             sentval.i_int = p_vout->p_sys->p_vout->output.i_height
 443                              - sentval.i_int;
 444         case TRANSFORM_MODE_270:
 445             var_Set( p_vout, "mouse-x", sentval );
 446             return VLC_SUCCESS;
 447
 448         case TRANSFORM_MODE_180:
 449         case TRANSFORM_MODE_VFLIP:
 450             sentval.i_int = p_vout->p_sys->p_vout->output.i_height
 451                              - sentval.i_int;
 452             break;
 453
 454         case TRANSFORM_MODE_HFLIP:
 455         default:
 456             break;
 457         }
 458     }
 459
 460     var_Set( p_vout, psz_var, sentval );
 461
 462     return VLC_SUCCESS;
 463 }
 464