warscript 0.0.1-dev.e0e46c4 → 0.0.1-dev.e147c49

package/engine/local-client.d.ts

@@ -3,19 +3,46 @@ import { Unit } from "../core/types/unit";
 import { Async } from "../core/types/async";
 import { Event, TriggerEvent } from "../event";
 import { GraphicsMode } from "./index";
+import { Color } from "../core/types/color";
 export declare class LocalClient {
     private constructor();
     static readonly locale: string;
     static get width(): number;
     static get height(): number;
+    static get isSD(): boolean;
     static get isHD(): boolean;
+    static get isDE(): boolean;
     static get graphicsMode(): GraphicsMode;
     static get isActive(): boolean;
+    static get isMinimized(): boolean;
+    static pingMinimap(x: number, y: number, duration: number, ...parameters: [] | [red: number, green: number, blue: number, flashy?: boolean] | [color: Color, flashy?: boolean]): void;
+    /** Local input state; do not use to branch synchronized game logic. */
+    static isKeyPressed(key: joskeytype): boolean;
+    /** Local input state. Uses the same modifier mask as Player key events. */
+    static isMetaKeyPressed(key: oskeymeta): boolean;
+    /** Local input state. */
+    static isMouseButtonPressed(button: jmousebuttontype): boolean;
+    static get mouseScreenX(): number;
+    static get mouseScreenY(): number;
+    static screenToFrameX(x: number): number;
+    static screenToFrameY(y: number): number;
+    static frameToScreenX(x: number): number;
+    static frameToScreenY(y: number): number;
+    static get mouseFrameX(): number;
+    static get mouseFrameY(): number;
     static get mouseFocusUnit(): Async<Unit> | undefined;
+    /**
+     * Local selection sampled every 1/64 second. Undefined before the first tick
+     * or until startup selection-frame initialization has finished.
+     */
     static get mainSelectedUnit(): Async<Unit> | undefined;
     static get mainSelectedUnitChangeEvent(): Event<[
         previousMainSelectedUnit: Unit | undefined,
         newMainSelectedUnit: Unit | undefined
     ]>;
+    static get isInTargetingMode(): boolean;
+    static readonly targetingModeEnterEvent: Event<[]>;
+    static readonly targetingModeLeaveEvent: Event<[]>;
+    static readonly targetingModeStateChangeEvent: Event<[]>;
     static readonly onDisconnect: TriggerEvent<[]>;
 }

package/engine/local-client.lua

@@ -1,9 +1,11 @@
 local ____lualib = require("lualib_bundle")
-local __TS__ArrayMap = ____lualib.__TS__ArrayMap
+local __TS__New = ____lualib.__TS__New
 local __TS__Class = ____lualib.__TS__Class
+local __TS__InstanceOf = ____lualib.__TS__InstanceOf
 local __TS__ObjectDefineProperty = ____lualib.__TS__ObjectDefineProperty
-local __TS__New = ____lualib.__TS__New
+local __TS__ArrayFilter = ____lualib.__TS__ArrayFilter
 local ____exports = {}
+local actualizeTargetingModeState
 local ____unit = require("core.types.unit")
 local Unit = ____unit.Unit
 local ____event = require("event")
@@ -15,32 +17,140 @@ local ____player = require("core.types.player")
 local Player = ____player.Player
 local ____timer = require("core.types.timer")
 local Timer = ____timer.Timer
-local loadTOCFile = BlzLoadTOCFile
-local getLocalClientWidth = BlzGetLocalClientWidth
-local getLocalClientHeight = BlzGetLocalClientHeight
-local isLocalClientActive = BlzIsLocalClientActive
-local isHeroUnitId = IsHeroUnitId
+local ____color = require("core.types.color")
+local Color = ____color.Color
+local ____arrays = require("utility.arrays")
+local array = ____arrays.array
+local ____unit = require("objutil.unit")
+local UnitDefinition = ____unit.UnitDefinition
+local ____socket = require("net.socket")
+local Socket = ____socket.Socket
+local frameGetChild = BlzFrameGetChild
+local frameGetChildrenCount = BlzFrameGetChildrenCount
+local frameGetParent = BlzFrameGetParent
+local frameToPixelX = BlzFrameToPixelX
+local frameToPixelY = BlzFrameToPixelY
+local getFrameByName = BlzGetFrameByName
 local getHandleId = GetHandleId
