Credit Andrew fix Alpha Bug of BT558 + small cleanup
[goodguy/cinelerra.git] / cinelerra-5.1 / cinelerra / filebase.C
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 #include "asset.h"
23 #include "assets.h"
24 #include "byteorder.h"
25 #include "bccmodels.h"
26 #include "file.h"
27 #include "filebase.h"
28 #include "overlayframe.h"
29 #include "sizes.h"
30
31 #include <stdlib.h>
32 #include <string.h>
33
34 FileBase::FileBase(Asset *asset, File *file)
35 {
36         this->file = file;
37         this->asset = asset;
38         internal_byte_order = get_byte_order();
39         reset_parameters();
40 }
41
42 FileBase::~FileBase()
43 {
44         close_file();
45 }
46
47 int FileBase::close_file()
48 {
49         delete [] row_pointers_in;   row_pointers_in = 0;
50         delete [] row_pointers_out;  row_pointers_out = 0;
51         delete [] float_buffer;      float_buffer = 0;
52
53         if( pcm_history ) {
54                 for(int i = 0; i < history_channels; i++)
55                         delete [] pcm_history[i];
56                 delete [] pcm_history;  pcm_history = 0;
57         }
58
59         close_file_derived();
60         reset_parameters();
61         return 0;
62 }
63
64 void FileBase::update_pcm_history(int64_t len)
65 {
66         decode_start = 0;
67         decode_len = 0;
68
69         if( !pcm_history ) {
70                 history_channels = asset->channels;
71                 pcm_history = new double*[history_channels];
72                 for(int i = 0; i < history_channels; i++)
73                         pcm_history[i] = new double[HISTORY_MAX];
74                 history_start = 0;
75                 history_size = 0;
76                 history_allocated = HISTORY_MAX;
77         }
78
79
80 //printf("FileBase::update_pcm_history current_sample=%jd history_start=%jd history_size=%jd\n",
81 //file->current_sample,
82 //history_start,
83 //history_size);
84 // Restart history.  Don't bother shifting history back.
85         if(file->current_sample < history_start ||
86                 file->current_sample > history_start + history_size)
87         {
88                 history_size = 0;
89                 history_start = file->current_sample;
90                 decode_start = file->current_sample;
91                 decode_len = len;
92         }
93         else
94 // Shift history forward to make room for new samples
95         if(file->current_sample > history_start + HISTORY_MAX)
96         {
97                 int diff = file->current_sample - (history_start + HISTORY_MAX);
98                 for(int i = 0; i < asset->channels; i++)
99                 {
100                         double *temp = pcm_history[i];
101                         memcpy(temp, temp + diff, (history_size - diff) * sizeof(double));
102                 }
103
104                 history_start += diff;
105                 history_size -= diff;
106
107 // Decode more data
108                 decode_start = history_start + history_size;
109                 decode_len = file->current_sample + len - (history_start + history_size);
110         }
111         else
112 // Starting somewhere in the buffer
113         {
114                 decode_start = history_start + history_size;
115                 decode_len = file->current_sample + len - (history_start + history_size);
116         }
117 }
118
119 void FileBase::append_history(float **new_data, int len)
120 {
121         allocate_history(len);
122
123         for(int i = 0; i < history_channels; i++)
124         {
125                 double *output = pcm_history[i] + history_size;
126                 float *input = new_data[i];
127                 for(int j = 0; j < len; j++)
128                         *output++ = *input++;
129         }
130
131         history_size += len;
132         decode_end += len;
133 }
134
135 void FileBase::append_history(short *new_data, int len)
136 {
137         allocate_history(len);
138
139         for(int i = 0; i < history_channels; i++)
140         {
141                 double *output = pcm_history[i] + history_size;
142                 short *input = new_data + i;
143                 for(int j = 0; j < len; j++)
144                 {
145                         *output++ = (double)*input / 32768;
146                         input += history_channels;
147                 }
148         }
149
150         history_size += len;
151         decode_end += len;
152 }
153
154 void FileBase::read_history(double *dst,
155         int64_t start_sample,
156         int channel,
157         int64_t len)
158 {
159         if(start_sample - history_start + len > history_size)
160                 len = history_size - (start_sample - history_start);
161 //printf("FileBase::read_history start_sample=%jd history_start=%jd history_size=%jd len=%jd\n",
162 //start_sample, history_start, history_size, len);
163         double *input = pcm_history[channel] + start_sample - history_start;
164         for(int i = 0; i < len; i++)
165         {
166                 *dst++ = *input++;
167         }
168 }
169
170 void FileBase::allocate_history(int len)
171 {
172         if(history_size + len > history_allocated)
173         {
174                 double **temp = new double*[history_channels];
175
176                 for(int i = 0; i < history_channels; i++)
177                 {
178                         temp[i] = new double[history_size + len];
179                         memcpy(temp[i], pcm_history[i], history_size * sizeof(double));
180                         delete [] pcm_history[i];
181                 }
182
183                 delete [] pcm_history;
184                 pcm_history = temp;
185                 history_allocated = history_size + len;
186         }
187 }
188
189 int64_t FileBase::get_history_sample()
190 {
191         return history_start + history_size;
192 }
193
194 int FileBase::set_dither()
195 {
196         dither = 1;
197         return 0;
198 }
199
200 void FileBase::reset_parameters()
201 {
202         dither = 0;
203         float_buffer = 0;
204         row_pointers_in = 0;
205         row_pointers_out = 0;
206         prev_buffer_position = -1;
207         prev_frame_position = -1;
208         prev_len = 0;
209         prev_bytes = 0;
210         prev_track = -1;
211         prev_layer = -1;
212         ulawtofloat_table = 0;
213         floattoulaw_table = 0;
214         rd = wr = 0;
215         pcm_history = 0;
216         history_start = 0;
217         history_size = 0;
218         history_allocated = 0;
219         history_channels = 0;
220         decode_end = 0;
221
222         delete_ulaw_tables();
223         reset_parameters_derived();
224 }
225
226 void FileBase::get_mode(char *mode, int rd, int wr)
227 {
228         if(rd && !wr) sprintf(mode, "rb");
229         else if(!rd && wr) sprintf(mode, "wb");
230         else if(rd && wr) {
231                 int exists = 0;
232                 FILE *stream = fopen(asset->path, "rb");
233                 if( stream ) {
234                         exists = 1;
235                         fclose(stream);
236                 }
237                 sprintf(mode, exists ? "rb+" : "wb+");
238         }
239 }
240
241 int FileBase::get_best_colormodel(int driver, int vstream)
242 {
243         return File::get_best_colormodel(asset, driver);
244 }
245
246
247 // ======================================= audio codecs
248
249 int FileBase::get_video_buffer(unsigned char **buffer, int depth)
250 {
251 // get a raw video buffer for writing or compression by a library
252         if(!*buffer)
253         {
254 // Video compression is entirely done in the library.
255                 int64_t bytes = asset->width * asset->height * depth;
256                 *buffer = new unsigned char[bytes];
257         }
258         return 0;
259 }
260
261 int FileBase::get_row_pointers(unsigned char *buffer, unsigned char ***pointers, int depth)
262 {
263 // This might be fooled if a new VFrame is created at the same address with a different height.
264         if(*pointers && (*pointers)[0] != &buffer[0])
265         {
266                 delete [] *pointers;
267                 *pointers = 0;
268         }
269
270         if(!*pointers)
271         {
272                 *pointers = new unsigned char*[asset->height];
273                 for(int i = 0; i < asset->height; i++)
274                 {
275                         (*pointers)[i] = &buffer[i * asset->width * depth / 8];
276                 }
277         }
278         return 0;
279 }
280
281 int FileBase::match4(const char *in, const char *out)
282 {
283         return  in[0] == out[0] && in[1] == out[1] &&
284                 in[2] == out[2] && in[3] == out[3] ? 1 : 0;
285 }
286
287 int FileBase::search_render_strategies(ArrayList<int>* render_strategies, int render_strategy)
288 {
289         for(int i = 0; i < render_strategies->total; i++)
290                 if(render_strategies->values[i] == render_strategy) return 1;
291         return 0;
292 }
293
294
295 int64_t FileBase::base_memory_usage()
296 {
297 //printf("FileBase::base_memory_usage %d\n", __LINE__);
298         return !pcm_history ? 0 :
299                 history_allocated * history_channels * sizeof(double);
300 }
301