Setup for 1.14.4-Fabric

[BoundingBoxOutlineReloaded.git] / src / main / java / com / irtimaled / bbor / client / providers / MobSpawnerProvider.java

diff --git a/src/main/java/com/irtimaled/bbor/client/providers/MobSpawnerProvider.java b/src/main/java/com/irtimaled/bbor/client/providers/MobSpawnerProvider.java
index 655b335cf06c8163d82bfde176b2ba5d9121b0e1..5079aa048524b643c7d586e5de3541ef651081dc 100644 (file)
--- a/src/main/java/com/irtimaled/bbor/client/providers/MobSpawnerProvider.java
+++ b/src/main/java/com/irtimaled/bbor/client/providers/MobSpawnerProvider.java
@@ -5,17 +5,18 @@ import com.irtimaled.bbor.client.interop.TileEntitiesHelper;
 import com.irtimaled.bbor.client.models.BoundingBoxMobSpawner;
 import com.irtimaled.bbor.common.BoundingBoxType;
 import com.irtimaled.bbor.common.models.Coords;
-import net.minecraft.tileentity.TileEntityMobSpawner;
+import com.irtimaled.bbor.common.models.DimensionId;
+import net.minecraft.block.entity.MobSpawnerBlockEntity;
 
 public class MobSpawnerProvider implements IBoundingBoxProvider<BoundingBoxMobSpawner> {
     @Override
-    public boolean canProvide(int dimensionId) {
+    public boolean canProvide(DimensionId dimensionId) {
         return BoundingBoxTypeHelper.shouldRender(BoundingBoxType.MobSpawner);
     }
 
     @Override
-    public Iterable<BoundingBoxMobSpawner> get(int dimensionId) {
-        return TileEntitiesHelper.map(TileEntityMobSpawner.class, spawner -> {
+    public Iterable<BoundingBoxMobSpawner> get(DimensionId dimensionId) {
+        return TileEntitiesHelper.map(MobSpawnerBlockEntity.class, spawner -> {
             Coords coords = new Coords(spawner.getPos());
             return BoundingBoxMobSpawner.from(coords);
         });