/*

  Version: MPL 1.1/GPL 2.0/LGPL 2.1

  The contents of this file are subject to the Mozilla Public License Version

  1.1 (the "License"); you may not use this file except in compliance with

  the License. You may obtain a copy of the License at

  http://www.mozilla.org/MPL/

  Software distributed under the License is distributed on an "AS IS" basis,

  WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License

  for the specific language governing rights and limitations under the License.

  The Original Code is the Open Hardware Monitor code.

  The Initial Developer of the Original Code is 

  Michael Möller <m.moeller@gmx.ch>.

  Portions created by the Initial Developer are Copyright (C) 2009-2010

  the Initial Developer. All Rights Reserved.

  Contributor(s):

  Alternatively, the contents of this file may be used under the terms of

  either the GNU General Public License Version 2 or later (the "GPL"), or

  the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),

  in which case the provisions of the GPL or the LGPL are applicable instead

  of those above. If you wish to allow use of your version of this file only

  under the terms of either the GPL or the LGPL, and not to allow others to

  use your version of this file under the terms of the MPL, indicate your

  decision by deleting the provisions above and replace them with the notice

  and other provisions required by the GPL or the LGPL. If you do not delete

  the provisions above, a recipient may use your version of this file under

  the terms of any one of the MPL, the GPL or the LGPL.

*/

using System;

using System.Collections.Generic;

using System.Text;

namespace OpenHardwareMonitor.Hardware.Nvidia {

  internal class NvidiaGPU : Hardware, IHardware {

    private string name;

    private int adapterIndex;

    private NvPhysicalGpuHandle handle;

    private NvDisplayHandle? displayHandle;

    private Sensor[] temperatures;

    private Sensor fan = null;

    private Sensor[] clocks;

    private Sensor[] loads;

    private Sensor control;

    private Sensor memoryLoad;

    public NvidiaGPU(int adapterIndex, NvPhysicalGpuHandle handle, 

      NvDisplayHandle? displayHandle, ISettings settings) 

    {

      string gpuName;

      if (NVAPI.NvAPI_GPU_GetFullName(handle, out gpuName) == NvStatus.OK) {

        this.name = "NVIDIA " + gpuName.Trim();

      } else {

        this.name = "NVIDIA";

      }

      this.adapterIndex = adapterIndex;

      this.handle = handle;

      this.displayHandle = displayHandle;

      NvGPUThermalSettings thermalSettings = GetThermalSettings();

      temperatures = new Sensor[thermalSettings.Count];

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

        NvSensor sensor = thermalSettings.Sensor[i];

        string name;

        switch (sensor.Target) {

          case NvThermalTarget.BOARD: name = "GPU Board"; break;

          case NvThermalTarget.GPU: name = "GPU Core"; break;

          case NvThermalTarget.MEMORY: name = "GPU Memory"; break;

          case NvThermalTarget.POWER_SUPPLY: name = "GPU Power Supply"; break;

          case NvThermalTarget.UNKNOWN: name = "GPU Unknown"; break;

          default: name = "GPU"; break;

        }

        temperatures[i] = new Sensor(name, i, SensorType.Temperature, this,

          new ParameterDescription[0], settings);

        ActivateSensor(temperatures[i]);

      }

      int value;

      if (NVAPI.NvAPI_GPU_GetTachReading != null &&

        NVAPI.NvAPI_GPU_GetTachReading(handle, out value) == NvStatus.OK) {

        if (value > 0) {

          fan = new Sensor("GPU", 0, SensorType.Fan, this, settings);

          ActivateSensor(fan);

        }

      }

      clocks = new Sensor[3];

      clocks[0] = new Sensor("GPU Core", 0, SensorType.Clock, this, settings);

      clocks[1] = new Sensor("GPU Memory", 1, SensorType.Clock, this, settings);

      clocks[2] = new Sensor("GPU Shader", 2, SensorType.Clock, this, settings);

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

        ActivateSensor(clocks[i]);

      loads = new Sensor[3];

      loads[0] = new Sensor("GPU Core", 0, SensorType.Load, this, settings);

      loads[1] = new Sensor("GPU Memory Controller", 1, SensorType.Load, this, settings);

      loads[2] = new Sensor("GPU Video Engine", 2, SensorType.Load, this, settings);

      memoryLoad = new Sensor("GPU Memory", 3, SensorType.Load, this, settings);

      control = new Sensor("GPU Fan", 0, SensorType.Control, this, settings);

    }

    public override string Name {

      get { return name; }

    }

    public override Identifier Identifier {

      get { return new Identifier("nvidiagpu", adapterIndex.ToString()); }

    }

