initial

dot_tmux/plugins/tmux-mem-cpu-load/common/load.cc

@@ -0,0 +1,124 @@
+/* vim: tabstop=2 shiftwidth=2 expandtab textwidth=80 linebreak wrap
+ *
+ * Copyright 2012 Matthew McCormick
+ * Copyright 2013 Justin Crawford <Justasic@gmail.com>
+ * Copyright 2015 Pawel 'l0ner' Soltys
+ * Copyright 2016 Compilenix
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Based on: github.com/freebsd/freebsd/blob/master/usr.bin/top/machine.c
+// Based on: Apple.cpp for load_string/mem_string and apple's documentation
+
+#include <sstream>
+#include <string>
+#include <stdlib.h> // getloadavg()
+#include <cmath> // floorf()
+#include <sys/types.h>
+#include <stdio.h>
+
+#include "cpu.h"
+#include "load.h"
+#include "luts.h"
+
+#include "powerline.h"
+
+// Load Averages
+std::string load_string( bool use_colors,
+  bool use_powerline_left, bool use_powerline_right, short num_averages,
+  bool segments_to_right, short right_color )
+{
+  std::ostringstream ss;
+  ss.setf( std::ios::fixed, std::ios::floatfield );
+  ss.precision( 2 );
+  double averages[num_averages];
+  // based on: opensource.apple.com/source/Libc/Libc-262/gen/getloadavg.c
+
+  if( num_averages <= 0 || num_averages > 3)
+  {
+    ss << (char) 0;
+    return ss.str();
+  }
+
+  if( getloadavg( averages, num_averages ) < 0 )
+  {
+    ss << " 0.00 0.00 0.00"; // couldn't get averages.
+  }
+  else
+  {
+    unsigned load_percent = static_cast<unsigned int>( averages[0] /
+        get_cpu_count() * 0.5f * 100.0f );
+
+    if( load_percent > 100 )
+    {
+      load_percent = 100;
+    }
+    if( use_colors )
+    {
+      if( use_powerline_right )
+      {
+        powerline( ss, load_lut[load_percent], POWERLINE_RIGHT );
+      }
+      else if( use_powerline_left )
+      {
+        powerline( ss, load_lut[load_percent], POWERLINE_LEFT );
+      }
+      else
+      {
+        powerline( ss, load_lut[load_percent], NONE );
+      }
+    }
+
+    ss << ' ';
+    for( int i = 0; i < num_averages; ++i )
+    {
+      // Round to nearest, make sure this is only a 0.00 value not a 0.0000
+      float avg = floorf( static_cast<float>( averages[i] ) * 100 + 0.5 ) / 100;
+      // Don't print trailing whitespace for last element
+      if ( i == num_averages-1 )
+      {
+        ss << avg;
+      }
+      else
+      {
+        ss << avg << " ";
+      }
+    }
+
+    if( use_colors )
+    {
+      if( use_powerline_left && segments_to_right )
+      {
+        powerline( ss, load_lut[load_percent], POWERLINE_LEFT, true );
+        powerline_char( ss, load_lut[load_percent], right_color, POWERLINE_LEFT, true );
+      }
+      else if( use_powerline_left && !segments_to_right )
+      {
+        powerline( ss, load_lut[load_percent], POWERLINE_LEFT, true );
+        powerline( ss, "#[fg=default,bg=default]", POWERLINE_LEFT );
+      }
+      else if( !use_powerline_right )
+      {
+        ss << "#[fg=default,bg=default]";
+      }
+      else if ( segments_to_right && use_powerline_right )
+      {
+        powerline_char( ss, load_lut[load_percent], right_color, POWERLINE_RIGHT, true );
+      }
+    }
+  }
+
+  return ss.str();
+}
+