Hardware/CPU/CPULoad.cs
author moel.mich
Sat, 15 May 2010 10:38:47 +0000
changeset 112 69d29f1a2307
parent 90 3333b29a1746
child 165 813d8bc3192f
permissions -rw-r--r--
Fixed Issue 59.
     1 /*
     2   
     3   Version: MPL 1.1/GPL 2.0/LGPL 2.1
     4 
     5   The contents of this file are subject to the Mozilla Public License Version
     6   1.1 (the "License"); you may not use this file except in compliance with
     7   the License. You may obtain a copy of the License at
     8  
     9   http://www.mozilla.org/MPL/
    10 
    11   Software distributed under the License is distributed on an "AS IS" basis,
    12   WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
    13   for the specific language governing rights and limitations under the License.
    14 
    15   The Original Code is the Open Hardware Monitor code.
    16 
    17   The Initial Developer of the Original Code is 
    18   Michael Möller <m.moeller@gmx.ch>.
    19   Portions created by the Initial Developer are Copyright (C) 2009-2010
    20   the Initial Developer. All Rights Reserved.
    21 
    22   Contributor(s):
    23 
    24   Alternatively, the contents of this file may be used under the terms of
    25   either the GNU General Public License Version 2 or later (the "GPL"), or
    26   the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
    27   in which case the provisions of the GPL or the LGPL are applicable instead
    28   of those above. If you wish to allow use of your version of this file only
    29   under the terms of either the GPL or the LGPL, and not to allow others to
    30   use your version of this file under the terms of the MPL, indicate your
    31   decision by deleting the provisions above and replace them with the notice
    32   and other provisions required by the GPL or the LGPL. If you do not delete
    33   the provisions above, a recipient may use your version of this file under
    34   the terms of any one of the MPL, the GPL or the LGPL.
    35  
    36 */
    37 
    38 using System;
    39 using System.Collections.Generic;
    40 using System.Runtime.InteropServices;
    41 using System.Text;
    42 
    43 namespace OpenHardwareMonitor.Hardware.CPU {
    44   public class CPULoad {
    45 
    46     [StructLayout(LayoutKind.Sequential)]
    47     private struct SystemProcessorPerformanceInformation {
    48       public long IdleTime;
    49       public long KernelTime;
    50       public long UserTime;
    51       public long Reserved0;
    52       public long Reserved1;
    53       public ulong Reserved2;
    54     }
    55 
    56     private enum SystemInformationClass : int {
    57       SystemBasicInformation = 0,
    58       SystemCpuInformation = 1,
    59       SystemPerformanceInformation = 2,
    60       SystemTimeOfDayInformation = 3,
    61       SystemProcessInformation = 5,
    62       SystemProcessorPerformanceInformation = 8
    63     }
    64 
    65     [DllImport("ntdll.dll")]
    66     private static extern int NtQuerySystemInformation(
    67       SystemInformationClass informationClass,
    68       [Out] SystemProcessorPerformanceInformation[] informations, 
    69       int structSize, out IntPtr returnLength);
    70 
    71     private CPUID[][] cpuid;
    72 
    73     private long systemTime;
    74     private long[] idleTimes;
    75 
    76     private float totalLoad;
    77     private float[] coreLoads;
    78 
    79     private bool available = false;
    80 
    81     private long[] GetIdleTimes() {      
    82       SystemProcessorPerformanceInformation[] informations = new
    83         SystemProcessorPerformanceInformation[64];
    84 
    85       int size = Marshal.SizeOf(typeof(SystemProcessorPerformanceInformation));
    86 
    87       IntPtr returnLength;
    88       NtQuerySystemInformation(
    89         SystemInformationClass.SystemProcessorPerformanceInformation,
    90         informations, informations.Length * size, out returnLength);
    91 
    92       long[] result = new long[(int)returnLength / size];
    93 
    94       for (int i = 0; i < result.Length; i++)
    95         result[i] = informations[i].IdleTime;
    96 
    97       return result;
    98     }
    99 
   100     public CPULoad(CPUID[][] cpuid) {
   101       this.cpuid = cpuid;
   102       this.coreLoads = new float[cpuid.Length];         
   103       this.systemTime = DateTime.Now.Ticks;
   104       this.totalLoad = 0;
   105       try {
   106         this.idleTimes = GetIdleTimes();
   107       } catch (Exception) {
   108         this.idleTimes = null;
   109       }
   110       if (idleTimes != null)
   111         available = true;
   112     }
   113 
   114     public bool IsAvailable {
   115       get { return available; }
   116     }
   117 
   118     public float GetTotalLoad() {
   119       return totalLoad;
   120     }
   121 
   122     public float GetCoreLoad(int core) {
   123       return coreLoads[core];
   124     }
   125 
   126     public void Update() {
   127       if (this.idleTimes == null)
   128         return;
   129 
   130       long systemTime = DateTime.Now.Ticks;
   131       long[] idleTimes = GetIdleTimes();
   132 
   133       if (systemTime - this.systemTime < 10000)
   134         return;
   135 
   136       float total = 0;
   137       int count = 0;
   138       for (int i = 0; i < cpuid.Length; i++) {
   139         float value = 0;
   140         for (int j = 0; j < cpuid[i].Length; j++) {
   141           long index = cpuid[i][j].Thread;
   142           if (index < idleTimes.Length) {
   143             long delta = idleTimes[index] - this.idleTimes[index];
   144             value += delta;
   145             total += delta;
   146             count++;
   147           }
   148         }
   149         value = 1.0f - value / (cpuid[i].Length * 
   150           (systemTime - this.systemTime));
   151         value = value < 0 ? 0 : value;
   152         coreLoads[i] = value * 100;
   153       }
   154       if (count > 0) {
   155         total = 1.0f - total / (count * (systemTime - this.systemTime));
   156         total = total < 0 ? 0 : total;
   157       } else {
   158         total = 0;
   159       }
   160       this.totalLoad = total * 100;
   161 
   162       this.systemTime = systemTime;
   163       this.idleTimes = idleTimes;
   164     }
   165 
   166   }
   167 }