using Elwig.Models.Entities;
using ScottPlot;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text.Json.Nodes;

namespace Elwig.Helpers.Billing {
    public class Graph : ICloneable {

        private readonly Season Season;
        public double[] DataX { get; set; }
        public double[] DataY { get; set; }
        public int MinX { get; set; }
        public int MaxX { get; set; }

        public Graph(Season season, int minX, int maxX) {
            Season = season;
            MinX = minX;
            MaxX = maxX;
            DataX = DataGen.Range(minX, maxX + 1);
            DataY = DataGen.Zeros(maxX - minX + 1);
        }

        public Graph(Dictionary<double, decimal> data, Season season, int minX, int maxX) {
            Season = season;
            MinX = minX;
            MaxX = maxX;
            DataX = DataGen.Range(minX, maxX + 1);
            DataY = DataX.Select(i => (double)BillingData.GetCurveValueAt(data, i)).ToArray();
        }

        private Graph(double[] dataX, double[] dataY, Season season, int minX, int maxX) {
            Season = season;
            MinX = minX;
            MaxX = maxX;
            DataX = dataX;
            DataY = dataY;
        }

        public double GetOechsleAt(int index) {
            return DataX[index];
        }

        public void SetOechsleAt(int index, double oechsle) {
            DataX[index] = oechsle;
        }

        public void SetPriceAt(int index, double price) {
            DataY[index] = price;
        }

        public double GetPriceAt(int index) {
            return DataY[index];
        }

        private void FlattenGraph(int begin, int end, double value) {
            for (int i = begin; i <= end; i++) {
                DataY[i] = value;
            }
        }

        public void FlattenGraphLeft(int pointIndex) {
            FlattenGraph(0, pointIndex, DataY[pointIndex]);
        }

        public void FlattenGraphRight(int pointIndex) {
            FlattenGraph(pointIndex, DataY.Length - 1, DataY[pointIndex]);
        }

        private void LinearIncreaseGraph(int begin, int end, double inc) {
            for (int i = begin; i < end; i++) {
                DataY[i + 1] = Math.Round(DataY[i] + inc, Season.Precision);
            }
        }

        public void LinearIncreaseGraphToEnd(int begin, double inc) {
            LinearIncreaseGraph(begin, DataY.Length - 1, inc);
        }

        public void InterpolateGraph(int firstPoint, int secondPoint) {
            int steps = Math.Abs(firstPoint - secondPoint);
            if (firstPoint == -1 || secondPoint == -1 || steps < 2) {
                return;
            }
            var (lowIndex, highIndex) = firstPoint < secondPoint ? (firstPoint, secondPoint) : (secondPoint, firstPoint);
            double step = (DataY[highIndex] - DataY[lowIndex]) / steps;

            for (int i = lowIndex; i < highIndex - 1; i++) {
                DataY[i + 1] = Math.Round(DataY[i] + step, Season.Precision);
            }
        }

        public JsonNode ToJson(string mode) {
            if (DataY.Distinct().Count() == 1) {
                return JsonValue.Create(DataY[0]);
            }

            var data = new JsonObject();

            if (DataY[0] != DataY[1]) {
                data.Add(new KeyValuePair<string, JsonNode?>(DataX[0] + mode, Math.Round(DataY[0], Season.Precision)));
            }
            for (int i = 1; i < DataX.Length - 1; i++) {
                if (Math.Round(DataY[i] - DataY[i - 1], 10) != Math.Round(DataY[i + 1] - DataY[i], 10)) {
                    data.Add(new KeyValuePair<string, JsonNode?>(DataX[i] + mode, Math.Round(DataY[i], Season.Precision)));
                }
            }
            if (DataY[^1] != DataY[^2]) {
                data.Add(new KeyValuePair<string, JsonNode?>(DataX[^1] + mode, Math.Round(DataY[^1], Season.Precision)));
            }
            return data;
        }

        public object Clone() {
            return new Graph((double[])DataX.Clone(), (double[])DataY.Clone(), Season, MinX, MaxX);
        }
    }
}