using JSON3, Printf, Plots
gr()

resources_dir = joinpath(@__DIR__, "resources")
include(joinpath(resources_dir, "julia", "crc_runtime.jl"))
include(joinpath(resources_dir, "julia", "polynomials.jl"))
crc_open_model()

show_params = Dict(
    "Payload" => ["SamplesPerFrame", "ProbabilityOfZero", "InitialSeed", "SampleTime"],
    "CRC Encoder" => ["GeneratorPolynomial", "InitialStates", "DirectMethod", "ReflectInputBytes", "FinalXOR"],
    "Channel Noise" => ["SamplesPerFrame", "ProbabilityOfZero", "InitialSeed", "SampleTime"],
    "Binary Symmetric Channel" => ["Operator"],
    "CRC Detector" => ["GeneratorPolynomial", "InitialStates", "DirectMethod"],
    "Data Changed" => ["Operator"], "Any Bit Changed" => ["Operator", "Inputs"],
    "Undetected" => ["Operator"], "Correct" => ["Operator"],
    "Previous Totals" => ["InitialCondition", "SampleTime"],
    "Stop Limits" => ["Value"], "Limit Reached" => ["Operator"])
for b in ["Payload", "CRC Encoder", "Channel Noise", "Binary Symmetric Channel", "CRC Detector", "Data Changed",
          "Any Bit Changed", "Undetected", "Correct", "Previous Totals", "Stop Limits", "Limit Reached"]
    prm = engee.get_param("crc_error_detection/" * b)
    println(rpad(b, 26), prm["BlockPath"])
    println(" "^26, join(["$k = $(prm[k])" for k in show_params[b]], "; "))
end

println(rpad("n", 5), rpad("k", 5), rpad("t", 5), rpad("工程师座", 13), "多项式(度)")
for x in crc_presets
    println(rpad(x.n, 5), rpad(x.k, 5), rpad(x.t, 5), rpad(x.supported ? "是的" : "非也。", 13), x.poly)
end
@printf("Поддерживается %d из %d полиномов\n", count(x -> x.supported, crc_presets), length(crc_presets))

crc_example_config = Dict("n" => 19, "p" => 0.1, "poly" => [5, 3, 0],
                          "blocksToStop" => 2000, "errorsToStop" => 2000, "seed" => 12345)
crc_example_result = JSON3.read(crc_run_json(JSON3.write(crc_example_config)))
@printf("Всего %d: верно %d, обнаружено %d, не обнаружено %d; битовых ошибок %d; %s; %.1f с\n",
        crc_example_result.packets, crc_example_result.correct, crc_example_result.detected,
        crc_example_result.undetected, crc_example_result.bit_errors, crc_example_result.stop_reason,
        crc_example_result.seconds)
@assert crc_example_result.correct + crc_example_result.detected + crc_example_result.undetected == crc_example_result.packets

verification = JSON3.read(include(joinpath(resources_dir, "julia", "verify_model.jl")))
println("所有检查已完成: ", verification.passed)
for t in verification.tests
    if haskey(t, :result)
        r = t.result
        @printf("  %-48s n=%-3d p=%-4g пакетов %-5d верно %-5d обн. %-5d не обн. %d\n",
                t.case, r.n, r.p, r.packets, r.correct, r.detected, r.undetected)
    else
        @printf("  %-48s кадров %d, одиночных ошибок %d\n", t.case, t.frames, t.single_bit_cases)
    end
end

poly = [5, 3, 0]; n_c, k_c = 19, 14
A = zeros(Int, n_c + 1)                          # A[w+1]为权重w的码字个数
for m in 0:2^k_c-1
    payload = [(m >> (k_c - i)) & 1 for i in 1:k_c]
    A[count(==(1), crc_reference_encode(payload, poly))+1] += 1
end
println("权重谱A_w(w:字数): ", join(["$(w):$(A[w+1])" for w in 0:n_c if A[w+1] > 0], "  "))
@printf("Минимальный ненулевой вес d_min = %d\n", findfirst(>(0), A[2:end]))

theory(p) = begin
    pc = (1 - p)^n_c
    pu = sum(A[w+1] * p^w * (1 - p)^(n_c - w) for w in 1:n_c)
    (correct = pc, undetected = pu, detected = 1 - pc - pu)
end

