mirror of
https://github.com/yshui/picom.git
synced 2024-10-27 05:24:17 -04:00
75 lines
3.2 KiB
Python
75 lines
3.2 KiB
Python
import xcffib.xproto as xproto
|
|
import xcffib.randr as randr
|
|
import xcffib
|
|
import time
|
|
import random
|
|
import string
|
|
def set_window_name(conn, wid, name):
|
|
prop_name = "_NET_WM_NAME"
|
|
prop_name = conn.core.InternAtom(False, len(prop_name), prop_name).reply().atom
|
|
str_type = "UTF8_STRING"
|
|
str_type = conn.core.InternAtom(True, len(str_type), str_type).reply().atom
|
|
conn.core.ChangePropertyChecked(xproto.PropMode.Replace, wid, prop_name, str_type, 8, len(name), name).check()
|
|
prop_name = "WM_NAME"
|
|
prop_name = conn.core.InternAtom(False, len(prop_name), prop_name).reply().atom
|
|
str_type = "STRING"
|
|
str_type = conn.core.InternAtom(True, len(str_type), str_type).reply().atom
|
|
conn.core.ChangePropertyChecked(xproto.PropMode.Replace, wid, prop_name, str_type, 8, len(name), name).check()
|
|
|
|
def set_window_class(conn, wid, name):
|
|
if not isinstance(name, bytearray):
|
|
name = name.encode()
|
|
name = name+b"\0"+name+b"\0"
|
|
prop_name = "WM_CLASS"
|
|
prop_name = conn.core.InternAtom(False, len(prop_name), prop_name).reply().atom
|
|
str_type = "STRING"
|
|
str_type = conn.core.InternAtom(True, len(str_type), str_type).reply().atom
|
|
conn.core.ChangePropertyChecked(xproto.PropMode.Replace, wid, prop_name, str_type, 8, len(name), name).check()
|
|
|
|
def find_picom_window(conn):
|
|
prop_name = "WM_NAME"
|
|
prop_name = conn.core.InternAtom(True, len(prop_name), prop_name).reply().atom
|
|
setup = conn.get_setup()
|
|
root = setup.roots[0].root
|
|
windows = conn.core.QueryTree(root).reply()
|
|
|
|
ext = xproto.xprotoExtension(conn)
|
|
for w in windows.children:
|
|
name = ext.GetProperty(False, w, prop_name, xproto.GetPropertyType.Any, 0, (2 ** 32) - 1).reply()
|
|
if name.value.buf() == b"picom":
|
|
return w
|
|
|
|
def trigger_root_configure(conn):
|
|
setup = conn.get_setup()
|
|
root = setup.roots[0].root
|
|
# Xorg sends root ConfigureNotify when we add a new mode to an output
|
|
rr = conn(randr.key)
|
|
name = ''.join([random.choice(string.ascii_letters + string.digits) for n in range(0, 32)])
|
|
mode_info = randr.ModeInfo.synthetic(id = 0, width = 1000, height = 1000, dot_clock = 0,
|
|
hsync_start = 0, hsync_end = 0, htotal = 0, hskew = 0, vsync_start = 0, vsync_end = 0,
|
|
vtotal = 0, name_len = len(name), mode_flags = 0)
|
|
|
|
reply = rr.CreateMode(root, mode_info, len(name), name).reply()
|
|
mode = reply.mode
|
|
reply = rr.GetScreenResourcesCurrent(root).reply()
|
|
# our xvfb is setup to only have 1 output
|
|
output = reply.outputs[0]
|
|
rr.AddOutputModeChecked(output, mode).check()
|
|
rr.SetCrtcConfig(reply.crtcs[0], reply.timestamp, reply.config_timestamp, 0, 0, mode, randr.Rotation.Rotate_0, 1, [output]).reply()
|
|
|
|
def find_32bit_visual(conn):
|
|
setup = conn.get_setup()
|
|
render = conn(xcffib.render.key)
|
|
r = render.QueryPictFormats().reply()
|
|
pictfmt_ids = set()
|
|
for pictform in r.formats:
|
|
if (pictform.depth == 32 and
|
|
pictform.type == xcffib.render.PictType.Direct and
|
|
pictform.direct.alpha_mask != 0):
|
|
pictfmt_ids.add(pictform.id)
|
|
print(pictfmt_ids)
|
|
for screen in r.screens:
|
|
for depth in screen.depths:
|
|
for pv in depth.visuals:
|
|
if pv.format in pictfmt_ids:
|
|
return pv.visual
|