    public override HardwareType HardwareType {

      get { return HardwareType.GPU; }

    }

    private NvGPUThermalSettings GetThermalSettings() {

      NvGPUThermalSettings settings = new NvGPUThermalSettings();

      settings.Version = NVAPI.GPU_THERMAL_SETTINGS_VER;

      settings.Count = NVAPI.MAX_THERMAL_SENSORS_PER_GPU;

      settings.Sensor = new NvSensor[NVAPI.MAX_THERMAL_SENSORS_PER_GPU];

      if (NVAPI.NvAPI_GPU_GetThermalSettings != null &&

        NVAPI.NvAPI_GPU_GetThermalSettings(handle, (int)NvThermalTarget.ALL,

        ref settings) != NvStatus.OK) {

        settings.Count = 0;        

      }

      return settings;

    }

    private uint[] GetClocks() {

      NvClocks clocks = new NvClocks();

      clocks.Version = NVAPI.GPU_CLOCKS_VER;

      clocks.Clock = new uint[NVAPI.MAX_CLOCKS_PER_GPU];

      if (NVAPI.NvAPI_GPU_GetAllClocks != null &&

        NVAPI.NvAPI_GPU_GetAllClocks(handle, ref clocks) == NvStatus.OK) {

        return clocks.Clock;

      }

      return null;

    }

    public override void Update() {

      NvGPUThermalSettings settings = GetThermalSettings();

      foreach (Sensor sensor in temperatures) 

        sensor.Value = settings.Sensor[sensor.Index].CurrentTemp;

      if (fan != null) {

        int value = 0;

        NVAPI.NvAPI_GPU_GetTachReading(handle, out value);

        fan.Value = value;

      }

      uint[] values = GetClocks();

      if (values != null) {

        clocks[0].Value = 0.001f * values[0];

        clocks[1].Value = 0.001f * values[8];

        clocks[2].Value = 0.001f * values[14];

        if (values[30] != 0) {

          clocks[0].Value = 0.0005f * values[30];

          clocks[2].Value = 0.001f * values[30];

        }

      }

      NvPStates states = new NvPStates();

      states.Version = NVAPI.GPU_PSTATES_VER;

      states.PStates = new NvPState[NVAPI.MAX_PSTATES_PER_GPU];

      if (NVAPI.NvAPI_GPU_GetPStates != null && 

        NVAPI.NvAPI_GPU_GetPStates(handle, ref states) == NvStatus.OK) {

        for (int i = 0; i < 3; i++)

          if (states.PStates[i].Present) {

            loads[i].Value = states.PStates[i].Percentage;

            ActivateSensor(loads[i]);

          }

      } else {

        NvUsages usages = new NvUsages();

        usages.Version = NVAPI.GPU_USAGES_VER;

        usages.Usage = new uint[NVAPI.MAX_USAGES_PER_GPU];

        if (NVAPI.NvAPI_GPU_GetUsages != null &&

          NVAPI.NvAPI_GPU_GetUsages(handle, ref usages) == NvStatus.OK) {

          loads[0].Value = usages.Usage[2];

          loads[1].Value = usages.Usage[6];

          loads[2].Value = usages.Usage[10];

          for (int i = 0; i < 3; i++)

            ActivateSensor(loads[i]);

        }

      }

      NvGPUCoolerSettings coolerSettings = new NvGPUCoolerSettings();

      coolerSettings.Version = NVAPI.GPU_COOLER_SETTINGS_VER;

      coolerSettings.Cooler = new NvCooler[NVAPI.MAX_COOLER_PER_GPU];

      if (NVAPI.NvAPI_GPU_GetCoolerSettings != null && 

        NVAPI.NvAPI_GPU_GetCoolerSettings(handle, 0, ref coolerSettings) ==

        NvStatus.OK && coolerSettings.Count > 0) {

        control.Value = coolerSettings.Cooler[0].CurrentLevel;

        ActivateSensor(control);

      }

      NvMemoryInfo memoryInfo = new NvMemoryInfo();

      memoryInfo.Version = NVAPI.GPU_MEMORY_INFO_VER;

      memoryInfo.Values = new uint[NVAPI.MAX_MEMORY_VALUES_PER_GPU];

      if (NVAPI.NvAPI_GPU_GetMemoryInfo != null && displayHandle.HasValue &&

        NVAPI.NvAPI_GPU_GetMemoryInfo(displayHandle.Value, ref memoryInfo) == 

        NvStatus.OK) 

      {

        uint totalMemory = memoryInfo.Values[0];

        uint freeMemory = memoryInfo.Values[4];

        float usedMemory = Math.Max(totalMemory - freeMemory, 0);

        memoryLoad.Value = 100f * usedMemory / totalMemory;

        ActivateSensor(memoryLoad);

      }

    }

