Skip to content

Commit

Permalink
Add chain previews
Browse files Browse the repository at this point in the history
  • Loading branch information
@TheBlackParrot
TheBlackParrot committed Jan 3, 2025
1 parent 68d48e0 commit 72b73a8
Show file tree
Hide file tree
Showing 2 changed files with 184 additions and 27 deletions.
186 changes: 163 additions & 23 deletions NoteTweaks/UI/PreviewViewController.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
using System.Threading.Tasks;
using BeatSaberMarkupLanguage.Attributes;
using BeatSaberMarkupLanguage.ViewControllers;
using NoteTweaks.Utils;
using UnityEngine;

namespace NoteTweaks.UI
Expand All @@ -16,11 +17,12 @@ internal class NotePreviewViewController : BSMLAutomaticViewController
internal static GameObject NoteContainer = new GameObject("_NoteTweaks_NoteContainer");

private static readonly float NoteSize = 0.5f;
private static readonly Vector3 InitialPosition = new Vector3(-2.9f, 1.15f, 3.7f);
private static readonly Vector3 InitialPosition = new Vector3(-2.7f, 1.15f, 3.5f);

internal static bool HasInitialized;
private static Vector3 _initialArrowPosition = Vector3.one;
private static Vector3 _initialDotPosition = Vector3.one;
private static Vector3 _initialChainDotPosition = Vector3.one;

private static Material _replacementDotMaterial;
private static Material _dotGlowMaterial;
Expand All @@ -33,7 +35,7 @@ internal class NotePreviewViewController : BSMLAutomaticViewController

private static readonly int Color0 = Shader.PropertyToID("_Color");

private static readonly List<String> FaceNames = new List<String> { "NoteArrow", "NoteCircleGlow" };
private static readonly List<String> FaceNames = new List<String> { "NoteArrow", "NoteCircleGlow", "Circle" };
private static readonly List<String> GlowNames = new List<String> { "NoteArrowGlow", "AddedNoteCircleGlow" };

public static void UpdateDotMesh()
Expand All @@ -53,7 +55,15 @@ public static void UpdateDotMesh()
for (int i = 0; i < NoteContainer.transform.childCount; i++)
{
GameObject noteCube = NoteContainer.transform.GetChild(i).gameObject;
noteCube.transform.Find("NoteCircleGlow").GetComponent<MeshFilter>().mesh = _dotMesh;
Transform noteCircleGlowTransform = noteCube.transform.Find("NoteCircleGlow");
if (noteCircleGlowTransform != null)
{
noteCircleGlowTransform.GetComponent<MeshFilter>().mesh = _dotMesh;
}
else
{
noteCube.transform.Find("Circle").GetComponent<MeshFilter>().mesh = _dotMesh;
}
}
}

Expand All @@ -62,18 +72,33 @@ public static void UpdateVisibility()
for (int i = 0; i < NoteContainer.transform.childCount; i++)
{
GameObject noteCube = NoteContainer.transform.GetChild(i).gameObject;

if (noteCube.transform.Find("NoteArrow").gameObject.activeSelf)
Transform noteArrowTransform = noteCube.transform.Find("NoteArrow");

if (noteArrowTransform != null)
{
// is not a dot note
noteCube.transform.Find("NoteArrowGlow").gameObject.SetActive(Plugin.Config.EnableFaceGlow);
if (noteCube.transform.Find("NoteArrow").gameObject.activeSelf)
{
// is not a dot note
noteCube.transform.Find("NoteArrowGlow").gameObject.SetActive(Plugin.Config.EnableFaceGlow);
}
else
{
// is a dot note
Transform noteCircleGlowTransform = noteCube.transform.Find("NoteCircleGlow");
if (noteCircleGlowTransform != null)
{
GameObject dotObject = noteCircleGlowTransform.gameObject;
dotObject.SetActive(Plugin.Config.EnableDots);
noteCube.transform.Find("AddedNoteCircleGlow").gameObject.SetActive(Plugin.Config.EnableFaceGlow && Plugin.Config.EnableDots);
}
}
}
else
{
// is a dot note
GameObject dotObject = noteCube.transform.Find("NoteCircleGlow").gameObject;
dotObject.SetActive(Plugin.Config.EnableDots);
noteCube.transform.Find("AddedNoteCircleGlow").gameObject.SetActive(Plugin.Config.EnableFaceGlow && Plugin.Config.EnableDots);
// is a chain link
GameObject dotObject = noteCube.transform.Find("Circle").gameObject;
dotObject.SetActive(Plugin.Config.EnableChainDots);
noteCube.transform.Find("AddedNoteCircleGlow").gameObject.SetActive(Plugin.Config.EnableChainDotGlow && Plugin.Config.EnableChainDots);
}
}
}
Expand All @@ -85,27 +110,42 @@ public static void UpdateDotPosition()
_initialDotPosition = NoteContainer.transform.GetChild(0).Find("NoteCircleGlow").localPosition;
}

