using System;
using UnityEngine;
using Unity.Mathematics;

namespace NanoBrain {

    /// <summary>
    /// A Synapse connects the ouput of a Neuron to another Neuron
    /// </summary>
    [Serializable]
    public class Synapse {
        /// <summary>
        /// The neuron from which input is received
        /// </summary>
        [SerializeReference]
        [HideInInspector]
        public Neuron neuron;

        /// <summary>
        /// The weight value to apply to the Neuron input
        /// </summary>
        public float weight;

        /// <summary>
        /// Indicator whether the weight can be trained
        /// </summary>
        public bool trainable = false;

        /// <summary>
        /// Create a new Synapse
        /// </summary>
        /// <param name="nucleus">The neuron from which input is received</param>
        /// <param name="weight">The weight value to apply to the Neuron input</param>
        public Synapse(Neuron nucleus, float weight = 1.0f) {
            this.neuron = nucleus;
            this.weight = weight;
        }

        public virtual void BackPropagation(Neuron receiver, float derivative, float learningRate) {
            switch (receiver.activator) {
                case Neuron.ActivationType.Linear:
                    derivative *= 1;
                    break;
                case Neuron.ActivationType.Power:
                    // untested
                    derivative *= 2 * math.length(this.neuron.combination);
                    break;
                case Neuron.ActivationType.Reciprocal:
                    // untested
                    derivative *= -1 / Mathf.Pow(math.length(this.neuron.combination), 2);
                    break;
                default:
                    Debug.Log("other activator");
                    break;
            }

            this.neuron.BackPropagation1D(derivative * this.weight, learningRate);

            derivative *= math.length(this.neuron.activation);

            if (this.trainable) {
                float deltaWeight = learningRate * derivative;
                this.weight += deltaWeight;
            }
        }

        public virtual void BackPropagation(Neuron receiver, Vector3 derivative, float learningRate) {
            switch (receiver.activator) {
                case Neuron.ActivationType.Linear:
                    derivative *= 1;
                    break;
                case Neuron.ActivationType.Power:
                    // untested
                    derivative *= 2 * math.length(this.neuron.combination);
                    break;
                case Neuron.ActivationType.Reciprocal:
                    // untested
                    derivative *= -1 / Mathf.Pow(math.length(this.neuron.combination), 2);
                    break;
                default:
                    Debug.Log("other activator");
                    break;
            }

            this.neuron.BackPropagation3D(derivative * this.weight, learningRate);

            derivative *= math.length(this.neuron.activation);

            if (this.trainable) {
                float deltaWeight = learningRate * derivative.magnitude;
                // Compared to the 1D solution, this does not decrease the weight because magnitude is always positive
                // derivative.direction and derivative.sign are different....
                if (AreOpposed(derivative, this.neuron.activation))
                    this.weight -= deltaWeight;
                else
                    this.weight += deltaWeight;
            }
        }

        public static bool AreOpposed(Vector3 a, Vector3 b) {
            // Check if the angle between the vectors is > 90 degrees
            return Vector3.Dot(a, b) < 0f;
        }
    }

}