summaryrefslogtreecommitdiff
path: root/src/status/cpuusage.c
blob: 1bb34ae87ae068f66391a57b37ccea0d0a11e006 (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
#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>

#include "status/cpuusage.h"
#include "config.h"

#ifndef NUM_CPUS
  #define NUM_CPUS 1
#endif
#ifndef CPU_HISTORY_SIZE
  #define CPU_HISTORY_SIZE 1
#endif
#if CPU_HISTORY_SIZE < 1
  #define CPU_HISTORY_SIZE 1
#endif


unsigned long last_cpu_used = 0;
unsigned long last_cpu_total = 0;
float cpu_history[CPU_HISTORY_SIZE]; // don't care about init values

void status_cpuusage()
{
  double loadavg[3] = { -13.37, -13.37, -13.37 } ;

  char *stline = NULL;
  size_t stlen;
  FILE *stfile;

  unsigned long cpu_user;
  unsigned long cpu_nice;
  unsigned long cpu_sys;
  unsigned long cpu_used;
  unsigned long cpu_idle;
  unsigned long cpu_total;
  int i;


  // Error signaling color
  fputs("^fg(yellow)", stdout);

  stfile = fopen("/proc/stat", "r");
  if (stfile != NULL) {
    for(i = CPU_HISTORY_SIZE - 1; i > 0; i--) {
      cpu_history[i] = cpu_history[i - 1];
    }

    stlen = getline(&stline, &stlen, stfile);
    fclose(stfile);

    if ( 4 == sscanf(stline, "%*s %ld %ld %ld %ld",
                    &cpu_user, &cpu_nice, &cpu_sys, &cpu_idle) ) {
      cpu_used = cpu_user + cpu_nice + cpu_sys;
      cpu_total = cpu_used + cpu_idle;

      // Now do the percentage
      cpu_history[0] = (float)(cpu_used - last_cpu_used) /
                        (float)(cpu_total - last_cpu_total);

      last_cpu_used = cpu_used;
      last_cpu_total = cpu_total;


      if (cpu_history[0] < 0.4) {
        // CPU idling OK
        fputs("^fg(#0077FF)", stdout);
      } else {
        // CPU busy
        fputs("^fg(#FF00FF)", stdout);
      }

      //printf(" CPU: %.0f%% ", cpu_history[0] * 100);
    }

    free(stline);
  }


  if (getloadavg(loadavg, 3) > 0) {
    printf(" %s%.0f,%.0f,%.0f,%.0f ",
            cpu_history[0] < 0.1000 ? " " : "",
            cpu_history[0] * 100,
            loadavg[0] * (100 / 1),
            loadavg[1] * (100 / 1),  // (100 / NUM_CPUS)
            loadavg[2] * (100 / 1));
  }

  //fputs(" CPU usage ", stdout);
}