-
Notifications
You must be signed in to change notification settings - Fork 19
/
reward_distribution.go
97 lines (86 loc) · 2.86 KB
/
reward_distribution.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package rewards
import (
"sort"
"code.vegaprotocol.io/vega/core/types"
"code.vegaprotocol.io/vega/libs/num"
"code.vegaprotocol.io/vega/protos/vega"
)
func findRank(rankingTable []*vega.Rank, ind int) uint32 {
for i, r := range rankingTable {
if ind < int(r.StartRank) && i >= 1 {
return rankingTable[i-1].ShareRatio
}
}
return 0
}
func rankingRewardCalculator(partyMetric []*types.PartyContributionScore, rankingTable []*vega.Rank, partyRewardFactor map[string]num.Decimal) []*types.PartyContributionScore {
partyScores := []*types.PartyContributionScore{}
sort.Slice(partyMetric, func(i, j int) bool {
return partyMetric[i].Score.GreaterThan(partyMetric[j].Score)
})
shareRatio := num.DecimalZero()
totalScores := num.DecimalZero()
for i, ps := range partyMetric {
rewardFactor, ok := partyRewardFactor[ps.Party]
if !ok {
rewardFactor = num.DecimalOne()
}
if i == 0 || !ps.Score.Equal(partyMetric[i-1].Score) {
shareRatio = num.DecimalFromInt64(int64(findRank(rankingTable, i+1)))
}
score := shareRatio.Mul(rewardFactor)
if shareRatio.IsZero() {
break
}
if score.IsZero() {
continue
}
partyScores = append(partyScores, &types.PartyContributionScore{Party: ps.Party, Score: score})
totalScores = totalScores.Add(score)
}
if totalScores.IsZero() {
return []*types.PartyContributionScore{}
}
normalise(partyScores, totalScores)
return partyScores
}
func proRataRewardCalculator(partyContribution []*types.PartyContributionScore, partyRewardFactor map[string]num.Decimal) []*types.PartyContributionScore {
total := num.DecimalZero()
partiesWithScore := []*types.PartyContributionScore{}
for _, metric := range partyContribution {
factor, ok := partyRewardFactor[metric.Party]
if !ok {
factor = num.DecimalOne()
}
score := factor.Mul(metric.Score)
if score.IsZero() {
continue
}
total = total.Add(score)
partiesWithScore = append(partiesWithScore, &types.PartyContributionScore{Party: metric.Party, Score: score})
}
if total.IsZero() {
return []*types.PartyContributionScore{}
}
normalise(partiesWithScore, total)
return partiesWithScore
}
func normalise(partyRewardScores []*types.PartyContributionScore, total num.Decimal) {
normalisedTotal := num.DecimalZero()
for _, p := range partyRewardScores {
p.Score = p.Score.Div(total)
normalisedTotal = normalisedTotal.Add(p.Score)
}
if normalisedTotal.LessThanOrEqual(num.DecimalOne()) {
return
}
capAtOne(partyRewardScores, normalisedTotal)
}
func capAtOne(partyRewardScores []*types.PartyContributionScore, total num.Decimal) {
if total.LessThanOrEqual(num.DecimalOne()) {
return
}
sort.SliceStable(partyRewardScores, func(i, j int) bool { return partyRewardScores[i].Score.GreaterThan(partyRewardScores[j].Score) })
delta := total.Sub(num.DecimalFromInt64(1))
partyRewardScores[0].Score = num.MaxD(num.DecimalZero(), partyRewardScores[0].Score.Sub(delta))
}