Implemented #1093

core/src/mindustry/world/blocks/power/ThermalGenerator.java

@@ -13,11 +13,19 @@ import static mindustry.Vars.renderer;
 
 public class ThermalGenerator extends PowerGenerator{
     public Effect generateEffect = Fx.none;
+    public Attribute attribute = Attribute.heat;
 
     public ThermalGenerator(String name){
         super(name);
     }
 
+    @Override
+    public void setStats(){
+        super.setStats();
+
+        stats.add(BlockStat.tiles, attribute);
+    }
+
     @Override
     public void update(Tile tile){
         GeneratorEntity entity = tile.ent();
@@ -29,7 +37,7 @@ public class ThermalGenerator extends PowerGenerator{
 
     @Override
     public void drawPlace(int x, int y, int rotation, boolean valid){
-        drawPlaceText(Core.bundle.formatFloat("bar.efficiency", sumAttribute(Attribute.heat, x, y) * 100, 1), x, y, valid);
+        drawPlaceText(Core.bundle.formatFloat("bar.efficiency", sumAttribute(attribute, x, y) * 100, 1), x, y, valid);
     }
 
     @Override
@@ -43,7 +51,7 @@ public class ThermalGenerator extends PowerGenerator{
         super.onProximityAdded(tile);
 
         GeneratorEntity entity = tile.ent();
-        entity.productionEfficiency = sumAttribute(Attribute.heat, tile.x, tile.y);
+        entity.productionEfficiency = sumAttribute(attribute, tile.x, tile.y);
     }
 
     @Override
@@ -56,6 +64,6 @@ public class ThermalGenerator extends PowerGenerator{
     @Override
     public boolean canPlaceOn(Tile tile){
         //make sure there's heat at this location
-        return tile.getLinkedTilesAs(this, tempTiles).sumf(other -> other.floor().attributes.get(Attribute.heat)) > 0.01f;
+        return tile.getLinkedTilesAs(this, tempTiles).sumf(other -> other.floor().attributes.get(attribute)) > 0.01f;
     }
 }

core/src/mindustry/world/blocks/production/Cultivator.java

@@ -1,17 +1,16 @@
 package mindustry.world.blocks.production;
 
-import arc.Core;
-import arc.graphics.Color;
+import arc.*;
+import arc.graphics.*;
 import arc.graphics.g2d.*;
-import arc.math.Mathf;
-import arc.math.Rand;
-import arc.util.Time;
-import mindustry.content.Fx;
-import mindustry.entities.type.TileEntity;
-import mindustry.graphics.Pal;
-import mindustry.ui.Bar;
-import mindustry.world.Tile;
-import mindustry.world.meta.Attribute;
+import arc.math.*;
+import arc.util.*;
+import mindustry.content.*;
+import mindustry.entities.type.*;
+import mindustry.graphics.*;
+import mindustry.ui.*;
+import mindustry.world.*;
+import mindustry.world.meta.*;
 
 import java.io.*;
 
@@ -57,6 +56,13 @@ public class Cultivator extends GenericCrafter{
         () -> ((CultivatorEntity)entity).warmup));
     }
 
+    @Override
+    public void setStats(){
+        super.setStats();
+
+        stats.add(BlockStat.affinities, attribute);
+    }
+
     @Override
     public void drawPlace(int x, int y, int rotation, boolean valid){
         drawPlaceText(Core.bundle.formatFloat("bar.efficiency", (1 + sumAttribute(attribute, x, y)) * 100, 1), x, y, valid);

core/src/mindustry/world/blocks/production/SolidPump.java

@@ -64,6 +64,7 @@ public class SolidPump extends Pump{
 
         stats.remove(BlockStat.output);
         stats.add(BlockStat.output, result, 60f * pumpAmount, true);
+        stats.add(BlockStat.affinities, attribute);
     }
 
     @Override