Modular Tree

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

public class ModularTree : MonoBehaviour
{
    public SpriteRenderer[] treeParts;
    [SerializeField] private Sprite[] rottenParts;
    [SerializeField] private Sprite[] healthyParts;
    [SerializeField] private Sprite babyTreeSprite;
    public bool[] isHealthy;
    public int growthStage;

    private void Update()
    {
        ChangeTree();
    }

    public void ChangeTree()
    {
        for(int i = 0; i <= growthStage; i++)
        {
            //Checks for each branch
            if(isHealthy[i])
            {
                //baby con
                if(growthStage == 0)
                {
                    treeParts[i].sprite = babyTreeSprite;
                    ClearTreeBranches();
                }
                else
                {
                    treeParts[i].sprite = healthyParts[i];
                }

            }
            else if(!isHealthy[i])
            {
                treeParts[i].sprite = rottenParts[i];
            }

        }
    }
    public void ClearTreeBranches()
    {
        for(int i = 1; i < treeParts.Length; i++)
        {
            treeParts[i].sprite = null;
        }
    }

}