Adjust active particle effect for factories

DarkShyMW · Jul 11, 2023 · fce28a4 · fce28a4
1 parent 669802d
commit fce28a4
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/main/java/mekanism/common/block/attribute/AttributeParticleFX.java b/src/main/java/mekanism/common/block/attribute/AttributeParticleFX.java
@@ -15,6 +15,14 @@ public List<Function<RandomSource, Particle>> getParticleFunctions() {
         return particleFunctions;
     }
 
+    public AttributeParticleFX addDense(ParticleOptions type, int density, Function<RandomSource, Pos3D> posSupplier) {
+        Function<RandomSource, Particle> particleFunction = random -> new Particle(type, posSupplier.apply(random));
+        for (int i = 0; i < density; i++) {
+            particleFunctions.add(particleFunction);
+        }
+        return this;
+    }
+
     public AttributeParticleFX add(ParticleOptions type, Function<RandomSource, Pos3D> posSupplier) {
         particleFunctions.add(random -> new Particle(type, posSupplier.apply(random)));
         return this;

diff --git a/src/main/java/mekanism/common/content/blocktype/Factory.java b/src/main/java/mekanism/common/content/blocktype/Factory.java
@@ -5,19 +5,22 @@
 import mekanism.common.block.attribute.AttributeEnergy;
 import mekanism.common.block.attribute.AttributeFactoryType;
 import mekanism.common.block.attribute.AttributeGui;
+import mekanism.common.block.attribute.AttributeParticleFX;
 import mekanism.common.block.attribute.AttributeSound;
 import mekanism.common.block.attribute.AttributeTier;
 import mekanism.common.block.attribute.AttributeUpgradeSupport;
 import mekanism.common.block.attribute.AttributeUpgradeable;
 import mekanism.common.content.blocktype.Machine.FactoryMachine;
 import mekanism.common.inventory.container.MekanismContainer;
+import mekanism.common.lib.math.Pos3D;
 import mekanism.common.registration.impl.ContainerTypeRegistryObject;
 import mekanism.common.registration.impl.TileEntityTypeRegistryObject;
 import mekanism.common.registries.MekanismBlocks;
 import mekanism.common.registries.MekanismContainerTypes;
 import mekanism.common.tier.FactoryTier;
 import mekanism.common.tile.factory.TileEntityFactory;
 import mekanism.common.util.EnumUtils;
+import net.minecraft.core.particles.ParticleTypes;
 
 public class Factory<TILE extends TileEntityFactory<?>> extends FactoryMachine<TILE> {
 
@@ -58,6 +61,11 @@ protected FactoryBuilder(FACTORY holder) {
             // assign the value here, and then return the builder itself as it is the same object
             builder.withComputerSupport(tier, type.getRegistryNameComponentCapitalized() + "Factory");
             builder.withCustomShape(BlockShapes.getShape(tier, type));
+            builder.replace(new AttributeParticleFX().addDense(ParticleTypes.SMOKE, 5, rand -> new Pos3D(
+                  rand.nextFloat() * 0.7F - 0.3F,
+                  rand.nextFloat() * 0.1F + 0.7F,
+                  rand.nextFloat() * 0.7F - 0.3F
+            )));
             return builder;
         }
     }