vlc-video-source.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743
  1. #include "vlc-video-plugin.h"
  2. #include <media-io/video-frame.h>
  3. #include <util/threading.h>
  4. #include <util/platform.h>
  5. #include <util/dstr.h>
  6. #define do_log(level, format, ...) \
  7. blog(level, "[vlc_source: '%s'] " format, \
  8. obs_source_get_name(ss->source), ##__VA_ARGS__)
  9. #define warn(format, ...) do_log(LOG_WARNING, format, ##__VA_ARGS__)
  10. #define S_PLAYLIST "playlist"
  11. #define S_LOOP "loop"
  12. #define S_BEHAVIOR "playback_behavior"
  13. #define S_BEHAVIOR_STOP_RESTART "stop_restart"
  14. #define S_BEHAVIOR_PAUSE_UNPAUSE "pause_unpause"
  15. #define S_BEHAVIOR_ALWAYS_PLAY "always_play"
  16. #define T_(text) obs_module_text(text)
  17. #define T_PLAYLIST T_("Playlist")
  18. #define T_LOOP T_("LoopPlaylist")
  19. #define T_BEHAVIOR T_("PlaybackBehavior")
  20. #define T_BEHAVIOR_STOP_RESTART T_("PlaybackBehavior.StopRestart")
  21. #define T_BEHAVIOR_PAUSE_UNPAUSE T_("PlaybackBehavior.PauseUnpause")
  22. #define T_BEHAVIOR_ALWAYS_PLAY T_("PlaybackBehavior.AlwaysPlay")
  23. /* ------------------------------------------------------------------------- */
  24. struct media_file_data {
  25. char *path;
  26. libvlc_media_t *media;
  27. };
  28. enum behavior {
  29. BEHAVIOR_STOP_RESTART,
  30. BEHAVIOR_PAUSE_UNPAUSE,
  31. BEHAVIOR_ALWAYS_PLAY,
  32. };
  33. struct vlc_source {
  34. obs_source_t *source;
  35. libvlc_media_player_t *media_player;
  36. libvlc_media_list_player_t *media_list_player;
  37. struct obs_source_frame frame;
  38. struct obs_source_audio audio;
  39. size_t audio_capacity;
  40. pthread_mutex_t mutex;
  41. DARRAY(struct media_file_data) files;
  42. enum behavior behavior;
  43. bool loop;
  44. };
  45. static libvlc_media_t *get_media(struct darray *array, const char *path)
  46. {
  47. DARRAY(struct media_file_data) files;
  48. libvlc_media_t *media = NULL;
  49. files.da = *array;
  50. for (size_t i = 0; i < files.num; i++) {
  51. const char *cur_path = files.array[i].path;
  52. if (strcmp(path, cur_path) == 0) {
  53. media = files.array[i].media;
  54. libvlc_media_retain_(media);
  55. break;
  56. }
  57. }
  58. return media;
  59. }
  60. static inline libvlc_media_t *create_media_from_file(const char *file)
  61. {
  62. return (file && strstr(file, "://") != NULL)
  63. ? libvlc_media_new_location_(libvlc, file)
  64. : libvlc_media_new_path_(libvlc, file);
  65. }
  66. static void free_files(struct darray *array)
  67. {
  68. DARRAY(struct media_file_data) files;
  69. files.da = *array;
  70. for (size_t i = 0; i < files.num; i++) {
  71. bfree(files.array[i].path);
  72. libvlc_media_release_(files.array[i].media);
  73. }
  74. da_free(files);
  75. }
  76. static inline bool chroma_is(const char *chroma, const char *val)
  77. {
  78. return *(uint32_t*)chroma == *(uint32_t*)val;
  79. }
  80. static enum video_format convert_vlc_video_format(char *chroma, bool *full)
  81. {
  82. *full = false;
  83. #define CHROMA_TEST(val, ret) \
  84. if (chroma_is(chroma, val)) return ret
  85. #define CHROMA_CONV(val, new_val, ret) \
  86. do { \
  87. if (chroma_is(chroma, val)) { \
  88. *(uint32_t*)chroma = *(uint32_t*)new_val; \
  89. return ret; \
  90. } \
  91. } while (false)
  92. #define CHROMA_CONV_FULL(val, new_val, ret) \
  93. do { \
  94. *full = true; \
  95. CHROMA_CONV(val, new_val, ret); \
  96. } while (false)
  97. CHROMA_TEST("RGBA", VIDEO_FORMAT_RGBA);
  98. CHROMA_TEST("BGRA", VIDEO_FORMAT_BGRA);
  99. /* 4:2:0 formats */
  100. CHROMA_TEST("NV12", VIDEO_FORMAT_NV12);
  101. CHROMA_TEST("I420", VIDEO_FORMAT_I420);
  102. CHROMA_TEST("IYUV", VIDEO_FORMAT_I420);
  103. CHROMA_CONV("NV21", "NV12", VIDEO_FORMAT_NV12);
  104. CHROMA_CONV("I422", "NV12", VIDEO_FORMAT_NV12);
  105. CHROMA_CONV("Y42B", "NV12", VIDEO_FORMAT_NV12);
  106. CHROMA_CONV("YV12", "NV12", VIDEO_FORMAT_NV12);
  107. CHROMA_CONV("yv12", "NV12", VIDEO_FORMAT_NV12);
  108. CHROMA_CONV_FULL("J420", "J420", VIDEO_FORMAT_I420);
  109. /* 4:2:2 formats */
  110. CHROMA_TEST("UYVY", VIDEO_FORMAT_UYVY);
  111. CHROMA_TEST("UYNV", VIDEO_FORMAT_UYVY);
  112. CHROMA_TEST("UYNY", VIDEO_FORMAT_UYVY);
  113. CHROMA_TEST("Y422", VIDEO_FORMAT_UYVY);
  114. CHROMA_TEST("HDYC", VIDEO_FORMAT_UYVY);
  115. CHROMA_TEST("AVUI", VIDEO_FORMAT_UYVY);
  116. CHROMA_TEST("uyv1", VIDEO_FORMAT_UYVY);
  117. CHROMA_TEST("2vuy", VIDEO_FORMAT_UYVY);
  118. CHROMA_TEST("2Vuy", VIDEO_FORMAT_UYVY);
  119. CHROMA_TEST("2Vu1", VIDEO_FORMAT_UYVY);
  120. CHROMA_TEST("YUY2", VIDEO_FORMAT_YUY2);
  121. CHROMA_TEST("YUYV", VIDEO_FORMAT_YUY2);
  122. CHROMA_TEST("YUNV", VIDEO_FORMAT_YUY2);
  123. CHROMA_TEST("V422", VIDEO_FORMAT_YUY2);
  124. CHROMA_TEST("YVYU", VIDEO_FORMAT_YVYU);
  125. CHROMA_CONV("v210", "UYVY", VIDEO_FORMAT_UYVY);
  126. CHROMA_CONV("cyuv", "UYVY", VIDEO_FORMAT_UYVY);
  127. CHROMA_CONV("CYUV", "UYVY", VIDEO_FORMAT_UYVY);
  128. CHROMA_CONV("VYUY", "UYVY", VIDEO_FORMAT_UYVY);
  129. CHROMA_CONV("NV16", "UYVY", VIDEO_FORMAT_UYVY);
  130. CHROMA_CONV("NV61", "UYVY", VIDEO_FORMAT_UYVY);
  131. CHROMA_CONV("I410", "UYVY", VIDEO_FORMAT_UYVY);
  132. CHROMA_CONV("I422", "UYVY", VIDEO_FORMAT_UYVY);
  133. CHROMA_CONV("Y42B", "UYVY", VIDEO_FORMAT_UYVY);
  134. CHROMA_CONV("J422", "UYVY", VIDEO_FORMAT_UYVY);
  135. /* 4:4:4 formats */
  136. CHROMA_TEST("I444", VIDEO_FORMAT_I444);
  137. CHROMA_CONV_FULL("J444", "J444", VIDEO_FORMAT_I444);
  138. CHROMA_CONV("YUVA", "RGBA", VIDEO_FORMAT_RGBA);
  139. /* 4:4:0 formats */
  140. CHROMA_CONV("I440", "I444", VIDEO_FORMAT_I444);
  141. CHROMA_CONV("J440", "I444", VIDEO_FORMAT_I444);
  142. /* 4:1:0 formats */
  143. CHROMA_CONV("YVU9", "NV12", VIDEO_FORMAT_UYVY);
  144. CHROMA_CONV("I410", "NV12", VIDEO_FORMAT_UYVY);
  145. /* 4:1:1 formats */
  146. CHROMA_CONV("I411", "NV12", VIDEO_FORMAT_UYVY);
  147. CHROMA_CONV("Y41B", "NV12", VIDEO_FORMAT_UYVY);
  148. /* greyscale formats */
  149. CHROMA_TEST("GREY", VIDEO_FORMAT_Y800);
  150. CHROMA_TEST("Y800", VIDEO_FORMAT_Y800);
  151. CHROMA_TEST("Y8 ", VIDEO_FORMAT_Y800);
  152. #undef CHROMA_CONV_FULL
  153. #undef CHROMA_CONV
  154. #undef CHROMA_TEST
  155. *(uint32_t*)chroma = *(uint32_t*)"BGRA";
  156. return VIDEO_FORMAT_BGRA;
  157. }
  158. static inline unsigned get_format_lines(enum video_format format,
  159. unsigned height, size_t plane)
  160. {
  161. switch (format) {
  162. case VIDEO_FORMAT_I420:
  163. case VIDEO_FORMAT_NV12:
  164. return (plane == 0) ? height : height / 2;
  165. case VIDEO_FORMAT_YVYU:
  166. case VIDEO_FORMAT_YUY2:
  167. case VIDEO_FORMAT_UYVY:
  168. case VIDEO_FORMAT_I444:
  169. case VIDEO_FORMAT_RGBA:
  170. case VIDEO_FORMAT_BGRA:
  171. case VIDEO_FORMAT_BGRX:
  172. case VIDEO_FORMAT_Y800:
  173. return height;
  174. case VIDEO_FORMAT_NONE:;
  175. }
  176. return 0;
  177. }
  178. static enum audio_format convert_vlc_audio_format(char *format)
  179. {
  180. #define AUDIO_TEST(val, ret) \
  181. if (chroma_is(format, val)) return ret
  182. #define AUDIO_CONV(val, new_val, ret) \
  183. do { \
  184. if (chroma_is(format, val)) { \
  185. *(uint32_t*)format = *(uint32_t*)new_val; \
  186. return ret; \
  187. } \
  188. } while (false)
  189. AUDIO_TEST("S16N", AUDIO_FORMAT_16BIT);
  190. AUDIO_TEST("S32N", AUDIO_FORMAT_32BIT);
  191. AUDIO_TEST("FL32", AUDIO_FORMAT_FLOAT);
  192. AUDIO_CONV("U16N", "S16N", AUDIO_FORMAT_16BIT);
  193. AUDIO_CONV("U32N", "S32N", AUDIO_FORMAT_32BIT);
  194. AUDIO_CONV("S24N", "S32N", AUDIO_FORMAT_32BIT);
  195. AUDIO_CONV("U24N", "S32N", AUDIO_FORMAT_32BIT);
  196. AUDIO_CONV("FL64", "FL32", AUDIO_FORMAT_FLOAT);
  197. AUDIO_CONV("S16I", "S16N", AUDIO_FORMAT_16BIT);
  198. AUDIO_CONV("U16I", "S16N", AUDIO_FORMAT_16BIT);
  199. AUDIO_CONV("S24I", "S32N", AUDIO_FORMAT_32BIT);
  200. AUDIO_CONV("U24I", "S32N", AUDIO_FORMAT_32BIT);
  201. AUDIO_CONV("S32I", "S32N", AUDIO_FORMAT_32BIT);
  202. AUDIO_CONV("U32I", "S32N", AUDIO_FORMAT_32BIT);
  203. #undef AUDIO_CONV
  204. #undef AUDIO_TEST
  205. *(uint32_t*)format = *(uint32_t*)"FL32";
  206. return AUDIO_FORMAT_FLOAT;
  207. }
  208. /* ------------------------------------------------------------------------- */
  209. static const char *vlcs_get_name(void *unused)
  210. {
  211. UNUSED_PARAMETER(unused);
  212. return obs_module_text("VLCSource");
  213. }
  214. static void vlcs_destroy(void *data)
  215. {
  216. struct vlc_source *c = data;
  217. if (c->media_list_player) {
  218. libvlc_media_list_player_stop_(c->media_list_player);
  219. libvlc_media_list_player_release_(c->media_list_player);
  220. }
  221. if (c->media_player) {
  222. libvlc_media_player_release_(c->media_player);
  223. }
  224. bfree((void*)c->audio.data[0]);
  225. obs_source_frame_free(&c->frame);
  226. free_files(&c->files.da);
  227. pthread_mutex_destroy(&c->mutex);
  228. bfree(c);
  229. }
  230. static void *vlcs_video_lock(void *data, void **planes)
  231. {
  232. struct vlc_source *c = data;
  233. for (size_t i = 0; i < MAX_AV_PLANES && c->frame.data[i] != NULL; i++)
  234. planes[i] = c->frame.data[i];
  235. return NULL;
  236. }
  237. static void vlcs_video_display(void *data, void *picture)
  238. {
  239. struct vlc_source *c = data;
  240. c->frame.timestamp = (uint64_t)libvlc_clock_() * 1000ULL - time_start;
  241. obs_source_output_video(c->source, &c->frame);
  242. UNUSED_PARAMETER(picture);
  243. }
  244. static unsigned vlcs_video_format(void **p_data, char *chroma, unsigned *width,
  245. unsigned *height, unsigned *pitches, unsigned *lines)
  246. {
  247. struct vlc_source *c = *p_data;
  248. enum video_format new_format;
  249. enum video_range_type range;
  250. bool new_range;
  251. unsigned new_width = 0;
  252. unsigned new_height = 0;
  253. size_t i = 0;
  254. new_format = convert_vlc_video_format(chroma, &new_range);
  255. /* This is used because VLC will by default try to use a different
  256. * scaling than what the file uses (probably for optimization reasons).
  257. * For example, if the file is 1920x1080, it will try to render it by
  258. * 1920x1088, which isn't what we want. Calling libvlc_video_get_size
  259. * gets the actual video file's size, and thus fixes the problem.
  260. * However this doesn't work with URLs, so if it returns a 0 value, it
  261. * shouldn't be used. */
  262. libvlc_video_get_size_(c->media_player, 0, &new_width, &new_height);
  263. if (new_width && new_height) {
  264. *width = new_width;
  265. *height = new_height;
  266. }
  267. /* don't allocate a new frame if format/width/height hasn't changed */
  268. if (c->frame.format != new_format ||
  269. c->frame.width != *width ||
  270. c->frame.height != *height) {
  271. obs_source_frame_free(&c->frame);
  272. obs_source_frame_init(&c->frame, new_format, *width, *height);
  273. c->frame.format = new_format;
  274. c->frame.full_range = new_range;
  275. range = c->frame.full_range ?
  276. VIDEO_RANGE_FULL : VIDEO_RANGE_PARTIAL;
  277. video_format_get_parameters(VIDEO_CS_DEFAULT, range,
  278. c->frame.color_matrix,
  279. c->frame.color_range_min,
  280. c->frame.color_range_max);
  281. }
  282. while (c->frame.data[i]) {
  283. pitches[i] = (unsigned)c->frame.linesize[i];
  284. lines[i] = get_format_lines(c->frame.format, *height, i);
  285. i++;
  286. }
  287. return 1;
  288. }
  289. static void vlcs_audio_play(void *data, const void *samples, unsigned count,
  290. int64_t pts)
  291. {
  292. struct vlc_source *c = data;
  293. size_t size = get_audio_size(c->audio.format, c->audio.speakers, count);
  294. if (c->audio_capacity < count) {
  295. c->audio.data[0] = brealloc((void*)c->audio.data[0], size);
  296. c->audio_capacity = count;
  297. }
  298. memcpy((void*)c->audio.data[0], samples, size);
  299. c->audio.timestamp = (uint64_t)pts * 1000ULL - time_start;
  300. c->audio.frames = count;
  301. obs_source_output_audio(c->source, &c->audio);
  302. }
  303. static int vlcs_audio_setup(void **p_data, char *format, unsigned *rate,
  304. unsigned *channels)
  305. {
  306. struct vlc_source *c = *p_data;
  307. enum audio_format new_audio_format;
  308. new_audio_format = convert_vlc_audio_format(format);
  309. if (*channels > 2)
  310. *channels = 2;
  311. /* don't free audio data if the data is the same format */
  312. if (c->audio.format == new_audio_format &&
  313. c->audio.samples_per_sec == *rate &&
  314. c->audio.speakers == (enum speaker_layout)*channels)
  315. return 0;
  316. c->audio_capacity = 0;
  317. bfree((void*)c->audio.data[0]);
  318. memset(&c->audio, 0, sizeof(c->audio));
  319. c->audio.speakers = (enum speaker_layout)*channels;
  320. c->audio.samples_per_sec = *rate;
  321. c->audio.format = new_audio_format;
  322. return 0;
  323. }
  324. static void add_file(struct vlc_source *c, struct darray *array,
  325. const char *path)
  326. {
  327. DARRAY(struct media_file_data) new_files;
  328. struct media_file_data data;
  329. struct dstr new_path = {0};
  330. libvlc_media_t *new_media;
  331. bool is_url = path && strstr(path, "://") != NULL;
  332. new_files.da = *array;
  333. dstr_copy(&new_path, path);
  334. #ifdef _WIN32
  335. if (!is_url)
  336. dstr_replace(&new_path, "/", "\\");
  337. #endif
  338. path = new_path.array;
  339. new_media = get_media(&c->files.da, path);
  340. if (!new_media)
  341. new_media = get_media(&new_files.da, path);
  342. if (!new_media)
  343. new_media = create_media_from_file(path);
  344. if (new_media) {
  345. if (is_url)
  346. libvlc_media_add_option_(new_media,
  347. ":network-caching=100");
  348. data.path = new_path.array;
  349. data.media = new_media;
  350. da_push_back(new_files, &data);
  351. } else {
  352. dstr_free(&new_path);
  353. }
  354. *array = new_files.da;
  355. }
  356. static bool valid_extension(const char *ext)
  357. {
  358. struct dstr test = {0};
  359. bool valid = false;
  360. const char *b;
  361. const char *e;
  362. if (!ext || !*ext)
  363. return false;
  364. b = EXTENSIONS_MEDIA + 1;
  365. e = strchr(b, ';');
  366. for (;;) {
  367. if (e) dstr_ncopy(&test, b, e - b);
  368. else dstr_copy(&test, b);
  369. if (dstr_cmp(&test, ext) == 0) {
  370. valid = true;
  371. break;
  372. }
  373. if (!e)
  374. break;
  375. b = e + 2;
  376. e = strchr(b, ';');
  377. }
  378. dstr_free(&test);
  379. return valid;
  380. }
  381. static void vlcs_update(void *data, obs_data_t *settings)
  382. {
  383. DARRAY(struct media_file_data) new_files;
  384. DARRAY(struct media_file_data) old_files;
  385. libvlc_media_list_t *media_list;
  386. struct vlc_source *c = data;
  387. obs_data_array_t *array;
  388. const char *behavior;
  389. size_t count;
  390. da_init(new_files);
  391. da_init(old_files);
  392. array = obs_data_get_array(settings, S_PLAYLIST);
  393. count = obs_data_array_count(array);
  394. c->loop = obs_data_get_bool(settings, S_LOOP);
  395. behavior = obs_data_get_string(settings, S_BEHAVIOR);
  396. if (astrcmpi(behavior, S_BEHAVIOR_PAUSE_UNPAUSE) == 0) {
  397. c->behavior = BEHAVIOR_PAUSE_UNPAUSE;
  398. } else if (astrcmpi(behavior, S_BEHAVIOR_ALWAYS_PLAY) == 0) {
  399. c->behavior = BEHAVIOR_ALWAYS_PLAY;
  400. } else { /* S_BEHAVIOR_STOP_RESTART */
  401. c->behavior = BEHAVIOR_STOP_RESTART;
  402. }
  403. /* ------------------------------------- */
  404. /* create new list of sources */
  405. for (size_t i = 0; i < count; i++) {
  406. obs_data_t *item = obs_data_array_item(array, i);
  407. const char *path = obs_data_get_string(item, "value");
  408. os_dir_t *dir = os_opendir(path);
  409. if (dir) {
  410. struct dstr dir_path = {0};
  411. struct os_dirent *ent;
  412. for (;;) {
  413. const char *ext;
  414. ent = os_readdir(dir);
  415. if (!ent)
  416. break;
  417. if (ent->directory)
  418. continue;
  419. ext = os_get_path_extension(ent->d_name);
  420. if (!valid_extension(ext))
  421. continue;
  422. dstr_copy(&dir_path, path);
  423. dstr_cat_ch(&dir_path, '/');
  424. dstr_cat(&dir_path, ent->d_name);
  425. add_file(c, &new_files.da, dir_path.array);
  426. }
  427. dstr_free(&dir_path);
  428. os_closedir(dir);
  429. } else {
  430. add_file(c, &new_files.da, path);
  431. }
  432. obs_data_release(item);
  433. }
  434. /* ------------------------------------- */
  435. /* update settings data */
  436. libvlc_media_list_player_stop_(c->media_list_player);
  437. pthread_mutex_lock(&c->mutex);
  438. old_files.da = c->files.da;
  439. c->files.da = new_files.da;
  440. pthread_mutex_unlock(&c->mutex);
  441. /* ------------------------------------- */
  442. /* clean up and restart playback */
  443. free_files(&old_files.da);
  444. media_list = libvlc_media_list_new_(libvlc);
  445. libvlc_media_list_lock_(media_list);
  446. for (size_t i = 0; i < c->files.num; i++)
  447. libvlc_media_list_add_media_(media_list,
  448. c->files.array[i].media);
  449. libvlc_media_list_unlock_(media_list);
  450. libvlc_media_list_player_set_media_list_(c->media_list_player,
  451. media_list);
  452. libvlc_media_list_release_(media_list);
  453. libvlc_media_list_player_set_playback_mode_(c->media_list_player,
  454. c->loop ? libvlc_playback_mode_loop :
  455. libvlc_playback_mode_default);
  456. if (c->files.num &&
  457. (c->behavior == BEHAVIOR_ALWAYS_PLAY || obs_source_active(c->source)))
  458. libvlc_media_list_player_play_(c->media_list_player);
  459. else
  460. obs_source_output_video(c->source, NULL);
  461. obs_data_array_release(array);
  462. }
  463. static void vlcs_stopped(const struct libvlc_event_t *event, void *data)
  464. {
  465. struct vlc_source *c = data;
  466. if (!c->loop)
  467. obs_source_output_video(c->source, NULL);
  468. UNUSED_PARAMETER(event);
  469. }
  470. static void *vlcs_create(obs_data_t *settings, obs_source_t *source)
  471. {
  472. struct vlc_source *c = bzalloc(sizeof(*c));
  473. c->source = source;
  474. pthread_mutex_init_value(&c->mutex);
  475. if (pthread_mutex_init(&c->mutex, NULL) != 0)
  476. goto error;
  477. if (!load_libvlc())
  478. goto error;
  479. c->media_list_player = libvlc_media_list_player_new_(libvlc);
  480. if (!c->media_list_player)
  481. goto error;
  482. c->media_player = libvlc_media_player_new_(libvlc);
  483. if (!c->media_player)
  484. goto error;
  485. libvlc_media_list_player_set_media_player_(c->media_list_player,
  486. c->media_player);
  487. libvlc_video_set_callbacks_(c->media_player,
  488. vlcs_video_lock, NULL, vlcs_video_display,
  489. c);
  490. libvlc_video_set_format_callbacks_(c->media_player,
  491. vlcs_video_format, NULL);
  492. libvlc_audio_set_callbacks_(c->media_player,
  493. vlcs_audio_play, NULL, NULL, NULL, NULL, c);
  494. libvlc_audio_set_format_callbacks_(c->media_player,
  495. vlcs_audio_setup, NULL);
  496. libvlc_event_manager_t *event_manager;
  497. event_manager = libvlc_media_player_event_manager_(c->media_player);
  498. libvlc_event_attach_(event_manager, libvlc_MediaPlayerEndReached,
  499. vlcs_stopped, c);
  500. obs_source_update(source, NULL);
  501. UNUSED_PARAMETER(settings);
  502. return c;
  503. error:
  504. vlcs_destroy(c);
  505. return NULL;
  506. }
  507. static void vlcs_activate(void *data)
  508. {
  509. struct vlc_source *c = data;
  510. if (c->behavior == BEHAVIOR_STOP_RESTART) {
  511. libvlc_media_list_player_play_(c->media_list_player);
  512. } else if (c->behavior == BEHAVIOR_PAUSE_UNPAUSE) {
  513. libvlc_media_list_player_play_(c->media_list_player);
  514. }
  515. }
  516. static void vlcs_deactivate(void *data)
  517. {
  518. struct vlc_source *c = data;
  519. if (c->behavior == BEHAVIOR_STOP_RESTART) {
  520. libvlc_media_list_player_stop_(c->media_list_player);
  521. obs_source_output_video(c->source, NULL);
  522. } else if (c->behavior == BEHAVIOR_PAUSE_UNPAUSE) {
  523. libvlc_media_list_player_pause_(c->media_list_player);
  524. }
  525. }
  526. static void vlcs_defaults(obs_data_t *settings)
  527. {
  528. obs_data_set_default_bool(settings, S_LOOP, true);
  529. obs_data_set_default_string(settings, S_BEHAVIOR,
  530. S_BEHAVIOR_STOP_RESTART);
  531. }
  532. static obs_properties_t *vlcs_properties(void *data)
  533. {
  534. obs_properties_t *ppts = obs_properties_create();
  535. struct vlc_source *c = data;
  536. struct dstr filter = {0};
  537. struct dstr exts = {0};
  538. struct dstr path = {0};
  539. obs_property_t *p;
  540. obs_properties_add_bool(ppts, S_LOOP, T_LOOP);
  541. if (c) {
  542. pthread_mutex_lock(&c->mutex);
  543. if (c->files.num) {
  544. struct media_file_data *last = da_end(c->files);
  545. const char *slash;
  546. dstr_copy(&path, last->path);
  547. dstr_replace(&path, "\\", "/");
  548. slash = strrchr(path.array, '/');
  549. if (slash)
  550. dstr_resize(&path, slash - path.array + 1);
  551. }
  552. pthread_mutex_unlock(&c->mutex);
  553. }
  554. p = obs_properties_add_list(ppts, S_BEHAVIOR, T_BEHAVIOR,
  555. OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_STRING);
  556. obs_property_list_add_string(p, T_BEHAVIOR_STOP_RESTART,
  557. S_BEHAVIOR_STOP_RESTART);
  558. obs_property_list_add_string(p, T_BEHAVIOR_PAUSE_UNPAUSE,
  559. S_BEHAVIOR_PAUSE_UNPAUSE);
  560. obs_property_list_add_string(p, T_BEHAVIOR_ALWAYS_PLAY,
  561. S_BEHAVIOR_ALWAYS_PLAY);
  562. dstr_cat(&filter, "Media Files (");
  563. dstr_copy(&exts, EXTENSIONS_MEDIA);
  564. dstr_replace(&exts, ";", " ");
  565. dstr_cat_dstr(&filter, &exts);
  566. dstr_cat(&filter, ");;Video Files (");
  567. dstr_copy(&exts, EXTENSIONS_VIDEO);
  568. dstr_replace(&exts, ";", " ");
  569. dstr_cat_dstr(&filter, &exts);
  570. dstr_cat(&filter, ");;Audio Files (");
  571. dstr_copy(&exts, EXTENSIONS_AUDIO);
  572. dstr_replace(&exts, ";", " ");
  573. dstr_cat_dstr(&filter, &exts);
  574. dstr_cat(&filter, ");;Playlist Files (");
  575. dstr_copy(&exts, EXTENSIONS_PLAYLIST);
  576. dstr_replace(&exts, ";", " ");
  577. dstr_cat_dstr(&filter, &exts);
  578. dstr_cat(&filter, ")");
  579. obs_properties_add_editable_list(ppts, S_PLAYLIST, T_PLAYLIST,
  580. OBS_EDITABLE_LIST_TYPE_FILES_AND_URLS,
  581. filter.array, path.array);
  582. dstr_free(&path);
  583. dstr_free(&filter);
  584. dstr_free(&exts);
  585. return ppts;
  586. }
  587. struct obs_source_info vlc_source_info = {
  588. .id = "vlc_source",
  589. .type = OBS_SOURCE_TYPE_INPUT,
  590. .output_flags = OBS_SOURCE_ASYNC_VIDEO | OBS_SOURCE_AUDIO,
  591. .get_name = vlcs_get_name,
  592. .create = vlcs_create,
  593. .destroy = vlcs_destroy,
  594. .update = vlcs_update,
  595. .get_defaults = vlcs_defaults,
  596. .get_properties = vlcs_properties,
  597. .activate = vlcs_activate,
  598. .deactivate = vlcs_deactivate
  599. };