123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407 |
- #include <obs-module.h>
- #include <graphics/image-file.h>
- #include <util/threading.h>
- #include <util/platform.h>
- #include <util/dstr.h>
- #include <sys/stat.h>
- #define blog(log_level, format, ...) \
- blog(log_level, "[image_source: '%s'] " format, obs_source_get_name(context->source), ##__VA_ARGS__)
- #define debug(format, ...) blog(LOG_DEBUG, format, ##__VA_ARGS__)
- #define info(format, ...) blog(LOG_INFO, format, ##__VA_ARGS__)
- #define warn(format, ...) blog(LOG_WARNING, format, ##__VA_ARGS__)
- struct image_source {
- obs_source_t *source;
- char *file;
- bool persistent;
- bool is_slide;
- bool linear_alpha;
- time_t file_timestamp;
- float update_time_elapsed;
- uint64_t last_time;
- bool active;
- bool restart_gif;
- volatile bool file_decoded;
- volatile bool texture_loaded;
- gs_image_file4_t if4;
- };
- static time_t get_modified_timestamp(const char *filename)
- {
- struct stat stats;
- if (os_stat(filename, &stats) != 0)
- return -1;
- return stats.st_mtime;
- }
- static const char *image_source_get_name(void *unused)
- {
- UNUSED_PARAMETER(unused);
- return obs_module_text("ImageInput");
- }
- void image_source_preload_image(void *data)
- {
- struct image_source *context = data;
- if (os_atomic_load_bool(&context->file_decoded))
- return;
- context->file_timestamp = get_modified_timestamp(context->file);
- gs_image_file4_init(&context->if4, context->file,
- context->linear_alpha ? GS_IMAGE_ALPHA_PREMULTIPLY_SRGB : GS_IMAGE_ALPHA_PREMULTIPLY);
- os_atomic_set_bool(&context->file_decoded, true);
- }
- static void image_source_load_texture(void *data)
- {
- struct image_source *context = data;
- if (os_atomic_load_bool(&context->texture_loaded))
- return;
- debug("loading texture '%s'", context->file);
- obs_enter_graphics();
- gs_image_file4_init_texture(&context->if4);
- obs_leave_graphics();
- if (!context->if4.image3.image2.image.loaded)
- warn("failed to load texture '%s'", context->file);
- context->update_time_elapsed = 0;
- os_atomic_set_bool(&context->texture_loaded, true);
- }
- static void image_source_unload(void *data)
- {
- struct image_source *context = data;
- os_atomic_set_bool(&context->file_decoded, false);
- os_atomic_set_bool(&context->texture_loaded, false);
- obs_enter_graphics();
- gs_image_file4_free(&context->if4);
- obs_leave_graphics();
- }
- static void image_source_load(struct image_source *context)
- {
- image_source_unload(context);
- if (context->file && *context->file) {
- image_source_preload_image(context);
- image_source_load_texture(context);
- }
- }
- static void image_source_update(void *data, obs_data_t *settings)
- {
- struct image_source *context = data;
- const char *file = obs_data_get_string(settings, "file");
- const bool unload = obs_data_get_bool(settings, "unload");
- const bool linear_alpha = obs_data_get_bool(settings, "linear_alpha");
- const bool is_slide = obs_data_get_bool(settings, "is_slide");
- if (context->file)
- bfree(context->file);
- context->file = bstrdup(file);
- context->persistent = !unload;
- context->linear_alpha = linear_alpha;
- context->is_slide = is_slide;
- if (is_slide)
- return;
- /* Load the image if the source is persistent or showing */
- if (context->persistent || obs_source_showing(context->source))
- image_source_load(data);
- else
- image_source_unload(data);
- }
- static void image_source_defaults(obs_data_t *settings)
- {
- obs_data_set_default_bool(settings, "unload", false);
- obs_data_set_default_bool(settings, "linear_alpha", false);
- }
- static void image_source_show(void *data)
- {
- struct image_source *context = data;
- if (!context->persistent && !context->is_slide)
- image_source_load(context);
- }
- static void image_source_hide(void *data)
- {
- struct image_source *context = data;
- if (!context->persistent && !context->is_slide)
- image_source_unload(context);
- }
- static void restart_gif(void *data)
- {
- struct image_source *context = data;
- if (context->if4.image3.image2.image.is_animated_gif) {
- context->if4.image3.image2.image.cur_frame = 0;
- context->if4.image3.image2.image.cur_loop = 0;
- context->if4.image3.image2.image.cur_time = 0;
- obs_enter_graphics();
- gs_image_file4_update_texture(&context->if4);
- obs_leave_graphics();
- context->restart_gif = false;
- }
- }
- static void image_source_activate(void *data)
- {
- struct image_source *context = data;
- context->restart_gif = true;
- }
- static void *image_source_create(obs_data_t *settings, obs_source_t *source)
- {
- struct image_source *context = bzalloc(sizeof(struct image_source));
- context->source = source;
- image_source_update(context, settings);
- return context;
- }
- static void image_source_destroy(void *data)
- {
- struct image_source *context = data;
- image_source_unload(context);
- if (context->file)
- bfree(context->file);
- bfree(context);
- }
- static uint32_t image_source_getwidth(void *data)
- {
- struct image_source *context = data;
- return context->if4.image3.image2.image.cx;
- }
- static uint32_t image_source_getheight(void *data)
- {
- struct image_source *context = data;
- return context->if4.image3.image2.image.cy;
- }
- static void image_source_render(void *data, gs_effect_t *effect)
- {
- struct image_source *context = data;
- if (!os_atomic_load_bool(&context->texture_loaded))
- return;
- struct gs_image_file *const image = &context->if4.image3.image2.image;
- gs_texture_t *const texture = image->texture;
- if (!texture)
- return;
- const bool previous = gs_framebuffer_srgb_enabled();
- gs_enable_framebuffer_srgb(true);
- gs_blend_state_push();
- gs_blend_function(GS_BLEND_ONE, GS_BLEND_INVSRCALPHA);
- gs_eparam_t *const param = gs_effect_get_param_by_name(effect, "image");
- gs_effect_set_texture_srgb(param, texture);
- gs_draw_sprite(texture, 0, image->cx, image->cy);
- gs_blend_state_pop();
- gs_enable_framebuffer_srgb(previous);
- }
- static void image_source_tick(void *data, float seconds)
- {
- struct image_source *context = data;
- if (!os_atomic_load_bool(&context->texture_loaded)) {
- if (os_atomic_load_bool(&context->file_decoded))
- image_source_load_texture(context);
- else
- return;
- }
- uint64_t frame_time = obs_get_video_frame_time();
- context->update_time_elapsed += seconds;
- if (obs_source_showing(context->source)) {
- if (context->update_time_elapsed >= 1.0f) {
- time_t t = get_modified_timestamp(context->file);
- context->update_time_elapsed = 0.0f;
- if (context->file_timestamp != t) {
- image_source_load(context);
- }
- }
- }
- if (obs_source_showing(context->source)) {
- if (!context->active) {
- if (context->if4.image3.image2.image.is_animated_gif)
- context->last_time = frame_time;
- context->active = true;
- }
- if (context->restart_gif)
- restart_gif(context);
- } else {
- if (context->active) {
- restart_gif(context);
- context->active = false;
- }
- return;
- }
- if (context->last_time && context->if4.image3.image2.image.is_animated_gif) {
- uint64_t elapsed = frame_time - context->last_time;
- bool updated = gs_image_file4_tick(&context->if4, elapsed);
- if (updated) {
- obs_enter_graphics();
- gs_image_file4_update_texture(&context->if4);
- obs_leave_graphics();
- }
- }
- context->last_time = frame_time;
- }
- static const char *image_filter =
- #ifdef _WIN32
- "All formats (*.bmp *.tga *.png *.jpeg *.jpg *.jxr *.gif *.psd *.webp);;"
- #else
- "All formats (*.bmp *.tga *.png *.jpeg *.jpg *.gif *.psd *.webp);;"
- #endif
- "BMP Files (*.bmp);;"
- "Targa Files (*.tga);;"
- "PNG Files (*.png);;"
- "JPEG Files (*.jpeg *.jpg);;"
- #ifdef _WIN32
- "JXR Files (*.jxr);;"
- #endif
- "GIF Files (*.gif);;"
- "PSD Files (*.psd);;"
- "WebP Files (*.webp);;"
- "All Files (*.*)";
- static obs_properties_t *image_source_properties(void *data)
- {
- UNUSED_PARAMETER(data);
- obs_properties_t *props = obs_properties_create();
- obs_properties_add_path(props, "file", obs_module_text("File"), OBS_PATH_FILE, image_filter, NULL);
- obs_properties_add_bool(props, "unload", obs_module_text("UnloadWhenNotShowing"));
- obs_properties_add_bool(props, "linear_alpha", obs_module_text("LinearAlpha"));
- return props;
- }
- uint64_t image_source_get_memory_usage(void *data)
- {
- struct image_source *s = data;
- return s->if4.image3.image2.mem_usage;
- }
- static void missing_file_callback(void *src, const char *new_path, void *data)
- {
- struct image_source *s = src;
- obs_source_t *source = s->source;
- obs_data_t *settings = obs_source_get_settings(source);
- obs_data_set_string(settings, "file", new_path);
- obs_source_update(source, settings);
- obs_data_release(settings);
- UNUSED_PARAMETER(data);
- }
- static obs_missing_files_t *image_source_missingfiles(void *data)
- {
- struct image_source *s = data;
- obs_missing_files_t *files = obs_missing_files_create();
- if (strcmp(s->file, "") != 0) {
- if (!os_file_exists(s->file)) {
- obs_missing_file_t *file = obs_missing_file_create(s->file, missing_file_callback,
- OBS_MISSING_FILE_SOURCE, s->source, NULL);
- obs_missing_files_add_file(files, file);
- }
- }
- return files;
- }
- static enum gs_color_space image_source_get_color_space(void *data, size_t count,
- const enum gs_color_space *preferred_spaces)
- {
- UNUSED_PARAMETER(count);
- UNUSED_PARAMETER(preferred_spaces);
- struct image_source *const s = data;
- gs_image_file4_t *const if4 = &s->if4;
- return if4->image3.image2.image.texture ? if4->space : GS_CS_SRGB;
- }
- static struct obs_source_info image_source_info = {
- .id = "image_source",
- .type = OBS_SOURCE_TYPE_INPUT,
- .output_flags = OBS_SOURCE_VIDEO | OBS_SOURCE_SRGB,
- .get_name = image_source_get_name,
- .create = image_source_create,
- .destroy = image_source_destroy,
- .update = image_source_update,
- .get_defaults = image_source_defaults,
- .show = image_source_show,
- .hide = image_source_hide,
- .get_width = image_source_getwidth,
- .get_height = image_source_getheight,
- .video_render = image_source_render,
- .video_tick = image_source_tick,
- .missing_files = image_source_missingfiles,
- .get_properties = image_source_properties,
- .icon_type = OBS_ICON_TYPE_IMAGE,
- .activate = image_source_activate,
- .video_get_color_space = image_source_get_color_space,
- };
- OBS_DECLARE_MODULE()
- OBS_MODULE_USE_DEFAULT_LOCALE("image-source", "en-US")
- MODULE_EXPORT const char *obs_module_description(void)
- {
- return "Image/color/slideshow sources";
- }
- extern struct obs_source_info slideshow_info;
- extern struct obs_source_info slideshow_info_mk2;
- extern struct obs_source_info color_source_info_v1;
- extern struct obs_source_info color_source_info_v2;
- extern struct obs_source_info color_source_info_v3;
- bool obs_module_load(void)
- {
- obs_register_source(&image_source_info);
- obs_register_source(&color_source_info_v1);
- obs_register_source(&color_source_info_v2);
- obs_register_source(&color_source_info_v3);
- obs_register_source(&slideshow_info);
- obs_register_source(&slideshow_info_mk2);
- return true;
- }
|