#include #include #include #include #include #include #include #define S_RESOLUTION "resolution" #define S_SAMPLING "sampling" #define S_UNDISTORT "undistort" #define T_RESOLUTION obs_module_text("Resolution") #define T_NONE obs_module_text("None") #define T_SAMPLING obs_module_text("ScaleFiltering") #define T_SAMPLING_POINT obs_module_text("ScaleFiltering.Point") #define T_SAMPLING_BILINEAR obs_module_text("ScaleFiltering.Bilinear") #define T_SAMPLING_BICUBIC obs_module_text("ScaleFiltering.Bicubic") #define T_SAMPLING_LANCZOS obs_module_text("ScaleFiltering.Lanczos") #define T_SAMPLING_AREA obs_module_text("ScaleFiltering.Area") #define T_UNDISTORT obs_module_text("UndistortCenter") #define T_BASE obs_module_text("Base.Canvas") #define S_SAMPLING_POINT "point" #define S_SAMPLING_BILINEAR "bilinear" #define S_SAMPLING_BICUBIC "bicubic" #define S_SAMPLING_LANCZOS "lanczos" #define S_SAMPLING_AREA "area" struct scale_filter_data { obs_source_t *context; gs_effect_t *effect; gs_eparam_t *image_param; gs_eparam_t *dimension_param; gs_eparam_t *undistort_factor_param; struct vec2 dimension_i; double undistort_factor; int cx_in; int cy_in; int cx_out; int cy_out; enum obs_scale_type sampling; gs_samplerstate_t *point_sampler; bool aspect_ratio_only; bool target_valid; bool valid; bool undistort; bool base_canvas_resolution; }; static const char *scale_filter_name(void *unused) { UNUSED_PARAMETER(unused); return obs_module_text("ScaleFilter"); } static void scale_filter_update(void *data, obs_data_t *settings) { struct scale_filter_data *filter = data; int ret; const char *res_str = obs_data_get_string(settings, S_RESOLUTION); const char *sampling = obs_data_get_string(settings, S_SAMPLING); filter->valid = true; filter->base_canvas_resolution = false; if (strcmp(res_str, T_BASE) == 0) { struct obs_video_info ovi; obs_get_video_info(&ovi); filter->aspect_ratio_only = false; filter->base_canvas_resolution = true; filter->cx_in = ovi.base_width; filter->cy_in = ovi.base_height; } else { ret = sscanf(res_str, "%dx%d", &filter->cx_in, &filter->cy_in); if (ret == 2) { filter->aspect_ratio_only = false; } else { ret = sscanf(res_str, "%d:%d", &filter->cx_in, &filter->cy_in); if (ret != 2) { filter->valid = false; return; } filter->aspect_ratio_only = true; } } if (astrcmpi(sampling, S_SAMPLING_POINT) == 0) { filter->sampling = OBS_SCALE_POINT; } else if (astrcmpi(sampling, S_SAMPLING_BILINEAR) == 0) { filter->sampling = OBS_SCALE_BILINEAR; } else if (astrcmpi(sampling, S_SAMPLING_LANCZOS) == 0) { filter->sampling = OBS_SCALE_LANCZOS; } else if (astrcmpi(sampling, S_SAMPLING_AREA) == 0) { filter->sampling = OBS_SCALE_AREA; } else { /* S_SAMPLING_BICUBIC */ filter->sampling = OBS_SCALE_BICUBIC; } filter->undistort = obs_data_get_bool(settings, S_UNDISTORT); } static void scale_filter_destroy(void *data) { struct scale_filter_data *filter = data; obs_enter_graphics(); gs_samplerstate_destroy(filter->point_sampler); obs_leave_graphics(); bfree(data); } static void *scale_filter_create(obs_data_t *settings, obs_source_t *context) { struct scale_filter_data *filter = bzalloc(sizeof(struct scale_filter_data)); struct gs_sampler_info sampler_info = {0}; filter->context = context; obs_enter_graphics(); filter->point_sampler = gs_samplerstate_create(&sampler_info); obs_leave_graphics(); scale_filter_update(filter, settings); return filter; } static void scale_filter_tick(void *data, float seconds) { struct scale_filter_data *filter = data; enum obs_base_effect type; obs_source_t *target; bool lower_than_2x; double cx_f; double cy_f; int cx; int cy; if (filter->base_canvas_resolution) { struct obs_video_info ovi; obs_get_video_info(&ovi); filter->cx_in = ovi.base_width; filter->cy_in = ovi.base_height; } target = obs_filter_get_target(filter->context); filter->cx_out = 0; filter->cy_out = 0; filter->target_valid = !!target; if (!filter->target_valid) return; cx = obs_source_get_base_width(target); cy = obs_source_get_base_height(target); if (!cx || !cy) { filter->target_valid = false; return; } filter->cx_out = cx; filter->cy_out = cy; if (!filter->valid) return; /* ------------------------- */ cx_f = (double)cx; cy_f = (double)cy; double old_aspect = cx_f / cy_f; double new_aspect = (double)filter->cx_in / (double)filter->cy_in; if (filter->aspect_ratio_only) { if (fabs(old_aspect - new_aspect) <= EPSILON) { filter->target_valid = false; return; } else { if (new_aspect > old_aspect) { filter->cx_out = (int)(cy_f * new_aspect); filter->cy_out = cy; } else { filter->cx_out = cx; filter->cy_out = (int)(cx_f / new_aspect); } } } else { filter->cx_out = filter->cx_in; filter->cy_out = filter->cy_in; } vec2_set(&filter->dimension_i, 1.0f / (float)cx, 1.0f / (float)cy); if (filter->undistort) { filter->undistort_factor = new_aspect / old_aspect; } else { filter->undistort_factor = 1.0; } /* ------------------------- */ lower_than_2x = filter->cx_out < cx / 2 || filter->cy_out < cy / 2; if (lower_than_2x && filter->sampling != OBS_SCALE_POINT) { type = OBS_EFFECT_BILINEAR_LOWRES; } else { switch (filter->sampling) { default: case OBS_SCALE_POINT: case OBS_SCALE_BILINEAR: type = OBS_EFFECT_DEFAULT; break; case OBS_SCALE_BICUBIC: type = OBS_EFFECT_BICUBIC; break; case OBS_SCALE_LANCZOS: type = OBS_EFFECT_LANCZOS; break; case OBS_SCALE_AREA: type = OBS_EFFECT_AREA; break; } } filter->effect = obs_get_base_effect(type); filter->image_param = gs_effect_get_param_by_name(filter->effect, "image"); if (type != OBS_EFFECT_DEFAULT) { filter->dimension_param = gs_effect_get_param_by_name( filter->effect, "base_dimension_i"); } else { filter->dimension_param = NULL; } if (type == OBS_EFFECT_BICUBIC || type == OBS_EFFECT_LANCZOS) { filter->undistort_factor_param = gs_effect_get_param_by_name( filter->effect, "undistort_factor"); } else { filter->undistort_factor_param = NULL; } UNUSED_PARAMETER(seconds); } static void scale_filter_render(void *data, gs_effect_t *effect) { struct scale_filter_data *filter = data; const char *technique = filter->undistort ? "DrawUndistort" : "Draw"; if (!filter->valid || !filter->target_valid) { obs_source_skip_video_filter(filter->context); return; } if (!obs_source_process_filter_begin(filter->context, GS_RGBA, OBS_NO_DIRECT_RENDERING)) return; if (filter->dimension_param) gs_effect_set_vec2(filter->dimension_param, &filter->dimension_i); if (filter->undistort_factor_param) gs_effect_set_float(filter->undistort_factor_param, (float)filter->undistort_factor); if (filter->sampling == OBS_SCALE_POINT) gs_effect_set_next_sampler(filter->image_param, filter->point_sampler); obs_source_process_filter_tech_end(filter->context, filter->effect, filter->cx_out, filter->cy_out, technique); UNUSED_PARAMETER(effect); } static const double downscale_vals[] = { 1.0, 1.25, (1.0/0.75), 1.5, (1.0/0.6), 1.75, 2.0, 2.25, 2.5, 2.75, 3.0 }; #define NUM_DOWNSCALES (sizeof(downscale_vals) / sizeof(double)) static const char *aspects[] = { "16:9", "16:10", "4:3", "1:1" }; #define NUM_ASPECTS (sizeof(aspects) / sizeof(const char *)) static bool sampling_modified(obs_properties_t *props, obs_property_t *p, obs_data_t *settings) { const char *sampling = obs_data_get_string(settings, S_SAMPLING); bool has_undistort; if (astrcmpi(sampling, S_SAMPLING_POINT) == 0) { has_undistort = false; } else if (astrcmpi(sampling, S_SAMPLING_BILINEAR) == 0) { has_undistort = false; } else if (astrcmpi(sampling, S_SAMPLING_LANCZOS) == 0) { has_undistort = true; } else if (astrcmpi(sampling, S_SAMPLING_AREA) == 0) { has_undistort = false; } else { /* S_SAMPLING_BICUBIC */ has_undistort = true; } obs_property_set_visible(obs_properties_get(props, S_UNDISTORT), has_undistort); UNUSED_PARAMETER(p); return true; } static obs_properties_t *scale_filter_properties(void *data) { obs_properties_t *props = obs_properties_create(); struct obs_video_info ovi; obs_property_t *p; uint32_t cx; uint32_t cy; struct { int cx; int cy; } downscales[NUM_DOWNSCALES]; /* ----------------- */ obs_get_video_info(&ovi); cx = ovi.base_width; cy = ovi.base_height; for (size_t i = 0; i < NUM_DOWNSCALES; i++) { downscales[i].cx = (int)((double)cx / downscale_vals[i]); downscales[i].cy = (int)((double)cy / downscale_vals[i]); } p = obs_properties_add_list(props, S_SAMPLING, T_SAMPLING, OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_STRING); obs_property_set_modified_callback(p, sampling_modified); obs_property_list_add_string(p, T_SAMPLING_POINT, S_SAMPLING_POINT); obs_property_list_add_string(p, T_SAMPLING_BILINEAR, S_SAMPLING_BILINEAR); obs_property_list_add_string(p, T_SAMPLING_BICUBIC, S_SAMPLING_BICUBIC); obs_property_list_add_string(p, T_SAMPLING_LANCZOS, S_SAMPLING_LANCZOS); obs_property_list_add_string(p, T_SAMPLING_AREA, S_SAMPLING_AREA); /* ----------------- */ p = obs_properties_add_list(props, S_RESOLUTION, T_RESOLUTION, OBS_COMBO_TYPE_EDITABLE, OBS_COMBO_FORMAT_STRING); obs_property_list_add_string(p, T_NONE, T_NONE); obs_property_list_add_string(p, T_BASE, T_BASE); for (size_t i = 0; i < NUM_ASPECTS; i++) obs_property_list_add_string(p, aspects[i], aspects[i]); for (size_t i = 0; i < NUM_DOWNSCALES; i++) { char str[32]; snprintf(str, 32, "%dx%d", downscales[i].cx, downscales[i].cy); obs_property_list_add_string(p, str, str); } obs_properties_add_bool(props, S_UNDISTORT, T_UNDISTORT); /* ----------------- */ UNUSED_PARAMETER(data); return props; } static void scale_filter_defaults(obs_data_t *settings) { obs_data_set_default_string(settings, S_SAMPLING, S_SAMPLING_BICUBIC); obs_data_set_default_string(settings, S_RESOLUTION, T_NONE); obs_data_set_default_bool(settings, S_UNDISTORT, 0); } static uint32_t scale_filter_width(void *data) { struct scale_filter_data *filter = data; return (uint32_t)filter->cx_out; } static uint32_t scale_filter_height(void *data) { struct scale_filter_data *filter = data; return (uint32_t)filter->cy_out; } struct obs_source_info scale_filter = { .id = "scale_filter", .type = OBS_SOURCE_TYPE_FILTER, .output_flags = OBS_SOURCE_VIDEO, .get_name = scale_filter_name, .create = scale_filter_create, .destroy = scale_filter_destroy, .video_tick = scale_filter_tick, .video_render = scale_filter_render, .update = scale_filter_update, .get_properties = scale_filter_properties, .get_defaults = scale_filter_defaults, .get_width = scale_filter_width, .get_height = scale_filter_height };