using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Text;
using static Tensorflow.Binding;

namespace Tensorflow
{
    public class MemoryMonitor
    {
        public void WarmUp()
        {
            print(tf.VERSION);
        }

        public void Execute(int epoch, int iterate, Action<int> process)
        {
            /*GC.Collect();
            GC.WaitForPendingFinalizers();
            GC.Collect();*/

            print($"{process.Method.Name} started...");
            for (int i = 0; i < epoch; i++)
            {
                var initialMemory = Process.GetCurrentProcess().PrivateMemorySize64;// GC.GetTotalMemory(true);
                process(iterate);
                var finalMemory = Process.GetCurrentProcess().PrivateMemorySize64; //GC.GetTotalMemory(true);
                print($"Epoch {i}: {Format(finalMemory - initialMemory)}.");
            }

            GC.Collect();
            GC.WaitForPendingFinalizers();
            GC.Collect();

            print($"Total {process.Method.Name} usage {Format(Process.GetCurrentProcess().PrivateMemorySize64)}");
        }

        private string Format(long usage)
        {
            if (usage < 0)
                return $"-{Format(0 - usage)}";

            if (usage <= 1024 && usage >= 0)
                return $"{usage} Bytes";
            else if (usage > 1024 && usage <= 1024 * 1024)
                return $"{usage / 1024} KB";
            else
                return $"{usage / 1024 / 1024} MB";
        }
    }
}