vlc-video-source.c 20 KB

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