Reorganizing

media/lua/client/TOC/Handlers/ModDataHandler.lua

@@ -0,0 +1,179 @@
+local StaticData = require("TOC/StaticData")
+
+----------------
+---@alias partData { isCut : boolean?, isInfected : boolean?, isOperated : boolean?, isCicatrized : boolean?, isCauterized : boolean?, isVisible : boolean?, cicatrizationTime : number }
+---@alias tocModData {Hand_L : partData, ForeArm_L : partData, UpperArm_L : partData, Hand_R : partData, ForeArm_R : partData, UpperArm_R : partData, isIgnoredPartInfected : boolean, isAnyLimbCut : boolean}
+----------------
+-- TODO This class should handle all the stuff related to the mod data
+
+---@class ModDataHandler
+---@field playerObj IsoPlayer
+---@field tocData tocModData 
+local ModDataHandler = {}
+
+---@param playerObj IsoPlayer
+---@return ModDataHandler
+function ModDataHandler:new(playerObj)
+    local o = {}
+    setmetatable(o, self)
+    self.__index = self
+
+    o.playerObj = playerObj
+    o.tocData = playerObj:getModData()[StaticData.MOD_NAME]
+
+    ModDataHandler.instance = o
+
+    return o
+end
+
+---Setup a newly instanced ModDataHandler
+---@param force boolean?
+function ModDataHandler:setup(force)
+    local tocData = self.playerObj:getModData()[StaticData.MOD_NAME]
+    if force or tocData == nil or tocData.Hand_L == nil or tocData.Hand_L.isCut == nil then
+        self:createData()
+    end
+    -- TODO Check compatibility or do we just skip it at this point?
+end
+
+function ModDataHandler:createData()
+    print("TOC: createData")
+
+    local modData = self.playerObj:getModData()
+    modData[StaticData.MOD_NAME] = {
+
+        -- Generic stuff that does not belong anywhere else
+        isIgnoredPartInfected = false,
+        isAnyLimbCut = false
+    }
+
+    -- Set a reference to TOC data in ModData
+    self.tocData = self.playerObj:getModData()[StaticData.MOD_NAME]
+
+    ---@type partData
+    local defaultParams = {isCut = false, isInfected = false, isOperated = false, isCicatrized = false, isCauterized = false, isVisible = false}
+
+
+    -- Initialize limbs
+    for i=1, #StaticData.LIMBS_STRINGS do
+        local limbName = StaticData.LIMBS_STRINGS[i]
+        modData[StaticData.MOD_NAME][limbName] = {}
+        self:setLimbParams(StaticData.LIMBS_STRINGS[i], defaultParams, 0)
+    end
+end
+
+-----------------
+--* Setters *--
+
+---Set a generic boolean that toggles varies function of the mod
+---@param isAnyLimbCut boolean
+function ModDataHandler:setIsAnyLimbCut(isAnyLimbCut)
+    self.tocData.isAnyLimbCut = true
+end
+
+---Set isCut 
+---@param limbName string
+---@param isCut boolean
+function ModDataHandler:setIsCut(limbName, isCut)
+    self.tocData[limbName].isCut = isCut
+end
+
+---Set isInfected
+---@param limbName string
+---@param isInfected boolean
+function ModDataHandler:setIsInfected(limbName, isInfected)
+    self.tocData[limbName].isInfected = isInfected
+end
+
+---Set isIgnoredPartInfected
+---@param isIgnoredPartInfected boolean
+function ModDataHandler:setIsIgnoredPartInfected(isIgnoredPartInfected)
+    self.tocData.isIgnoredPartInfected = isIgnoredPartInfected
+end
+
+-----------------
+--* Getters *--
+
+---Set a generic boolean that toggles varies function of the mod
+---@return boolean
+function ModDataHandler:getIsAnyLimbCut()
+    return self.tocData.isAnyLimbCut
+end
+
+---Get isCut
+---@param limbName string
+---@return boolean
+function ModDataHandler:getIsCut(limbName)
+    return self.tocData[limbName].isCut
+end
+
+---Get isIgnoredPartInfected
+---@return boolean
+function ModDataHandler:getIsIgnoredPartInfected()
+    return self.tocData.isIgnoredPartInfected
+end
+
+---Get isVisible
+---@return boolean
+function ModDataHandler:getIsVisible(limbName)
+    return self.tocData[limbName].isVisible
+end
+
+--* Limbs data handling *--
+
+---Set a limb and its dependend limbs as cut
+---@param limbName string
+---@param isOperated boolean
+---@param isCicatrized boolean
+---@param isCauterized boolean
+---@param surgeonFactor number?
+function ModDataHandler:setCutLimb(limbName, isOperated, isCicatrized, isCauterized, surgeonFactor)
+    local cicatrizationTime = 0
+    if isCicatrized == false or isCauterized == false then
+        cicatrizationTime = StaticData.LIMBS_CICATRIZATION_TIME[limbName] - surgeonFactor
+    end
+
+    ---@type partData
+    local params = {isCut = true, isInfected = false, isOperated = isOperated, isCicatrized = isCicatrized, isCauterized = isCauterized, isVisible = true}
+    self:setLimbParams(limbName, params, cicatrizationTime)
+
+    for i=1, #StaticData.LIMBS_DEPENDENCIES[limbName] do
+        local dependedLimbName = StaticData.LIMBS_DEPENDENCIES[limbName][i]
+
+        -- We don't care about isOperated, isCicatrized, isCauterized since this is depending on another limb
+        -- Same story for cicatrizationTime, which will be 0
+        self:setLimbParams(dependedLimbName, {isCut = true, isInfected = false, isVisible = false}, 0)
+    end
+
+    -- Set that a limb has been cut, to activate some functions without having to loop through the parts
+    self:setIsAnyLimbCut(true)
+
+end
+
+---Internal use only, set a limb data
+---@param limbName string
+---@param ampStatus partData {isCut, isInfected, isOperated, isCicatrized, isCauterized, isVisible}
+---@param cicatrizationTime integer?
+---@private
+function ModDataHandler:setLimbParams(limbName, ampStatus, cicatrizationTime)
+    local limbData = self.tocData[limbName]
+    if ampStatus.isCut ~= nil then limbData.isCut = ampStatus.isCut end
+    if ampStatus.isInfected ~= nil then limbData.isInfected = ampStatus.isInfected end
+    if ampStatus.isOperated ~= nil then limbData.isOperated = ampStatus.isOperated end
+    if ampStatus.isCicatrized ~= nil then limbData.isCicatrized = ampStatus.isCicatrized end
+    if ampStatus.isCauterized ~= nil then limbData.isCauterized = ampStatus.isCauterized end
+    if ampStatus.isVisible ~= nil then limbData.isVisible = ampStatus.isVisible end
+
+    if cicatrizationTime ~= nil then limbData.cicatrizationTime = cicatrizationTime end
+end
+
+---@return ModDataHandler
+function ModDataHandler.GetInstance()
+    if ModDataHandler.instance ~= nil then
+        return ModDataHandler.instance
+    else
+        return ModDataHandler:new(getPlayer())
+    end
+end
+
+return ModDataHandler