add cpu

Signed-off-by: Mattias Andrée <maandree@operamail.com>

1 files changed, 126 insertions, 0 deletions

diff --git a/src/plugins/cpu.py b/src/plugins/cpu.py
new file mode 100644
index 0000000..be19e30
--- /dev/null
+++ b/src/plugins/cpu.py
@@ -0,0 +1,126 @@
+# -*- python -*-
+'''
+xpybar – xmobar replacement written in python
+Copyright © 2014  Mattias Andrée (maandree@member.fsf.org)
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+'''
+
+
+class CPU:
+    '''
+    Retrieve processor statistics
+    
+    @variable  cpu_total:list<int>   Accumulative processors statistics
+    @variable  cpus:list<list<int>>  Individual processors statistics
+    @variable  intr:list<int>        Counts of individual interrupts serviced since boot time
+    @variable  intr_total:int        Sum of `intr`
+    @variable  ctxt:int              The number of context switches that the system underwent.
+    @variable  btime:int             Boot time, in seconds since the Epoch, 1970-01-01 00:00:00 UTC
+    @variable  processes:int         Number of forks and clones (and similar) created since boot
+    @variable  procs_running:int?    Number of processes in runnable state (linux>=2.5.45)
+    @variable  procs_blocked:int?    Number of processes blocked waiting for I/O to complete (linux>=2.5.45)
+    @variable  softirq:list<int>?    Counts of individual software IRQ since boot time
+    @variable  softirq_total:int?    Sum of `softirq`
+    @variable  fields:dict<str, list<int>>  Table with all information
+    '''
+    
+    
+    # These contants are index in the CPU statistcs, use for example `CPU_instance.cpu[UPC.user]`
+    
+    user = 0
+    '''
+    :int  Time spent in user mode, in USER_HZ (normally 1/100ths of a second, it is a time not a frequency)
+    '''
+    
+    nice = 1
+    '''
+    :int  Time spent in user mode with low priority, in USER_HZ
+    '''
+    
+    system = 2
+    '''
+    :int  Time spent in system mode, in USER_HZ
+    '''
+    
+    idle = 3
+    '''
+    :int  Time spent in the idle task, in USER_HZ
+    '''
+    
+    iowait = 4
+    '''
+    :int  Time waiting for I/O to complete, in USER_HZ (linux>=2.5.41)
+    '''
+    
+    irq = 5
+    '''
+    :int  Time servicing interrupts, in USER_HZ (linux>=2.6.0-test4)
+    '''
+    
+    softirq = 6
+    '''
+    :int  Time servicing softirqs, in USER_HZ (linux>=2.6.0-test4)
+    '''
+    
+    steal = 7
+    '''
+    :int  Stolen time, which is the time spent in other operating systems
+          when running in a virtualized environment, in USER_HZ (linux>=2.6.11)
+    '''
+    
+    guest = 8
+    '''
+    :int  Time spent running a virtual CPU for guest operating systems
+          under the control of the Linux kernel, in USER_HZ (linux>=2.6.24)
+    '''
+    
+    guest_nice = 9
+    '''
+    :int  Time spent running a niced guest, in USER_HZ (linux>=2.6.33).
+    '''
+    
+    
+    def __init__(self):
+        '''
+        Constructor
+        '''
+        stat = None
+        with open('/proc/stat', 'rb') as file:
+            stat = file.read()
+        stat = stat.decode('utf-8', 'replace')
+        stat = filter(lambda x : not x == '', stat.split('\n'))
+        
+        fields = {}
+        for line in stat:
+            line = list(filter(lambda x : not x == '', line.split(' ')))
+            fields[line[0]] = [int(x) for x in line[1:]]
+        
+        self.cpu = fields['cpu']
+        i = 0
+        self.cpus = []
+        while ('cpu%i' % i) in fields:
+            self.cpus.append(fields['cpu%i' % i])
+            i += 1
+        self.ctxt = fields['ctxt'][0]
+        self.btime = fields['btime'][0]
+        self.processes = fields['processes'][0]
+        self.procs_running = fields['procs_running'][0] if 'procs_running' in fields else None
+        self.procs_blocked = fields['procs_blocked'][0] if 'procs_blocked' in fields else None
+        self.intr = fields['intr'][1:]
+        self.intr_total = fields['intr'][0]
+        self.softirq = fields['softirq'][1:] if 'softirq' in fields else None
+        self.softirq_total = fields['softirq'][0] if 'softirq' in fields else None
+        self.fields = fields
+