Delete image.go
This commit is contained in:
Eccentric
parent
db3d60748a
commit
3cea75e5a0
|
|
@ -1,298 +0,0 @@
|
||||||
package fortnite
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"image"
|
|
||||||
"image/draw"
|
|
||||||
"image/png"
|
|
||||||
"io"
|
|
||||||
"math"
|
|
||||||
"net/http"
|
|
||||||
|
|
||||||
"github.com/ectrc/snow/aid"
|
|
||||||
"github.com/ectrc/snow/storage"
|
|
||||||
"github.com/nfnt/resize"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
PlaylistImages = map[string][]byte{
|
|
||||||
"Playlist_DefaultSolo": {},
|
|
||||||
"Playlist_DefaultDuo": {},
|
|
||||||
"Playlist_DefaultTrio": {},
|
|
||||||
"Playlist_DefaultSquad": {},
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
type colours struct {
|
|
||||||
averageRed uint8
|
|
||||||
averageGreen uint8
|
|
||||||
averageBlue uint8
|
|
||||||
}
|
|
||||||
|
|
||||||
var SETS_NOT_ALLOWED = []string{
|
|
||||||
"Soccer",
|
|
||||||
"Football",
|
|
||||||
"Waypoint",
|
|
||||||
}
|
|
||||||
|
|
||||||
func getAverageColour(img image.Image) colours {
|
|
||||||
var red, green, blue uint64
|
|
||||||
var count uint64
|
|
||||||
|
|
||||||
bounds := img.Bounds()
|
|
||||||
for x := bounds.Min.X; x < bounds.Max.X; x++ {
|
|
||||||
for y := bounds.Min.Y; y < bounds.Max.Y; y++ {
|
|
||||||
r, g, b, _ := img.At(x, y).RGBA()
|
|
||||||
red += uint64(r)
|
|
||||||
green += uint64(g)
|
|
||||||
blue += uint64(b)
|
|
||||||
count++
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return colours{
|
|
||||||
averageRed: uint8(red / count),
|
|
||||||
averageGreen: uint8(green / count),
|
|
||||||
averageBlue: uint8(blue / count),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetAverageHexColour(img image.Image) string {
|
|
||||||
colour := getAverageColour(img)
|
|
||||||
return "#" + aid.ToHex(int(colour.averageRed)) + aid.ToHex(int(colour.averageGreen)) + aid.ToHex(int(colour.averageBlue))
|
|
||||||
}
|
|
||||||
|
|
||||||
func colorDifference(c1, c2 colours) float64 {
|
|
||||||
diffRed := int(c1.averageRed) - int(c2.averageRed)
|
|
||||||
diffGreen := int(c1.averageGreen) - int(c2.averageGreen)
|
|
||||||
diffBlue := int(c1.averageBlue) - int(c2.averageBlue)
|
|
||||||
|
|
||||||
return math.Sqrt(float64(diffRed*diffRed + diffGreen*diffGreen + diffBlue*diffBlue))
|
|
||||||
}
|
|
||||||
|
|
||||||
func getRandomCharacterImage() image.Image {
|
|
||||||
character := RandomItemWithFeaturedImage()
|
|
||||||
response, err := http.Get(character.Images.Featured)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
defer response.Body.Close()
|
|
||||||
|
|
||||||
b, err := io.ReadAll(response.Body)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
image, _, err := image.Decode(bytes.NewReader(b))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return image
|
|
||||||
}
|
|
||||||
|
|
||||||
func getRandomCharacterImageWithSimilarColour(colour colours) image.Image {
|
|
||||||
character := getRandomCharacterImage()
|
|
||||||
characterColour := getAverageColour(character)
|
|
||||||
|
|
||||||
if colorDifference(characterColour, colour) <= 140 {
|
|
||||||
return character
|
|
||||||
}
|
|
||||||
|
|
||||||
return getRandomCharacterImageWithSimilarColour(colour)
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenerateSoloImage() {
|
|
||||||
background := *storage.Asset("background.png")
|
|
||||||
|
|
||||||
soloPlayer := getRandomCharacterImage()
|
|
||||||
bg, _, err := image.Decode(bytes.NewReader(background))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
m := image.NewRGBA(bg.Bounds())
|
|
||||||
draw.Draw(m, m.Bounds(), bg, image.Point{0, 0}, draw.Src)
|
|
||||||
|
|
||||||
resized := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 1.4), soloPlayer, resize.Lanczos3)
|
|
||||||
centre := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resized.Bounds().Dx()/2,
|
|
||||||
(m.Bounds().Dy()/2 - resized.Bounds().Dy()/2) + 200,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resized.Bounds().Add(centre), resized, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
var bytes bytes.Buffer
|
|
||||||
err = png.Encode(&bytes, m)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
PlaylistImages["Playlist_DefaultSolo"] = bytes.Bytes()
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenerateDuoImage() {
|
|
||||||
background := *storage.Asset("background.png")
|
|
||||||
|
|
||||||
player1 := getRandomCharacterImage()
|
|
||||||
player2 := getRandomCharacterImageWithSimilarColour(getAverageColour(player1))
|
|
||||||
|
|
||||||
bg, _, err := image.Decode(bytes.NewReader(background))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
m := image.NewRGBA(bg.Bounds())
|
|
||||||
draw.Draw(m, m.Bounds(), bg, image.Point{0, 0}, draw.Src)
|
|
||||||
|
|
||||||
resizedPlayerLeft := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 1.25), player1, resize.Lanczos3)
|
|
||||||
leftPosition := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resizedPlayerLeft.Bounds().Dx()/2 - 200,
|
|
||||||
(m.Bounds().Dy()/2 - resizedPlayerLeft.Bounds().Dy()/2) + 200,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resizedPlayerLeft.Bounds().Add(leftPosition), resizedPlayerLeft, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
resizedPlayerRight := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 1.4), player2, resize.Lanczos3)
|
|
||||||
rightPosition := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resizedPlayerRight.Bounds().Dx()/2 + 240,
|
|
||||||
(m.Bounds().Dy()/2 - resizedPlayerRight.Bounds().Dy()/2) + 200,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resizedPlayerRight.Bounds().Add(rightPosition), resizedPlayerRight, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
var bytes bytes.Buffer
|
|
||||||
err = png.Encode(&bytes, m)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
PlaylistImages["Playlist_DefaultDuo"] = bytes.Bytes()
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenerateTrioImage() {
|
|
||||||
background := *storage.Asset("background.png")
|
|
||||||
glow := *storage.Asset("glow.png")
|
|
||||||
|
|
||||||
player1 := getRandomCharacterImage()
|
|
||||||
player2 := getRandomCharacterImageWithSimilarColour(getAverageColour(player1))
|
|
||||||
player3 := getRandomCharacterImageWithSimilarColour(getAverageColour(player1))
|
|
||||||
|
|
||||||
bg, _, err := image.Decode(bytes.NewReader(background))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
glowImage, _, err := image.Decode(bytes.NewReader(glow))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
m := image.NewRGBA(bg.Bounds())
|
|
||||||
draw.Draw(m, m.Bounds(), bg, image.Point{0, 0}, draw.Src)
|
|
||||||
|
|
||||||
resizedPlayer1 := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 1), player1, resize.Lanczos3)
|
|
||||||
player1Position := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resizedPlayer1.Bounds().Dx()/2 - 400,
|
|
||||||
(m.Bounds().Dy()/2 - resizedPlayer1.Bounds().Dy()/2) + 150,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resizedPlayer1.Bounds().Add(player1Position), resizedPlayer1, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
resizedPlayer2 := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 1.1), player2, resize.Lanczos3)
|
|
||||||
player2Position := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resizedPlayer2.Bounds().Dx()/2 + 350,
|
|
||||||
(m.Bounds().Dy()/2 - resizedPlayer2.Bounds().Dy()/2) + 100,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resizedPlayer2.Bounds().Add(player2Position), resizedPlayer2, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
centre := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - glowImage.Bounds().Dx()/2,
|
|
||||||
m.Bounds().Dy()/2 - glowImage.Bounds().Dy()/2,
|
|
||||||
}
|
|
||||||
draw.Draw(m, glowImage.Bounds().Add(centre), glowImage, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
resizedPlayer3 := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 1.4), player3, resize.Lanczos3)
|
|
||||||
player3Position := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resizedPlayer3.Bounds().Dx()/2,
|
|
||||||
(m.Bounds().Dy()/2 - resizedPlayer3.Bounds().Dy()/2) + 200,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resizedPlayer3.Bounds().Add(player3Position), resizedPlayer3, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
var bytes bytes.Buffer
|
|
||||||
err = png.Encode(&bytes, m)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
PlaylistImages["Playlist_DefaultTrio"] = bytes.Bytes()
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenerateSquadImage() {
|
|
||||||
background := *storage.Asset("background.png")
|
|
||||||
glow := *storage.Asset("glow.png")
|
|
||||||
|
|
||||||
player1 := getRandomCharacterImage()
|
|
||||||
player2 := getRandomCharacterImageWithSimilarColour(getAverageColour(player1))
|
|
||||||
player3 := getRandomCharacterImageWithSimilarColour(getAverageColour(player1))
|
|
||||||
player4 := getRandomCharacterImageWithSimilarColour(getAverageColour(player1))
|
|
||||||
|
|
||||||
bg, _, err := image.Decode(bytes.NewReader(background))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
glowImage, _, err := image.Decode(bytes.NewReader(glow))
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
m := image.NewRGBA(bg.Bounds())
|
|
||||||
draw.Draw(m, m.Bounds(), bg, image.Point{0, 0}, draw.Src)
|
|
||||||
|
|
||||||
resizedPlayer4 := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 0.8), player4, resize.Lanczos3)
|
|
||||||
player4Position := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resizedPlayer4.Bounds().Dx()/2 - 600,
|
|
||||||
(m.Bounds().Dy()/2 - resizedPlayer4.Bounds().Dy()/2) + 110,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resizedPlayer4.Bounds().Add(player4Position), resizedPlayer4, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
resizedPlayer1 := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 1), player1, resize.Lanczos3)
|
|
||||||
player1Position := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resizedPlayer1.Bounds().Dx()/2 - 350,
|
|
||||||
(m.Bounds().Dy()/2 - resizedPlayer1.Bounds().Dy()/2) + 100,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resizedPlayer1.Bounds().Add(player1Position), resizedPlayer1, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
resizedPlayer2 := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 1.1), player2, resize.Lanczos3)
|
|
||||||
player2Position := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resizedPlayer2.Bounds().Dx()/2 + 400,
|
|
||||||
(m.Bounds().Dy()/2 - resizedPlayer2.Bounds().Dy()/2) + 100,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resizedPlayer2.Bounds().Add(player2Position), resizedPlayer2, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
centre := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - glowImage.Bounds().Dx()/2,
|
|
||||||
m.Bounds().Dy()/2 - glowImage.Bounds().Dy()/2,
|
|
||||||
}
|
|
||||||
draw.Draw(m, glowImage.Bounds().Add(centre), glowImage, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
resizedPlayer3 := resize.Resize(0, uint(float64(m.Bounds().Dy()) * 1.4), player3, resize.Lanczos3)
|
|
||||||
player3Position := image.Point{
|
|
||||||
m.Bounds().Dx()/2 - resizedPlayer3.Bounds().Dx()/2 + 150,
|
|
||||||
(m.Bounds().Dy()/2 - resizedPlayer3.Bounds().Dy()/2) + 200,
|
|
||||||
}
|
|
||||||
draw.Draw(m, resizedPlayer3.Bounds().Add(player3Position), resizedPlayer3, image.Point{0, 0}, draw.Over)
|
|
||||||
|
|
||||||
var bytes bytes.Buffer
|
|
||||||
err = png.Encode(&bytes, m)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
PlaylistImages["Playlist_DefaultSquad"] = bytes.Bytes()
|
|
||||||
}
|
|
||||||
|
|
||||||
func GeneratePlaylistImages() {
|
|
||||||
GenerateSoloImage()
|
|
||||||
GenerateDuoImage()
|
|
||||||
GenerateTrioImage()
|
|
||||||
GenerateSquadImage()
|
|
||||||
aid.Print("(snow) generated playlist images")
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue
Block a user