debugging.lua

--------------------------------------------------------------------------------------------
--- AUTHOR: Keithen
--- GITHUB REPO: https://github.com/Nostrademous/Dota2-FullOverwrite
--------------------------------------------------------------------------------------------

_G._savedEnv = getfenv()
module( "debugging", package.seeall )

local utils = require(GetScriptDirectory() .. "/utility")
local gHeroVar = require( GetScriptDirectory().."/global_hero_data" )

local retreatMode = dofile( GetScriptDirectory().."/modes/retreat" )

local last_draw_time = -500

local bot_states = {}
local team_states = {}
local circles = {}

local LINE_HEIGHT = 10
local TITLE_VALUE_DELTA_X = 10

local BOT_STATES_MAX_LINES = 2
local BOT_STATES_X = 1600
local BOT_STATES_Y = 100

local TEAM_STATES_MAX_LINES = 6
local TEAM_STATES_X = 1550
local TEAM_STATES_Y = 400

local function updateBotStates()
    local listAllies = GetUnitList(UNIT_LIST_ALLIED_HEROES)
    for _, ally in pairs(listAllies) do
        if ally:IsBot() and not ally:IsIllusion() and gHeroVar.HasID(ally:GetPlayerID()) then
            local hMyBot = ally.SelfRef
            local mode = hMyBot:getCurrentMode()
            local state = mode:GetName()
            if state == "laning" then
                local cLane = hMyBot:getHeroVar("CurLane")
                state = state .. " Lane: " .. tostring(cLane) .. " Info: " .. hMyBot:getHeroVar("LaningStateInfo")
            elseif state == "fight" then
                local target = hMyBot:getHeroVar("Target")
                if utils.ValidTarget(target) then
                    state = state .. " " .. utils.GetHeroName(target)
                end
            elseif state == "roam" then
                local target = hMyBot:getHeroVar("RoamTarget")
                if utils.ValidTarget(target) then
                    state = state .. " " .. utils.GetHeroName(target)
                end
            elseif state == "retreat" then
                state = state .. " " .. ally.retreat_desire_debug
            end
            SetBotState(hMyBot.Name, 1, state)
        end
    end
end

-- gets called by the framework
function draw()
    if last_draw_time > GameTime() - 0.010 then return end -- TODO: check actual frame time
    last_draw_time = GameTime()

    updateBotStates()
    
    local y = BOT_STATES_Y
    for name, v in utils.Spairs(bot_states) do
        DebugDrawText( BOT_STATES_X, y, name, 255, 0, 0 )
        for line,text in pairs(v) do
            DebugDrawText( BOT_STATES_X + TITLE_VALUE_DELTA_X, y + line * LINE_HEIGHT, text, 255, 0, 0 )
        end
        y = y + (BOT_STATES_MAX_LINES + 1) * LINE_HEIGHT
    end

    y = TEAM_STATES_Y
    for name, v in utils.Spairs(team_states) do
        DebugDrawText( TEAM_STATES_X, y, name, 255, 0, 0 )
        for line,text in pairs(v) do
            DebugDrawText( TEAM_STATES_X + TITLE_VALUE_DELTA_X, y + line * LINE_HEIGHT, text, 255, 0, 0 )
        end
        y = y + (TEAM_STATES_MAX_LINES + 1) * LINE_HEIGHT
    end

    for name, circle in pairs(circles) do
        -- TODO: draw name? (kinda hard to do, but could be nice)
        DebugDrawCircle( circle.center, circle.radius, circle.r, circle.g, circle.b )
    end
end

-- set a line in the specified bot's text area
function SetBotState(name, line, text)
    if line < 1 or BOT_STATES_MAX_LINES > 2 then
        print("SetBotState: line out of bounds!")
        return
    end
    if bot_states[name] == nil then
        bot_states[name] = {}
    end
    bot_states[name][line] = text
end

-- set a line in the specified category's text area
function SetTeamState(category, line, text)
    if line < 1 or line > TEAM_STATES_MAX_LINES then
        print("SetBotState: line out of bounds!")
        return
    end
    if team_states[category] == nil then
        team_states[category] = {}
    end
    team_states[category][line] = text
end

-- draw a (filled) circle
function SetCircle(name, center, r, g, b, radius)
    if radius == nil then radius = 50 end

    circles[name] = {["center"] = center, ["r"] = r, ["g"] = g, ["b"] = b, ["radius"] = radius}
end

-- remove a circle
function DeleteCircle(name)
    circles[name] = nil
end

for k,v in pairs( debugging ) do _G._savedEnv[k] = v end