+local getLocalClientHeight = BlzGetLocalClientHeight
+local getLocalClientWidth = BlzGetLocalClientWidth
+local getLocale = BlzGetLocale
 local getMouseFocusUnit = BlzGetMouseFocusUnit
+local getMouseScreenPosX = BlzGetMouseScreenPosX
+local getMouseScreenPosY = BlzGetMouseScreenPosY
 local getUnitRealField = BlzGetUnitRealField
 local getUnitTypeId = GetUnitTypeId
-local getLocale = BlzGetLocale
+local isHeroUnitId = IsHeroUnitId
+local isKeyPressed = BlzIsKeyPressed
+local isLocalClientActive = BlzIsLocalClientActive
+local isMetaKeyPressed = BlzIsMetaKeyPressed
+local isMouseButtonPressed = BlzIsMouseButtonPressed
+local loadTOCFile = BlzLoadTOCFile
+local location = Location
+local pingMinimap = PingMinimap
+local pingMinimapEx = PingMinimapEx
+local pixelToFrameX = BlzPixelToFrameX
+local pixelToFrameY = BlzPixelToFrameY
 local tableSort = table.sort
-local tocPath = "_warscript\\IsHD.toc"
+local hdTocPath = "_warscript\\IsHD.toc"
+local deTocPath = "_warscript\\IsDE.toc"
 compiletime(function()
     if currentMap then
-        local fdfPath = "_warscript\\IsHD.fdf"
-        currentMap:addFileString("_HD.w3mod\\" .. fdfPath, "\r\n")
-        currentMap:addFileString("_HD.w3mod\\" .. tocPath, fdfPath .. "\r\n")
+        local hdFdfPath = "_warscript\\IsHD.fdf"
+        currentMap:addFileString("_HD.w3mod\\" .. hdFdfPath, "\r\n")
+        currentMap:addFileString("_HD.w3mod\\" .. hdTocPath, hdFdfPath .. "\r\n")
+        local deFdfPath = "_warscript\\IsDE.fdf"
+        currentMap:addFileString("_DE.w3mod\\" .. deFdfPath, "\r\n")
+        currentMap:addFileString("_DE.w3mod\\" .. deTocPath, deFdfPath .. "\r\n")
     end
 end)
