Reorganizing

media/lua/client/TOC/UI/CutLimbInteractions.lua

@@ -0,0 +1,115 @@
+local BaseHandler = require("UI/TOC_HealthPanelBaseHandler")
+local CutLimbAction = require("TimedActions/TOC_CutLimbAction")
+local StaticData = require("TOC/StaticData")
+local ModDataHandler = require("TOC/Handlers/ModDataHandler")
+
+---------------------
+
+
+---Check if the item name corresponds to a compatible saw
+---@param itemName string
+local function CheckIfSaw(itemName)
+    return itemName == "Saw" or itemName == "GardenSaw" or itemName == "Chainsaw"
+end
+
+---Add the action to the queue
+---@param limbName string
+---@param surgeon IsoPlayer
+---@param patient IsoPlayer
+local function PerformAction(limbName, surgeon, patient)
+    ISTimedActionQueue.add(CutLimbAction:new(surgeon, patient, limbName))
+end
+
+---Adds the actions to the inventory context menu
+---@param surgeonNum number
+---@param context ISContextMenu
+local function AddInventoryAmputationOptions(surgeonNum, context)
+    local surgeonObj = getSpecificPlayer(surgeonNum)
+    local option = context:addOption(getText("ContextMenu_Amputate"), nil)
+    local subMenu = context:getNew(context)
+    context:addSubMenu(option, subMenu)
+    for i=1, #StaticData.LIMBS_STRINGS do
+        local limbName = StaticData.LIMBS_STRINGS[i]
+        if not ModDataHandler.GetInstance():getIsCut(limbName) then
+            local limbTranslatedName = getText("ContextMenu_Limb_" .. limbName)
+            subMenu:addOption(limbTranslatedName, limbName, PerformAction, surgeonObj, surgeonObj)    -- TODO Should be patient, not surgeon
+        end
+    end
+end
+
+---Handler for OnFillInventoryObjectContextMenu
+---@param player number
+---@param context ISUIElement
+---@param items table
+local function AddInventoryAmputationMenu(player, context, items)
+    local item = items[1]       -- Selected item
+    if CheckIfSaw(item.name) then
+        AddInventoryAmputationOptions(player, context)
+    end
+end
+
+
+Events.OnFillInventoryObjectContextMenu.Add(AddInventoryAmputationMenu)
+
+-------------------------------------
+
+---@class CutLimbHandler : BaseHandler
+local CutLimbHandler = BaseHandler:derive("CutLimbHandler")
+
+
+---Creates new CutLimbHandler
+---@param panel any
+---@param bodyPart any
+---@return CutLimbHandler
+function CutLimbHandler:new(panel, bodyPart)
+    local o = BaseHandler.new(self, panel, bodyPart)
+    o.items.ITEMS = {}
+    return o
+end
+
+function CutLimbHandler:checkItem(item)
+    local itemType = item:getType()
+    if CheckIfSaw(itemType) then
+        self:addItem(self.items.ITEMS, item)
+    end
+end
+
+function CutLimbHandler:addToMenu(context)
+    --local types = self:getAllItemTypes(self.items.ITEMS)
+    --if #types > 0 then
+    local option = context:addOption(getText("ContextMenu_Amputate"), nil)
+    local subMenu = context:getNew(context)
+    context:addSubMenu(option, subMenu)
+    for i=1, #StaticData.LIMBS_STRINGS do
+        local limbName = StaticData.LIMBS_STRINGS[i]
+        if not ModDataHandler.GetInstance():getIsCut(limbName) then
+            local limbTranslatedName = getText("ContextMenu_Limb_" .. limbName)
+            subMenu:addOption(limbTranslatedName, self.onMenuOptionSelected, nil)
+        end
+    end
+    --end
+end
+
+function CutLimbHandler:dropItems(items)
+    local types = self:getAllItemTypes(items)
+    if #self.items.ITEMS > 0 and #types == 1 then
+        self:onMenuOptionSelected(types[1])
+        return true
+    end
+    return false
+end
+
+function CutLimbHandler:isValid(itemType)
+    return self:getItemOfType(self.items.ITEMS, itemType)
+end
+
+function CutLimbHandler:perform(previousAction, itemType)
+    print("perform cutlimbhandler")
+    local item = self:getItemOfType(self.items.ITEMS, itemType)
+    previousAction = self:toPlayerInventory(item, previousAction)
+
+    local action = CutLimbAction:new(self:getPatient(), self:getDoctor(), self.bodyPart)
+    ISTimedActionQueue.addAfter(previousAction, action)
+end
+
+return CutLimbHandler

