HeavenStudioPlus/Assets/Scripts/Games/CropStomp/CropStomp.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using System;
using NaughtyBezierCurves;
using DG.Tweening;

using RhythmHeavenMania.Util;

namespace RhythmHeavenMania.Games.CropStomp
{
    public class CropStomp : Minigame
    {
        const float stepDistance = 2.115f;

        float scrollRate => stepDistance / (Conductor.instance.secPerBeat * 2f / Conductor.instance.musicSource.pitch);
        float grassWidth;

        private float newBeat = -1f; // So that marching can happen on beat 0.
        private float marchStartBeat = -1f;
        private float marchOffset;
        private int currentMarchBeat;
        private int stepCount;
        private bool isStepping;

        public bool isMarching => marchStartBeat != -1f && Conductor.instance.isPlaying;

        [NonSerialized] public bool isFlicking;

        public GameObject baseVeggie;
        public GameObject baseMole;
        public Animator legsAnim;
        public Animator bodyAnim;
        public Transform farmerTrans;
        public SpriteRenderer grass;
        public Transform grassTrans;
        public Transform scrollingHolder;
        public Transform veggieHolder;
        public Farmer farmer;
        public BezierCurve3D pickCurve;
        public BezierCurve3D moleCurve;

        private Tween shakeTween;

        public static CropStomp instance;

        private void Awake()
        {
            instance = this;
        }

        private async void Start()
        {
            // Finding grass sprite width for grass scrolling.
            var grassSprite = grass.sprite;
            var borderLeft = grassSprite.rect.xMin + grassSprite.border.x;
            var borderRight = grassSprite.rect.xMax - grassSprite.border.z;
            var borderWidthPixels = borderRight - borderLeft;
            grassWidth = borderWidthPixels / grassSprite.pixelsPerUnit;

            // Initialize vegetables.
            var cond = Conductor.instance;
            var entities = GameManager.instance.Beatmap.entities;

            // Find the beat of the closest "start marching" event.
            // If not found, default to current beat.
            float startBeat = cond.songPositionInBeats;

            var marchStarts = entities.FindAll(m => m.datamodel == "cropStomp/start marching");
            for (int i = 0; i < marchStarts.Count; i++)
            {
                var sampleBeat = marchStarts[i].beat;
                if (cond.songPositionInBeats <= sampleBeat + 0.25f) // 0.25-beat buffer in case the start marching event is directly next to the game switch event.
                {
                    startBeat = sampleBeat;
                    break;
                }
            }

            // Veggie and mole events.
            var vegEvents = entities.FindAll(v => v.datamodel == "cropStomp/veggies");
            var moleEvents = entities.FindAll(m => m.datamodel == "cropStomp/mole");

            // Spawn veggies.
            for (int i = 0; i < vegEvents.Count; i++)
            {
                var vegBeat = vegEvents[i].beat;
                var vegLength = vegEvents[i].length;

                // Only consider veggie events that aren't past the start point.
                if (startBeat < vegBeat + vegLength)
                {
                    int veggiesInEvent = Mathf.CeilToInt(vegLength + 1) / 2;

                    for (int b = 0; b < veggiesInEvent; b++)
                    {
                        var targetVeggieBeat = vegBeat + 2f * b;
                        if (startBeat < targetVeggieBeat)
                        {
                            SpawnVeggie(targetVeggieBeat, startBeat, false);
                        }
                    }
                }
            }

            // Spawn moles.
            for (int i = 0; i < moleEvents.Count; i++)
            {
                var moleBeat = moleEvents[i].beat;

                if (startBeat < moleBeat)
                {
                    SpawnVeggie(moleBeat, startBeat, true);
                }
            }
        }

        private void Update()
        {
            if (!isMarching)
                return;

            var cond = Conductor.instance;

            if (cond.ReportBeat(ref newBeat, marchOffset, true))
            {
                currentMarchBeat += 1;

                // Step.
                if (currentMarchBeat % 2 != 0)
                {
                    // Don't step if already stomped.
                    if (!isStepping)
                    {
                        stepCount += 1;
                        var stepAnim = (stepCount % 2 != 0 ? "StepFront" : "StepBack");
                        
                        legsAnim.Play(stepAnim, 0, 0);

                        isStepping = true;
                    }

                    Jukebox.PlayOneShotGame("cropStomp/hmm");
                }
                // Lift.
                else
                {
                    var liftAnim = (stepCount % 2 != 0 ? "LiftBack" : "LiftFront");
                    legsAnim.Play(liftAnim, 0, 0);

                    var farmerPos = farmerTrans.localPosition;
                    farmerTrans.localPosition = new Vector3(farmerPos.x - stepDistance, farmerPos.y, farmerPos.z);

                    isStepping = false;
                }
            }

            // Object scroll.
            var scrollPos = scrollingHolder.localPosition;
            var newScrollX = scrollPos.x + (scrollRate * Time.deltaTime);
            scrollingHolder.localPosition = new Vector3(newScrollX, scrollPos.y, scrollPos.z);

            // Grass scroll.
            var grassPos = grassTrans.localPosition;

            var newGrassX = grassPos.x + (scrollRate * Time.deltaTime);
            newGrassX = (newGrassX % (grassWidth * 4.5f));

            grassTrans.localPosition = new Vector3(newGrassX, grassPos.y, grassPos.z);
        }

        private void LateUpdate()
        {
            if (!isMarching)
                return;

            if (PlayerInput.PressedUp())
            {
                // Don't play raise animation if successfully flicked.
                if (!isFlicking)
                    bodyAnim.Play("Raise");
            }

            isFlicking = false;
        }

        public void StartMarching(float beat)
        {
            marchStartBeat = beat;
            marchOffset = (marchStartBeat % 1) * Conductor.instance.secPerBeat / Conductor.instance.musicSource.pitch;
            currentMarchBeat = 0;
            stepCount = 0;

            farmer.nextStompBeat = beat;
        }

        public void Stomp()
        {
            // Don't increment step counter if autostep stepped already.
            if (!isStepping)
                stepCount += 1;

            var stompAnim = (stepCount % 2 != 0 ? "StompFront" : "StompBack");
            
            legsAnim.Play(stompAnim, 0, 0);

            Jukebox.PlayOneShotGame("cropStomp/stomp");

            if (shakeTween != null)
                shakeTween.Kill(true);
            
            var camTrans = GameCamera.instance.transform;
            camTrans.localPosition = new Vector3(camTrans.localPosition.x, 0.75f, camTrans.localPosition.z);
            camTrans.DOLocalMoveY(0f, 0.5f).SetEase(Ease.OutElastic, 1f);

            isStepping = true;
        }

        private void SpawnVeggie(float beat, float startBeat, bool isMole)
        {
            var newVeggie = GameObject.Instantiate(isMole ? baseMole : baseVeggie, veggieHolder).GetComponent<Veggie>();

            newVeggie.targetBeat = beat;

            var veggieX = (beat - startBeat) * -stepDistance / 2f;
            newVeggie.transform.localPosition = new Vector3(veggieX, 0f, 0f);

            newVeggie.gameObject.SetActive(true);
        }
    }
}