version update, msg txt
[goodguy/history.git] / cinelerra-5.1 / libzmpeg3 / yy.C
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <alsa/asoundlib.h>
4 #include <sys/time.h>
5 #include <gtk/gtk.h>
6 #include <gdk/gdk.h>
7
8 #ifdef __cplusplus
9 #include "libzmpeg3.h"
10 #else
11 #include "../libmpeg3/libmpeg3.h"
12 #define zmpeg3_t mpeg3_t
13 #endif
14
15 #define AUDIO
16 #define VIDEO
17
18 /* c++ `pkg-config --cflags --libs gtk+-2.0` y.C ./x86_64/libzmpeg3.a -lpthread -lasound -lm */
19
20 #ifdef AUDIO
21 snd_pcm_t *zpcm;
22 snd_pcm_uframes_t zpcm_ufrm_size;
23 snd_pcm_sframes_t zpcm_sbfr_size;
24 snd_pcm_sframes_t zpcm_sper_size;
25 unsigned int zpcm_rate;
26 static const char *zpcm_device = "plughw:0,0";
27 static unsigned int zpcm_bfr_time_us = 500000; 
28 static unsigned int zpcm_per_time_us = 200000;
29 static snd_pcm_channel_area_t *zpcm_areas = 0;
30 static int zpcm_channels = 0;
31 static short **zpcm_buffers = 0;
32 static short *zpcm_samples = 0;
33
34 void alsa_close()
35 {
36 #ifdef __cplusplus
37   if( zpcm_buffers != 0 ) { delete [] zpcm_buffers; zpcm_buffers = 0; }
38   if( zpcm_areas != 0 ) { delete [] zpcm_areas; zpcm_areas = 0; }
39   if( zpcm_samples != 0 ) { delete [] zpcm_samples; zpcm_samples = 0; }
40 #else
41   if( zpcm_buffers != 0 ) { free(zpcm_buffers); zpcm_buffers = 0; }
42   if( zpcm_areas != 0 ) { free(zpcm_areas); zpcm_areas = 0; }
43   if( zpcm_samples != 0 ) { free(zpcm_samples); zpcm_samples = 0; }
44 #endif
45   if( zpcm != 0 ) { snd_pcm_close(zpcm);  zpcm = 0; }
46 }
47
48 void alsa_open(int chs,int rate)
49 {
50   int ich, bits, byts, dir, ret;
51   snd_pcm_format_t fmt = SND_PCM_FORMAT_S16;
52   snd_pcm_hw_params_t *phw;
53   snd_pcm_sw_params_t *psw;
54   snd_pcm_hw_params_alloca(&phw);
55   snd_pcm_sw_params_alloca(&psw);
56
57   zpcm = 0;
58   ret = snd_pcm_open(&zpcm, zpcm_device,
59      SND_PCM_STREAM_PLAYBACK, SND_PCM_NONBLOCK);
60   if( ret >= 0 )
61     ret = snd_pcm_hw_params_any(zpcm, phw);
62   if( ret >= 0 )
63     ret = snd_pcm_hw_params_set_rate_resample(zpcm, phw, 1);
64   if( ret >= 0 )
65     ret = snd_pcm_hw_params_set_access(zpcm, phw,
66       SND_PCM_ACCESS_RW_NONINTERLEAVED);
67   if( ret >= 0 )
68     ret = snd_pcm_hw_params_set_format(zpcm, phw, fmt);
69   if( ret >= 0 ) {
70     zpcm_channels = chs;
71     ret = snd_pcm_hw_params_set_channels(zpcm, phw, chs);
72   }
73   if( ret >= 0 ) {
74     zpcm_rate = rate;
75     ret = snd_pcm_hw_params_set_rate_near(zpcm, phw, &zpcm_rate, 0);
76     if( (int)zpcm_rate != rate )
77       printf("nearest audio_rate for %d is %u\n",rate,zpcm_rate);
78   }
79   if( ret >= 0 )
80     ret = snd_pcm_hw_params_set_buffer_time_near(zpcm, phw,
81       &zpcm_bfr_time_us, &dir);
82   if( ret >= 0 )
83     ret = snd_pcm_hw_params_get_buffer_size(phw, &zpcm_ufrm_size);
84   if( ret >= 0 ) {
85     zpcm_sbfr_size = zpcm_ufrm_size;
86     ret = snd_pcm_hw_params_set_period_time_near(zpcm, phw,
87       &zpcm_per_time_us, &dir);
88   }
89   if( ret >= 0 )
90     ret = snd_pcm_hw_params_get_period_size(phw, &zpcm_ufrm_size, &dir);
91   if( ret >= 0 ) {
92     zpcm_sper_size = zpcm_ufrm_size;
93     ret = snd_pcm_hw_params(zpcm, phw);
94   }
95   if( ret >= 0 )
96     ret = snd_pcm_sw_params_current(zpcm, psw);
97   if( ret >= 0 )
98     ret = snd_pcm_sw_params_set_start_threshold(zpcm, psw,
99       (zpcm_sbfr_size / zpcm_sper_size) * zpcm_sper_size);
100   if( ret >= 0 )
101     ret = snd_pcm_sw_params_set_avail_min(zpcm, psw, zpcm_sper_size);
102   if( ret >= 0 )
103     ret = snd_pcm_sw_params(zpcm, psw);
104   /* snd_pcm_dump(zpcm, stdout); */
105
106   if( ret >= 0 ) {
107 #ifdef __cplusplus
108      zpcm_areas = new snd_pcm_channel_area_t[chs];
109      byts = snd_pcm_format_physical_width(fmt) / 8;
110      zpcm_samples = new short[zpcm_sper_size * chs * byts/2];
111      zpcm_buffers = new short *[chs];
112 #else
113      zpcm_areas = calloc(chs, sizeof(snd_pcm_channel_area_t));
114      bits = snd_pcm_format_physical_width(fmt);
115      byts = bits / 8;
116      zpcm_samples = malloc(zpcm_sper_size * chs * byts);
117      zpcm_buffers = malloc(chs * sizeof(short*));
118 #endif
119      if( zpcm_samples ) {
120        for( ich = 0; ich < chs; ++ich ) {
121          zpcm_areas[ich].addr = zpcm_samples;
122          zpcm_areas[ich].first = ich * zpcm_sper_size * bits;
123          zpcm_areas[ich].step = bits;
124          zpcm_buffers[ich] = zpcm_samples + ich*zpcm_sper_size;
125        }
126      }
127      else {
128        fprintf(stderr,"alsa sample buffer allocation failure.\n");
129        ret = -999;
130      }
131   }
132   if( ret < 0 ) {
133     if( ret > -999 )
134       printf("audio error: %s\n", snd_strerror(ret));
135     alsa_close();
136   }
137 }
138
139 short *alsa_bfr(int ch)
140 {
141   return zpcm_buffers[ch];
142 }
143
144 int alsa_bfrsz()
145 {
146   return zpcm_sper_size;
147 }
148
149 int alsa_recovery(int ret)
150 {
151   printf("alsa recovery\n");
152   switch( ret ) {
153   case -ESTRPIPE:
154     /* wait until the suspend flag is released, then fall through */
155     while( (ret=snd_pcm_resume(zpcm)) == -EAGAIN ) usleep(100000);
156   case -EPIPE:
157     ret = snd_pcm_prepare(zpcm);
158     if( ret < 0 )
159       printf("underrun, prepare failed: %s\n", snd_strerror(ret));
160     break;
161   default:
162     printf("unhandled error: %s\n",snd_strerror(ret));
163     break;
164   }
165   return ret;
166 }
167
168 int alsa_write(int len)
169 {
170   int i, ret;
171 #ifdef __cplusplus
172   short *bfrs[zpcm_channels];
173 #else
174   short **bfrs = alloca(zpcm_channels*sizeof(short *));
175 #endif
176   ret = 0;
177   for( i=0; i<zpcm_channels; ++i ) bfrs[i] = zpcm_buffers[i];
178
179   while( len > 0 ) {
180     ret = snd_pcm_writen(zpcm,(void **)bfrs, len);
181     if( ret == -EAGAIN ) continue;
182     if ( ret < 0 ) {
183       alsa_recovery(ret);
184       break;
185     }
186     for( i=0; i<zpcm_channels; ++i ) bfrs[i] += ret;
187     len -= ret;
188   }
189   return ret < 0 ? ret : 0;
190 }
191
192 #endif
193
194 int done = 0;
195
196 void sigint(int n)
197 {
198   done = 1;
199 }
200
201
202 void dst_exit(GtkWidget *widget, gpointer data)
203 {
204    exit(0);
205 }
206
207 double the_time()
208 {
209   struct timeval tv;
210   gettimeofday(&tv,NULL);
211   return tv.tv_sec + tv.tv_usec/1000000.0;
212 }
213
214 int main(int ac, char **av)
215 {
216   int ret, more_data, frame, astream, vstream;
217   int has_audio, total_astreams, audio_channels, sample_rate;
218   int has_video, total_vstreams, width, height, colormodel;
219   int owidth, oheight;
220   long audio_samples, video_frames;
221   float frame_rate;
222
223 #ifdef VIDEO
224   GtkWidget *window;
225   GtkWidget *panel_hbox;
226   GtkWidget *image;
227   GdkPixbuf *pbuf0, *pbuf1;
228   GdkImage  *img0, *img1;
229   GdkVisual *visual;
230   double start, zstart, now, znow, delay;
231   unsigned char **rows, **row0, **row1, *cap, *cap0, *cap1;
232   int row;
233
234   start = the_time();
235   zstart = -1;
236 #endif
237
238   setbuf(stdout,NULL);
239
240   zmpeg3_t* zsrc = mpeg3_open("/tmp/dat",&ret);
241   //zmpeg3_t* zsrc = mpeg3_open("/root/LimeWire/Shared/Britney Spears - Pepsi Commercial with Bob Dole.mpeg",&ret);
242   //zmpeg3_t* zsrc = mpeg3_open("/tmp/dat.mp3",&ret);
243   //zmpeg3_t* zsrc = mpeg3_open("/dvd/VIDEO_TS/VIDEO_TS.IFO",&ret);
244   printf(" ret = %d\n",ret);
245   if( ret != 0 ) exit(1);
246   has_audio = mpeg3_has_audio(zsrc);
247   printf(" has_audio = %d\n", has_audio);
248   total_astreams = mpeg3_total_astreams(zsrc);
249   printf(" total_astreams = %d\n", total_astreams);
250   for( astream=0; astream<total_astreams; ++astream ) {
251     audio_channels = mpeg3_audio_channels(zsrc, astream);
252     printf("   audio_channels = %d\n", audio_channels);
253     sample_rate = mpeg3_sample_rate(zsrc, astream);
254     printf("   sample_rate = %d\n", sample_rate);
255     audio_samples = mpeg3_audio_samples(zsrc, astream);
256     printf("   audio_samples = %ld\n", audio_samples);
257   }
258   printf("\n");
259   has_video = mpeg3_has_video(zsrc);
260   printf(" has_video = %d\n", has_video);
261   total_vstreams = mpeg3_total_vstreams(zsrc);
262   printf(" total_vstreams = %d\n", total_vstreams);
263   for( vstream=0; vstream<total_vstreams; ++vstream ) {
264     width = mpeg3_video_width(zsrc, vstream);
265     printf("   video_width = %d\n", width);
266     height = mpeg3_video_height(zsrc, vstream);
267     printf("   video_height = %d\n", height);
268     frame_rate = mpeg3_frame_rate(zsrc, vstream);
269     printf("   frame_rate = %f\n", frame_rate);
270     video_frames = mpeg3_video_frames(zsrc, vstream);
271     printf("   video_frames = %ld\n", video_frames);
272     colormodel = mpeg3_colormodel(zsrc, vstream);
273     printf("   colormodel = %d\n", colormodel);
274   }
275
276   //mpeg3_set_cpus(zsrc,2);
277   mpeg3_seek_byte(zsrc,0);
278   //oheight = 1050-96;
279   //owidth = 1680-64;
280   oheight = height;
281   owidth = width;
282
283 #ifdef AUDIO
284   alsa_open(audio_channels,sample_rate);
285 #endif
286 #ifdef VIDEO
287   gtk_set_locale();
288   gtk_init(&ac, &av);
289   visual = gdk_visual_get_system();
290   /* toplevel window */
291   window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
292   gtk_signal_connect(GTK_OBJECT(window),"destroy",
293      GTK_SIGNAL_FUNC(dst_exit),NULL);
294   gtk_window_set_position(GTK_WINDOW(window), GTK_WIN_POS_NONE);
295   /* try for shared image bfr, only seems to work with gtk_rgb */
296   img0 = gdk_image_new(GDK_IMAGE_SHARED, visual, owidth, oheight);
297   pbuf0 = gdk_pixbuf_new_from_data((const guchar *)img0->mem,
298              GDK_COLORSPACE_RGB,FALSE,8,owidth,oheight,owidth*3,NULL,NULL);
299   cap0 = gdk_pixbuf_get_pixels(pbuf0);
300   image = gtk_image_new_from_pixbuf(pbuf0);
301   /* double buffered */
302   img1 = gdk_image_new(GDK_IMAGE_SHARED, visual, owidth, oheight);
303   pbuf1 = gdk_pixbuf_new_from_data((const guchar *)img1->mem,
304              GDK_COLORSPACE_RGB,FALSE,8,owidth,oheight,owidth*3,NULL,NULL);
305   cap1 = gdk_pixbuf_get_pixels(pbuf1);
306
307   panel_hbox = gtk_hbox_new(FALSE,0);
308   gtk_container_add(GTK_CONTAINER(window), panel_hbox);
309   /* pack image into panel */
310   gtk_box_pack_start(GTK_BOX(panel_hbox), image, TRUE, TRUE, 0);
311
312
313 #ifdef __cplusplus
314   row0 = new unsigned char *[oheight];
315   row1 = new unsigned char *[oheight];
316   int cmdl = zmpeg3_t::cmdl_RGB888;
317 #else
318   row0 = malloc(oheight*sizeof(unsigned char *));
319   row1 = malloc(oheight*sizeof(unsigned char *));
320   int cmdl = MPEG3_RGB888;
321 #endif
322   for( row=0; row<oheight; ++row ) {
323     row0[row] = cap0 + row*owidth*3;
324     row1[row] = cap1 + row*owidth*3;
325   }
326
327   gtk_widget_show_all(window);
328   cap = cap0;
329   rows = row0;
330 #endif
331
332   signal(SIGINT,sigint);
333   frame = 0;
334   more_data = 1;
335
336   while( done == 0 && more_data ) {
337 #ifdef VIDEO
338     if( !gtk_events_pending() ) {
339 #endif
340       more_data = 0;
341 #ifdef AUDIO
342       int ich;
343       int n = alsa_bfrsz();
344       for( ich=0; ich<audio_channels; ++ich) {
345         short *bfr = alsa_bfr(ich);
346         ret = ich == 0 ?
347           mpeg3_read_audio(zsrc, 0, bfr, ich, n, 0) :
348           mpeg3_reread_audio(zsrc, 0, bfr, ich, n, 0);
349         printf("read_audio(stream=%d,channel=%d) = %d\n", 0, ich, ret);
350       }
351       alsa_write(n);
352       more_data |= !mpeg3_end_of_audio(zsrc,0);
353 #endif
354 #ifdef VIDEO
355       ret = mpeg3_read_frame(zsrc, rows, 0, 0, width, height,
356                owidth, oheight, cmdl, 0);
357       printf("read_video(stream=%d, frame=%d) = %d\n", 0, frame, ret);
358       GdkGC *blk = image->style->black_gc;
359       gdk_draw_rgb_image(image->window,blk, 0,0,owidth,oheight,
360          GDK_RGB_DITHER_NONE,cap,owidth*3);
361       *(unsigned long *)&cap ^= ((unsigned long)cap0 ^ (unsigned long)cap1);
362       *(unsigned long *)&rows ^= ((unsigned long)row0 ^ (unsigned long)row1);
363       now = the_time();
364       znow = mpeg3_get_time(zsrc);
365       if( zstart < 0. ) zstart = znow;
366       delay = (znow-zstart) - (now-start);
367       if( delay > 0 ) usleep((int)(delay*100000.0));
368       more_data |= !mpeg3_end_of_video(zsrc,0);
369 #endif
370       ++frame;
371 #ifdef VIDEO
372     }
373     else
374       gtk_main_iteration();
375 #endif
376   }
377
378 #ifdef AUDIO
379   alsa_close();
380 #endif
381   mpeg3_close(zsrc);
382   return 0;
383 }
384