-
Notifications
You must be signed in to change notification settings - Fork 18
/
reward_distribution.go
119 lines (105 loc) · 3.62 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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
// Copyright (C) 2023 Gobalsky Labs Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
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 {
var lastSeen *vega.Rank
for _, rank := range rankingTable {
if int(rank.StartRank) > ind {
break
}
lastSeen = rank
}
if lastSeen == nil {
return 0
}
return lastSeen.ShareRatio
}
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))
}