Metabolism refactor (#4395)

* metabolism -> respirator, add reageanteffect and reagenteffectcondition, start on metabolizercomp/system * move LiverBehavior metabolism logic to Metabolizer * minor tweaks and update all YAML * how about actually taking conditions into account * off by one * removals * reviews
2021-07-31 04:50:32 -07:00
parent dc18997bf8
commit 8aa4f998de
36 changed files with 572 additions and 558 deletions
--- a/Content.Server/Body/Metabolism/MetabolizerSystem.cs
+++ b/Content.Server/Body/Metabolism/MetabolizerSystem.cs
@@ -0,0 +1,107 @@
+using System.Collections.Generic;
+using System.Linq;
+using Content.Server.Body.Circulatory;
+using Content.Server.Chemistry.Components;
+using Content.Shared.Body.Components;
+using Content.Shared.Body.Mechanism;
+using Content.Shared.Chemistry.Reagent;
+using Content.Shared.Chemistry.Solution;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Prototypes;
+
+namespace Content.Server.Body.Metabolism
+{
+    // TODO mirror in the future working on mechanisms move updating here to BodySystem so it can be ordered?
+    public class MetabolizerSystem : EntitySystem
+    {
+        public override void Update(float frameTime)
+        {
+            base.Update(frameTime);
+
+            foreach (var metab in ComponentManager.EntityQuery<MetabolizerComponent>(false))
+            {
+                metab.AccumulatedFrametime += frameTime;
+
+                // Only update as frequently as it should
+                if (metab.AccumulatedFrametime >= metab.UpdateFrequency)
+                {
+                    metab.AccumulatedFrametime = 0.0f;
+                    TryMetabolize(metab);
+                }
+            }
+        }
+
+        private void TryMetabolize(MetabolizerComponent comp)
+        {
+            var owner = comp.Owner;
+            var reagentList = new List<Solution.ReagentQuantity>();
+            SolutionContainerComponent? solution = null;
+            SharedBodyComponent? body = null;
+
+            // if this field is passed we should try and take from the bloodstream over anything else
+            if (comp.TakeFromBloodstream && owner.TryGetComponent<SharedMechanismComponent>(out var mech))
+            {
+                body = mech.Body;
+                if (body != null)
+                {
+                    if (body.Owner.TryGetComponent<BloodstreamComponent>(out var bloodstream)
+                        && bloodstream.Solution.CurrentVolume >= ReagentUnit.Zero)
+                    {
+                        solution = bloodstream.Solution;
+                        reagentList = bloodstream.Solution.ReagentList.ToList();
+                    }
+                }
+            }
+            else if (owner.TryGetComponent<SolutionContainerComponent>(out var sol))
+            {
+                // if we have no mechanism/body but a solution container instead,
+                // we'll just use that to metabolize from
+                solution = sol;
+                reagentList = sol.ReagentList.ToList();
+            }
+            if (solution == null || reagentList.Count == 0)
+            {
+                // We're all outta ideas on where to metabolize from
+                return;
+            }
+
+            // Run metabolism for each reagent, remove metabolized reagents
+            foreach (var reagent in reagentList)
+            {
+                if (!comp.Metabolisms.ContainsKey(reagent.ReagentId))
+                    continue;
+
+                var metabolism = comp.Metabolisms[reagent.ReagentId];
+                // Run metabolism code for each reagent
+                foreach (var effect in metabolism.Effects)
+                {
+                    var ent = body != null ? body.Owner : owner;
+                    var conditionsMet = true;
+                    if (effect.Conditions != null)
+                    {
+                        // yes this is 3 nested for loops, but all of these lists are
+                        // basically guaranteed to be small or empty
+                        foreach (var condition in effect.Conditions)
+                        {
+                            if (!condition.Condition(ent, reagent))
+                            {
+                                conditionsMet = false;
+                                break;
+                            }
+                        }
+                    }
+
+                    if (!conditionsMet)
+                        return;
+
+                    // If we're part of a body, pass that entity to Metabolize
+                    // Otherwise, just pass our owner entity, maybe we're a plant or something
+                    effect.Metabolize(ent, reagent);
+                }
+
+                solution.TryRemoveReagent(reagent.ReagentId, metabolism.MetabolismRate);
+            }
+        }
+    }
+}