cinelerra-5.1/cinelerra/asset.C

   1 /*
   2  * CINELERRA
   3  * Copyright (C) 2010 Adam Williams <broadcast at earthling dot net>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18  *
  19  */
  20
  21
  22 #include "asset.h"
  23 #include "assets.h"
  24 #include "awindowgui.h"
  25 #include "bchash.h"
  26 #include "bcsignals.h"
  27 #include "clip.h"
  28 #include "cstrdup.h"
  29 #include "edl.h"
  30 #include "file.h"
  31 #include "filesystem.h"
  32 #include "filexml.h"
  33 #include "indexstate.h"
  34 #include "interlacemodes.h"
  35
  36 #include <stdio.h>
  37 #include <string.h>
  38
  39
  40 Asset::Asset()
  41  : Indexable(1), ListItem<Asset>()
  42 {
  43         init_values();
  44 }
  45
  46 Asset::Asset(Asset &asset)
  47  : Indexable(1), ListItem<Asset>()
  48 {
  49         init_values();
  50         this->copy_from(&asset, 1);
  51 }
  52
  53 Asset::Asset(const char *path)
  54  : Indexable(1), ListItem<Asset>()
  55 {
  56         init_values();
  57         strcpy(this->path, path);
  58 }
  59
  60 Asset::Asset(const int plugin_type, const char *plugin_title)
  61  : Indexable(1), ListItem<Asset>()
  62 {
  63         init_values();
  64 }
  65
  66 Asset::~Asset()
  67 {
  68 }
  69
  70 int Asset::init_values()
  71 {
  72         path[0] = 0;
  73 // Has to be unknown for file probing to succeed
  74         format = FILE_UNKNOWN;
  75         bits = 0;
  76         byte_order = 0;
  77         signed_ = 0;
  78         header = 0;
  79         dither = 0;
  80         reset_audio();
  81         reset_video();
  82
  83         strcpy(vcodec, "");
  84         strcpy(acodec, "");
  85
  86         strcpy(fformat,"mp4");
  87         ff_format_options[0] = 0;
  88         ff_audio_options[0] = 0;
  89         ff_sample_format[0] = 0;
  90         ff_audio_bitrate = 0;
  91         ff_audio_quality = -1;
  92         ff_video_options[0] = 0;
  93         ff_pixel_format[0] = 0;
  94         ff_video_bitrate = 0;
  95         ff_video_quality = -1;
  96         ff_color_space = -1;
  97         ff_color_range = -1;
  98
  99         jpeg_quality = 80;
 100         aspect_ratio = -1;
 101         interlace_mode = ILACE_MODE_UNDETECTED;
 102
 103         mp3_bitrate = 224;
 104         ampeg_bitrate = 256;
 105         ampeg_derivative = 3;
 106
 107         vorbis_vbr = 0;
 108         vorbis_min_bitrate = -1;
 109         vorbis_bitrate = 128000;
 110         vorbis_max_bitrate = -1;
 111
 112         theora_fix_bitrate = 1;
 113         theora_bitrate = 860000;
 114         theora_quality = 16;
 115         theora_sharpness = 2;
 116         theora_keyframe_frequency = 64;
 117         theora_keyframe_force_frequency = 64;
 118
 119 // mpeg parameters
 120         vmpeg_iframe_distance = 45;
 121         vmpeg_pframe_distance = 0;
 122         vmpeg_progressive = 0;
 123         vmpeg_denoise = 1;
 124         vmpeg_bitrate = 1000000;
 125         vmpeg_derivative = 1;
 126         vmpeg_quantization = 15;
 127         vmpeg_cmodel = BC_YUV420P;
 128         vmpeg_fix_bitrate = 0;
 129         vmpeg_seq_codes = 0;
 130         vmpeg_preset = 0;
 131         vmpeg_field_order = 0;
 132
 133         ac3_bitrate = 128;
 134
 135         png_use_alpha = 0;
 136         png_depth = 8;
 137         png_compression = 0;
 138
 139         exr_use_alpha = 0;
 140         exr_compression = 0;
 141
 142         tiff_cmodel = 0;
 143         tiff_compression = 0;
 144         mov_sphere = 0;
 145         jpeg_sphere = 0;
 146
 147         use_header = 1;
 148         id = EDL::next_id();
 149
 150         return 0;
 151 }
 152
 153 void Asset::reset_audio()
 154 {
 155         audio_data = 0;
 156         channels = 0;
 157         sample_rate = 0;
 158         audio_length = 0;
 159 }
 160
 161 void Asset::reset_video()
 162 {
 163         video_data = 0;
 164         layers = 0;
 165         actual_width = width = 0;
 166         actual_height = height = 0;
 167         proxy_scale = 0; // not a proxy
 168         proxy_edl = 0; // not proxy from edl
 169         video_length = 0;
 170         timecode = -2; // unknown
 171         single_frame = 0;
 172         vmpeg_cmodel = BC_YUV420P;
 173         frame_rate = 0;
 174         program = -1;
 175 }
 176
 177
 178 void Asset::boundaries()
 179 {
 180 //printf("Asset::boundaries %d %d %f\n", __LINE__, sample_rate, frame_rate);
 181 // sample_rate & frame_rate are user defined
 182 //      CLAMP(sample_rate, 1, 1000000);
 183 //      CLAMP(frame_rate, 0.001, 1000000);
 184         CLAMP(channels, 0, 16);
 185         CLAMP(width, 0, 10000);
 186         CLAMP(height, 0, 10000);
 187 //printf("Asset::boundaries %d %d %f\n", __LINE__, sample_rate, frame_rate);
 188 }
 189
 190 void Asset::copy_from(Asset *asset, int do_index)
 191 {
 192         copy_location(asset);
 193         copy_format(asset, do_index);
 194 }
 195
 196 void Asset::copy_location(Asset *asset)
 197 {
 198         strcpy(path, asset->path);
 199         folder_no = asset->folder_no;
 200 }
 201
 202 void Asset::copy_format(Asset *asset, int do_index)
 203 {
 204         if(do_index) update_index(asset);
 205
 206         audio_data = asset->audio_data;
 207         format = asset->format;
 208         strcpy(fformat, asset->fformat);
 209         strcpy(ff_format_options, asset->ff_format_options);
 210         channels = asset->channels;
 211         sample_rate = asset->sample_rate;
 212         bits = asset->bits;
 213         byte_order = asset->byte_order;
 214         signed_ = asset->signed_;
 215         header = asset->header;
 216         dither = asset->dither;
 217         mp3_bitrate = asset->mp3_bitrate;
 218         use_header = asset->use_header;
 219         aspect_ratio = asset->aspect_ratio;
 220         interlace_mode = asset->interlace_mode;
 221         timecode = asset->timecode;
 222
 223         video_data = asset->video_data;
 224         layers = asset->layers;
 225         program = asset->program;
 226         frame_rate = asset->frame_rate;
 227         width = asset->width;
 228         height = asset->height;
 229         actual_width = asset->actual_width;
 230         actual_height = asset->actual_height;
 231         proxy_scale = asset->proxy_scale;
 232         proxy_edl = asset->proxy_edl;
 233         strcpy(vcodec, asset->vcodec);
 234         strcpy(acodec, asset->acodec);
 235
 236         strcpy(ff_audio_options, asset->ff_audio_options);
 237         strcpy(ff_sample_format, asset->ff_sample_format);
 238         ff_audio_bitrate = asset->ff_audio_bitrate;
 239         ff_audio_quality = asset->ff_audio_quality;
 240         strcpy(ff_video_options, asset->ff_video_options);
 241         strcpy(ff_pixel_format, asset->ff_pixel_format);
 242         ff_video_bitrate = asset->ff_video_bitrate;
 243         ff_video_quality = asset->ff_video_quality;
 244         ff_color_space = asset->ff_color_space;
 245         ff_color_range = asset->ff_color_range;
 246
 247         this->audio_length = asset->audio_length;
 248         this->video_length = asset->video_length;
 249         this->single_frame = asset->single_frame;
 250
 251         ampeg_bitrate = asset->ampeg_bitrate;
 252         ampeg_derivative = asset->ampeg_derivative;
 253
 254
 255         vorbis_vbr = asset->vorbis_vbr;
 256         vorbis_min_bitrate = asset->vorbis_min_bitrate;
 257         vorbis_bitrate = asset->vorbis_bitrate;
 258         vorbis_max_bitrate = asset->vorbis_max_bitrate;
 259
 260
 261         theora_fix_bitrate = asset->theora_fix_bitrate;
 262         theora_bitrate = asset->theora_bitrate;
 263         theora_quality = asset->theora_quality;
 264         theora_sharpness = asset->theora_sharpness;
 265         theora_keyframe_frequency = asset->theora_keyframe_frequency;
 266         theora_keyframe_force_frequency = asset->theora_keyframe_frequency;
 267
 268
 269         jpeg_quality = asset->jpeg_quality;
 270
 271 // mpeg parameters
 272         vmpeg_iframe_distance = asset->vmpeg_iframe_distance;
 273         vmpeg_pframe_distance = asset->vmpeg_pframe_distance;
 274         vmpeg_progressive = asset->vmpeg_progressive;
 275         vmpeg_denoise = asset->vmpeg_denoise;
 276         vmpeg_bitrate = asset->vmpeg_bitrate;
 277         vmpeg_derivative = asset->vmpeg_derivative;
 278         vmpeg_quantization = asset->vmpeg_quantization;
 279         vmpeg_cmodel = asset->vmpeg_cmodel;
 280         vmpeg_fix_bitrate = asset->vmpeg_fix_bitrate;
 281         vmpeg_seq_codes = asset->vmpeg_seq_codes;
 282         vmpeg_preset = asset->vmpeg_preset;
 283         vmpeg_field_order = asset->vmpeg_field_order;
 284
 285         ac3_bitrate = asset->ac3_bitrate;
 286
 287         png_use_alpha = asset->png_use_alpha;
 288         png_depth = asset->png_depth;
 289         png_compression = asset->png_compression;
 290         exr_use_alpha = asset->exr_use_alpha;
 291         exr_compression = asset->exr_compression;
 292
 293         tiff_cmodel = asset->tiff_cmodel;
 294         tiff_compression = asset->tiff_compression;
 295
 296         mov_sphere = asset->mov_sphere;
 297         jpeg_sphere = asset->jpeg_sphere;
 298 }
 299
 300 int64_t Asset::get_index_offset(int channel)
 301 {
 302         return index_state->get_index_offset(channel);
 303 }
 304
 305 int64_t Asset::get_index_size(int channel)
 306 {
 307         return index_state->get_index_size(channel);
 308 }
 309
 310
 311 char* Asset::get_compression_text(int audio, int video)
 312 {
 313         if(audio) {
 314                 switch(format) {
 315                 case FILE_FFMPEG:
 316                         if( acodec[0] ) return acodec;
 317                         break;
 318                 }
 319         }
 320         if(video) {
 321                 switch(format) {
 322                 case FILE_FFMPEG:
 323                         if( vcodec[0] ) return vcodec;
 324                         break;
 325                 }
 326         }
 327         return 0;
 328 }
 329
 330 int Asset::equivalent(Asset &asset, int test_audio, int test_video, EDL *edl)
 331 {
 332         int result = format == asset.format ? 1 : 0;
 333         if( result && strcmp(asset.path, path) ) {
 334                 char *out_path = edl ? FileSystem::basepath(edl->path) : 0;
 335                 char *sp = out_path ? strrchr(out_path,'/') : 0;
 336                 if( sp ) *++sp = 0;
 337                 char *apath = FileSystem::basepath(asset.path);
 338                 char *tpath = FileSystem::basepath(this->path);
 339                 if( out_path ) {
 340                         if( *apath != '/' ) {
 341                                 char *cp = cstrcat(2, out_path, apath);
 342                                 delete [] apath;  apath = FileSystem::basepath(cp);
 343                                 delete [] cp;
 344                         }
 345                         if( *tpath != '/' ) {
 346                                 char *cp = cstrcat(2, out_path, tpath);
 347                                 delete [] tpath;  tpath = FileSystem::basepath(cp);
 348                                 delete [] cp;
 349                         }
 350                 }
 351                 if( strcmp(apath, tpath) ) result = 0;
 352                 delete [] apath;
 353                 delete [] tpath;
 354                 delete [] out_path;
 355         }
 356
 357         if( result && format == FILE_FFMPEG &&
 358             (strcmp(fformat, asset.fformat) ||
 359              strcmp(ff_format_options, asset.ff_format_options)) )
 360                 result = 0;
 361
 362         if(test_audio && result) {
 363                 result = (channels == asset.channels &&
 364                         sample_rate == asset.sample_rate &&
 365                         bits == asset.bits &&
 366                         byte_order == asset.byte_order &&
 367                         signed_ == asset.signed_ &&
 368                         header == asset.header &&
 369                         dither == asset.dither &&
 370                         !strcmp(acodec, asset.acodec));
 371                 if( result && format == FILE_FFMPEG )
 372                         result = !strcmp(ff_audio_options, asset.ff_audio_options) &&
 373                                 !strcmp(ff_sample_format, asset.ff_sample_format) &&
 374                                 ff_audio_bitrate == asset.ff_audio_bitrate &&
 375                                 ff_audio_quality == asset.ff_audio_quality;
 376         }
 377
 378
 379         if(test_video && result) {
 380                 result = (layers == asset.layers &&
 381                         program == asset.program &&
 382                         frame_rate == asset.frame_rate &&
 383                         asset.interlace_mode == interlace_mode &&
 384                         asset.timecode == timecode &&
 385                         width == asset.width &&
 386                         height == asset.height &&
 387                         !strcmp(vcodec, asset.vcodec) &&
 388                         mov_sphere == asset.mov_sphere &&
 389                         jpeg_sphere == asset.jpeg_sphere);
 390                 if( result && format == FILE_FFMPEG )
 391                         result = !strcmp(ff_video_options, asset.ff_video_options) &&
 392                                 !strcmp(ff_pixel_format, asset.ff_pixel_format) &&
 393                                 ff_video_bitrate == asset.ff_video_bitrate &&
 394                                 ff_video_quality == asset.ff_video_quality &&
 395                                 ff_color_space == asset.ff_color_space &&
 396                                 ff_color_range == asset.ff_color_range;
 397         }
 398
 399         return result;
 400 }
 401
 402 int Asset::test_path(const char *path)
 403 {
 404         if(!strcasecmp(this->path, path))
 405                 return 1;
 406         else
 407                 return 0;
 408 }
 409
 410 int Asset::read(FileXML *file, int expand_relative)
 411 {
 412         int result = 0;
 413 // Check for relative path.
 414         if( expand_relative ) {
 415                 char new_path[BCTEXTLEN];
 416                 char asset_directory[BCTEXTLEN];
 417                 char input_directory[BCTEXTLEN];
 418                 FileSystem fs;
 419                 strcpy(new_path, path);
 420                 fs.set_current_dir("");
 421                 fs.extract_dir(asset_directory, path);
 422 // No path in asset.
 423 // Take path of XML file.
 424                 if( !asset_directory[0] ) {
 425                         fs.extract_dir(input_directory, file->filename);
 426 // Input file has a path
 427                         if(input_directory[0])
 428                                 fs.join_names(path, input_directory, new_path);
 429                 }
 430         }
 431
 432         while( !(result=file->read_tag()) ) {
 433                 if( file->tag.title_is("/ASSET") ) break;
 434                 if( file->tag.title_is("AUDIO")  ) { read_audio(file);  continue; }
 435                 if( file->tag.title_is("AUDIO_OMIT") ) { read_audio(file);  continue; }
 436                 if( file->tag.title_is("FORMAT") ) { read_format(file);  continue; }
 437                 if( file->tag.title_is("VIDEO")  ) { read_video(file);  continue; }
 438                 if( file->tag.title_is("VIDEO_OMIT") ) { read_video(file);  continue; }
 439                 if( file->tag.title_is("INDEX")  ) { read_index(file);  continue; }
 440                 if( file->tag.title_is("FOLDER") )
 441                         folder_no = file->tag.get_property("NUMBER", AW_MEDIA_FOLDER);
 442         }
 443
 444         boundaries();
 445 //printf("Asset::read 2\n");
 446         return 0;
 447 }
 448
 449 int Asset::read_audio(FileXML *file)
 450 {
 451         if(file->tag.title_is("AUDIO")) audio_data = 1;
 452         channels = file->tag.get_property("CHANNELS", 2);
 453 // This is loaded from the index file after the EDL but this
 454 // should be overridable in the EDL.
 455         if(!sample_rate) sample_rate = file->tag.get_property("RATE", 48000);
 456         bits = file->tag.get_property("BITS", 16);
 457         byte_order = file->tag.get_property("BYTE_ORDER", 1);
 458         signed_ = file->tag.get_property("SIGNED", 1);
 459         header = file->tag.get_property("HEADER", 0);
 460         dither = file->tag.get_property("DITHER", 0);
 461
 462         audio_length = file->tag.get_property("AUDIO_LENGTH", (int64_t)0);
 463         acodec[0] = 0;
 464         file->tag.get_property("ACODEC", acodec);
 465         return 0;
 466 }
 467
 468 int Asset::read_video(FileXML *file)
 469 {
 470         char string[BCTEXTLEN];
 471
 472         if(file->tag.title_is("VIDEO")) video_data = 1;
 473         actual_height = file->tag.get_property("ACTUAL_HEIGHT", actual_height);
 474         actual_width = file->tag.get_property("ACTUAL_WIDTH", actual_width);
 475         height = file->tag.get_property("HEIGHT", height);
 476         width = file->tag.get_property("WIDTH", width);
 477         proxy_scale = file->tag.get_property("PROXY_SCALE", 0);
 478         proxy_edl = file->tag.get_property("PROXY_EDL", 0);
 479         layers = file->tag.get_property("LAYERS", layers);
 480         program = file->tag.get_property("PROGRAM", program);
 481 // This is loaded from the index file after the EDL but this
 482 // should be overridable in the EDL.
 483         if(EQUIV(frame_rate, 0)) frame_rate = file->tag.get_property("FRAMERATE", frame_rate);
 484         vcodec[0] = 0;
 485         file->tag.get_property("VCODEC", vcodec);
 486
 487         video_length = file->tag.get_property("VIDEO_LENGTH", (int64_t)0);
 488         timecode = file->tag.get_property("TIMECODE", -2);
 489         mov_sphere = file->tag.get_property("MOV_SPHERE", 0);
 490         jpeg_sphere = file->tag.get_property("JPEG_SPHERE", 0);
 491         single_frame = file->tag.get_property("SINGLE_FRAME", (int64_t)0);
 492         ilacemode_to_xmltext(string, ILACE_MODE_NOTINTERLACED);
 493         interlace_mode = ilacemode_from_xmltext(file->tag.get_property("INTERLACE_MODE",string), ILACE_MODE_NOTINTERLACED);
 494
 495         return 0;
 496 }
 497
 498 int Asset::read_format(FileXML *file)
 499 {
 500         const char *string = file->tag.get_property("TYPE");
 501         format = File::strtoformat(string);
 502         use_header = file->tag.get_property("USE_HEADER", use_header);
 503         fformat[0] = 0;
 504         file->tag.get_property("FFORMAT", fformat);
 505         ff_format_options[0] = 0;
 506         file->tag.get_property("FF_FORMAT_OPTIONS", ff_format_options);
 507         return 0;
 508 }
 509
 510 int Asset::read_index(FileXML *file)
 511 {
 512         index_state->read_xml(file, channels);
 513         return 0;
 514 }
 515
 516 // Output path is the path of the output file if name truncation is desired.
 517 // It is a "" if complete names should be used.
 518
 519 int Asset::write(FileXML *file,
 520         int include_index,
 521         const char *output_path)
 522 {
 523         char new_path[BCTEXTLEN];
 524         char asset_directory[BCTEXTLEN];
 525         char output_directory[BCTEXTLEN];
 526         FileSystem fs;
 527
 528 // Make path relative
 529         fs.extract_dir(asset_directory, path);
 530         if(output_path && output_path[0])
 531                 fs.extract_dir(output_directory, output_path);
 532         else
 533                 output_directory[0] = 0;
 534
 535 // Asset and EDL are in same directory.  Extract just the name.
 536         if(!strcmp(asset_directory, output_directory))
 537         {
 538                 fs.extract_name(new_path, path);
 539         }
 540         else
 541         {
 542                 strcpy(new_path, path);
 543         }
 544
 545         file->tag.set_title("ASSET");
 546         file->tag.set_property("SRC", new_path);
 547         file->append_tag();
 548         file->append_newline();
 549
 550         file->tag.set_title("FOLDER");
 551         file->tag.set_property("NUMBER", folder_no);
 552         file->append_tag();
 553         file->tag.set_title("/FOLDER");
 554         file->append_tag();
 555         file->append_newline();
 556
 557 // Write the format information
 558         file->tag.set_title("FORMAT");
 559
 560         file->tag.set_property("TYPE",
 561                 File::formattostr(format));
 562         file->tag.set_property("USE_HEADER", use_header);
 563         file->tag.set_property("FFORMAT", fformat);
 564         file->tag.set_property("FF_FORMAT_OPTIONS", ff_format_options);
 565
 566         file->append_tag();
 567         file->tag.set_title("/FORMAT");
 568         file->append_tag();
 569         file->append_newline();
 570
 571 // Requiring data to exist caused batch render to lose settings.
 572 // But the only way to know if an asset doesn't have audio or video data
 573 // is to not write the block.
 574 // So change the block name if the asset doesn't have the data.
 575         write_audio(file);
 576         write_video(file);
 577 // index goes after source
 578         if( include_index && index_state &&
 579             index_state->index_status == INDEX_READY )
 580                 write_index(file);
 581
 582         file->tag.set_title("/ASSET");
 583         file->append_tag();
 584         file->append_newline();
 585         return 0;
 586 }
 587
 588 int Asset::write_audio(FileXML *file)
 589 {
 590 // Let the reader know if the asset has the data by naming the block.
 591         if(audio_data)
 592                 file->tag.set_title("AUDIO");
 593         else
 594                 file->tag.set_title("AUDIO_OMIT");
 595 // Necessary for PCM audio
 596         file->tag.set_property("CHANNELS", channels);
 597         file->tag.set_property("RATE", sample_rate);
 598         file->tag.set_property("BITS", bits);
 599         file->tag.set_property("BYTE_ORDER", byte_order);
 600         file->tag.set_property("SIGNED", signed_);
 601         file->tag.set_property("HEADER", header);
 602         file->tag.set_property("DITHER", dither);
 603         if(acodec[0])
 604                 file->tag.set_property("ACODEC", acodec);
 605
 606         file->tag.set_property("AUDIO_LENGTH", audio_length);
 607
 608
 609
 610 // Rely on defaults operations for these.
 611
 612 //      file->tag.set_property("AMPEG_BITRATE", ampeg_bitrate);
 613 //      file->tag.set_property("AMPEG_DERIVATIVE", ampeg_derivative);
 614 //
 615 //      file->tag.set_property("VORBIS_VBR", vorbis_vbr);
 616 //      file->tag.set_property("VORBIS_MIN_BITRATE", vorbis_min_bitrate);
 617 //      file->tag.set_property("VORBIS_BITRATE", vorbis_bitrate);
 618 //      file->tag.set_property("VORBIS_MAX_BITRATE", vorbis_max_bitrate);
 619 //
 620 //      file->tag.set_property("MP3_BITRATE", mp3_bitrate);
 621 //
 622
 623
 624
 625         file->append_tag();
 626         if(audio_data)
 627           file->tag.set_title("/AUDIO");
 628         else
 629           file->tag.set_title("/AUDIO_OMIT");
 630         file->append_tag();
 631         file->append_newline();
 632         return 0;
 633 }
 634
 635 int Asset::write_video(FileXML *file)
 636 {
 637         char string[BCTEXTLEN];
 638
 639         if(video_data)
 640                 file->tag.set_title("VIDEO");
 641         else
 642                 file->tag.set_title("VIDEO_OMIT");
 643         file->tag.set_property("ACTUAL_HEIGHT", actual_height);
 644         file->tag.set_property("ACTUAL_WIDTH", actual_width);
 645         file->tag.set_property("HEIGHT", height);
 646         file->tag.set_property("WIDTH", width);
 647         file->tag.set_property("PROXY_SCALE", proxy_scale);
 648         file->tag.set_property("PROXY_EDL", proxy_edl);
 649         file->tag.set_property("LAYERS", layers);
 650         file->tag.set_property("PROGRAM", program);
 651         file->tag.set_property("FRAMERATE", frame_rate);
 652         if(vcodec[0])
 653                 file->tag.set_property("VCODEC", vcodec);
 654
 655         file->tag.set_property("VIDEO_LENGTH", video_length);
 656         file->tag.set_property("TIMECODE", timecode);
 657         file->tag.set_property("MOV_SPHERE", mov_sphere);
 658         file->tag.set_property("JPEG_SPHERE", jpeg_sphere);
 659         file->tag.set_property("SINGLE_FRAME", single_frame);
 660
 661         ilacemode_to_xmltext(string, interlace_mode);
 662         file->tag.set_property("INTERLACE_MODE", string);
 663
 664         file->append_tag();
 665         if(video_data)
 666                 file->tag.set_title("/VIDEO");
 667         else
 668                 file->tag.set_title("/VIDEO_OMIT");
 669
 670         file->append_tag();
 671         file->append_newline();
 672         return 0;
 673 }
 674
 675 int Asset::write_index(FileXML *file)
 676 {
 677         index_state->write_xml(file);
 678         return 0;
 679 }
 680
 681
 682 char* Asset::construct_param(const char *param, const char *prefix, char *return_value)
 683 {
 684         if(prefix)
 685                 sprintf(return_value, "%s%s", prefix, param);
 686         else
 687                 strcpy(return_value, param);
 688         return return_value;
 689 }
 690
 691 #define UPDATE_DEFAULT(x, y) defaults->update(construct_param(x, prefix, string), y);
 692 #define GET_DEFAULT(x, y) defaults->get(construct_param(x, prefix, string), y);
 693
 694 void Asset::load_defaults(BC_Hash *defaults,
 695         const char *prefix,
 696         int do_format,
 697         int do_compression,
 698         int do_path,
 699         int do_data_types,
 700         int do_bits)
 701 {
 702         char string[BCTEXTLEN];
 703
 704 // Can't save codec here because it's specific to render, record, and effect.
 705 // The codec has to be UNKNOWN for file probing to work.
 706
 707         if(do_path)
 708         {
 709                 GET_DEFAULT("PATH", path);
 710         }
 711
 712         if(do_compression)
 713         {
 714                 GET_DEFAULT("AUDIO_CODEC", acodec);
 715                 GET_DEFAULT("VIDEO_CODEC", vcodec);
 716         }
 717
 718         if(do_format)
 719         {
 720                 format = GET_DEFAULT("FORMAT", format);
 721                 use_header = GET_DEFAULT("USE_HEADER", use_header);
 722                 GET_DEFAULT("FFORMAT", fformat);
 723                 GET_DEFAULT("FF_FORMAT_OPTIONS", ff_format_options);
 724         }
 725
 726         if(do_data_types)
 727         {
 728                 audio_data = GET_DEFAULT("AUDIO", 1);
 729                 video_data = GET_DEFAULT("VIDEO", 1);
 730         }
 731
 732         if(do_bits)
 733         {
 734                 bits = GET_DEFAULT("BITS", 16);
 735                 dither = GET_DEFAULT("DITHER", 0);
 736                 signed_ = GET_DEFAULT("SIGNED", 1);
 737                 byte_order = GET_DEFAULT("BYTE_ORDER", 1);
 738
 739
 740
 741                 channels = GET_DEFAULT("CHANNELS", 2);
 742                 if(!sample_rate) sample_rate = GET_DEFAULT("RATE", 48000);
 743                 header = GET_DEFAULT("HEADER", 0);
 744                 audio_length = GET_DEFAULT("AUDIO_LENGTH", (int64_t)0);
 745
 746
 747
 748                 height = GET_DEFAULT("HEIGHT", height);
 749                 width = GET_DEFAULT("WIDTH", width);
 750                 actual_height = GET_DEFAULT("ACTUAL_HEIGHT", actual_height);
 751                 actual_width = GET_DEFAULT("ACTUAL_WIDTH", actual_width);
 752                 proxy_scale = GET_DEFAULT("PROXY_SCALE", proxy_scale);
 753                 proxy_edl = GET_DEFAULT("PROXY_SCALE", proxy_edl);
 754                 program = GET_DEFAULT("PROGRAM", program);
 755                 layers = GET_DEFAULT("LAYERS", layers);
 756                 if(EQUIV(frame_rate, 0)) frame_rate = GET_DEFAULT("FRAMERATE", frame_rate);
 757                 video_length = GET_DEFAULT("VIDEO_LENGTH", (int64_t)0);
 758                 single_frame = GET_DEFAULT("SINGLE_FRAME", (int64_t)0);
 759         }
 760
 761         ampeg_bitrate = GET_DEFAULT("AMPEG_BITRATE", ampeg_bitrate);
 762         ampeg_derivative = GET_DEFAULT("AMPEG_DERIVATIVE", ampeg_derivative);
 763
 764         vorbis_vbr = GET_DEFAULT("VORBIS_VBR", vorbis_vbr);
 765         vorbis_min_bitrate = GET_DEFAULT("VORBIS_MIN_BITRATE", vorbis_min_bitrate);
 766         vorbis_bitrate = GET_DEFAULT("VORBIS_BITRATE", vorbis_bitrate);
 767         vorbis_max_bitrate = GET_DEFAULT("VORBIS_MAX_BITRATE", vorbis_max_bitrate);
 768
 769         theora_fix_bitrate = GET_DEFAULT("THEORA_FIX_BITRATE", theora_fix_bitrate);
 770         theora_bitrate = GET_DEFAULT("THEORA_BITRATE", theora_bitrate);
 771         theora_quality = GET_DEFAULT("THEORA_QUALITY", theora_quality);
 772         theora_sharpness = GET_DEFAULT("THEORA_SHARPNESS", theora_sharpness);
 773         theora_keyframe_frequency = GET_DEFAULT("THEORA_KEYFRAME_FREQUENCY", theora_keyframe_frequency);
 774         theora_keyframe_force_frequency = GET_DEFAULT("THEORA_FORCE_KEYFRAME_FREQUENCY", theora_keyframe_force_frequency);
 775
 776         GET_DEFAULT("FF_AUDIO_OPTIONS", ff_audio_options);
 777         GET_DEFAULT("FF_SAMPLE_FORMAT", ff_sample_format);
 778         ff_audio_bitrate = GET_DEFAULT("FF_AUDIO_BITRATE", ff_audio_bitrate);
 779         ff_audio_quality = GET_DEFAULT("FF_AUDIO_QUALITY", ff_audio_quality);
 780         GET_DEFAULT("FF_VIDEO_OPTIONS", ff_video_options);
 781         GET_DEFAULT("FF_PIXEL_FORMAT", ff_pixel_format);
 782         ff_video_bitrate = GET_DEFAULT("FF_VIDEO_BITRATE", ff_video_bitrate);
 783         ff_video_quality = GET_DEFAULT("FF_VIDEO_QUALITY", ff_video_quality);
 784         ff_color_space = GET_DEFAULT("FF_COLOR_SPACE", ff_color_space);
 785         ff_color_range = GET_DEFAULT("FF_COLOR_RANGE", ff_color_range);
 786         GET_DEFAULT("FF_FORMAT_OPTIONS", ff_format_options);
 787
 788         mp3_bitrate = GET_DEFAULT("MP3_BITRATE", mp3_bitrate);
 789
 790         jpeg_quality = GET_DEFAULT("JPEG_QUALITY", jpeg_quality);
 791         aspect_ratio = GET_DEFAULT("ASPECT_RATIO", aspect_ratio);
 792         interlace_mode          = ILACE_MODE_UNDETECTED;
 793
 794 // MPEG format information
 795         vmpeg_iframe_distance = GET_DEFAULT("VMPEG_IFRAME_DISTANCE", vmpeg_iframe_distance);
 796         vmpeg_pframe_distance = GET_DEFAULT("VMPEG_PFRAME_DISTANCE", vmpeg_pframe_distance);
 797         vmpeg_progressive = GET_DEFAULT("VMPEG_PROGRESSIVE", vmpeg_progressive);
 798         vmpeg_denoise = GET_DEFAULT("VMPEG_DENOISE", vmpeg_denoise);
 799         vmpeg_bitrate = GET_DEFAULT("VMPEG_BITRATE", vmpeg_bitrate);
 800         vmpeg_derivative = GET_DEFAULT("VMPEG_DERIVATIVE", vmpeg_derivative);
 801         vmpeg_quantization = GET_DEFAULT("VMPEG_QUANTIZATION", vmpeg_quantization);
 802         vmpeg_cmodel = GET_DEFAULT("VMPEG_CMODEL", vmpeg_cmodel);
 803         vmpeg_fix_bitrate = GET_DEFAULT("VMPEG_FIX_BITRATE", vmpeg_fix_bitrate);
 804         vmpeg_seq_codes = GET_DEFAULT("VMPEG_SEQ_CODES", vmpeg_seq_codes);
 805         vmpeg_preset = GET_DEFAULT("VMPEG_PRESET", vmpeg_preset);
 806         vmpeg_field_order = GET_DEFAULT("VMPEG_FIELD_ORDER", vmpeg_field_order);
 807
 808         theora_fix_bitrate = GET_DEFAULT("THEORA_FIX_BITRATE", theora_fix_bitrate);
 809         theora_bitrate = GET_DEFAULT("THEORA_BITRATE", theora_bitrate);
 810         theora_quality = GET_DEFAULT("THEORA_QUALITY", theora_quality);
 811         theora_sharpness = GET_DEFAULT("THEORA_SHARPNESS", theora_sharpness);
 812         theora_keyframe_frequency = GET_DEFAULT("THEORA_KEYFRAME_FREQUENCY", theora_keyframe_frequency);
 813         theora_keyframe_force_frequency = GET_DEFAULT("THEORA_FORCE_KEYFRAME_FEQUENCY", theora_keyframe_force_frequency);
 814
 815
 816         ac3_bitrate = GET_DEFAULT("AC3_BITRATE", ac3_bitrate);
 817
 818         png_use_alpha = GET_DEFAULT("PNG_USE_ALPHA", png_use_alpha);
 819         png_depth = GET_DEFAULT("PNG_DEPTH", png_depth);
 820         png_compression = GET_DEFAULT("PNG_COMPRESSION", png_compression);
 821         exr_use_alpha = GET_DEFAULT("EXR_USE_ALPHA", exr_use_alpha);
 822         exr_compression = GET_DEFAULT("EXR_COMPRESSION", exr_compression);
 823         tiff_cmodel = GET_DEFAULT("TIFF_CMODEL", tiff_cmodel);
 824         tiff_compression = GET_DEFAULT("TIFF_COMPRESSION", tiff_compression);
 825
 826         mov_sphere = GET_DEFAULT("MOV_SPHERE", mov_sphere);
 827         jpeg_sphere = GET_DEFAULT("JPEG_SPHERE", jpeg_sphere);
 828         boundaries();
 829 }
 830
 831 void Asset::save_defaults(BC_Hash *defaults,
 832         const char *prefix,
 833         int do_format,
 834         int do_compression,
 835         int do_path,
 836         int do_data_types,
 837         int do_bits)
 838 {
 839         char string[BCTEXTLEN];
 840
 841         UPDATE_DEFAULT("PATH", path);
 842
 843
 844
 845
 846         if(do_format)
 847         {
 848                 UPDATE_DEFAULT("FORMAT", format);
 849                 UPDATE_DEFAULT("USE_HEADER", use_header);
 850                 UPDATE_DEFAULT("FFORMAT", fformat);
 851                 UPDATE_DEFAULT("FF_FORMAT_OPTIONS", ff_format_options);
 852         }
 853
 854         if(do_data_types)
 855         {
 856                 UPDATE_DEFAULT("AUDIO", audio_data);
 857                 UPDATE_DEFAULT("VIDEO", video_data);
 858         }
 859
 860         if(do_compression)
 861         {
 862                 UPDATE_DEFAULT("AUDIO_CODEC", acodec);
 863                 UPDATE_DEFAULT("VIDEO_CODEC", vcodec);
 864
 865                 UPDATE_DEFAULT("AMPEG_BITRATE", ampeg_bitrate);
 866                 UPDATE_DEFAULT("AMPEG_DERIVATIVE", ampeg_derivative);
 867
 868                 UPDATE_DEFAULT("VORBIS_VBR", vorbis_vbr);
 869                 UPDATE_DEFAULT("VORBIS_MIN_BITRATE", vorbis_min_bitrate);
 870                 UPDATE_DEFAULT("VORBIS_BITRATE", vorbis_bitrate);
 871                 UPDATE_DEFAULT("VORBIS_MAX_BITRATE", vorbis_max_bitrate);
 872
 873                 UPDATE_DEFAULT("FF_AUDIO_OPTIONS", ff_audio_options);
 874                 UPDATE_DEFAULT("FF_SAMPLE_FORMAT", ff_sample_format);
 875                 UPDATE_DEFAULT("FF_AUDIO_BITRATE", ff_audio_bitrate);
 876                 UPDATE_DEFAULT("FF_AUDIO_QUALITY", ff_audio_quality);
 877                 UPDATE_DEFAULT("FF_VIDEO_OPTIONS", ff_video_options);
 878                 UPDATE_DEFAULT("FF_PIXEL_FORMAT",  ff_pixel_format);
 879                 UPDATE_DEFAULT("FF_VIDEO_BITRATE", ff_video_bitrate);
 880                 UPDATE_DEFAULT("FF_VIDEO_QUALITY", ff_video_quality);
 881                 UPDATE_DEFAULT("FF_COLOR_SPACE", ff_color_space);
 882                 UPDATE_DEFAULT("FF_COLOR_RANGE", ff_color_range);
 883
 884                 UPDATE_DEFAULT("THEORA_FIX_BITRATE", theora_fix_bitrate);
 885                 UPDATE_DEFAULT("THEORA_BITRATE", theora_bitrate);
 886                 UPDATE_DEFAULT("THEORA_QUALITY", theora_quality);
 887                 UPDATE_DEFAULT("THEORA_SHARPNESS", theora_sharpness);
 888                 UPDATE_DEFAULT("THEORA_KEYFRAME_FREQUENCY", theora_keyframe_frequency);
 889                 UPDATE_DEFAULT("THEORA_FORCE_KEYFRAME_FREQUENCY", theora_keyframe_force_frequency);
 890
 891
 892
 893                 UPDATE_DEFAULT("MP3_BITRATE", mp3_bitrate);
 894
 895                 UPDATE_DEFAULT("JPEG_QUALITY", jpeg_quality);
 896                 UPDATE_DEFAULT("ASPECT_RATIO", aspect_ratio);
 897
 898 // MPEG format information
 899                 UPDATE_DEFAULT("VMPEG_IFRAME_DISTANCE", vmpeg_iframe_distance);
 900                 UPDATE_DEFAULT("VMPEG_PFRAME_DISTANCE", vmpeg_pframe_distance);
 901                 UPDATE_DEFAULT("VMPEG_PROGRESSIVE", vmpeg_progressive);
 902                 UPDATE_DEFAULT("VMPEG_DENOISE", vmpeg_denoise);
 903                 UPDATE_DEFAULT("VMPEG_BITRATE", vmpeg_bitrate);
 904                 UPDATE_DEFAULT("VMPEG_DERIVATIVE", vmpeg_derivative);
 905                 UPDATE_DEFAULT("VMPEG_QUANTIZATION", vmpeg_quantization);
 906                 UPDATE_DEFAULT("VMPEG_CMODEL", vmpeg_cmodel);
 907                 UPDATE_DEFAULT("VMPEG_FIX_BITRATE", vmpeg_fix_bitrate);
 908                 UPDATE_DEFAULT("VMPEG_SEQ_CODES", vmpeg_seq_codes);
 909                 UPDATE_DEFAULT("VMPEG_PRESET", vmpeg_preset);
 910                 UPDATE_DEFAULT("VMPEG_FIELD_ORDER", vmpeg_field_order);
 911
 912
 913                 UPDATE_DEFAULT("AC3_BITRATE", ac3_bitrate);
 914
 915
 916                 UPDATE_DEFAULT("PNG_USE_ALPHA", png_use_alpha);
 917                 UPDATE_DEFAULT("PNG_DEPTH", png_depth);
 918                 UPDATE_DEFAULT("PNG_COMPRESSION", png_compression);
 919                 UPDATE_DEFAULT("EXR_USE_ALPHA", exr_use_alpha);
 920                 UPDATE_DEFAULT("EXR_COMPRESSION", exr_compression);
 921                 UPDATE_DEFAULT("TIFF_CMODEL", tiff_cmodel);
 922                 UPDATE_DEFAULT("TIFF_COMPRESSION", tiff_compression);
 923
 924
 925
 926                 UPDATE_DEFAULT("MOV_SPHERE", mov_sphere);
 927                 UPDATE_DEFAULT("JPEG_SPHERE", jpeg_sphere);
 928         }
 929
 930
 931
 932
 933         if(do_bits)
 934         {
 935                 UPDATE_DEFAULT("BITS", bits);
 936                 UPDATE_DEFAULT("DITHER", dither);
 937                 UPDATE_DEFAULT("SIGNED", signed_);
 938                 UPDATE_DEFAULT("BYTE_ORDER", byte_order);
 939
 940
 941
 942
 943
 944
 945                 UPDATE_DEFAULT("CHANNELS", channels);
 946                 UPDATE_DEFAULT("RATE", sample_rate);
 947                 UPDATE_DEFAULT("HEADER", header);
 948                 UPDATE_DEFAULT("AUDIO_LENGTH", audio_length);
 949
 950
 951
 952                 UPDATE_DEFAULT("HEIGHT", height);
 953                 UPDATE_DEFAULT("WIDTH", width);
 954                 UPDATE_DEFAULT("ACTUAL_HEIGHT", actual_height);
 955                 UPDATE_DEFAULT("ACTUAL_WIDTH", actual_width);
 956                 UPDATE_DEFAULT("PROXY_SCALE", proxy_scale);
 957                 UPDATE_DEFAULT("PROXY_EDL", proxy_edl);
 958                 UPDATE_DEFAULT("PROGRAM", program);
 959                 UPDATE_DEFAULT("LAYERS", layers);
 960                 UPDATE_DEFAULT("FRAMERATE", frame_rate);
 961                 UPDATE_DEFAULT("VIDEO_LENGTH", video_length);
 962                 UPDATE_DEFAULT("SINGLE_FRAME", single_frame);
 963
 964         }
 965 }
 966
 967
 968
 969
 970
 971
 972
 973
 974
 975 int Asset::dump(FILE *fp)
 976 {
 977         fprintf(fp,"  asset::dump\n");
 978         fprintf(fp,"   this=%p path=%s\n", this, path);
 979         fprintf(fp,"   index_status %d\n", index_state->index_status);
 980         fprintf(fp,"   format %d\n", format);
 981         fprintf(fp,"   fformat=\"%s\"\n", fformat);
 982         fprintf(fp,"   ff_audio_options=\"%s\"\n", ff_audio_options);
 983         fprintf(fp,"   ff_sample_format=\"%s\"\n", ff_sample_format);
 984         fprintf(fp,"   ff_audio_bitrate=%d\n", ff_audio_bitrate);
 985         fprintf(fp,"   ff_audio_quality=%d\n", ff_audio_quality);
 986         fprintf(fp,"   ff_video_options=\"%s\"\n", ff_video_options);
 987         fprintf(fp,"   ff_pixel_format=\"%s\"\n", ff_pixel_format);
 988         fprintf(fp,"   ff_video_bitrate=%d\n", ff_video_bitrate);
 989         fprintf(fp,"   ff_video_quality=%d\n", ff_video_quality);
 990         fprintf(fp,"   ff_color_space=%d\n", ff_color_space);
 991         fprintf(fp,"   ff_color_range=%d\n", ff_color_range);
 992         fprintf(fp,"   audio_data %d channels %d samplerate %d bits %d"
 993                 " byte_order %d signed %d header %d dither %d acodec %4.4s\n",
 994                 audio_data, channels, sample_rate, bits, byte_order, signed_,
 995                 header, dither, acodec);
 996         fprintf(fp,"   audio_length %jd\n", audio_length);
 997         char string[BCTEXTLEN];
 998         ilacemode_to_xmltext(string, interlace_mode);
 999         fprintf(fp,"   video_data %d program %d layers %d framerate %f width %d"
1000                 " height %d vcodec %s aspect_ratio %f ilace_mode %s\n",
1001                 video_data, layers, program, frame_rate, width, height,
1002                 vcodec, aspect_ratio,string);
1003         fprintf(fp,"   actual_width %d actual_height %d proxy_scale %d proxy_edl %d\n",
1004                 actual_width, actual_height, proxy_scale, proxy_edl);
1005         fprintf(fp,"   video_length %jd repeat %d timecode %f\n",
1006                 video_length, single_frame, timecode);
1007         fprintf(fp,"   mov_sphere=%d jpeg_sphere=%d\n", mov_sphere, jpeg_sphere);
1008         return 0;
1009 }
1010
1011
1012 // For Indexable
1013 int Asset::get_audio_channels()
1014 {
1015         return channels;
1016 }
1017
1018 int Asset::get_sample_rate()
1019 {
1020         return sample_rate;
1021 }
1022
1023 int64_t Asset::get_audio_samples()
1024 {
1025         return audio_length;
1026 }
1027
1028 int Asset::have_audio()
1029 {
1030         return audio_data;
1031 }
1032
1033 int Asset::have_video()
1034 {
1035         return video_data;
1036 }
1037
1038 int Asset::get_w()
1039 {
1040         return width;
1041 }
1042
1043 int Asset::get_h()
1044 {
1045         return height;
1046 }
1047
1048 double Asset::get_frame_rate()
1049 {
1050         return frame_rate;
1051 }
1052
1053
1054 int Asset::get_video_layers()
1055 {
1056         return layers;
1057 }
1058
1059 int Asset::get_program()
1060 {
1061         return program;
1062 }
1063
1064 int64_t Asset::get_video_frames()
1065 {
1066         return video_length;
1067 }
1068
1069 double Asset::total_length_framealigned(double fps)
1070 {
1071         if (video_data && audio_data) {
1072                 double aud = floor(( (double)audio_length / sample_rate) * fps) / fps;
1073                 double vid = floor(( (double)video_length / frame_rate) * fps) / fps;
1074                 return MIN(aud,vid);
1075         }
1076
1077         if (audio_data)
1078                 return (double)audio_length / sample_rate;
1079
1080         if (video_data)
1081                 return (double)video_length / frame_rate;
1082
1083         return 0;
1084 }
1085