Store duration in NBT

iopleke · Nov 5, 2014 · 18b25d7 · 18b25d7
1 parent 031ef53
commit 18b25d7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/java/alchemyplusplus/block/complex/diffuser/DiffuserTileEntity.java b/src/main/java/alchemyplusplus/block/complex/diffuser/DiffuserTileEntity.java
@@ -213,8 +213,9 @@ public void writeToNBT(NBTTagCompound nbt)
 			int count = 0;
 			while (potionEffects.hasNext())
 			{
-				int potionID = ((PotionEffect) potionEffects.next()).getPotionID();
-				nbt.setInteger("effect" + count, potionID);
+				PotionEffect effect = (PotionEffect) potionEffects.next();
+				nbt.setInteger("effect" + count, effect.getPotionID());
+				nbt.setInteger("duration" + count, effect.getDuration());
 				count++;
 			}
 		}
@@ -247,7 +248,8 @@ public void readFromNBT(NBTTagCompound nbt)
 			while (count >= 0)
 			{
 				int potionID = nbt.getInteger("effect" + count);
-				PotionEffect effect = new PotionEffect(potionID, 40, 0);
+				int duration = nbt.getInteger("duration" + count);
+				PotionEffect effect = new PotionEffect(potionID, duration, 0);
 
 				this.fluidTank.potionEffects.add(effect);
 				count--;