/*

   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-2010 Michael Möller <mmoeller@openhardwaremonitor.org>

   Copyright (C) 2010 Paul Werelds <paul@werelds.net>

 */

 using System;

 using System.Globalization;

 using System.Text;

 using System.Threading;

 namespace OpenHardwareMonitor.Hardware.CPU {

   internal sealed class AMD0FCPU : AMDCPU {

     private readonly Sensor[] coreTemperatures;

     private readonly Sensor[] coreClocks;

     private readonly Sensor busClock;

     private const uint FIDVID_STATUS = 0xC0010042;

     private const byte MISCELLANEOUS_CONTROL_FUNCTION = 3;

     private const ushort MISCELLANEOUS_CONTROL_DEVICE_ID = 0x1103;

     private const uint THERMTRIP_STATUS_REGISTER = 0xE4;

     private readonly byte thermSenseCoreSelCPU0;

     private readonly byte thermSenseCoreSelCPU1;

     private readonly uint miscellaneousControlAddress;

     public AMD0FCPU(int processorIndex, CPUID[][] cpuid, ISettings settings)

       : base(processorIndex, cpuid, settings) 

     {

       float offset = -49.0f;

       // AM2+ 65nm +21 offset

       uint model = cpuid[0][0].Model;

       if (model >= 0x69 && model != 0xc1 && model != 0x6c && model != 0x7c) 

         offset += 21;

       if (model < 40) {

         // AMD Athlon 64 Processors

         thermSenseCoreSelCPU0 = 0x0;

         thermSenseCoreSelCPU1 = 0x4;

       } else {

         // AMD NPT Family 0Fh Revision F, G have the core selection swapped

         thermSenseCoreSelCPU0 = 0x4;

         thermSenseCoreSelCPU1 = 0x0;

       }

       // check if processor supports a digital thermal sensor 

       if (cpuid[0][0].ExtData.GetLength(0) > 7 && 

         (cpuid[0][0].ExtData[7, 3] & 1) != 0) 

       {

         coreTemperatures = new Sensor[coreCount];

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

           coreTemperatures[i] =

             new Sensor("Core #" + (i + 1), i, SensorType.Temperature,

               this, new [] { new ParameterDescription("Offset [°C]", 

                   "Temperature offset of the thermal sensor.\n" + 

                   "Temperature = Value + Offset.", offset)

           }, settings);

         }

       } else {

         coreTemperatures = new Sensor[0];

       }

       miscellaneousControlAddress = GetPciAddress(

         MISCELLANEOUS_CONTROL_FUNCTION, MISCELLANEOUS_CONTROL_DEVICE_ID);

       busClock = new Sensor("Bus Speed", 0, SensorType.Clock, this, settings);

       coreClocks = new Sensor[coreCount];

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

         coreClocks[i] = new Sensor(CoreString(i), i + 1, SensorType.Clock,

           this, settings);

         if (HasTimeStampCounter)

           ActivateSensor(coreClocks[i]);

       }

       Update();                   

     }

     protected override uint[] GetMSRs() {

       return new [] { FIDVID_STATUS };

     }

     public override string GetReport() {

       StringBuilder r = new StringBuilder();

       r.Append(base.GetReport());

       r.Append("Miscellaneous Control Address: 0x");

       r.AppendLine((miscellaneousControlAddress).ToString("X", 

         CultureInfo.InvariantCulture));

       r.AppendLine();

       return r.ToString();

     }

     public override void Update() {

       base.Update();

       if (miscellaneousControlAddress != Ring0.InvalidPciAddress) {

         for (uint i = 0; i < coreTemperatures.Length; i++) {

           if (Ring0.WritePciConfig(

             miscellaneousControlAddress, THERMTRIP_STATUS_REGISTER,

             i > 0 ? thermSenseCoreSelCPU1 : thermSenseCoreSelCPU0)) {

             uint value;

             if (Ring0.ReadPciConfig(

               miscellaneousControlAddress, THERMTRIP_STATUS_REGISTER, 

               out value)) 

             {

               coreTemperatures[i].Value = ((value >> 16) & 0xFF) + 

                 coreTemperatures[i].Parameters[0].Value;

               ActivateSensor(coreTemperatures[i]);

             } else {

               DeactivateSensor(coreTemperatures[i]);

             }

           }

         }

       }

       if (HasTimeStampCounter) {

         double newBusClock = 0;

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

           Thread.Sleep(1);

           uint eax, edx;

           if (Ring0.RdmsrTx(FIDVID_STATUS, out eax, out edx,

             1UL << cpuid[i][0].Thread)) {

             // CurrFID can be found in eax bits 0-5, MaxFID in 16-21

             // 8-13 hold StartFID, we don't use that here.

             double curMP = 0.5 * ((eax & 0x3F) + 8);

             double maxMP = 0.5 * ((eax >> 16 & 0x3F) + 8);

             coreClocks[i].Value = 

               (float)(curMP * TimeStampCounterFrequency / maxMP);

             newBusClock = (float)(TimeStampCounterFrequency / maxMP);

           } else {

             // Fail-safe value - if the code above fails, we'll use this instead

             coreClocks[i].Value = (float)TimeStampCounterFrequency;

           }

         }

         if (newBusClock > 0) {

           this.busClock.Value = (float)newBusClock;

           ActivateSensor(this.busClock);

         }

       }

     }  

   }

 }