fileogg.h

Go to the documentation of this file.

#ifndef FILEOGG_H
#define FILEOGG_H

#include "../config.h"
#include "filebase.h"
#include "file.inc"

#include <theora/theora.h>
#include <vorbis/codec.h>
#include <vorbis/vorbisenc.h>


#include <libdv/dv.h>

/* This code was aspired by ffmpeg2theora */
/* Special thanks for help on this code goes out to j@v2v.cc */

typedef struct 
{
        ogg_sync_state sync;
        off_t file_bufpos; // position of the start of the buffer inside the file
        off_t file_pagepos; // position of the page that will be next read 
        off_t file_pagepos_found; // position of last page that was returned (in seeking operations)
        int wlen;
} sync_window_t;

typedef struct
{
    ogg_page audiopage;
    ogg_page videopage;

    double audiotime;
    double videotime;
    ogg_int64_t audio_bytesout;
    ogg_int64_t video_bytesout;

    ogg_page og;    /* one Ogg bitstream page.  Vorbis packets are inside */
    ogg_packet op;  /* one raw packet of data for decode */

    theora_info ti;
    theora_comment tc;
    theora_state td;

    vorbis_info vi;       /* struct that stores all the static vorbis bitstream settings */
    vorbis_comment vc;    /* struct that stores all the user comments */
    vorbis_dsp_state vd; /* central working state for the packet<->PCM encoder/decoder */
    vorbis_block vb;     /* local working space for packet<->PCM encode/decode */

    /* used for muxing */
    ogg_stream_state to;    /* take physical pages, weld into a logical
                             * stream of packets */
    ogg_stream_state vo;    /* take physical pages, weld into a logical
                             * stream of packets */

    int apage_valid;
    int vpage_valid;
    unsigned char *apage;
    unsigned char *vpage;
    int vpage_len;
    int apage_len;
    int vpage_buffer_length;
    int apage_buffer_length;


// stuff needed for reading only
        sync_window_t *audiosync;
        sync_window_t *videosync;
        

}
theoraframes_info_t;

class FileOGG : public FileBase
{
public:
        FileOGG(Asset *asset, File *file);
        ~FileOGG();

        static void get_parameters(BC_WindowBase *parent_window,
                Asset *asset,
                BC_WindowBase* &format_window,
                int audio_options,
                int video_options);
        
        int reset_parameters_derived();
        int open_file(int rd, int wr);
        static int check_sig(Asset *asset);
        int close_file();
        int close_file_derived();
        int64_t get_video_position();
        int64_t get_audio_position();
        int set_video_position(int64_t x);
        int set_audio_position(int64_t x);
        int colormodel_supported(int colormodel);
        int get_best_colormodel(Asset *asset, int driver);
        int write_samples(double **buffer, int64_t len);
        int write_frames(VFrame ***frames, int len);
        int read_samples(double *buffer, int64_t len);
        int read_frame(VFrame *frame);

private:
        int write_samples_vorbis(double **buffer, int64_t len, int e_o_s);
        int write_frames_theora(VFrame ***frames, int len, int e_o_s);
        void flush_ogg(int e_o_s);
        int write_audio_page();
        int write_video_page();
        
        FILE *stream;
        off_t file_length;
        
        theoraframes_info_t *tf;
        VFrame *temp_frame;
        Mutex *flush_lock;

        off_t filedata_begin;

        int ogg_get_last_page(sync_window_t *sw, long serialno, ogg_page *og);
        int ogg_get_prev_page(sync_window_t *sw, long serialno, ogg_page *og);
        int ogg_get_first_page(sync_window_t *sw, long serialno, ogg_page *og);
        int ogg_get_next_page(sync_window_t *sw, long serialno, ogg_page *og);
        int ogg_sync_and_get_next_page(sync_window_t *sw, long serialno, ogg_page *og);

        int ogg_get_page_of_sample(sync_window_t *sw, long serialno, ogg_page *og, int64_t sample);
        int ogg_seek_to_sample(sync_window_t *sw, long serialno, int64_t sample);
        int ogg_decode_more_samples(sync_window_t *sw, long serialno);

        int ogg_get_page_of_frame(sync_window_t *sw, long serialno, ogg_page *og, int64_t frame);
        int ogg_seek_to_keyframe(sync_window_t *sw, long serialno, int64_t frame, int64_t *keyframe_number);


