motion draw_vectors using VFrame draw_pixel brush
[goodguy/history.git] / 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
25 #include "edl.inc"
26 #include "filebase.h"
27 #include "file.inc"
28 #include "packagingengine.h"
29
30 #include <theora/theora.h>
31 #include <vorbis/codec.h>
32 #include <vorbis/vorbisenc.h>
33
34
35 #include <libdv/dv.h>
36
37 /* This code was aspired by ffmpeg2theora */
38 /* Special thanks for help on this code goes out to j@v2v.cc */
39
40 typedef struct
41 {
42         ogg_sync_state sync;
43         off_t file_bufpos; // position of the start of the buffer inside the file
44         off_t file_pagepos; // position of the page that will be next read
45         off_t file_pagepos_found; // position of last page that was returned (in seeking operations)
46         int wlen;
47 } sync_window_t;
48
49 typedef struct
50 {
51     ogg_page audiopage;
52     ogg_page videopage;
53
54     double audiotime;
55     double videotime;
56     ogg_int64_t audio_bytesout;
57     ogg_int64_t video_bytesout;
58
59     ogg_page og;    /* one Ogg bitstream page.  Vorbis packets are inside */
60     ogg_packet op;  /* one raw packet of data for decode */
61
62     theora_info ti;
63     theora_comment tc;
64     theora_state td;
65
66     vorbis_info vi;       /* struct that stores all the static vorbis bitstream settings */
67     vorbis_comment vc;    /* struct that stores all the user comments */
68     vorbis_dsp_state vd; /* central working state for the packet<->PCM encoder/decoder */
69     vorbis_block vb;     /* local working space for packet<->PCM encode/decode */
70
71     /* used for muxing */
72     ogg_stream_state to;    /* take physical pages, weld into a logical
73                              * stream of packets */
74     ogg_stream_state vo;    /* take physical pages, weld into a logical
75                              * stream of packets */
76
77     int apage_valid;
78     int vpage_valid;
79     unsigned char *apage;
80     unsigned char *vpage;
81     int vpage_len;
82     int apage_len;
83     int vpage_buffer_length;
84     int apage_buffer_length;
85
86
87 // stuff needed for reading only
88         sync_window_t *audiosync;
89         sync_window_t *videosync;
90
91     //to do some manual page flusing
92     int v_pkg;
93     int a_pkg;
94
95 }
96 theoraframes_info_t;
97
98 class FileOGG : public FileBase
99 {
100 friend class PackagingEngineOGG;
101 public:
102         FileOGG(Asset *asset, File *file);
103         ~FileOGG();
104
105         static void get_parameters(BC_WindowBase *parent_window,
106                 Asset *asset, BC_WindowBase* &format_window,
107                 int audio_options, int video_options, EDL *edl);
108
109         int reset_parameters_derived();
110         int open_file(int rd, int wr);
111         static int check_sig(Asset *asset);
112         int close_file();
113         int close_file_derived();
114         int64_t get_video_position();
115         int64_t get_audio_position();
116         int set_video_position(int64_t x);
117         int set_audio_position(int64_t x);
118         int colormodel_supported(int colormodel);
119         int get_best_colormodel(Asset *asset, int driver);
120         int write_samples(double **buffer, int64_t len);
121         int write_frames(VFrame ***frames, int len);
122         int read_samples(double *buffer, int64_t len);
123         int read_frame(VFrame *frame);
124
125 private:
126         int write_samples_vorbis(double **buffer, int64_t len, int e_o_s);
127         int write_frames_theora(VFrame ***frames, int len, int e_o_s);
128         void flush_ogg(int e_o_s);
129         int write_audio_page();
130         int write_video_page();
131
132         FILE *stream;
133         off_t file_length;
134
135         theoraframes_info_t *tf;
136         VFrame *temp_frame;
137         Mutex *flush_lock;
138
139         off_t filedata_begin;
140
141         int ogg_get_last_page(sync_window_t *sw, long serialno, ogg_page *og);
142         int ogg_get_prev_page(sync_window_t *sw, long serialno, ogg_page *og);
143         int ogg_get_first_page(sync_window_t *sw, long serialno, ogg_page *og);
144         int ogg_get_next_page(sync_window_t *sw, long serialno, ogg_page *og);
145         int ogg_sync_and_get_next_page(sync_window_t *sw, long serialno, ogg_page *og);
146
147         int ogg_get_page_of_sample(sync_window_t *sw, long serialno, ogg_page *og, int64_t sample);
148         int ogg_seek_to_sample(sync_window_t *sw, long serialno, int64_t sample);
149         int ogg_decode_more_samples(sync_window_t *sw, long serialno);
150
151         int ogg_get_page_of_frame(sync_window_t *sw, long serialno, ogg_page *og, int64_t frame);
152         int ogg_seek_to_keyframe(sync_window_t *sw, long serialno, int64_t frame, int64_t *position);
153         int ogg_seek_to_databegin(sync_window_t *sw, long serialno);
154
155
156         int64_t start_sample; // first and last sample inside this file
157         int64_t last_sample;
158         int64_t start_frame; // first and last frame inside this file
159         int64_t last_frame;
160
161
162         int64_t ogg_sample_position;  // what will be the next sample taken from vorbis decoder
163         int64_t next_sample_position; // what is the next sample read_samples must deliver
164
165         int move_history(int from, int to, int len);
166
167         float **pcm_history;
168 #ifndef HISTORY_MAX
169 #define HISTORY_MAX 0x100000
170 #endif
171         int64_t history_start;
172         int64_t history_size;
173
174         int theora_cmodel;
175         int64_t ogg_frame_position;    // LAST decoded frame position
176         int64_t next_frame_position;   // what is the next sample read_frames must deliver
177         char theora_keyframe_granule_shift;
178         int final_write;
179 };
180
181 class OGGConfigAudio;
182 class OGGConfigVideo;
183
184 class OGGVorbisFixedBitrate : public BC_Radial
185 {
186 public:
187         OGGVorbisFixedBitrate(int x, int y, OGGConfigAudio *gui);
188         int handle_event();
189         OGGConfigAudio *gui;
190 };
191
192 class OGGVorbisVariableBitrate : public BC_Radial
193 {
194 public:
195         OGGVorbisVariableBitrate(int x, int y, OGGConfigAudio *gui);
196         int handle_event();
197         OGGConfigAudio *gui;
198 };
199
200 class OGGVorbisMinBitrate : public BC_TextBox
201 {
202 public:
203         OGGVorbisMinBitrate(int x,
204                 int y,
205                 OGGConfigAudio *gui,
206                 char *text);
207         int handle_event();
208         OGGConfigAudio *gui;
209 };
210
211 class OGGVorbisMaxBitrate : public BC_TextBox
212 {
213 public:
214         OGGVorbisMaxBitrate(int x,
215                 int y,
216                 OGGConfigAudio *gui,
217                 char *text);
218         int handle_event();
219         OGGConfigAudio *gui;
220 };
221
222 class OGGVorbisAvgBitrate : public BC_TextBox
223 {
224 public:
225         OGGVorbisAvgBitrate(int x,
226                 int y,
227                 OGGConfigAudio *gui,
228                 char *text);
229         int handle_event();
230         OGGConfigAudio *gui;
231 };
232
233
234 class OGGConfigAudio: public BC_Window
235 {
236 public:
237         OGGConfigAudio(BC_WindowBase *parent_window, Asset *asset);
238         ~OGGConfigAudio();
239
240         void create_objects();
241         int close_event();
242
243         Asset *asset;
244         OGGVorbisFixedBitrate *fixed_bitrate;
245         OGGVorbisVariableBitrate *variable_bitrate;
246 private:
247         BC_WindowBase *parent_window;
248         char string[BCTEXTLEN];
249 };
250
251
252 class OGGTheoraBitrate : public BC_TextBox
253 {
254 public:
255         OGGTheoraBitrate(int x, int y, OGGConfigVideo *gui);
256         int handle_event();
257         OGGConfigVideo *gui;
258 };
259
260 class OGGTheoraKeyframeFrequency : public BC_TumbleTextBox
261 {
262 public:
263         OGGTheoraKeyframeFrequency(int x, int y, OGGConfigVideo *gui);
264         int handle_event();
265         OGGConfigVideo *gui;
266 };
267
268 class OGGTheoraKeyframeForceFrequency : public BC_TumbleTextBox
269 {
270 public:
271         OGGTheoraKeyframeForceFrequency(int x, int y, OGGConfigVideo *gui);
272         int handle_event();
273         OGGConfigVideo *gui;
274 };
275
276 class OGGTheoraSharpness : public BC_TumbleTextBox
277 {
278 public:
279         OGGTheoraSharpness(int x, int y, OGGConfigVideo *gui);
280         int handle_event();
281         OGGConfigVideo *gui;
282 };
283
284 class OGGTheoraFixedBitrate : public BC_Radial
285 {
286 public:
287         OGGTheoraFixedBitrate(int x, int y, OGGConfigVideo *gui);
288         int handle_event();
289         OGGConfigVideo *gui;
290 };
291
292 class OGGTheoraFixedQuality : public BC_Radial
293 {
294 public:
295         OGGTheoraFixedQuality(int x, int y, OGGConfigVideo *gui);
296         int handle_event();
297         OGGConfigVideo *gui;
298 };
299
300
301
302 class OGGConfigVideo: public BC_Window
303 {
304 public:
305         OGGConfigVideo(BC_WindowBase *parent_window, Asset *asset);
306         ~OGGConfigVideo();
307
308         void create_objects();
309         int close_event();
310
311         OGGTheoraFixedBitrate *fixed_bitrate;
312         OGGTheoraFixedQuality *fixed_quality;
313         Asset *asset;
314 private:
315         BC_WindowBase *parent_window;
316 };
317
318 class PackagingEngineOGG : public PackagingEngine
319 {
320 public:
321         PackagingEngineOGG();
322         ~PackagingEngineOGG();
323         int create_packages_single_farm(
324                 EDL *edl,
325                 Preferences *preferences,
326                 Asset *default_asset,
327                 double total_start,
328                 double total_end);
329         RenderPackage* get_package_single_farm(double frames_per_second,
330                 int client_number,
331                 int use_local_rate);
332         int64_t get_progress_max();
333         void get_package_paths(ArrayList<char*> *path_list);
334         int packages_are_done();
335
336 private:
337         EDL *edl;
338
339         RenderPackage **packages;
340         int total_packages;
341         double video_package_len;    // Target length of a single package
342
343         Asset *default_asset;
344         Preferences *preferences;
345         int current_package;
346         double total_start;
347         double total_end;
348         int64_t audio_position;
349         int64_t video_position;
350         int64_t audio_start;
351         int64_t video_start;
352         int64_t audio_end;
353         int64_t video_end;
354
355 };
356
357
358 #endif