Vector3 position = new Vector3(_initialDotPosition.x + Plugin.Config.DotPosition.x, _initialDotPosition.y + Plugin.Config.DotPosition.y, _initialDotPosition.z);
Vector3 glowPosition = new Vector3(_initialDotPosition.x + Plugin.Config.DotPosition.x, _initialDotPosition.y + Plugin.Config.DotPosition.y, _initialDotPosition.z + 0.001f);
Vector3 dotPosition = new Vector3(_initialDotPosition.x + Plugin.Config.DotPosition.x, _initialDotPosition.y + Plugin.Config.DotPosition.y, _initialDotPosition.z);
Vector3 dotGlowPosition = new Vector3(_initialDotPosition.x + Plugin.Config.DotPosition.x, _initialDotPosition.y + Plugin.Config.DotPosition.y, _initialDotPosition.z + 0.001f);

for (int i = 0; i < NoteContainer.transform.childCount; i++)
{
GameObject noteCube = NoteContainer.transform.GetChild(i).gameObject;
noteCube.transform.Find("NoteCircleGlow").localPosition = position;
noteCube.transform.Find("AddedNoteCircleGlow").localPosition = glowPosition;
Transform noteCircleGlowTransform = noteCube.transform.Find("NoteCircleGlow");
if (noteCircleGlowTransform != null)
{
noteCircleGlowTransform.localPosition = dotPosition;
noteCube.transform.Find("AddedNoteCircleGlow").localPosition = dotGlowPosition;
}
}
}

public static void UpdateDotScale()
{
Vector3 scale = new Vector3(Plugin.Config.DotScale.x / 5f, Plugin.Config.DotScale.y / 5f, 1.0f);
Vector3 glowScale = new Vector3((Plugin.Config.DotScale.x / 1.5f) * Plugin.Config.DotGlowScale, (Plugin.Config.DotScale.y / 1.5f) * Plugin.Config.DotGlowScale, 1.0f);
Vector3 chainLinkDotScale = new Vector3(Plugin.Config.ChainDotScale.x / 18f, Plugin.Config.ChainDotScale.y / 18f, 1.0f);
Vector3 chainLinkGlowScale = new Vector3((Plugin.Config.ChainDotScale.x / 5.4f) * Plugin.Config.DotGlowScale, (Plugin.Config.ChainDotScale.y / 5.4f) * Plugin.Config.DotGlowScale, 1.0f);

for (int i = 0; i < NoteContainer.transform.childCount; i++)
{
GameObject noteCube = NoteContainer.transform.GetChild(i).gameObject;
noteCube.transform.Find("NoteCircleGlow").localScale = scale;
noteCube.transform.Find("AddedNoteCircleGlow").localScale = glowScale;
Transform noteCircleGlowTransform = noteCube.transform.Find("NoteCircleGlow");
if (noteCircleGlowTransform != null)
{
noteCircleGlowTransform.localScale = scale;
noteCube.transform.Find("AddedNoteCircleGlow").localScale = glowScale;
}
else
{
noteCube.transform.Find("Circle").localScale = chainLinkDotScale;
noteCube.transform.Find("AddedNoteCircleGlow").localScale = chainLinkGlowScale;
}
}
}

Expand All @@ -115,11 +155,23 @@ public static void UpdateDotRotation()
{
GameObject noteCube = NoteContainer.transform.GetChild(i).gameObject;

noteCube.transform.Find("NoteCircleGlow").localRotation = Quaternion.identity;
noteCube.transform.Find("AddedNoteCircleGlow").localRotation = Quaternion.identity;
Transform circleTransform = noteCube.transform.Find("Circle");
Transform noteCircleGlowTransform = noteCube.transform.Find("NoteCircleGlow");
Transform addedNoteCircleGlowTransform = noteCube.transform.Find("AddedNoteCircleGlow");

if (noteCircleGlowTransform != null)
{
noteCircleGlowTransform.localRotation = Quaternion.identity;
noteCircleGlowTransform.Rotate(0f, 0f, Plugin.Config.RotateDot);
}
else
{
circleTransform.localRotation = Quaternion.identity;
circleTransform.Rotate(0f, 0f, Plugin.Config.RotateDot);
}

noteCube.transform.Find("NoteCircleGlow").Rotate(0f, 0f, Plugin.Config.RotateDot);
noteCube.transform.Find("AddedNoteCircleGlow").Rotate(0f, 0f, Plugin.Config.RotateDot);
addedNoteCircleGlowTransform.localRotation = Quaternion.identity;
addedNoteCircleGlowTransform.Rotate(0f, 0f, Plugin.Config.RotateDot);
}
}

