your pets should receive oxygen too

wiidotmom · Sep 6, 2024 · 0bf595c · 0bf595c
1 parent 047d02e
commit 0bf595c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/main/java/dev/igalaxy/marspopsicle/MarsPopsicle.java b/src/main/java/dev/igalaxy/marspopsicle/MarsPopsicle.java
@@ -14,6 +14,8 @@
 import org.apache.logging.log4j.Logger;
 import net.minecraftforge.fml.common.Mod;
 
+import static com.momosoftworks.coldsweat.common.capability.handler.EntityTempManager.TEMPERATURE_ENABLED_ENTITIES;
+
 @Mod(MarsPopsicle.MODID)
 public class MarsPopsicle {
 
@@ -34,10 +36,9 @@ public static void onModifiersRegister(TempModifierRegisterEvent event)
   @SubscribeEvent
   public static void defaultModifiersInit(GatherDefaultTempModifiersEvent event)
   {
-    if (event.getEntity() instanceof Player && event.getTrait() == Temperature.Trait.WORLD)
+    if ((event.getEntity() instanceof Player || TEMPERATURE_ENABLED_ENTITIES.contains(event.getEntity().getType())) && event.getTrait() == Temperature.Trait.WORLD)
     {
-      LOGGER.info("Initialized oxygen temp modifier for " + ((Player) event.getEntity()).getName().getString());
-      event.addModifier(new OxygenTempModifier().tickRate(60), Placement.Duplicates.BY_CLASS, Placement.of(Placement.Mode.AFTER, Placement.Order.LAST, (mod2) -> mod2 instanceof BiomeTempModifier));
+      event.addModifier(new OxygenTempModifier().tickRate(40), Placement.Duplicates.BY_CLASS, Placement.of(Placement.Mode.AFTER, Placement.Order.LAST, (mod) -> mod instanceof BiomeTempModifier));
     }
   }
 }
diff --git a/src/main/java/dev/igalaxy/marspopsicle/tempmodifiers/OxygenTempModifier.java b/src/main/java/dev/igalaxy/marspopsicle/tempmodifiers/OxygenTempModifier.java
@@ -3,8 +3,6 @@
 import com.momosoftworks.coldsweat.api.temperature.modifier.TempModifier;
 import com.momosoftworks.coldsweat.api.util.Temperature;
 import com.momosoftworks.coldsweat.api.util.Temperature.Units;
-import com.petrolpark.destroy.capability.level.pollution.LevelPollution.PollutionType;
-import com.petrolpark.destroy.util.PollutionHelper;
 import earth.terrarium.adastra.api.systems.OxygenApi;
 import earth.terrarium.adastra.common.constants.PlanetConstants;
 import net.minecraft.world.entity.LivingEntity;