source: orxonox.OLD/trunk/src/world_entities/recorder.h @ 6826

/*!
 * @file recorder.h
 * 
 */

#ifndef _RECORDER_H
#define _RECORDER_H

#include "sdlincl.h"

#ifdef HAVE_AVFORMAT_H
  #include <avformat.h>
#else
  #include <ffmpeg/avformat.h>
#endif

#include "glincl.h"
#include "world_entity.h"

class Recorder : public WorldEntity
{
  private:
    AVFrame* picture;
    uint8_t* buffer;
    int frame_count;
    int buffer_size;

    AVOutputFormat* output_format;
    AVFormatContext* format_context;
    AVCodecContext* codec_context;
    AVCodec *codec;
    AVStream* video_stream;
    AVPacket packet;
    double video_pts;
    
    uint8_t *picture_buf;
    int size;

    int height;
    int width;
    float time;

    float stream_duration;
    float stream_frame_rate;
    int stream_nb_frames;

AVFrame* RGB_frame;
  public:
    Recorder (const TiXmlElement* root = NULL);
    ~Recorder ();

    virtual void draw () const;
    virtual void tick (float dt);

    virtual void loadParams(const TiXmlElement* root);

  private:
    void initVideo(const char* filename);
    void addVideoStream();
    void openVideo();
    void closeVideo();
    void allocPicture();
    void writeVideoFrame();
    void fillYuvImage();

    void setStreamDuration(float duration);
    void setFPS(float fps);
};

#endif /* _RECORDER_H */