ps = [0.001, 0.003, 0.01, 0.03, 0.1, 0.2, 0.3, 0.5]
sim = Dict{Float64, Any}()
for p in ps
    cfg = Dict("n" => 19, "p" => p, "poly" => poly, "blocksToStop" => 20000, "errorsToStop" => 1_000_000, "seed" => 777)
    sim[p] = JSON3.read(crc_run_json(JSON3.write(cfg)))
    th = theory(p); r = sim[p]
    @printf("p = %-5g верно %.4f (теор. %.4f)  обн. %.4f (%.4f)  не обн. %.2e (%.2e)\n", p,
            r.correct / r.packets, th.correct, r.detected / r.packets, th.detected,
            r.undetected / r.packets, th.undetected)
end

# 理论概率由精确的权重谱a_w计算。
pgrid = 10 .^ range(-3, log10(0.5); length = 200)

function wilson95(k, n)
    z = 1.959963984540054
    q = k / n
    den = 1 + z^2 / n
    mid = (q + z^2 / (2n)) / den
    half = z * sqrt(q * (1 - q) / n + z^2 / (4n^2)) / den
    (max(0.0, mid - half), min(1.0, mid + half))
end

main_plot = plot(pgrid, [theory(p).correct for p in pgrid];
    color = :seagreen, lw = 2, label = "没错，理论上", xscale = :log10,
    ylims = (-0.03, 1.05), ylabel = "包的份额",
    title = "CRC x^5+x^3+1，n=19：模型和精确理论",
    legend = :outertopright, left_margin = 5Plots.mm)
plot!(main_plot, pgrid, [theory(p).detected for p in pgrid];
    color = :royalblue, lw = 2, label = "发现的，理论的")
scatter!(main_plot, ps, [sim[p].correct / sim[p].packets for p in ps];
    color = :seagreen, marker = :circle, ms = 6, label = "没错，模特")
scatter!(main_plot, ps, [sim[p].detected / sim[p].packets for p in ps];
    color = :royalblue, marker = :square, ms = 6, label = "检测到，模型")

rare_plot = plot(pgrid, [theory(p).undetected for p in pgrid];
    color = :orangered, lw = 2, label = "未检测到，理论",
    xscale = :log10, yscale = :log10, ylims = (1e-8, 0.1),
    xlabel = "P位的反转概率", ylabel = "包的份额",
    legend = :outertopright, left_margin = 5Plots.mm)
nonzero_ps = [p for p in ps if sim[p].undetected > 0]
zero_ps = [p for p in ps if sim[p].undetected == 0]
for (j, p) in enumerate(nonzero_ps)
    r = sim[p]
    low, high = wilson95(r.undetected, r.packets)
    plot!(rare_plot, [p, p], [low, high]; color = :orangered, lw = 2,
        label = j == 1 ? "95% интервал Уилсона" : "")
end
scatter!(rare_plot, nonzero_ps,
    [sim[p].undetected / sim[p].packets for p in nonzero_ps];
    color = :orangered, marker = :diamond, ms = 6, label = "未检测到，模型")
scatter!(rare_plot, zero_ps,
    [wilson95(0, sim[p].packets)[2] for p in zero_ps];
    color = :orangered, marker = :dtriangle, ms = 7,
    label = "0活动：上限95%")

println("内的事件份额预计会有95%的间隔")
for p in ps
    r = sim[p]
    low, high = wilson95(r.undetected, r.packets)
    expected = r.packets * theory(p).undetected
    @printf("%-7g %6d    %7.3f    [%.3g, %.3g]       %s\n",
        p, r.undetected, expected, low, high,
        low <= theory(p).undetected <= high ? "да" : "нет")
end
@assert all(p -> begin
    r = sim[p]
    low, high = wilson95(r.undetected, r.packets)
    low <= theory(p).undetected <= high
end, ps)

plot(main_plot, rare_plot; layout = (2, 1), size = (1050, 760))

include(joinpath(resources_dir, "julia", "start.jl"))
app_url = match(r"https://[^\s']+?/genie/CRCApp/", repr(crc_app_status))
println("应用程序正在运行: ", isnothing(app_url) ? crc_app_status : app_url.match)

th = theory(0.1); r = sim[0.1]
string("CRC ok; p=0.1: correct=", round(r.correct / r.packets, digits = 4), " (theory ", round(th.correct, digits = 4),
       "), undetected=", round(r.undetected / r.packets, sigdigits = 3), " (theory ", round(th.undetected, sigdigits = 3), ")")