vlc-video-source.c 23 KB

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