        int64_t start_sample; // first and last sample inside this file
        int64_t last_sample;    
        int64_t start_frame; // first and last frame inside this file
        int64_t last_frame;     
        

        int64_t ogg_sample_position;  // what will be the next sample taken from vorbis decoder
        int64_t next_sample_position; // what is the next sample read_samples must deliver

        int move_history(int from, int to, int len);

        float **pcm_history;
#ifndef HISTORY_MAX
#define HISTORY_MAX 0x100000
#endif
        int64_t history_start;
        int64_t history_size;

        int theora_cmodel;
        int64_t ogg_frame_position;    // LAST decoded frame position
        int64_t next_frame_position;   // what is the next sample read_frames must deliver
        char theora_keyframe_granule_shift;
};

class OGGConfigAudio;
class OGGConfigVideo;

class OGGVorbisFixedBitrate : public BC_Radial
{
public:
        OGGVorbisFixedBitrate(int x, int y, OGGConfigAudio *gui);
        int handle_event();
        OGGConfigAudio *gui;
};

class OGGVorbisVariableBitrate : public BC_Radial
{
public:
        OGGVorbisVariableBitrate(int x, int y, OGGConfigAudio *gui);
        int handle_event();
        OGGConfigAudio *gui;
};

class OGGVorbisMinBitrate : public BC_TextBox
{
public:
        OGGVorbisMinBitrate(int x, 
                int y, 
                OGGConfigAudio *gui, 
                char *text);
        int handle_event();
        OGGConfigAudio *gui;
};

class OGGVorbisMaxBitrate : public BC_TextBox
{
public:
        OGGVorbisMaxBitrate(int x, 
                int y, 
                OGGConfigAudio *gui, 
                char *text);
        int handle_event();
        OGGConfigAudio *gui;
};

class OGGVorbisAvgBitrate : public BC_TextBox
{
public:
        OGGVorbisAvgBitrate(int x, 
                int y, 
                OGGConfigAudio *gui, 
                char *text);
        int handle_event();
        OGGConfigAudio *gui;
};


class OGGConfigAudio: public BC_Window
{
public:
        OGGConfigAudio(BC_WindowBase *parent_window, Asset *asset);
        ~OGGConfigAudio();

        int create_objects();
        int close_event();

        Asset *asset;
        OGGVorbisFixedBitrate *fixed_bitrate;
        OGGVorbisVariableBitrate *variable_bitrate;
private:
        BC_WindowBase *parent_window;
        char string[BCTEXTLEN];
};


class OGGTheoraBitrate : public BC_TextBox
{
public:
        OGGTheoraBitrate(int x, int y, OGGConfigVideo *gui);
        int handle_event();
        OGGConfigVideo *gui;
};

class OGGTheoraKeyframeFrequency : public BC_TumbleTextBox
{
public:
        OGGTheoraKeyframeFrequency(int x, int y, OGGConfigVideo *gui);
        int handle_event();
        OGGConfigVideo *gui;
};

class OGGTheoraKeyframeForceFrequency : public BC_TumbleTextBox
{
public:
        OGGTheoraKeyframeForceFrequency(int x, int y, OGGConfigVideo *gui);
        int handle_event();
        OGGConfigVideo *gui;
};

class OGGTheoraSharpness : public BC_TumbleTextBox
{
public:
        OGGTheoraSharpness(int x, int y, OGGConfigVideo *gui);
        int handle_event();
        OGGConfigVideo *gui;
};

class OGGTheoraFixedBitrate : public BC_Radial
{
public:
        OGGTheoraFixedBitrate(int x, int y, OGGConfigVideo *gui);
        int handle_event();
        OGGConfigVideo *gui;
};

class OGGTheoraFixedQuality : public BC_Radial
{
public:
        OGGTheoraFixedQuality(int x, int y, OGGConfigVideo *gui);
        int handle_event();
        OGGConfigVideo *gui;
};



class OGGConfigVideo: public BC_Window
{
public:
        OGGConfigVideo(BC_WindowBase *parent_window, Asset *asset);
        ~OGGConfigVideo();

        int create_objects();
        int close_event();

        OGGTheoraFixedBitrate *fixed_bitrate;
        OGGTheoraFixedQuality *fixed_quality;
        Asset *asset;
private:
        BC_WindowBase *parent_window;
};


#endif

---