Trio/FreeAPS/Sources/Modules/Bolus/View/Predictions.swift

import Charts
import CoreData
import SwiftUI
import Swinject

struct PredictionView: View {
    @Binding var predictions: Predictions?
    @Binding var units: GlucoseUnits
    @Binding var eventualBG: Int
    @Binding var target: Decimal

    var body: some View {
        VStack {
            chart()
            HStack {
                let conversion = units == .mmolL ? 0.0555 : 1
                Text("Eventual Glucose")
                Spacer()
                Text(
                    (Double(eventualBG) * conversion)
                        .formatted(.number.grouping(.never).rounded().precision(.fractionLength(units == .mmolL ? 1 : 0)))
                )
                Text(units.rawValue).foregroundStyle(.secondary)
                Divider()
            }.font(.callout)
        }
    }

    func chart() -> some View {
        // Data Source
        let iob = predictions?.iob ?? [Int]()
        let cob = predictions?.cob ?? [Int]()
        let uam = predictions?.uam ?? [Int]()
        let zt = predictions?.zt ?? [Int]()
        let count = max(iob.count, cob.count, uam.count, zt.count)
        var now = Date.now
        var startIndex = 0
        let conversion = units == .mmolL ? 0.0555 : 1
        // Organize the data needed for prediction chart.
        var data = [ChartData]()
        repeat {
            now = now.addingTimeInterval(5.minutes.timeInterval)
            if startIndex < count {
                let addedData = ChartData(
                    date: now,
                    iob: startIndex < iob.count ? Double(iob[startIndex]) * conversion : 0,
                    zt: startIndex < zt.count ? Double(zt[startIndex]) * conversion : 0,
                    cob: startIndex < cob.count ? Double(cob[startIndex]) * conversion : 0,
                    uam: startIndex < uam.count ? Double(uam[startIndex]) * conversion : 0,
                    id: UUID()
                )
                data.append(addedData)
            }
            startIndex += 1
        } while startIndex < count
        // Chart
        return Chart(data) {
            // Remove 0 (empty) values
            if $0.iob != 0 {
                LineMark(
                    x: .value("Time", $0.date),
                    y: .value("IOB", $0.iob),
                    series: .value("IOB", "A")
                )
                .foregroundStyle(Color(.insulin))
                .lineStyle(StrokeStyle(lineWidth: 2))
            }
            if $0.uam != 0 {
                LineMark(
                    x: .value("Time", $0.date),
                    y: .value("UAM", $0.uam),
                    series: .value("UAM", "B")
                )
                .foregroundStyle(Color(.UAM))
                .lineStyle(StrokeStyle(lineWidth: 2))
            }
            if $0.cob != 0 {
                LineMark(
                    x: .value("Time", $0.date),
                    y: .value("COB", $0.cob),
                    series: .value("COB", "C")
                )
                .foregroundStyle(Color(.loopYellow))
                .lineStyle(StrokeStyle(lineWidth: 2))
            }
            if $0.zt != 0 {
                LineMark(
                    x: .value("Time", $0.date),
                    y: .value("ZT", $0.zt),
                    series: .value("ZT", "D")
                )
                .foregroundStyle(Color(.ZT))
                .lineStyle(StrokeStyle(lineWidth: 2))
            }
        }
        .frame(minHeight: 150)
        .chartForegroundStyleScale([
            "IOB": Color(.insulin),
            "UAM": Color(.UAM),
            "COB": Color(.loopYellow),
            "ZT": Color(.ZT)
        ])
        .chartYAxisLabel("Glucose, " + units.rawValue, alignment: .center)
        .chartXAxis {
            AxisMarks(values: .stride(by: .hour)) { _ in
                AxisValueLabel(format: .dateTime.hour())
            }
        }
    }
}