Record more statistics

[liqimai]

I notice that there is an env.state.stats variable to record some game statistics. This is exactly what we RL trainers need. During these days' training, I really want to visualize some statistics about the gameplay, but most statistics are not easy to get outside env.step, because env.step is the actual place where the game logic runs. Here are some statistics I suggest recording.

Suggested statistics

Resource production statistics

• ore dug so far by light/heavy
• ice dug so far by light/heavy
• water/metal refined so far
• power generated so far by
  • factory
  • light robots
  • heavy robots

Resource consumption statistics

• water spent on
  • watering
  • keeping factories alive
• metal spent on
  • building light robots
  • building heavy robots
• power spent on each action type, including
  • move/dig/self_destruct
  • updating action queue
  • building light/heavy robots

Action statistics

• the successfully executed number of each action type
• for dig
  • rubble removed so far by light/heavy
  • own/opponent lichen dug so far by light/heavy
• for pickup
  • the total amount of each type of resource picked by light/heavy
• for transfer
  • for each resource type
    • amount transferred from light/heavy
    • amount transferred to light/heavy/factory
• for move
  • average rubble in the target tile of light/heavy
  • the number of collisions with own/opponent robots
  • survival light/heavy in collisions
  • destroyed light/heavy in collisions

To be continued

Other proposals are welcome. Please leave a comment if you want other statistics.

[StoneT2000]

https://github.com/Lux-AI-Challenge/Lux-Design-S2/tree/main/luxai_s2/luxai_s2/state/stats.py some of the ones we record now @liqimai

[liqimai]

Thanks. This really helps, especially with dense rewards design!

[Nashgari]

I would like to add there is a place for the power consumption of the units / factories.
But only factories are recording the power consumption.
Env.py contains the code where the units uses power
line 503 unit.power -= self.state.env_cfg.ROBOTS[unit.unit_type.name].DIG_COST
but it is not recorded in the stats in the same way as a factory
line 568: self.state.stats[factory.team.agent]["consumption"]["power"][
"FACTORY"
] += spent_power