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@35: using System.Text;
moel@1: 
moel@1: namespace OpenHardwareMonitor.Hardware.Nvidia {
moel@1: 
moel@1:   public class NvidiaGroup : IGroup {
moel@35:    
moel@1:     private List<IHardware> hardware = new List<IHardware>();
moel@35:     private StringBuilder report = new StringBuilder();
moel@1: 
moel@1:     public NvidiaGroup() {
moel@1:       if (!NVAPI.IsAvailable)
moel@1:         return;
moel@1: 
moel@35:       report.AppendLine("NVAPI");
moel@35:       report.AppendLine();
moel@35: 
moel@140:       string version;
moel@140:       if (NVAPI.NvAPI_GetInterfaceVersionString(out version) == NvStatus.OK) {
moel@140:         report.Append("Version: ");
moel@140:         report.AppendLine(version);
moel@140:       }
moel@140: 
moel@1:       NvPhysicalGpuHandle[] handles = 
moel@1:         new NvPhysicalGpuHandle[NVAPI.MAX_PHYSICAL_GPUS];
moel@140:       int count;
moel@115:       if (NVAPI.NvAPI_EnumPhysicalGPUs == null) {
moel@115:         report.AppendLine("Error: NvAPI_EnumPhysicalGPUs not available");
moel@115:         report.AppendLine();
moel@115:         return;
moel@140:       } else {        
moel@140:         NvStatus status = NVAPI.NvAPI_EnumPhysicalGPUs(handles, out count);
moel@140:         if (status != NvStatus.OK) {
moel@140:           report.AppendLine("Status: " + status.ToString());
moel@140:           report.AppendLine();
moel@140:           return;
moel@140:         }
moel@115:       }
moel@115: 
moel@140:       IDictionary<NvPhysicalGpuHandle, NvDisplayHandle> displayHandles =
moel@140:         new Dictionary<NvPhysicalGpuHandle, NvDisplayHandle>();
moel@140: 
moel@140:       if (NVAPI.NvAPI_EnumNvidiaDisplayHandle != null &&
moel@140:         NVAPI.NvAPI_GetPhysicalGPUsFromDisplay != null) 
moel@140:       {
moel@140:         NvStatus status = NvStatus.OK;
moel@140:         int i = 0;
moel@140:         while (status == NvStatus.OK) {
moel@140:           NvDisplayHandle displayHandle = new NvDisplayHandle();
moel@140:           status = NVAPI.NvAPI_EnumNvidiaDisplayHandle(i, ref displayHandle);
moel@140:           i++;
moel@140: 
moel@140:           if (status == NvStatus.OK) {
moel@140:             NvPhysicalGpuHandle[] handlesFromDisplay =
moel@140:               new NvPhysicalGpuHandle[NVAPI.MAX_PHYSICAL_GPUS];
moel@140:             uint countFromDisplay;
moel@140:             if (NVAPI.NvAPI_GetPhysicalGPUsFromDisplay(displayHandle,
moel@140:               handlesFromDisplay, out countFromDisplay) == NvStatus.OK) {
moel@140:               for (int j = 0; j < countFromDisplay; j++) {
moel@140:                 if (!displayHandles.ContainsKey(handlesFromDisplay[j]))
moel@140:                   displayHandles.Add(handlesFromDisplay[j], displayHandle);
moel@140:               }
moel@140:             }
moel@140:           }
moel@140:         }
moel@140:       }
moel@1: 
moel@35:       report.Append("Number of GPUs: ");
moel@140:       report.AppendLine(count.ToString());      
moel@140:       
moel@140:       for (int i = 0; i < count; i++) {    
moel@140:         NvDisplayHandle displayHandle;
moel@140:         if (displayHandles.TryGetValue(handles[i], out displayHandle))
moel@140:           hardware.Add(new NvidiaGPU(i, handles[i], displayHandle));                            
moel@140:         else
moel@140:           hardware.Add(new NvidiaGPU(i, handles[i], null));   
moel@140:       }
moel@140: 
moel@35:       report.AppendLine();
moel@1:     }
moel@1: 
moel@1:     public IHardware[] Hardware {
moel@1:       get {
moel@1:         return hardware.ToArray();
moel@1:       }
moel@1:     }
moel@1: 
moel@1:     public string GetReport() {
moel@35:       return report.ToString();
moel@1:     }
moel@1: 
moel@140:     public void Close() { }
moel@1:   }
moel@1: }