!7

// All numbers per core

import com.sun.management.OperatingSystemMXBean; 

static boolean verbose = false;
static long interval = 1000; // ms
sbool measureThreads = true;

static Map<Long, Thread> idToThread;
static new WeakHashMap<Thread, Info> infoMap;

static class Info {
  long cpuTime, measuredWhen; // all in milliseconds
  double percentage; // percentage of CPU used
}

static ThreadMXBean mx;
static boolean cpuTimeSupported;

static volatile double totalPercentage; // all threads combined
static long lastMeasured;

// info about HotSpot compilation
static double compilationPercentage;
static long lastCompilationTime, compilationTimeMeasured;

p {
  mx = ManagementFactory.getThreadMXBean();
  cpuTimeSupported = mx.isThreadCpuTimeSupported();
  if (!cpuTimeSupported) fail("No CPU time info available on platform.");
  
  makeAndroid3("CPU Monitor Bot.");
  
  thread "Java CPU Monitor" {
    while (licensed()) {
      pcall {
        monitor();
        if (verbose && measureThreads)
          printMostActiveThread();
      }
      sleep(interval);
    }
  }
}
  
svoid monitor {
  lock programLock();
  
  long ct = hotSpotCompilationTime();
  long ctm = now();
  if (compilationTimeMeasured != 0)
    compilationPercentage = 100*doubleRatio(ct-lastCompilationTime, ctm-compilationTimeMeasured);
  compilationTimeMeasured = ctm;
  lastCompilationTime = ct;

  if (!measureThreads) ret;
  L<Thread> threads = allThreads_fast();
  new WeakHashMap<Thread, Info> newInfoMap;
  double _totalPercentage = 0;
  
  for (Thread t : threads) {
    Info info = infoMap.get(t);
    if (info == null) info = new Info;
    newInfoMap.put(t, info);
    
    long nano = 0;
    pcall { nano = mx.getThreadCpuTime(t.getId()); }
    long measured = now();
    long ms = nano/1000000;
    
    if (info.measuredWhen != 0) {
      long elapsed = measured - info.measuredWhen;
      long used = ms-info.cpuTime;
      info.percentage = max(0, min(100, used*100.0/elapsed));
      _totalPercentage += info.percentage;
    }
    
    info.measuredWhen = measured;
    info.cpuTime = ms;
  }
  
  totalPercentage = _totalPercentage;
  lastMeasured = now();
  infoMap = newInfoMap;
}

static Thread getMostActiveThread() {
  lock programLock();
  Thread mostActive = null;
  double highest = 0.0;
  
  for (Map.Entry<Thread, Info> e : infoMap.entrySet()) {
    Thread t = e.getKey();
    Info info = e.getValue();
    if (mostActive == null || info.percentage > highest) {
      mostActive = t;
      highest = info.percentage;
    }
  }
  
  return mostActive;
}
  
static void printMostActiveThread() {
  lock programLock();
  Thread mostActive = getMostActiveThread();
  if (mostActive == null)
    print("No info");
  else
    print("Total CPU used by this VM: " + formatDouble(totalPercentage, 1) + "%. Most active thread: " + mostActive.getName() + " (" + formatDouble(infoMap.get(mostActive).percentage, 1) + "% CPU)");
}

answer {
  lock programLock();
  if (match("cpu", s) || match("cpu usage", s)) {
    S a = formatDouble(totalPercentage, 2) + "% Java threads";
    pcall {
      OperatingSystemMXBean osBean = ManagementFactory.getPlatformMXBean(OperatingSystemMXBean.class);
      
      double x = osBean.getProcessCpuLoad()*numberOfCores()*100, y = osBean.getSystemCpuLoad()*numberOfCores()*100;
      a += ". " + formatDouble(x, 1) + "% this process, " 
        + formatDouble(max(0, y-x), 1) + "% other processes"; 
    }
    return a;
  }
    
  if (match3("most active thread", s)) {
    Thread t = getMostActiveThread();
    return t == null ? "Sorry, no info." : format3("Thread name: *, cpu: *", t.getName(), formatDouble(totalPercentage, 2) + "%");
  }
  
  if (match3("cpu: get measurement interval", s))
    ret interval + " ms";
    
  if (match3("cpu: are you measuring", s))
    ret "Yes.";
    
  if (match3("cpu: get last measurement time", s))
    ret "" + lastMeasured;
    
  if "cpu: list all threads" {
    new L<S> l;
    for (Map.Entry<Thread, Info> e : infoMap.entrySet()) {
      Thread t = e.getKey();
      Info info = e.getValue();
      l.add(formatDouble(info.percentage, 1) + "% " + format("*", t.getName()));
    }
    ret slackSnippet(fromLines(l));
  }
  
  if "system load"
    ret "System load: " + formatDouble(systemLoad(), 2);
    
  if "cpu: processors" {
    java.lang.management.OperatingSystemMXBean bean = ManagementFactory.getOperatingSystemMXBean();
    ret "I have " + n(bean.getAvailableProcessors(), bean.getArch() +  " processor") + ".";
  }
}

// in percent
static double getInProcessCPU() {
  ret totalPercentage;
}