namespace AdventOfCode2022.App

open System.Diagnostics
open BenchmarkDotNet.Attributes
open BenchmarkDotNet.Configs
open BenchmarkDotNet.Running

module Benchmarks =
    type Benchmark1To5 () =
        [<GlobalSetup>]
        member _.Setup () = Run.shouldWrite <- false

        [<Params(1, 2, 3, 4, 5)>]
        member val Day = 0 with get, set

        [<Params(false, true)>]
        member val IsPartOne = false with get, set

        [<Benchmark>]
        member this.Benchmark () : unit =
            Run.allRuns.[this.Day - 1] (not this.IsPartOne) (Inputs.day this.Day)

        [<GlobalCleanup>]
        member _.Cleanup () = Run.shouldWrite <- true


    type Benchmark6To10 () =
        [<GlobalSetup>]
        member _.Setup () = Run.shouldWrite <- false

        [<Params(6, 7, 8, 9, 10)>]
        member val Day = 0 with get, set

        [<Params(false, true)>]
        member val IsPartOne = false with get, set

        [<Benchmark>]
        member this.Benchmark () : unit =
            Run.allRuns.[this.Day - 1] (not this.IsPartOne) (Inputs.day this.Day)

        [<GlobalCleanup>]
        member _.Cleanup () = Run.shouldWrite <- true

    type Benchmark11To15 () =
        [<GlobalSetup>]
        member _.Setup () = Run.shouldWrite <- false

        [<Params(11, 12, 13, 14, 15)>]
        member val Day = 0 with get, set

        [<Params(false, true)>]
        member val IsPartOne = false with get, set

        [<Benchmark>]
        member this.Benchmark () : unit =
            Run.allRuns.[this.Day - 1] (not this.IsPartOne) (Inputs.day this.Day)

        [<GlobalCleanup>]
        member _.Cleanup () = Run.shouldWrite <- true

    type Benchmark16To20 () =
        [<GlobalSetup>]
        member _.Setup () = Run.shouldWrite <- false

        [<Params(16, 17, 18, 19, 20)>]
        member val Day = 0 with get, set

        [<Params(false, true)>]
        member val IsPartOne = false with get, set

        [<Benchmark>]
        member this.Benchmark () : unit =
            Run.allRuns.[this.Day - 1] (not this.IsPartOne) (Inputs.day this.Day)

        [<GlobalCleanup>]
        member _.Cleanup () = Run.shouldWrite <- true


    type Benchmark21To25 () =
        static member finalBenchmarkArgs =
            (25, true) :: List.allPairs [ 21..24 ] [ false ; true ]
            |> Seq.ofList
            |> Seq.map box

        [<GlobalSetup>]
        member _.Setup () = Run.shouldWrite <- false

        [<Benchmark>]
        [<ArgumentsSource "finalBenchmarkArgs">]
        member this.Benchmark (args : obj) : unit =
            let day, isPartOne = unbox<int * bool> args

            if day <= 24 then
                Run.allRuns.[day - 1] (not isPartOne) (Inputs.day day)
            else if not isPartOne || day <> 25 then
                failwithf "Unexpected args: %+A" args
            else
                Run.day25 (Inputs.day 25)

        [<GlobalCleanup>]
        member _.Cleanup () = Run.shouldWrite <- true

module Program =

    [<EntryPoint>]
    let main args =
        match args with
        | [| "bench" |] ->
            let config =
                ManualConfig
                    .Create(DefaultConfig.Instance)
                    .WithOptions ConfigOptions.DisableOptimizationsValidator

            let _summary = BenchmarkRunner.Run<Benchmarks.Benchmark1To5> config
            let _summary = BenchmarkRunner.Run<Benchmarks.Benchmark6To10> config
            let _summary = BenchmarkRunner.Run<Benchmarks.Benchmark11To15> config
            let _summary = BenchmarkRunner.Run<Benchmarks.Benchmark16To20> config
            let _summary = BenchmarkRunner.Run<Benchmarks.Benchmark21To25> config
            0
        | _ ->

        let time = Stopwatch.StartNew ()
        time.Restart ()

        for day = 1 to Run.allRuns.Length do
            Run.allRuns.[day - 1] false (Inputs.day day)
            Run.allRuns.[day - 1] true (Inputs.day day)

        Run.day25 (Inputs.day 25)

        time.Stop ()
        printfn $"Took %i{time.ElapsedMilliseconds}ms"
        0