/* Distributed under the Apache License, Version 2.0.
   See accompanying NOTICE file for details.*/

using UnityEngine;

// Pulse data consumer which draws a vital line
[ExecuteInEditMode]
[RequireComponent(typeof(RectTransform))]
public class PulseDataLineRenderer : PulseDataConsumer
{
  [Range(.1f, 10f)]
  public float thickness = 1f;    // Thickness of the line renderer
  public Color color = Color.red; // Color of the line renderer

  public bool traceInitialLine = true;    // Trace line at start
  public float yMin = 0f;                 // Data minimum value
  public float yMax = 100f;               // Data maximum value
  public float xRange = 10f;              // Time period shown

  double previousTime = 0f;    // Used to determine dt and shift the line
  public LineRenderer lineRenderer;  // Inner component tracing the line
  RectTransform T;            // Inner component holding the canvas transform


  // MARK: Monobehavior methods

  // Called when application or editor opens
  void Awake()
  {
    InitInnerComponents();
    UpdateLineProperties();

    // Show line placeholder in editor only
    if (!Application.isPlaying)
      TracePlaceholderLine();
  }

  // Called when the component is being enabled
  void OnEnable()
  {
    lineRenderer.enabled = true;
  }

  // Called when the component is being disabled
  void OnDisable()
  {
    lineRenderer.enabled = false;
  }

  // Called when the inspector inputs are modified
  void OnValidate()
  {
    // Ensure min < max
    if (yMax < yMin)
      yMax = yMin + .1f;

    // Ensure time period > 0
    if (xRange <= 0)
      xRange = .1f;

    // Update properties and placeholder if needed
    UpdateLineProperties();
    if (!Application.isPlaying)
      TracePlaceholderLine();    
  }


  // MARK: PulseDataConsumer methods

  // Consume pulse data
  override internal void UpdateFromPulse(DoubleList times, DoubleList values)
  {
    for (int i = 0; i < times.Count; ++i)
      AddPoint(times.Get(i), values.Get(i));
  }


  // MARK: Custom methods

  // Initializes line and transform components
  void InitInnerComponents()
  {
    // Transform
    T = (RectTransform)this.transform;

    // Line
    lineRenderer = gameObject.GetComponent<LineRenderer>();
    if (!lineRenderer)
    {
      lineRenderer = gameObject.AddComponent<LineRenderer>();
      lineRenderer.hideFlags = HideFlags.None;
      lineRenderer.material = new Material(Shader.Find("Sprites/Default"));
    }
    lineRenderer.shadowCastingMode = UnityEngine.Rendering.ShadowCastingMode.Off;
    lineRenderer.receiveShadows = false;
    lineRenderer.allowOcclusionWhenDynamic = false;
    lineRenderer.useWorldSpace = false; // use canvas relative space
    lineRenderer.alignment = LineAlignment.TransformZ; // orthogonal to canvas
    lineRenderer.sortingOrder = 1;  // in front of image canvas (?)
    lineRenderer.positionCount = 0; // no points
  }

  // Show line placeholder
  void TracePlaceholderLine()
  {
    if (!lineRenderer)
      return;

    // Remove any points
    lineRenderer.positionCount = 0;

    // Trace a flat line in the middle
    var mean = (yMax + yMin) / 2;
    AddPoint(0, mean);
  }

  // Update line color and width
  void UpdateLineProperties()
  {
    if (!lineRenderer)
      return;

    T = (RectTransform)this.transform;
    lineRenderer.widthMultiplier = thickness / 100 * T.rect.height;
    lineRenderer.startColor = color;
    lineRenderer.endColor = color;
    lineRenderer.hideFlags = HideFlags.None;
  }

  // Append data to the line
  void AddPoint(double t, double val)
  {
    // Add a point on the left
    lineRenderer.positionCount++;

    // Compute dx to shift previous points
    double dt = t - previousTime;
    previousTime = t;
    double dx = DeltaTimeToDeltaX(dt);

    // Move previous position to the left
    for (int i = lineRenderer.positionCount - 1; i > 0; --i)
    {
      var pos = lineRenderer.GetPosition(i - 1);
      pos.x = pos.x - (float)dx;
      lineRenderer.SetPosition(i, pos);
    }

    // Put new point on far right (far right, id = 0)
    float x = (1 - T.pivot.x) * T.rect.width;
    float y = ValueToY(Mathf.Clamp((float)val, yMin, yMax));
    lineRenderer.SetPosition(0, new Vector3(x, y));

    // Check if points out of bounds need to be removed
    float originX = -T.pivot.x * T.rect.width;
    for (int i = lineRenderer.positionCount - 1; i > 0; --i)
    {
      // If point not out of bounds, break
      var pos1 = lineRenderer.GetPosition(i);
      if (pos1.x >= originX)
        break;

      // If next point in bound, move current point to the intersection
      // of the line with the left bound of the canvas
      var pos2 = lineRenderer.GetPosition(i - 1);
      if (pos2.x > originX)
      {
        // Calculate slop-intercept between those two points
        var m = (pos2.y - pos1.y) / (pos2.x - pos1.x);
        var b = pos1.y - m * pos1.x;

        // Place last point on the far left
        pos1.x = originX;
        pos1.y = m * originX + b;
        lineRenderer.SetPosition(i, pos1);
        break;
      }

      // If next point out of bounds too, decrease and go to next
      lineRenderer.positionCount--;
    }

    // Add point of far left if we only have one point on far right
    if (lineRenderer.positionCount == 1 && traceInitialLine)
    {
      lineRenderer.positionCount++;
      lineRenderer.SetPosition(1, new Vector3(originX, y));
    }
  }

  double DeltaTimeToDeltaX(double t)
  {
    double p = t / xRange;
    return p * T.rect.width;
  }

  float ValueToY(float val)
  {
    float p = (val - yMin) / (yMax - yMin);
    return (p - T.pivot.y) * T.rect.height;
  }
}