using Microsoft.Xna.Framework; using System; using System.Collections.Generic; using System.Linq; using System.Reflection; namespace Barotrauma { [Flags] enum MapEntityCategory { Structure = 1, Machine = 2, Equipment = 4, Electrical = 8, Material = 16, Misc = 32, Alien = 64, ItemAssembly = 128, Legacy = 256 } partial class MapEntityPrefab { public readonly static List List = new List(); protected string name; protected string identifier; public Sprite sprite; //the position where the structure is being placed (needed when stretching the structure) protected static Vector2 placePosition; protected ConstructorInfo constructor; //is it possible to stretch the entity horizontally/vertically [Serialize(false, false)] public bool ResizeHorizontal { get; protected set; } [Serialize(false, false)] public bool ResizeVertical { get; protected set; } //which prefab has been selected for placing protected static MapEntityPrefab selected; public string Name { get { return name; } } //Used to differentiate between items when saving/loading //Allows changing the name of an item without breaking existing subs or having multiple items with the same name public string Identifier { get { return identifier; } } public HashSet Tags { get; protected set; } = new HashSet(); public static MapEntityPrefab Selected { get { return selected; } set { selected = value; } } [Serialize("", false)] public string Description { get; protected set; } [Serialize(false, false)] public bool Linkable { get; private set; } /// /// Links defined to identifiers. /// public List AllowedLinks { get; protected set; } = new List(); public MapEntityCategory Category { get; protected set; } [Serialize("1.0,1.0,1.0,1.0", false)] public Color SpriteColor { get; protected set; } [Serialize(1f, true), Editable(0.1f, 10f, DecimalCount = 3)] public float Scale { get; protected set; } //If a matching prefab is not found when loading a sub, the game will attempt to find a prefab with a matching alias. //(allows changing names while keeping backwards compatibility with older sub files) public string[] Aliases { get; protected set; } public static void Init() { MapEntityPrefab ep = new MapEntityPrefab { identifier = "hull", name = TextManager.Get("EntityName.hull"), Description = TextManager.Get("EntityDescription.hull"), constructor = typeof(Hull).GetConstructor(new Type[] { typeof(MapEntityPrefab), typeof(Rectangle) }), ResizeHorizontal = true, ResizeVertical = true, Linkable = true }; ep.AllowedLinks.Add("hull"); List.Add(ep); ep = new MapEntityPrefab { identifier = "gap", name = TextManager.Get("EntityName.gap"), Description = TextManager.Get("EntityDescription.gap"), constructor = typeof(Gap).GetConstructor(new Type[] { typeof(MapEntityPrefab), typeof(Rectangle) }), ResizeHorizontal = true, ResizeVertical = true }; List.Add(ep); ep = new MapEntityPrefab { identifier = "waypoint", name = TextManager.Get("EntityName.waypoint"), Description = TextManager.Get("EntityDescription.waypoint"), constructor = typeof(WayPoint).GetConstructor(new Type[] { typeof(MapEntityPrefab), typeof(Rectangle) }) }; List.Add(ep); ep = new MapEntityPrefab { identifier = "spawnpoint", name = TextManager.Get("EntityName.spawnpoint"), Description = TextManager.Get("EntityDescription.spawnpoint"), constructor = typeof(WayPoint).GetConstructor(new Type[] { typeof(MapEntityPrefab), typeof(Rectangle) }) }; List.Add(ep); } public MapEntityPrefab() { Category = MapEntityCategory.Structure; } public virtual void UpdatePlacing(Camera cam) { Vector2 placeSize = Submarine.GridSize; if (placePosition == Vector2.Zero) { Vector2 position = Submarine.MouseToWorldGrid(cam, Submarine.MainSub); if (PlayerInput.LeftButtonHeld()) placePosition = position; } else { Vector2 position = Submarine.MouseToWorldGrid(cam, Submarine.MainSub); if (ResizeHorizontal) placeSize.X = position.X - placePosition.X; if (ResizeVertical) placeSize.Y = placePosition.Y - position.Y; Rectangle newRect = Submarine.AbsRect(placePosition, placeSize); newRect.Width = (int)Math.Max(newRect.Width, Submarine.GridSize.X); newRect.Height = (int)Math.Max(newRect.Height, Submarine.GridSize.Y); if (Submarine.MainSub != null) { newRect.Location -= MathUtils.ToPoint(Submarine.MainSub.Position); } if (PlayerInput.LeftButtonReleased()) { CreateInstance(newRect); placePosition = Vector2.Zero; selected = null; } newRect.Y = -newRect.Y; } if (PlayerInput.RightButtonHeld()) { placePosition = Vector2.Zero; selected = null; } } protected virtual void CreateInstance(Rectangle rect) { object[] lobject = new object[] { this, rect }; constructor.Invoke(lobject); } public static bool SelectPrefab(object selection) { if ((selected = selection as MapEntityPrefab) != null) { placePosition = Vector2.Zero; return true; } else { return false; } } /// /// Find a matching map entity prefab /// /// The name of the item (can be omitted when searching based on identifier) /// The identifier of the item (if null, the identifier is ignored and the search is done only based on the name) public static MapEntityPrefab Find(string name, string identifier = null, bool showErrorMessages = true) { if (name != null) name = name.ToLowerInvariant(); foreach (MapEntityPrefab prefab in List) { if (identifier != null) { if (prefab.identifier != identifier) { continue; } else { if (string.IsNullOrEmpty(name)) return prefab; } } if (!string.IsNullOrEmpty(name)) { if (prefab.name.ToLowerInvariant() == name || (prefab.Aliases != null && prefab.Aliases.Any(a => a.ToLowerInvariant() == name))) return prefab; } } if (showErrorMessages) { DebugConsole.ThrowError("Failed to find a matching MapEntityPrefab (name: \"" + name + "\", identifier: \"" + identifier + "\").\n" + Environment.StackTrace); } return null; } /// /// Check if the name or any of the aliases of this prefab match the given name. /// public bool NameMatches(string name, bool caseSensitive = false) { if (caseSensitive) { return this.name == name || (Aliases != null && Aliases.Any(a => a == name)); } else { name = name.ToLowerInvariant(); return this.name.ToLowerInvariant() == name || (Aliases != null && Aliases.Any(a => a.ToLowerInvariant() == name)); } } public bool NameMatches(IEnumerable allowedNames, bool caseSensitive = false) { foreach (string name in allowedNames) { if (NameMatches(name, caseSensitive)) return true; } return false; } public bool IsLinkAllowed(MapEntityPrefab target) { if (target == null) { return false; } return AllowedLinks.Contains(target.Identifier) || target.AllowedLinks.Contains(identifier) || target.Tags.Any(t => AllowedLinks.Contains(t)) || Tags.Any(t => target.AllowedLinks.Contains(t)); } //a method that allows the GUIListBoxes to check through a delegate if the entityprefab is still selected public static object GetSelected() { return (object)selected; } } }