/*

   This Source Code Form is subject to the terms of the Mozilla Public

   License, v. 2.0. If a copy of the MPL was not distributed with this

   file, You can obtain one at http://mozilla.org/MPL/2.0/.

   Copyright (C) 2009-2011 Michael Möller <mmoeller@openhardwaremonitor.org>

 */

 using System;

 using System.Collections.Generic;

 using System.Globalization;

 using System.Text;

 namespace OpenHardwareMonitor.Hardware.CPU {

   internal class CPUGroup : IGroup {

     private readonly List<GenericCPU> hardware = new List<GenericCPU>();

     private readonly CPUID[][][] threads;

     private static CPUID[][] GetProcessorThreads() {

       List<CPUID> threads = new List<CPUID>();

       for (int i = 0; i < 32; i++) {

         try {

           threads.Add(new CPUID(i));

         } catch (ArgumentOutOfRangeException) { }

       }

       SortedDictionary<uint, List<CPUID>> processors =

         new SortedDictionary<uint, List<CPUID>>();

       foreach (CPUID thread in threads) {

         List<CPUID> list;

         processors.TryGetValue(thread.ProcessorId, out list);

         if (list == null) {

           list = new List<CPUID>();

           processors.Add(thread.ProcessorId, list);

         }

         list.Add(thread);

       }

       CPUID[][] processorThreads = new CPUID[processors.Count][];

       int index = 0;

       foreach (List<CPUID> list in processors.Values) {

         processorThreads[index] = list.ToArray();

         index++;

       }

       return processorThreads;

     }

     private static CPUID[][] GroupThreadsByCore(IEnumerable<CPUID> threads) {

       SortedDictionary<uint, List<CPUID>> cores = 

         new SortedDictionary<uint, List<CPUID>>();

       foreach (CPUID thread in threads) {

         List<CPUID> coreList;

         cores.TryGetValue(thread.CoreId, out coreList);

         if (coreList == null) {

           coreList = new List<CPUID>();

           cores.Add(thread.CoreId, coreList);

         }

         coreList.Add(thread);

       }

       CPUID[][] coreThreads = new CPUID[cores.Count][];

       int index = 0;

       foreach (List<CPUID> list in cores.Values) {

         coreThreads[index] = list.ToArray();

         index++;

       }

       return coreThreads;

     }

     public CPUGroup(ISettings settings) {

       CPUID[][] processorThreads = GetProcessorThreads();

       this.threads = new CPUID[processorThreads.Length][][];

       int index = 0;

       foreach (CPUID[] threads in processorThreads) {

         if (threads.Length == 0)

           continue;

         CPUID[][] coreThreads = GroupThreadsByCore(threads);

         this.threads[index] = coreThreads;

         switch (threads[0].Vendor) {

           case Vendor.Intel:

             hardware.Add(new IntelCPU(index, coreThreads, settings));

             break;

           case Vendor.AMD:

             switch (threads[0].Family) {

               case 0x0F:

                 hardware.Add(new AMD0FCPU(index, coreThreads, settings));

                 break;

               case 0x10:

               case 0x11:

               case 0x12:

               case 0x14:

               case 0x15:

                 hardware.Add(new AMD10CPU(index, coreThreads, settings));

                 break;

               default:

                 hardware.Add(new GenericCPU(index, coreThreads, settings));

                 break;

             } break;

           default:

             hardware.Add(new GenericCPU(index, coreThreads, settings));

             break;

         }

         index++;

       }

     }

     public IHardware[] Hardware {

       get {

         return hardware.ToArray();

       }

     }

     private static void AppendCpuidData(StringBuilder r, uint[,] data, 

       uint offset) 

     {

       for (int i = 0; i < data.GetLength(0); i++) {

         r.Append(" ");

         r.Append((i + offset).ToString("X8", CultureInfo.InvariantCulture));

         for (int j = 0; j < 4; j++) {

           r.Append("  ");

           r.Append(data[i, j].ToString("X8", CultureInfo.InvariantCulture));

         }

         r.AppendLine();

       }

     }

     public string GetReport() {

       if (threads == null)

         return null;

       StringBuilder r = new StringBuilder();

       r.AppendLine("CPUID");

       r.AppendLine();

       for (int i = 0; i < threads.Length; i++) {

         r.AppendLine("Processor " + i);

         r.AppendLine();

         r.AppendFormat("Processor Vendor: {0}{1}", threads[i][0][0].Vendor,

           Environment.NewLine);

         r.AppendFormat("Processor Brand: {0}{1}", threads[i][0][0].BrandString,

           Environment.NewLine);

         r.AppendFormat("Family: 0x{0}{1}", 

           threads[i][0][0].Family.ToString("X", CultureInfo.InvariantCulture), 

           Environment.NewLine);

         r.AppendFormat("Model: 0x{0}{1}",

           threads[i][0][0].Model.ToString("X", CultureInfo.InvariantCulture), 

           Environment.NewLine);

         r.AppendFormat("Stepping: 0x{0}{1}",

           threads[i][0][0].Stepping.ToString("X", CultureInfo.InvariantCulture), 

           Environment.NewLine);

         r.AppendLine();

         r.AppendLine("CPUID Return Values");

         r.AppendLine();

         for (int j = 0; j < threads[i].Length; j++)

           for (int k = 0; k < threads[i][j].Length; k++) {

             r.AppendLine(" CPU Thread: " + threads[i][j][k].Thread);

             r.AppendLine(" APIC ID: " + threads[i][j][k].ApicId);

             r.AppendLine(" Processor ID: " + threads[i][j][k].ProcessorId);

             r.AppendLine(" Core ID: " + threads[i][j][k].CoreId);

             r.AppendLine(" Thread ID: " + threads[i][j][k].ThreadId);

             r.AppendLine();

             r.AppendLine(" Function  EAX       EBX       ECX       EDX");

             AppendCpuidData(r, threads[i][j][k].Data, CPUID.CPUID_0);

             AppendCpuidData(r, threads[i][j][k].ExtData, CPUID.CPUID_EXT);

             r.AppendLine();

           }

       }

       return r.ToString(); 

     }

     public void Close() {

       foreach (GenericCPU cpu in hardware) {

         cpu.Close();

       }

     }

   }

 }