/*

   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.Collections.Generic;

 using System.Globalization;

 using System.Text;

 namespace OpenHardwareMonitor.Hardware.Nvidia {

   internal class NvidiaGroup : IGroup {

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

     private readonly StringBuilder report = new StringBuilder();

     public NvidiaGroup(ISettings settings) {

       if (!NVAPI.IsAvailable)

         return;

       report.AppendLine("NVAPI");

       report.AppendLine();

       string version;

       if (NVAPI.NvAPI_GetInterfaceVersionString(out version) == NvStatus.OK) {

         report.Append("Version: ");

         report.AppendLine(version);

       }

       NvPhysicalGpuHandle[] handles = 

         new NvPhysicalGpuHandle[NVAPI.MAX_PHYSICAL_GPUS];

       int count;

       if (NVAPI.NvAPI_EnumPhysicalGPUs == null) {

         report.AppendLine("Error: NvAPI_EnumPhysicalGPUs not available");

         report.AppendLine();

         return;

       } else {        

         NvStatus status = NVAPI.NvAPI_EnumPhysicalGPUs(handles, out count);

         if (status != NvStatus.OK) {

           report.AppendLine("Status: " + status);

           report.AppendLine();

           return;

         }

       }

       IDictionary<NvPhysicalGpuHandle, NvDisplayHandle> displayHandles =

         new Dictionary<NvPhysicalGpuHandle, NvDisplayHandle>();

       if (NVAPI.NvAPI_EnumNvidiaDisplayHandle != null &&

         NVAPI.NvAPI_GetPhysicalGPUsFromDisplay != null) 

       {

         NvStatus status = NvStatus.OK;

         int i = 0;

         while (status == NvStatus.OK) {

           NvDisplayHandle displayHandle = new NvDisplayHandle();

           status = NVAPI.NvAPI_EnumNvidiaDisplayHandle(i, ref displayHandle);

           i++;

           if (status == NvStatus.OK) {

             NvPhysicalGpuHandle[] handlesFromDisplay =

               new NvPhysicalGpuHandle[NVAPI.MAX_PHYSICAL_GPUS];

             uint countFromDisplay;

             if (NVAPI.NvAPI_GetPhysicalGPUsFromDisplay(displayHandle,

               handlesFromDisplay, out countFromDisplay) == NvStatus.OK) {

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

                 if (!displayHandles.ContainsKey(handlesFromDisplay[j]))

                   displayHandles.Add(handlesFromDisplay[j], displayHandle);

               }

             }

           }

         }

       }

       report.Append("Number of GPUs: ");

       report.AppendLine(count.ToString(CultureInfo.InvariantCulture));

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

         NvDisplayHandle displayHandle;

         displayHandles.TryGetValue(handles[i], out displayHandle);

         hardware.Add(new NvidiaGPU(i, handles[i], displayHandle, settings));

       }

       report.AppendLine();

     }

     public IHardware[] Hardware {

       get {

         return hardware.ToArray();

       }

     }

     public string GetReport() {

       return report.ToString();

     }

     public void Close() {

       foreach (Hardware gpu in hardware)

         gpu.Close();      

     }

   }

 }