Code:

do

local OldCommonLogic = CommonLogic
function CommonLogic()
   OldCommonLogic()
   
    local oldSecChoiceSetControlToType = controls.secondaryChoices.SetControlToType
    controls.secondaryChoices.SetControlToType = function(control, type)
      oldSecChoiceSetControlToType(control, type)
      
      if __blueprints[control.Data.id].Categories.CONSTRUCTION + categories.FACTORY + categories.ENGINEER then
         local ConstBar = '/mods/BattlePack/textures/ui/common/construction_bar.dds'
         if DiskGetFileInfo(UIUtil.UIFile('/mods/BattlePack/textures/ui/common/construction_bar.dds')) then
            control.ConstBar:SetTexture(UIUtil.UIFile('/mods/BattlePack/textures/ui/common/construction_bar.dds'))
            control.ConstBar.Height:Set(48)
            control.ConstBar.Width:Set(48)
         else
            control.ConstBar:SetSolidColor('ff00ff00')
         end
      else
         control.ConstBar:SetSolidColor('00000000')
      end       
    end
   
   local oldSecChoCreateElement = controls.secondaryChoices.CreateElement
    controls.secondaryChoices.CreateElement = function()
        local btn = oldSecChoCreateElement()
              
      btn.ConstBar = Bitmap(btn.Icon)
        btn.ConstBar:DisableHitTest()
        LayoutHelpers.AtCenterIn(btn.ConstBar, btn)
               
        return btn
    end
   
   local choiceCreateElement = controls.choices.CreateElement
    controls.choices.CreateElement = function()
        local btn = choiceCreateElement()
              
      btn.ConstBar = Bitmap(btn.Icon)
        btn.ConstBar:DisableHitTest()
        LayoutHelpers.AtCenterIn(btn.ConstBar, btn)
      
        return btn
    end
   
   local choiceSetControlToType = controls.choices.SetControlToType
    controls.choices.SetControlToType = function(control, type)
      
      if __blueprints[control.Data.id].Categories.CONSTRUCTION + categories.FACTORY + categories.ENGINEER then
      local ConstBar = '/mods/BattlePack/textures/ui/common/construction_bar.dds'
         if DiskGetFileInfo(UIUtil.UIFile('/mods/BattlePack/textures/ui/common/construction_bar.dds')) then
            control.ConstBar:SetTexture(UIUtil.UIFile('/mods/BattlePack/textures/ui/common/construction_bar.dds'))
            control.ConstBar.Height:Set(20)
            control.ConstBar.Width:Set(20)   
         else
            control.ConstBar:SetSolidColor('ff00ff00')
         end
      else
         control.ConstBar:SetSolidColor('00000000')
      end
   end
end


end

Statistics: Posted by brandonpotter0 — 31 Oct 2013, 01:10