7cc6dcd7941516e1dcd1b1cad57bad87c627c4a5
[vlc.git] / modules / visualization / scope / scope.c
1 /*****************************************************************************
2  * scope.c : Scope effect module
3  *****************************************************************************
4  * Copyright (C) 2002 VideoLAN
5  * $Id: scope.c,v 1.3 2003/08/18 14:57:09 sigmunau 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>                                              /* strdup() */
29 #include <errno.h>
30
31 #include <vlc/vlc.h>
32 #include <vlc/aout.h>
33 #include <vlc/vout.h>
34 #include "aout_internal.h"
35
36 #define SCOPE_WIDTH 320
37 #define SCOPE_HEIGHT 240
38 #define SCOPE_ASPECT (VOUT_ASPECT_FACTOR*SCOPE_WIDTH/SCOPE_HEIGHT)
39
40 /*****************************************************************************
41  * aout_sys_t: scope audio output method descriptor
42  *****************************************************************************
43  * This structure is part of the audio output thread descriptor.
44  * It describes some scope specific variables.
45  *****************************************************************************/
46 typedef struct aout_filter_sys_t
47 {
48     aout_fifo_t *p_aout_fifo;
49     vout_thread_t *p_vout;
50 } aout_filter_sys_t;
51
52 /*****************************************************************************
53  * Local prototypes
54  *****************************************************************************/
55 static int  Open         ( vlc_object_t * );             
56 static void Close        ( vlc_object_t * );                   
57
58 static void DoWork    ( aout_instance_t *, aout_filter_t *, aout_buffer_t *,
59                         aout_buffer_t * );
60
61 /*****************************************************************************
62  * Module descriptor
63  *****************************************************************************/
64 vlc_module_begin();
65     set_description( _("scope effect") ); 
66     set_capability( "audio filter", 0 );
67     set_callbacks( Open, Close );
68     add_shortcut( "scope" );
69 vlc_module_end();
70
71 /*****************************************************************************
72  * Open: open a scope effect plugin
73  *****************************************************************************/
74 static int Open( vlc_object_t *p_this )
75 {
76     aout_filter_t *p_aout = (aout_filter_t *)p_this;
77     aout_filter_t * p_filter = (aout_filter_t *)p_this;
78
79     if ( p_filter->input.i_format != VLC_FOURCC('f','l','3','2') //AOUT_FMT_U16_NE
80          || p_filter->output.i_format != VLC_FOURCC('f','l','3','2') )//AOUT_FMT_U16_NE )
81     {
82         msg_Warn( p_filter, "Bad input or output format" );
83         return -1;
84     }
85
86     if ( !AOUT_FMTS_SIMILAR( &p_filter->input, &p_filter->output ) )
87     {
88         msg_Warn( p_filter, "input and output formats are not similar" );
89         return -1;
90     }
91
92     p_filter->pf_do_work = DoWork;
93     p_filter->b_in_place = 1;
94
95     /* Allocate structure */
96     p_aout->p_sys = malloc( sizeof( aout_filter_sys_t ) );
97     if( p_aout->p_sys == NULL )
98     {
99         msg_Err( p_aout, "out of memory" );
100         return -1;
101     }
102
103     /* Open video output */
104     p_aout->p_sys->p_vout =
105         vout_Create( p_aout, SCOPE_WIDTH, SCOPE_HEIGHT,
106                      VLC_FOURCC('I','4','2','0'), SCOPE_ASPECT );
107
108     if( p_aout->p_sys->p_vout == NULL )
109     {
110         msg_Err( p_aout, "no suitable vout module" );
111         free( p_aout->p_sys );
112         return -1;
113     }
114
115     return( 0 );
116 }
117
118 /*****************************************************************************
119  * Play: play a sound samples buffer
120  *****************************************************************************
121  * This function writes a buffer of i_length bytes in the socket
122  *****************************************************************************/
123 static void DoWork( aout_instance_t * p_aout, aout_filter_t * p_filter,
124                     aout_buffer_t * p_in_buf, aout_buffer_t * p_out_buf )
125 {
126     picture_t *p_outpic;
127     int i_index, i_image;
128     int i_size = p_in_buf->i_size;
129     byte_t *p_buffer = p_in_buf->p_buffer;
130     uint8_t  *ppp_area[2][3];
131     float *p_sample;
132
133     p_out_buf->i_nb_samples = p_in_buf->i_nb_samples;
134     p_out_buf->i_nb_bytes = p_in_buf->i_nb_bytes;
135     for( i_image = 0; (i_image + 1) * SCOPE_WIDTH * 8 < i_size ; i_image++ )
136     {
137         /* Don't stay here forever */
138         if( mdate() >= p_in_buf->start_date - 10000 )
139         {
140             break;
141         }
142         /* This is a new frame. Get a structure from the video_output. */
143         while( ( p_outpic = vout_CreatePicture( p_filter->p_sys->p_vout, 0, 0, 0 ) )
144                   == NULL )
145         {
146             if( p_aout->b_die )
147             {
148                 return;
149             }
150             msleep( 1 );/* Not sleeping here makes us use 100% cpu,
151                          * sleeping too much absolutly kills audio
152                          * quality. 1 seems to be a good value */
153         }
154
155         /* Blank the picture */
156         for( i_index = 0 ; i_index < p_outpic->i_planes ; i_index++ )
157         {
158             memset( p_outpic->p[i_index].p_pixels, i_index ? 0x80 : 0x00,
159                     p_outpic->p[i_index].i_lines * p_outpic->p[i_index].i_pitch );
160         }
161
162         /* We only support 2 channels for now */
163         for( i_index = 0 ; i_index < 2 ; i_index++ )
164         {
165             int j;
166             for( j = 0 ; j < 3 ; j++ )
167             {
168                 ppp_area[i_index][j] =
169                     p_outpic->p[j].p_pixels + i_index * p_outpic->p[j].i_lines
170                                 / p_filter->input.i_original_channels * p_outpic->p[j].i_pitch;
171             }
172         }
173
174         for( i_index = 0, p_sample = (float*)p_buffer;
175              i_index < SCOPE_WIDTH;
176              i_index++ )
177         {
178             int i;
179             int i_tmp_value;
180             uint8_t i_value;
181
182             for( i = 0 ; i < 2 ; i++ )
183             {
184                 /* Left channel */
185                 if ( *p_sample >= 1.0 ) i_tmp_value = 32767;
186                 else if ( *p_sample < -1.0 ) i_tmp_value = -32768;
187                 else i_tmp_value = *p_sample * 32768.0;
188                 p_sample++;
189                 i_value = i_tmp_value / 256 + 128;
190                 *(ppp_area[0][0]
191                    + p_outpic->p[0].i_pitch * i_index / SCOPE_WIDTH
192                    + p_outpic->p[0].i_lines * i_value / 512
193                        * p_outpic->p[0].i_pitch) = 0xbf;
194                 *(ppp_area[0][1]
195                    + p_outpic->p[1].i_pitch * i_index / SCOPE_WIDTH
196                    + p_outpic->p[1].i_lines * i_value / 512
197                       * p_outpic->p[1].i_pitch) = 0xff;
198
199                 /* Right channel */
200                 if ( *p_sample >= 1.0 ) i_tmp_value = 32767;
201                 else if ( *p_sample < -1.0 ) i_tmp_value = -32768;
202                 else i_tmp_value = *p_sample * 32768.0;
203                 p_sample++;
204                 i_value = i_tmp_value / 256 + 128;
205                 *(ppp_area[1][0]
206                    + p_outpic->p[0].i_pitch * i_index / SCOPE_WIDTH
207                    + p_outpic->p[0].i_lines * i_value / 512
208                       * p_outpic->p[0].i_pitch) = 0x9f;
209                 *(ppp_area[1][2]
210                    + p_outpic->p[2].i_pitch * i_index / SCOPE_WIDTH
211                    + p_outpic->p[2].i_lines * i_value / 512
212                       * p_outpic->p[2].i_pitch) = 0xdd;
213             }
214         }
215
216         /* Display the picture - FIXME: find a better date :-) */
217         vout_DatePicture( p_filter->p_sys->p_vout, p_outpic,
218                           p_in_buf->start_date + i_image * 20000 );
219         vout_DisplayPicture( p_filter->p_sys->p_vout, p_outpic );
220
221         p_buffer += SCOPE_WIDTH * 8;
222     }
223 }
224
225 /*****************************************************************************
226  * Close: close the plugin
227  *****************************************************************************/
228 static void Close( vlc_object_t *p_this )
229 {
230     aout_filter_t * p_filter = (aout_filter_t *)p_this;
231
232     /* Kill video output */
233     vout_Destroy( p_filter->p_sys->p_vout );
234
235     free( p_filter->p_sys );
236 }
237