-local selectionButtons
-Timer:run(function()
-    selectionButtons = __TS__ArrayMap(
-        Frame:byName("SimpleInfoPanelUnitDetail").parent:getChild(5):getChild(0).children,
-        function(____, frame) return frame:getChild(1) end
-    )
+local selectionWarmupUnitTypeId = compiletime(function()
+    local dummy = __TS__New(UnitDefinition, "hfoo", nil, {
+        name = "[Warscript/Selection Warmup] Unit",
+        model = "",
+        modelHD = "",
+        foodCost = 0,
+        movementType = "none",
+        collisionSize = 0,
+        selectionScale = 0,
+        sightRadiusDay = 0,
+        sightRadiusNight = 0,
+        abilitiesNormal = {"Avul"}
+    })
+    dummy.weapons[1].enabled = false
+    dummy.weapons[2].enabled = false
+    return fourCC(dummy.id)
 end)
+local SELECTION_DETAIL_FRAME_NAME = "SimpleInfoPanelUnitDetail"
+local SELECTION_GROUP_PANEL_INDEX = 5
+local SELECTION_BUTTON_ICON_INDEX = 1
+--- The detail panel, its parent, the group panel and the button grid.
+local SELECTION_LAYOUT_ROOT_HANDLE_COUNT = 4
+--- The group selection layouts by button count, in the order the startup warmup goes through them.
+local SELECTION_LAYOUT_BUTTON_COUNTS = {24, 12}
+local function getSelectionButtons()
+    return Frame:byName(SELECTION_DETAIL_FRAME_NAME).parent:getChild(SELECTION_GROUP_PANEL_INDEX):getChild(0)
+end
+local function getMainSelectedUnitIndex(selectionButtons, buttonCount)
+    local mainSelectedUnitIndex
+    local maxButtonWidth = 0
+    for i = 0, buttonCount - 1 do
+        local width = selectionButtons:getChild(i):getChild(SELECTION_BUTTON_ICON_INDEX).width
+        if width > maxButtonWidth then
+            maxButtonWidth = width
+            mainSelectedUnitIndex = i
+        end
+    end
+    return mainSelectedUnitIndex
+end
+local function getChildIfPresent(parent, index)
+    if parent == nil or index >= frameGetChildrenCount(parent) then
+        return nil
+    end
+    return frameGetChild(parent, index)
+end
+--- Wraps every frame of the group selection layout currently shown on this client, in the
+-- order `actualizeMainSelectedUnit` later walks them. Game-owned frames receive a JASS handle
+-- id the first time a script retrieves them, so this must run on every client in the same
+-- tick and register the same number of handles everywhere: every frame reached for the first
+-- time counts, and the remainder up to the layout's total is padded with locations, the way
+-- `Frame.byOrigin` balances origin frames that appear asynchronously.
+-- 
+-- Returns whether the local layout has `buttonCount` buttons and every frame was reached.
+local function registerSelectionLayoutFrames(buttonCount)
+    local registeredCount = 0
+    local function register(handle)
+        if handle == nil or getHandleId(handle) == 0 then
+            return nil
+        end
+        if not Frame:isWrapped(handle) then
+            registeredCount = registeredCount + 1
+        end
+        Frame:of(handle)
+        return handle
+    end
+    local detail = register(getFrameByName(SELECTION_DETAIL_FRAME_NAME, 0))
+    local container = register(detail ~= nil and frameGetParent(detail) or nil)
+    local groupPanel = register(getChildIfPresent(container, SELECTION_GROUP_PANEL_INDEX))
+    local grid = register(getChildIfPresent(groupPanel, 0))
+    local complete = grid ~= nil and frameGetChildrenCount(grid) == buttonCount
+    for i = 0, buttonCount - 1 do
+        local button = register(getChildIfPresent(grid, i))
+        local icon = register(getChildIfPresent(button, SELECTION_BUTTON_ICON_INDEX))
+        if icon == nil then
+            complete = false
+        end
+    end
+    local totalHandleCount = SELECTION_LAYOUT_ROOT_HANDLE_COUNT + 2 * buttonCount
+    do
+        local i = registeredCount
+        while i < totalHandleCount do
+            location(0, 0)
+            i = i + 1
+        end
+    end
+    return complete
+end
 local localSelectedUnits = {}
 local indexByLocalSelectedUnit = {}
 local function compareUnitsSelectionPriority(a, b)
@@ -57,11 +167,62 @@ local function compareUnitsSelectionPriority(a, b)
 end
 local mainSelectedUnitChangeEvent
 local previousMainSelectedUnit
+local currentMainSelectedUnit
+local lastTargetingModeState = false
+local targetingModeEnterEvent = __TS__New(Event)
+local targetingModeLeaveEvent = __TS__New(Event)
+local targetingModeStateChangeEvent = __TS__New(Event)
 ____exports.LocalClient = __TS__Class()
 local LocalClient = ____exports.LocalClient
 LocalClient.name = "LocalClient"
 function LocalClient.prototype.____constructor(self)
 end
+function LocalClient.pingMinimap(self, x, y, duration, redOrColor, greenOrFlashy, blue, flashy)
+    if redOrColor == nil then
+        pingMinimap(x, y, duration)
+    elseif __TS__InstanceOf(redOrColor, Color) then
+        pingMinimapEx(
+            x,
+            y,
+            duration,
+            redOrColor.r,
+            redOrColor.g,
+            redOrColor.b,
+            greenOrFlashy or false
+        )
+    else
+        pingMinimapEx(
+            x,
+            y,
+            duration,
+            redOrColor,
+            greenOrFlashy,
+            blue,
+            flashy or false
+        )
+    end
+end
+function LocalClient.isKeyPressed(self, key)
+    return isKeyPressed(key)
+end
+function LocalClient.isMetaKeyPressed(self, key)
+    return isMetaKeyPressed(key)
+end
+function LocalClient.isMouseButtonPressed(self, button)
+    return isMouseButtonPressed(button)
+end
+function LocalClient.screenToFrameX(self, x)
+    return pixelToFrameX(x)
+end
+function LocalClient.screenToFrameY(self, y)
+    return pixelToFrameY(y)
+end
+function LocalClient.frameToScreenX(self, x)
+    return frameToPixelX(x)
+end
+function LocalClient.frameToScreenY(self, y)
+    return frameToPixelY(y)
+end
 LocalClient.locale = getLocale()
 __TS__ObjectDefineProperty(
     LocalClient,
@@ -77,6 +238,13 @@ __TS__ObjectDefineProperty(
         return getLocalClientHeight()
     end}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "isSD",
+    {get = function(self)
+        return true
+    end}
+)
 __TS__ObjectDefineProperty(
     LocalClient,
     "isHD",
@@ -84,11 +252,18 @@ __TS__ObjectDefineProperty(
         return false
     end}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "isDE",
+    {get = function(self)
+        return false
+    end}
+)
 __TS__ObjectDefineProperty(
     LocalClient,
     "graphicsMode",
     {get = function(self)
-        return self.isHD and 1 or 0
+        return 0
     end}
 )
 __TS__ObjectDefineProperty(
@@ -98,6 +273,41 @@ __TS__ObjectDefineProperty(
         return isLocalClientActive()
     end}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "isMinimized",
+    {get = function(self)
+        return getLocalClientHeight() == 0
+    end}
+)
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "mouseScreenX",
+    {get = function(self)
+        return getMouseScreenPosX()
+    end}
+)
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "mouseScreenY",
+    {get = function(self)
+        return getMouseScreenPosY()
+    end}
+)
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "mouseFrameX",
+    {get = function(self)
+        return pixelToFrameX(getMouseScreenPosX())
+    end}
+)
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "mouseFrameY",
+    {get = function(self)
+        return pixelToFrameY(getMouseScreenPosY())
+    end}
+)
 __TS__ObjectDefineProperty(
     LocalClient,
     "mouseFocusUnit",
@@ -109,33 +319,7 @@ __TS__ObjectDefineProperty(
     LocalClient,
     "mainSelectedUnit",
     {get = function(self)
-        Unit:getSelectionOf(Player["local"], localSelectedUnits)
-        for i = 1, #localSelectedUnits do
-            indexByLocalSelectedUnit[localSelectedUnits[i]] = i
-        end
-        tableSort(localSelectedUnits, compareUnitsSelectionPriority)
-        local mainSelectedUnitIndex
-        if selectionButtons and #localSelectedUnits > 1 then
-            local maxButtonWidth = 0
-            for i = 0, #selectionButtons - 1 do
-                local width = selectionButtons[i + 1].width
-                if width > maxButtonWidth then
-                    maxButtonWidth = width
-                    mainSelectedUnitIndex = i
-                end
-            end
-        end
-        local mainSelectedUnit = localSelectedUnits[(mainSelectedUnitIndex or 0) + 1]
-        for i = 1, #localSelectedUnits do
-            indexByLocalSelectedUnit[localSelectedUnits[i]] = nil
-            localSelectedUnits[i] = nil
-        end
-        if mainSelectedUnitChangeEvent ~= nil and mainSelectedUnit ~= previousMainSelectedUnit then
-            local previousPreviousMainSelectedUnit = previousMainSelectedUnit
-            previousMainSelectedUnit = mainSelectedUnit
-            Event.invoke(mainSelectedUnitChangeEvent, previousPreviousMainSelectedUnit, previousMainSelectedUnit)
-        end
-        return mainSelectedUnit
+        return currentMainSelectedUnit
     end}
 )
 __TS__ObjectDefineProperty(
@@ -144,13 +328,20 @@ __TS__ObjectDefineProperty(
     {get = function(self)
         if mainSelectedUnitChangeEvent == nil then
             mainSelectedUnitChangeEvent = __TS__New(Event)
-            Timer.onPeriod[1 / 64]:addListener(function()
-                local _ = ____exports.LocalClient.mainSelectedUnit
-            end)
         end
         return mainSelectedUnitChangeEvent
     end}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "isInTargetingMode",
+    {get = function(self)
+        return actualizeTargetingModeState()
+    end}
+)
+LocalClient.targetingModeEnterEvent = targetingModeEnterEvent
+LocalClient.targetingModeLeaveEvent = targetingModeLeaveEvent
+LocalClient.targetingModeStateChangeEvent = targetingModeStateChangeEvent
 LocalClient.onDisconnect = __TS__New(
     TriggerEvent,
     function(trigger)
@@ -158,11 +349,284 @@ LocalClient.onDisconnect = __TS__New(
     end,
     function() return  end
 )
+local function actualizeMainSelectedUnit()
+    Unit:getSelectionOf(Player["local"], localSelectedUnits)
+    for i = 1, #localSelectedUnits do
+        indexByLocalSelectedUnit[localSelectedUnits[i]] = i
+    end
+    tableSort(localSelectedUnits, compareUnitsSelectionPriority)
+    local mainSelectedUnitIndex
+    if #localSelectedUnits > 1 then
+        local selectionButtons = getSelectionButtons()
+        mainSelectedUnitIndex = getMainSelectedUnitIndex(
+            selectionButtons,
+            selectionButtons:getChildrenCount()
+        )
+    end
+    local mainSelectedUnit = localSelectedUnits[(#localSelectedUnits > 1 and (mainSelectedUnitIndex or 0) or 0) + 1]
+    for i = 1, #localSelectedUnits do
+        indexByLocalSelectedUnit[localSelectedUnits[i]] = nil
+        localSelectedUnits[i] = nil
+    end
+    currentMainSelectedUnit = mainSelectedUnit
+    if mainSelectedUnitChangeEvent ~= nil and mainSelectedUnit ~= previousMainSelectedUnit then
+        local previousPreviousMainSelectedUnit = previousMainSelectedUnit
+        previousMainSelectedUnit = mainSelectedUnit
+        Event.invoke(mainSelectedUnitChangeEvent, previousPreviousMainSelectedUnit, previousMainSelectedUnit)
+    end
+end
+local commandButtons = array(
+    12,
+    function(i) return Frame:byOrigin(ORIGIN_FRAME_COMMAND_BUTTON, i) end
+)
+local function getTargetingModeState()
+    for i = 0, 10 do
+        if commandButtons[i + 1].visible then
+            return false
+        end
+    end
+    return commandButtons[12].visible
+end
+actualizeTargetingModeState = function()
+    if getTargetingModeState() then
+        if not lastTargetingModeState then
+            lastTargetingModeState = true
+            Event.invoke(targetingModeEnterEvent)
+            Event.invoke(targetingModeStateChangeEvent)
+        end
+        return true
+    end
+    if lastTargetingModeState then
+        lastTargetingModeState = false
+        Event.invoke(targetingModeLeaveEvent)
+        Event.invoke(targetingModeStateChangeEvent)
+    end
+    return false
+end
+Timer.onPeriod[1 / 64]:addListener(function()
+    actualizeTargetingModeState()
+end)
+local function initializeSelectionFrames()
+    local checkProgress, registerStage, checkRegistration, onMessage, onPlayerLeave, observeSelection, retryTimer
+    local warmups = {}
+    local warmupByPlayer = {}
+    local dummyUnits = {}
+    local socket = __TS__New(Socket)
+    local stageIndex = 0
+    local round = 0
+    local registering = false
+    --- The round whose frames this client has registered.
+    local registeredRound = 0
+    local finished = false
+    local sentSeenRound = 0
+    local seenMessage = ""
+    local registeredMessage = ""
+    local failedMessage = ""
+    local function getStageButtonCount()
+        return SELECTION_LAYOUT_BUTTON_COUNTS[stageIndex + 1]
+    end
+    local function retrySelection()
+        local count = getStageButtonCount()
+        for ____, warmup in ipairs(warmups) do
+            do
+                if not warmup.player.isPlaying then
+                    goto __continue69
+                end
+                if warmup.player == Player["local"] then
+                    local selection = Unit:getSelectionOf(Player["local"])
+                    local normalSelection = __TS__ArrayFilter(
+                        selection,
+                        function(____, unit) return not (dummyUnits[unit] ~= nil) end
+                    )
+                    if #normalSelection > 0 or #selection == 0 then
+                        warmup.previousSelection = normalSelection
+                    end
+                end
+                warmup.player:clearSelection()
+                for i = 0, count - 1 do
+                    warmup.player:select(warmup.units[i + 1])
+                end
+            end
+            ::__continue69::
+        end
+    end
+    local function startRound()
+        round = round + 1
+        registering = false
+        local count = getStageButtonCount()
+        seenMessage = "seen" .. tostring(count)
+        registeredMessage = "registered" .. tostring(count)
+        failedMessage = "failed" .. tostring(count)
+        retrySelection()
+        checkProgress()
+    end
+    local function finish()
+        finished = true
+        retryTimer:destroy()
+        Timer.onPeriod[1 / 64]:removeListener(observeSelection)
+        socket.onMessage:removeListener(onMessage)
+        Player.onLeave:removeListener(onPlayerLeave)
+        local restoreSelection
+        local selection = Unit:getSelectionOf(Player["local"])
+        for ____, unit in ipairs(selection) do
+            if dummyUnits[unit] ~= nil then
+                restoreSelection = __TS__ArrayFilter(
+                    selection,
+                    function(____, selected) return not (dummyUnits[selected] ~= nil) end
+                )
+                if #restoreSelection == 0 then
+                    local ____opt_0 = warmupByPlayer[Player["local"]]
+                    restoreSelection = ____opt_0 and ____opt_0.previousSelection
+                end
+                Player["local"]:clearSelection()
+                break
+            end
+        end
+        for ____, warmup in ipairs(warmups) do
+            for ____, unit in ipairs(warmup.units) do
+                unit:destroy()
+            end
+            warmup.units = {}
+        end
+        if restoreSelection ~= nil then
+            for ____, unit in ipairs(restoreSelection) do
+                if getUnitTypeId(unit.handle) ~= 0 then
+                    Player["local"]:select(unit)
+                end
+            end
+        end
+        Timer.onPeriod[1 / 64]:addListener(actualizeMainSelectedUnit)
+    end
+    checkProgress = function()
+        if finished or registering then
+            return
+        end
+        for ____, warmup in ipairs(warmups) do
+            if warmup.player.isPlaying and warmup.seenRound ~= round then
+                return
+            end
+        end
+        registering = true
+        Timer:simple(1 / 64, registerStage)
+    end
+    registerStage = function()
+        if finished then
+            return
+        end
+        registeredRound = round
+        local succeeded = registerSelectionLayoutFrames(getStageButtonCount())
+        if warmupByPlayer[Player["local"]] ~= nil and Player["local"].isPlaying then
+            socket:send(succeeded and registeredMessage or failedMessage)
+        end
+        checkRegistration()
+    end
+    checkRegistration = function()
+        if finished or not registering then
+            return
+        end
+        local succeeded = true
+        for ____, warmup in ipairs(warmups) do
+            do
+                if not warmup.player.isPlaying then
+                    goto __continue101
+                end
+                if warmup.registrationRound ~= round then
+                    return
+                end
+                if not warmup.registrationSucceeded then
+                    succeeded = false
+                end
+            end
+            ::__continue101::
+        end
+        if not succeeded then
+            startRound()
+            return
+        end
+        if stageIndex + 1 >= #SELECTION_LAYOUT_BUTTON_COUNTS then
+            finish()
+            return
+        end
+        stageIndex = stageIndex + 1
+        local count = getStageButtonCount()
+        for ____, warmup in ipairs(warmups) do
+            for i = #warmup.units - 1, count, -1 do
+                warmup.units[i + 1]:destroy()
+                warmup.units[i + 1] = nil
+            end
+        end
+        startRound()
+    end
+    onMessage = function(player, message)
+        local warmup = warmupByPlayer[player]
+        if warmup == nil or not player.isPlaying then
+            return
+        end
+        if message == seenMessage then
+            warmup.seenRound = round
+            checkProgress()
+        elseif (message == registeredMessage or message == failedMessage) and registeredRound == round then
+            warmup.registrationRound = round
+            warmup.registrationSucceeded = message == registeredMessage
+            checkRegistration()
+        end
+    end
+    onPlayerLeave = function()
+        checkProgress()
+        checkRegistration()
+    end
+    observeSelection = function()
+        if sentSeenRound == round or not (warmupByPlayer[Player["local"]] ~= nil) or not Player["local"].isPlaying then
+            return
+        end
+        local localSelectionCount = #Unit:getSelectionOf(Player["local"])
+        local maximumCount = getStageButtonCount()
+        local minimumCount = maximumCount == 12 and 2 or 13
+        if localSelectionCount >= minimumCount and localSelectionCount <= maximumCount then
+            socket:send(seenMessage)
+            sentSeenRound = round
+        end
+    end
+    for ____, player in ipairs(Player.all) do
+        if player.isUser and player.isPlaying then
+            local warmup = {
+                player = player,
+                units = {},
+                previousSelection = player == Player["local"] and Unit:getSelectionOf(player) or ({}),
+                seenRound = 0,
+                registrationRound = 0,
+                registrationSucceeded = false
+            }
+            warmups[#warmups + 1] = warmup
+            warmupByPlayer[player] = warmup
+            for i = 0, SELECTION_LAYOUT_BUTTON_COUNTS[1] - 1 do
+                local unit = assert(Unit:create(
+                    player,
+                    selectionWarmupUnitTypeId,
+                    0,
+                    0,
+                    0
+                ))
+                unit.isPaused = true
+                warmup.units[i + 1] = unit
+                dummyUnits[unit] = true
+            end
+        end
+    end
+    socket.onMessage:addListener(onMessage)
+    Player.onLeave:addListener(onPlayerLeave)
+    Timer.onPeriod[1 / 64]:addListener(observeSelection)
+    retryTimer = Timer:periodic(1, retrySelection)
+    startRound()
+    observeSelection()
+end
 warpack.afterMapInit(function()
-    rawset(
-        ____exports.LocalClient,
-        "isHD",
-        loadTOCFile(tocPath)
-    )
+    local isDE = loadTOCFile(deTocPath)
+    local isHD = not isDE and loadTOCFile(hdTocPath)
+    rawset(____exports.LocalClient, "isDE", isDE)
+    rawset(____exports.LocalClient, "isHD", isHD)
+    rawset(____exports.LocalClient, "isSD", not isDE and not isHD)
+    rawset(____exports.LocalClient, "graphicsMode", isDE and 2 or (isHD and 1 or 0))
 end)
+Timer:run(initializeSelectionFrames)
 return ____exports