widgets/cpu.lua

   1
   2 --[[
   3
   4      Licensed under GNU General Public License v2
   5       * (c) 2013,      Luke Bonham
   6       * (c) 2010-2012, Peter Hofmann
   7
   8 --]]
   9
  10 local lines_match  = require("lain.helpers").lines_match
  11 local newtimer     = require("lain.helpers").newtimer
  12
  13 local wibox        = require("wibox")
  14
  15 local math         = { ceil   = math.ceil }
  16 local string       = { format = string.format,
  17                        gmatch = string.gmatch }
  18 local tostring     = tostring
  19
  20 local setmetatable = setmetatable
  21
  22 -- CPU usage
  23 -- lain.widgets.cpu
  24 local cpu = { core = {} }
  25
  26 local function worker(args)
  27     local args     = args or {}
  28           local timeout  = args.timeout or 2
  29           local settings = args.settings or function() end
  30
  31           cpu.widget = wibox.widget.textbox('')
  32
  33           function update()
  34         -- Read the amount of time the CPUs have spent performing
  35                     -- different kinds of work. Read the first line of /proc/stat
  36                     -- which is the sum of all CPUs.
  37         local times = lines_match("cpu","/proc/stat")
  38
  39         for index,time in pairs(times)
  40         do
  41             local coreid = index - 1
  42             local core = cpu.core[coreid] or
  43                   { last_active = 0 , last_total = 0, usage = 0 }
  44             local at = 1
  45             local idle = 0
  46             local total = 0
  47
  48             for field in string.gmatch(time, "[%s]+([^%s]+)")
  49             do
  50                 -- 4 = idle, 5 = ioWait. Essentially, the CPUs have done
  51                 -- nothing during these times.
  52                 if at == 4 or at == 5 then
  53                     idle = idle + field
  54                 end
  55                 total = total + field
  56                 at = at + 1
  57             end
  58
  59             local active = total - idle
  60
  61             if core.last_active ~= active or core.last_total ~= total then
  62                 -- Read current data and calculate relative values.
  63                 local dactive = active - core.last_active
  64                 local dtotal  = total - core.last_total
  65
  66                 local usage = math.ceil((dactive / dtotal) * 100)
  67
  68                 core.last_active = active
  69                 core.last_total  = total
  70                 core.usage       = usage
  71
  72                 -- Save current data for the next run.
  73                 cpu.core[coreid] = core;
  74             end
  75         end
  76
  77         widget = cpu.widget
  78         cpu_now = cpu.core
  79         cpu_now.usage = cpu_now[0].usage
  80
  81         settings()
  82     end
  83
  84     newtimer("cpu", timeout, update)
  85     return cpu.widget
  86 end
  87
  88 return setmetatable(cpu, { __call = function(_, ...) return worker(...) end })