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.12 2003/05/15 22:27:37 massiot 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, VLC_FALSE );
  64     add_string_from_list( "transform-type", "90", type_list, NULL,
  65                           TYPE_TEXT, TYPE_LONGTEXT, VLC_FALSE);
  66     set_description( _("video transformation filter") );
  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_SUCCESS;
 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     vlc_object_detach( p_vout->p_sys->p_vout );
 234     vout_Destroy( p_vout->p_sys->p_vout );
 235
 236     free( p_vout->p_sys );
 237 }
 238
 239 /*****************************************************************************
 240  * Render: displays previously rendered output
 241  *****************************************************************************
 242  * This function send the currently rendered image to Transform image, waits
 243  * until it is displayed and switch the two rendering buffers, preparing next
 244  * frame.
 245  *****************************************************************************/
 246 static void Render( vout_thread_t *p_vout, picture_t *p_pic )
 247 {
 248     picture_t *p_outpic;
 249     int i_index;
 250
 251     /* This is a new frame. Get a structure from the video_output. */
 252     while( ( p_outpic = vout_CreatePicture( p_vout->p_sys->p_vout, 0, 0, 0 ) )
 253               == NULL )
 254     {
 255         if( p_vout->b_die || p_vout->b_error )
 256         {
 257             return;
 258         }
 259         msleep( VOUT_OUTMEM_SLEEP );
 260     }
 261
 262     vout_DatePicture( p_vout->p_sys->p_vout, p_outpic, p_pic->date );
 263     vout_LinkPicture( p_vout->p_sys->p_vout, p_outpic );
 264
 265     switch( p_vout->p_sys->i_mode )
 266     {
 267         case TRANSFORM_MODE_90:
 268             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 269             {
 270                 int i_pitch = p_pic->p[i_index].i_pitch;
 271
 272                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 273
 274                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 275                 uint8_t *p_out_end = p_out + p_outpic->p[i_index].i_lines
 276                                               * p_outpic->p[i_index].i_pitch;
 277
 278                 for( ; p_out < p_out_end ; )
 279                 {
 280                     uint8_t *p_line_end;
 281
 282                     p_out_end -= p_outpic->p[i_index].i_pitch
 283                                   - p_outpic->p[i_index].i_visible_pitch;
 284                     p_line_end = p_in + p_pic->p[i_index].i_lines * i_pitch;
 285
 286                     for( ; p_in < p_line_end ; )
 287                     {
 288                         p_line_end -= i_pitch;
 289                         *(--p_out_end) = *p_line_end;
 290                     }
 291
 292                     p_in++;
 293                 }
 294             }
 295             break;
 296
 297         case TRANSFORM_MODE_180:
 298             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 299             {
 300                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 301                 uint8_t *p_in_end = p_in + p_pic->p[i_index].i_lines
 302                                             * p_pic->p[i_index].i_pitch;
 303
 304                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 305
 306                 for( ; p_in < p_in_end ; )
 307                 {
 308                     uint8_t *p_line_start = p_in_end
 309                                              - p_pic->p[i_index].i_pitch;
 310                     p_in_end -= p_pic->p[i_index].i_pitch
 311                                  - p_pic->p[i_index].i_visible_pitch;
 312
 313                     for( ; p_line_start < p_in_end ; )
 314                     {
 315                         *p_out++ = *(--p_in_end);
 316                     }
 317
 318                     p_out += p_outpic->p[i_index].i_pitch
 319                               - p_outpic->p[i_index].i_visible_pitch;
 320                 }
 321             }
 322             break;
 323
 324         case TRANSFORM_MODE_270:
 325             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 326             {
 327                 int i_pitch = p_pic->p[i_index].i_pitch;
 328
 329                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 330
 331                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 332                 uint8_t *p_out_end = p_out + p_outpic->p[i_index].i_lines
 333                                               * p_outpic->p[i_index].i_pitch;
 334
 335                 for( ; p_out < p_out_end ; )
 336                 {
 337                     uint8_t *p_in_end;
 338
 339                     p_in_end = p_in + p_pic->p[i_index].i_lines * i_pitch;
 340
 341                     for( ; p_in < p_in_end ; )
 342                     {
 343                         p_in_end -= i_pitch;
 344                         *p_out++ = *p_in_end;
 345                     }
 346
 347                     p_out += p_outpic->p[i_index].i_pitch
 348                               - p_outpic->p[i_index].i_visible_pitch;
 349                     p_in++;
 350                 }
 351             }
 352             break;
 353
 354         case TRANSFORM_MODE_VFLIP:
 355             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 356             {
 357                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 358                 uint8_t *p_in_end = p_in + p_pic->p[i_index].i_lines
 359                                             * p_pic->p[i_index].i_pitch;
 360
 361                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 362
 363                 for( ; p_in < p_in_end ; )
 364                 {
 365                     p_in_end -= p_pic->p[i_index].i_pitch;
 366                     p_vout->p_vlc->pf_memcpy( p_out, p_in_end,
 367                                            p_pic->p[i_index].i_visible_pitch );
 368                     p_out += p_pic->p[i_index].i_pitch;
 369                 }
 370             }
 371             break;
 372
 373         case TRANSFORM_MODE_HFLIP:
 374             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 375             {
 376                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
 377                 uint8_t *p_in_end = p_in + p_pic->p[i_index].i_lines
 378                                             * p_pic->p[i_index].i_pitch;
 379
 380                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
 381
 382                 for( ; p_in < p_in_end ; )
 383                 {
 384                     uint8_t *p_line_end = p_in
 385                                            + p_pic->p[i_index].i_visible_pitch;
 386
 387                     for( ; p_in < p_line_end ; )
 388                     {
 389                         *p_out++ = *(--p_line_end);
 390                     }
 391
 392                     p_in += p_pic->p[i_index].i_pitch;
 393                 }
 394             }
 395             break;
 396
 397         default:
 398             break;
 399     }
 400
 401     vout_UnlinkPicture( p_vout->p_sys->p_vout, p_outpic );
 402
 403     vout_DisplayPicture( p_vout->p_sys->p_vout, p_outpic );
 404 }
 405
 406 /*****************************************************************************
 407  * SendEvents: forward mouse and keyboard events to the parent p_vout
 408  *****************************************************************************/
 409 static int SendEvents( vlc_object_t *p_this, char const *psz_var,
 410                        vlc_value_t oldval, vlc_value_t newval, void *_p_vout )
 411 {
 412     vout_thread_t *p_vout = (vout_thread_t *)_p_vout;
 413     vlc_value_t sentval = newval;
 414
 415     /* Translate the mouse coordinates */
 416     if( !strcmp( psz_var, "mouse-x" ) )
 417     {
 418         switch( p_vout->p_sys->i_mode )
 419         {
 420         case TRANSFORM_MODE_270:
 421             sentval.i_int = p_vout->p_sys->p_vout->output.i_width
 422                              - sentval.i_int;
 423         case TRANSFORM_MODE_90:
 424             var_Set( p_vout, "mouse-y", sentval );
 425             return VLC_SUCCESS;
 426
 427         case TRANSFORM_MODE_180:
 428         case TRANSFORM_MODE_HFLIP:
 429             sentval.i_int = p_vout->p_sys->p_vout->output.i_width
 430                              - sentval.i_int;
 431             break;
 432
 433         case TRANSFORM_MODE_VFLIP:
 434         default:
 435             break;
 436         }
 437     }
 438     else if( !strcmp( psz_var, "mouse-y" ) )
 439     {
 440         switch( p_vout->p_sys->i_mode )
 441         {
 442         case TRANSFORM_MODE_90:
 443             sentval.i_int = p_vout->p_sys->p_vout->output.i_height
 444                              - sentval.i_int;
 445         case TRANSFORM_MODE_270:
 446             var_Set( p_vout, "mouse-x", sentval );
 447             return VLC_SUCCESS;
 448
 449         case TRANSFORM_MODE_180:
 450         case TRANSFORM_MODE_VFLIP:
 451             sentval.i_int = p_vout->p_sys->p_vout->output.i_height
 452                              - sentval.i_int;
 453             break;
 454
 455         case TRANSFORM_MODE_HFLIP:
 456         default:
 457             break;
 458         }
 459     }
 460
 461     var_Set( p_vout, psz_var, sentval );
 462
 463     return VLC_SUCCESS;
 464 }
 465