image-source.c 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. #include <obs-module.h>
  2. #define blog(log_level, format, ...) \
  3. blog(log_level, "[image_source: '%s'] " format, \
  4. obs_source_get_name(context->source), ##__VA_ARGS__)
  5. #define debug(format, ...) \
  6. blog(LOG_DEBUG, format, ##__VA_ARGS__)
  7. #define info(format, ...) \
  8. blog(LOG_INFO, format, ##__VA_ARGS__)
  9. #define warn(format, ...) \
  10. blog(LOG_WARNING, format, ##__VA_ARGS__)
  11. struct image_source {
  12. obs_source_t *source;
  13. gs_texture_t *tex;
  14. uint32_t cx;
  15. uint32_t cy;
  16. };
  17. static const char *image_source_get_name(void)
  18. {
  19. return obs_module_text("ImageInput");
  20. }
  21. static void image_source_update(void *data, obs_data_t *settings)
  22. {
  23. struct image_source *context = data;
  24. const char *file = obs_data_get_string(settings, "file");
  25. obs_enter_graphics();
  26. if (context->tex) {
  27. gs_texture_destroy(context->tex);
  28. context->tex = NULL;
  29. }
  30. if (file && *file) {
  31. context->tex = gs_texture_create_from_file(file);
  32. if (context->tex) {
  33. context->cx = gs_texture_get_width(context->tex);
  34. context->cy = gs_texture_get_height(context->tex);
  35. } else {
  36. warn("failed to load texture '%s'", file);
  37. context->cx = 0;
  38. context->cy = 0;
  39. }
  40. }
  41. obs_leave_graphics();
  42. }
  43. static void *image_source_create(obs_data_t *settings, obs_source_t *source)
  44. {
  45. struct image_source *context = bzalloc(sizeof(struct image_source));
  46. context->source = source;
  47. image_source_update(context, settings);
  48. return context;
  49. }
  50. static void image_source_destroy(void *data)
  51. {
  52. struct image_source *context = data;
  53. obs_enter_graphics();
  54. gs_texture_destroy(context->tex);
  55. obs_leave_graphics();
  56. bfree(context);
  57. }
  58. static uint32_t image_source_getwidth(void *data)
  59. {
  60. struct image_source *context = data;
  61. return context->cx;
  62. }
  63. static uint32_t image_source_getheight(void *data)
  64. {
  65. struct image_source *context = data;
  66. return context->cy;
  67. }
  68. static void image_source_render(void *data, gs_effect_t *effect)
  69. {
  70. struct image_source *context = data;
  71. if (!context->tex)
  72. return;
  73. gs_reset_blend_state();
  74. gs_effect_set_texture(gs_effect_get_param_by_name(effect, "image"),
  75. context->tex);
  76. gs_draw_sprite(context->tex, 0, context->cx, context->cy);
  77. }
  78. static const char *image_filter =
  79. "All formats (*.bmp *.tga *.png *.jpeg *.jpg *.gif);;"
  80. "BMP Files (*.bmp);;"
  81. "Targa Files (*.tga);;"
  82. "PNG Files (*.png);;"
  83. "JPEG Files (*.jpeg *.jpg);;"
  84. "GIF Files (*.gif)";
  85. static obs_properties_t *image_source_properties(void *unused)
  86. {
  87. UNUSED_PARAMETER(unused);
  88. obs_properties_t *props = obs_properties_create();
  89. obs_properties_add_path(props,
  90. "file", obs_module_text("File"),
  91. OBS_PATH_FILE, image_filter, NULL);
  92. return props;
  93. }
  94. static struct obs_source_info image_source_info = {
  95. .id = "image_source",
  96. .type = OBS_SOURCE_TYPE_INPUT,
  97. .output_flags = OBS_SOURCE_VIDEO,
  98. .get_name = image_source_get_name,
  99. .create = image_source_create,
  100. .destroy = image_source_destroy,
  101. .update = image_source_update,
  102. .get_width = image_source_getwidth,
  103. .get_height = image_source_getheight,
  104. .video_render = image_source_render,
  105. .get_properties = image_source_properties
  106. };
  107. OBS_DECLARE_MODULE()
  108. OBS_MODULE_USE_DEFAULT_LOCALE("image-source", "en-US")
  109. bool obs_module_load(void)
  110. {
  111. obs_register_source(&image_source_info);
  112. return true;
  113. }