108 lines
2.9 KiB
Go
Raw Normal View History

2019-06-03 09:32:43 +00:00
package arn
import (
"fmt"
"sort"
"time"
)
const (
currentlyAiringBonus = 5.0
longSummaryBonus = 0.1
popularityThreshold = 5
popularityPenalty = 8.0
watchingPopularityWeight = 0.07
completedPopularityWeight = watchingPopularityWeight
plannedPopularityWeight = watchingPopularityWeight * (2.0 / 3.0)
droppedPopularityWeight = -plannedPopularityWeight
visualsWeight = 0.0075
storyWeight = 0.0075
soundtrackWeight = 0.0075
movieBonus = 0.28
agePenalty = 11.0
ageThreshold = 6 * 30 * 24 * time.Hour
)
// SortAnimeByPopularity sorts the given slice of anime by popularity.
func SortAnimeByPopularity(animes []*Anime) {
sort.Slice(animes, func(i, j int) bool {
aPopularity := animes[i].Popularity.Total()
bPopularity := animes[j].Popularity.Total()
if aPopularity == bPopularity {
return animes[i].Title.Canonical < animes[j].Title.Canonical
}
return aPopularity > bPopularity
})
}
// SortAnimeByQuality sorts the given slice of anime by quality.
func SortAnimeByQuality(animes []*Anime) {
SortAnimeByQualityDetailed(animes, "")
}
// SortAnimeByQualityDetailed sorts the given slice of anime by quality.
func SortAnimeByQualityDetailed(animes []*Anime, filterStatus string) {
sort.Slice(animes, func(i, j int) bool {
a := animes[i]
b := animes[j]
scoreA := a.Score()
scoreB := b.Score()
// If we show currently running shows, rank shows that started a long time ago a bit lower
if filterStatus == "current" {
if a.StartDate != "" && time.Since(a.StartDateTime()) > ageThreshold {
scoreA -= agePenalty
}
if b.StartDate != "" && time.Since(b.StartDateTime()) > ageThreshold {
scoreB -= agePenalty
}
}
if scoreA == scoreB {
return a.Title.Canonical < b.Title.Canonical
}
return scoreA > scoreB
})
}
// Score returns the score used for the anime ranking.
func (anime *Anime) Score() float64 {
score := anime.Rating.Overall
score += anime.Rating.Story * storyWeight
score += anime.Rating.Visuals * visualsWeight
score += anime.Rating.Soundtrack * soundtrackWeight
score += float64(anime.Popularity.Watching) * watchingPopularityWeight
score += float64(anime.Popularity.Planned) * plannedPopularityWeight
score += float64(anime.Popularity.Completed) * completedPopularityWeight
score += float64(anime.Popularity.Dropped) * droppedPopularityWeight
if anime.Status == "current" {
score += currentlyAiringBonus
}
if anime.Type == "movie" {
score += movieBonus
}
if anime.Popularity.Total() < popularityThreshold {
score -= popularityPenalty
}
if len(anime.Summary) >= 140 {
score += longSummaryBonus
}
return score
}
// ScoreHumanReadable returns the score used for the anime ranking in human readable format.
func (anime *Anime) ScoreHumanReadable() string {
return fmt.Sprintf("%.1f", anime.Score())
}