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