LuaCsForBarotraumaEP/Subsurface/Items/Components/OxygenGenerator.cs

using System;
using System.Collections.Generic;
using System.Xml.Linq;

namespace Subsurface.Items.Components
{
    class OxygenGenerator : Powered
    {
        PropertyTask powerUpTask;

        bool running;

        List<Vent> ventList;

        public bool IsRunning()
        {
            return running && item.Condition>0.0f;
        }

        public OxygenGenerator(Item item, XElement element)
            : base(item, element)
        {
            isActive = true;

            ventList = new List<Vent>();

            item.linkedTo.CollectionChanged += new System.Collections.Specialized.NotifyCollectionChangedEventHandler(
                delegate(object sender, System.Collections.Specialized.NotifyCollectionChangedEventArgs e)
                { GetVents(); }
            );
        }

        public override void Update(float deltaTime, Camera cam)
        {
            base.Update(deltaTime, cam);

            currPowerConsumption = powerConsumption;

            if (item.currentHull == null) return;

            if (voltage < minVoltage)
            {
                running = false;
                if (powerUpTask==null || powerUpTask.IsFinished)
                {
                    powerUpTask = new PropertyTask(Game1.gameSession.taskManager, item, IsRunning, 30.0f, "Turn on the oxygen generator");
                }
                return;                
            }

            running = true;
            
            float deltaOxygen = Math.Min(voltage, 1.0f) * 50000.0f;
            item.currentHull.Oxygen += deltaOxygen * deltaTime;

            UpdateVents(deltaOxygen);

            voltage = 0.0f;
        }

        private void GetVents()
        {            
            foreach (MapEntity entity in item.linkedTo)
            {
                Item linkedItem = entity as Item;
                if (linkedItem == null) continue;

                Vent vent = linkedItem.GetComponent<Vent>();
                if (vent != null) ventList.Add(vent);                
            }
        }
                
        private void UpdateVents(float deltaOxygen)
        {
            if (ventList.Count == 0) return;

            deltaOxygen = deltaOxygen / ventList.Count;
            foreach (Vent v in ventList)
            {
                v.OxygenFlow = deltaOxygen;
                v.IsActive = true;
            }
        }
    }
}