using System;
using System.IO;
using System.Runtime.Serialization.Formatters.Binary;
using MSI.Afterburner;
using MSI.Afterburner.Exceptions;

namespace CSharpMACMDemo
{
    class CSharpMACMDemo
    {
        static void Main(string[] args)
        {
            bool isDirty = false;

            try
            {
                // connect to MACM shared memory
                ControlMemory macm = new ControlMemory();

                // print out current MACM Header values
                Console.WriteLine("***** MSI AFTERTERBURNER CONTROL HEADER *****");
                Console.WriteLine(macm.Header.ToString().Replace(";", "\n"));
                Console.WriteLine();

                // print out current MACM GPU Entry values
                for(int i = 0; i < macm.Header.GpuEntryCount; i++)
                {
                    Console.WriteLine("***** MSI AFTERTERBURNER GPU " + i + " *****");
                    Console.WriteLine(macm.GpuEntries[i].ToString().Replace(";", "\n"));
                    Console.WriteLine();
                }
                //return;

                // save existing settings for GPU 0 to file
                string filename = System.Environment.GetEnvironmentVariable("TEMP") + @"\gpuEntry0.gpu";
                Stream stream = File.Open(filename, FileMode.Create);
                BinaryFormatter bformatter = new BinaryFormatter();
                bformatter.Serialize(stream, macm.GpuEntries[0]);
                stream.Close();

                // prompt user to change fan speed
                Console.WriteLine("Enter a new Fan speed % for GPU 0");
                Console.WriteLine("Current Fan speed for GPU 0 is: " + macm.GpuEntries[0].FanSpeedCur + "%");
                Console.WriteLine("New value must be between " + macm.GpuEntries[0].FanSpeedMin + " and " + macm.GpuEntries[0].FanSpeedMax);                
                String input = Console.ReadLine();
                UInt32 newValue;
                if (UInt32.TryParse(input, out newValue))
                {
                    try
                    {
                        macm.GpuEntries[0].FanSpeedCur = newValue;
                        isDirty = true;
                    }
                    catch (MACMFanControlNotManual e)
                    {
                        Console.WriteLine("\nYour fan is currently set to auto.  Would you like to change it to manual?  Enter 'Y'");
                        ConsoleKeyInfo cki = Console.ReadKey(true);
                        if (cki.Key == ConsoleKey.Y)
                        {
                            macm.GpuEntries[0].FanFlagsCur = MACM_SHARED_MEMORY_GPU_ENTRY_FAN_FLAG.None;
                            macm.GpuEntries[0].FanSpeedCur = newValue;
                            isDirty = true;
                        }
                        else
                        {
                            Console.WriteLine(e.Message);
                        }
                    }
                }
                else
                {
                    Console.WriteLine("New fan speed entered is not valid: " + input);
                }

                // prompt user to change core clock speed
                Console.WriteLine("\nEnter a new Core Clock speed for GPU 0");
                Console.WriteLine("Current Core Clock speed for GPU 0 is: " + macm.GpuEntries[0].CoreClockCur);
                Console.WriteLine("New value must be between " + macm.GpuEntries[0].CoreClockMin + " and " + macm.GpuEntries[0].CoreClockMax);                
                input = Console.ReadLine();
                if (UInt32.TryParse(input, out newValue))
                {
                    try
                    {
                        macm.GpuEntries[0].CoreClockCur = newValue;
                        isDirty = true;
                    }
                    catch (Exception e)
                    {
                        Console.WriteLine(e.Message);
                    }
                }
                else
                {
                    Console.WriteLine("New clock speed entered is not valid: " + input);
                }         

                // prompt user to apply changes in Afterburner
                if (isDirty)
                {
                    Console.WriteLine("\nWould you like to apply these changes?  Enter 'Y'");
                    ConsoleKeyInfo cki = Console.ReadKey(true);
                    if (cki.Key == ConsoleKey.Y)
                    {
                        // apply these changes in MSI Afterburner
                        Console.WriteLine("Committing Changes.....Pausing 2 seconds\n");
                        macm.CommitChanges();
                        System.Threading.Thread.Sleep(2000);
                        // reread changes from Afterburner to ensure the change took place
                        macm.ReloadAll();
                        Console.WriteLine("Fan Speed for GPU 0 is: " + macm.GpuEntries[0].FanSpeedCur);
                        Console.WriteLine("Core Clock Speed for GPU 0 is: " + macm.GpuEntries[0].CoreClockCur);

                        // prompt user to undo changes
                        Console.WriteLine("\nWould you like to undo these changes?  Enter 'Y'");
                        cki = Console.ReadKey(true);
                        if (cki.Key == ConsoleKey.Y)
                        {
                            // undo changes by reading back saved profile and applying it
                            stream = File.Open(filename, FileMode.Open);
                            bformatter = new BinaryFormatter();
                            ControlMemoryGpuEntry gpu = new ControlMemoryGpuEntry(macm.Header, 0);
                            gpu = (ControlMemoryGpuEntry)bformatter.Deserialize(stream);
                            stream.Close();
                            macm.GpuEntries[0] = gpu;
                            Console.WriteLine("Undoing Changes.....Pausing 2 seconds\n");
                            macm.CommitChanges();
                            System.Threading.Thread.Sleep(2000);
                            // reread changes from Afterburner to ensure the change took place
                            macm.ReloadAll();
                            Console.WriteLine("Fan Speed for GPU 0 is: " + macm.GpuEntries[0].FanSpeedCur);
                            Console.WriteLine("Core Clock Speed for GPU 0 is: " + macm.GpuEntries[0].CoreClockCur);
                        }
                    }
                    else
                    {
                        Console.WriteLine("\n*** Changes aborted ***");
                        Console.WriteLine();
                    }
                }
            }
            catch (Exception e)
            {
                Console.WriteLine(e.Message);
                if (e.InnerException != null)
                    Console.WriteLine(e.InnerException.Message);
            }

            Console.WriteLine("\nPress any key to exit");
            Console.ReadKey();
        }
    }
}