Expand All @@ -135,6 +187,13 @@ public static void UpdateArrowPosition()
for (int i = 0; i < NoteContainer.transform.childCount; i++)
{
GameObject noteCube = NoteContainer.transform.GetChild(i).gameObject;

if (noteCube.name.Contains("_Chain_"))
{
// chain link, move on
continue;
}

noteCube.transform.Find("NoteArrow").localPosition = position;
noteCube.transform.Find("NoteArrowGlow").localPosition = position;
}
Expand All @@ -148,6 +207,13 @@ public static void UpdateArrowScale()
for (int i = 0; i < NoteContainer.transform.childCount; i++)
{
GameObject noteCube = NoteContainer.transform.GetChild(i).gameObject;

if (noteCube.name.Contains("_Chain_"))
{
// chain link, move on
continue;
}

noteCube.transform.Find("NoteArrow").localScale = scale;
noteCube.transform.Find("NoteArrowGlow").localScale = glowScale;
}
Expand All @@ -158,7 +224,14 @@ public static void UpdateNoteScale()
for (int i = 0; i < NoteContainer.transform.childCount; i++)
{
GameObject noteCube = NoteContainer.transform.GetChild(i).gameObject;
noteCube.transform.localScale = Plugin.Config.NoteScale;
if (noteCube.name.Contains("_Chain_"))
{
noteCube.transform.localScale = Vectors.Max(Plugin.Config.NoteScale * Plugin.Config.LinkScale, 0.1f);
}
else
{
noteCube.transform.localScale = Plugin.Config.NoteScale;
}
}
}

Expand Down Expand Up @@ -236,6 +309,61 @@ public static void UpdateColors()
}
}

private static void CreateChainNote(BurstSliderGameNoteController chainPrefab, string extraName, int cell, int linkNum)
{
GameObject chainNote = Instantiate(chainPrefab.transform.GetChild(0).gameObject, NoteContainer.transform);
chainNote.gameObject.SetActive(false);

chainNote.name = "_NoteTweaks_PreviewNote_" + extraName + $"_{linkNum}";
DestroyImmediate(chainNote.transform.Find("BigCuttable").gameObject);
DestroyImmediate(chainNote.transform.Find("SmallCuttable").gameObject);

Vector3 position = new Vector3((cell - 2.25f) * NoteSize, (linkNum / 6.667f) - 0.375f, 0);
chainNote.transform.localPosition = position;
chainNote.transform.Rotate(90f, 0f, 0f);

Transform originalDot = chainNote.transform.Find("Circle");
if (originalDot)
{
Transform originalDotTransform = originalDot.transform;

if (_initialChainDotPosition == Vector3.one)
{
_initialChainDotPosition = originalDotTransform.localPosition;
}

Vector3 dotPosition = new Vector3(_initialChainDotPosition.x, _initialChainDotPosition.y, _initialChainDotPosition.z - 0.002f);
Vector3 glowPosition = new Vector3(_initialChainDotPosition.x, _initialChainDotPosition.y, _initialChainDotPosition.z - 0.001f);
Vector3 dotScale = new Vector3(Plugin.Config.ChainDotScale.x / 18f, Plugin.Config.ChainDotScale.y / 18f, 1.0f);
Vector3 glowScale = new Vector3((Plugin.Config.ChainDotScale.x / 5.4f) * Plugin.Config.DotGlowScale, (Plugin.Config.ChainDotScale.y / 5.4f) * Plugin.Config.DotGlowScale, 1.0f);

originalDotTransform.localScale = dotScale;
originalDotTransform.localPosition = dotPosition;

MeshRenderer meshRenderer = originalDot.GetComponent<MeshRenderer>();

meshRenderer.GetComponent<MeshFilter>().mesh = _dotMesh;

meshRenderer.material = _replacementDotMaterial;
meshRenderer.sharedMaterial = _replacementDotMaterial;

GameObject newGlowObject = Instantiate(originalDot.gameObject, originalDot.parent);
newGlowObject.name = "AddedNoteCircleGlow";

newGlowObject.GetComponent<MeshFilter>().mesh = _dotGlowMesh;
newGlowObject.transform.localPosition = glowPosition;
newGlowObject.transform.localScale = glowScale;

if (newGlowObject.TryGetComponent(out MeshRenderer newGlowMeshRenderer))
{
newGlowMeshRenderer.material = _dotGlowMaterial;
newGlowMeshRenderer.sharedMaterial = _dotGlowMaterial;
}
}

chainNote.gameObject.SetActive(true);
}

