2022-05-22 18:48:23 -07:00

564 lines
24 KiB
C#

using NewHorizons.Components;
using NewHorizons.External.Modules;
using NewHorizons.External.Modules.VariableSize;
using NewHorizons.Handlers;
using NewHorizons.Utility;
using System;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;
using UnityEngine.UI;
using Logger = NewHorizons.Utility.Logger;
using Object = UnityEngine.Object;
namespace NewHorizons.Builder.ShipLog
{
public static class MapModeBuilder
{
#region General
public static ShipLogAstroObject[][] ConstructMapMode(string systemName, GameObject transformParent,
ShipLogAstroObject[][] currentNav, int layer)
{
var greyScaleMaterial = GameObject.Find(ShipLogHandler.PAN_ROOT_PATH + "/TimberHearth/Sprite")
.GetComponent<Image>().material;
var bodies = Main.BodyDict[systemName].Where(
b => !(b.Config.ShipLog?.mapMode?.remove ?? false) && !b.Config.isQuantumState
).ToList();
var flagManualPositionUsed = systemName == "SolarSystem";
var flagAutoPositionUsed = false;
foreach (var body in bodies.Where(b => ShipLogHandler.IsVanillaBody(b) == false))
{
if (body.Config.ShipLog == null) continue;
if (body.Config.ShipLog?.mapMode?.manualPosition == null)
{
flagAutoPositionUsed = true;
}
else
{
flagManualPositionUsed = true;
if (body.Config.ShipLog?.mapMode?.manualNavigationPosition == null)
{
Logger.LogError("Navigation position is missing for: " + body.Config.name);
return null;
}
}
}
if (flagManualPositionUsed)
{
if (flagAutoPositionUsed && flagManualPositionUsed)
Logger.LogWarning(
"Can't mix manual and automatic layout of ship log map mode, defaulting to manual");
return ConstructMapModeManual(bodies, transformParent, greyScaleMaterial, currentNav, layer);
}
if (flagAutoPositionUsed)
{
return ConstructMapModeAuto(bodies, transformParent, greyScaleMaterial, layer);
}
return null;
}
public static string GetAstroBodyShipLogName(string id) =>
TranslationHandler.GetTranslation(ShipLogHandler.GetNameFromAstroID(id) ?? id,
TranslationHandler.TextType.UI);
private static GameObject CreateImage(GameObject nodeGO, Texture2D texture, string name, int layer)
{
var newImageGO = new GameObject(name);
newImageGO.layer = layer;
newImageGO.transform.SetParent(nodeGO.transform);
var transform = newImageGO.AddComponent<RectTransform>();
transform.localPosition = Vector3.zero;
transform.localRotation = Quaternion.identity;
transform.localScale = Vector3.one;
var newImage = newImageGO.AddComponent<Image>();
var rect = new Rect(0, 0, texture.width, texture.height);
var pivot = new Vector2(texture.width / 2, texture.height / 2);
newImage.sprite = Sprite.Create(texture, rect, pivot);
return newImageGO;
}
private static GameObject CreateMapModeGameObject(NewHorizonsBody body, GameObject parent, int layer,
Vector2 position)
{
var newGameObject = new GameObject(body.Config.name + "_ShipLog");
newGameObject.layer = layer;
newGameObject.transform.SetParent(parent.transform);
var transform = newGameObject.AddComponent<RectTransform>();
var scale = body.Config.ShipLog?.mapMode?.scale ?? 1f;
transform.localPosition = position;
transform.localRotation = Quaternion.identity;
transform.localScale = Vector3.one * scale;
transform.SetAsFirstSibling();
return newGameObject;
}
private static ShipLogAstroObject AddShipLogAstroObject(GameObject gameObject, NewHorizonsBody body,
Material greyScaleMaterial, int layer)
{
const float unviewedIconOffset = 15;
var unviewedReference =
SearchUtilities.CachedFind(ShipLogHandler.PAN_ROOT_PATH + "/TimberHearth/UnviewedIcon");
var astroObject = gameObject.AddComponent<ShipLogAstroObject>();
astroObject._id = ShipLogHandler.GetAstroObjectId(body);
Texture2D image = null;
Texture2D outline = null;
var imagePath = body.Config.ShipLog?.mapMode?.revealedSprite;
var outlinePath = body.Config.ShipLog?.mapMode?.outlineSprite;
if (imagePath != null) image = ImageUtilities.GetTexture(body.Mod, imagePath);
if (image == null) image = AutoGenerateMapModePicture(body);
if (outlinePath != null) outline = ImageUtilities.GetTexture(body.Mod, outlinePath);
if (outline == null) outline = ImageUtilities.MakeOutline(image, Color.white, 10);
astroObject._imageObj = CreateImage(gameObject, image, body.Config.name + " Revealed", layer);
astroObject._outlineObj = CreateImage(gameObject, outline, body.Config.name + " Outline", layer);
if (ShipLogHandler.BodyHasEntries(body))
{
var revealedImage = astroObject._imageObj.GetComponent<Image>();
astroObject._greyscaleMaterial = greyScaleMaterial;
revealedImage.material = greyScaleMaterial;
revealedImage.color = Color.white;
astroObject._image = revealedImage;
}
astroObject._unviewedObj = Object.Instantiate(unviewedReference, gameObject.transform, false);
astroObject._invisibleWhenHidden = body.Config.ShipLog?.mapMode?.invisibleWhenHidden ?? false;
var imageRect = astroObject._imageObj.GetComponent<RectTransform>().rect;
astroObject._unviewedObj.transform.localPosition = new Vector3(imageRect.width / 2 + unviewedIconOffset,
imageRect.height / 2 + unviewedIconOffset, 0);
return astroObject;
}
#endregion
#region Details
private static void MakeDetail(ShipLogModule.ShipLogDetailInfo info, Transform parent, NewHorizonsBody body,
Material greyScaleMaterial)
{
var detailGameObject = new GameObject("Detail");
detailGameObject.transform.SetParent(parent);
detailGameObject.SetActive(false);
var detailTransform = detailGameObject.AddComponent<RectTransform>();
detailTransform.localPosition = (Vector2)(info.position ?? new MVector2(0, 0));
detailTransform.localRotation = Quaternion.Euler(0f, 0f, info.rotation);
detailTransform.localScale = (Vector2)(info.scale ?? new MVector2(0, 0));
Texture2D image;
Texture2D outline;
var imagePath = info.revealedSprite;
var outlinePath = info.outlineSprite;
if (imagePath != null) image = ImageUtilities.GetTexture(body.Mod, imagePath);
else image = AutoGenerateMapModePicture(body);
if (outlinePath != null) outline = ImageUtilities.GetTexture(body.Mod, outlinePath);
else outline = ImageUtilities.MakeOutline(image, Color.white, 10);
var revealedImage = CreateImage(detailGameObject, image, "Detail Revealed", parent.gameObject.layer)
.GetComponent<Image>();
var outlineImage = CreateImage(detailGameObject, outline, "Detail Outline", parent.gameObject.layer)
.GetComponent<Image>();
var detail = detailGameObject.AddComponent<ShipLogDetail>();
detail.Init(info, revealedImage, outlineImage, greyScaleMaterial);
detailGameObject.SetActive(true);
}
private static void MakeDetails(NewHorizonsBody body, Transform parent, Material greyScaleMaterial)
{
if (body.Config.ShipLog?.mapMode?.details?.Length > 0)
{
var detailsParent = new GameObject("Details");
detailsParent.transform.SetParent(parent);
detailsParent.SetActive(false);
var detailsTransform = detailsParent.AddComponent<RectTransform>();
detailsTransform.localPosition = Vector3.zero;
detailsTransform.localRotation = Quaternion.identity;
detailsTransform.localScale = Vector3.one;
foreach (var detailInfo in body.Config.ShipLog.mapMode.details)
MakeDetail(detailInfo, detailsTransform, body, greyScaleMaterial);
detailsParent.SetActive(true);
}
}
#endregion
#region Manual Map Mode
private static ShipLogAstroObject[][] ConstructMapModeManual(List<NewHorizonsBody> bodies,
GameObject transformParent, Material greyScaleMaterial, ShipLogAstroObject[][] currentNav, int layer)
{
var maxAmount = bodies.Count + 20;
var navMatrix = new ShipLogAstroObject[maxAmount][];
for (var i = 0; i < maxAmount; i++) navMatrix[i] = new ShipLogAstroObject[maxAmount];
var astroIdToNavIndex = new Dictionary<string, int[]>();
if (Main.Instance.CurrentStarSystem == "SolarSystem")
for (var y = 0; y < currentNav.Length; y++)
for (var x = 0; x < currentNav[y].Length; x++)
{
navMatrix[y][x] = currentNav[y][x];
astroIdToNavIndex.Add(currentNav[y][x].GetID(), new[] { y, x });
}
foreach (var body in bodies)
{
if (body.Config.ShipLog?.mapMode?.manualNavigationPosition == null) continue;
// Sometimes they got other names idk
var name = body.Config.name.Replace(" ", "");
var existingBody = AstroObjectLocator.GetAstroObject(body.Config.name);
if (existingBody != null)
{
var astroName = existingBody.GetAstroObjectName();
if (astroName == AstroObject.Name.RingWorld) name = "InvisiblePlanet";
else if (astroName != AstroObject.Name.CustomString) name = astroName.ToString();
}
// Should probably also just fix the IsVanilla method
var isVanilla = ShipLogHandler.IsVanillaBody(body);
if (!isVanilla)
{
var newMapModeGO = CreateMapModeGameObject(body, transformParent, layer,
body.Config.ShipLog?.mapMode?.manualPosition);
var newAstroObject = AddShipLogAstroObject(newMapModeGO, body, greyScaleMaterial, layer);
MakeDetails(body, newMapModeGO.transform, greyScaleMaterial);
Vector2 navigationPosition = body.Config.ShipLog?.mapMode?.manualNavigationPosition;
navMatrix[(int)navigationPosition.y][(int)navigationPosition.x] = newAstroObject;
}
else if (Main.Instance.CurrentStarSystem == "SolarSystem")
{
var gameObject = SearchUtilities.CachedFind(ShipLogHandler.PAN_ROOT_PATH + "/" + name);
if (body.Config.destroy || (body.Config.ShipLog?.mapMode?.remove ?? false))
{
var astroObject = gameObject.GetComponent<ShipLogAstroObject>();
if (astroObject != null)
{
var navIndex = astroIdToNavIndex[astroObject.GetID()];
navMatrix[navIndex[0]][navIndex[1]] = null;
if (astroObject.GetID() == "CAVE_TWIN" || astroObject.GetID() == "TOWER_TWIN")
GameObject.Find(ShipLogHandler.PAN_ROOT_PATH + "/" + "SandFunnel").SetActive(false);
}
else if (name == "SandFunnel")
{
GameObject.Find(ShipLogHandler.PAN_ROOT_PATH + "/" + "SandFunnel").SetActive(false);
}
gameObject.SetActive(false);
}
else
{
if (body.Config.ShipLog?.mapMode?.manualPosition != null)
gameObject.transform.localPosition = (Vector2)body.Config.ShipLog.mapMode.manualPosition;
if (body.Config.ShipLog?.mapMode?.manualNavigationPosition != null)
{
Vector2 navigationPosition = body.Config.ShipLog?.mapMode?.manualNavigationPosition;
navMatrix[(int)navigationPosition.y][(int)navigationPosition.x] =
gameObject.GetComponent<ShipLogAstroObject>();
}
if (body.Config.ShipLog?.mapMode?.scale != null)
gameObject.transform.localScale = Vector3.one * body.Config.ShipLog.mapMode.scale;
}
}
}
navMatrix = navMatrix.Where(a => a.Count(c => c != null && c.gameObject != null) > 0)
.Prepend(new ShipLogAstroObject[1]).ToArray();
for (var index = 0; index < navMatrix.Length; index++)
navMatrix[index] = navMatrix[index].Where(a => a != null && a.gameObject != null).ToArray();
return navMatrix;
}
#endregion
#region Automatic Map Mode
private class MapModeObject
{
public int x;
public int y;
public int branch_width;
public int branch_height;
public int level;
public NewHorizonsBody mainBody;
public ShipLogAstroObject astroObject;
public List<MapModeObject> children;
public MapModeObject parent;
public MapModeObject lastSibling;
public void IncrementWidth()
{
branch_width++;
parent?.IncrementWidth();
}
public void IncrementHeight()
{
branch_height++;
parent?.IncrementHeight();
}
}
private static ShipLogAstroObject[][] ConstructMapModeAuto(List<NewHorizonsBody> bodies,
GameObject transformParent, Material greyScaleMaterial, int layer)
{
var rootObject = ConstructPrimaryNode(bodies);
if (rootObject.mainBody != null) MakeAllNodes(ref rootObject, transformParent, greyScaleMaterial, layer);
var maxAmount = bodies.Count;
var navMatrix = new ShipLogAstroObject[maxAmount][];
for (var i = 0; i < maxAmount; i++) navMatrix[i] = new ShipLogAstroObject[maxAmount];
CreateNavigationMatrix(rootObject, ref navMatrix);
navMatrix = navMatrix.Where(a => a.Count(c => c != null) > 0).Prepend(new ShipLogAstroObject[1]).ToArray();
for (var index = 0; index < navMatrix.Length; index++)
navMatrix[index] = navMatrix[index].Where(a => a != null).ToArray();
return navMatrix;
}
private static void CreateNavigationMatrix(MapModeObject root, ref ShipLogAstroObject[][] navMatrix)
{
if (root.astroObject != null) navMatrix[root.y][root.x] = root.astroObject;
foreach (var child in root.children) CreateNavigationMatrix(child, ref navMatrix);
}
private static void MakeAllNodes(ref MapModeObject parentNode, GameObject parent, Material greyScaleMaterial,
int layer)
{
MakeNode(ref parentNode, parent, greyScaleMaterial, layer);
for (var i = 0; i < parentNode.children.Count; i++)
{
var child = parentNode.children[i];
MakeAllNodes(ref child, parent, greyScaleMaterial, layer);
parentNode.children[i] = child;
}
}
private static MapModeObject ConstructPrimaryNode(List<NewHorizonsBody> bodies)
{
foreach (var body in bodies.Where(b => b.Config.Base.centerOfSolarSystem))
{
bodies.Sort((b, o) => b.Config.Orbit.semiMajorAxis.CompareTo(o.Config.Orbit.semiMajorAxis));
var newNode = new MapModeObject
{
mainBody = body,
level = 0,
x = 0,
y = 0
};
newNode.children = ConstructChildrenNodes(newNode, bodies);
return newNode;
}
Logger.LogError("Couldn't find center of system!");
return new MapModeObject();
}
private static List<MapModeObject> ConstructChildrenNodes(MapModeObject parent,
List<NewHorizonsBody> searchList, string secondaryName = "")
{
var children = new List<MapModeObject>();
var newX = parent.x;
var newY = parent.y;
var newLevel = parent.level + 1;
var lastSibling = parent;
foreach (var body in searchList.Where(b =>
b.Config.Orbit.PrimaryBody == parent.mainBody.Config.name || b.Config.name == secondaryName))
{
var even = newLevel % 2 == 0;
newX = even ? newX : newX + 1;
newY = even ? newY + 1 : newY;
var newNode = new MapModeObject
{
mainBody = body,
level = newLevel,
x = newX,
y = newY,
parent = parent,
lastSibling = lastSibling
};
var newSecondaryName = "";
if (body.Config.FocalPoint != null)
{
newNode.mainBody = searchList.Find(b => b.Config.name == body.Config.FocalPoint.primary);
newSecondaryName = searchList.Find(b => b.Config.name == body.Config.FocalPoint.secondary).Config
.name;
}
newNode.children = ConstructChildrenNodes(newNode, searchList, newSecondaryName);
if (even)
{
newY += newNode.branch_height;
parent.IncrementHeight();
}
else
{
newX += newNode.branch_width;
parent.IncrementWidth();
}
lastSibling = newNode;
children.Add(newNode);
}
return children;
}
private static void ConnectNodeToLastSibling(MapModeObject node, Material greyScaleMaterial)
{
Vector2 fromPosition = node.astroObject.transform.localPosition;
Vector2 toPosition = node.lastSibling.astroObject.transform.localPosition;
var newLink = new GameObject("Line_ShipLog");
newLink.layer = node.astroObject.gameObject.layer;
newLink.SetActive(false);
var transform = newLink.AddComponent<RectTransform>();
transform.SetParent(node.astroObject.transform.parent);
var center = toPosition + (fromPosition - toPosition) / 2;
transform.localPosition = new Vector3(center.x, center.y, -1);
transform.localRotation = Quaternion.identity;
transform.localScale = node.level % 2 == 0
? new Vector3(node.astroObject.transform.localScale.x / 5f,
Mathf.Abs(fromPosition.y - toPosition.y) / 100f, 1)
: new Vector3(Mathf.Abs(fromPosition.x - toPosition.x) / 100f,
node.astroObject.transform.localScale.y / 5f, 1);
var linkImage = newLink.AddComponent<Image>();
linkImage.color = new Color(0.28f, 0.28f, 0.5f, 0.12f);
var linkDetailInfo = new ShipLogModule.ShipLogDetailInfo
{
invisibleWhenHidden = node.mainBody.Config.ShipLog?.mapMode?.invisibleWhenHidden ?? false
};
var linkDetail = newLink.AddComponent<ShipLogDetail>();
linkDetail.Init(linkDetailInfo, linkImage, linkImage, greyScaleMaterial);
transform.SetParent(node.astroObject.transform);
transform.SetAsFirstSibling();
newLink.SetActive(true);
}
private static void MakeNode(ref MapModeObject node, GameObject parent, Material greyScaleMaterial, int layer)
{
const float padding = 100f;
var position = Vector2.zero;
if (node.lastSibling != null)
{
var lastAstroObject = node.lastSibling.astroObject;
var lastPosition = lastAstroObject.transform.localPosition;
position = lastPosition;
var extraDistance = (node.mainBody.Config.ShipLog?.mapMode?.offset ?? 0f) * 100;
if (node.level % 2 == 0)
position.y += padding * (node.y - node.lastSibling.y) + extraDistance;
else
position.x += padding * (node.x - node.lastSibling.x) + extraDistance;
}
var newNodeGO = CreateMapModeGameObject(node.mainBody, parent, layer, position);
var astroObject = AddShipLogAstroObject(newNodeGO, node.mainBody, greyScaleMaterial, layer);
if (node.mainBody.Config.FocalPoint != null)
{
astroObject._imageObj.GetComponent<Image>().enabled = false;
astroObject._outlineObj.GetComponent<Image>().enabled = false;
astroObject._unviewedObj.GetComponent<Image>().enabled = false;
}
node.astroObject = astroObject;
if (node.lastSibling != null) ConnectNodeToLastSibling(node, greyScaleMaterial);
MakeDetails(node.mainBody, newNodeGO.transform, greyScaleMaterial);
}
#endregion
private static Texture2D AutoGenerateMapModePicture(NewHorizonsBody body)
{
Texture2D texture;
if (body.Config.Star != null)
texture = ImageUtilities.GetTexture(Main.Instance, "AssetBundle/DefaultMapModeStar.png");
else if (body.Config.Atmosphere != null)
texture = ImageUtilities.GetTexture(Main.Instance, "AssetBundle/DefaultMapModNoAtmo.png");
else texture = ImageUtilities.GetTexture(Main.Instance, "AssetBundle/DefaultMapModePlanet.png");
var color = GetDominantPlanetColor(body);
var darkColor = new Color(color.r / 3f, color.g / 3f, color.b / 3f);
texture = ImageUtilities.LerpGreyscaleImage(texture, color, darkColor);
return texture;
}
private static Color GetDominantPlanetColor(NewHorizonsBody body)
{
try
{
switch (body.Config?.Singularity?.type)
{
case SingularityModule.SingularityType.BlackHole:
return Color.black;
case SingularityModule.SingularityType.WhiteHole:
return Color.white;
}
var starColor = body.Config?.Star?.tint;
if (starColor != null) return starColor;
var atmoColor = body.Config.Atmosphere?.atmosphereTint;
if (body.Config.Atmosphere?.clouds != null && atmoColor != null) return atmoColor;
if (body.Config?.HeightMap?.textureMap != null)
try
{
var texture = ImageUtilities.GetTexture(body.Mod, body.Config.HeightMap.textureMap);
var landColor = ImageUtilities.GetAverageColor(texture);
if (landColor != null) return landColor;
}
catch (Exception) { }
var waterColor = body.Config.Water?.tint;
if (waterColor != null) return waterColor;
var lavaColor = body.Config.Lava?.tint;
if (lavaColor != null) return lavaColor;
var sandColor = body.Config.Sand?.Tint;
if (sandColor != null) return sandColor;
}
catch (Exception)
{
Logger.LogWarning(
$"Something went wrong trying to pick the colour for {body.Config.name} but I'm too lazy to fix it.");
}
return Color.white;
}
}
}