Apply trait-specific corruption when using a spell

src/main/java/com/minelittlepony/unicopia/ability/magic/spell/Spell.java

@@ -25,6 +25,11 @@ public interface Spell extends NbtSerialisable, Affine {
      */
     SpellType<?> getType();
 
+    /**
+     * Gets the traits of this spell.
+     */
+    SpellTraits getTraits();
+
     /**
      * The unique id of this particular spell instance.
      */

src/main/java/com/minelittlepony/unicopia/ability/magic/spell/effect/AbstractSpell.java

@@ -35,7 +35,7 @@ public abstract class AbstractSpell implements Spell {
         return type;
     }
 
-    protected SpellTraits getTraits() {
+    public final SpellTraits getTraits() {
         return traits == null ? SpellTraits.EMPTY : traits;
     }
 

src/main/java/com/minelittlepony/unicopia/ability/magic/spell/trait/SpellTraits.java

@@ -48,6 +48,10 @@ public final class SpellTraits implements Iterable<Map.Entry<Trait, Float>> {
         this(new EnumMap<>(from.traits));
     }
 
+    public float getCorruption() {
+        return (float)stream().filter(e -> e.getValue() != 0).mapToDouble(e -> e.getKey().getGroup().getCorruption()).sum();
+    }
+
     public SpellTraits multiply(float factor) {
         return factor == 0 ? EMPTY : map(v -> v * factor);
     }

src/main/java/com/minelittlepony/unicopia/entity/player/Pony.java

@@ -606,6 +606,7 @@ public class Pony extends Living<PlayerEntity> implements Transmittable, Copieab
             if (spell.getAffinity() == Affinity.BAD && entity.getWorld().random.nextInt(120) == 0) {
                 getCorruption().add(1);
             }
+            getCorruption().add((int)spell.getTraits().getCorruption());
         }
         setDirty();
     }