private static void CreateNote(GameNoteController notePrefab, string extraName, int cell)
{
GameObject noteCube = Instantiate(notePrefab.transform.GetChild(0).gameObject, NoteContainer.transform);
Expand Down Expand Up @@ -450,7 +578,7 @@ protected void OnEnable()
}

NoteContainer.transform.position = InitialPosition;
NoteContainer.transform.localRotation = Quaternion.Euler(0, 330, 0);
NoteContainer.transform.localRotation = Quaternion.Euler(0, 320, 0);

MenuTransitionsHelper menuTransitionsHelper = Resources.FindObjectsOfTypeAll<MenuTransitionsHelper>().FirstOrDefault();
StandardLevelScenesTransitionSetupDataSO standardLevelScenesTransitionSetupData = menuTransitionsHelper._standardLevelScenesTransitionSetupData;
Expand All @@ -464,14 +592,26 @@ protected void OnEnable()
operation2 =>
{
BeatmapObjectsInstaller beatmapObjectsInstaller = Resources.FindObjectsOfTypeAll<BeatmapObjectsInstaller>().FirstOrDefault();

GameNoteController notePrefab = beatmapObjectsInstaller._normalBasicNotePrefab;
BurstSliderGameNoteController chainPrefab = beatmapObjectsInstaller._burstSliderNotePrefab;

List<String> noteNames = new List<string> { "L_Arrow", "R_Arrow", "L_Dot", "R_Dot" };
for (int i = 0; i < noteNames.Count; i++)
{
CreateNote(notePrefab, noteNames[i], i);
}

List<String> chainNames = new List<string> { "L_Chain", "R_Chain" };
int chainLinkCount = 6;
for (int j = 0; j < chainLinkCount; j++)
{
for (int i = 0; i < chainNames.Count; i++)
{
CreateChainNote(chainPrefab, chainNames[i], i, j);
}
}

UpdateColors();
UpdateArrowPosition();
UpdateArrowScale();
Expand Down
25 changes: 21 additions & 4 deletions NoteTweaks/UI/SettingsViewController.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -192,9 +192,13 @@ protected float NoteScaleZ
protected float LinkScale
{
get => _config.LinkScale;
set => _config.LinkScale = value;
set
{
_config.LinkScale = value;
NotePreviewViewController.UpdateNoteScale();
}
}

protected float ColorBoostLeft
{
get => _config.ColorBoostLeft;
Expand Down Expand Up @@ -258,8 +262,13 @@ protected float RightGlowIntensity
protected bool EnableChainDots
{
get => _config.EnableChainDots;
set => _config.EnableChainDots = value;
set
{
_config.EnableChainDots = value;
NotePreviewViewController.UpdateVisibility();
}
}

protected float ChainDotScaleX
{
get => _config.ChainDotScale.x;
Expand All @@ -268,6 +277,8 @@ protected float ChainDotScaleX
Vector3 scale = _config.ChainDotScale;
scale.x = value;
_config.ChainDotScale = scale;

NotePreviewViewController.UpdateDotScale();
}
}

Expand All @@ -279,13 +290,19 @@ protected float ChainDotScaleY
Vector3 scale = _config.ChainDotScale;
scale.y = value;
_config.ChainDotScale = scale;

NotePreviewViewController.UpdateDotScale();
}
}

protected bool EnableChainDotGlow
{
get => _config.EnableChainDotGlow;
set => _config.EnableChainDotGlow = value;
set
{
_config.EnableChainDotGlow = value;
NotePreviewViewController.UpdateVisibility();
}
}

protected Color LeftFaceColor
Expand Down

0 comments on commit 72b73a8

Please sign in to comment.