    public override string GetReport() {

      StringBuilder r = new StringBuilder();

      r.AppendLine("Nvidia GPU");

      r.AppendLine();

      r.AppendFormat("Name: {0}{1}", name, Environment.NewLine);

      r.AppendFormat("Index: {0}{1}", adapterIndex, Environment.NewLine);

      if (displayHandle.HasValue && NVAPI.NvAPI_GetDisplayDriverVersion != null) 

      {

        NvDisplayDriverVersion driverVersion = new NvDisplayDriverVersion();

        driverVersion.Version = NVAPI.DISPLAY_DRIVER_VERSION_VER;

        if (NVAPI.NvAPI_GetDisplayDriverVersion(displayHandle.Value,

          ref driverVersion) == NvStatus.OK) {

          r.Append("Driver Version: ");

          r.Append(driverVersion.DriverVersion / 100);

          r.Append(".");

          r.Append((driverVersion.DriverVersion % 100).ToString("00"));

          r.AppendLine();

          r.Append("Driver Branch: ");

          r.AppendLine(driverVersion.BuildBranch);

        }

      }

      r.AppendLine();

      if (NVAPI.NvAPI_GPU_GetThermalSettings != null) {

        NvGPUThermalSettings settings = new NvGPUThermalSettings();

        settings.Version = NVAPI.GPU_THERMAL_SETTINGS_VER;

        settings.Count = NVAPI.MAX_THERMAL_SENSORS_PER_GPU;

        settings.Sensor = new NvSensor[NVAPI.MAX_THERMAL_SENSORS_PER_GPU];

        NvStatus status = NVAPI.NvAPI_GPU_GetThermalSettings(handle,

          (int)NvThermalTarget.ALL, ref settings);

        r.AppendLine("Thermal Settings");

        r.AppendLine();

        if (status == NvStatus.OK) {

          for (int i = 0; i < settings.Count; i++) {

            r.AppendFormat(" Sensor[{0}].Controller: {1}{2}", i,

              settings.Sensor[i].Controller, Environment.NewLine);

            r.AppendFormat(" Sensor[{0}].DefaultMinTemp: {1}{2}", i,

              settings.Sensor[i].DefaultMinTemp, Environment.NewLine);

            r.AppendFormat(" Sensor[{0}].DefaultMaxTemp: {1}{2}", i,

              settings.Sensor[i].DefaultMaxTemp, Environment.NewLine);

            r.AppendFormat(" Sensor[{0}].CurrentTemp: {1}{2}", i,

              settings.Sensor[i].CurrentTemp, Environment.NewLine);

            r.AppendFormat(" Sensor[{0}].Target: {1}{2}", i,

              settings.Sensor[i].Target, Environment.NewLine);

          }

        } else {

          r.Append(" Status: ");

          r.AppendLine(status.ToString());

        }

        r.AppendLine();

      }      

      if (NVAPI.NvAPI_GPU_GetAllClocks != null) {

        NvClocks clocks = new NvClocks();

        clocks.Version = NVAPI.GPU_CLOCKS_VER;

        clocks.Clock = new uint[NVAPI.MAX_CLOCKS_PER_GPU];

        NvStatus status = NVAPI.NvAPI_GPU_GetAllClocks(handle, ref clocks);

        r.AppendLine("Clocks");

        r.AppendLine();

        if (status == NvStatus.OK) {

          for (int i = 0; i < clocks.Clock.Length; i++)

            if (clocks.Clock[i] > 0) {

              r.AppendFormat(" Clock[{0}]: {1}{2}", i, clocks.Clock[i],

                Environment.NewLine);

            }

        } else {

          r.Append(" Status: ");

          r.AppendLine(status.ToString());

        }

        r.AppendLine();

      }         

      if (NVAPI.NvAPI_GPU_GetTachReading != null) {

        int tachValue; 

        NvStatus status = NVAPI.NvAPI_GPU_GetTachReading(handle, out tachValue);

        r.AppendLine("Tachometer");

        r.AppendLine();

        if (status == NvStatus.OK) {

          r.AppendFormat(" Value: {0}{1}", tachValue, Environment.NewLine);

        } else {

          r.Append(" Status: ");

          r.AppendLine(status.ToString());

        }

        r.AppendLine();

      }

      if (NVAPI.NvAPI_GPU_GetPStates != null) {

        NvPStates states = new NvPStates();

        states.Version = NVAPI.GPU_PSTATES_VER;

        states.PStates = new NvPState[NVAPI.MAX_PSTATES_PER_GPU];

        NvStatus status = NVAPI.NvAPI_GPU_GetPStates(handle, ref states);

        r.AppendLine("P-States");

        r.AppendLine();

        if (status == NvStatus.OK) {

          for (int i = 0; i < states.PStates.Length; i++)

            if (states.PStates[i].Present)

              r.AppendFormat(" Percentage[{0}]: {1}{2}", i,

                states.PStates[i].Percentage, Environment.NewLine);

        } else {

          r.Append(" Status: ");

          r.AppendLine(status.ToString());

        }

        r.AppendLine();

      }

      if (NVAPI.NvAPI_GPU_GetUsages != null) {

        NvUsages usages = new NvUsages();

        usages.Version = NVAPI.GPU_USAGES_VER;

        usages.Usage = new uint[NVAPI.MAX_USAGES_PER_GPU];

        NvStatus status = NVAPI.NvAPI_GPU_GetUsages(handle, ref usages);

        r.AppendLine("Usages");

        r.AppendLine();

        if (status == NvStatus.OK) {

          for (int i = 0; i < usages.Usage.Length; i++)

            if (usages.Usage[i] > 0)

              r.AppendFormat(" Usage[{0}]: {1}{2}", i,

                usages.Usage[i], Environment.NewLine);

        } else {

          r.Append(" Status: ");

          r.AppendLine(status.ToString());

        }

        r.AppendLine();

      }

      if (NVAPI.NvAPI_GPU_GetCoolerSettings != null) {

        NvGPUCoolerSettings settings = new NvGPUCoolerSettings();

        settings.Version = NVAPI.GPU_COOLER_SETTINGS_VER;

        settings.Cooler = new NvCooler[NVAPI.MAX_COOLER_PER_GPU];

        NvStatus status =

          NVAPI.NvAPI_GPU_GetCoolerSettings(handle, 0, ref settings);

        r.AppendLine("Cooler Settings");

        r.AppendLine();

        if (status == NvStatus.OK) {

          for (int i = 0; i < settings.Count; i++) {

            r.AppendFormat(" Cooler[{0}].Type: {1}{2}", i,

              settings.Cooler[i].Type, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].Controller: {1}{2}", i,

              settings.Cooler[i].Controller, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].DefaultMin: {1}{2}", i,

              settings.Cooler[i].DefaultMin, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].DefaultMax: {1}{2}", i,

              settings.Cooler[i].DefaultMax, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].CurrentMin: {1}{2}", i,

              settings.Cooler[i].CurrentMin, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].CurrentMax: {1}{2}", i,

