using System; using System.Xml.Linq; namespace Barotrauma { class SetPriceMultiplierAction : EventAction { public enum OperationType { Set, Multiply, Min, Max } public enum PriceMultiplierType { Store, Mechanical } [Serialize(1.0f, IsPropertySaveable.Yes)] public float Multiplier { get; set; } [Serialize(OperationType.Set, IsPropertySaveable.Yes)] public OperationType Operation { get; set; } [Serialize(PriceMultiplierType.Store, IsPropertySaveable.Yes)] public PriceMultiplierType TargetMultiplier { get; set; } public SetPriceMultiplierAction(ScriptedEvent parentEvent, ContentXElement element) : base(parentEvent, element) { } private bool isFinished = false; public override bool IsFinished(ref string goTo) { return isFinished; } public override void Reset() { isFinished = false; } public override void Update(float deltaTime) { if (isFinished) { return; } if (GameMain.GameSession?.GameMode is CampaignMode campaign && campaign.Map?.CurrentLocation != null) { float newMultiplier = GetCurrentMultiplier(campaign.Map.CurrentLocation); switch (Operation) { case OperationType.Set: newMultiplier = Multiplier; break; case OperationType.Multiply: newMultiplier *= Multiplier; break; case OperationType.Min: newMultiplier = Math.Min(Multiplier, campaign.Map.CurrentLocation.PriceMultiplier); break; case OperationType.Max: newMultiplier = Math.Max(Multiplier, campaign.Map.CurrentLocation.PriceMultiplier); break; default: throw new NotImplementedException(); } SetCurrentMultiplier(campaign.Map.CurrentLocation, newMultiplier); } isFinished = true; } private float GetCurrentMultiplier(Location location) { return TargetMultiplier switch { PriceMultiplierType.Store => location.PriceMultiplier, PriceMultiplierType.Mechanical => location.MechanicalPriceMultiplier, _ => throw new NotImplementedException() }; } private void SetCurrentMultiplier(Location location, float value) { switch (TargetMultiplier) { case PriceMultiplierType.Store: location.PriceMultiplier = value; break; case PriceMultiplierType.Mechanical: location.MechanicalPriceMultiplier = value; break; default: throw new NotImplementedException(); } } public override string ToDebugString() { return $"{ToolBox.GetDebugSymbol(isFinished)} {nameof(SetPriceMultiplierAction)} -> (Multiplier: {Multiplier.ColorizeObject()}, " + $"Operation: {Operation.ColorizeObject()}, Target: {TargetMultiplier})"; } } }