1
0

obs-ffmpeg-source.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793
  1. /*
  2. * Copyright (c) 2015 John R. Bradley <[email protected]>
  3. *
  4. * Permission to use, copy, modify, and distribute this software for any
  5. * purpose with or without fee is hereby granted, provided that the above
  6. * copyright notice and this permission notice appear in all copies.
  7. *
  8. * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9. * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10. * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11. * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12. * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13. * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14. * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15. */
  16. #include <obs-module.h>
  17. #include <util/platform.h>
  18. #include <util/dstr.h>
  19. #include "obs-ffmpeg-compat.h"
  20. #include "obs-ffmpeg-formats.h"
  21. #include <media-playback/media.h>
  22. #define FF_LOG(level, format, ...) \
  23. blog(level, "[Media Source]: " format, ##__VA_ARGS__)
  24. #define FF_LOG_S(source, level, format, ...) \
  25. blog(level, "[Media Source '%s']: " format, \
  26. obs_source_get_name(source), ##__VA_ARGS__)
  27. #define FF_BLOG(level, format, ...) \
  28. FF_LOG_S(s->source, level, format, ##__VA_ARGS__)
  29. struct ffmpeg_source {
  30. mp_media_t media;
  31. bool media_valid;
  32. bool destroy_media;
  33. struct SwsContext *sws_ctx;
  34. int sws_width;
  35. int sws_height;
  36. enum AVPixelFormat sws_format;
  37. uint8_t *sws_data;
  38. int sws_linesize;
  39. enum video_range_type range;
  40. bool is_linear_alpha;
  41. obs_source_t *source;
  42. obs_hotkey_id hotkey;
  43. char *input;
  44. char *input_format;
  45. int buffering_mb;
  46. int speed_percent;
  47. bool is_looping;
  48. bool is_local_file;
  49. bool is_hw_decoding;
  50. bool is_clear_on_media_end;
  51. bool restart_on_activate;
  52. bool close_when_inactive;
  53. bool seekable;
  54. pthread_t reconnect_thread;
  55. bool stop_reconnect;
  56. bool reconnect_thread_valid;
  57. volatile bool reconnecting;
  58. int reconnect_delay_sec;
  59. enum obs_media_state state;
  60. obs_hotkey_pair_id play_pause_hotkey;
  61. obs_hotkey_id stop_hotkey;
  62. };
  63. static void set_media_state(void *data, enum obs_media_state state)
  64. {
  65. struct ffmpeg_source *s = data;
  66. s->state = state;
  67. }
  68. static bool is_local_file_modified(obs_properties_t *props,
  69. obs_property_t *prop, obs_data_t *settings)
  70. {
  71. UNUSED_PARAMETER(prop);
  72. bool enabled = obs_data_get_bool(settings, "is_local_file");
  73. obs_property_t *input = obs_properties_get(props, "input");
  74. obs_property_t *input_format =
  75. obs_properties_get(props, "input_format");
  76. obs_property_t *local_file = obs_properties_get(props, "local_file");
  77. obs_property_t *looping = obs_properties_get(props, "looping");
  78. obs_property_t *buffering = obs_properties_get(props, "buffering_mb");
  79. obs_property_t *seekable = obs_properties_get(props, "seekable");
  80. obs_property_t *speed = obs_properties_get(props, "speed_percent");
  81. obs_property_t *reconnect_delay_sec =
  82. obs_properties_get(props, "reconnect_delay_sec");
  83. obs_property_set_visible(input, !enabled);
  84. obs_property_set_visible(input_format, !enabled);
  85. obs_property_set_visible(buffering, !enabled);
  86. obs_property_set_visible(local_file, enabled);
  87. obs_property_set_visible(looping, enabled);
  88. obs_property_set_visible(speed, enabled);
  89. obs_property_set_visible(seekable, !enabled);
  90. obs_property_set_visible(reconnect_delay_sec, !enabled);
  91. return true;
  92. }
  93. static void ffmpeg_source_defaults(obs_data_t *settings)
  94. {
  95. obs_data_set_default_bool(settings, "is_local_file", true);
  96. obs_data_set_default_bool(settings, "looping", false);
  97. obs_data_set_default_bool(settings, "clear_on_media_end", true);
  98. obs_data_set_default_bool(settings, "restart_on_activate", true);
  99. obs_data_set_default_bool(settings, "linear_alpha", false);
  100. obs_data_set_default_int(settings, "reconnect_delay_sec", 10);
  101. obs_data_set_default_int(settings, "buffering_mb", 2);
  102. obs_data_set_default_int(settings, "speed_percent", 100);
  103. }
  104. static const char *media_filter =
  105. " (*.mp4 *.ts *.mov *.flv *.mkv *.avi *.mp3 *.ogg *.aac *.wav *.gif *.webm);;";
  106. static const char *video_filter =
  107. " (*.mp4 *.ts *.mov *.flv *.mkv *.avi *.gif *.webm);;";
  108. static const char *audio_filter = " (*.mp3 *.aac *.ogg *.wav);;";
  109. static obs_properties_t *ffmpeg_source_getproperties(void *data)
  110. {
  111. struct ffmpeg_source *s = data;
  112. struct dstr filter = {0};
  113. struct dstr path = {0};
  114. UNUSED_PARAMETER(data);
  115. obs_properties_t *props = obs_properties_create();
  116. obs_properties_set_flags(props, OBS_PROPERTIES_DEFER_UPDATE);
  117. obs_property_t *prop;
  118. // use this when obs allows non-readonly paths
  119. prop = obs_properties_add_bool(props, "is_local_file",
  120. obs_module_text("LocalFile"));
  121. obs_property_set_modified_callback(prop, is_local_file_modified);
  122. dstr_copy(&filter, obs_module_text("MediaFileFilter.AllMediaFiles"));
  123. dstr_cat(&filter, media_filter);
  124. dstr_cat(&filter, obs_module_text("MediaFileFilter.VideoFiles"));
  125. dstr_cat(&filter, video_filter);
  126. dstr_cat(&filter, obs_module_text("MediaFileFilter.AudioFiles"));
  127. dstr_cat(&filter, audio_filter);
  128. dstr_cat(&filter, obs_module_text("MediaFileFilter.AllFiles"));
  129. dstr_cat(&filter, " (*.*)");
  130. if (s && s->input && *s->input) {
  131. const char *slash;
  132. dstr_copy(&path, s->input);
  133. dstr_replace(&path, "\\", "/");
  134. slash = strrchr(path.array, '/');
  135. if (slash)
  136. dstr_resize(&path, slash - path.array + 1);
  137. }
  138. obs_properties_add_path(props, "local_file",
  139. obs_module_text("LocalFile"), OBS_PATH_FILE,
  140. filter.array, path.array);
  141. dstr_free(&filter);
  142. dstr_free(&path);
  143. obs_properties_add_bool(props, "looping", obs_module_text("Looping"));
  144. obs_properties_add_bool(props, "restart_on_activate",
  145. obs_module_text("RestartWhenActivated"));
  146. prop = obs_properties_add_int_slider(props, "buffering_mb",
  147. obs_module_text("BufferingMB"), 0,
  148. 16, 1);
  149. obs_property_int_set_suffix(prop, " MB");
  150. obs_properties_add_text(props, "input", obs_module_text("Input"),
  151. OBS_TEXT_DEFAULT);
  152. obs_properties_add_text(props, "input_format",
  153. obs_module_text("InputFormat"),
  154. OBS_TEXT_DEFAULT);
  155. prop = obs_properties_add_int_slider(
  156. props, "reconnect_delay_sec",
  157. obs_module_text("ReconnectDelayTime"), 1, 60, 1);
  158. obs_property_int_set_suffix(prop, " S");
  159. obs_properties_add_bool(props, "hw_decode",
  160. obs_module_text("HardwareDecode"));
  161. obs_properties_add_bool(props, "clear_on_media_end",
  162. obs_module_text("ClearOnMediaEnd"));
  163. prop = obs_properties_add_bool(
  164. props, "close_when_inactive",
  165. obs_module_text("CloseFileWhenInactive"));
  166. obs_property_set_long_description(
  167. prop, obs_module_text("CloseFileWhenInactive.ToolTip"));
  168. prop = obs_properties_add_int_slider(props, "speed_percent",
  169. obs_module_text("SpeedPercentage"),
  170. 1, 200, 1);
  171. obs_property_int_set_suffix(prop, "%");
  172. prop = obs_properties_add_list(props, "color_range",
  173. obs_module_text("ColorRange"),
  174. OBS_COMBO_TYPE_LIST,
  175. OBS_COMBO_FORMAT_INT);
  176. obs_property_list_add_int(prop, obs_module_text("ColorRange.Auto"),
  177. VIDEO_RANGE_DEFAULT);
  178. obs_property_list_add_int(prop, obs_module_text("ColorRange.Partial"),
  179. VIDEO_RANGE_PARTIAL);
  180. obs_property_list_add_int(prop, obs_module_text("ColorRange.Full"),
  181. VIDEO_RANGE_FULL);
  182. obs_properties_add_bool(props, "linear_alpha",
  183. obs_module_text("LinearAlpha"));
  184. obs_properties_add_bool(props, "seekable", obs_module_text("Seekable"));
  185. return props;
  186. }
  187. static void dump_source_info(struct ffmpeg_source *s, const char *input,
  188. const char *input_format)
  189. {
  190. FF_BLOG(LOG_INFO,
  191. "settings:\n"
  192. "\tinput: %s\n"
  193. "\tinput_format: %s\n"
  194. "\tspeed: %d\n"
  195. "\tis_looping: %s\n"
  196. "\tis_linear_alpha: %s\n"
  197. "\tis_hw_decoding: %s\n"
  198. "\tis_clear_on_media_end: %s\n"
  199. "\trestart_on_activate: %s\n"
  200. "\tclose_when_inactive: %s",
  201. input ? input : "(null)",
  202. input_format ? input_format : "(null)", s->speed_percent,
  203. s->is_looping ? "yes" : "no", s->is_linear_alpha ? "yes" : "no",
  204. s->is_hw_decoding ? "yes" : "no",
  205. s->is_clear_on_media_end ? "yes" : "no",
  206. s->restart_on_activate ? "yes" : "no",
  207. s->close_when_inactive ? "yes" : "no");
  208. }
  209. static void get_frame(void *opaque, struct obs_source_frame *f)
  210. {
  211. struct ffmpeg_source *s = opaque;
  212. obs_source_output_video(s->source, f);
  213. }
  214. static void preload_frame(void *opaque, struct obs_source_frame *f)
  215. {
  216. struct ffmpeg_source *s = opaque;
  217. if (s->close_when_inactive)
  218. return;
  219. if (s->is_clear_on_media_end || s->is_looping)
  220. obs_source_preload_video(s->source, f);
  221. if (!s->is_local_file && os_atomic_set_bool(&s->reconnecting, false))
  222. FF_BLOG(LOG_INFO, "Reconnected.");
  223. }
  224. static void seek_frame(void *opaque, struct obs_source_frame *f)
  225. {
  226. struct ffmpeg_source *s = opaque;
  227. obs_source_set_video_frame(s->source, f);
  228. }
  229. static void get_audio(void *opaque, struct obs_source_audio *a)
  230. {
  231. struct ffmpeg_source *s = opaque;
  232. obs_source_output_audio(s->source, a);
  233. if (!s->is_local_file && os_atomic_set_bool(&s->reconnecting, false))
  234. FF_BLOG(LOG_INFO, "Reconnected.");
  235. }
  236. static void media_stopped(void *opaque)
  237. {
  238. struct ffmpeg_source *s = opaque;
  239. if (s->is_clear_on_media_end) {
  240. obs_source_output_video(s->source, NULL);
  241. }
  242. if ((s->close_when_inactive || !s->is_local_file) && s->media_valid)
  243. s->destroy_media = true;
  244. set_media_state(s, OBS_MEDIA_STATE_ENDED);
  245. obs_source_media_ended(s->source);
  246. }
  247. static void ffmpeg_source_open(struct ffmpeg_source *s)
  248. {
  249. if (s->input && *s->input) {
  250. struct mp_media_info info = {
  251. .opaque = s,
  252. .v_cb = get_frame,
  253. .v_preload_cb = preload_frame,
  254. .v_seek_cb = seek_frame,
  255. .a_cb = get_audio,
  256. .stop_cb = media_stopped,
  257. .path = s->input,
  258. .format = s->input_format,
  259. .buffering = s->buffering_mb * 1024 * 1024,
  260. .speed = s->speed_percent,
  261. .force_range = s->range,
  262. .is_linear_alpha = s->is_linear_alpha,
  263. .hardware_decoding = s->is_hw_decoding,
  264. .is_local_file = s->is_local_file || s->seekable,
  265. .reconnecting = s->reconnecting,
  266. };
  267. s->media_valid = mp_media_init(&s->media, &info);
  268. }
  269. }
  270. static void ffmpeg_source_start(struct ffmpeg_source *s)
  271. {
  272. if (!s->media_valid)
  273. ffmpeg_source_open(s);
  274. if (!s->media_valid)
  275. return;
  276. mp_media_play(&s->media, s->is_looping, s->reconnecting);
  277. if (s->is_local_file && (s->is_clear_on_media_end || s->is_looping))
  278. obs_source_show_preloaded_video(s->source);
  279. else
  280. obs_source_output_video(s->source, NULL);
  281. set_media_state(s, OBS_MEDIA_STATE_PLAYING);
  282. obs_source_media_started(s->source);
  283. }
  284. static void *ffmpeg_source_reconnect(void *data)
  285. {
  286. struct ffmpeg_source *s = data;
  287. os_sleep_ms(s->reconnect_delay_sec * 1000);
  288. if (s->stop_reconnect || s->media_valid)
  289. goto finish;
  290. bool active = obs_source_active(s->source);
  291. if (!s->close_when_inactive || active)
  292. ffmpeg_source_open(s);
  293. if (!s->restart_on_activate || active)
  294. ffmpeg_source_start(s);
  295. finish:
  296. s->reconnect_thread_valid = false;
  297. return NULL;
  298. }
  299. static void ffmpeg_source_tick(void *data, float seconds)
  300. {
  301. UNUSED_PARAMETER(seconds);
  302. struct ffmpeg_source *s = data;
  303. if (s->destroy_media) {
  304. if (s->media_valid) {
  305. mp_media_free(&s->media);
  306. s->media_valid = false;
  307. }
  308. s->destroy_media = false;
  309. if (!s->is_local_file) {
  310. if (!os_atomic_set_bool(&s->reconnecting, true)) {
  311. FF_BLOG(LOG_WARNING, "Disconnected. "
  312. "Reconnecting...");
  313. }
  314. if (pthread_create(&s->reconnect_thread, NULL,
  315. ffmpeg_source_reconnect, s) != 0) {
  316. FF_BLOG(LOG_WARNING, "Could not create "
  317. "reconnect thread");
  318. return;
  319. }
  320. s->reconnect_thread_valid = true;
  321. }
  322. }
  323. }
  324. static void ffmpeg_source_update(void *data, obs_data_t *settings)
  325. {
  326. struct ffmpeg_source *s = data;
  327. bool is_local_file = obs_data_get_bool(settings, "is_local_file");
  328. char *input;
  329. char *input_format;
  330. bfree(s->input);
  331. bfree(s->input_format);
  332. if (is_local_file) {
  333. input = (char *)obs_data_get_string(settings, "local_file");
  334. input_format = NULL;
  335. s->is_looping = obs_data_get_bool(settings, "looping");
  336. } else {
  337. input = (char *)obs_data_get_string(settings, "input");
  338. input_format =
  339. (char *)obs_data_get_string(settings, "input_format");
  340. s->reconnect_delay_sec =
  341. (int)obs_data_get_int(settings, "reconnect_delay_sec");
  342. s->reconnect_delay_sec = s->reconnect_delay_sec == 0
  343. ? 10
  344. : s->reconnect_delay_sec;
  345. s->is_looping = false;
  346. if (s->reconnect_thread_valid) {
  347. s->stop_reconnect = true;
  348. pthread_join(s->reconnect_thread, NULL);
  349. s->stop_reconnect = false;
  350. }
  351. }
  352. s->close_when_inactive =
  353. obs_data_get_bool(settings, "close_when_inactive");
  354. s->input = input ? bstrdup(input) : NULL;
  355. s->input_format = input_format ? bstrdup(input_format) : NULL;
  356. s->is_hw_decoding = obs_data_get_bool(settings, "hw_decode");
  357. s->is_clear_on_media_end =
  358. obs_data_get_bool(settings, "clear_on_media_end");
  359. s->restart_on_activate =
  360. obs_data_get_bool(settings, "restart_on_activate");
  361. s->range = (enum video_range_type)obs_data_get_int(settings,
  362. "color_range");
  363. s->is_linear_alpha = obs_data_get_bool(settings, "linear_alpha");
  364. s->buffering_mb = (int)obs_data_get_int(settings, "buffering_mb");
  365. s->speed_percent = (int)obs_data_get_int(settings, "speed_percent");
  366. s->is_local_file = is_local_file;
  367. s->seekable = obs_data_get_bool(settings, "seekable");
  368. if (s->speed_percent < 1 || s->speed_percent > 200)
  369. s->speed_percent = 100;
  370. if (s->media_valid) {
  371. mp_media_free(&s->media);
  372. s->media_valid = false;
  373. }
  374. bool active = obs_source_active(s->source);
  375. if (!s->close_when_inactive || active)
  376. ffmpeg_source_open(s);
  377. dump_source_info(s, input, input_format);
  378. if (!s->restart_on_activate || active)
  379. ffmpeg_source_start(s);
  380. }
  381. static const char *ffmpeg_source_getname(void *unused)
  382. {
  383. UNUSED_PARAMETER(unused);
  384. return obs_module_text("FFMpegSource");
  385. }
  386. static void restart_hotkey(void *data, obs_hotkey_id id, obs_hotkey_t *hotkey,
  387. bool pressed)
  388. {
  389. UNUSED_PARAMETER(id);
  390. UNUSED_PARAMETER(hotkey);
  391. if (!pressed)
  392. return;
  393. struct ffmpeg_source *s = data;
  394. if (obs_source_showing(s->source))
  395. obs_source_media_restart(s->source);
  396. }
  397. static void restart_proc(void *data, calldata_t *cd)
  398. {
  399. restart_hotkey(data, 0, NULL, true);
  400. UNUSED_PARAMETER(cd);
  401. }
  402. static void get_duration(void *data, calldata_t *cd)
  403. {
  404. struct ffmpeg_source *s = data;
  405. int64_t dur = 0;
  406. if (s->media.fmt)
  407. dur = s->media.fmt->duration;
  408. calldata_set_int(cd, "duration", dur * 1000);
  409. }
  410. static void get_nb_frames(void *data, calldata_t *cd)
  411. {
  412. struct ffmpeg_source *s = data;
  413. int64_t frames = 0;
  414. if (!s->media.fmt) {
  415. calldata_set_int(cd, "num_frames", frames);
  416. return;
  417. }
  418. int video_stream_index = av_find_best_stream(
  419. s->media.fmt, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
  420. if (video_stream_index < 0) {
  421. FF_BLOG(LOG_WARNING, "Getting number of frames failed: No "
  422. "video stream in media file!");
  423. calldata_set_int(cd, "num_frames", frames);
  424. return;
  425. }
  426. AVStream *stream = s->media.fmt->streams[video_stream_index];
  427. if (stream->nb_frames > 0) {
  428. frames = stream->nb_frames;
  429. } else {
  430. FF_BLOG(LOG_DEBUG, "nb_frames not set, estimating using frame "
  431. "rate and duration");
  432. AVRational avg_frame_rate = stream->avg_frame_rate;
  433. frames = (int64_t)ceil((double)s->media.fmt->duration /
  434. (double)AV_TIME_BASE *
  435. (double)avg_frame_rate.num /
  436. (double)avg_frame_rate.den);
  437. }
  438. calldata_set_int(cd, "num_frames", frames);
  439. }
  440. static bool ffmpeg_source_play_hotkey(void *data, obs_hotkey_pair_id id,
  441. obs_hotkey_t *hotkey, bool pressed)
  442. {
  443. UNUSED_PARAMETER(id);
  444. UNUSED_PARAMETER(hotkey);
  445. if (!pressed)
  446. return false;
  447. struct ffmpeg_source *s = data;
  448. if (s->state == OBS_MEDIA_STATE_PLAYING ||
  449. !obs_source_showing(s->source))
  450. return false;
  451. obs_source_media_play_pause(s->source, false);
  452. return true;
  453. }
  454. static bool ffmpeg_source_pause_hotkey(void *data, obs_hotkey_pair_id id,
  455. obs_hotkey_t *hotkey, bool pressed)
  456. {
  457. UNUSED_PARAMETER(id);
  458. UNUSED_PARAMETER(hotkey);
  459. if (!pressed)
  460. return false;
  461. struct ffmpeg_source *s = data;
  462. if (s->state != OBS_MEDIA_STATE_PLAYING ||
  463. !obs_source_showing(s->source))
  464. return false;
  465. obs_source_media_play_pause(s->source, true);
  466. return true;
  467. }
  468. static void ffmpeg_source_stop_hotkey(void *data, obs_hotkey_id id,
  469. obs_hotkey_t *hotkey, bool pressed)
  470. {
  471. UNUSED_PARAMETER(id);
  472. UNUSED_PARAMETER(hotkey);
  473. if (!pressed)
  474. return;
  475. struct ffmpeg_source *s = data;
  476. if (obs_source_showing(s->source))
  477. obs_source_media_stop(s->source);
  478. }
  479. static void *ffmpeg_source_create(obs_data_t *settings, obs_source_t *source)
  480. {
  481. UNUSED_PARAMETER(settings);
  482. struct ffmpeg_source *s = bzalloc(sizeof(struct ffmpeg_source));
  483. s->source = source;
  484. s->hotkey = obs_hotkey_register_source(source, "MediaSource.Restart",
  485. obs_module_text("RestartMedia"),
  486. restart_hotkey, s);
  487. s->play_pause_hotkey = obs_hotkey_pair_register_source(
  488. s->source, "MediaSource.Play", obs_module_text("Play"),
  489. "MediaSource.Pause", obs_module_text("Pause"),
  490. ffmpeg_source_play_hotkey, ffmpeg_source_pause_hotkey, s, s);
  491. s->stop_hotkey = obs_hotkey_register_source(source, "MediaSource.Stop",
  492. obs_module_text("Stop"),
  493. ffmpeg_source_stop_hotkey,
  494. s);
  495. proc_handler_t *ph = obs_source_get_proc_handler(source);
  496. proc_handler_add(ph, "void restart()", restart_proc, s);
  497. proc_handler_add(ph, "void get_duration(out int duration)",
  498. get_duration, s);
  499. proc_handler_add(ph, "void get_nb_frames(out int num_frames)",
  500. get_nb_frames, s);
  501. ffmpeg_source_update(s, settings);
  502. return s;
  503. }
  504. static void ffmpeg_source_destroy(void *data)
  505. {
  506. struct ffmpeg_source *s = data;
  507. if (s->hotkey)
  508. obs_hotkey_unregister(s->hotkey);
  509. if (!s->is_local_file) {
  510. s->stop_reconnect = true;
  511. if (s->reconnect_thread_valid)
  512. pthread_join(s->reconnect_thread, NULL);
  513. }
  514. if (s->media_valid)
  515. mp_media_free(&s->media);
  516. if (s->sws_ctx != NULL)
  517. sws_freeContext(s->sws_ctx);
  518. bfree(s->sws_data);
  519. bfree(s->input);
  520. bfree(s->input_format);
  521. bfree(s);
  522. }
  523. static void ffmpeg_source_activate(void *data)
  524. {
  525. struct ffmpeg_source *s = data;
  526. if (s->restart_on_activate)
  527. obs_source_media_restart(s->source);
  528. }
  529. static void ffmpeg_source_deactivate(void *data)
  530. {
  531. struct ffmpeg_source *s = data;
  532. if (s->restart_on_activate) {
  533. if (s->media_valid) {
  534. mp_media_stop(&s->media);
  535. if (s->is_clear_on_media_end)
  536. obs_source_output_video(s->source, NULL);
  537. }
  538. }
  539. }
  540. static void ffmpeg_source_play_pause(void *data, bool pause)
  541. {
  542. struct ffmpeg_source *s = data;
  543. if (!s->media_valid)
  544. ffmpeg_source_open(s);
  545. if (!s->media_valid)
  546. return;
  547. mp_media_play_pause(&s->media, pause);
  548. if (pause) {
  549. set_media_state(s, OBS_MEDIA_STATE_PAUSED);
  550. } else {
  551. set_media_state(s, OBS_MEDIA_STATE_PLAYING);
  552. obs_source_media_started(s->source);
  553. }
  554. }
  555. static void ffmpeg_source_stop(void *data)
  556. {
  557. struct ffmpeg_source *s = data;
  558. if (s->media_valid) {
  559. mp_media_stop(&s->media);
  560. obs_source_output_video(s->source, NULL);
  561. set_media_state(s, OBS_MEDIA_STATE_STOPPED);
  562. }
  563. }
  564. static void ffmpeg_source_restart(void *data)
  565. {
  566. struct ffmpeg_source *s = data;
  567. if (obs_source_showing(s->source))
  568. ffmpeg_source_start(s);
  569. set_media_state(s, OBS_MEDIA_STATE_PLAYING);
  570. }
  571. static int64_t ffmpeg_source_get_duration(void *data)
  572. {
  573. struct ffmpeg_source *s = data;
  574. int64_t dur = 0;
  575. if (s->media.fmt)
  576. dur = s->media.fmt->duration / INT64_C(1000);
  577. return dur;
  578. }
  579. static int64_t ffmpeg_source_get_time(void *data)
  580. {
  581. struct ffmpeg_source *s = data;
  582. return mp_get_current_time(&s->media);
  583. }
  584. static void ffmpeg_source_set_time(void *data, int64_t ms)
  585. {
  586. struct ffmpeg_source *s = data;
  587. if (!s->media_valid)
  588. return;
  589. mp_media_seek_to(&s->media, ms);
  590. }
  591. static enum obs_media_state ffmpeg_source_get_state(void *data)
  592. {
  593. struct ffmpeg_source *s = data;
  594. return s->state;
  595. }
  596. static void missing_file_callback(void *src, const char *new_path, void *data)
  597. {
  598. struct ffmpeg_source *s = src;
  599. obs_source_t *source = s->source;
  600. obs_data_t *settings = obs_source_get_settings(source);
  601. obs_data_set_string(settings, "local_file", new_path);
  602. obs_source_update(source, settings);
  603. obs_data_release(settings);
  604. UNUSED_PARAMETER(data);
  605. }
  606. static obs_missing_files_t *ffmpeg_source_missingfiles(void *data)
  607. {
  608. struct ffmpeg_source *s = data;
  609. obs_missing_files_t *files = obs_missing_files_create();
  610. if (s->is_local_file && strcmp(s->input, "") != 0) {
  611. if (!os_file_exists(s->input)) {
  612. obs_missing_file_t *file = obs_missing_file_create(
  613. s->input, missing_file_callback,
  614. OBS_MISSING_FILE_SOURCE, s->source, NULL);
  615. obs_missing_files_add_file(files, file);
  616. }
  617. }
  618. return files;
  619. }
  620. struct obs_source_info ffmpeg_source = {
  621. .id = "ffmpeg_source",
  622. .type = OBS_SOURCE_TYPE_INPUT,
  623. .output_flags = OBS_SOURCE_ASYNC_VIDEO | OBS_SOURCE_AUDIO |
  624. OBS_SOURCE_DO_NOT_DUPLICATE |
  625. OBS_SOURCE_CONTROLLABLE_MEDIA,
  626. .get_name = ffmpeg_source_getname,
  627. .create = ffmpeg_source_create,
  628. .destroy = ffmpeg_source_destroy,
  629. .get_defaults = ffmpeg_source_defaults,
  630. .get_properties = ffmpeg_source_getproperties,
  631. .activate = ffmpeg_source_activate,
  632. .deactivate = ffmpeg_source_deactivate,
  633. .video_tick = ffmpeg_source_tick,
  634. .missing_files = ffmpeg_source_missingfiles,
  635. .update = ffmpeg_source_update,
  636. .icon_type = OBS_ICON_TYPE_MEDIA,
  637. .media_play_pause = ffmpeg_source_play_pause,
  638. .media_restart = ffmpeg_source_restart,
  639. .media_stop = ffmpeg_source_stop,
  640. .media_get_duration = ffmpeg_source_get_duration,
  641. .media_get_time = ffmpeg_source_get_time,
  642. .media_set_time = ffmpeg_source_set_time,
  643. .media_get_state = ffmpeg_source_get_state,
  644. };