*

Assets/ABCUnity/Scripts/CustomStaff.cs

@@ -0,0 +1,75 @@
+using System.Collections.Generic;
+using UnityEngine;
+using UnityEngine.U2D;
+using TMPro;
+
+namespace ABCUnity
+{
+    public class CustomStaff
+    {
+        SpriteCache cache;
+        NoteCreator notes;
+
+        public Dictionary<int, List<string>> decorations { get; } = new Dictionary<int, List<string>>();
+
+        ABC.Clef clef;
+        public GameObject container { get; }
+
+        public bool isInit { get; private set; }
+        public float staffWidth { get { return currentWidth; } set { SetWidth(value);  } }
+        float staffSpriteBaseWidth = 0.0f;
+        float currentWidth = 0.0f;
+
+        public CustomStaff(GameObject parent, SpriteAtlas spriteAtlas, TextMeshPro textPrefab)
+        {
+            var staffContainer = new GameObject("Custom Staff");
+            staffContainer.transform.parent = parent.transform;
+            staffContainer.transform.localPosition = Vector3.zero;
+            this.container = staffContainer;
+
+            cache = new SpriteCache(spriteAtlas, textPrefab);
+            notes = new NoteCreator(cache);
+        }
+
+        public void Init(ABC.Clef clef)
+        {
+            this.clef = clef;
+            var info = notes.CreateStaff(this.clef, container, Vector3.zero);
+
+            SetWidth(info.totalBounding.size.x);
+            isInit = true;
+        }
+
+        private void SetWidth(float newWidth)
+        {
+            var staff = container.transform.GetChild(0).gameObject;
+            var staffRenderer = staff.GetComponent<SpriteRenderer>();
+
+            if (staffSpriteBaseWidth == 0.0f)
+                staffSpriteBaseWidth = staffRenderer.bounds.size.x;
+
+            var scaleX = newWidth / staffSpriteBaseWidth;
+            staffRenderer.transform.localScale = new Vector3(scaleX, 1.0f, 1.0f);
+            this.currentWidth = newWidth;
+        }
+
+        public GameObject AppendNote(ABC.Note note, float pos)
+        {
+            return AppendNote(note, null, pos);
+        }
+
+        public GameObject AppendNote(ABC.Note note, List<string> decorations, float pos)
+        {
+            GameObject noteContainer = new GameObject();
+            noteContainer.transform.parent = container.transform;
+
+            notes.CreateNote(note, this.clef, decorations, noteContainer);
+            noteContainer.transform.localPosition = new Vector3(pos, 0.0f, 0.0f);
+            noteContainer.transform.localScale = Vector3.one; // clear any scaling set on the parent
+
+            return noteContainer;
+        }
+    }
+
+}
+