aswi2021tri-musketyri-gitlab/Client/Client/Assets/Scripts/MapRenderer.cs @ 017c7108

using System;
using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.UI;

public class MapRenderer : MonoBehaviour
{
    [SerializeField]
    public SceneManager sceneManager;

    [SerializeField]
    private SpriteRenderer underlayTemplate;

    [SerializeField]
    private SpriteRenderer overlayTemplate;

    [SerializeField]
    public bool interactable = true;

    public Color minColor;

    public Color maxColor;


    private Vector3 startPos;

    private Vector3 startScale;


    public SpriteRenderer underlay;

    private Dictionary<string, SpriteRenderer> overlay = new Dictionary<string, SpriteRenderer>();

    private Dictionary<string, Image> overlay2 = new Dictionary<string, Image>();
    
    private bool first = true;


    // Start is called before the first frame update
    void Start()
    {
        startPos = transform.position;
        startScale = transform.localScale;
        
        Load();
        interactable = true;
    }

    void Update()
    {
        if (first)
        {
            first = false;

            // Get source directory
            string sourceDirectory = sceneManager.mapSourceDirectory;

            // Get file with asset relative positions
            TextAsset output = Resources.Load<TextAsset>($"{sourceDirectory}/output");

            if (output == null)
            {
                Debug.Log($"Layout file missing at {sourceDirectory}/output.csv. Returning.");
                return;
            }

            string[] lines = output.ToString().Split('\n');
            List<string> filenamesTMP = new List<string>();

            for (int i = 0; i < lines.Length; i++)
            {
                lines[i] = lines[i].Trim();
                if (lines[i].Length != 0 && !lines[i].StartsWith("//"))
                    filenamesTMP.Add(lines[i]);
            }

            lines = filenamesTMP.ToArray();

            for (int i = 0; i < lines.Length; i++)
            {
                var split = lines[i].Split(';');

                // Parse name minus the extension
                // X - percent offset from left btm corner
                // Y - percent offset from left btm corner
                var name = split[0].Substring(0, split[0].Length - 4);
                var xoff = float.Parse(split[1]);
                var yoff = float.Parse(split[2]);

                var img = overlay[name];

                if (img != null)
                {
                    var bounds = underlay.bounds;
                    var imgbounds = img.bounds;

                    var xshift = -bounds.size.x / 2 + imgbounds.size.x / 2 + bounds.size.x * (xoff / 100);
                    var yshift = -bounds.size.y / 2 + imgbounds.size.y / 2 + bounds.size.y * (yoff / 100);

                    img.gameObject.transform.position = bounds.center + new Vector3(xshift, yshift, 0);
                }
                else
                {
                    Debug.Log($"Error: Missing sprite {name}!");
                }
            }
        }
    }

    /// <summary>
    /// Loads map resources from an annotated directory
    /// A "layout.txt" file with directory annotation is required
    /// Creates the underlay map and building overlays addressable by their name as per the layout file
    /// </summary>
    /// <param name="sourceDirectory">Path to the source directory</param>
    public void Load()
    {
        // Get source directory
        string sourceDirectory = sceneManager.mapSourceDirectory;

        // Load directory layout
        // Read filenames
        // First filename is the underlay
        // All the others are the overlays, names as their respective building codes
        TextAsset layout = Resources.Load<TextAsset>($"{sourceDirectory}/layout");

        if (layout == null)
        {
            Debug.Log($"Layout file missing at {sourceDirectory}/layout.txt. Returning.");
            return;
        }

        string[] filenames = layout.ToString().Split('\n');

        // A map must contain at least an underlay
        if (filenames.Length < 1)
        {
            Debug.Log("Attempted to load an empty map. Returning.");
            return;
        }

        // Trim filenames (line breaks), remove empty line breaks and comments
        List<string> filenamesTMP = new List<string>();
        
        for (int i = 0; i < filenames.Length; i++)
        {
            filenames[i] = filenames[i].Trim();
            if (filenames[i].Length != 0 && !filenames[i].StartsWith("//"))
                filenamesTMP.Add(filenames[i]);
        }

        filenames = filenamesTMP.ToArray();

        // Load underlay sprite
        Sprite underlaySprite = Resources.Load<Sprite>($"{sourceDirectory}/{filenames[0]}");

        if (underlaySprite == null)
        {
            Debug.LogError($"Underlay sprite does not exist at path \"{sourceDirectory}/{filenames[0]}.png\". Returning.");
            return;
        }

        // Load overlay sprites
        Sprite[] overlaySprites = new Sprite[filenames.Length - 1];

        for (int i = 1; i < filenames.Length; i++)
        {
            overlaySprites[i-1] = Resources.Load<Sprite>($"{sourceDirectory}/{filenames[i]}");
            
            if (overlaySprites[i - 1] == null)
            {
                Debug.LogError($"Overlay sprite does not exist at path \"{sourceDirectory}/{filenames[i]}.png\". Returning.");
                return;
            }
        }

        // Create game objects
        underlay = Instantiate(underlayTemplate, transform);
        underlay.name = "Underlay";
        underlay.sprite = underlaySprite;

        for (int i = 1; i < filenames.Length; i++)
        {
            overlay[filenames[i]] = Instantiate(overlayTemplate, transform);
            overlay[filenames[i]].name = filenames[i];
            overlay[filenames[i]].sprite = overlaySprites[i - 1];
            overlay[filenames[i]].sortingOrder = i;
        }

        // Get references to remaining building sprites
        BuildingTag[] images = (FindObjectsOfType(typeof(BuildingTag)) as BuildingTag[]);
        foreach (var building in images)
        {
            overlay2[building.builidngName] = building.GetComponent<Image>();
            //Debug.Log(building.builidngName);
        }
    }

    public void SetColors(double[] predictions)
    {
        var buildings = Buildings.buildings;

        if (predictions.Length != buildings.Length)
        {
            Debug.LogError("Mismatch in prediction and building count.");
            return;
        }

        for (int i = 0; i < buildings.Length; i++)
        {
            if (predictions[i] < 0)
            {
                // TODO set different material
            }
            else
            {
                if (overlay.ContainsKey(buildings[i]))
                {
                    float a = overlay[buildings[i]].color.a;
                    var c = Color.Lerp(minColor, maxColor, (float)predictions[i] / 100f);
                    c.a = a;
                    overlay[buildings[i]].color = c;
                }

                else if (overlay2.ContainsKey(buildings[i]))
                {
                    RectTransform rt = overlay2[buildings[i]].GetComponent(typeof(RectTransform)) as RectTransform;
                    rt.SetSizeWithCurrentAnchors(RectTransform.Axis.Horizontal, 25 + ( 200 * ((float)predictions[i] / 100f)));
                    rt.ForceUpdateRectTransforms();

                    float a = overlay2[buildings[i]].color.a;
                    var c = Color.Lerp(minColor, maxColor, (float)predictions[i] / 100f);
                    c.a = a;
                    overlay2[buildings[i]].color = c;
                }

            }
        }

    }

}