#!/usr/bin/env julia
using KM3NeT
using Plots
using PlotThemes
using Dates
using Measures
GR.inline("png")

if length(ARGS) < 2
    println("Usage: ./live_royfit.jl DETX TIME_RES")
    exit(1)
end


const calib = KM3NeT.read_calibration(ARGS[1])
const LIGIER_PORT = parse(Int, ARGS[2])
const TIME_RES = ARGS[3]

function main()
    println("Starting live ROyFit")

    for message in CHClient(ip"127.0.0.1", LIGIER_PORT, ["IO_EVT"])
        event = KM3NeT.read_io(IOBuffer(message.data), KM3NeT.DAQEvent)

        hits = calibrate(event.hits, calib)
        triggered_hits = filter(h->h.triggered, hits)
        dus = sort(unique(map(h->h.du, hits)))
        n_dus = length(dus)
        n_doms = length(unique(h->h.dom_id, triggered_hits))

        if n_doms < 4
            continue
        end

        colours = palette(:default)
        plot()
        Q = []
        for (idx, du) in enumerate(dus)
            du_hits = filter(h->h.du == du, hits)
            if length(triggered(du_hits))== 0
                println("No triggered hits")
                continue
            end
            fit = KM3NeT.single_du_fit(du_hits)
            push!(Q, fit.Q)
            plot!(du_hits, fit, markercolor=colours[idx], label="DU $(du)", max_z=calib.max_z)
            write_time_residuals(TIME_RES, event, hits, fit)
        end
        if sum(Q) < 200 && n_doms > 12 && n_dus > 1
            println("Plotting...")
            fit_params = "ROy live reconstruction (combined single line): Q=$([round(_Q,digits=2) for _Q in Q])"
            event_params = "Det ID $(event.det_id), Run $(event.run_id), FrameIndex $(event.timeslice_id), TriggerCounter $(event.trigger_counter), Overlays $(event.overlays)"
            time_params = "$(unix2datetime(event.timestamp)) UTC"
            trigger_params = "Trigger: $(is_mxshower(event) ? "MX " : "")$(is_3dmuon(event) ? "3DM " : "")$(is_3dshower(event) ? "3DS " : "")"
            time_params = "$(unix2datetime(event.timestamp)) UTC"

            plot!(title="$(fit_params)\n$(event_params), $(trigger_params)\n$(time_params)", titlefontsize=8, margin=5mm)

            savefig("plots/ztplot_roy.png")
        end
    end
end

function write_time_residuals(filename, event, hits, fit)
    if !isfile(filename)
        fobj = open(filename, "w")
        write(fobj, "run,timestamp,du,floor,dom_id,t_res,Q\n")
    else
        fobj = open(filename, "a")
    end
    dγ, ccalc = KM3NeT.make_cherenkov_calculator(fit.sdp)
    for hit in hits
        Δt = hit.t - ccalc(hit.pos.z)
        write(fobj, "$(event.run_id),$(event.timestamp),$(hit.du),$(hit.floor),$(hit.dom_id),$(Δt),$(fit.Q)\n")
    end
    close(fobj)
end

main()