All entity types

[darkfrei]

Hi all!

Is here easy way to get if some prototype an entity?
Now I use this code:

code

Code: Select all

local item_types = {}
local item_types_list = {}
local entity_types = {}
local entity_types_list = {}

function is_value_in_list (value, list)
  for i, v in pairs (list) do
    if value == v then
      return true
    end
  end
  return false
end

for type_name, type_prot in pairs (data.raw) do
  for element_name, element_prot in pairs (type_prot) do
    if element_prot.minable 
    or element_prot.collision_box 
    or element_prot.selection_box 
    then
      if not entity_types[type_name] then
        entity_types[type_name] = {element_name}
      else
        if not is_value_in_list (element_name, entity_types[type_name]) then
          table.insert(entity_types[type_name], element_name)
        end
      end
      
      if not is_value_in_list (type_name, entity_types_list) then
        table.insert(entity_types_list, type_name)
      end
    end
    
    if element_prot.name and element_prot.type == "item" 
    or element_prot.stack_size
    or element_prot.place_result
    or element_prot.place_as_tile
    then
      if not item_types[type_name] then
        item_types[type_name] = {element_name}
      else
        if not is_value_in_list (element_name, item_types[type_name]) then
          table.insert(item_types[type_name], element_name)
        end
      end
      
      if not is_value_in_list (type_name, item_types_list) then
        table.insert(item_types_list, type_name)
      end
    end
  end
end

log ('item_types = '.. serpent.block(item_types))
log ('item_types_list = '.. serpent.block(item_types_list))
log ('entity_types = '.. serpent.block(entity_types))
log ('entity_types_list = '.. serpent.block(entity_types_list))

and this code gives me [0.16.36 with mods]:

Code: Select all

entity_types_list = {
  "character-corpse",
  "player",
  "furnace",
  "transport-belt",
  "fish",
  "boiler",
  "container",
  "electric-pole",
  "generator",
  "offshore-pump",
  "inserter",
  "item-entity",
  "pipe",
  "radar",
  "lamp",
  "arrow",
  "pipe-to-ground",
  "assembling-machine",
  "corpse",
  "entity-ghost",
  "tile-ghost",
  "deconstructible-tile-proxy",
  "item-request-proxy",
  "cliff",
  "wall",
  "mining-drill",
  "projectile",
  "resource",
  "turret",
  "ammo-turret",
  "unit",
  "unit-spawner",
  "rail-remnants",
  "simple-entity",
  "tree",
  "tile",
  "optimized-decorative",
  "underground-belt",
  "loader",
  "splitter",
  "car",
  "solar-panel",
  "locomotive",
  "cargo-wagon",
  "fluid-wagon",
  "artillery-wagon",
  "gate",
  "player-port",
  "straight-rail",
  "curved-rail",
  "land-mine",
  "train-stop",
  "rail-signal",
  "rail-chain-signal",
  "lab",
  "logistic-robot",
  "construction-robot",
  "logistic-container",
  "rocket-silo",
  "rocket-silo-rocket",
  "rocket-silo-rocket-shadow",
  "roboport",
  "storage-tank",
  "pump",
  "market",
  "accumulator",
  "beacon",
  "combat-robot",
  "arithmetic-combinator",
  "decider-combinator",
  "constant-combinator",
  "power-switch",
  "programmable-speaker",
  "electric-energy-interface",
  "reactor",
  "heat-pipe",
  "simple-entity-with-force",
  "simple-entity-with-owner",
  "infinity-container",
  "artillery-turret",
  "electric-turret",
  "fluid-turret"
}

and

Code: Select all

item_types_list = {
  "ammo",
  "armor",
  "gun",
  "item",
  "capsule",
  "repair-tool",
  "mining-tool",
  "item-with-entity-data",
  "rail-planner",
  "tool",
  "blueprint",
  "deconstruction-item",
  "blueprint-book",
  "selection-tool",
  "item-with-tags",
  "item-with-label",
  "item-with-inventory",
  "module"
}

Is it correct? Do we have some easy way to do that?