media/lua/client/TOC/UI/HealthPanel.lua

@@ -0,0 +1,96 @@
+local PlayerHandler = require("TOC/Handlers/PlayerHandler")
+local StaticData = require("TOC/StaticData")
+local CommonMethods = require("TOC/CommonMethods")
+
+---@diagnostic disable: duplicate-set-field
+local CutLimbHandler = require("UI/TOC_CutLimbInteractions")
+
+---------------------------------
+
+-- We're overriding ISHealthPanel to add custom textures to the body panel.
+-- By doing so we can show the player which limbs have been cut without having to use another menu
+-- We can show prosthesis too this way
+-- We also manage the drag'n drop of items on the body to let the players use the saw this way too
+
+ISHealthBodyPartPanel = ISBodyPartPanel:derive("ISHealthBodyPartPanel")
+
+--* Handling drag n drop of the saw *--
+
+local og_ISHealthPanel_dropItemsOnBodyPart = ISHealthPanel.dropItemsOnBodyPart
+function ISHealthPanel:dropItemsOnBodyPart(bodyPart, items)
+    og_ISHealthPanel_dropItemsOnBodyPart(self, bodyPart, items)
+    local cutLimbHandler = CutLimbHandler:new(self, bodyPart)
+    for _,item in ipairs(items) do
+        cutLimbHandler:checkItem(item)
+    end
+    if cutLimbHandler:dropItems(items) then
+        return
+    end
+
+end
+
+local og_ISHealthPanel_doBodyPartContextMenu = ISHealthPanel.doBodyPartContextMenu
+function ISHealthPanel:doBodyPartContextMenu(bodyPart, x, y)
+    og_ISHealthPanel_doBodyPartContextMenu(self, bodyPart, x, y)
+    local playerNum = self.otherPlayer and self.otherPlayer:getPlayerNum() or self.character:getPlayerNum()
+
+    -- To not recreate it but reuse the one that has been created in the original method
+    local context = getPlayerContextMenu(playerNum) 
+    local cutLimbHandler = CutLimbHandler:new(self, bodyPart)
+    cutLimbHandler:addToMenu(context)
+end
+
+
+--* Modification to handle visible amputation on the health menu *--
+
+function ISHealthPanel.GetHighestAmputation()
+    ISHealthPanel.highestAmputations = {}
+    for i=1, #PlayerHandler.amputatedLimbs do
+        local limbName = PlayerHandler.amputatedLimbs[i]
+        local index = CommonMethods.GetSide(limbName)
+        if PlayerHandler.modDataHandler:getIsCut(limbName) and PlayerHandler.modDataHandler:getIsVisible(limbName) then
+            ISHealthPanel.highestAmputations[index] = limbName
+        end
+    end
+end
+
+
+local og_ISHealthPanel_initialise = ISHealthPanel.initialise
+function ISHealthPanel:initialise()
+    if self.character:isFemale() then
+        self.sexPl = "Female"
+    else
+        self.sexPl = "Male"
+    end
+    og_ISHealthPanel_initialise(self)
+end
+
+local og_ISHealthPanel_render = ISHealthPanel.render
+function ISHealthPanel:render()
+    og_ISHealthPanel_render(self)
+
+    -- TODO Handle another player health panel
+
+    if ISHealthPanel.highestAmputations then
+        -- Left Texture
+        if ISHealthPanel.highestAmputations["L"] then
+            local textureL = StaticData.HEALTH_PANEL_TEXTURES[self.sexPl][ISHealthPanel.highestAmputations["L"]]
+            self:drawTexture(textureL, self.healthPanel.x/2 - 2, self.healthPanel.y/2, 1, 1, 0, 0)
+        end
+
+        -- Right Texture
+        if ISHealthPanel.highestAmputations["R"] then
+            local textureR = StaticData.HEALTH_PANEL_TEXTURES[self.sexPl][ISHealthPanel.highestAmputations["R"]]
+            self:drawTexture(textureR, self.healthPanel.x/2 + 2, self.healthPanel.y/2, 1, 1, 0, 0)
+        end
+    else
+        ISHealthPanel.GetHighestAmputation()
+    end
+end
+
+-- We need to override this to force the alpha to 1
+local og_ISCharacterInfoWindow_render = ISCharacterInfoWindow.prerender
+function ISCharacterInfoWindow:prerender()
+    og_ISCharacterInfoWindow_render(self)
+    self.backgroundColor.a = 1
+end

