bel
/
ana-ledger
1
0
Fork 0
Code Actions Packages Projects Releases Activity

wow taht passes

main
bel 2023-10-26 22:10:27 -06:00
parent c1237b4f5e
commit c044a48455
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ana/predictor.go
View File

@ -46,16 +46,17 @@ func NewContributionPredictor(reg ledger.Register) Predictor {
} }
func getMonthlyContributionRates(reg ledger.Register) map[string]ledger.Balance { func getMonthlyContributionRates(reg ledger.Register) map[string]ledger.Balance {
contributions := getRecentContributions(reg) window := time.Hour * 24 * 365 / 2
contributions := getRecentContributions(reg, window)
result := make(map[string]ledger.Balance) result := make(map[string]ledger.Balance)
for name := range contributions { for name := range contributions {
result[name] = getMonthlyContributionRate(contributions[name]) result[name] = getMonthlyContributionRate(contributions[name], window)
} }
return result return result
} }
// TODO better than average // TODO better than average
func getMonthlyContributionRate(contributions []ledger.Balance) ledger.Balance { func getMonthlyContributionRate(contributions []ledger.Balance, window time.Duration) ledger.Balance {
sumPerCurrency := map[ledger.Currency]float64{} sumPerCurrency := map[ledger.Currency]float64{}
for _, balance := range contributions { for _, balance := range contributions {
for currency, v := range balance { for currency, v := range balance {
@ -64,13 +65,13 @@ func getMonthlyContributionRate(contributions []ledger.Balance) ledger.Balance {
} }
result := make(ledger.Balance) result := make(ledger.Balance)
for currency, summed := range sumPerCurrency { for currency, summed := range sumPerCurrency {
result[currency] = summed / float64(len(contributions)) result[currency] = summed / (float64(window) / float64(time.Hour*24*365/12))
} }
return result return result
} }
func getRecentContributions(reg ledger.Register) map[string][]ledger.Balance { func getRecentContributions(reg ledger.Register, window time.Duration) map[string][]ledger.Balance {
return getContributions(reg.Between(time.Now().Add(-1*time.Hour*365/2), time.Now())) return getContributions(reg.Between(time.Now().Add(-1*window), time.Now()))
} }
func getContributions(reg ledger.Register) map[string][]ledger.Balance { func getContributions(reg ledger.Register) map[string][]ledger.Balance {

2
ana/predictor_test.go
View File

@ -94,7 +94,7 @@ func TestGetMonthlyContributionRate(t *testing.T) {
ledger.Balance{"x": 4}, ledger.Balance{"x": 4},
ledger.Balance{"y": 3}, ledger.Balance{"y": 3},
} }
got := getMonthlyContributionRate(input) got := getMonthlyContributionRate(input, time.Hour*24*365/4)
if len(got) != 2 { if len(got) != 2 {
t.Error(got) t.Error(got)
} }