Add support for spirit balls effect.

2 files changed, 58 insertions, 2 deletions

diff --git a/src/being/being.cpp b/src/being/being.cpp
index 58fb9b2b7..ae0491372 100644
--- a/src/being/being.cpp
+++ b/src/being/being.cpp
@@ -200,6 +200,7 @@ Being::Being(const BeingId id,
     mHorseInfo(nullptr),
     mDownHorseSprites(),
     mUpHorseSprites(),
+    mSpiritParticles(),
 #endif
     mX(0),
     mY(0),
@@ -340,6 +341,9 @@ Being::~Being()
     mPets.clear();
 
     removeAllItemsParticles();
+#ifdef EATHENA_SUPPORT
+    mSpiritParticles.clear();
+#endif
 }
 
 void Being::createSpeechBubble()
@@ -4244,4 +4248,51 @@ void Being::setTrickDead(const bool b)
     }
 }
 
+void Being::setSpiritBalls(const unsigned int balls)
+{
+    if (!Particle::enabled)
+    {
+        mSpiritBalls = balls;
+        return;
+    }
+
+    if (balls > mSpiritBalls)
+    {
+        const int effectId = paths.getIntValue("spiritEffectId");
+        if (effectId != -1)
+            addSpiritBalls(balls - mSpiritBalls, effectId);
+    }
+    else if (balls < mSpiritBalls)
+    {
+        removeSpiritBalls(mSpiritBalls - balls);
+    }
+    mSpiritBalls = balls;
+}
+
+void Being::addSpiritBalls(const unsigned int balls,
+                           const int effectId)
+{
+    if (!effectManager)
+        return;
+    for (unsigned int f = 0; f < balls; f ++)
+    {
+        Particle *const particle = effectManager->triggerReturn(
+            effectId,
+            this);
+        mSpiritParticles.push_back(particle);
+    }
+}
+
+void Being::removeSpiritBalls(const unsigned int balls)
+{
+    if (!particleEngine)
+        return;
+    for (unsigned int f = 0; f < balls && !mSpiritParticles.empty(); f ++)
+    {
+        Particle *const particle = mSpiritParticles.back();
+        mChildParticleEffects.removeLocally(particle);
+        mSpiritParticles.pop_back();
+    }
+}
+
 #endif
diff --git a/src/being/being.h b/src/being/being.h
index 10e1b8482..bab1cc254 100644
--- a/src/being/being.h
+++ b/src/being/being.h
@@ -921,8 +921,7 @@ class Being notfinal : public ActorSprite,
         std::string getBuyBoard() const A_WARN_UNUSED
         { return mBuyBoard; }
 
-        void setSpiritBalls(const unsigned int balls)
-        { mSpiritBalls = balls; }
+        void setSpiritBalls(const unsigned int balls);
 
         unsigned int getSpiritBalls() const A_WARN_UNUSED
         { return mSpiritBalls; }
@@ -1081,6 +1080,11 @@ class Being notfinal : public ActorSprite,
                                        std::vector<int> &slotRemap,
                                        const int val);
 
+        void addSpiritBalls(const unsigned int balls,
+                            const int effectId);
+
+        void removeSpiritBalls(const unsigned int balls);
+
         void dumpSprites() const;
 
         const ActorTypeT mType;
@@ -1112,6 +1116,7 @@ class Being notfinal : public ActorSprite,
         HorseInfo *mHorseInfo;
         std::vector<AnimatedSprite*> mDownHorseSprites;
         std::vector<AnimatedSprite*> mUpHorseSprites;
+        std::vector<Particle*> mSpiritParticles;
 #endif
 
         int mX;             // position in tiles