media/lua/client/TOC/UI/HealthPanelBaseHandler.lua

@@ -0,0 +1,132 @@
+-- Had to copy and paste this stuff from the base game since this is a local only class. Kinda shit, but eh
+
+---@class BaseHandler : ISBaseObject
+---@field panel ISUIElement
+---@field bodyPart BodyPart
+---@field items table
+local BaseHandler = ISBaseObject:derive("BaseHandler")
+
+function BaseHandler:new(panel, bodyPart)
+    local o = {}
+    setmetatable(o, self)
+    self.__index = self
+    o.panel = panel
+    o.bodyPart = bodyPart
+    o.items = {}
+    return o
+end
+
+function BaseHandler:isInjured()
+    local bodyPart = self.bodyPart
+    return (bodyPart:HasInjury() or bodyPart:stitched() or bodyPart:getSplintFactor() > 0) and not bodyPart:bandaged()
+end
+
+function BaseHandler:checkItems()
+    for k,v in pairs(self.items) do
+        table.wipe(v)
+    end
+
+    local containers = ISInventoryPaneContextMenu.getContainers(self:getDoctor())
+    local done = {}
+    local childContainers = {}
+    for i=1,containers:size() do
+        local container = containers:get(i-1)
+        done[container] = true
+        table.wipe(childContainers)
+        self:checkContainerItems(container, childContainers)
+        for _,container2 in ipairs(childContainers) do
+            if not done[container2] then
+                done[container2] = true
+                self:checkContainerItems(container2, nil)
+            end
+        end
+    end
+end
+
+function BaseHandler:checkContainerItems(container, childContainers)
+    local containerItems = container:getItems()
+    for i=1,containerItems:size() do
+        local item = containerItems:get(i-1)
+        if item:IsInventoryContainer() then
+            if childContainers then
+                table.insert(childContainers, item:getInventory())
+            end
+        else
+            self:checkItem(item)
+        end
+    end
+end
+
+function BaseHandler:dropItems(items)
+    return false
+end
+
+function BaseHandler:addItem(items, item)
+    table.insert(items, item)
+end
+
+function BaseHandler:getAllItemTypes(items)
+    local done = {}
+    local types = {}
+    for _,item in ipairs(items) do
+        if not done[item:getFullType()] then
+            table.insert(types, item:getFullType())
+            done[item:getFullType()] = true
+        end
+    end
+    return types
+end
+
+function BaseHandler:getItemOfType(items, type)
+    for _,item in ipairs(items) do
+        if item:getFullType() == type then
+            return item
+        end
+    end
+    return nil
+end
+
+function BaseHandler:getItemOfTag(items, type)
+    for _,item in ipairs(items) do
+        if item:hasTag(type) then
+            return item
+        end
+    end
+    return nil
+end
+
+function BaseHandler:getAllItemsOfType(items, type)
+    local items = {}
+    for _,item in ipairs(items) do
+        if item:getFullType() == type then
+            table.insert(items, item)
+        end
+    end
+    return items
+end
+
+function BaseHandler:onMenuOptionSelected(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8)
+    ISTimedActionQueue.add(HealthPanelAction:new(self:getDoctor(), self, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8))
+end
+
+function BaseHandler:toPlayerInventory(item, previousAction)
+    if item:getContainer() ~= self:getDoctor():getInventory() then
+        local action = ISInventoryTransferAction:new(self:getDoctor(), item, item:getContainer(), self:getDoctor():getInventory())
+        ISTimedActionQueue.addAfter(previousAction, action)
+        -- FIXME: ISHealthPanel.actions never gets cleared
+        self.panel.actions = self.panel.actions or {}
+        self.panel.actions[action] = self.bodyPart
+        return action
+    end
+    return previousAction
+end
+
+function BaseHandler:getDoctor()
+    return self.panel.otherPlayer or self.panel.character
+end
+
+function BaseHandler:getPatient()
+    return self.panel.character
+end
+
+return BaseHandler

