using BenchmarkDotNet.Attributes;
using BenchmarkDotNet.Diagnosers;
using BenchmarkDotNet.Diagnostics.dotTrace;
using BenchmarkDotNet.Jobs;
using Craftimizer.Simulator;
using Craftimizer.Solver;
using System.Runtime.CompilerServices;

namespace Craftimizer.Benchmark;

[SimpleJob(RuntimeMoniker.Net80, baseline: true)]
//[SimpleJob(RuntimeMoniker.Net90)]
[MinColumn, Q1Column, Q3Column, MaxColumn]
//[DotTraceDiagnoser]
//[MemoryDiagnoser]
[DisassemblyDiagnoser(maxDepth: 500, exportGithubMarkdown: false, exportHtml: true)]
public class Bench
{
    public record struct HashWrapper<T>(T Data) where T : notnull
    {
        public static implicit operator T(HashWrapper<T> wrapper) => wrapper.Data;

        public override readonly string ToString() =>
            $"{HashCode.Combine(Data.ToString()!):X8}";
    }

    private static SimulationInput[] Inputs { get; } =
    [
        // https://craftingway.app/rotation/loud-namazu-jVe9Y
        // Chondrite Saw
        new(new()
        {
            Craftsmanship = 3304,
            Control = 3374,
            CP = 575,
            Level = 90,
            CanUseManipulation = true,
            HasSplendorousBuff = false,
            IsSpecialist = false,
        },
        new()
        {
            IsExpert = false,
            ClassJobLevel = 90,
            ConditionsFlag = 0b1111,
            MaxDurability = 80,
            MaxQuality = 7200,
            MaxProgress = 3500,
            QualityModifier = 80,
            QualityDivider = 115,
            ProgressModifier = 90,
            ProgressDivider = 130
        }),

        // https://craftingway.app/rotation/sandy-fafnir-doVCs
        // Classical Longsword
        new(new()
        {
            Craftsmanship = 3290,
            Control = 3541,
            CP = 649,
            Level = 90,
            CanUseManipulation = true,
            HasSplendorousBuff = false,
            IsSpecialist = false,
        },
        new()
        {
            IsExpert = false,
            ClassJobLevel = 90,
            ConditionsFlag = 0b1111,
            MaxDurability = 70,
            MaxQuality = 10920,
            MaxProgress = 3900,
            QualityModifier = 70,
            QualityDivider = 115,
            ProgressModifier = 80,
            ProgressDivider = 130
        })
    ];

    public static IEnumerable<HashWrapper<SimulationState>> States => Inputs.Select(i => new HashWrapper<SimulationState>(new(i)));

    public static IEnumerable<HashWrapper<SolverConfig>> Configs => new HashWrapper<SolverConfig>[]
    {
        new(new()
        {
            Algorithm = SolverAlgorithm.Stepwise,
            Iterations = 30_000,
        })
    };

    [ParamsSource(nameof(States))]
    public HashWrapper<SimulationState> State { get; set; }

    [ParamsSource(nameof(Configs))]
    public HashWrapper<SolverConfig> Config { get; set; }

    // [Benchmark]
    public async Task<float> SolveAsync()
    {
        var solver = new Solver.Solver(Config, State);
        solver.Start();
        var (_, s) = await solver.GetTask().ConfigureAwait(false);
        return (float)s.Quality / s.Input.Recipe.MaxQuality;
    }

    [Benchmark]
    [MethodImpl(MethodImplOptions.NoInlining)]
    public (float MaxScore, SolverSolution Solution) Solve()
    {
        var config = new MCTSConfig(Config.Data);

        var solver = new MCTS(config, State);
        var progress = 0;
        solver.Search(Config.Data.Iterations, ref progress, CancellationToken.None);
        var solution = solver.Solution();

        return (solver.MaxScore, solution);
    }
}