Planet
navi homePPSaboutscreenshotsdownloaddevelopmentforum

source: orxonox.OLD/branches/avi_play/src/lib/graphics/importer/media_container.cc @ 6333

Last change on this file since 6333 was 6333, checked in by hdavid, 18 years ago

branches\avi_play: fixed a bug, still memory leak somewhere in MediaContainer
File size: 12.5 KB
Line 
1/*
2   orxonox - the future of 3D-vertical-scrollers
3
4   Copyright (C) 2004 orx
5
6   This program is free software; you can redistribute it and/or modify
7   it under the terms of the GNU General Public License as published by
8   the Free Software Foundation; either version 2, or (at your option)
9   any later version.
10
11### File Specific:
12   main-programmer: David Hasenfratz, Stephan Lienhard
13   co-programmer:
14*/
15
16
17
18/* this is for debug output. It just says, that all calls to PRINT() belong to the DEBUG_MODULE_MEDIA module
19   For more information refere to https://www.orxonox.net/cgi-bin/trac.cgi/wiki/DebugOutput
20*/
21#define DEBUG_MODULE_MEDIA
22
23
24/* include your own header */
25#include "media_container.h"
26
27/* header for debug output */
28#include "debug.h"
29
30
31/**
32 * Default constructor
33 */
34MediaContainer::MediaContainer(const char* filename)
35{
36  this->init();
37
38  if (filename != NULL)
39    this->loadMedia(filename);
40}
41
42MediaContainer::MediaContainer()
43{
44  this->init();
45}
46
47/**
48 * Default destructor
49 */
50MediaContainer::~MediaContainer()
51{
52  // delete all textures.
53  while(!this->texture_list.empty())
54  {
55    if (glIsTexture(this->texture_list.back()))
56      glDeleteTextures(1, &this->texture_list.back());
57    this->texture_list.pop_back();
58  }
59  glDeleteTextures(1, &texture);
60  SDL_FreeSurface(surface);
61
62  // Free the RGB image
63  delete [] buffer;
64  av_free(RGB_frame);
65
66  /* Free the frame */
67  av_free(frame);
68
69  /* Close the codec */
70  avcodec_close(codec_context);
71
72  /* Close the video file */
73  av_close_input_file(format_context);
74
75}
76
77void MediaContainer::init()
78{
79  /* set the class id for the base object */
80  this->setClassID(CL_MEDIA_CONTAINER, "MediaContainer");
81
82  /* register all formats and codecs */
83  av_register_all();
84
85  fps = 0;
86  frame_num = 0;
87}
88
89void MediaContainer::gotoFrame(int frame_number)
90{
91  // seek doesnt work for the first two frames
92  // you will get ugly fragments
93  if(frame_number < 2)
94  {
95    // go to the begin of the video
96    av_seek_frame(format_context, video_stream, 0, AVSEEK_FLAG_BACKWARD);
97    frame_num = 0;
98  }
99  else
100  {
101    // seeks to the nearest keyframe
102    // NOTE: there is only about every 5s a keyframe!
103    av_seek_frame(format_context, video_stream, frame_number, AVSEEK_FLAG_BACKWARD);
104   
105    // go from the keyframe to the exact position
106    codec_context->hurry_up = 1;
107    do {
108      av_read_frame(format_context, &packet);
109      if(packet.pts >= frame_number-1)
110        break;
111      int frame_finished;
112      avcodec_decode_video(codec_context, frame, &frame_finished, packet.data, packet.size);
113      av_free_packet(&packet);
114    } while(1);
115    codec_context->hurry_up = 0;
116 
117    frame_num = frame_number;
118  }
119}
120
121GLuint MediaContainer::getNextFrame()
122{
123  /* get next frame */
124  if(av_read_frame(format_context, &packet) >= 0)
125  {
126    //this->printPacketInformation();
127
128    /* Is this a packet from the video stream? */
129    if(packet.stream_index == video_stream)
130    {
131      int frame_finished;
132      // Decode video frame
133      avcodec_decode_video(codec_context, frame, &frame_finished,
134                           packet.data, packet.size);
135
136      // Free the packet that was allocated by av_read_frame
137      av_free_packet(&packet);
138     
139      // Did we get a video frame?
140      if(frame_finished)
141      {
142        frame_num++;
143        //PRINTF(1)("frame_number: %i\n", this->getFrameNumber());
144        // Conversion from YUV to RGB
145        // Most codecs return images in YUV 420 format
146        // (one luminance and two chrominance channels, with the chrominance
147        // channels samples at half the spatial resolution of the luminance channel)
148        img_convert((AVPicture*)RGB_frame, PIX_FMT_RGB24, (AVPicture*)frame,
149                    codec_context->pix_fmt, codec_context->width, codec_context->height);
150
151        picture = (AVPicture*)RGB_frame;
152
153
154        data = 0;
155        data = new uint8_t[codec_context->width*codec_context->height*3*sizeof(uint8_t)];
156        for(int i = 0; i < codec_context->height; i++)
157          memcpy(&data[i*codec_context->width*3], picture->data[0]+i *
158                 picture->linesize[0],codec_context->width*sizeof(uint8_t)*3);
159
160        surface = SDL_CreateRGBSurfaceFrom(data, codec_context->width,
161                                           codec_context->height,24,
162                                           codec_context->width*sizeof(uint8_t)*3,
163#if SDL_BYTEORDER == SDL_LIL_ENDIAN /* OpenGL RGBA masks */
164                                           0x000000FF,
165                                           0x0000FF00,
166                                           0x00FF0000,
167                                           0
168#else
169                                           0xFF000000,
170                                           0x00FF0000,
171                                           0x0000FF00,
172                                           0
173#endif
174                                            );
175
176        if(frame_num == 1)
177        {
178          /* Create an OpenGL texture from the surface */
179          glGenTextures(1, &texture);
180          glBindTexture(GL_TEXTURE_2D, texture);
181          glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
182          glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
183          // create the texture
184          glTexImage2D(GL_TEXTURE_2D,
185                      0,
186                      GL_RGB,
187                      surface->w, surface->h,
188                      0,
189                      GL_RGB,
190                      GL_UNSIGNED_BYTE,
191                      surface->pixels);
192          // build the MipMaps
193          gluBuild2DMipmaps(GL_TEXTURE_2D,
194                          GL_RGB,
195                          surface->w,
196                          surface->h,
197                          GL_RGB,
198                          GL_UNSIGNED_BYTE,
199                          surface->pixels);
200          glBindTexture(GL_TEXTURE_2D, 0);
201        }
202        else
203        {
204          /* Create an OpenGL texture from the surface */
205          glGenTextures(1, &texture);
206          glBindTexture(GL_TEXTURE_2D, texture);
207          glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
208          glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
209          // update the texture
210          glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, surface->w, surface->h, GL_RGB, GL_UNSIGNED_BYTE, surface->pixels);
211
212          // build the MipMaps
213          gluBuild2DMipmaps(GL_TEXTURE_2D,
214                          GL_RGB,
215                          surface->w,
216                          surface->h,
217                          GL_RGB,
218                          GL_UNSIGNED_BYTE,
219                          surface->pixels);
220          glBindTexture(GL_TEXTURE_2D, 0);
221        }
222
223        return texture;
224      }
225    }
226    else
227    {
228      av_free_packet(&packet);
229      this->getNextFrame();
230    }
231  }
232  else
233    return NULL;
234}
235
236GLuint MediaContainer::skipFrame(int num_frames)
237{
238  frame_num += num_frames;
239 
240  while(num_frames != 0)
241  {
242    if(av_read_frame(format_context, &packet) < 0)
243      break;
244    if(packet.stream_index == video_stream)
245    {
246      int frame_finished;
247      // We have to decode the frame to not get ugly fragments
248      avcodec_decode_video(codec_context, frame, &frame_finished,
249                            packet.data, packet.size);
250       
251      // Did we get a video frame?
252      if(frame_finished)
253        num_frames--;
254    }
255    av_free_packet(&packet);
256  }
257 
258  return this->getNextFrame();
259}
260
261vector<GLuint> MediaContainer::getFrameList()
262{
263
264  while((texture = this->getNextFrame()) != NULL)
265    texture_list.push_back(texture);
266
267  return texture_list;
268}
269
270void MediaContainer::saveCurrentFrame()
271{
272  FILE *file;
273  char filename[32];
274  int  y;
275
276  // Open file
277  sprintf(filename, "frame%i.ppm", frame_num);
278  file = fopen(filename, "wb");
279  if(file == NULL)
280        return;
281
282  // Write header
283  fprintf(file, "P6\n%d %d\n255\n", codec_context->width, codec_context->height);
284  // Write pixel data
285  for(y = 0; y < codec_context->height; y++)
286    fwrite(picture->data[0]+y * picture->linesize[0], 1, codec_context->width*3, file);
287  // Close file
288  fclose(file);
289
290  PRINTF(1)("created file: %s\n", filename);
291}
292
293void MediaContainer::loadMedia(const char* filename)
294{
295  /* Open video file */
296  if (av_open_input_file(&format_context, filename, NULL, 0, NULL) !=0 )
297    PRINTF(1)("Could not open %s\n", filename);
298
299  /* Retrieve stream information */
300  if (av_find_stream_info(format_context) < 0)
301    PRINTF(1)("Could not find stream information in %s\n", filename);
302
303  // Dump information about file onto standard error
304  //dump_format(pFormatCtx, 0, argv[1], false);
305
306  /* Find the first video stream and take it */
307  video_stream = -1;
308  for(int i = 0; i < format_context->nb_streams; i++)
309  {
310    // NOTE: different code for the 0.4.9-pre1 release of ffmpeg (tardis)
311    // if(format_context->streams[i]->codec.codec_type == CODEC_TYPE_VIDEO)
312    if(format_context->streams[i]->codec->codec_type == CODEC_TYPE_VIDEO)
313    {
314      video_stream = i;
315      break;
316    }
317  }
318
319  if(video_stream == -1)
320    PRINTF(1)("Could not find a video stream in %s\n", filename);
321
322  /* Get a pointer to the codec context for the video stream */
323  // NOTE: different code for the 0.4.9-pre1 release of ffmpeg (tardis)
324  // codec_context = &format_context->streams[video_stream]->codec;
325  codec_context = format_context->streams[video_stream]->codec;
326
327  /* Find the decoder for the video stream */
328  codec = avcodec_find_decoder(codec_context->codec_id);
329  if (codec == NULL)
330    PRINTF(1)("Could not find codec\n");
331
332  /* Open codec */
333  if (avcodec_open(codec_context, codec) < 0)
334    PRINTF(1)("Could not open codec\n");
335
336  // Allocate video frame
337  frame = avcodec_alloc_frame();
338  RGB_frame = avcodec_alloc_frame();
339
340  // Determine required buffer size and allocate buffer
341  num_bytes = avpicture_get_size(PIX_FMT_RGB24, codec_context->width, codec_context->height);
342  buffer=new uint8_t[num_bytes];
343
344  // Assign appropriate parts of buffer to image planes in pFrameRGB
345  avpicture_fill((AVPicture *)RGB_frame, buffer, PIX_FMT_RGB24, codec_context->width, codec_context->height);
346
347  // Calculate fps
348  fps = av_q2d(format_context->streams[video_stream]->r_frame_rate);
349
350  // duration
351  duration = format_context->duration / 1000000LL;
352
353  frame_num = 0;
354}
355
356int MediaContainer::getHeight()
357{
358  return codec_context->height;
359}
360
361int MediaContainer::getWidth()
362{
363  return codec_context->width;
364}
365
366int MediaContainer::getFrameNumber()
367{
368  return frame_num;
369}
370
371double MediaContainer::getFPS()
372{
373  return this->fps;
374}
375
376void MediaContainer::printMediaInformation()
377{
378  PRINTF(1)("========================\n");
379  PRINTF(1)("========================\n");
380  PRINTF(1)("=    MEDIACONTAINER    =\n");
381  PRINTF(1)("========================\n");
382  PRINTF(1)("========================\n");
383  PRINTF(1)("=    AVFormatContext   =\n");
384  PRINTF(1)("========================\n");
385  PRINTF(1)("filename: %s\n", format_context->filename);
386  PRINTF(1)("nb_streams: %i\n", format_context->nb_streams);
387  PRINTF(1)("duration: (%02d:%02d:%02d)\n", duration/3600, (duration%3600)/60, duration%60);
388  PRINTF(1)("file_size: %ikb\n", format_context->file_size/1024);
389  PRINTF(1)("bit_rate: %ikb/s\n", format_context->bit_rate/1000);
390  PRINTF(1)("nb_frames: %i\n", format_context->streams[video_stream]->nb_frames);
391  PRINTF(1)("r_frame_rate: %i\n", format_context->streams[video_stream]->r_frame_rate.num);
392  PRINTF(1)("fps: %0.2f\n", av_q2d(format_context->streams[video_stream]->r_frame_rate));
393  PRINTF(1)("========================\n");
394  PRINTF(1)("=    AVCodecContext    =\n");
395  PRINTF(1)("========================\n");
396  PRINTF(1)("width: %i\n", codec_context->width);
397  PRINTF(1)("height: %i\n", codec_context->height);
398  PRINTF(1)("time_base.den: %i\n", codec_context->time_base.den);
399  PRINTF(1)("time_base.num: %i\n", codec_context->time_base.num);
400  PRINTF(1)("========================\n");
401  PRINTF(1)("=       AVCodec        =\n");
402  PRINTF(1)("========================\n");
403  PRINTF(1)("codec name: %s\n", codec->name);
404  PRINTF(1)("========================\n");
405  PRINTF(1)("========================\n");
406}
407
408void MediaContainer::printPacketInformation()
409{
410  PRINTF(1)("========================\n");
411  PRINTF(1)("========================\n");
412  PRINTF(1)("=       AVPacket       =\n");
413  PRINTF(1)("========================\n");
414  PRINTF(1)("pts: %i\n", packet.pts);
415  PRINTF(1)("dts: %i\n", packet.dts);
416  PRINTF(1)("size: %i\n", packet.size);
417  PRINTF(1)("stream_index: %i\n", packet.stream_index);
418  PRINTF(1)("duration: %i\n", packet.duration);
419  PRINTF(1)("pos: %i\n", packet.pos);
420  PRINTF(1)("========================\n");
421  PRINTF(1)("========================\n");
422}
Note: See TracBrowser for help on using the repository browser.