Changeset 6068 in orxonox.OLD for branches/avi_play/src/lib/graphics/importer/media_container.cc

Timestamp:

Dec 12, 2005, 4:32:11 PM (18 years ago)

Author:

hdavid

Message:

branches\avi_play: MediaContainer::getNextFrame() grabs a frame and saves it as a ppm image, would not try it with large video files

File:

• branches/avi_play/src/lib/graphics/importer/media_container.cc (modified) (7 diffs)

branches/avi_play/src/lib/graphics/importer/media_container.cc

@@ -41 +41 @@
 
   current_frame = 0;
-  num_frames = 0;
-        frames_left = true;
 
   if (filename != NULL)
@@ -54 +52 @@
 MediaContainer::~MediaContainer()
 {
+  // Free the RGB image
+  delete [] buffer;
+  av_free(RGB_frame);
 
   /* Free the frame */
@@ -66 +67 @@
 }
 
-/*GLuint MediaContainer::getFrame(int frame_number)
+GLuint MediaContainer::getFrame(int frame_number)
 {
 
@@ -73 +74 @@
 GLuint MediaContainer::getNextFrame()
 {
-
-}*/
-
-SDL_Surface* MediaContainer::getFrame(int frame_number)
-{
-
-}
-
-SDL_Surface* MediaContainer::getNextFrame()
-{
-        /* get next frame */
-        if(av_read_frame(format_context, &packet) >= 0)
-        {
-                frames_left = true;
-                current_frame++;
-                PRINTF(1)("current_frame: %i\n", current_frame);
-
-
-                /* work with the frame */
-                /* packet -> SDL_Surface */
-
-
-
-
-        }
-        else
-                frames_left = false;
-
+  /* get next frame */
+  if(av_read_frame(format_context, &packet) >= 0)
+  {
+    //this->printPacketInformation();
+
+    /* Is this a packet from the video stream? */
+    if(packet.stream_index == video_stream)
+    {
+      int frame_finished;
+      // Decode video frame
+      avcodec_decode_video(codec_context, frame, &frame_finished, packet.data, packet.size);
+
+      // Did we get a video frame?
+      if(frame_finished)
+      {
+        current_frame++;
+        PRINTF(1)("current_frame: %i\n", current_frame);
+
+        // Convert the image from its native format to RGB
+        img_convert((AVPicture*)RGB_frame, PIX_FMT_RGB24, (AVPicture*)frame, codec_context->pix_fmt,
+                                        codec_context->width, codec_context->height);
+                        
+        // save frame
+        this->saveCurrentFrame();
+
+
+        /* RGB_picture -> texture */
+
+
+        return texture;
+      }
+    }
+    // Free the packet that was allocated by av_read_frame
+    av_free_packet(&packet);
+  }
+  else
+    return NULL;
+}
+
+void MediaContainer::saveCurrentFrame()
+{
+  FILE *file;
+  char filename[32];
+  int  y;
+
+  picture = (AVPicture*)RGB_frame;
+
+  // Open file
+  sprintf(filename, "frame%i.ppm", current_frame);
+  file = fopen(filename, "wb");
+  if(file == NULL)
+        return;
+
+  // Write header
+  fprintf(file, "P6\n%d %d\n255\n", codec_context->width, codec_context->height);
+  // Write pixel data
+  for(y = 0; y < codec_context->height; y++)
+    fwrite(picture->data[0]+y * picture->linesize[0], 1, codec_context->width*3, file);
+  // Close file
+  fclose(file);
+
+  PRINTF(1)("created file: %s\n", filename);
 }
 
@@ -144 +179 @@
   /* Open codec */
   if (avcodec_open(codec_context, codec) < 0)
-    PRINTF(1)("Could not open codec\n");        
+    PRINTF(1)("Could not open codec\n");
+
+  // Allocate video frame
+  frame = avcodec_alloc_frame();
+  RGB_frame = avcodec_alloc_frame();
+
+  // Determine required buffer size and allocate buffer
+  num_bytes = avpicture_get_size(PIX_FMT_RGB24, codec_context->width, codec_context->height);
+  buffer=new uint8_t[num_bytes];
+
+  // Assign appropriate parts of buffer to image planes in pFrameRGB
+  avpicture_fill((AVPicture *)RGB_frame, buffer, PIX_FMT_RGB24, codec_context->width, codec_context->height);   
 
 }
