447 lines
19 KiB
C#
447 lines
19 KiB
C#
using Barotrauma.Networking;
|
|
using Microsoft.Xna.Framework;
|
|
using Microsoft.Xna.Framework.Graphics;
|
|
using Microsoft.Xna.Framework.Input;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Xml.Linq;
|
|
|
|
namespace Barotrauma.Items.Components
|
|
{
|
|
partial class Wire : ItemComponent, IDrawableComponent, IServerSerializable, IClientSerializable
|
|
{
|
|
public static Color higlightColor = Color.LightGreen;
|
|
public static Color editorHighlightColor = Color.Yellow;
|
|
public static Color editorSelectedColor = Color.Red;
|
|
|
|
partial class WireSection
|
|
{
|
|
public void Draw(SpriteBatch spriteBatch, Wire wire, Color color, Vector2 offset, float depth, float width = 0.3f)
|
|
{
|
|
spriteBatch.Draw(wire.wireSprite.Texture,
|
|
new Vector2(start.X + offset.X, -(start.Y + offset.Y)), null, color,
|
|
-angle,
|
|
new Vector2(0.0f, wire.wireSprite.size.Y / 2.0f),
|
|
new Vector2(length / wire.wireSprite.Texture.Width, width),
|
|
SpriteEffects.None,
|
|
depth);
|
|
}
|
|
|
|
public static void Draw(SpriteBatch spriteBatch, Wire wire, Vector2 start, Vector2 end, Color color, float depth, float width = 0.3f)
|
|
{
|
|
start.Y = -start.Y;
|
|
end.Y = -end.Y;
|
|
|
|
spriteBatch.Draw(wire.wireSprite.Texture,
|
|
start, null, color,
|
|
MathUtils.VectorToAngle(end - start),
|
|
new Vector2(0.0f, wire.wireSprite.size.Y / 2.0f),
|
|
new Vector2((Vector2.Distance(start, end)) / wire.wireSprite.Texture.Width, width),
|
|
SpriteEffects.None,
|
|
depth);
|
|
}
|
|
}
|
|
private static Sprite defaultWireSprite;
|
|
private Sprite overrideSprite;
|
|
private Sprite wireSprite;
|
|
|
|
private static Wire draggingWire;
|
|
private static int? selectedNodeIndex;
|
|
private static int? highlightedNodeIndex;
|
|
|
|
public Vector2 DrawSize
|
|
{
|
|
get { return sectionExtents; }
|
|
}
|
|
|
|
public static Wire DraggingWire
|
|
{
|
|
get => draggingWire;
|
|
}
|
|
|
|
partial void InitProjSpecific(XElement element)
|
|
{
|
|
if (defaultWireSprite == null)
|
|
{
|
|
defaultWireSprite = new Sprite("Content/Items/wireHorizontal.png", new Vector2(0.5f, 0.5f))
|
|
{
|
|
Depth = 0.85f
|
|
};
|
|
}
|
|
|
|
foreach (XElement subElement in element.Elements())
|
|
{
|
|
if (subElement.Name.ToString().Equals("wiresprite", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
overrideSprite = new Sprite(subElement);
|
|
break;
|
|
}
|
|
}
|
|
|
|
wireSprite = overrideSprite ?? defaultWireSprite;
|
|
}
|
|
|
|
public void Draw(SpriteBatch spriteBatch, bool editing, float itemDepth = -1)
|
|
{
|
|
if (sections.Count == 0 && !IsActive || Hidden)
|
|
{
|
|
Drawable = false;
|
|
return;
|
|
}
|
|
|
|
Vector2 drawOffset = Vector2.Zero;
|
|
Submarine sub = item.Submarine;
|
|
if (IsActive && sub == null) // currently being rewired, we need to get the sub from the connections in case the wire has been taken outside
|
|
{
|
|
if (connections[0] != null && connections[0].Item.Submarine != null) { sub = connections[0].Item.Submarine; }
|
|
if (connections[1] != null && connections[1].Item.Submarine != null) { sub = connections[1].Item.Submarine; }
|
|
}
|
|
|
|
if (sub != null)
|
|
{
|
|
drawOffset = sub.DrawPosition + sub.HiddenSubPosition;
|
|
}
|
|
|
|
float depth = item.IsSelected ? 0.0f : wireSprite.Depth + ((item.ID % 100) * 0.00001f);
|
|
|
|
if (item.IsHighlighted)
|
|
{
|
|
foreach (WireSection section in sections)
|
|
{
|
|
section.Draw(spriteBatch, this, Screen.Selected == GameMain.GameScreen ? higlightColor : editorHighlightColor, drawOffset, depth + 0.00001f, 0.7f);
|
|
}
|
|
}
|
|
else if (item.IsSelected)
|
|
{
|
|
foreach (WireSection section in sections)
|
|
{
|
|
section.Draw(spriteBatch, this, editorSelectedColor, drawOffset, depth + 0.00001f, 0.7f);
|
|
}
|
|
}
|
|
|
|
foreach (WireSection section in sections)
|
|
{
|
|
section.Draw(spriteBatch, this, item.Color, drawOffset, depth, 0.3f);
|
|
}
|
|
|
|
if (nodes.Count > 0)
|
|
{
|
|
if (!IsActive)
|
|
{
|
|
if (connections[0] == null) { DrawHangingWire(spriteBatch, nodes[0] + drawOffset, depth); }
|
|
if (connections[1] == null) { DrawHangingWire(spriteBatch, nodes.Last() + drawOffset, depth); }
|
|
}
|
|
if (IsActive && item.ParentInventory?.Owner is Character user && user == Character.Controlled)// && Vector2.Distance(newNodePos, nodes[nodes.Count - 1]) > nodeDistance)
|
|
{
|
|
if (user.CanInteract)
|
|
{
|
|
Vector2 gridPos = Character.Controlled.Position;
|
|
Vector2 roundedGridPos = new Vector2(
|
|
MathUtils.RoundTowardsClosest(Character.Controlled.Position.X, Submarine.GridSize.X),
|
|
MathUtils.RoundTowardsClosest(Character.Controlled.Position.Y, Submarine.GridSize.Y));
|
|
//Vector2 attachPos = GetAttachPosition(user);
|
|
|
|
if (item.Submarine == null)
|
|
{
|
|
Structure attachTarget = Structure.GetAttachTarget(item.WorldPosition);
|
|
if (attachTarget != null)
|
|
{
|
|
if (attachTarget.Submarine != null)
|
|
{
|
|
//set to submarine-relative position
|
|
gridPos += attachTarget.Submarine.Position;
|
|
roundedGridPos += attachTarget.Submarine.Position;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
gridPos += item.Submarine.Position;
|
|
roundedGridPos += item.Submarine.Position;
|
|
}
|
|
|
|
Submarine.DrawGrid(spriteBatch, 14, gridPos, roundedGridPos, alpha: 0.7f);
|
|
|
|
WireSection.Draw(
|
|
spriteBatch, this,
|
|
new Vector2(nodes[nodes.Count - 1].X, nodes[nodes.Count - 1].Y) + drawOffset,
|
|
new Vector2(newNodePos.X, newNodePos.Y) + drawOffset,
|
|
item.Color, 0.0f, 0.3f);
|
|
|
|
WireSection.Draw(
|
|
spriteBatch, this,
|
|
new Vector2(newNodePos.X, newNodePos.Y) + drawOffset,
|
|
item.DrawPosition,
|
|
item.Color, itemDepth, 0.3f);
|
|
|
|
GUI.DrawRectangle(spriteBatch, new Vector2(newNodePos.X + drawOffset.X, -(newNodePos.Y + drawOffset.Y)) - Vector2.One * 3, Vector2.One * 6, item.Color);
|
|
}
|
|
else
|
|
{
|
|
WireSection.Draw(
|
|
spriteBatch, this,
|
|
new Vector2(nodes[nodes.Count - 1].X, nodes[nodes.Count - 1].Y) + drawOffset,
|
|
item.DrawPosition,
|
|
item.Color, 0.0f, 0.3f);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!editing || !GameMain.SubEditorScreen.WiringMode) { return; }
|
|
|
|
for (int i = 0; i < nodes.Count; i++)
|
|
{
|
|
Vector2 drawPos = nodes[i];
|
|
if (item.Submarine != null) drawPos += item.Submarine.Position + item.Submarine.HiddenSubPosition;
|
|
drawPos.Y = -drawPos.Y;
|
|
|
|
if ((highlightedNodeIndex == i && item.IsHighlighted) || (selectedNodeIndex == i && item.IsSelected))
|
|
{
|
|
GUI.DrawRectangle(spriteBatch, drawPos + new Vector2(-10, -10), new Vector2(20, 20), editorHighlightColor, false, 0.0f);
|
|
}
|
|
|
|
if (item.IsSelected)
|
|
{
|
|
GUI.DrawRectangle(spriteBatch, drawPos + new Vector2(-5, -5), new Vector2(10, 10), item.Color, true, 0.0f);
|
|
|
|
}
|
|
else
|
|
{
|
|
GUI.DrawRectangle(spriteBatch, drawPos + new Vector2(-3, -3), new Vector2(6, 6), item.Color, true, 0.0f);
|
|
}
|
|
}
|
|
}
|
|
|
|
private void DrawHangingWire(SpriteBatch spriteBatch, Vector2 start, float depth)
|
|
{
|
|
float angle = (float)Math.Sin(GameMain.GameScreen.GameTime * 2.0f + item.ID) * 0.2f;
|
|
Vector2 endPos = start + new Vector2((float)Math.Sin(angle), -(float)Math.Cos(angle)) * 50.0f;
|
|
|
|
WireSection.Draw(
|
|
spriteBatch, this,
|
|
start, endPos,
|
|
GUI.Style.Orange, depth + 0.00001f, 0.2f);
|
|
|
|
WireSection.Draw(
|
|
spriteBatch, this,
|
|
start, start + (endPos - start) * 0.7f,
|
|
item.Color, depth, 0.3f);
|
|
}
|
|
|
|
|
|
public static void UpdateEditing(List<Wire> wires)
|
|
{
|
|
Wire equippedWire =
|
|
Character.Controlled?.SelectedItems[0]?.GetComponent<Wire>() ??
|
|
Character.Controlled?.SelectedItems[1]?.GetComponent<Wire>();
|
|
if (equippedWire != null)
|
|
{
|
|
if (PlayerInput.PrimaryMouseButtonClicked() && Character.Controlled.SelectedConstruction == null)
|
|
{
|
|
equippedWire.Use(1.0f, Character.Controlled);
|
|
}
|
|
return;
|
|
}
|
|
|
|
//dragging a node of some wire
|
|
if (draggingWire != null)
|
|
{
|
|
if (Character.Controlled != null)
|
|
{
|
|
Character.Controlled.FocusedItem = null;
|
|
Character.Controlled.ResetInteract = true;
|
|
Character.Controlled.ClearInputs();
|
|
}
|
|
//cancel dragging
|
|
if (!PlayerInput.PrimaryMouseButtonHeld())
|
|
{
|
|
draggingWire = null;
|
|
selectedNodeIndex = null;
|
|
}
|
|
//update dragging
|
|
else
|
|
{
|
|
MapEntity.DisableSelect = true;
|
|
|
|
Submarine sub = draggingWire.item.Submarine;
|
|
if (draggingWire.connections[0] != null && draggingWire.connections[0].Item.Submarine != null) sub = draggingWire.connections[0].Item.Submarine;
|
|
if (draggingWire.connections[1] != null && draggingWire.connections[1].Item.Submarine != null) sub = draggingWire.connections[1].Item.Submarine;
|
|
|
|
Vector2 nodeWorldPos = GameMain.SubEditorScreen.Cam.ScreenToWorld(PlayerInput.MousePosition);
|
|
if (sub != null)
|
|
{
|
|
nodeWorldPos = nodeWorldPos - sub.HiddenSubPosition - sub.Position;
|
|
}
|
|
|
|
if (selectedNodeIndex.HasValue)
|
|
{
|
|
nodeWorldPos.X = MathUtils.Round(nodeWorldPos.X, Submarine.GridSize.X / 2.0f);
|
|
nodeWorldPos.Y = MathUtils.Round(nodeWorldPos.Y, Submarine.GridSize.Y / 2.0f);
|
|
|
|
draggingWire.nodes[(int)selectedNodeIndex] = nodeWorldPos;
|
|
draggingWire.UpdateSections();
|
|
}
|
|
else
|
|
{
|
|
if (Vector2.DistanceSquared(nodeWorldPos, draggingWire.nodes[(int)highlightedNodeIndex]) > Submarine.GridSize.X * Submarine.GridSize.X)
|
|
{
|
|
selectedNodeIndex = highlightedNodeIndex;
|
|
}
|
|
}
|
|
|
|
|
|
MapEntity.SelectEntity(draggingWire.item);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
//a wire has been selected -> check if we should start dragging one of the nodes
|
|
float nodeSelectDist = 10, sectionSelectDist = 5;
|
|
highlightedNodeIndex = null;
|
|
if (MapEntity.SelectedList.Count == 1 && MapEntity.SelectedList[0] is Item)
|
|
{
|
|
Wire selectedWire = ((Item)MapEntity.SelectedList[0]).GetComponent<Wire>();
|
|
|
|
if (selectedWire != null)
|
|
{
|
|
Vector2 mousePos = GameMain.SubEditorScreen.Cam.ScreenToWorld(PlayerInput.MousePosition);
|
|
if (selectedWire.item.Submarine != null) mousePos -= (selectedWire.item.Submarine.Position + selectedWire.item.Submarine.HiddenSubPosition);
|
|
|
|
//left click while holding ctrl -> check if the cursor is on a wire section,
|
|
//and add a new node if it is
|
|
if (PlayerInput.KeyDown(Keys.RightControl) || PlayerInput.KeyDown(Keys.LeftControl))
|
|
{
|
|
if (PlayerInput.PrimaryMouseButtonClicked())
|
|
{
|
|
if (Character.Controlled != null)
|
|
{
|
|
Character.Controlled.ResetInteract = true;
|
|
Character.Controlled.ClearInputs();
|
|
}
|
|
int closestSectionIndex = selectedWire.GetClosestSectionIndex(mousePos, sectionSelectDist, out _);
|
|
if (closestSectionIndex > -1)
|
|
{
|
|
selectedWire.nodes.Insert(closestSectionIndex + 1, mousePos);
|
|
selectedWire.UpdateSections();
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
//check if close enough to a node
|
|
int closestIndex = selectedWire.GetClosestNodeIndex(mousePos, nodeSelectDist, out _);
|
|
if (closestIndex > -1)
|
|
{
|
|
highlightedNodeIndex = closestIndex;
|
|
//start dragging the node
|
|
if (PlayerInput.PrimaryMouseButtonHeld())
|
|
{
|
|
if (Character.Controlled != null)
|
|
{
|
|
Character.Controlled.ResetInteract = true;
|
|
Character.Controlled.ClearInputs();
|
|
}
|
|
draggingWire = selectedWire;
|
|
//selectedNodeIndex = closestIndex;
|
|
return;
|
|
}
|
|
//remove the node
|
|
else if (PlayerInput.SecondaryMouseButtonClicked() && closestIndex > 0 && closestIndex < selectedWire.nodes.Count - 1)
|
|
{
|
|
selectedWire.nodes.RemoveAt(closestIndex);
|
|
selectedWire.UpdateSections();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
Wire highlighted = null;
|
|
|
|
//check which wire is highlighted with the cursor
|
|
if (GUI.MouseOn == null)
|
|
{
|
|
float closestDist = float.PositiveInfinity;
|
|
foreach (Wire w in wires)
|
|
{
|
|
Vector2 mousePos = GameMain.SubEditorScreen.Cam.ScreenToWorld(PlayerInput.MousePosition);
|
|
if (w.item.Submarine != null) { mousePos -= (w.item.Submarine.Position + w.item.Submarine.HiddenSubPosition); }
|
|
|
|
int highlightedNode = w.GetClosestNodeIndex(mousePos, highlighted == null ? nodeSelectDist : closestDist, out float dist);
|
|
if (highlightedNode > -1)
|
|
{
|
|
if (dist < closestDist)
|
|
{
|
|
highlightedNodeIndex = highlightedNode;
|
|
highlighted = w;
|
|
closestDist = dist;
|
|
}
|
|
}
|
|
|
|
if (w.GetClosestSectionIndex(mousePos, highlighted == null ? sectionSelectDist : closestDist, out dist) > -1)
|
|
{
|
|
//prefer nodes over sections
|
|
if (dist + nodeSelectDist * 0.5f < closestDist)
|
|
{
|
|
highlightedNodeIndex = null;
|
|
highlighted = w;
|
|
closestDist = dist + nodeSelectDist * 0.5f;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (highlighted != null)
|
|
{
|
|
highlighted.item.IsHighlighted = true;
|
|
if (PlayerInput.PrimaryMouseButtonClicked())
|
|
{
|
|
MapEntity.DisableSelect = true;
|
|
MapEntity.SelectEntity(highlighted.item);
|
|
}
|
|
}
|
|
}
|
|
|
|
public void ClientRead(ServerNetObject type, IReadMessage msg, float sendingTime)
|
|
{
|
|
int eventIndex = msg.ReadRangedInteger(0, (int)Math.Ceiling(MaxNodeCount / (float)MaxNodesPerNetworkEvent));
|
|
int nodeCount = msg.ReadRangedInteger(0, MaxNodesPerNetworkEvent);
|
|
int nodeStartIndex = eventIndex * MaxNodesPerNetworkEvent;
|
|
|
|
Vector2[] nodePositions = new Vector2[nodeStartIndex + nodeCount];
|
|
for (int i = 0; i < nodes.Count && i < nodePositions.Length; i++)
|
|
{
|
|
nodePositions[i] = nodes[i];
|
|
}
|
|
|
|
for (int i = 0; i < nodeCount; i++)
|
|
{
|
|
nodePositions[nodeStartIndex + i] = new Vector2(msg.ReadSingle(), msg.ReadSingle());
|
|
}
|
|
|
|
if (nodePositions.Any(n => !MathUtils.IsValid(n)))
|
|
{
|
|
nodes.Clear();
|
|
return;
|
|
}
|
|
|
|
nodes = nodePositions.ToList();
|
|
UpdateSections();
|
|
Drawable = nodes.Any();
|
|
}
|
|
|
|
public void ClientWrite(IWriteMessage msg, object[] extraData = null)
|
|
{
|
|
int nodeCount = (int)extraData[2];
|
|
msg.Write((byte)nodeCount);
|
|
if (nodeCount > 0)
|
|
{
|
|
msg.Write(nodes.Last().X);
|
|
msg.Write(nodes.Last().Y);
|
|
}
|
|
}
|
|
}
|
|
}
|