              settings.Cooler[i].CurrentMax, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].CurrentLevel: {1}{2}", i,

              settings.Cooler[i].CurrentLevel, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].DefaultPolicy: {1}{2}", i,

              settings.Cooler[i].DefaultPolicy, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].CurrentPolicy: {1}{2}", i,

              settings.Cooler[i].CurrentPolicy, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].Target: {1}{2}", i,

              settings.Cooler[i].Target, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].ControlType: {1}{2}", i,

              settings.Cooler[i].ControlType, Environment.NewLine);

            r.AppendFormat(" Cooler[{0}].Active: {1}{2}", i,

              settings.Cooler[i].Active, Environment.NewLine);

          }

        } else {

          r.Append(" Status: ");

          r.AppendLine(status.ToString());

        }

        r.AppendLine();

      }

      if (NVAPI.NvAPI_GPU_GetMemoryInfo != null && displayHandle.HasValue) {

        NvMemoryInfo memoryInfo = new NvMemoryInfo();

        memoryInfo.Version = NVAPI.GPU_MEMORY_INFO_VER;

        memoryInfo.Values = new uint[NVAPI.MAX_MEMORY_VALUES_PER_GPU];

        NvStatus status = NVAPI.NvAPI_GPU_GetMemoryInfo(displayHandle.Value, 

          ref memoryInfo);

        r.AppendLine("Memory Info");

        r.AppendLine();

        if (status == NvStatus.OK) {

          for (int i = 0; i < memoryInfo.Values.Length; i++)

            r.AppendFormat(" Value[{0}]: {1}{2}", i,

                memoryInfo.Values[i], Environment.NewLine);

        } else {

          r.Append(" Status: ");

          r.AppendLine(status.ToString());

        }

        r.AppendLine();

      }

      return r.ToString();

    }

  }

}