@@ -150 +196 @@
 int MediaContainer::getHeight()
 {
-        return codec_context->height;
+  return codec_context->height;
 }
 
 int MediaContainer::getWidth()
 {
-        return codec_context->width;
+  return codec_context->width;
 }
 
@@ -168 +214 @@
 }
 
-bool MediaContainer::framesLeft()
-{
-        return frames_left;
-}
-
 void MediaContainer::printMediaInformation()
 {
-        PRINTF(1)("========================\n");
-        PRINTF(1)("========================\n");
-        PRINTF(1)("=    MEDIACONTAINER    =\n");
-        PRINTF(1)("========================\n");
-        PRINTF(1)("========================\n");
-        PRINTF(1)("=    AVFormatContext   =\n");
-        PRINTF(1)("========================\n");
-        PRINTF(1)("filename: %s\n", format_context->filename);
-        PRINTF(1)("nb_streams: %i\n", format_context->nb_streams);
-        PRINTF(1)("duration: %fs\n", format_context->duration/1000000.);
-        PRINTF(1)("file_size: %ikb\n", format_context->file_size/1024);
-        PRINTF(1)("bit_rate: %ikb/s\n", format_context->bit_rate/1000);
-        PRINTF(1)("nb_frames: %i\n", format_context->streams[video_stream]->nb_frames);
-        //PRINTF(1)("r_frame_rate: %ifps\n", format_context->streams[video_stream]->r_frame_rate.num);
-        PRINTF(1)("========================\n");
-        PRINTF(1)("=    AVCodecContext    =\n");
-        PRINTF(1)("========================\n");
-        PRINTF(1)("width: %i\n", codec_context->width);
-        PRINTF(1)("height: %i\n", codec_context->height);
-        PRINTF(1)("========================\n");
-        PRINTF(1)("=       AVCodec        =\n");
-        PRINTF(1)("========================\n");
-        PRINTF(1)("codec name: %s\n", codec->name);
-        PRINTF(1)("========================\n");
-        PRINTF(1)("========================\n");
-}
+  PRINTF(1)("========================\n");
+  PRINTF(1)("========================\n");
+  PRINTF(1)("=    MEDIACONTAINER    =\n");
+  PRINTF(1)("========================\n");
+  PRINTF(1)("========================\n");
+  PRINTF(1)("=    AVFormatContext   =\n");
+  PRINTF(1)("========================\n");
+  PRINTF(1)("filename: %s\n", format_context->filename);
+  PRINTF(1)("nb_streams: %i\n", format_context->nb_streams);
+  PRINTF(1)("duration: %fs\n", format_context->duration/1000000.);
+  PRINTF(1)("file_size: %ikb\n", format_context->file_size/1024);
+  PRINTF(1)("bit_rate: %ikb/s\n", format_context->bit_rate/1000);
+  PRINTF(1)("nb_frames: %i\n", format_context->streams[video_stream]->nb_frames);
+  PRINTF(1)("r_frame_rate: %i\n", format_context->streams[video_stream]->r_frame_rate.num);
+  PRINTF(1)("========================\n");
+  PRINTF(1)("=    AVCodecContext    =\n");
+  PRINTF(1)("========================\n");
+  PRINTF(1)("width: %i\n", codec_context->width);
+  PRINTF(1)("height: %i\n", codec_context->height);
+  PRINTF(1)("========================\n");
+  PRINTF(1)("=       AVCodec        =\n");
+  PRINTF(1)("========================\n");
+  PRINTF(1)("codec name: %s\n", codec->name);
+  PRINTF(1)("========================\n");
+  PRINTF(1)("========================\n");
+}
+
+void MediaContainer::printPacketInformation()
+{
+  PRINTF(1)("========================\n");
+  PRINTF(1)("========================\n");
+  PRINTF(1)("=       AVPacket       =\n");
+  PRINTF(1)("========================\n");
+  PRINTF(1)("pts: %i\n", packet.pts);
+  PRINTF(1)("dts: %i\n", packet.dts);
+  PRINTF(1)("size: %i\n", packet.size);
+  PRINTF(1)("stream_index: %i\n", packet.stream_index);
+  PRINTF(1)("duration: %i\n", packet.duration);
+  PRINTF(1)("pos: %i\n", packet.pos);
+  PRINTF(1)("========================\n");
+  PRINTF(1)("========================\n");
+}