100 lines
3.3 KiB
Swift
100 lines
3.3 KiB
Swift
//
|
|
// HeaderStatsView.swift
|
|
// Feels
|
|
//
|
|
// Created by Trey Tartt on 1/8/22.
|
|
//
|
|
|
|
import SwiftUI
|
|
import Charts
|
|
|
|
struct HeaderStatsView : UIViewRepresentable {
|
|
//Bar chart accepts data as array of BarChartDataEntry objects
|
|
var entries : [BarChartDataEntry]
|
|
|
|
init(fakeData: Bool, backDays: Int) {
|
|
entries = [BarChartDataEntry]()
|
|
|
|
var moodEntries: [MoodEntry]?
|
|
|
|
if fakeData {
|
|
moodEntries = PersistenceController.shared.randomEntries(count: 10)
|
|
} else {
|
|
var daysAgo = Calendar.current.date(byAdding: .day, value: -backDays, to: Date())!
|
|
daysAgo = Calendar.current.date(bySettingHour: 0, minute: 0, second: 0, of: daysAgo)!
|
|
|
|
moodEntries = PersistenceController.shared.getData(startDate: daysAgo, endDate: Date(), includedDays: [1,2,3,4,5,6,7])
|
|
}
|
|
if let moodEntries = moodEntries {
|
|
for (index, mood) in Mood.allValues.enumerated() {
|
|
entries.append(BarChartDataEntry(x: Double(index + 1),
|
|
y: Double(moodEntries.filter({
|
|
Int($0.moodValue) == mood.rawValue
|
|
}).count)))
|
|
}
|
|
}
|
|
}
|
|
|
|
// this func is required to conform to UIViewRepresentable protocol
|
|
func makeUIView(context: Context) -> BarChartView {
|
|
//crate new chart
|
|
let chart = BarChartView()
|
|
chart.drawGridBackgroundEnabled = false
|
|
chart.drawValueAboveBarEnabled = false
|
|
|
|
chart.xAxis.drawAxisLineEnabled = false
|
|
chart.xAxis.labelTextColor = .clear
|
|
|
|
chart.rightAxis.drawAxisLineEnabled = false
|
|
chart.rightAxis.labelTextColor = .clear
|
|
|
|
chart.leftAxis.drawAxisLineEnabled = false
|
|
chart.leftAxis.labelTextColor = .clear
|
|
|
|
chart.xAxis.drawGridLinesEnabled = false
|
|
chart.leftAxis.drawGridLinesEnabled = false
|
|
chart.leftAxis.axisLineColor = .clear
|
|
chart.rightAxis.axisLineColor = .clear
|
|
|
|
chart.legend.textColor = .clear
|
|
chart.legend.enabled = false
|
|
|
|
chart.drawBordersEnabled = false
|
|
chart.drawMarkers = false
|
|
// chart.yAxis.drawGridLinesEnabled = false
|
|
chart.rightAxis.drawGridLinesEnabled = false
|
|
chart.borderColor = .clear
|
|
//it is convenient to form chart data in a separate func
|
|
chart.data = addData()
|
|
return chart
|
|
}
|
|
|
|
// this func is required to conform to UIViewRepresentable protocol
|
|
func updateUIView(_ uiView: BarChartView, context: Context) {
|
|
//when data changes chartd.data update is required
|
|
uiView.data = addData()
|
|
}
|
|
|
|
func addData() -> BarChartData{
|
|
let data = BarChartData()
|
|
//BarChartDataSet is an object that contains information about your data, styling and more
|
|
let dataSet = BarChartDataSet(entries: entries)
|
|
// change bars color to green
|
|
dataSet.colors = Mood.allValues.map({ NSUIColor( $0.color ) })
|
|
//change data label
|
|
data.append(dataSet)
|
|
return data
|
|
}
|
|
|
|
typealias UIViewType = BarChartView
|
|
|
|
}
|
|
|
|
|
|
|
|
struct HeaderStatsView_Previews: PreviewProvider {
|
|
static var previews: some View {
|
|
HeaderStatsView(fakeData: true, backDays: 30).frame(minHeight: 85, maxHeight: 90)
|
|
}
|
|
}
|