using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using Ichni.RhythmGame.Beatmap;
using UnityEngine;

namespace Ichni.RhythmGame.ThemeBundles.DepartureToMultiverse
{
    public partial class DTMNoteVisual : NoteVisualBase
    {
        public List<List<Material>> normalMaterialList;
        public List<List<Material>> highlightMaterialList;
        
        public new static DTMNoteVisual GenerateElement(string elementName, Guid id, List<string> tags,
            bool isFirstGenerated, string themeBundleName, string objectName, GameElement parentElement, bool isHighlighted)
        {
            DTMNoteVisual noteVisual = NoteVisualBase.GenerateElement(elementName, id, tags,
                isFirstGenerated, themeBundleName, objectName, parentElement, isHighlighted).GetComponent<DTMNoteVisual>();

            return noteVisual;
        }

        public override void FirstSetUpObject(bool isFirstGenerated)
        {
            NoteBase note = parentElement as NoteBase;
            if (note == null) throw new System.Exception("NoteVisual只能生成在Note下。");
            this.note = note;
            note.noteVisual = this;
            
            if (note is Flick flick)
            {
                flick.availableFlickDirections = new List<Vector2>() { Vector2.left, Vector2.right };
            }
        }

        public override void AfterInitialize()
        {
            base.AfterInitialize();
            Recover();
        }

        public override void Recover()
        {
            foreach (GameObject part in notePartList)
            {
                Renderer rend = part.GetComponent<Renderer>();
                if (rend != null)
                {
                    rend.material.SetFloat("_MainAlpha", 1f);
                }
            }

            foreach (EffectBase effect in effectSubmodule.effectCollection["Generate"])
            {
                effect.Recover();
            }
        }
    }

    public partial class DTMNoteVisual
    {
        public override void SaveBM()
        {
            matchedBM = new Beatmap.DTMNoteVisual_BM(elementName, elementGuid, tags,
                parentElement.matchedBM as GameElement_BM, themeBundleName, objectName, isHighlighted);
        }

        public override void SetHighlight()
        {
            //List<Renderer> partRendererList = notePartList.Select(part => part.GetComponent<Renderer>()).ToList();
            
            //partRendererList.ForEach(rend => Destroy(rend.material));

            if (note is Tap or Hold)
            {
                extraPartList[0].gameObject.SetActive(isHighlighted);
            }

            /*if (!isHighlighted)
            {
                for (int i = 0; i < partRendererList.Count; i++)
                {
                    for (int j = 0; j < partRendererList[i].materials.Length; j++)
                    {
                        partRendererList[i].materials[j] = normalMaterialList[i][j];
                    }
                }
            }
            else
            {
                for (int i = 0; i < partRendererList.Count; i++)
                {
                    for (int j = 0; j < partRendererList[i].materials.Length; j++)
                    {
                        partRendererList[i].materials[j] = highlightMaterialList[i][j];
                    }
                }
            }*/
        }
    }

    namespace Beatmap
    {
        public class DTMNoteVisual_BM : NoteVisualBase_BM
        {
            public DTMNoteVisual_BM()
            {

            }

            public DTMNoteVisual_BM(string elementName, Guid id, List<string> tags,
                GameElement_BM parent, string themeBundleName, string objectName, bool isHighlighted) :
                base(elementName, id, tags, parent, themeBundleName, objectName, isHighlighted)
            {

            }

            public override void ExecuteBM()
            {
                matchedElement = DTMNoteVisual.GenerateElement(elementName, elementGuid, tags, false,
                    themeBundleName, objectName, GetElement(attachedElementGuid), isHighlighted);
            }
        }
    }
}