/* preset.c Copyright (c) 2003-2015 HandBrake Team This file is part of the HandBrake source code Homepage: . It may be used under the terms of the GNU General Public License v2. For full terms see the file COPYING file or visit http://www.gnu.org/licenses/gpl-2.0.html */ #include "builtin_presets.h" #include "hb.h" #include "hb_dict.h" #include "plist.h" #if defined(SYS_LINUX) #define HB_PRESET_PLIST_FILE "ghb/presets" #define HB_PRESET_JSON_FILE "ghb/presets.json" #elif defined(SYS_MINGW) #define HB_PRESET_PLIST_FILE "HandBrake\\user_presets.xml" #define HB_PRESET_JSON_FILE "HandBrake\\user_presets.json" #elif defined(SYS_DARWIN) #define HB_PRESET_PLIST_FILE "HandBrake/UserPresets.plist" #define HB_PRESET_JSON_FILE "HandBrake/UserPresets.json" #endif int hb_preset_version_major; int hb_preset_version_minor; int hb_preset_version_micro; static hb_value_t *hb_preset_template = NULL; static hb_value_t *hb_presets = NULL; static hb_value_t *hb_presets_builtin = NULL; static void preset_clean(hb_value_t *preset, hb_value_t *template); static void preset_import(hb_value_t *preset, int major, int minor, int micro); enum { PRESET_DO_SUCCESS, PRESET_DO_FAIL, PRESET_DO_PARTIAL, PRESET_DO_NEXT, PRESET_DO_SKIP, PRESET_DO_DELETE, PRESET_DO_DONE }; typedef struct { hb_preset_index_t path; } preset_do_context_t; typedef struct { preset_do_context_t do_ctx; hb_value_t *template; } preset_clean_context_t; typedef struct { preset_do_context_t do_ctx; int major; int minor; int micro; } preset_import_context_t; typedef struct { preset_do_context_t do_ctx; const char *name; int recurse; int last_match_idx; } preset_search_context_t; typedef int (*preset_do_f)(hb_value_t *preset, preset_do_context_t *ctx); static int preset_cmp_idx(hb_value_t *preset, int idx, const char *name) { const char *next, *preset_name; int ii, len; // Strip leading '/' if (name[0] == '/') name++; // Find the part of the "name" path we want to match. for (ii = 0; ii < idx; ii++) { next = strchr(name, '/'); if (next == NULL) return PRESET_DO_SKIP; next++; name = next; } // Find the end of the part we want to match next = strchr(name, '/'); if (next != NULL) len = next - name; else len = strlen(name); if (len <= 0) return PRESET_DO_SKIP; preset_name = hb_value_get_string(hb_dict_get(preset, "PresetName")); if (strlen(preset_name) > len) len = strlen(preset_name); // If match found and it's the last component of the "name", success! if (!strncmp(name, preset_name, len)) { if (name[len] == 0) return PRESET_DO_SUCCESS; else return PRESET_DO_PARTIAL; } return PRESET_DO_NEXT; } static int do_preset_search(hb_value_t *preset, preset_do_context_t *do_ctx) { preset_search_context_t *ctx = (preset_search_context_t*)do_ctx; int idx, result; idx = ctx->do_ctx.path.depth - 1; if (ctx->last_match_idx >= 0 && idx > ctx->last_match_idx) { // If there was a previous partial match, try to continue the match idx -= ctx->last_match_idx; } result = preset_cmp_idx(preset, idx, ctx->name); if (ctx->recurse && result == PRESET_DO_SKIP) { result = preset_cmp_idx(preset, 0, ctx->name); ctx->last_match_idx = idx; } if (result == PRESET_DO_PARTIAL) { return PRESET_DO_NEXT; } else { ctx->last_match_idx = -1; } return result; } static int do_preset_import(hb_value_t *preset, preset_do_context_t *do_ctx) { preset_import_context_t *ctx = (preset_import_context_t*)do_ctx; preset_import(preset, ctx->major, ctx->minor, ctx->micro); return PRESET_DO_NEXT; } static int do_preset_clean(hb_value_t *preset, preset_do_context_t *do_ctx) { preset_clean_context_t *ctx = (preset_clean_context_t*)do_ctx; preset_clean(preset, ctx->template); return PRESET_DO_NEXT; } static int do_delete_builtin(hb_value_t *preset, preset_do_context_t *ctx) { if (hb_value_get_int(hb_dict_get(preset, "Type")) == 0) return PRESET_DO_DELETE; return PRESET_DO_NEXT; } static int do_clear_default(hb_value_t *preset, preset_do_context_t *ctx) { hb_dict_set(preset, "Default", hb_value_bool(0)); return PRESET_DO_NEXT; } static int do_find_default(hb_value_t *preset, preset_do_context_t *ctx) { if (!hb_value_get_bool(hb_dict_get(preset, "Folder")) && hb_value_get_bool(hb_dict_get(preset, "Default"))) { return PRESET_DO_SUCCESS; } return PRESET_DO_NEXT; } static int presets_do(preset_do_f do_func, hb_value_t *preset, preset_do_context_t *ctx) { int result; hb_value_t *next; if (hb_value_type(preset) == HB_VALUE_TYPE_ARRAY) { // An array of presets, clean each one int ii; for (ii = 0; ii < hb_value_array_len(preset); ) { ctx->path.index[ctx->path.depth-1] = ii; next = hb_value_array_get(preset, ii); result = presets_do(do_func, next, ctx); if (result == PRESET_DO_DELETE) { hb_value_array_remove(preset, ii); continue; } ii++; if (result == PRESET_DO_SKIP) return PRESET_DO_NEXT; if (result != PRESET_DO_NEXT) return result; } return PRESET_DO_NEXT; } else if (hb_value_type(preset) == HB_VALUE_TYPE_DICT && hb_dict_get(preset, "VersionMajor") != NULL) { // A packaged preset list next = hb_dict_get(preset, "PresetList"); return presets_do(do_func, next, ctx); } else if (hb_value_type(preset) == HB_VALUE_TYPE_DICT && hb_value_get_bool(hb_dict_get(preset, "Folder"))) { // Perform do_func on the folder... result = do_func(preset, ctx); if (result != PRESET_DO_NEXT) return result; // Then perform preset action on the children of the folder ctx->path.depth++; next = hb_dict_get(preset, "ChildrenArray"); result = presets_do(do_func, next, ctx); if (result == PRESET_DO_SUCCESS) return result; ctx->path.depth--; return result; } else if (hb_value_type(preset) == HB_VALUE_TYPE_DICT && hb_dict_get(preset, "PresetName") != NULL) { // An individual, non-folder, preset return do_func(preset, ctx); } else { hb_error("Error: invalid preset format in presets_do()"); return PRESET_DO_NEXT; } return PRESET_DO_DONE; } hb_preset_index_t* hb_preset_index_init(const int *index, int depth) { hb_preset_index_t *path; path = malloc(sizeof(hb_preset_index_t)); path->depth = depth; if (index != NULL) memcpy(path->index, index, depth * sizeof(int)); return path; } hb_preset_index_t* hb_preset_index_dup(const hb_preset_index_t *path) { if (path == NULL) return NULL; return hb_preset_index_init(path->index, path->depth); } void hb_preset_index_append(hb_preset_index_t *dst, const hb_preset_index_t *src) { int ii; for (ii = 0; ii < src->depth && dst->depth < HB_MAX_PRESET_FOLDER_DEPTH; ii++, dst->depth++) { dst->index[dst->depth] = src->index[ii]; } } static int get_job_mux(hb_dict_t *job_dict) { int mux; hb_dict_t *dest_dict = hb_dict_get(job_dict, "Destination"); hb_value_t *mux_value = hb_dict_get(dest_dict, "Mux"); if (hb_value_type(mux_value) == HB_VALUE_TYPE_STRING) { mux = hb_container_get_from_name(hb_value_get_string(mux_value)); if (mux == 0) mux = hb_container_get_from_extension( hb_value_get_string(mux_value)); } else { mux = hb_value_get_int(mux_value); } hb_container_t *container = hb_container_get_from_format(mux); if (container == NULL) { char *str = hb_value_get_string_xform(mux_value); hb_error("Invalid container (%s)", str); free(str); return HB_MUX_INVALID; } return mux; } static hb_value_t* get_audio_copy_mask(const hb_dict_t * preset, int *mask) { int copy_mask = 0; hb_value_array_t *out_copy_mask, *in_copy_mask; if (mask != NULL) *mask = 0; in_copy_mask = hb_dict_get(preset, "AudioCopyMask"); out_copy_mask = hb_value_array_init(); if (in_copy_mask != NULL) { int count, ii; count = hb_value_array_len(in_copy_mask); for (ii = 0; ii < count; ii++) { int codec; hb_value_t *value; value = hb_value_array_get(in_copy_mask, ii); if (hb_value_type(value) == HB_VALUE_TYPE_STRING) { char *tmp = NULL; const char * s = hb_value_get_string(value); // Only codecs that start with 'copy:' can be copied if (strncmp(s, "copy:", 5)) { s = tmp = hb_strdup_printf("copy:%s", s); } codec = hb_audio_encoder_get_from_name(s); if (codec == 0) { hb_error("Invalid audio codec in autopassthru copy mask (%s)", s); hb_error("Codec name is invalid or can not be copied"); free(tmp); hb_value_free(&out_copy_mask); return NULL; } free(tmp); } else { codec = hb_value_get_int(value); } hb_value_array_append(out_copy_mask, hb_value_string( hb_audio_encoder_get_short_name(codec))); copy_mask |= codec; } } if (mask != NULL) *mask = copy_mask; return out_copy_mask; } static hb_dict_t * source_audio_track_used(hb_dict_t *track_dict, int track) { // Kind of hacky, but keys must be strings char key[8]; snprintf(key, sizeof(key), "%d", track); hb_dict_t *used = hb_dict_get(track_dict, key); if (used == NULL) { used = hb_dict_init(); hb_dict_set(track_dict, key, used); } return used; } // Find a source audio track matching given language static int find_audio_track(const hb_title_t *title, const char *lang, int start) { hb_audio_config_t * audio; int ii, count; count = hb_list_count(title->list_audio); for (ii = start; ii < count; ii++) { audio = hb_list_audio_config_item(title->list_audio, ii); // Ignore secondary audio types if ((audio->lang.type == HB_AUDIO_TYPE_NONE || audio->lang.type == HB_AUDIO_TYPE_NORMAL) && (!strcmp(lang, audio->lang.iso639_2) || !strcmp(lang, "und"))) { return ii; } } return -1; } static int validate_audio_encoders(const hb_dict_t *preset) { hb_value_array_t * encoder_list = hb_dict_get(preset, "AudioList"); int count = hb_value_array_len(encoder_list); int ii; for (ii = 0; ii < count; ii++) { hb_value_t *audio_dict = hb_value_array_get(encoder_list, ii); hb_value_t *value; int codec, mix, sr; value = hb_dict_get(audio_dict, "AudioEncoder"); if (hb_value_type(value) == HB_VALUE_TYPE_STRING) { codec = hb_audio_encoder_get_from_name(hb_value_get_string(value)); } else { codec = hb_value_get_int(value); } if (hb_audio_encoder_get_from_codec(codec) == NULL) { char *str = hb_value_get_string_xform(value); hb_error("Invalid audio encoder (%s)", str); free(str); return -1; } value = hb_dict_get(audio_dict, "AudioMixdown"); if (hb_value_type(value) == HB_VALUE_TYPE_STRING) { mix = hb_audio_encoder_get_from_name(hb_value_get_string(value)); } else { mix = hb_value_get_int(value); } if (hb_mixdown_get_from_mixdown(mix) == NULL) { char *str = hb_value_get_string_xform(value); hb_error("Invalid audio mixdown (%s)", str); free(str); return -1; } value = hb_dict_get(audio_dict, "AudioSamplerate"); if (hb_value_type(value) == HB_VALUE_TYPE_STRING) { const char *str = hb_value_get_string(value); if (!strcasecmp(str, "source") || !strcasecmp(str, "auto") || !strcasecmp(str, "same as source")) { sr = 0; } else { sr = hb_audio_samplerate_get_from_name(str); } } else { sr = hb_value_get_int(value); } if (sr != 0 && hb_audio_samplerate_get_name(sr) == NULL) { char *str = hb_value_get_string_xform(value); hb_error("Invalid audio samplerate (%s)", str); free(str); return -1; } } return 0; } static int sanitize_audio_codec(int in_codec, int out_codec, int copy_mask, int fallback, int mux) { int codec = out_codec; if (out_codec == HB_ACODEC_AUTO_PASS) { codec = hb_autopassthru_get_encoder(in_codec, copy_mask, fallback, mux); } else if ((out_codec & HB_ACODEC_PASS_FLAG) && !(in_codec & out_codec & HB_ACODEC_PASS_MASK)) { codec = hb_audio_encoder_get_fallback_for_passthru(out_codec); if (codec == 0) codec = fallback; } // Check that encoder is valid for mux const hb_encoder_t *encoder = NULL; while ((encoder = hb_audio_encoder_get_next(encoder)) != NULL) { if (encoder->codec == codec && !(encoder->muxers & mux)) { codec = hb_audio_encoder_get_default(mux); break; } } if (codec == 0) codec = hb_audio_encoder_get_default(mux); return codec; } static void add_audio_for_lang(hb_value_array_t *list, const hb_dict_t *preset, hb_title_t *title, int mux, int copy_mask, int fallback, const char *lang, int behavior, int mode, hb_dict_t *track_dict) { hb_value_array_t * encoder_list = hb_dict_get(preset, "AudioList"); int count = hb_value_array_len(encoder_list); int track = find_audio_track(title, lang, 0); int current_mode = 0; while (track >= 0) { char key[8]; snprintf(key, sizeof(key), "%d", track); count = current_mode ? 1 : count; int ii; for (ii = 0; ii < count; ii++) { // Check if this source track has already been added using these // same encoder settings. If so, continue to next track. hb_dict_t *used = source_audio_track_used(track_dict, ii); if (hb_value_get_bool(hb_dict_get(used, key))) continue; hb_dict_set(used, key, hb_value_bool(1)); // Create new audio output track settings hb_dict_t *audio_dict = hb_dict_init(); hb_value_t *acodec_value; hb_dict_t *encoder_dict = hb_value_array_get(encoder_list, ii); int out_codec; acodec_value = hb_dict_get(encoder_dict, "AudioEncoder"); if (hb_value_type(acodec_value) == HB_VALUE_TYPE_STRING) { out_codec = hb_audio_encoder_get_from_name( hb_value_get_string(acodec_value)); } else { out_codec = hb_value_get_int(acodec_value); } // Save the encoder value before sanitizing. This value is // useful to the frontends. hb_dict_set(audio_dict, "PresetEncoder", hb_value_string(hb_audio_encoder_get_short_name(out_codec))); hb_audio_config_t *aconfig; aconfig = hb_list_audio_config_item(title->list_audio, track); out_codec = sanitize_audio_codec(aconfig->in.codec, out_codec, copy_mask, fallback, mux); hb_dict_set(audio_dict, "Track", hb_value_int(track)); hb_dict_set(audio_dict, "Encoder", hb_value_string( hb_audio_encoder_get_short_name(out_codec))); if (hb_dict_get(encoder_dict, "AudioTrackName") != NULL) { hb_dict_set(audio_dict, "Name", hb_value_dup( hb_dict_get(encoder_dict, "AudioTrackName"))); } if (!(out_codec & HB_ACODEC_PASS_FLAG)) { if (hb_dict_get(encoder_dict, "AudioTrackGainSlider") != NULL) { hb_dict_set(audio_dict, "Gain", hb_value_dup( hb_dict_get(encoder_dict, "AudioTrackGainSlider"))); } if (hb_dict_get(encoder_dict, "AudioTrackDRCSlider") != NULL) { hb_dict_set(audio_dict, "DRC", hb_value_dup( hb_dict_get(encoder_dict, "AudioTrackDRCSlider"))); } if (hb_dict_get(encoder_dict, "AudioMixdown") != NULL) { hb_dict_set(audio_dict, "Mixdown", hb_value_dup( hb_dict_get(encoder_dict, "AudioMixdown"))); } if (hb_dict_get(encoder_dict, "AudioNormalizeMixLevel") != NULL) { hb_dict_set(audio_dict, "NormalizeMixLevel", hb_value_dup( hb_dict_get(encoder_dict, "AudioNormalizeMixLevel"))); } if (hb_dict_get(encoder_dict, "AudioDitherMethod") != NULL) { hb_dict_set(audio_dict, "DitherMethod", hb_value_dup( hb_dict_get(encoder_dict, "AudioDitherMethod"))); } if (hb_dict_get(encoder_dict, "AudioSamplerate") != NULL) { hb_dict_set(audio_dict, "Samplerate", hb_value_dup( hb_dict_get(encoder_dict, "AudioSamplerate"))); } if (hb_dict_get(encoder_dict, "AudioCompressionLevel") != NULL) { hb_dict_set(audio_dict, "CompressionLevel", hb_value_dup( hb_dict_get(encoder_dict, "AudioCompressionLevel"))); } if (hb_value_get_bool(hb_dict_get(encoder_dict, "AudioTrackQualityEnable"))) { hb_dict_set(audio_dict, "Quality", hb_value_xform( hb_dict_get(encoder_dict, "AudioTrackQuality"), HB_VALUE_TYPE_DOUBLE)); } else { hb_dict_set(audio_dict, "Bitrate", hb_value_xform( hb_dict_get(encoder_dict, "AudioBitrate"), HB_VALUE_TYPE_INT)); } } hb_value_array_append(list, audio_dict); } if (behavior == 2) track = find_audio_track(title, lang, track + 1); else break; } } // This function assumes that Mux has already been initialized in // the job_dict int hb_preset_job_add_audio(hb_handle_t *h, int title_index, const hb_dict_t *preset, hb_dict_t *job_dict) { hb_title_t *title = hb_find_title_by_index(h, title_index); if (title == NULL) { // Can't create audio track list without knowing source audio tracks hb_error("Invalid title index (%d)", title_index); return -1; } if (hb_list_count(title->list_audio) <= 0) { // Source has no audio return 0; } int mux = get_job_mux(job_dict); if (mux == HB_MUX_INVALID) { return -1; } hb_dict_t *audio_dict = hb_dict_get(job_dict, "Audio"); if (audio_dict == NULL) { audio_dict = hb_dict_init(); hb_dict_set(job_dict, "Audio", audio_dict); } int copy_mask; hb_value_t *copy_mask_array = get_audio_copy_mask(preset, ©_mask); if (copy_mask_array == NULL) { return -1; } int fallback = 0; hb_dict_set(audio_dict, "CopyMask", copy_mask_array); hb_value_t *fallback_value = hb_dict_get(preset, "AudioEncoderFallback"); if (fallback_value != NULL) { hb_dict_set(audio_dict, "FallbackEncoder", hb_value_dup(fallback_value)); if (hb_value_type(fallback_value) == HB_VALUE_TYPE_STRING) { const char * s = hb_value_get_string(fallback_value); fallback = hb_audio_encoder_get_from_name(s); if (fallback == 0) { hb_error("Invalid fallback audio codec (%s)", s); return -1; } } else { fallback = hb_value_get_int(fallback_value); } } if (validate_audio_encoders(preset) < 0) return -1; hb_value_array_t *list = hb_dict_get(audio_dict, "AudioList"); if (list == NULL) { list = hb_value_array_init(); hb_dict_set(audio_dict, "AudioList", list); } int behavior = 1; // default first const char *s; s = hb_value_get_string(hb_dict_get(preset, "AudioTrackSelectionBehavior")); if (s != NULL) { if (!strcasecmp(s, "none")) return 0; else if (!strcasecmp(s, "all")) behavior = 2; } // Create hash that is used to track which tracks have been already added // We do not want to add the same track with the same settings twice hb_dict_t *track_dict = hb_dict_init(); // Add tracks for all languages in the language list int mode; hb_value_array_t *lang_list = hb_dict_get(preset, "AudioLanguageList"); mode = hb_value_get_bool(hb_dict_get(preset, "AudioSecondaryEncoderMode")); int count = hb_value_array_len(lang_list); int ii; for (ii = 0; ii < count; ii++) { const char *lang; lang = hb_value_get_string(hb_value_array_get(lang_list, ii)); add_audio_for_lang(list, preset, title, mux, copy_mask, fallback, lang, behavior, mode, track_dict); } // If no audios found, try "und" language option if (hb_value_array_len(list) <= 0) { add_audio_for_lang(list, preset, title, mux, copy_mask, fallback, "und", behavior, mode, track_dict); } hb_dict_free(&track_dict); return 0; } // Find a source audio track matching given language static int find_subtitle_track(const hb_title_t *title, const char *lang, int start) { hb_subtitle_t * subtitle; int ii, count; count = hb_list_count(title->list_subtitle); for (ii = start; ii < count; ii++) { subtitle = hb_list_item(title->list_subtitle, ii); if (!strcmp(lang, subtitle->iso639_2) || !strcmp(lang, "und")) { return ii; } } return -1; } static void add_subtitle(hb_value_array_t *list, int track, int make_default, int force, int burn) { hb_dict_t *subtitle_dict = hb_dict_init(); hb_dict_set(subtitle_dict, "Track", hb_value_int(track)); hb_dict_set(subtitle_dict, "Default", hb_value_bool(make_default)); hb_dict_set(subtitle_dict, "Forced", hb_value_bool(force)); hb_dict_set(subtitle_dict, "Burn", hb_value_bool(burn)); hb_value_array_append(list, subtitle_dict); } typedef struct subtitle_behavior_s { int one; int burn_foreign; int make_default; int burn_first; int burn_dvd; int burn_bd; int one_burned; uint8_t *used; } subtitle_behavior_t; static void add_subtitle_for_lang(hb_value_array_t *list, hb_title_t *title, int mux, const char *lang, subtitle_behavior_t *behavior) { int t; t = find_subtitle_track(title, lang, 0); for (t = find_subtitle_track(title, lang, 0); t >= 0; t = behavior->one ? -1 : find_subtitle_track(title, lang, t + 1)) { if (behavior->used[t]) { if (behavior->one) break; continue; } int burn, make_default; hb_subtitle_t *subtitle; subtitle = hb_list_item(title->list_subtitle, t); burn = !behavior->one_burned && ((subtitle->source == VOBSUB && behavior->burn_dvd) || (subtitle->source == PGSSUB && behavior->burn_bd) || !hb_subtitle_can_pass(subtitle->source, mux) || behavior->burn_first || behavior->burn_foreign); make_default = !burn && behavior->make_default; behavior->burn_first &= !burn; behavior->one_burned |= burn; behavior->used[t] = 1; add_subtitle(list, t, make_default, 0 /*!force*/, burn); } } // This function assumes that the AudioList and Mux have already been // initialized in the job_dict int hb_preset_job_add_subtitles(hb_handle_t *h, int title_index, const hb_dict_t *preset, hb_dict_t *job_dict) { hb_title_t *title = hb_find_title_by_index(h, title_index); if (title == NULL) { // Can't create subtitle track list without knowing source hb_error("Invalid title index (%d)", title_index); return -1; } int mux = get_job_mux(job_dict); if (mux == HB_MUX_INVALID) { return -1; } // Get the language of the first audio output track // Needed for subtitle track selection hb_dict_t *audio_dict = hb_dict_get(job_dict, "Audio"); hb_value_array_t *audio_list = hb_dict_get(audio_dict, "AudioList"); const char *first_audio_lang = NULL; if (hb_value_array_len(audio_list) > 0) { int track; hb_value_t *audio = hb_value_array_get(audio_list, 0); track = hb_value_get_int(hb_dict_get(audio, "Track")); if (hb_list_count(title->list_audio) > track) { hb_audio_config_t *aconfig; aconfig = hb_list_audio_config_item(title->list_audio, track); if (aconfig != NULL) first_audio_lang = aconfig->lang.iso639_2; } } int source_subtitle_count = hb_list_count(title->list_subtitle); if (source_subtitle_count == 0) return 0; hb_dict_t *subtitle_dict = hb_dict_get(job_dict, "Subtitle"); if (subtitle_dict == NULL) { subtitle_dict = hb_dict_init(); hb_dict_set(job_dict, "Subtitle", subtitle_dict); } hb_value_array_t *list = hb_dict_get(subtitle_dict, "SubtitleList"); if (list == NULL) { list = hb_value_array_init(); hb_dict_set(subtitle_dict, "SubtitleList", list); } int track_behavior = 0; // default no subtitles int burn_behavior = 0; int burn_foreign; struct subtitle_behavior_s behavior; behavior.one = 0; behavior.burn_foreign = 0; behavior.make_default = 0; behavior.burn_first = 0; behavior.burn_dvd = 0; behavior.burn_bd = 0; behavior.one_burned = 0; // Create array that is used to track which tracks have been already added // We do not want to add the same track with the same settings twice behavior.used = calloc(source_subtitle_count, sizeof(*behavior.used)); // Since this function can be called multiple times, we need to // initialize the "used" array from the existing subtitles in the list. int count, ii; count = hb_value_array_len(list); for (ii = 0; ii < count; ii++) { hb_value_t *sub = hb_value_array_get(list, ii); int track = hb_value_get_int(hb_dict_get(sub, "Track")); behavior.used[track] = 1; } const char *s; s = hb_value_get_string(hb_dict_get(preset, "SubtitleTrackSelectionBehavior")); if (s != NULL) { if (!strcasecmp(s, "first")) track_behavior = 1; else if (!strcasecmp(s, "all")) track_behavior = 2; } s = hb_value_get_string(hb_dict_get(preset, "SubtitleBurnBehavior")); if (s != NULL) { if (!strcasecmp(s, "foreign")) burn_behavior = 1; else if (!strcasecmp(s, "first")) burn_behavior = 2; else if (!strcasecmp(s, "foreign_first")) burn_behavior = 3; } behavior.burn_dvd = hb_value_get_int(hb_dict_get(preset, "SubtitleBurnDVDSub")); behavior.burn_bd = hb_value_get_int(hb_dict_get(preset, "SubtitleBurnBDSub")); burn_foreign = burn_behavior == 1 || burn_behavior == 3; behavior.burn_first = burn_behavior == 2 || burn_behavior == 3; int foreign_audio_search, foreign_first_audio; foreign_audio_search = hb_value_get_bool(hb_dict_get(preset, "SubtitleAddForeignAudioSearch")); foreign_first_audio = hb_value_get_bool(hb_dict_get(preset, "SubtitleAddForeignAudioSubtitle")); // Add tracks for all languages in the language list hb_value_array_t *lang_list = hb_dict_get(preset, "SubtitleLanguageList"); count = hb_value_array_len(lang_list); const char *pref_lang = "und"; if (count > 0) { pref_lang = hb_value_get_string(hb_value_array_get(lang_list, 0)); } if (!strcmp(pref_lang, "und")) { foreign_audio_search = foreign_first_audio = 0; } int track; if (first_audio_lang != NULL && foreign_first_audio && strncmp(first_audio_lang, pref_lang, 4)) { // First audio lang does not match the preferred subittle lang. // Preset says to add pref lang subtitle. // Foreign audio search is not necessary since entire audio track // is foreign. foreign_audio_search = 0; behavior.one = 1; behavior.burn_foreign = burn_foreign; behavior.make_default = 1; add_subtitle_for_lang(list, title, mux, pref_lang, &behavior); } hb_dict_t *search_dict = hb_dict_get(subtitle_dict, "Search"); if (search_dict == NULL) { search_dict = hb_dict_init(); hb_dict_set(subtitle_dict, "Search", search_dict); } if (first_audio_lang != NULL && foreign_audio_search && !strncmp(first_audio_lang, pref_lang, 4)) { // First audio lang matches the preferred subittle lang. // Preset says to add search for foreign audio subtitles. int burn = burn_foreign || behavior.burn_first; // If not burning, make this the default track. hb_dict_set(search_dict, "Enable", hb_value_bool(1)); hb_dict_set(search_dict, "Default", hb_value_bool(!burn)); hb_dict_set(search_dict, "Forced", hb_value_bool(1)); hb_dict_set(search_dict, "Burn", hb_value_bool(burn)); } else { hb_dict_set(search_dict, "Enable", hb_value_bool(0)); } if (track_behavior > 0) { int ii; behavior.one = track_behavior == 1; behavior.burn_foreign = 0; behavior.make_default = 0; for (ii = 0; ii < count; ii++) { const char *lang; lang = hb_value_get_string(hb_value_array_get(lang_list, ii)); add_subtitle_for_lang(list, title, mux, lang, &behavior); } if (count <= 0) { add_subtitle_for_lang(list, title, mux, "und", &behavior); } } if (hb_value_get_bool(hb_dict_get(preset, "SubtitleAddCC"))) { // Add Closed Caption track for (track = 0; track < source_subtitle_count; track++) { if (behavior.used[track]) { continue; } hb_subtitle_t *subtitle = hb_list_item(title->list_subtitle, track); if (subtitle->source == CC608SUB || subtitle->source == CC708SUB) { int burn; burn = !behavior.one_burned && (!hb_subtitle_can_pass(subtitle->source, mux) || behavior.burn_first); behavior.used[track] = 1; behavior.one_burned |= burn; add_subtitle(list, track, 0 /*default*/, 0 /*!force*/, burn); break; } } } free(behavior.used); return 0; } static int get_video_framerate(hb_value_t *rate_value) { // Predefined by name if (hb_value_type(rate_value) == HB_VALUE_TYPE_STRING) { int rate = 0; const char *rate_name = hb_value_get_string(rate_value); if (!strcasecmp(rate_name, "source") || !strcasecmp(rate_name, "auto") || !strcasecmp(rate_name, "same as source")) { return rate; } else { rate = hb_video_framerate_get_from_name(rate_name); if (rate != -1) { return rate; } } } // Arbitrary int clock_min, clock_max, clock, frame_min, frame_max; hb_video_framerate_get_limits(&clock_min, &clock_max, &clock); frame_min = clock / clock_max; frame_max = clock / clock_min; double rate_d = hb_value_get_double(rate_value); if (rate_d >= frame_min && rate_d <= frame_max) { // Value is a framerate, return clockrate return (int)(clock / rate_d); } else if (rate_d >= clock_min && rate_d <= clock_max) { // Value is already a clockrate return (int)(rate_d); } // Value out of bounds return -1; } int hb_preset_apply_filters(const hb_dict_t *preset, hb_dict_t *job_dict) { hb_value_t *filters_dict, *filter_list, *filter_dict; char *filter_str; int clock_min, clock_max, clock; hb_video_framerate_get_limits(&clock_min, &clock_max, &clock); // Create new filters filters_dict = hb_dict_init(); hb_dict_set(job_dict, "Filters", filters_dict); filter_list = hb_value_array_init(); hb_dict_set(filters_dict, "FilterList", filter_list); hb_dict_set(filters_dict, "Grayscale", hb_value_xform( hb_dict_get(preset, "VideoGrayScale"), HB_VALUE_TYPE_BOOL)); // Detelecine filter hb_value_t *detel_val = hb_dict_get(preset, "PictureDetelecine"); if (detel_val != NULL) { const char *custom; custom = hb_value_get_string(hb_dict_get(preset, "PictureDetelecineCustom")); if (hb_value_type(detel_val) == HB_VALUE_TYPE_STRING) { filter_str = hb_generate_filter_settings( HB_FILTER_DETELECINE, hb_value_get_string(detel_val), custom); } else { filter_str = hb_generate_filter_settings_by_index( HB_FILTER_DETELECINE, hb_value_get_int(detel_val), custom); } if (filter_str == NULL) { char *s = hb_value_get_string_xform(detel_val); hb_error("Invalid detelecine filter settings (%s)", s); free(s); return -1; } else if (filter_str != hb_filter_off) { filter_dict = hb_dict_init(); hb_dict_set(filter_dict, "ID", hb_value_int(HB_FILTER_DETELECINE)); hb_dict_set(filter_dict, "Settings", hb_value_string(filter_str)); hb_value_array_append(filter_list, filter_dict); free(filter_str); } } // Decomb or deinterlace filters int decomb_or_deint; decomb_or_deint = hb_value_get_bool(hb_dict_get(preset, "PictureDecombDeinterlace")); hb_value_t *decomb_val = hb_dict_get(preset, "PictureDecomb"); if (decomb_or_deint && decomb_val != NULL) { const char *custom; custom = hb_value_get_string(hb_dict_get(preset, "PictureDecombCustom")); if (hb_value_type(decomb_val) == HB_VALUE_TYPE_STRING) { filter_str = hb_generate_filter_settings( HB_FILTER_DECOMB, hb_value_get_string(decomb_val), custom); } else { filter_str = hb_generate_filter_settings_by_index( HB_FILTER_DECOMB, hb_value_get_int(decomb_val), custom); } if (filter_str == NULL) { char *s = hb_value_get_string_xform(decomb_val); hb_error("Invalid decomb filter settings (%s)", s); free(s); return -1; } else if (filter_str != hb_filter_off) { filter_dict = hb_dict_init(); hb_dict_set(filter_dict, "ID", hb_value_int(HB_FILTER_DECOMB)); hb_dict_set(filter_dict, "Settings", hb_value_string(filter_str)); hb_value_array_append(filter_list, filter_dict); free(filter_str); } } hb_value_t *deint_val = hb_dict_get(preset, "PictureDeinterlace"); if (!decomb_or_deint && deint_val != NULL) { const char *custom; custom = hb_value_get_string(hb_dict_get(preset, "PictureDeinterlaceCustom")); if (hb_value_type(deint_val) == HB_VALUE_TYPE_STRING) { filter_str = hb_generate_filter_settings( HB_FILTER_DEINTERLACE, hb_value_get_string(deint_val), custom); } else { filter_str = hb_generate_filter_settings_by_index( HB_FILTER_DEINTERLACE, hb_value_get_int(deint_val), custom); } if (filter_str == NULL) { char *s = hb_value_get_string_xform(deint_val); hb_error("Invalid deinterlace filter settings (%s)", s); free(s); return -1; } else if (filter_str != hb_filter_off) { filter_dict = hb_dict_init(); hb_dict_set(filter_dict, "ID", hb_value_int(HB_FILTER_DEINTERLACE)); hb_dict_set(filter_dict, "Settings", hb_value_string(filter_str)); hb_value_array_append(filter_list, filter_dict); free(filter_str); } } // Denoise filter int denoise; hb_value_t *denoise_value = hb_dict_get(preset, "PictureDenoiseFilter"); denoise = hb_value_type(denoise_value) == HB_VALUE_TYPE_STRING ? ( !strcasecmp(hb_value_get_string(denoise_value), "off") ? 0 : !strcasecmp(hb_value_get_string(denoise_value), "nlmeans") ? 1 : 2) : hb_value_get_int(denoise_value); if (denoise != 0) { int filter_id = denoise == 1 ? HB_FILTER_NLMEANS : HB_FILTER_HQDN3D; const char *denoise_preset, *denoise_tune; denoise_preset = hb_value_get_string( hb_dict_get(preset, "PictureDenoisePreset")); if (denoise_preset != NULL) { if (strcasecmp(denoise_preset, "custom")) denoise_tune = hb_value_get_string( hb_dict_get(preset, "PictureDenoiseTune")); else denoise_tune = hb_value_get_string( hb_dict_get(preset, "PictureDenoiseCustom")); filter_str = hb_generate_filter_settings( filter_id, denoise_preset, denoise_tune); if (filter_str == NULL) { hb_error("Invalid denoise filter settings (%s%s%s)", denoise_preset, denoise_tune ? "," : "", denoise_tune ? denoise_tune : ""); return -1; } else if (filter_str != hb_filter_off) { filter_dict = hb_dict_init(); hb_dict_set(filter_dict, "ID", hb_value_int(filter_id)); hb_dict_set(filter_dict, "Settings", hb_value_string(filter_str)); hb_value_array_append(filter_list, filter_dict); free(filter_str); } } } // Deblock filter char *deblock = hb_value_get_string_xform( hb_dict_get(preset, "PictureDeblock")); if (deblock != NULL) { filter_str = hb_generate_filter_settings(HB_FILTER_DEBLOCK, deblock, NULL); if (filter_str == NULL) { hb_error("Invalid deblock filter settings (%s)", deblock); return -1; } else if (filter_str != hb_filter_off) { filter_dict = hb_dict_init(); hb_dict_set(filter_dict, "ID", hb_value_int(HB_FILTER_DEBLOCK)); hb_dict_set(filter_dict, "Settings", hb_value_string(filter_str)); hb_value_array_append(filter_list, filter_dict); free(filter_str); } } free(deblock); // Rotate filter char *rotate = hb_value_get_string_xform( hb_dict_get(preset, "PictureRotate")); if (rotate != NULL) { filter_str = hb_generate_filter_settings(HB_FILTER_ROTATE, rotate, NULL); if (filter_str == NULL) { hb_error("Invalid rotate filter settings (%s)", rotate); return -1; } else if (filter_str != hb_filter_off) { filter_dict = hb_dict_init(); hb_dict_set(filter_dict, "ID", hb_value_int(HB_FILTER_ROTATE)); hb_dict_set(filter_dict, "Settings", hb_value_string(filter_str)); hb_value_array_append(filter_list, filter_dict); free(filter_str); } } free(rotate); hb_value_t *fr_value = hb_dict_get(preset, "VideoFramerate"); int vrate_den = get_video_framerate(fr_value); if (vrate_den < 0) { char *str = hb_value_get_string_xform(fr_value); hb_error("Invalid video framerate (%s)", str); free(str); return -1; } int fr_mode; hb_value_t *fr_mode_value = hb_dict_get(preset, "VideoFramerateMode"); fr_mode = hb_value_type(fr_mode_value) == HB_VALUE_TYPE_STRING ? ( !strcasecmp(hb_value_get_string(fr_mode_value), "cfr") ? 1 : !strcasecmp(hb_value_get_string(fr_mode_value), "pfr") ? 2 : 0) : hb_value_get_int(fr_mode_value); if (vrate_den == 0) filter_str = hb_strdup_printf("%d", fr_mode); else filter_str = hb_strdup_printf("%d:%d:%d", fr_mode, clock, vrate_den); filter_dict = hb_dict_init(); hb_dict_set(filter_dict, "ID", hb_value_int(HB_FILTER_VFR)); hb_dict_set(filter_dict, "Settings", hb_value_string(filter_str)); hb_value_array_append(filter_list, filter_dict); free(filter_str); return 0; } int hb_preset_apply_video(const hb_dict_t *preset, hb_dict_t *job_dict) { hb_dict_t *dest_dict, *video_dict, *qsv; hb_value_t *value, *vcodec_value, *color_value; int mux, vcodec, vqtype; hb_encoder_t *encoder; dest_dict = hb_dict_get(job_dict, "Destination"); mux = hb_container_get_from_name(hb_value_get_string( hb_dict_get(dest_dict, "Mux"))); vcodec_value = hb_dict_get(preset, "VideoEncoder"); if (hb_value_type(vcodec_value) == HB_VALUE_TYPE_STRING) { vcodec = hb_video_encoder_get_from_name( hb_value_get_string(vcodec_value)); } else { vcodec = hb_value_get_int(vcodec_value); } encoder = hb_video_encoder_get_from_codec(vcodec); if (encoder == NULL) { char *str = hb_value_get_string_xform(vcodec_value); hb_error("Invalid video encoder (%s)", str); free(str); return -1; } if (!(encoder->muxers & mux)) { hb_error("Incompatible video encoder (%s) for muxer (%s)", hb_video_encoder_get_name(vcodec), hb_container_get_name(mux)); return -1; } video_dict = hb_dict_get(job_dict, "Video"); hb_dict_set(video_dict, "Encoder", hb_value_string(encoder->short_name)); if ((color_value = hb_dict_get(preset, "VideoColorMatrixCode")) != NULL) hb_dict_set(video_dict, "ColorMatrixCode", hb_value_dup(color_value)); hb_dict_set(video_dict, "Encoder", hb_value_dup(vcodec_value)); if (vcodec == HB_VCODEC_X264 && hb_value_get_bool(hb_dict_get(preset, "x264UseAdvancedOptions"))) { hb_dict_set(video_dict, "Options", hb_value_dup(hb_dict_get(preset, "x264Option"))); } else { if ((value = hb_dict_get(preset, "VideoPreset")) != NULL) hb_dict_set(video_dict, "Preset", hb_value_dup(value)); if ((value = hb_dict_get(preset, "VideoProfile")) != NULL) hb_dict_set(video_dict, "Profile", hb_value_dup(value)); if ((value = hb_dict_get(preset, "VideoLevel")) != NULL) hb_dict_set(video_dict, "Level", hb_value_dup(value)); if ((value = hb_dict_get(preset, "VideoTune")) != NULL) hb_dict_set(video_dict, "Tune", hb_value_dup(value)); if ((value = hb_dict_get(preset, "VideoOptionExtra")) != NULL) hb_dict_set(video_dict, "Options", hb_value_dup(value)); } vqtype = hb_value_get_int(hb_dict_get(preset, "VideoQualityType")); if (vqtype == 2) // Constant quality { hb_dict_set(video_dict, "Quality", hb_value_xform(hb_dict_get(preset, "VideoQualitySlider"), HB_VALUE_TYPE_DOUBLE)); hb_dict_set(video_dict, "Bitrate", hb_value_int(-1)); } else if (vqtype == 1) // ABR { hb_dict_set(video_dict, "Bitrate", hb_value_xform(hb_dict_get(preset, "VideoAvgBitrate"), HB_VALUE_TYPE_INT)); hb_dict_set(video_dict, "TwoPass", hb_value_xform(hb_dict_get(preset, "VideoTwoPass"), HB_VALUE_TYPE_BOOL)); hb_dict_set(video_dict, "Turbo", hb_value_xform(hb_dict_get(preset, "VideoTurboTwoPass"), HB_VALUE_TYPE_BOOL)); hb_dict_set(video_dict, "Quality", hb_value_double(-1.0)); } else { value = hb_dict_get(preset, "VideoQualitySlider"); if (value != NULL && hb_value_get_double(value) >= 0) { hb_dict_set(video_dict, "Quality", hb_value_xform(value, HB_VALUE_TYPE_DOUBLE)); hb_dict_set(video_dict, "Bitrate", hb_value_int(-1)); } else { hb_dict_set(video_dict, "Bitrate", hb_value_xform(hb_dict_get(preset, "VideoAvgBitrate"), HB_VALUE_TYPE_INT)); hb_dict_set(video_dict, "TwoPass", hb_value_xform(hb_dict_get(preset, "VideoTwoPass"), HB_VALUE_TYPE_BOOL)); hb_dict_set(video_dict, "Turbo", hb_value_xform(hb_dict_get(preset, "VideoTurboTwoPass"), HB_VALUE_TYPE_BOOL)); hb_dict_set(video_dict, "Quality", hb_value_double(-1.0)); } } qsv = hb_dict_get(video_dict, "QSV"); if (qsv == NULL) { qsv = hb_dict_init(); hb_dict_set(video_dict, "QSV", qsv); } if ((value = hb_dict_get(preset, "VideoQSVDecode")) != NULL) { hb_dict_set(qsv, "Decode", hb_value_xform(value, HB_VALUE_TYPE_BOOL)); } if ((value = hb_dict_get(preset, "VideoQSVAsyncDepth")) != NULL) { hb_dict_set(qsv, "AsyncDepth", hb_value_xform(value, HB_VALUE_TYPE_INT)); } if ((value = hb_dict_get(preset, "VideoScaler")) != NULL) { const char *s = hb_value_get_string(value); if (!strcasecmp(s, "opencl")) { hb_dict_set(video_dict, "OpenCL", hb_value_bool(1)); } } if ((value = hb_dict_get(preset, "VideoHWDecode")) != NULL) { hb_dict_set(video_dict, "HWDecode", hb_value_xform(value, HB_VALUE_TYPE_BOOL)); } return 0; } int hb_preset_apply_mux(const hb_dict_t *preset, hb_dict_t *job_dict) { hb_value_t *mux_value = hb_dict_get(preset, "FileFormat"); int mux; if (hb_value_type(mux_value) == HB_VALUE_TYPE_STRING) { mux = hb_container_get_from_name(hb_value_get_string(mux_value)); if (mux == 0) mux = hb_container_get_from_extension( hb_value_get_string(mux_value)); } else { mux = hb_value_get_int(mux_value); } hb_container_t *container = hb_container_get_from_format(mux); if (container == NULL) { char *str = hb_value_get_string_xform(mux_value); hb_error("Invalid container (%s)", str); free(str); return -1; } hb_dict_t *dest_dict = hb_dict_get(job_dict, "Destination"); hb_dict_set(dest_dict, "Mux", hb_value_string(container->short_name)); if (mux & HB_MUX_MASK_MP4) { hb_dict_t *mp4_dict = hb_dict_init(); hb_dict_set(mp4_dict, "Mp4Optimize", hb_value_xform(hb_dict_get(preset, "Mp4HttpOptimize"), HB_VALUE_TYPE_BOOL)); hb_dict_set(mp4_dict, "IpodAtom", hb_value_xform(hb_dict_get(preset, "Mp4iPodCompatible"), HB_VALUE_TYPE_BOOL)); hb_dict_set(dest_dict, "Mp4Options", mp4_dict); } return 0; } int hb_preset_apply_title(hb_handle_t *h, int title_index, const hb_dict_t *preset, hb_dict_t *job_dict) { // Apply preset settings that requires the title hb_title_t *title = hb_find_title_by_index(h, title_index); if (title == NULL) return -1; int chapters; chapters = hb_value_get_bool(hb_dict_get(preset, "ChapterMarkers")); if (title != NULL && hb_list_count(title->list_chapter) <= 1) chapters = 0; // Set "Destination" settings in job hb_dict_t *dest_dict = hb_dict_get(job_dict, "Destination"); hb_dict_set(dest_dict, "ChapterMarkers", hb_value_bool(chapters)); hb_dict_t *filters_dict = hb_dict_get(job_dict, "Filters"); hb_value_array_t *filter_list = hb_dict_get(filters_dict, "FilterList"); // Calculate default job geometry settings hb_geometry_t srcGeo, resultGeo; hb_geometry_settings_t geo; int keep_aspect; srcGeo = title->geometry; if (!hb_value_get_bool(hb_dict_get(preset, "PictureAutoCrop"))) { geo.crop[0] = hb_value_get_int(hb_dict_get(preset, "PictureTopCrop")); geo.crop[1] = hb_value_get_int(hb_dict_get(preset, "PictureBottomCrop")); geo.crop[2] = hb_value_get_int(hb_dict_get(preset, "PictureLeftCrop")); geo.crop[3] = hb_value_get_int(hb_dict_get(preset, "PictureRightCrop")); } else { memcpy(geo.crop, title->crop, sizeof(geo.crop)); } geo.modulus = hb_value_get_int(hb_dict_get(preset, "PictureModulus")); if (geo.modulus < 2) geo.modulus = 2; if (hb_value_get_bool(hb_dict_get(preset, "PictureLooseCrop"))) { // Crop a few extra pixels to avoid scaling to fit Modulus int extra1, extra2, crop_width, crop_height, width, height; crop_width = srcGeo.width - geo.crop[2] - geo.crop[3]; crop_height = srcGeo.height - geo.crop[0] - geo.crop[1]; width = MULTIPLE_MOD_DOWN(crop_width, geo.modulus); height = MULTIPLE_MOD_DOWN(crop_height, geo.modulus); extra1 = EVEN((crop_height - height) / 2); extra2 = crop_height - height - extra1; geo.crop[0] += extra1; geo.crop[1] += extra2; extra1 = EVEN((crop_width - width) / 2); extra2 = crop_width - width - extra1; geo.crop[2] += extra1; geo.crop[3] += extra2; } hb_value_t *ana_mode_value = hb_dict_get(preset, "PicturePAR"); if (hb_value_type(ana_mode_value) == HB_VALUE_TYPE_STRING) { const char *s = hb_value_get_string(ana_mode_value); if (!strcasecmp(s, "none")) geo.mode = 0; else if (!strcasecmp(s, "strict")) geo.mode = 1; else if (!strcasecmp(s, "custom")) geo.mode = 3; else // default loose geo.mode = 2; } else { geo.mode = hb_value_get_int(hb_dict_get(preset, "PicturePAR")); } keep_aspect = hb_value_get_bool(hb_dict_get(preset, "PictureKeepRatio")); if (geo.mode == HB_ANAMORPHIC_STRICT || geo.mode == HB_ANAMORPHIC_LOOSE) keep_aspect = 1; geo.keep = keep_aspect * HB_KEEP_DISPLAY_ASPECT; geo.itu_par = hb_value_get_bool(hb_dict_get(preset, "PictureItuPAR")); geo.maxWidth = hb_value_get_int(hb_dict_get(preset, "PictureWidth")); geo.maxHeight = hb_value_get_int(hb_dict_get(preset, "PictureHeight")); geo.geometry = title->geometry; int width = hb_value_get_int(hb_dict_get(preset, "PictureForceWidth")); int height = hb_value_get_int(hb_dict_get(preset, "PictureForceHeight")); if (width > 0) { geo.geometry.width = width; geo.keep |= HB_KEEP_WIDTH; } else { geo.geometry.width -= geo.crop[2] + geo.crop[3]; } if (height > 0) { geo.geometry.height = height; geo.keep |= HB_KEEP_HEIGHT; } else { geo.geometry.height -= geo.crop[0] + geo.crop[1]; } if (geo.mode == HB_ANAMORPHIC_CUSTOM && !keep_aspect) { int dar_width; dar_width = hb_value_get_int(hb_dict_get(preset, "PictureDARWidth")); if (dar_width > 0) { geo.geometry.par.num = dar_width; geo.geometry.par.den = geo.geometry.width; } else { geo.geometry.par.num = hb_value_get_int(hb_dict_get(preset, "PicturePARWidth")); geo.geometry.par.den = hb_value_get_int(hb_dict_get(preset, "PicturePARHeight")); } } hb_set_anamorphic_size2(&srcGeo, &geo, &resultGeo); hb_dict_t *par_dict = hb_dict_get(job_dict, "PAR"); hb_dict_set(par_dict, "Num", hb_value_int(resultGeo.par.num)); hb_dict_set(par_dict, "Den", hb_value_int(resultGeo.par.den)); par_dict = NULL; hb_dict_t *filter_dict; char *filter_str; // Setup scale filter filter_str = hb_strdup_printf("%d:%d:%d:%d:%d:%d", resultGeo.width, resultGeo.height, geo.crop[0], geo.crop[1], geo.crop[2], geo.crop[3]); filter_dict = hb_dict_init(); hb_dict_set(filter_dict, "ID", hb_value_int(HB_FILTER_CROP_SCALE)); hb_dict_set(filter_dict, "Settings", hb_value_string(filter_str)); free(filter_str); hb_value_array_append(filter_list, filter_dict); // Audio settings if (hb_preset_job_add_audio(h, title_index, preset, job_dict) != 0) { goto fail; } // Subtitle settings if (hb_preset_job_add_subtitles(h, title_index, preset, job_dict) != 0) { goto fail; } return 0; fail: return -1; } /** * Initialize an hb_job_t and return a hb_dict_t representation of the job. * This dict will have key/value pairs compatible with json jobs. * @param h - Pointer to hb_handle_t instance that contains the * specified title_index * @param title_index - Index of hb_title_t to use for job initialization. * Index comes from title->index or "Index" key * in json representation of a title. * @param preset - Preset to initialize job with */ hb_dict_t* hb_preset_job_init(hb_handle_t *h, int title_index, const hb_dict_t *preset) { hb_title_t *title = hb_find_title_by_index(h, title_index); if (title == NULL) { hb_error("Invalid title index (%d)", title_index); return NULL; } hb_job_t *job = hb_job_init(title); hb_dict_t *job_dict = hb_job_to_dict(job); hb_job_close(&job); if (hb_preset_apply_mux(preset, job_dict) < 0) goto fail; if (hb_preset_apply_video(preset, job_dict) < 0) goto fail; if (hb_preset_apply_filters(preset, job_dict) < 0) goto fail; if (hb_preset_apply_title(h, title_index, preset, job_dict) < 0) goto fail; return job_dict; fail: hb_value_free(&job_dict); return NULL; } // Clean a dictionary of unwanted keys // Used to make sure only valid keys are in output presets static void dict_clean(hb_value_t *dict, hb_value_t *template) { hb_value_t *tmp = hb_value_dup(dict); hb_dict_iter_t iter; const char *key; hb_value_t *val; hb_value_t *template_val; hb_value_type_t template_type, val_type; const char *preset_name = NULL; val = hb_dict_get(dict, "PresetName"); if (val != NULL) preset_name = hb_value_get_string(val); // Remove keys that are not in the template and translate compatible // data types to the types used in the template. for (iter = hb_dict_iter_init(tmp); iter != HB_DICT_ITER_DONE; iter = hb_dict_iter_next(tmp, iter)) { key = hb_dict_iter_key(iter); val = hb_dict_iter_value(iter); val_type = hb_value_type(val); template_val = hb_dict_get(template, key); template_type = hb_value_type(template_val); if (template_val == NULL) { // Unknown key. These can be keys used privately by the // frontend. So don't make noise about them. hb_dict_remove(dict, key); } else if (val_type != template_type) { if (val_type == HB_VALUE_TYPE_DICT || val_type == HB_VALUE_TYPE_ARRAY || template_type == HB_VALUE_TYPE_DICT || template_type == HB_VALUE_TYPE_ARRAY) { hb_error("Preset %s: Incompatible value types for key %s. " "Dropping.", preset_name, key); hb_dict_remove(dict, key); } else if (hb_value_is_number(val) && hb_value_is_number(template_val)) { // Silently convert compatible numbers hb_value_t *v; v = hb_value_xform(val, template_type); hb_dict_set(dict, key, v); } else { hb_value_t *v; hb_error("Preset %s: Incorrect value type for key %s. " "Converting.", preset_name, key); v = hb_value_xform(val, template_type); hb_dict_set(dict, key, v); } } else if (val_type == HB_VALUE_TYPE_DICT && template_type == HB_VALUE_TYPE_DICT) { val = hb_dict_get(dict, key); dict_clean(val, template_val); } else if (val_type == HB_VALUE_TYPE_ARRAY && template_type == HB_VALUE_TYPE_ARRAY && hb_value_array_len(template_val) > 0) { template_val = hb_value_array_get(template_val, 0); if (hb_value_type(template_val) == HB_VALUE_TYPE_DICT) { val = hb_dict_get(dict, key); int count = hb_value_array_len(val); int ii; for (ii = 0; ii < count; ii++) { hb_value_t *array_val; array_val = hb_value_array_get(val, ii); if (hb_value_type(array_val) == HB_VALUE_TYPE_DICT) { dict_clean(array_val, template_val); } } } } } hb_value_free(&tmp); if (!hb_value_get_bool(hb_dict_get(dict, "Folder"))) { // Add key/value pairs that are in the template but not in the dict. for (iter = hb_dict_iter_init(template); iter != HB_DICT_ITER_DONE; iter = hb_dict_iter_next(template, iter)) { key = hb_dict_iter_key(iter); template_val = hb_dict_iter_value(iter); val = hb_dict_get(dict, key); if (val == NULL) { hb_dict_set(dict, key, hb_value_dup(template_val)); } } } } static void preset_clean(hb_value_t *preset, hb_value_t *template) { dict_clean(preset, template); // Check for proper "short name" values. // Convert as necessary. hb_value_t *val; const char *preset_name = NULL; int muxer; val = hb_dict_get(preset, "PresetName"); if (val != NULL) preset_name = hb_value_get_string(val); val = hb_dict_get(preset, "FileFormat"); if (val != NULL) { const char *s, *mux; s = hb_value_get_string(val); muxer = hb_container_get_from_name(s); if (muxer == HB_MUX_INVALID) { const hb_container_t *c = hb_container_get_next(NULL); muxer = c->format; hb_error("Preset %s: Invalid container (%s)", preset_name, s); } mux = hb_container_get_short_name(muxer); val = hb_value_string(mux); hb_dict_set(preset, "FileFormat", val); } else { const hb_container_t *c = hb_container_get_next(NULL); muxer = c->format; } val = hb_dict_get(preset, "VideoEncoder"); if (val != NULL) { const char *s, *enc; int vcodec; s = hb_value_get_string(val); vcodec = hb_video_encoder_get_from_name(s); if (vcodec == HB_VCODEC_INVALID) { vcodec = hb_video_encoder_get_default(muxer); hb_error("Preset %s: Invalid video encoder (%s)", preset_name, s); } enc = hb_video_encoder_get_short_name(vcodec); val = hb_value_string(enc); hb_dict_set(preset, "VideoEncoder", val); } val = hb_dict_get(preset, "VideoFramerate"); if (val != NULL) { const char *s; s = hb_value_get_string(val); if (strcasecmp(s, "auto")) { int fr = hb_video_framerate_get_from_name(s); if (fr < 0) { if (strcasecmp(s, "same as source")) { hb_error("Preset %s: Invalid video framerate (%s)", preset_name, s); } val = hb_value_string("auto"); hb_dict_set(preset, "VideoFramerate", val); } } } val = hb_dict_get(preset, "AudioEncoderFallback"); if (val != NULL) { const char *s, *enc; int acodec; s = hb_value_get_string(val); acodec = hb_audio_encoder_get_from_name(s); if (acodec == HB_ACODEC_INVALID) { acodec = hb_audio_encoder_get_default(muxer); hb_error("Preset %s: Invalid audio fallback encoder (%s)", preset_name, s); } enc = hb_audio_encoder_get_short_name(acodec); val = hb_value_string(enc); hb_dict_set(preset, "AudioEncoderFallback", val); } hb_value_t *alist = hb_dict_get(preset, "AudioList"); int count = hb_value_array_len(alist); int ii; for (ii = 0; ii < count; ii++) { hb_value_t *adict = hb_value_array_get(alist, ii); val = hb_dict_get(adict, "AudioEncoder"); if (val != NULL) { const char *s, *enc; int acodec; s = hb_value_get_string(val); acodec = hb_audio_encoder_get_from_name(s); if (acodec == HB_ACODEC_INVALID) { acodec = hb_audio_encoder_get_default(muxer); hb_error("Preset %s: Invalid audio encoder (%s)", preset_name, s); } enc = hb_audio_encoder_get_short_name(acodec); val = hb_value_string(enc); hb_dict_set(adict, "AudioEncoder", val); } val = hb_dict_get(adict, "AudioSamplerate"); if (val != NULL) { const char *s; s = hb_value_get_string(val); if (strcasecmp(s, "auto")) { int sr = hb_audio_samplerate_get_from_name(s); if (sr < 0) { hb_error("Preset %s: Invalid audio samplerate (%s)", preset_name, s); val = hb_value_string("auto"); hb_dict_set(adict, "AudioSamplerate", val); } } } val = hb_dict_get(adict, "AudioMixdown"); if (val != NULL) { const char *s, *mix; s = hb_value_get_string(val); int mixdown = hb_mixdown_get_from_name(s); if (mixdown == HB_INVALID_AMIXDOWN) { // work.c do_job() sanitizes NONE to default mixdown mixdown = HB_AMIXDOWN_NONE; hb_error("Preset %s: Invalid audio mixdown (%s)", preset_name, s); } mix = hb_mixdown_get_short_name(mixdown); val = hb_value_string(mix); hb_dict_set(adict, "AudioMixdown", val); } } } static void presets_clean(hb_value_t *presets, hb_value_t *template) { preset_clean_context_t ctx; ctx.do_ctx.path.depth = 1; ctx.template = template; presets_do(do_preset_clean, presets, (preset_do_context_t*)&ctx); } void hb_presets_clean(hb_value_t *preset) { presets_clean(preset, hb_preset_template); } static const char* import_indexed_filter(int filter_id, int index) { hb_filter_param_t *filter_presets; filter_presets = hb_filter_param_get_presets(filter_id); int ii; for (ii = 0; filter_presets[ii].name != NULL; ii++) { if (filter_presets[ii].index == index) break; } return filter_presets[ii].short_name; } static void import_decomb(hb_value_t *preset) { hb_value_t *val = hb_dict_get(preset, "PictureDecomb"); if (hb_value_is_number(val)) { const char *s; int index = hb_value_get_int(val); s = import_indexed_filter(HB_FILTER_DECOMB, index); if (s != NULL) { hb_dict_set(preset, "PictureDecomb", hb_value_string(s)); } else { hb_error("Invalid decomb index %d", index); hb_dict_set(preset, "PictureDecomb", hb_value_string("off")); } } } static void import_deint(hb_value_t *preset) { hb_value_t *val = hb_dict_get(preset, "PictureDeinterlace"); if (hb_value_is_number(val)) { const char *s; int index = hb_value_get_int(val); s = import_indexed_filter(HB_FILTER_DEINTERLACE, index); if (s != NULL) { hb_dict_set(preset, "PictureDeinterlace", hb_value_string(s)); } else { hb_error("Invalid deinterlace index %d", index); hb_dict_set(preset, "PictureDeinterlace", hb_value_string("off")); } } } static void import_detel(hb_value_t *preset) { hb_value_t *val = hb_dict_get(preset, "PictureDetelecine"); if (hb_value_is_number(val)) { const char *s; int index = hb_value_get_int(val); s = import_indexed_filter(HB_FILTER_DETELECINE, index); if (s != NULL) { hb_dict_set(preset, "PictureDetelecine", hb_value_string(s)); } else { hb_error("Invalid detelecine index %d", index); hb_dict_set(preset, "PictureDetelecine", hb_value_string("off")); } } } static void import_denoise(hb_value_t *preset) { hb_value_t *val = hb_dict_get(preset, "PictureDenoise"); if (hb_value_is_number(val)) { const char *s; int index = hb_value_get_int(val); s = import_indexed_filter(HB_FILTER_HQDN3D, index); if (s != NULL) { hb_dict_set(preset, "PictureDenoiseFilter", hb_value_string("hqdn3d")); hb_dict_set(preset, "PictureDenoisePreset", hb_value_string(s)); } else { if (index != 0) hb_error("Invalid denoise index %d", index); hb_dict_set(preset, "PictureDenoiseFilter", hb_value_string("off")); } } } static void import_pic(hb_value_t *preset) { if (hb_value_get_bool(hb_dict_get(preset, "UsesMaxPictureSettings"))) { // UsesMaxPictureSettings was deprecated hb_dict_set(preset, "UsesPictureSettings", hb_value_int(2)); } hb_value_t *val = hb_dict_get(preset, "PicturePAR"); if (hb_value_is_number(val)) { const char *s; int pic_par = hb_value_get_int(val); switch (pic_par) { default: case 0: s = "off"; break; case 1: s = "strict"; break; case 2: s = "loose"; break; case 3: s = "custom"; break; } hb_dict_set(preset, "PicturePAR", hb_value_string(s)); } } static void import_audio(hb_value_t *preset) { hb_value_t *copy = hb_dict_get(preset, "AudioCopyMask"); if (copy != NULL) return; copy = hb_value_array_init(); hb_dict_set(preset, "AudioCopyMask", copy); if (hb_value_get_bool(hb_dict_get(preset, "AudioAllowMP3Pass"))) hb_value_array_append(copy, hb_value_string("copy:mp3")); if (hb_value_get_bool(hb_dict_get(preset, "AudioAllowAACPass"))) hb_value_array_append(copy, hb_value_string("copy:aac")); if (hb_value_get_bool(hb_dict_get(preset, "AudioAllowAC3Pass"))) hb_value_array_append(copy, hb_value_string("copy:ac3")); if (hb_value_get_bool(hb_dict_get(preset, "AudioAllowDTSPass"))) hb_value_array_append(copy, hb_value_string("copy:dts")); if (hb_value_get_bool(hb_dict_get(preset, "AudioAllowDTSHDPass"))) hb_value_array_append(copy, hb_value_string("copy:dtshd")); if (hb_value_get_bool(hb_dict_get(preset, "AudioAllowEAC3Pass"))) hb_value_array_append(copy, hb_value_string("copy:eac3")); if (hb_value_get_bool(hb_dict_get(preset, "AudioAllowFLACPass"))) hb_value_array_append(copy, hb_value_string("copy:flac")); if (hb_value_get_bool(hb_dict_get(preset, "AudioAllowTRUEHDPass"))) hb_value_array_append(copy, hb_value_string("copy:truehd")); } static void import_video(hb_value_t *preset) { hb_value_t *val; if ((val = hb_dict_get(preset, "x264Preset")) != NULL) hb_dict_set(preset, "VideoPreset", hb_value_dup(val)); if ((val = hb_dict_get(preset, "x264Tune")) != NULL) hb_dict_set(preset, "VideoTune", hb_value_dup(val)); if ((val = hb_dict_get(preset, "h264Profile")) != NULL) hb_dict_set(preset, "VideoProfile", hb_value_dup(val)); if ((val = hb_dict_get(preset, "h264Level")) != NULL) hb_dict_set(preset, "VideoLevel", hb_value_dup(val)); if ((val = hb_dict_get(preset, "x264OptionExtra")) != NULL) hb_dict_set(preset, "VideoOptionExtra", hb_value_dup(val)); // Remove invalid "none" tune from VideoTune. Frontends should // be removing this before saving a preset. if ((val = hb_dict_get(preset, "VideoTune")) != NULL) { const char *tune; tune = hb_value_get_string(val); // "none" is not a valid tune, but is used by HandBrake // to indicate no tune options. if (tune != NULL && !strncasecmp(tune, "none", 4)) { tune += 4; if (tune[0] == ',') { tune++; } } if (tune != NULL) { hb_dict_set(preset, "VideoTune", hb_value_string(tune)); } } if (hb_value_get_int(hb_dict_get(preset, "VideoQualityType")) == 0) { // Target size no longer supported hb_dict_set(preset, "VideoQualityType", hb_value_int(1)); } if (hb_value_get_bool(hb_dict_get(preset, "VideoFrameratePFR"))) { hb_dict_set(preset, "VideoFramerateMode", hb_value_string("pfr")); } else if (hb_value_get_bool(hb_dict_get(preset, "VideoFramerateCFR"))) { hb_dict_set(preset, "VideoFramerateMode", hb_value_string("cfr")); } else if (hb_value_get_bool(hb_dict_get(preset, "VideoFramerateVFR"))) { hb_dict_set(preset, "VideoFramerateMode", hb_value_string("vfr")); } const char *enc; int codec; enc = hb_value_get_string(hb_dict_get(preset, "VideoEncoder")); codec = hb_video_encoder_get_from_name(enc); if (codec & HB_VCODEC_FFMPEG_MASK) { if ((val = hb_dict_get(preset, "lavcOption")) != NULL) hb_dict_set(preset, "VideoOptionExtra", hb_value_dup(val)); } } static void preset_import(hb_value_t *preset, int major, int minor, int micro) { if (!hb_value_get_bool(hb_dict_get(preset, "Folder"))) { if (major == 0 && minor == 0 && micro == 0) { // Convert legacy presets (before versioning introduced) import_video(preset); import_pic(preset); import_audio(preset); import_decomb(preset); import_deint(preset); import_detel(preset); import_denoise(preset); } preset_clean(preset, hb_preset_template); } } int hb_presets_version(hb_value_t *preset, int *major, int *minor, int *micro) { *major = 0; *minor = 0; *micro = 0; if (hb_value_type(preset) == HB_VALUE_TYPE_DICT) { // Is this a single preset or a packaged collection of presets? hb_value_t *val = hb_dict_get(preset, "PresetName"); if (val == NULL) { val = hb_dict_get(preset, "VersionMajor"); if (val != NULL) { *major = hb_value_get_int(hb_dict_get(preset, "VersionMajor")); *minor = hb_value_get_int(hb_dict_get(preset, "VersionMinor")); *micro = hb_value_get_int(hb_dict_get(preset, "VersionMicro")); return 0; } } } return -1; } void hb_presets_import(hb_value_t *preset) { preset_import_context_t ctx; ctx.do_ctx.path.depth = 1; hb_presets_version(preset, &ctx.major, &ctx.minor, &ctx.micro); presets_do(do_preset_import, preset, (preset_do_context_t*)&ctx); } char * hb_presets_import_json(const char *json) { hb_value_t * dict = hb_value_json(json); if (dict == NULL) return NULL; hb_presets_import(dict); char * result = hb_value_get_json(dict); hb_value_free(&dict); return result; } char * hb_presets_clean_json(const char *json) { hb_value_t * dict = hb_value_json(json); if (dict == NULL) return NULL; presets_clean(dict, hb_preset_template); char * result = hb_value_get_json(dict); hb_value_free(&dict); return result; } // Note that unpackage does not make any copies. // In one increases the reference count. static hb_value_t * presets_unpackage(const hb_value_t *packaged_presets) { // Do any legacy translations. hb_value_t *tmp = hb_value_dup(packaged_presets); hb_presets_import(tmp); if (hb_value_type(tmp) == HB_VALUE_TYPE_ARRAY) { // Not packaged return tmp; } if (hb_dict_get(tmp, "PresetName") != NULL) { // Bare single preset return tmp; } hb_value_t *presets = hb_dict_get(tmp, "PresetList"); hb_value_incref(presets); hb_value_free(&tmp); return presets; } static hb_value_t * presets_package(const hb_value_t *presets) { hb_dict_t *packaged_presets; if (hb_value_type(presets) != HB_VALUE_TYPE_DICT || hb_dict_get(presets, "VersionMajor") == NULL) { // Preset is not packaged packaged_presets = hb_dict_init(); hb_dict_set(packaged_presets, "VersionMajor", hb_value_int(hb_preset_version_major)); hb_dict_set(packaged_presets, "VersionMinor", hb_value_int(hb_preset_version_minor)); hb_dict_set(packaged_presets, "VersionMicro", hb_value_int(hb_preset_version_micro)); // TODO: What else do we want in the preset containers header? hb_dict_t *tmp = hb_value_dup(presets); if (hb_value_type(presets) == HB_VALUE_TYPE_DICT) { hb_value_array_t *array = hb_value_array_init(); hb_value_array_append(array, tmp); tmp = array; } presets_clean(tmp, hb_preset_template); hb_dict_set(packaged_presets, "PresetList", tmp); } else { // Preset is already packaged hb_dict_t *tmp = hb_value_dup(presets); presets_clean(tmp, hb_preset_template); packaged_presets = tmp; } return packaged_presets; } void hb_presets_builtin_init(void) { hb_value_t * dict = hb_value_json(hb_builtin_presets_json); hb_value_t * template = hb_dict_get(dict, "PresetTemplate"); hb_preset_version_major = hb_value_get_int( hb_dict_get(template, "VersionMajor")); hb_preset_version_minor = hb_value_get_int( hb_dict_get(template, "VersionMinor")); hb_preset_version_micro = hb_value_get_int( hb_dict_get(template, "VersionMicro")); hb_preset_template = hb_value_dup(hb_dict_get(template, "Preset")); hb_presets_builtin = hb_value_dup(hb_dict_get(dict, "PresetBuiltin")); hb_presets_clean(hb_presets_builtin); hb_presets = hb_value_array_init(); hb_value_free(&dict); } void hb_presets_current_version(int *major, int* minor, int *micro) { *major = hb_preset_version_major; *minor = hb_preset_version_minor; *micro = hb_preset_version_micro; } int hb_presets_gui_init(void) { char path[1024]; hb_value_t * dict = NULL; #if defined(HB_PRESET_JSON_FILE) hb_get_user_config_filename(path, "%s", HB_PRESET_JSON_FILE); dict = hb_value_read_json(path); #endif #if defined(HB_PRESET_PLIST_FILE) if (dict == NULL) { hb_get_user_config_filename(path, "%s", HB_PRESET_PLIST_FILE); dict = hb_plist_parse_file(path); } #endif if (dict == NULL) { hb_error("Failed to load GUI presets file"); #if defined(HB_PRESET_JSON_FILE) hb_error("Attempted: %s", HB_PRESET_JSON_FILE); #endif #if defined(HB_PRESET_PLIST_FILE) hb_error("Attempted: %s", HB_PRESET_PLIST_FILE); #endif return -1; } else { preset_do_context_t ctx; ctx.path.depth = 1; presets_do(do_delete_builtin, dict, &ctx); int result = hb_presets_add(dict); hb_value_free(&dict); return result; } return -1; } hb_value_t * hb_presets_builtin_get(void) { return hb_value_dup(hb_presets_builtin); } char * hb_presets_builtin_get_json(void) { char *json = hb_value_get_json(hb_presets_builtin); return json; } // Lookup a preset in the preset list. The "name" may contain '/' // separators to explicitely specify a preset within the preset lists // folder structure. // // If 'recurse' is specified, a recursive search for the first component // in the name will be performed. // // I assume that the actual preset name does not include any '/' // // A reference to the preset is returned static hb_preset_index_t * preset_lookup_path(const char *name, int recurse) { preset_search_context_t ctx; int result; ctx.do_ctx.path.depth = 1; ctx.name = name; ctx.recurse = recurse; ctx.last_match_idx = -1; result = presets_do(do_preset_search, hb_presets, (preset_do_context_t*)&ctx); if (result != PRESET_DO_SUCCESS) ctx.do_ctx.path.depth = 0; return hb_preset_index_dup(&ctx.do_ctx.path); } // Lookup a preset in the preset list. The "name" may contain '/' // separators to explicitely specify a preset within the preset lists // folder structure. // // If 'recurse' is specified, a recursive search for the first component // in the name will be performed. // // I assume that the actual preset name does not include any '/' // // A copy of the preset is returned hb_preset_index_t * hb_preset_search_index(const char *name, int recurse) { return preset_lookup_path(name, recurse); } hb_value_t * hb_preset_search(const char *name, int recurse) { hb_preset_index_t *path = preset_lookup_path(name, recurse); hb_value_t *preset = hb_preset_get(path); free(path); return preset; } char * hb_preset_search_json(const char *name, int recurse) { hb_value_t * preset; char *json; preset = hb_preset_search(name, recurse); if (preset == NULL) return NULL; json = hb_value_get_json(preset); return json; } static hb_preset_index_t * lookup_default_index(hb_value_t *list) { preset_do_context_t ctx; int result; ctx.path.depth = 1; result = presets_do(do_find_default, list, &ctx); if (result != PRESET_DO_SUCCESS) ctx.path.depth = 0; return hb_preset_index_dup(&ctx.path); } hb_preset_index_t * hb_presets_get_default_index(void) { hb_preset_index_t *path = lookup_default_index(hb_presets); return path; } hb_dict_t * hb_presets_get_default(void) { hb_preset_index_t *path = hb_presets_get_default_index(); return hb_preset_get(path); } char * hb_presets_get_default_json(void) { // Look for default preset hb_value_t *def = hb_presets_get_default(); return hb_value_get_json(def); } void hb_presets_clear_default() { preset_do_context_t ctx; ctx.path.depth = 1; presets_do(do_clear_default, hb_presets, &ctx); } void hb_presets_builtin_update(void) { preset_do_context_t ctx; hb_preset_index_t *path; hb_value_t *builtin; int ii; ctx.path.depth = 1; presets_do(do_delete_builtin, hb_presets, &ctx); builtin = hb_value_dup(hb_presets_builtin); path = lookup_default_index(hb_presets); if (path != NULL && path->depth != 0) { // The "Default" preset is an existing custom preset. // Clear the default preset in builtins ctx.path.depth = 1; presets_do(do_clear_default, builtin, &ctx); } free(path); for (ii = hb_value_array_len(builtin) - 1; ii >= 0; ii--) { hb_value_t *dict; dict = hb_value_array_get(builtin, ii); hb_value_incref(dict); hb_value_array_insert(hb_presets, 0, dict); } hb_value_free(&builtin); } int hb_presets_add(hb_value_t *preset) { hb_preset_index_t *path; int added = 0; if (preset == NULL) return -1; preset = presets_unpackage(preset); if (preset == NULL) return -1; path = lookup_default_index(preset); if (path != NULL && path->depth != 0) { // There is a "Default" preset in the preset(s) being added. // Clear any existing default preset. hb_presets_clear_default(); } free(path); int index = hb_value_array_len(hb_presets); if (hb_value_type(preset) == HB_VALUE_TYPE_DICT) { // A standalone preset or folder of presets. Add to preset array. hb_value_array_append(hb_presets, hb_value_dup(preset)); added++; } else if (hb_value_type(preset) == HB_VALUE_TYPE_ARRAY) { // An array of presets. Add each element. int count = hb_value_array_len(preset); int ii; for (ii = 0; ii < count; ii++) { hb_value_t *value = hb_value_array_get(preset, ii); hb_value_array_append(hb_presets, hb_value_dup(value)); added++; } } hb_value_free(&preset); if (added == 0) { return -1; } return index; } int hb_presets_add_json(const char *json) { hb_value_t *preset = hb_value_json(json); if (preset == NULL) return -1; int result = hb_presets_add(preset); hb_value_free(&preset); return result; } int hb_presets_version_file(const char *filename, int *major, int *minor, int *micro) { int result; hb_value_t *preset = hb_value_read_json(filename); if (preset == NULL) preset = hb_plist_parse_file(filename); if (preset == NULL) return -1; result = hb_presets_version(preset, major, minor, micro); hb_value_free(&preset); return result; } hb_value_t* hb_presets_read_file(const char *filename) { hb_value_t *preset = hb_value_read_json(filename); if (preset == NULL) preset = hb_plist_parse_file(filename); if (preset == NULL) return NULL; return preset; } char * hb_presets_read_file_json(const char *filename) { char *result; hb_value_t *preset = hb_value_read_json(filename); if (preset == NULL) preset = hb_plist_parse_file(filename); if (preset == NULL) return NULL; result = hb_value_get_json(preset); return result; } int hb_presets_add_file(const char *filename) { hb_value_t *preset = hb_value_read_json(filename); if (preset == NULL) preset = hb_plist_parse_file(filename); if (preset == NULL) return -1; int result = hb_presets_add(preset); hb_value_free(&preset); return result; } static int compare_str(const void *a, const void *b) { return strncmp(*(const char**)a, *(const char**)b, PATH_MAX); } int hb_presets_add_path(char * path) { hb_stat_t sb; HB_DIR * dir; struct dirent * entry; char * filename; int count, ii; char ** files; int result = -1; if (hb_stat(path, &sb)) return -1; if (S_ISREG(sb.st_mode)) { return hb_presets_add_file(path); } if (!S_ISDIR(sb.st_mode)) return -1; dir = hb_opendir(path); if ( dir == NULL ) return -1; // Count the total number of entries count = 0; while ((entry = hb_readdir(dir))) { count++; } files = malloc(count * sizeof(char*)); // Find all regular files ii = 0; hb_rewinddir(dir); while ((entry = hb_readdir(dir))) { filename = hb_strdup_printf("%s" DIR_SEP_STR "%s", path, entry->d_name); if (hb_stat(filename, &sb)) { free(filename); continue; } // Only load regular files if (!S_ISREG(sb.st_mode)) { free(filename); continue; } // Only load files with .json extension if (strcmp(".json", filename + strlen(filename) - 5)) { free(filename); continue; } files[ii++] = filename; } count = ii; // Sort the files so presets get added in a consistent order qsort(files, count, sizeof(char*), compare_str); // Add preset files to preset list for (ii = 0; ii < count; ii++) { int res = hb_presets_add_file(files[ii]); // return success if any one of the files is successfully loaded if (res >= 0) result = res; } hb_closedir( dir ); free(files); return result; } hb_value_t * hb_presets_get(void) { return hb_presets; } char * hb_presets_get_json(void) { char * result; hb_value_t *presets = hb_presets_get(); result = hb_value_get_json(presets); return result; } int hb_presets_write_json(const hb_value_t *preset, const char *path) { hb_value_t *packaged_preset = presets_package(preset); // Packaging does some validity checks and can fail if (packaged_preset == NULL) return -1; int result = hb_value_write_json(packaged_preset, path); hb_value_free(&packaged_preset); return result; } char * hb_presets_package_json(const hb_value_t *preset) { hb_value_t *packaged_preset = presets_package(preset); // Packaging does some validity checks and can fail if (packaged_preset == NULL) return NULL; char *out_json = hb_value_get_json(packaged_preset); hb_value_free(&packaged_preset); return out_json; } char * hb_presets_json_package(const char *in_json) { hb_value_t *preset = hb_value_json(in_json); hb_value_t *packaged_preset = presets_package(preset); // Packaging does some validity checks and can fail if (packaged_preset == NULL) return NULL; char *out_json = hb_value_get_json(packaged_preset); hb_value_free(&packaged_preset); hb_value_free(&preset); return out_json; } void hb_presets_free(void) { hb_value_free(&hb_preset_template); hb_value_free(&hb_presets); hb_value_free(&hb_presets_builtin); } hb_value_t * hb_presets_get_folder_children(const hb_preset_index_t *path) { int ii, count, folder; hb_value_t *dict; if (path == NULL) return hb_presets; hb_value_t *presets = hb_presets; for (ii = 0; ii < path->depth; ii++) { count = hb_value_array_len(presets); if (path->index[ii] >= count) return NULL; dict = hb_value_array_get(presets, path->index[ii]); folder = hb_value_get_bool(hb_dict_get(dict, "Folder")); if (!folder) break; presets = hb_dict_get(dict, "ChildrenArray"); } if (ii < path->depth) return NULL; return presets; } hb_value_t * hb_preset_get(const hb_preset_index_t *path) { hb_value_t *folder = NULL; if (path == NULL || path->depth <= 0) return NULL; hb_preset_index_t folder_path = *path; folder_path.depth--; folder = hb_presets_get_folder_children(&folder_path); if (folder) { if (hb_value_array_len(folder) <= path->index[path->depth-1]) { hb_error("hb_preset_get: not found"); } else { return hb_value_array_get(folder, path->index[path->depth-1]); } } else { hb_error("hb_preset_get: not found"); } return NULL; } int hb_preset_set(const hb_preset_index_t *path, const hb_value_t *dict) { hb_value_t *folder = NULL; if (dict == NULL || path == NULL || path->depth <= 0) return -1; hb_preset_index_t folder_path = *path; folder_path.depth--; folder = hb_presets_get_folder_children(&folder_path); if (folder) { if (hb_value_array_len(folder) <= path->index[path->depth-1]) { hb_error("hb_preset_replace: not found"); return -1; } else { hb_value_t *dup = hb_value_dup(dict); presets_clean(dup, hb_preset_template); hb_value_array_set(folder, path->index[path->depth-1], dup); } } else { hb_error("hb_preset_replace: not found"); return -1; } return 0; } int hb_preset_insert(const hb_preset_index_t *path, const hb_value_t *dict) { hb_value_t *folder = NULL; if (dict == NULL || path == NULL || path->depth < 0) return -1; int index = path->index[path->depth - 1]; hb_preset_index_t folder_path = *path; folder_path.depth--; folder = hb_presets_get_folder_children(&folder_path); if (folder) { hb_value_t *dup = hb_value_dup(dict); presets_clean(dup, hb_preset_template); if (hb_value_array_len(folder) <= index) { index = hb_value_array_len(folder); hb_value_array_append(folder, dup); } else { hb_value_array_insert(folder, index, dup); } } else { hb_error("hb_preset_insert: not found"); return -1; } return index; } int hb_preset_append(const hb_preset_index_t *path, const hb_value_t *dict) { hb_value_t *folder = NULL; if (dict == NULL) return -1; folder = hb_presets_get_folder_children(path); if (folder) { int index; hb_value_t *dup = hb_value_dup(dict); presets_clean(dup, hb_preset_template); index = hb_value_array_len(folder); hb_value_array_append(folder, dup); return index; } else { hb_error("hb_preset_append: not found"); return -1; } return 0; } int hb_preset_delete(const hb_preset_index_t *path) { hb_value_t *folder = NULL; if (path == NULL) return -1; hb_preset_index_t folder_path = *path; folder_path.depth--; folder = hb_presets_get_folder_children(&folder_path); if (folder) { if (hb_value_array_len(folder) <= path->index[path->depth-1]) { hb_error("hb_preset_delete: not found"); return -1; } else { hb_value_array_remove(folder, path->index[path->depth-1]); } } else { hb_error("hb_preset_delete: not found"); return -1; } return 0; } int hb_preset_move(const hb_preset_index_t *src_path, const hb_preset_index_t *dst_path) { hb_value_t *src_folder = NULL; hb_value_t *dst_folder = NULL; hb_preset_index_t src_folder_path = *src_path; hb_preset_index_t dst_folder_path = *dst_path; src_folder_path.depth--; dst_folder_path.depth--; src_folder = hb_presets_get_folder_children(&src_folder_path); dst_folder = hb_presets_get_folder_children(&dst_folder_path); if (src_folder == NULL || dst_folder == NULL) { hb_error("hb_preset_move: not found"); return -1; } hb_value_t *dict; int src_index, dst_index; src_index = src_path->index[src_path->depth-1]; dst_index = dst_path->index[src_path->depth-1]; dict = hb_value_array_get(src_folder, src_index); hb_value_incref(dict); hb_value_array_remove(src_folder, src_index); // Be careful about indexes in the case that they are in the same folder if (src_folder == dst_folder && src_index < dst_index) dst_index--; if (hb_value_array_len(dst_folder) <= dst_index) hb_value_array_append(dst_folder, dict); else hb_value_array_insert(dst_folder, dst_index, dict); return 0; }