moel@1: /*
moel@1:   
moel@1:   Version: MPL 1.1/GPL 2.0/LGPL 2.1
moel@1: 
moel@1:   The contents of this file are subject to the Mozilla Public License Version
moel@1:   1.1 (the "License"); you may not use this file except in compliance with
moel@1:   the License. You may obtain a copy of the License at
moel@1:  
moel@1:   http://www.mozilla.org/MPL/
moel@1: 
moel@1:   Software distributed under the License is distributed on an "AS IS" basis,
moel@1:   WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
moel@1:   for the specific language governing rights and limitations under the License.
moel@1: 
moel@1:   The Original Code is the Open Hardware Monitor code.
moel@1: 
moel@1:   The Initial Developer of the Original Code is 
moel@1:   Michael Möller <m.moeller@gmx.ch>.
moel@1:   Portions created by the Initial Developer are Copyright (C) 2009-2010
moel@1:   the Initial Developer. All Rights Reserved.
moel@1: 
moel@1:   Contributor(s):
moel@1: 
moel@1:   Alternatively, the contents of this file may be used under the terms of
moel@1:   either the GNU General Public License Version 2 or later (the "GPL"), or
moel@1:   the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
moel@1:   in which case the provisions of the GPL or the LGPL are applicable instead
moel@1:   of those above. If you wish to allow use of your version of this file only
moel@1:   under the terms of either the GPL or the LGPL, and not to allow others to
moel@1:   use your version of this file under the terms of the MPL, indicate your
moel@1:   decision by deleting the provisions above and replace them with the notice
moel@1:   and other provisions required by the GPL or the LGPL. If you do not delete
moel@1:   the provisions above, a recipient may use your version of this file under
moel@1:   the terms of any one of the MPL, the GPL or the LGPL.
moel@1:  
moel@1: */
moel@1: 
moel@1: using System;
moel@1: using System.Collections.Generic;
moel@90: using System.Diagnostics;
moel@1: using System.Text;
moel@1: 
moel@1: namespace OpenHardwareMonitor.Hardware.CPU {
moel@1: 
moel@1:   public class CPUGroup : IGroup { 
moel@1:     private List<IHardware> hardware = new List<IHardware>();
moel@1: 
moel@90:     private CPUID[][][] threads;
moel@1: 
moel@90:     private CPUID[][] GetProcessorThreads() {
moel@1: 
moel@90:       List<CPUID> threads = new List<CPUID>();
moel@90:       for (int i = 0; i < 32; i++) {
moel@90:         try {
moel@90:           threads.Add(new CPUID(i));
moel@90:         } catch (ArgumentException) { }
moel@90:       }
moel@1: 
moel@90:       SortedDictionary<uint, List<CPUID>> processors =
moel@90:         new SortedDictionary<uint, List<CPUID>>();
moel@90:       foreach (CPUID thread in threads) {
moel@90:         List<CPUID> list;
moel@90:         processors.TryGetValue(thread.ProcessorId, out list);
moel@90:         if (list == null) {
moel@90:           list = new List<CPUID>();
moel@90:           processors.Add(thread.ProcessorId, list);
moel@90:         }
moel@90:         list.Add(thread);
moel@90:       }
moel@90: 
moel@90:       CPUID[][] processorThreads = new CPUID[processors.Count][];
moel@90:       int index = 0;
moel@90:       foreach (List<CPUID> list in processors.Values) {
moel@90:         processorThreads[index] = list.ToArray();
moel@90:         index++;
moel@90:       }
moel@90:       return processorThreads;
moel@90:     }
moel@90: 
moel@90:     private CPUID[][] GroupThreadsByCore(CPUID[] threads) {
moel@90: 
moel@90:       SortedDictionary<uint, List<CPUID>> cores = 
moel@90:         new SortedDictionary<uint, List<CPUID>>();
moel@90:       foreach (CPUID thread in threads) {
moel@90:         List<CPUID> coreList;
moel@90:         cores.TryGetValue(thread.CoreId, out coreList);
moel@90:         if (coreList == null) {
moel@90:           coreList = new List<CPUID>();
moel@90:           cores.Add(thread.CoreId, coreList);
moel@90:         }
moel@90:         coreList.Add(thread);
moel@90:       }
moel@90: 
moel@90:       CPUID[][] coreThreads = new CPUID[cores.Count][];
moel@90:       int index = 0;
moel@90:       foreach (List<CPUID> list in cores.Values) {
moel@90:         coreThreads[index] = list.ToArray();
moel@90:         index++;
moel@90:       }
moel@90:       return coreThreads;
moel@1:     }
moel@1: 
moel@1:     public CPUGroup() {
moel@90:       if (!WinRing0.IsCpuid())
moel@1:         return;
moel@1: 
moel@90:       CPUID[][] processorThreads = GetProcessorThreads();
moel@90:       this.threads = new CPUID[processorThreads.Length][][];
moel@1: 
moel@90:       int index = 0;
moel@90:       foreach (CPUID[] threads in processorThreads) {
moel@90:         if (threads.Length == 0)
moel@90:           continue;
moel@90:             
moel@90:         CPUID[][] coreThreads = GroupThreadsByCore(threads);
moel@1: 
moel@101:         this.threads[index] = coreThreads;        
moel@1: 
moel@90:         switch (threads[0].Vendor) {
moel@90:           case Vendor.Intel:
moel@100:             hardware.Add(new IntelCPU(index, coreThreads));
moel@1:             break;
moel@90:           case Vendor.AMD:
moel@90:             switch (threads[0].Family) {
moel@90:               case 0x0F:
moel@100:                 hardware.Add(new AMD0FCPU(index, coreThreads));
moel@90:                 break;
moel@90:               case 0x10:
moel@100:                 hardware.Add(new AMD10CPU(index, coreThreads));
moel@90:                 break;
moel@90:               default:
moel@90:                 break;
moel@90:             } break;
moel@1:           default:
moel@1:             break;
moel@101:         }
moel@101: 
moel@101:         index++;
moel@1:       }
moel@1:     }
moel@90:     
moel@1:     public IHardware[] Hardware {
moel@1:       get {
moel@1:         return hardware.ToArray();
moel@1:       }
moel@1:     }
moel@1: 
moel@1:     private void AppendCpuidData(StringBuilder r, uint[,] data, uint offset) {
moel@1:       for (int i = 0; i < data.GetLength(0); i++) {
moel@1:         r.Append(" ");
moel@1:         r.Append((i + offset).ToString("X8"));
moel@1:         for (int j = 0; j < 4; j++) {
moel@1:           r.Append("  ");
moel@1:           r.Append(data[i, j].ToString("X8"));
moel@1:         }
moel@1:         r.AppendLine();
moel@1:       }
moel@1:     }
moel@1: 
moel@1:     public string GetReport() {
moel@1: 
moel@1:       StringBuilder r = new StringBuilder();
moel@90:       
moel@1:       r.AppendLine("CPUID");
moel@1:       r.AppendLine();
moel@1: 
moel@90:       for (int i = 0; i < threads.Length; i++) {
moel@1: 
moel@90:         r.AppendLine("Processor " + i);
moel@18:         r.AppendLine();
moel@90:         r.AppendFormat("Processor Vendor: {0}{1}", threads[i][0][0].Vendor,
moel@90:           Environment.NewLine);
moel@90:         r.AppendFormat("Processor Brand: {0}{1}", threads[i][0][0].BrandString,
moel@90:           Environment.NewLine);
moel@90:         r.AppendFormat("Family: 0x{0}{1}", 
moel@90:           threads[i][0][0].Family.ToString("X"), Environment.NewLine);
moel@90:         r.AppendFormat("Model: 0x{0}{1}", 
moel@90:           threads[i][0][0].Model.ToString("X"), Environment.NewLine);
moel@90:         r.AppendFormat("Stepping: 0x{0}{1}", 
moel@90:           threads[i][0][0].Stepping.ToString("X"), Environment.NewLine);
moel@90:         r.AppendLine();
moel@90: 
moel@90:         r.AppendLine("CPUID Return Values");
moel@90:         r.AppendLine();
moel@90:         for (int j = 0; j < threads[i].Length; j++)
moel@90:           for (int k = 0; k < threads[i][j].Length; k++) {
moel@90:             r.AppendLine(" CPU Thread: " + threads[i][j][k].Thread);
moel@90:             r.AppendLine(" APIC ID: " + threads[i][j][k].ApicId);
moel@90:             r.AppendLine(" Processor ID: " + threads[i][j][k].ProcessorId);
moel@90:             r.AppendLine(" Core ID: " + threads[i][j][k].CoreId);
moel@90:             r.AppendLine(" Thread ID: " + threads[i][j][k].ThreadId);
moel@90:             r.AppendLine();
moel@90:             r.AppendLine(" Function  EAX       EBX       ECX       EDX");
moel@90:             AppendCpuidData(r, threads[i][j][k].Data, CPUID.CPUID_0);
moel@90:             AppendCpuidData(r, threads[i][j][k].ExtData, CPUID.CPUID_EXT);
moel@90:             r.AppendLine();
moel@90:           }
moel@18:       }
moel@90:       return r.ToString(); 
moel@1:     }
moel@1: 
moel@1:     public void Close() { }
moel@1:   }
moel@1: }