obs-ffmpeg-source.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808
  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. #define SRT_PROTO "srt"
  325. #define RIST_PROTO "rist"
  326. static bool requires_mpegts(char *path)
  327. {
  328. return !astrcmpi_n(path, SRT_PROTO, sizeof(SRT_PROTO) - 1) ||
  329. !astrcmpi_n(path, RIST_PROTO, sizeof(RIST_PROTO) - 1);
  330. }
  331. static void ffmpeg_source_update(void *data, obs_data_t *settings)
  332. {
  333. struct ffmpeg_source *s = data;
  334. bool is_local_file = obs_data_get_bool(settings, "is_local_file");
  335. char *input;
  336. char *input_format;
  337. bfree(s->input);
  338. bfree(s->input_format);
  339. if (is_local_file) {
  340. input = (char *)obs_data_get_string(settings, "local_file");
  341. input_format = NULL;
  342. s->is_looping = obs_data_get_bool(settings, "looping");
  343. } else {
  344. input = (char *)obs_data_get_string(settings, "input");
  345. input_format =
  346. (char *)obs_data_get_string(settings, "input_format");
  347. if (requires_mpegts(input)) {
  348. input_format = "mpegts";
  349. obs_data_set_string(settings, "input_format", "mpegts");
  350. }
  351. s->reconnect_delay_sec =
  352. (int)obs_data_get_int(settings, "reconnect_delay_sec");
  353. s->reconnect_delay_sec = s->reconnect_delay_sec == 0
  354. ? 10
  355. : s->reconnect_delay_sec;
  356. s->is_looping = false;
  357. if (s->reconnect_thread_valid) {
  358. s->stop_reconnect = true;
  359. pthread_join(s->reconnect_thread, NULL);
  360. s->stop_reconnect = false;
  361. }
  362. }
  363. s->close_when_inactive =
  364. obs_data_get_bool(settings, "close_when_inactive");
  365. s->input = input ? bstrdup(input) : NULL;
  366. s->input_format = input_format ? bstrdup(input_format) : NULL;
  367. s->is_hw_decoding = obs_data_get_bool(settings, "hw_decode");
  368. s->is_clear_on_media_end =
  369. obs_data_get_bool(settings, "clear_on_media_end");
  370. s->restart_on_activate =
  371. !astrcmpi_n(input, RIST_PROTO, sizeof(RIST_PROTO) - 1)
  372. ? false
  373. : obs_data_get_bool(settings, "restart_on_activate");
  374. s->range = (enum video_range_type)obs_data_get_int(settings,
  375. "color_range");
  376. s->is_linear_alpha = obs_data_get_bool(settings, "linear_alpha");
  377. s->buffering_mb = (int)obs_data_get_int(settings, "buffering_mb");
  378. s->speed_percent = (int)obs_data_get_int(settings, "speed_percent");
  379. s->is_local_file = is_local_file;
  380. s->seekable = obs_data_get_bool(settings, "seekable");
  381. if (s->speed_percent < 1 || s->speed_percent > 200)
  382. s->speed_percent = 100;
  383. if (s->media_valid) {
  384. mp_media_free(&s->media);
  385. s->media_valid = false;
  386. }
  387. bool active = obs_source_active(s->source);
  388. if (!s->close_when_inactive || active)
  389. ffmpeg_source_open(s);
  390. dump_source_info(s, input, input_format);
  391. if (!s->restart_on_activate || active)
  392. ffmpeg_source_start(s);
  393. }
  394. static const char *ffmpeg_source_getname(void *unused)
  395. {
  396. UNUSED_PARAMETER(unused);
  397. return obs_module_text("FFMpegSource");
  398. }
  399. static void restart_hotkey(void *data, obs_hotkey_id id, obs_hotkey_t *hotkey,
  400. bool pressed)
  401. {
  402. UNUSED_PARAMETER(id);
  403. UNUSED_PARAMETER(hotkey);
  404. if (!pressed)
  405. return;
  406. struct ffmpeg_source *s = data;
  407. if (obs_source_showing(s->source))
  408. obs_source_media_restart(s->source);
  409. }
  410. static void restart_proc(void *data, calldata_t *cd)
  411. {
  412. restart_hotkey(data, 0, NULL, true);
  413. UNUSED_PARAMETER(cd);
  414. }
  415. static void get_duration(void *data, calldata_t *cd)
  416. {
  417. struct ffmpeg_source *s = data;
  418. int64_t dur = 0;
  419. if (s->media.fmt)
  420. dur = s->media.fmt->duration;
  421. calldata_set_int(cd, "duration", dur * 1000);
  422. }
  423. static void get_nb_frames(void *data, calldata_t *cd)
  424. {
  425. struct ffmpeg_source *s = data;
  426. int64_t frames = 0;
  427. if (!s->media.fmt) {
  428. calldata_set_int(cd, "num_frames", frames);
  429. return;
  430. }
  431. int video_stream_index = av_find_best_stream(
  432. s->media.fmt, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, 0);
  433. if (video_stream_index < 0) {
  434. FF_BLOG(LOG_WARNING, "Getting number of frames failed: No "
  435. "video stream in media file!");
  436. calldata_set_int(cd, "num_frames", frames);
  437. return;
  438. }
  439. AVStream *stream = s->media.fmt->streams[video_stream_index];
  440. if (stream->nb_frames > 0) {
  441. frames = stream->nb_frames;
  442. } else {
  443. FF_BLOG(LOG_DEBUG, "nb_frames not set, estimating using frame "
  444. "rate and duration");
  445. AVRational avg_frame_rate = stream->avg_frame_rate;
  446. frames = (int64_t)ceil((double)s->media.fmt->duration /
  447. (double)AV_TIME_BASE *
  448. (double)avg_frame_rate.num /
  449. (double)avg_frame_rate.den);
  450. }
  451. calldata_set_int(cd, "num_frames", frames);
  452. }
  453. static bool ffmpeg_source_play_hotkey(void *data, obs_hotkey_pair_id id,
  454. obs_hotkey_t *hotkey, bool pressed)
  455. {
  456. UNUSED_PARAMETER(id);
  457. UNUSED_PARAMETER(hotkey);
  458. if (!pressed)
  459. return false;
  460. struct ffmpeg_source *s = data;
  461. if (s->state == OBS_MEDIA_STATE_PLAYING ||
  462. !obs_source_showing(s->source))
  463. return false;
  464. obs_source_media_play_pause(s->source, false);
  465. return true;
  466. }
  467. static bool ffmpeg_source_pause_hotkey(void *data, obs_hotkey_pair_id id,
  468. obs_hotkey_t *hotkey, bool pressed)
  469. {
  470. UNUSED_PARAMETER(id);
  471. UNUSED_PARAMETER(hotkey);
  472. if (!pressed)
  473. return false;
  474. struct ffmpeg_source *s = data;
  475. if (s->state != OBS_MEDIA_STATE_PLAYING ||
  476. !obs_source_showing(s->source))
  477. return false;
  478. obs_source_media_play_pause(s->source, true);
  479. return true;
  480. }
  481. static void ffmpeg_source_stop_hotkey(void *data, obs_hotkey_id id,
  482. obs_hotkey_t *hotkey, bool pressed)
  483. {
  484. UNUSED_PARAMETER(id);
  485. UNUSED_PARAMETER(hotkey);
  486. if (!pressed)
  487. return;
  488. struct ffmpeg_source *s = data;
  489. if (obs_source_showing(s->source))
  490. obs_source_media_stop(s->source);
  491. }
  492. static void *ffmpeg_source_create(obs_data_t *settings, obs_source_t *source)
  493. {
  494. UNUSED_PARAMETER(settings);
  495. struct ffmpeg_source *s = bzalloc(sizeof(struct ffmpeg_source));
  496. s->source = source;
  497. s->hotkey = obs_hotkey_register_source(source, "MediaSource.Restart",
  498. obs_module_text("RestartMedia"),
  499. restart_hotkey, s);
  500. s->play_pause_hotkey = obs_hotkey_pair_register_source(
  501. s->source, "MediaSource.Play", obs_module_text("Play"),
  502. "MediaSource.Pause", obs_module_text("Pause"),
  503. ffmpeg_source_play_hotkey, ffmpeg_source_pause_hotkey, s, s);
  504. s->stop_hotkey = obs_hotkey_register_source(source, "MediaSource.Stop",
  505. obs_module_text("Stop"),
  506. ffmpeg_source_stop_hotkey,
  507. s);
  508. proc_handler_t *ph = obs_source_get_proc_handler(source);
  509. proc_handler_add(ph, "void restart()", restart_proc, s);
  510. proc_handler_add(ph, "void get_duration(out int duration)",
  511. get_duration, s);
  512. proc_handler_add(ph, "void get_nb_frames(out int num_frames)",
  513. get_nb_frames, s);
  514. ffmpeg_source_update(s, settings);
  515. return s;
  516. }
  517. static void ffmpeg_source_destroy(void *data)
  518. {
  519. struct ffmpeg_source *s = data;
  520. if (s->hotkey)
  521. obs_hotkey_unregister(s->hotkey);
  522. if (!s->is_local_file) {
  523. s->stop_reconnect = true;
  524. if (s->reconnect_thread_valid)
  525. pthread_join(s->reconnect_thread, NULL);
  526. }
  527. if (s->media_valid)
  528. mp_media_free(&s->media);
  529. if (s->sws_ctx != NULL)
  530. sws_freeContext(s->sws_ctx);
  531. bfree(s->sws_data);
  532. bfree(s->input);
  533. bfree(s->input_format);
  534. bfree(s);
  535. }
  536. static void ffmpeg_source_activate(void *data)
  537. {
  538. struct ffmpeg_source *s = data;
  539. if (s->restart_on_activate)
  540. obs_source_media_restart(s->source);
  541. }
  542. static void ffmpeg_source_deactivate(void *data)
  543. {
  544. struct ffmpeg_source *s = data;
  545. if (s->restart_on_activate) {
  546. if (s->media_valid) {
  547. mp_media_stop(&s->media);
  548. if (s->is_clear_on_media_end)
  549. obs_source_output_video(s->source, NULL);
  550. }
  551. }
  552. }
  553. static void ffmpeg_source_play_pause(void *data, bool pause)
  554. {
  555. struct ffmpeg_source *s = data;
  556. if (!s->media_valid)
  557. ffmpeg_source_open(s);
  558. if (!s->media_valid)
  559. return;
  560. mp_media_play_pause(&s->media, pause);
  561. if (pause) {
  562. set_media_state(s, OBS_MEDIA_STATE_PAUSED);
  563. } else {
  564. set_media_state(s, OBS_MEDIA_STATE_PLAYING);
  565. obs_source_media_started(s->source);
  566. }
  567. }
  568. static void ffmpeg_source_stop(void *data)
  569. {
  570. struct ffmpeg_source *s = data;
  571. if (s->media_valid) {
  572. mp_media_stop(&s->media);
  573. obs_source_output_video(s->source, NULL);
  574. set_media_state(s, OBS_MEDIA_STATE_STOPPED);
  575. }
  576. }
  577. static void ffmpeg_source_restart(void *data)
  578. {
  579. struct ffmpeg_source *s = data;
  580. if (obs_source_showing(s->source))
  581. ffmpeg_source_start(s);
  582. set_media_state(s, OBS_MEDIA_STATE_PLAYING);
  583. }
  584. static int64_t ffmpeg_source_get_duration(void *data)
  585. {
  586. struct ffmpeg_source *s = data;
  587. int64_t dur = 0;
  588. if (s->media.fmt)
  589. dur = s->media.fmt->duration / INT64_C(1000);
  590. return dur;
  591. }
  592. static int64_t ffmpeg_source_get_time(void *data)
  593. {
  594. struct ffmpeg_source *s = data;
  595. return mp_get_current_time(&s->media);
  596. }
  597. static void ffmpeg_source_set_time(void *data, int64_t ms)
  598. {
  599. struct ffmpeg_source *s = data;
  600. if (!s->media_valid)
  601. return;
  602. mp_media_seek_to(&s->media, ms);
  603. }
  604. static enum obs_media_state ffmpeg_source_get_state(void *data)
  605. {
  606. struct ffmpeg_source *s = data;
  607. return s->state;
  608. }
  609. static void missing_file_callback(void *src, const char *new_path, void *data)
  610. {
  611. struct ffmpeg_source *s = src;
  612. obs_source_t *source = s->source;
  613. obs_data_t *settings = obs_source_get_settings(source);
  614. obs_data_set_string(settings, "local_file", new_path);
  615. obs_source_update(source, settings);
  616. obs_data_release(settings);
  617. UNUSED_PARAMETER(data);
  618. }
  619. static obs_missing_files_t *ffmpeg_source_missingfiles(void *data)
  620. {
  621. struct ffmpeg_source *s = data;
  622. obs_missing_files_t *files = obs_missing_files_create();
  623. if (s->is_local_file && strcmp(s->input, "") != 0) {
  624. if (!os_file_exists(s->input)) {
  625. obs_missing_file_t *file = obs_missing_file_create(
  626. s->input, missing_file_callback,
  627. OBS_MISSING_FILE_SOURCE, s->source, NULL);
  628. obs_missing_files_add_file(files, file);
  629. }
  630. }
  631. return files;
  632. }
  633. struct obs_source_info ffmpeg_source = {
  634. .id = "ffmpeg_source",
  635. .type = OBS_SOURCE_TYPE_INPUT,
  636. .output_flags = OBS_SOURCE_ASYNC_VIDEO | OBS_SOURCE_AUDIO |
  637. OBS_SOURCE_DO_NOT_DUPLICATE |
  638. OBS_SOURCE_CONTROLLABLE_MEDIA,
  639. .get_name = ffmpeg_source_getname,
  640. .create = ffmpeg_source_create,
  641. .destroy = ffmpeg_source_destroy,
  642. .get_defaults = ffmpeg_source_defaults,
  643. .get_properties = ffmpeg_source_getproperties,
  644. .activate = ffmpeg_source_activate,
  645. .deactivate = ffmpeg_source_deactivate,
  646. .video_tick = ffmpeg_source_tick,
  647. .missing_files = ffmpeg_source_missingfiles,
  648. .update = ffmpeg_source_update,
  649. .icon_type = OBS_ICON_TYPE_MEDIA,
  650. .media_play_pause = ffmpeg_source_play_pause,
  651. .media_restart = ffmpeg_source_restart,
  652. .media_stop = ffmpeg_source_stop,
  653. .media_get_duration = ffmpeg_source_get_duration,
  654. .media_get_time = ffmpeg_source_get_time,
  655. .media_set_time = ffmpeg_source_set_time,
  656. .media_get_state = ffmpeg_source_get_state,
  657. };