awesome/tests/examples/awful/template.lua

316 lines
8.7 KiB
Lua

local file_path, image_path = ...
require("_common_template")(...)
local cairo = require("lgi").cairo
local color = require( "gears.color" )
local shape = require( "gears.shape" )
local beautiful = require( "beautiful" )
local wibox = require( "wibox" )
local screenshot = require( "awful.screenshot")
local function wrap_titlebar(tb, width, height, args)
local bg, fg
if args.honor_titlebar_colors then
bg = tb.drawable.background_color or tb.args.bg_normal
fg = tb.drawable.foreground_color or tb.args.fg_normal
else
bg, fg = tb.args.bg_normal, tb.args.fg_normal
end
return wibox.widget {
tb.drawable.widget,
bg = bg,
fg = fg,
forced_width = width,
forced_height = height,
widget = wibox.container.background
}
end
local function client_widget(c, col, label, args)
local geo = c:geometry()
local bw = c.border_width or beautiful.border_width or 0
local bc = c.border_color or beautiful.border_color
local l = wibox.layout.align.vertical()
l.fill_space = true
local tbs = c._private and c._private.titlebars or {}
local map = {
top = "set_first",
left = "set_first",
bottom = "set_third",
right = "set_third",
}
for _, position in ipairs{"top", "bottom"} do
local tb = tbs[position]
if tb then
l[map[position]](l, wrap_titlebar(tb, c:geometry().width, tb.args.height or 16, args))
end
end
for _, position in ipairs{"left", "right"} do
local tb = tbs[position]
if tb then
l[map[position]](l, wrap_titlebar(tb, tb.args.width or 16, c:geometry().height), args)
end
end
local l2 = wibox.layout.align.horizontal()
l2.fill_space = true
l:set_second(l2)
l.forced_width = c.width
l.forced_height = c.height
return wibox.widget {
{
l,
{
text = label or "",
halign = "center",
valign = "center",
widget = wibox.widget.textbox
},
layout = wibox.layout.stack
},
border_width = bw,
border_color = bc,
shape_clip = true,
border_strategy = "inner",
opacity = c.opacity,
fg = beautiful.fg_normal or "#000000",
bg = col,
shape = function(cr2, w, h)
if c.shape then
c.shape(cr2, w, h)
else
return shape.rounded_rect(cr2, w, h, args.radius or 5)
end
end,
forced_width = geo.width + 2*bw,
forced_height = geo.height + 2*bw,
widget = wibox.container.background,
}
end
-- Mock the c:content(), it cannot be shimmed because the "real" one uses
-- a raw surface rather than a LGI one.
function screenshot._screenshot_methods.client(self)
local c = self.client
local geo = c:geometry()
local wdg = client_widget(
c, c.color or geo._color or beautiful.bg_normal, "", {}
)
local sur = wibox.widget.draw_to_image_surface(wdg, geo.width, geo.height)
return sur, geo
end
function screenshot._screenshot_methods.root()
local w, h = root.size()
local img = cairo.ImageSurface.create(cairo.Format.ARGB32, 1, 1)
return img, {x=0,y=0,width=w,height=h}
end
-- Run the test
local args = loadfile(file_path)() or {}
-- Emulate the event loop for 5 iterations
for _ = 1, 5 do
require("gears.timer").run_delayed_calls_now()
end
-- Draw the result
local img = cairo.SvgSurface.create(image_path..".svg", screen._get_extents() )
local cr = cairo.Context(img)
-- Draw a mouse cursor at [x,y]
local function draw_mouse(x, y)
cr:move_to(x, y)
cr:rel_line_to( 0, 10)
cr:rel_line_to( 3, -2)
cr:rel_line_to( 3, 4)
cr:rel_line_to( 2, 0)
cr:rel_line_to(-3, -4)
cr:rel_line_to( 4, 0)
cr:close_path()
cr:fill()
end
-- Print an outline for the screens
if not rawget(screen, "no_outline") then
for s in screen do
cr:save()
-- Draw the screen outline
if s._no_outline ~= true then
cr:set_source(color("#00000044"))
cr:set_line_width(1.5)
cr:set_dash({10,4},1)
cr:rectangle(s.geometry.x+0.75,s.geometry.y+0.75,s.geometry.width-1.5,s.geometry.height-1.5)
cr:stroke()
-- Draw the workarea outline
cr:set_source(color("#00000033"))
cr:rectangle(s.workarea.x,s.workarea.y,s.workarea.width,s.workarea.height)
cr:stroke()
end
-- Draw the padding outline
--TODO
cr:restore()
end
end
-- Draw the wallpaper (if any).
if root._wallpaper_pattern then
cr:set_source_rgb(1,0,0)
cr:set_line_width(2)
cr:rectangle(0, 0, root.size())
cr:stroke()
cr:set_source(root._wallpaper_pattern)
cr:rectangle(0, 0, root.size())
cr:fill()
cr:paint()
end
cr:set_line_width(beautiful.border_width/2)
cr:set_source(color(beautiful.border_color))
local rect = {x1 = 0 ,y1 = 0 , x2 = 0 , y2 = 0}
-- Get the region with wiboxes
for _, obj in ipairs {drawin, client} do
for _, d in ipairs(obj.get()) do
local w = d.get_wibox and d:get_wibox() or d
if w and w.geometry and w.visible then
local geo = w:geometry()
rect.x1 = math.min(rect.x1, geo.x )
rect.y1 = math.min(rect.y1, geo.y )
rect.x2 = math.max(rect.x2, geo.x + geo.width + 2*(w.border_width or 0))
rect.y2 = math.max(rect.y2, geo.y + geo.height + 2*(w.border_width or 0))
end
end
end
local total_area = wibox.layout {
forced_width = rect.x2,
forced_height = rect.y2,
layout = wibox.layout.manual,
}
-- Fix the wibox geometries that have a dependency on their content
for _, d in ipairs(drawin.get()) do
local w = d.get_wibox and d:get_wibox() or nil
if w then
-- Force a full layout first as widgets with as the awful.popup have
-- interdependencies between the content and the container
if w._apply_size_now then
w:_apply_size_now()
end
end
end
-- Emulate the event loop for another 5 iterations
for _ = 1, 5 do
require("gears.timer").run_delayed_calls_now()
end
for _, d in ipairs(drawin.get()) do
local w = d.get_wibox and d:get_wibox() or nil
if w and w.visible then
local geo = w:geometry()
total_area:add_at(w:to_widget(), {x = geo.x, y = geo.y})
end
end
-- Loop each clients geometry history and paint it
for _, c in ipairs(client.get()) do
local is_displayed = false
for _, t in pairs(c:tags()) do
is_displayed = is_displayed or t.selected
end
if (not c.minimized) and is_displayed then
local pgeo = nil
for _, geo in ipairs(c._old_geo) do
if not geo._hide then
total_area:add_at(
client_widget(c, c.color or geo._color or beautiful.bg_normal, geo._label, args),
{x=geo.x, y=geo.y}
)
end
-- Draw lines between the old and new corners
if pgeo and not args.hide_lines then
cr:save()
cr:set_source_rgba(0,0,0,.1)
-- Top left
cr:move_to(pgeo.x, pgeo.y)
cr:line_to(geo.x, geo.y)
cr:stroke()
-- Top right
cr:move_to(pgeo.x+pgeo.width, pgeo.y)
cr:line_to(geo.x+pgeo.width, geo.y)
-- Bottom left
cr:move_to(pgeo.x, pgeo.y+pgeo.height)
cr:line_to(geo.x, geo.y+geo.height)
cr:stroke()
-- Bottom right
cr:move_to(pgeo.x+pgeo.width, pgeo.y+pgeo.height)
cr:line_to(geo.x+pgeo.width, geo.y+geo.height)
cr:stroke()
cr:restore()
end
pgeo = geo
end
end
end
-- Draw the wiboxes/clients on top of the screen
wibox.widget.draw_to_cairo_context(
total_area, cr, screen._get_extents()
)
cr:set_source_rgba(1,0,0,1)
cr:set_dash({1,1},1)
-- Paint the mouse cursor position history
for _, h in ipairs(mouse.old_histories) do
local pos = nil
for _, coords in ipairs(h) do
draw_mouse(coords.x, coords.y)
cr:fill()
if pos then
cr:move_to(pos.x, pos.y)
cr:line_to(coords.x, coords.y)
cr:stroke()
end
pos = coords
end
end
img:finish()
-- vim: filetype=lua:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:textwidth=80