cinelerra-5.1/cinelerra/fileflac.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 <errno.h>
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <unistd.h>
  26
  27 #include "FLAC/stream_decoder.h"
  28 #include "FLAC/stream_encoder.h"
  29
  30 #define IS_FILEFLAC
  31 #include "asset.h"
  32 #include "bcsignals.h"
  33 #include "bitspopup.h"
  34 #include "byteorder.h"
  35 #include "clip.h"
  36 #include "file.h"
  37 #include "fileflac.h"
  38 #include "filesystem.h"
  39 #include "guicast.h"
  40 #include "language.h"
  41 #include "mwindow.inc"
  42
  43
  44 FileFLAC::FileFLAC(Asset *asset, File *file)
  45  : FileBase(asset, file)
  46 {
  47         reset_parameters();
  48         if(asset->format == FILE_UNKNOWN) asset->format = FILE_FLAC;
  49         asset->byte_order = 0;
  50 }
  51
  52 FileFLAC::~FileFLAC()
  53 {
  54         close_file();
  55 }
  56
  57 void FileFLAC::get_parameters(BC_WindowBase *parent_window,
  58         Asset *asset, BC_WindowBase* &format_window,
  59         int audio_options, int video_options, EDL *edl)
  60 {
  61         if(audio_options)
  62         {
  63                 FLACConfigAudio *window = new FLACConfigAudio(parent_window, asset);
  64                 format_window = window;
  65                 window->create_objects();
  66                 window->run_window();
  67                 delete window;
  68         }
  69 }
  70
  71
  72
  73 int FileFLAC::check_sig(Asset *asset, char *test)
  74 {
  75         return test[0]=='f' && test[1]=='L' && test[2]=='a' && test[3]=='C' ? 1 : 0;
  76 }
  77
  78 int FileFLAC::reset_parameters_derived()
  79 {
  80         pcm_history = 0;
  81         flac_decode = 0;
  82         flac_encode = 0;
  83         temp_buffer = 0;
  84         temp_allocated = 0;
  85         temp_channels = 0;
  86         initialized = 0;
  87         is_reading = 0;
  88         return 0;
  89 }
  90
  91
  92 static FLAC__StreamDecoderWriteStatus write_callback(
  93         const FLAC__StreamDecoder *decoder,
  94         const FLAC__Frame *frame,
  95         const FLAC__int32 * const buffer[],
  96         void *client_data)
  97 {
  98         FileFLAC *ptr = (FileFLAC*)client_data;
  99
 100         if(!ptr->initialized)
 101         {
 102                 ptr->initialized = 1;
 103                 ptr->asset->audio_data = 1;
 104                 ptr->asset->channels = FLAC__stream_decoder_get_channels(ptr->flac_decode);
 105                 ptr->asset->bits = FLAC__stream_decoder_get_bits_per_sample(ptr->flac_decode);
 106                 ptr->asset->sample_rate = FLAC__stream_decoder_get_sample_rate(ptr->flac_decode);
 107                 ptr->asset->audio_length = FLAC__stream_decoder_get_total_samples(ptr->flac_decode);
 108                 return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
 109         }
 110         else
 111         if(ptr->is_reading)
 112         {
 113                 int fragment = FLAC__stream_decoder_get_blocksize(ptr->flac_decode);
 114                 ptr->samples_read += fragment;
 115 //printf("write_callback 2 %d\n", fragment);
 116
 117                 if(ptr->temp_allocated < fragment)
 118                 {
 119                         if(ptr->temp_buffer)
 120                         {
 121                                 for(int i = 0; i < ptr->temp_channels; i++)
 122                                 {
 123                                         delete [] ptr->temp_buffer[i];
 124                                 }
 125                                 delete [] ptr->temp_buffer;
 126                         }
 127
 128                         ptr->temp_channels = ptr->asset->channels;
 129                         ptr->temp_buffer = new int32_t*[ptr->temp_channels];
 130                         for(int i = 0; i < ptr->temp_channels; i++)
 131                         {
 132                                 ptr->temp_buffer[i] = new int32_t[fragment];
 133                         }
 134                         ptr->temp_allocated = fragment;
 135                 }
 136
 137                 float audio_max = (float)0x7fff;
 138                 if(FLAC__stream_decoder_get_bits_per_sample(ptr->flac_decode) == 24)
 139                         audio_max = (float)0x7fffff;
 140
 141                 int nchannels = FLAC__stream_decoder_get_channels(ptr->flac_decode);
 142                 if( nchannels > ptr->temp_channels ) nchannels = ptr->temp_channels;
 143                 for(int j = 0; j < nchannels; j++)
 144                 {
 145                         float *dst = (float*)ptr->temp_buffer[j];
 146                         int32_t *src = (int32_t*)buffer[j];
 147                         for(int i = 0; i < fragment; i++)
 148                         {
 149                                 *dst++ = (float)*src++ / audio_max;
 150                         }
 151                 }
 152
 153                 ptr->append_history((float**)ptr->temp_buffer, fragment);
 154
 155 //printf("write_callback 3\n");
 156                 return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
 157         }
 158
 159         return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
 160 }
 161
 162
 163 static void metadata_callback(const FLAC__StreamDecoder *decoder,
 164         const FLAC__StreamMetadata *metadata,
 165         void *client_data)
 166 {
 167 //      printf("metadata_callback\n");
 168 }
 169
 170 static void error_callback(const FLAC__StreamDecoder *decoder,
 171         FLAC__StreamDecoderErrorStatus status,
 172         void *client_data)
 173 {
 174 //      printf("error_callback\n");
 175 }
 176
 177
 178
 179
 180
 181 int FileFLAC::open_file(int rd, int wr)
 182 {
 183         int result = 0;
 184
 185         if(rd)
 186         {
 187                 file_bytes = FileSystem::get_size(asset->path);
 188                 flac_decode = FLAC__stream_decoder_new();
 189                 FLAC__stream_decoder_init_file(flac_decode,
 190                         asset->path,
 191                         write_callback,
 192                         metadata_callback,
 193                         error_callback,
 194                         this);
 195
 196                 initialized = 0;
 197                 while(!initialized)
 198                 {
 199                         if(!FLAC__stream_decoder_process_single(flac_decode)) break;
 200                 }
 201
 202                 if(!initialized)
 203                         result = 1;
 204                 else
 205                 {
 206                         FLAC__stream_decoder_seek_absolute(flac_decode, 0);
 207                 }
 208         }
 209
 210         if(wr)
 211         {
 212                 flac_encode = FLAC__stream_encoder_new();
 213                 FLAC__stream_encoder_set_channels(flac_encode, asset->channels);
 214                 FLAC__stream_encoder_set_bits_per_sample(flac_encode, asset->bits);
 215                 FLAC__stream_encoder_set_sample_rate(flac_encode, asset->sample_rate);
 216                 FLAC__stream_encoder_init_file(flac_encode, asset->path, 0, 0);
 217         }
 218
 219         return result;
 220 }
 221
 222
 223 int FileFLAC::close_file_derived()
 224 {
 225         if(flac_decode)
 226         {
 227                 FLAC__stream_decoder_finish(flac_decode);
 228                 FLAC__stream_decoder_delete(flac_decode);
 229         }
 230
 231         if(flac_encode)
 232         {
 233                 FLAC__stream_encoder_finish(flac_encode);
 234                 FLAC__stream_encoder_delete(flac_encode);
 235         }
 236
 237
 238         if(temp_buffer)
 239         {
 240                 for(int i = 0; i < temp_channels; i++)
 241                         delete [] temp_buffer[i];
 242                 delete [] temp_buffer;
 243         }
 244         return 0;
 245 }
 246
 247
 248 int FileFLAC::write_samples(double **buffer, int64_t len)
 249 {
 250 // Create temporary buffer of samples
 251         if(temp_allocated < len)
 252         {
 253                 if(temp_buffer)
 254                 {
 255                         for(int i = 0; i < asset->channels; i++)
 256                         {
 257                                 delete [] temp_buffer[i];
 258                         }
 259                         delete [] temp_buffer;
 260                 }
 261
 262                 temp_buffer = new int32_t*[asset->channels];
 263                 for(int i = 0; i < asset->channels; i++)
 264                 {
 265                         temp_buffer[i] = new int32_t[len];
 266                 }
 267                 temp_allocated = len;
 268         }
 269
 270         float audio_max = (float)0x7fff;
 271
 272         switch(asset->bits)
 273         {
 274                 case 24:
 275                         audio_max = (float)0x7fffff;
 276                         break;
 277         }
 278
 279         for(int i = 0; i < asset->channels; i++)
 280         {
 281                 int32_t *dst = temp_buffer[i];
 282                 double *src = buffer[i];
 283                 for(int j = 0; j < len; j++)
 284                 {
 285                         double sample = *src++ * audio_max;
 286                         CLAMP(sample, -audio_max, audio_max);
 287                         *dst++ = (int32_t)sample;
 288                 }
 289         }
 290
 291         int result = FLAC__stream_encoder_process(flac_encode,
 292                 temp_buffer,
 293                 len);
 294
 295         return !result;
 296 }
 297
 298 int FileFLAC::read_samples(double *buffer, int64_t len)
 299 {
 300         int result = 0;
 301         update_pcm_history(len);
 302
 303         if(decode_end != decode_start)
 304         {
 305                 FLAC__stream_decoder_seek_absolute(flac_decode, decode_start);
 306                 decode_end = decode_start;
 307         }
 308
 309         samples_read = 0;
 310         is_reading = 1;
 311
 312         while(samples_read < decode_len)
 313         {
 314                 FLAC__uint64 position;
 315                 FLAC__stream_decoder_get_decode_position(flac_decode, &position);
 316 // EOF
 317                 if((int64_t)position >= file_bytes) break;
 318                 if(!FLAC__stream_decoder_process_single(flac_decode)) {
 319                         result = 1;
 320                         break;
 321                 }
 322         }
 323         is_reading = 0;
 324
 325         if(!result)
 326         {
 327                 read_history(buffer,
 328                         file->current_sample,
 329                         file->current_channel,
 330                         len);
 331         }
 332
 333 //printf("FileFLAC::read_samples 10\n");
 334         return result;
 335 }
 336
 337
 338 FLACConfigAudio::FLACConfigAudio(BC_WindowBase *parent_window,
 339         Asset *asset)
 340  : BC_Window(_(PROGRAM_NAME ": Audio Compression"),
 341         parent_window->get_abs_cursor_x(1), parent_window->get_abs_cursor_y(1),
 342         xS(350), yS(170), -1, -1, 0, 0, 1)
 343 {
 344         this->parent_window = parent_window;
 345         this->asset = asset;
 346 }
 347
 348 FLACConfigAudio::~FLACConfigAudio()
 349 {
 350 }
 351
 352 void FLACConfigAudio::create_objects()
 353 {
 354         int x = xS(10), y = yS(10);
 355         lock_window("FLACConfigAudio::create_objects");
 356         bits_popup = new BitsPopup(this, x, y, &asset->bits, 0, 0, 0, 0, 0);
 357         bits_popup->create_objects();
 358
 359         add_subwindow(new BC_OKButton(this));
 360         show_window(1);
 361         unlock_window();
 362 }
 363
 364 int FLACConfigAudio::close_event()
 365 {
 366         set_done(0);
 367         return 1;
 368 }
 369