local Event = require "utils.event"
local Scrap_table = require "maps.scrapyard.table"

local function balance(t)
  local p=#t local g=0 local c=0
  for k,v in pairs(t)do if(v.valid)then g=g+v.energy c=c+v.electric_buffer_size end end
  for k,v in pairs(t)do if(v.valid)then local r=(v.electric_buffer_size/c) v.energy=g*r end end
end

local function tick()
  local this = Scrap_table.get_table()
  if not this.energy then
    this.energy = {}
  end
  if not this.energy["scrapyard"] then
    this.energy["scrapyard"] = this.ow_energy
  end

  if not this.energy["loco"] then 
    this.energy["loco"] = this.lo_energy
  end

  local scrapyard = this.energy["scrapyard"]
  local loco = this.energy["loco"]
  if not scrapyard or not loco then return end
  if not scrapyard.valid or not loco.valid then return end
  balance(this.energy)
end

local function built_entity(event)
  local entity = event.created_entity
  if not entity.valid then return end
  local player = game.players[event.player_index]
  local surface = entity.surface
  if surface.name ~= "scrapyard" then return end
  if entity.name == "steam-engine" or entity.name == "steam-turbine" or entity.name == "lab" or entity.name == "rocket-silo" then 
    if not entity.valid then return end
    player.print("\""..entity.name.."\" Does not seem to work down here, thats strange!", {r = 1, g = 0, b = 0})
    entity.active = false
  end
end

Event.add(defines.events.on_tick, tick)
--Event.add(defines.events.on_built_entity, built_entity)