blob: f24b9ee4745802d965cdfbbbdf51bc9a6dc37e55 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
-----------------------------------------------------------------------------
-- |
-- Module : Monitors.Cpu
-- Copyright : (c) Andrea Rossato
-- License : BSD-style (see LICENSE)
--
-- Maintainer : Andrea Rossato <andrea.rossato@unibz.it>
-- Stability : unstable
-- Portability : unportable
--
-- A cpu monitor for XMobar
--
-----------------------------------------------------------------------------
module Main where
import Numeric
import Control.Concurrent
import Control.Concurrent.MVar
import qualified Data.ByteString.Lazy.Char8 as B
data Config =
Config { intervall :: Int
, cpuNormal :: Integer
, cpuNormalColor :: String
, cpuCritical :: Integer
, cpuCriticalColor :: String
}
defaultConfig :: Config
defaultConfig =
Config { intervall = 500000
, cpuNormal = 2
, cpuNormalColor = "#00FF00"
, cpuCritical = 60
, cpuCriticalColor = "#FF0000"
}
config :: Config
config = defaultConfig
-- Utilities
floatToPercent :: Float -> String
floatToPercent n =
showFFloat (Just 2) (n*100) "%"
fileCPU :: IO B.ByteString
fileCPU = B.readFile "/proc/stat"
getData :: MVar [Float] -> Int -> IO ()
getData var d =
do threadDelay d
s <- fileCPU
modifyMVar_ var (\_ -> return $! cpuParser s)
cpuParser :: B.ByteString -> [Float]
cpuParser =
map read . map B.unpack . tail . B.words . flip (!!) 0 . B.lines
parseCPU :: IO [Float]
parseCPU =
do v1 <- newMVar []
forkIO $! getData v1 0
v2 <- newMVar []
forkIO $! getData v2 500000
threadDelay 750000
a <- readMVar v1
b <- readMVar v2
let dif = zipWith (-) b a
tot = foldr (+) 0 dif
percent = map (/ tot) dif
return percent
formatCpu :: [Float] -> String
formatCpu [] = ""
formatCpu (us:ni:sy:_)
| x >= c = setColor z cpuCriticalColor
| x >= n = setColor z cpuNormalColor
| otherwise = floatToPercent y
where x = (us * 100) + (sy * 100) + (ni * 100)
y = us + sy + ni
z = floatToPercent y
c = fromInteger (cpuCritical config)
n = fromInteger (cpuNormal config)
formatCpu _ = ""
setColor :: String -> (Config -> String) -> String
setColor str ty =
"<fc=" ++ ty config ++ ">" ++
str ++ "</fc>"
cpu :: IO String
cpu =
do l <- parseCPU
return $ "Cpu: " ++ formatCpu l
main :: IO ()
main =
do c <- cpu
putStrLn c
|