bbgo_origin/pkg/style/pnl.go

package style

import (
	"github.com/c9s/bbgo/pkg/fixedpoint"
)

var LossEmoji = "🔥"
var ProfitEmoji = "💰"

// 0.1% = 10 bps
var DefaultPnLLevelResolution = fixedpoint.NewFromFloat(0.001)

const MaxEmojiRepeat = 6

func PnLColor(pnl fixedpoint.Value) string {
	if pnl.Sign() > 0 {
		return GreenColor
	}
	return RedColor
}

func PnLSignString(pnl fixedpoint.Value) string {
	if pnl.Sign() > 0 {
		return "+" + pnl.String()
	}
	return pnl.String()
}

func PnLEmojiSimple(pnl fixedpoint.Value) string {
	if pnl.Sign() < 0 {
		return LossEmoji
	}

	if pnl.IsZero() {
		return ""
	}

	return ProfitEmoji
}

func PnLEmojiMargin(pnl, margin, resolution fixedpoint.Value) (out string) {
	if margin.IsZero() {
		return PnLEmojiSimple(pnl)
	}

	if pnl.Sign() < 0 {
		out = LossEmoji
		level := max((margin.Neg()).Div(resolution).Int(), MaxEmojiRepeat)
		for i := 1; i < level; i++ {
			out += LossEmoji
		}
		return out
	}

	if pnl.IsZero() {
		return out
	}

	out = ProfitEmoji
	level := max(margin.Div(resolution).Int(), MaxEmojiRepeat)
	for i := 1; i < level; i++ {
		out += ProfitEmoji
	}

	return out
}