Detection Tracker with Calibration Replay

-- rebind: min_sdk=0.1.0 tick_rate=1000
-- rebind: name=File-Based Detection

local function getDetection()
    if File.Exists("detection.json") then
        local data = File.ReadJSON("detection.json")
        return data
    end
    return nil
end

function OnTick()
    local detection = getDetection()
    if detection and detection.x and detection.y then
        -- Move to detection center
        local mx, my = System.Mouse()
        local dx = detection.x - mx
        local dy = detection.y - my
        HID.Move(dx, dy)
    end
end

-- rebind: min_sdk=0.1.0 permission=net
-- rebind: name=HTTP Detection Server

local server = nil

function OnStart()
    server = Net.Listen(8888, function(req)
        if req.method == "POST" then
            local data = JSON.Parse(req.body)
            -- data = { x: 500, y = 300 }
            local mx, my = System.Mouse()
            local dx = data.x - mx
            local dy = data.y - my
            HID.Move(dx, dy)
            return { status = 200, body = "OK" }
        end
        return { status = 404 }
    end)
    Log.Info("Detection server on :8888")
end

function OnStop()
    if server then server:Stop() end
end

-- rebind: min_sdk=0.1.0 permission=net
-- rebind: name=WebSocket Detection

local server = nil

function OnStart()
    server = Net.WSListen(9000, {
        OnMessage = function(client, payload, is_binary)
            local data = JSON.Parse(payload)
            local mx, my = System.Mouse()
            local dx = data.x - mx
            local dy = data.y - my
            HID.Move(dx, dy)
        end,
    })
    Log.Info("WS detection on :9000")
end

function OnStop()
    if server then server:Stop() end
end

-- rebind: min_sdk=0.1.0
-- rebind: name=Detection Tracker
-- rebind: tick_rate=1000

local cfg = UI.Schema({
    enabled = UI.Toggle(true, { label = "Enable Tracking" }),
    smoothing = UI.Slider(3, { min = 1, max = 10, label = "Smoothing Window" }),
    jitter_scale = UI.Slider(0.5, { min = 0, max = 2, step = 0.1, label = "Jitter Scale" }),
    velocity_lead = UI.Slider(0.3, { min = 0, max = 1, step = 0.1, label = "Velocity Lead" }),
})

local SAMPLES_DIR = "calibration_samples"
local tracking_data = {
    enemy_heights = {},
    mouse_deltas = {},
    enemy_velocities = {},
    tracking_error = {},
    time_on_target = {}
}
local currentIndex = 1
local lastFileTime = 0
local lastFileName = nil

local sma_window = {}

local function addSMA(value)
    table.insert(sma_window, value)
    if #sma_window > cfg.smoothing then
        table.remove(sma_window, 1)
    end
    local sum = 0
    for _, v in ipairs(sma_window) do sum = sum + v end
    return sum / #sma_window
end

-- Find most recent calibration file
local function findLatestFile()
    if not File.Exists(SAMPLES_DIR) then
        File.MkDir(SAMPLES_DIR)
        return nil
    end
    
    local files = File.List(SAMPLES_DIR)
    local latest = nil
    local latestTime = 0
    
    for _, fname in ipairs(files) do
        if fname:match("calibration_samples_.*%.json$") or fname:match("calibration_samples_.*%.lua$") then
            -- Files are returned in modification order; first match is latest
            latest = fname
            break
        end
    end
    
    return latest and (SAMPLES_DIR .. "/" .. latest) or nil
end

-- Parse calibration file (JSON or Lua table format)
local function parseCalibrationFile(filepath)
    local content = File.Read(filepath)
    if not content then return nil end
    
    local data = {
        enemy_heights = {},
        mouse_deltas = {},
        enemy_velocities = {},
        tracking_error = {},
        time_on_target = {}
    }
    
    -- Try JSON first
    local ok, result = pcall(function() return JSON.Parse(content) end)
    if ok and result then
        data.enemy_heights = result.heights or result.enemy_heights or {}
        data.mouse_deltas = result.deltas or result.mouse_deltas or {}
        data.enemy_velocities = result.velocities or result.enemy_velocities or {}
        data.tracking_error = result.errors or result.tracking_error or {}
        data.time_on_target = result.times or result.time_on_target or {}
        return data
    end
    
    -- Fallback: try simple CSV/line format
    for line in content:gmatch("[^\n]+") do
        local height, delta, velocity, error, time = line:match("([%d.]+)[,%s]+([%d.]+)[,%s]+([%d.]+)[,%s]+([%d.]+)[,%s]+([%d.]+)")
        if height then
            table.insert(data.enemy_heights, tonumber(height))
            table.insert(data.mouse_deltas, tonumber(delta) or 50)
            table.insert(data.enemy_velocities, tonumber(velocity) or 5)
            table.insert(data.tracking_error, tonumber(error) or 10)
            table.insert(data.time_on_target, tonumber(time) or 0.05)
        end
    end
    
    return #data.enemy_heights > 0 and data or nil
