cinelerra-5.1/cinelerra/fileogg.h

   1
   2 /*
   3  * CINELERRA
   4  * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
   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 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  */
  21
  22 #ifndef FILEOGG_H
  23 #define FILEOGG_H
  24 #ifdef HAVE_OGG
  25
  26 #include "edl.inc"
  27 #include "filebase.h"
  28 #include "file.inc"
  29 #include "mutex.inc"
  30
  31 #include <theora/theora.h>
  32 #include <theora/theoraenc.h>
  33 #include <theora/theoradec.h>
  34 #include <vorbis/codec.h>
  35 #include <vorbis/vorbisenc.h>
  36
  37
  38 /* This code was aspired by ffmpeg2theora */
  39 /* Special thanks for help on this code goes out to j@v2v.cc */
  40
  41 #define mn_pagesz 32
  42
  43 class sync_window_t : public ogg_sync_state
  44 {
  45 public:
  46         sync_window_t(FILE *fp, Mutex *sync_lock, int64_t begin=0, int64_t end=0);
  47         ~sync_window_t();
  48         int ogg_read_locked(int buflen);
  49         int ogg_read_buffer(int buflen);
  50         int ogg_read_buffer_at(off_t filepos, int buflen);
  51         int ogg_sync_and_take_page_out(ogg_page *og);
  52         int ogg_take_page_out_autoadvance(ogg_page *og);
  53         int ogg_sync_and_get_next_page(long serialno, ogg_page *og);
  54         int ogg_prev_page_search(long serialno,ogg_page *og,
  55                         off_t begin, off_t end);
  56         int ogg_get_prev_page(long serialno, ogg_page *og);
  57         int ogg_get_next_page(long serialno, ogg_page *og);
  58         int ogg_get_first_page(long serialno, ogg_page *og);
  59         int ogg_get_last_page(long serialno, ogg_page *og);
  60
  61         int64_t filepos; // current file position
  62         int64_t bufpos;  // position at start of read
  63         int64_t pagpos;  // last search target position
  64         FILE *fp;
  65         Mutex *sync_lock;
  66         int64_t file_begin, file_end;
  67 };
  68
  69 class OGG_PageBfr
  70 {
  71 public:
  72         int allocated, len;
  73         int valid, packets;
  74         int64_t position;
  75         uint8_t *page;
  76
  77         OGG_PageBfr();
  78         ~OGG_PageBfr();
  79         void demand(int sz);
  80         int write_page(FILE *fp);
  81         int64_t load(ogg_page *og);
  82 };
  83
  84 class FileOGG : public FileBase
  85 {
  86 public:
  87         FileOGG(Asset *asset, File *file);
  88         ~FileOGG();
  89
  90         static void get_parameters(BC_WindowBase *parent_window,
  91                 Asset *asset, BC_WindowBase *&format_window,
  92                 int audio_options,int video_options,EDL *edl);
  93         static int check_sig(Asset *asset);
  94         int open_file(int rd,int wr);
  95         int close_file();
  96
  97         int64_t get_video_position();
  98         int64_t get_audio_position();
  99         int set_video_position(int64_t pos);
 100         int colormodel_supported(int colormodel);
 101         int get_best_colormodel(Asset *asset,int driver);
 102         int read_frame(VFrame *frame);
 103         int set_audio_position(int64_t pos);
 104         int read_samples(double *buffer,int64_t len);
 105         int write_samples(double **buffer,int64_t len);
 106         int write_frames(VFrame ***frames,int len);
 107
 108 private:
 109         void init();
 110         int encode_theora_init();
 111         int encode_vorbis_init();
 112         int ogg_init_encode(FILE *out);
 113         void close_encoder();
 114         int decode_theora_init();
 115         int decode_vorbis_init();
 116         int ogg_init_decode(FILE *inp);
 117         void close_decoder();
 118
 119         int write_samples_vorbis(double **buffer, int64_t len, int e_o_s);
 120         int write_frames_theora(VFrame ***frames, int len, int e_o_s);
 121         void flush_ogg(int e_o_s);
 122         int write_audio_page();
 123         int write_video_page();
 124
 125         FILE *inp, *out;
 126         off_t file_length;
 127         int audio, video;
 128         int64_t frame_position;
 129         int64_t sample_position;
 130
 131         VFrame *temp_frame;
 132         Mutex *file_lock;
 133         float **pcm_history;
 134 #ifndef HISTORY_MAX
 135 #define HISTORY_MAX 0x100000
 136 #endif
 137         int64_t history_start;
 138         int64_t history_size;
 139         int pcm_channels;
 140         int ach, ahz;           // audio channels, sample rate
 141         int amn, amx;           // audio min/max bitrate
 142         int abr, avbr, aqu;     // audio bitrate, variable bitrate, quality
 143         int asz, afrmsz;        // audio sample size, frame size
 144
 145         off_t file_begin, file_end;
 146         sync_window_t *audiosync;
 147         sync_window_t *videosync;
 148
 149         int64_t ogg_sample_pos(ogg_page *og);
 150         int64_t ogg_next_sample_pos(ogg_page *og);
 151         int64_t ogg_frame_pos(ogg_page *og);
 152         int64_t ogg_next_frame_pos(ogg_page *og);
 153         int ogg_read_buffer(FILE *in, sync_window_t *sy, int buflen);
 154         int ogg_get_video_packet(ogg_packet *op);
 155         int ogg_get_audio_packet(ogg_packet *op);
 156
 157         int ogg_get_page_of_sample(ogg_page *og, int64_t sample);
 158         int ogg_seek_to_sample(int64_t sample);
 159         int ogg_decode_more_samples();
 160
 161         int ogg_get_page_of_frame(ogg_page *og, int64_t frame);
 162         int ogg_seek_to_keyframe(int64_t frame, int64_t *keyframe_number);
 163         int move_history(int from, int to, int len);
 164
 165         ogg_stream_state to;    // theora to ogg out
 166         ogg_stream_state vo;    // vorbis to ogg out
 167         int64_t ogg_sample_position, ogg_frame_position;
 168         int64_t next_sample_position, next_frame_position;
 169         int64_t start_sample, last_sample; // first and last sample inside this file
 170         int64_t start_frame, last_frame; // first and last frame inside this file
 171         int64_t audio_pos, video_pos;   // decoder last sample/frame in
 172         int audio_eos, video_eos;       // decoder sample/frame end of file
 173
 174         th_enc_ctx *enc;        // theora video encode context
 175         th_dec_ctx *dec;        // theora video decode context
 176         th_info ti;             // theora video encoder init parameters
 177         th_setup_info *ts;      // theora video setup huff/quant codes
 178         th_comment tc;          // header init parameters
 179         vorbis_info vi;         // vorbis audio encoder init parameters
 180         vorbis_comment vc;      // header init parameters
 181         vorbis_dsp_state vd;    // vorbis decode audio context
 182         vorbis_block vb;        // vorbis decode buffering
 183         int force_keyframes;
 184         int vp3_compatible;
 185         int soft_target;
 186
 187         int pic_x, pic_y, pic_w, pic_h;
 188         int frame_w, frame_h;
 189         int colorspace, pixfmt;
 190         int bitrate, quality;
 191         int keyframe_period, keyframe_force;
 192         int fps_num, fps_den;
 193         int aratio_num, aratio_den;
 194
 195         OGG_PageBfr apage, vpage;
 196         double audiotime, videotime;
 197         int keyframe_granule_shift;
 198         int iframe_granule_offset;
 199         int theora_cmodel;
 200 };
 201
 202 class OGGConfigAudio;
 203 class OGGConfigVideo;
 204
 205 class OGGVorbisFixedBitrate : public BC_Radial
 206 {
 207 public:
 208         OGGVorbisFixedBitrate(int x, int y, OGGConfigAudio *gui);
 209         int handle_event();
 210         OGGConfigAudio *gui;
 211 };
 212
 213 class OGGVorbisVariableBitrate : public BC_Radial
 214 {
 215 public:
 216         OGGVorbisVariableBitrate(int x, int y, OGGConfigAudio *gui);
 217         int handle_event();
 218         OGGConfigAudio *gui;
 219 };
 220
 221 class OGGVorbisMinBitrate : public BC_TextBox
 222 {
 223 public:
 224         OGGVorbisMinBitrate(int x,
 225                 int y,
 226                 OGGConfigAudio *gui,
 227                 char *text);
 228         int handle_event();
 229         OGGConfigAudio *gui;
 230 };
 231
 232 class OGGVorbisMaxBitrate : public BC_TextBox
 233 {
 234 public:
 235         OGGVorbisMaxBitrate(int x,
 236                 int y,
 237                 OGGConfigAudio *gui,
 238                 char *text);
 239         int handle_event();
 240         OGGConfigAudio *gui;
 241 };
 242
 243 class OGGVorbisAvgBitrate : public BC_TextBox
 244 {
 245 public:
 246         OGGVorbisAvgBitrate(int x,
 247                 int y,
 248                 OGGConfigAudio *gui,
 249                 char *text);
 250         int handle_event();
 251         OGGConfigAudio *gui;
 252 };
 253
 254
 255 class OGGConfigAudio: public BC_Window
 256 {
 257 public:
 258         OGGConfigAudio(BC_WindowBase *parent_window, Asset *asset);
 259         ~OGGConfigAudio();
 260
 261         void create_objects();
 262         int close_event();
 263
 264         Asset *asset;
 265         OGGVorbisFixedBitrate *fixed_bitrate;
 266         OGGVorbisVariableBitrate *variable_bitrate;
 267 private:
 268         BC_WindowBase *parent_window;
 269         char string[BCTEXTLEN];
 270 };
 271
 272
 273 class OGGTheoraBitrate : public BC_TextBox
 274 {
 275 public:
 276         OGGTheoraBitrate(int x, int y, OGGConfigVideo *gui);
 277         int handle_event();
 278         OGGConfigVideo *gui;
 279 };
 280
 281 class OGGTheoraKeyframeFrequency : public BC_TumbleTextBox
 282 {
 283 public:
 284         OGGTheoraKeyframeFrequency(int x, int y, OGGConfigVideo *gui);
 285         int handle_event();
 286         OGGConfigVideo *gui;
 287 };
 288
 289 class OGGTheoraKeyframeForceFrequency : public BC_TumbleTextBox
 290 {
 291 public:
 292         OGGTheoraKeyframeForceFrequency(int x, int y, OGGConfigVideo *gui);
 293         int handle_event();
 294         OGGConfigVideo *gui;
 295 };
 296
 297 class OGGTheoraSharpness : public BC_TumbleTextBox
 298 {
 299 public:
 300         OGGTheoraSharpness(int x, int y, OGGConfigVideo *gui);
 301         int handle_event();
 302         OGGConfigVideo *gui;
 303 };
 304
 305 class OGGTheoraFixedBitrate : public BC_Radial
 306 {
 307 public:
 308         OGGTheoraFixedBitrate(int x, int y, OGGConfigVideo *gui);
 309         int handle_event();
 310         OGGConfigVideo *gui;
 311 };
 312
 313 class OGGTheoraFixedQuality : public BC_Radial
 314 {
 315 public:
 316         OGGTheoraFixedQuality(int x, int y, OGGConfigVideo *gui);
 317         int handle_event();
 318         OGGConfigVideo *gui;
 319 };
 320
 321
 322
 323 class OGGConfigVideo: public BC_Window
 324 {
 325 public:
 326         OGGConfigVideo(BC_WindowBase *parent_window, Asset *asset);
 327         ~OGGConfigVideo();
 328
 329         void create_objects();
 330         int close_event();
 331
 332         OGGTheoraFixedBitrate *fixed_bitrate;
 333         OGGTheoraFixedQuality *fixed_quality;
 334         Asset *asset;
 335 private:
 336         BC_WindowBase *parent_window;
 337 };
 338
 339 #endif
 340 #endif