Merge pull request #303 from fryshorts/xshm-different-xserver
linux-xshm: Allow capturing from a different X servermaster
commit
963f052c2f
|
@ -1,6 +1,8 @@
|
|||
X11SharedMemoryScreenInput="Screen Capture (XSHM)"
|
||||
Screen="Screen"
|
||||
CaptureCursor="Capture Cursor"
|
||||
AdvancedSettings="Advanced Settings"
|
||||
XServer="X Server"
|
||||
XCCapture="Window Capture (Xcomposite)"
|
||||
Window="Window"
|
||||
CropTop="Crop Top (pixels)"
|
||||
|
|
|
@ -22,42 +22,56 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|||
#include <X11/Xutil.h>
|
||||
|
||||
#include <obs-module.h>
|
||||
#include <util/dstr.h>
|
||||
#include "xcursor.h"
|
||||
#include "xhelpers.h"
|
||||
|
||||
#define XSHM_DATA(voidptr) struct xshm_data *data = voidptr;
|
||||
|
||||
#define blog(level, msg, ...) blog(level, "xshm-input: " msg, ##__VA_ARGS__)
|
||||
|
||||
struct xshm_data {
|
||||
/** The source object */
|
||||
obs_source_t *source;
|
||||
/** Xlib display object */
|
||||
Display *dpy;
|
||||
/** Xlib screen object */
|
||||
Screen *screen;
|
||||
|
||||
/** user setting - the server name to capture from */
|
||||
char *server;
|
||||
/** user setting - the id of the screen that should be captured */
|
||||
uint_fast32_t screen_id;
|
||||
/** root coordinates for the capture */
|
||||
int_fast32_t x_org, y_org;
|
||||
/** size for the capture */
|
||||
int_fast32_t width, height;
|
||||
|
||||
/** shared memory management object */
|
||||
xshm_t *xshm;
|
||||
/** the texture used to display the capture */
|
||||
gs_texture_t *texture;
|
||||
|
||||
bool show_cursor;
|
||||
/** cursor object for displaying the server */
|
||||
xcursor_t *cursor;
|
||||
|
||||
/** user setting - if cursor should be displayed */
|
||||
bool show_cursor;
|
||||
/** set if xinerama is available and active on the screen */
|
||||
bool use_xinerama;
|
||||
/** user setting - if advanced settings should be displayed */
|
||||
bool advanced;
|
||||
};
|
||||
|
||||
/**
|
||||
* Resize the texture
|
||||
*
|
||||
* This will automatically create the texture if it does not exist
|
||||
*
|
||||
* @note requires to be called within the obs graphics context
|
||||
*/
|
||||
static void xshm_resize_texture(struct xshm_data *data)
|
||||
static inline void xshm_resize_texture(struct xshm_data *data)
|
||||
{
|
||||
obs_enter_graphics();
|
||||
|
||||
if (data->texture)
|
||||
gs_texture_destroy(data->texture);
|
||||
data->texture = gs_texture_create(data->width, data->height,
|
||||
GS_BGRA, 1, NULL, GS_DYNAMIC);
|
||||
|
||||
obs_leave_graphics();
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -65,15 +79,13 @@ static void xshm_resize_texture(struct xshm_data *data)
|
|||
*
|
||||
* @return < 0 on error, 0 when size is unchanged, > 1 on size change
|
||||
*/
|
||||
static int_fast32_t xshm_update_geometry(struct xshm_data *data,
|
||||
obs_data_t *settings)
|
||||
static int_fast32_t xshm_update_geometry(struct xshm_data *data)
|
||||
{
|
||||
int_fast32_t old_width = data->width;
|
||||
int_fast32_t old_height = data->height;
|
||||
int_fast32_t screen = obs_data_get_int(settings, "screen");
|
||||
|
||||
if (data->use_xinerama) {
|
||||
if (xinerama_screen_geo(data->dpy, screen,
|
||||
if (xinerama_screen_geo(data->dpy, data->screen_id,
|
||||
&data->x_org, &data->y_org,
|
||||
&data->width, &data->height) < 0) {
|
||||
return -1;
|
||||
|
@ -83,19 +95,19 @@ static int_fast32_t xshm_update_geometry(struct xshm_data *data,
|
|||
else {
|
||||
data->x_org = 0;
|
||||
data->y_org = 0;
|
||||
if (x11_screen_geo(data->dpy, screen,
|
||||
if (x11_screen_geo(data->dpy, data->screen_id,
|
||||
&data->width, &data->height) < 0) {
|
||||
return -1;
|
||||
}
|
||||
data->screen = XScreenOfDisplay(data->dpy, screen);
|
||||
data->screen = XScreenOfDisplay(data->dpy, data->screen_id);
|
||||
}
|
||||
|
||||
if (!data->width || !data->height) {
|
||||
blog(LOG_ERROR, "xshm-input: Failed to get geometry");
|
||||
blog(LOG_ERROR, "Failed to get geometry");
|
||||
return -1;
|
||||
}
|
||||
|
||||
blog(LOG_INFO, "xshm-input: Geometry %"PRIdFAST32"x%"PRIdFAST32
|
||||
blog(LOG_INFO, "Geometry %"PRIdFAST32"x%"PRIdFAST32
|
||||
" @ %"PRIdFAST32",%"PRIdFAST32,
|
||||
data->width, data->height, data->x_org, data->y_org);
|
||||
|
||||
|
@ -113,6 +125,87 @@ static const char* xshm_getname(void)
|
|||
return obs_module_text("X11SharedMemoryScreenInput");
|
||||
}
|
||||
|
||||
/**
|
||||
* Stop the capture
|
||||
*/
|
||||
static void xshm_capture_stop(struct xshm_data *data)
|
||||
{
|
||||
obs_enter_graphics();
|
||||
|
||||
if (data->texture) {
|
||||
gs_texture_destroy(data->texture);
|
||||
data->texture = NULL;
|
||||
}
|
||||
if (data->cursor) {
|
||||
xcursor_destroy(data->cursor);
|
||||
data->cursor = NULL;
|
||||
}
|
||||
|
||||
obs_leave_graphics();
|
||||
|
||||
if (data->xshm) {
|
||||
xshm_detach(data->xshm);
|
||||
data->xshm = NULL;
|
||||
}
|
||||
|
||||
if (data->dpy) {
|
||||
XSync(data->dpy, true);
|
||||
XCloseDisplay(data->dpy);
|
||||
data->dpy = NULL;
|
||||
}
|
||||
|
||||
if (data->server) {
|
||||
bfree(data->server);
|
||||
data->server = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Start the capture
|
||||
*/
|
||||
static void xshm_capture_start(struct xshm_data *data)
|
||||
{
|
||||
const char *server = (data->advanced && *data->server)
|
||||
? data->server : NULL;
|
||||
|
||||
data->dpy = XOpenDisplay(server);
|
||||
if (!data->dpy) {
|
||||
blog(LOG_ERROR, "Unable to open X display !");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
if (!XShmQueryExtension(data->dpy)) {
|
||||
blog(LOG_ERROR, "XShm extension not found !");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
data->use_xinerama = xinerama_is_active(data->dpy) ? true : false;
|
||||
|
||||
if (xshm_update_geometry(data) < 0) {
|
||||
blog(LOG_ERROR, "failed to update geometry !");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
data->xshm = xshm_attach(data->dpy, data->screen,
|
||||
data->width, data->height);
|
||||
if (!data->xshm) {
|
||||
blog(LOG_ERROR, "failed to attach shm !");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
obs_enter_graphics();
|
||||
|
||||
data->cursor = xcursor_init(data->dpy);
|
||||
xcursor_offset(data->cursor, data->x_org, data->y_org);
|
||||
xshm_resize_texture(data);
|
||||
|
||||
obs_leave_graphics();
|
||||
|
||||
return;
|
||||
fail:
|
||||
xshm_capture_stop(data);
|
||||
}
|
||||
|
||||
/**
|
||||
* Update the capture with changed settings
|
||||
*/
|
||||
|
@ -120,25 +213,14 @@ static void xshm_update(void *vptr, obs_data_t *settings)
|
|||
{
|
||||
XSHM_DATA(vptr);
|
||||
|
||||
xshm_capture_stop(data);
|
||||
|
||||
data->screen_id = obs_data_get_int(settings, "screen");
|
||||
data->show_cursor = obs_data_get_bool(settings, "show_cursor");
|
||||
data->advanced = obs_data_get_bool(settings, "advanced");
|
||||
data->server = bstrdup(obs_data_get_string(settings, "server"));
|
||||
|
||||
if (data->xshm)
|
||||
xshm_detach(data->xshm);
|
||||
|
||||
if (xshm_update_geometry(data, settings) < 0) {
|
||||
blog(LOG_ERROR, "xshm-input: failed to update geometry !");
|
||||
return;
|
||||
}
|
||||
|
||||
xshm_resize_texture(data);
|
||||
xcursor_offset(data->cursor, data->x_org, data->y_org);
|
||||
|
||||
data->xshm = xshm_attach(data->dpy, data->screen,
|
||||
data->width, data->height);
|
||||
if (!data->xshm) {
|
||||
blog(LOG_ERROR, "xshm-input: failed to attach shm !");
|
||||
return;
|
||||
}
|
||||
xshm_capture_start(data);
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -148,29 +230,116 @@ static void xshm_defaults(obs_data_t *defaults)
|
|||
{
|
||||
obs_data_set_default_int(defaults, "screen", 0);
|
||||
obs_data_set_default_bool(defaults, "show_cursor", true);
|
||||
obs_data_set_default_bool(defaults, "advanced", false);
|
||||
}
|
||||
|
||||
/**
|
||||
* Toggle visibility of advanced settings
|
||||
*/
|
||||
static bool xshm_toggle_advanced(obs_properties_t *props,
|
||||
obs_property_t *p, obs_data_t *settings)
|
||||
{
|
||||
UNUSED_PARAMETER(p);
|
||||
const bool visible = obs_data_get_bool(settings, "advanced");
|
||||
obs_property_t *server = obs_properties_get(props, "server");
|
||||
|
||||
obs_property_set_visible(server, visible);
|
||||
|
||||
/* trigger server changed callback so the screen list is refreshed */
|
||||
obs_property_modified(server, settings);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* The x server was changed
|
||||
*/
|
||||
static bool xshm_server_changed(obs_properties_t *props,
|
||||
obs_property_t *p, obs_data_t *settings)
|
||||
{
|
||||
UNUSED_PARAMETER(p);
|
||||
|
||||
bool advanced = obs_data_get_bool(settings, "advanced");
|
||||
int_fast32_t old_screen = obs_data_get_int(settings, "screen");
|
||||
const char *server = obs_data_get_string(settings, "server");
|
||||
obs_property_t *screens = obs_properties_get(props, "screen");
|
||||
|
||||
/* we want a real NULL here in case there is no string here */
|
||||
server = (advanced && *server) ? server : NULL;
|
||||
|
||||
obs_property_list_clear(screens);
|
||||
|
||||
Display *dpy = XOpenDisplay(server);
|
||||
if (!dpy) {
|
||||
obs_property_set_enabled(screens, false);
|
||||
return true;
|
||||
}
|
||||
|
||||
struct dstr screen_info;
|
||||
dstr_init(&screen_info);
|
||||
bool xinerama = xinerama_is_active(dpy);
|
||||
int_fast32_t count = (xinerama) ?
|
||||
xinerama_screen_count(dpy) : XScreenCount(dpy);
|
||||
|
||||
for (int_fast32_t i = 0; i < count; ++i) {
|
||||
int_fast32_t x, y, w, h;
|
||||
x = y = w = h = 0;
|
||||
|
||||
if (xinerama)
|
||||
xinerama_screen_geo(dpy, i, &x, &y, &w, &h);
|
||||
else
|
||||
x11_screen_geo(dpy, i, &w, &h);
|
||||
|
||||
dstr_printf(&screen_info, "Screen %"PRIuFAST32" (%"PRIuFAST32
|
||||
"x%"PRIuFAST32" @ %"PRIuFAST32
|
||||
",%"PRIuFAST32")", i, w, h, x, y);
|
||||
|
||||
obs_property_list_add_int(screens, screen_info.array, i);
|
||||
}
|
||||
|
||||
/* handle missing screen */
|
||||
if (old_screen + 1 > count) {
|
||||
dstr_printf(&screen_info, "Screen %"PRIuFAST32" (not found)",
|
||||
old_screen);
|
||||
size_t index = obs_property_list_add_int(screens,
|
||||
screen_info.array, old_screen);
|
||||
obs_property_list_item_disable(screens, index, true);
|
||||
|
||||
}
|
||||
|
||||
dstr_free(&screen_info);
|
||||
|
||||
XCloseDisplay(dpy);
|
||||
obs_property_set_enabled(screens, true);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the properties for the capture
|
||||
*/
|
||||
static obs_properties_t *xshm_properties(void *unused)
|
||||
static obs_properties_t *xshm_properties(void *vptr)
|
||||
{
|
||||
UNUSED_PARAMETER(unused);
|
||||
XSHM_DATA(vptr);
|
||||
|
||||
obs_properties_t *props = obs_properties_create();
|
||||
int_fast32_t screen_max;
|
||||
|
||||
Display *dpy = XOpenDisplay(NULL);
|
||||
screen_max = xinerama_is_active(dpy)
|
||||
? xinerama_screen_count(dpy)
|
||||
: XScreenCount(dpy);
|
||||
screen_max = (screen_max) ? screen_max - 1 : 0;
|
||||
XCloseDisplay(dpy);
|
||||
|
||||
obs_properties_add_int(props, "screen",
|
||||
obs_module_text("Screen"), 0, screen_max, 1);
|
||||
obs_properties_add_list(props, "screen", obs_module_text("Screen"),
|
||||
OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_INT);
|
||||
obs_properties_add_bool(props, "show_cursor",
|
||||
obs_module_text("CaptureCursor"));
|
||||
obs_property_t *advanced = obs_properties_add_bool(props, "advanced",
|
||||
obs_module_text("AdvancedSettings"));
|
||||
obs_property_t *server = obs_properties_add_text(props, "server",
|
||||
obs_module_text("XServer"), OBS_TEXT_DEFAULT);
|
||||
|
||||
obs_property_set_modified_callback(advanced, xshm_toggle_advanced);
|
||||
obs_property_set_modified_callback(server, xshm_server_changed);
|
||||
|
||||
/* trigger server callback to get screen count ... */
|
||||
obs_data_t *settings = obs_source_get_settings(data->source);
|
||||
obs_property_modified(server, settings);
|
||||
obs_data_release(settings);
|
||||
|
||||
return props;
|
||||
}
|
||||
|
@ -185,19 +354,7 @@ static void xshm_destroy(void *vptr)
|
|||
if (!data)
|
||||
return;
|
||||
|
||||
obs_enter_graphics();
|
||||
|
||||
if (data->texture)
|
||||
gs_texture_destroy(data->texture);
|
||||
if (data->cursor)
|
||||
xcursor_destroy(data->cursor);
|
||||
|
||||
obs_leave_graphics();
|
||||
|
||||
if (data->xshm)
|
||||
xshm_detach(data->xshm);
|
||||
if (data->dpy)
|
||||
XCloseDisplay(data->dpy);
|
||||
xshm_capture_stop(data);
|
||||
|
||||
bfree(data);
|
||||
}
|
||||
|
@ -207,33 +364,12 @@ static void xshm_destroy(void *vptr)
|
|||
*/
|
||||
static void *xshm_create(obs_data_t *settings, obs_source_t *source)
|
||||
{
|
||||
UNUSED_PARAMETER(source);
|
||||
|
||||
struct xshm_data *data = bzalloc(sizeof(struct xshm_data));
|
||||
|
||||
data->dpy = XOpenDisplay(NULL);
|
||||
if (!data->dpy) {
|
||||
blog(LOG_ERROR, "xshm-input: Unable to open X display !");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
if (!XShmQueryExtension(data->dpy)) {
|
||||
blog(LOG_ERROR, "xshm-input: XShm extension not found !");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
data->use_xinerama = xinerama_is_active(data->dpy) ? true : false;
|
||||
|
||||
obs_enter_graphics();
|
||||
data->cursor = xcursor_init(data->dpy);
|
||||
obs_leave_graphics();
|
||||
data->source = source;
|
||||
|
||||
xshm_update(data, settings);
|
||||
|
||||
return data;
|
||||
fail:
|
||||
xshm_destroy(data);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -244,7 +380,7 @@ static void xshm_video_tick(void *vptr, float seconds)
|
|||
UNUSED_PARAMETER(seconds);
|
||||
XSHM_DATA(vptr);
|
||||
|
||||
if (!data->xshm)
|
||||
if (!data->texture)
|
||||
return;
|
||||
|
||||
obs_enter_graphics();
|
||||
|
@ -266,7 +402,7 @@ static void xshm_video_render(void *vptr, gs_effect_t *effect)
|
|||
{
|
||||
XSHM_DATA(vptr);
|
||||
|
||||
if (!data->xshm)
|
||||
if (!data->texture)
|
||||
return;
|
||||
|
||||
gs_eparam_t *image = gs_effect_get_param_by_name(effect, "image");
|
||||
|
|
Loading…
Reference in New Issue