end

-- Convert height to screen position
local function heightToPosition(height)
    local sw, sh = System.Screen()
    local focal_length = 1310
    local distance = focal_length / (height > 0 and height or 1)
    
    -- Simulate detection box center
    local cx = sw / 2 + (math.random() - 0.5) * 200
    local cy = sh / 2 + (math.random() - 0.5) * 150
    
    return math.floor(cx), math.floor(cy), distance
end

-- Get next detection from loaded data
local function getNextDetection()
    if not tracking_data or #tracking_data.enemy_heights == 0 then
        return nil
    end
    
    if currentIndex > #tracking_data.enemy_heights then
        currentIndex = 1
    end
    
    local height = tracking_data.enemy_heights[currentIndex]
    local x, y, dist = heightToPosition(height)
    
    local detection = {
        x = x,
        y = y,
        height = height,
        distance = dist,
        mouse_delta = tracking_data.mouse_deltas[currentIndex] or 50,
        enemy_velocity = tracking_data.enemy_velocities[currentIndex] or 5,
        tracking_error = tracking_data.tracking_error[currentIndex] or 10,
        time_on_target = tracking_data.time_on_target[currentIndex] or 0.05,
    }
    
    currentIndex = currentIndex + 1
    return detection
end

function OnStart()
    File.MkDir(SAMPLES_DIR)
    Log.Info("Detection Tracker ready")
    Log.Info("Place calibration files in: " .. SAMPLES_DIR)
    Log.Info("Supported formats: calibration_samples_*.json or *.lua")
end

function OnTick()
    if not cfg.enabled then return end
    
    local now = System.Time()
    
    -- Poll for new files every 100ms
    if now - lastFileTime > 100 then
        lastFileTime = now
        local latest = findLatestFile()
        
        if latest and latest ~= lastFileName then
            lastFileName = latest
            Log.Info("Loading: " .. latest)
            tracking_data = parseCalibrationFile(latest)
            
            if tracking_data and #tracking_data.enemy_heights > 0 then
                Log.Info("Loaded " .. #tracking_data.enemy_heights .. " samples")
                currentIndex = 1
                sma_window = {}
            else
                Log.Warn("Failed to parse file or no data found")
            end
        end
    end
    
    if not tracking_data or #tracking_data.enemy_heights == 0 then
        return
    end
    
    local detection = getNextDetection()
    if not detection then return end
    
    -- Calculate movement
    local mx, my = System.Mouse()
    local dx = detection.x - mx
    local dy = detection.y - my
    
    -- Apply smoothing
    dx = addSMA(dx)
    dy = addSMA(dy)
    
    -- Add jitter
    local jitterX = (math.random() - 0.5) * detection.tracking_error * cfg.jitter_scale
    local jitterY = (math.random() - 0.5) * detection.tracking_error * cfg.jitter_scale
    
    -- Lead target
    dx = dx + detection.enemy_velocity * cfg.velocity_lead
    dy = dy + detection.enemy_velocity * cfg.velocity_lead * 0.5
    
    -- Move
    HID.Move(dx + jitterX, dy + jitterY)
end

-- Hotkeys
Bind("F5", function()
    local latest = findLatestFile()
    if latest then
        tracking_data = parseCalibrationFile(latest)
        currentIndex = 1
        sma_window = {}
        Log.Info("Reloaded calibration data")
    end
    return false
end)

Bind("F6", function()
    cfg.enabled = not cfg.enabled
    UI.Notify(cfg.enabled and "Tracking ON" or "Tracking OFF", "info")
    return false
end)

function OnBlur()
    sma_window = {}
end

function OnStop()
    Log.Info("Detection Tracker stopped")
end

{
  "heights": [50, 55, 48, 52],
  "deltas": [25, 30, 22, 28],
  "velocities": [5, 6, 4, 5.5],
  "errors": [10, 12, 8, 11],
  "times": [0.05, 0.06, 0.04, 0.05]
}