media/lua/client/TOC/UI/ProsthesisBuilderUI.lua

@@ -0,0 +1 @@
+-- TODO Separate UI to craft prosthesis... No just use the crafting menu you mook

media/lua/client/TOC/UI/SurgeryInteractions.lua

@@ -0,0 +1,67 @@
+local PlayerHandler = require("TOC/Handlers/PlayerHandler")
+local ModDataHandler = require("TOC/Handlers/ModDataHandler")
+
+---------------
+
+-- TODO Surgery Kits
+
+local function AddInventorySurgeryMenu(playerNum, context, items)
+
+end
+
+Events.OnFillInventoryObjectContextMenu.Add(AddInventorySurgeryMenu)
+
+
+-- TODO Oven
+
+-- TODO We need a class to handle operations, this is just a placeholder
+local function Cauterize(limbName)
+    
+end
+
+---comment
+---@param playerNum any
+---@param context ISContextMenu
+---@param worldObjects any
+---@param test any
+local function AddOvenContextMenu(playerNum, context, worldObjects, test)
+    local pl = getSpecificPlayer(playerNum)
+
+    if not ModDataHandler.GetInstance():getIsAnyLimbCut() then return end
+    local amputatedLimbs = PlayerHandler.GetAmputatedLimbs()
+
+    local stoveObj = nil
+    for _, obj in pairs(worldObjects) do
+        if instanceof(obj, "IsoStove") then
+            stoveObj = obj
+            break
+        end
+    end
+    if stoveObj == nil then return end
+    if pl:HasTrait("Brave") or pl:getPerkLevel(Perks.Strength) > 5 then
+        local isTempLow = stoveObj:getCurrentTemperature() < 250
+        local tempTooltip = ISToolTip:new()
+        tempTooltip:initialise()
+        tempTooltip:setName("ContextMenu_Cauterize_TempTooLow_tooltip")
+        tempTooltip.description = getText("Tooltip_Surgery_TempTooLow")
+        tempTooltip:setVisible(false)
+
+        local optionMain = context:addOption(getText("ContextMenu_Cauterize"), nil)
+        local subMenu = context:getNew(context)
+        context:addSubMenu(optionMain, subMenu)
+        for i=1, #amputatedLimbs do
+            local limbName = amputatedLimbs[i]
+            local option = subMenu:addOption(getText("ContextMenu_Limb_" .. limbName), limbName, Cauterize)
+            option.notAvailable = isTempLow
+            if isTempLow then
+                option.toolTip = tempTooltip
+            end
+        end
+    end
+
+end
+
+Events.OnFillWorldObjectContextMenu.Add(AddOvenContextMenu)
+
+
+-- TODO Other stuff?