split library & example

This commit is contained in:
6543 2021-02-02 02:43:02 +01:00
parent c02c75e2e3
commit fd3ef77fa6
No known key found for this signature in database
GPG Key ID: C99B82E40B027BAE
2 changed files with 68 additions and 62 deletions

View File

@ -1,32 +1,24 @@
package main
package avatars
import (
"fmt"
"math"
"math/bits"
"os"
"strings"
"strconv"
"strings"
)
func main() {
if len(os.Args) < 2 {
fmt.Println("please specify a seed string")
return
}
fmt.Println(makeAvatar(os.Args[1]))
}
func makeAvatar(seedString string) string {
// MakeAvatar create svg from seed
func MakeAvatar(seedString string) string {
var seed uint64
for _, c := range []byte(seedString) {
seed = bits.RotateLeft64(seed, 8)
seed ^= uint64(c)
}
if seed&1 == 0 {
return femaleAvatar(seed)
return femaleAvatar(seed, "")
} else {
return maleAvatar(seed)
return maleAvatar(seed, "")
}
}
@ -36,35 +28,35 @@ func makeAvatar(seedString string) string {
* cf. https://github.com/DiceBear/avatars/blob/master/packages/avatars-male-sprites/src/index.ts
*/
type LCG struct {
type lcg struct {
seed uint64
}
func (g *LCG) random() uint32 {
func (g *lcg) random() uint32 {
/* Linear Congruent Generator, POSIX/glibc [de]rand48 setting, bits [47..0] are output bits */
g.seed = (25214903917*g.seed + 11) % 281474976710656
return uint32(g.seed)
}
func (g *LCG) binomial(p float64) bool {
func (g *lcg) binomial(p float64) bool {
/* Sample from Binomial distribution with probability p */
var sample = float64(g.random()) * float64(1.0/4294967295.0)
return sample > p
}
func (g *LCG) pickOne(s []string) string {
func (g *lcg) pickOne(s []string) string {
/* Pick one element from list */
var N = uint32(len(s))
return s[g.random()%N]
}
func (g *LCG) pickOneFloat(s []float64) float64 {
func (g *lcg) pickOneFloat(s []float64) float64 {
/* Pick one element from list - float version*/
var N = uint32(len(s))
return s[g.random()%N]
}
func (g *LCG) pickAorB(p float64, a string, b string) string {
func (g *lcg) pickAorB(p float64, a string, b string) string {
/* Pick a or b with probability p of picking a */
if g.binomial(p) {
return a
@ -72,28 +64,28 @@ func (g *LCG) pickAorB(p float64, a string, b string) string {
return b
}
func LinearCongruentialGenerator(seed uint64) *LCG {
g := new(LCG)
func linearCongruentialGenerator(seed uint64) *lcg {
g := new(lcg)
g.seed = seed
return g
}
type RGB struct {
type rgb struct {
r uint8
g uint8
b uint8
a uint8
}
type HSV struct {
type hsv struct {
h float64
s float64
v float64
}
func f2rgb(r, g, b float64) *RGB {
// make RGB from 3 floats
c := new(RGB)
func f2rgb(r, g, b float64) *rgb {
// make rgb from 3 floats
c := new(rgb)
c.r = uint8(r * 1.0 / 255)
c.g = uint8(g * 1.0 / 255)
c.b = uint8(b * 1.0 / 255)
@ -101,23 +93,23 @@ func f2rgb(r, g, b float64) *RGB {
return c
}
func f2hsv(h, s, v float64) *HSV {
// make HSV from 3 floats
c := new(HSV)
func f2hsv(h, s, v float64) *hsv {
// make hsv from 3 floats
c := new(hsv)
c.h = h
c.s = s
c.v = v
return c
}
func rgb(s string) *RGB {
c := new(RGB)
func to_rgb(s string) *rgb {
c := new(rgb)
c.a = 255
fmt.Sscanf(s, "#%02x%02x%02x", &c.r, &c.g, &c.b)
return c
}
func (c *RGB) to_hsv() *HSV {
func (c *rgb) to_hsv() *hsv {
var h float64
var s float64
var v float64
@ -150,7 +142,7 @@ func (c *RGB) to_hsv() *HSV {
return f2hsv(h, s, v)
}
func (c *HSV) to_rgb() *RGB {
func (c *hsv) to_rgb() *rgb {
h := int((c.h / 60))
f := c.h/60 - float64(h)
p := c.v * (1 - c.s)
@ -174,7 +166,7 @@ func (c *HSV) to_rgb() *RGB {
return f2rgb(0, 0, 0)
}
func (c *RGB) brighterThan(ref *RGB, delta float64) *RGB {
func (c *rgb) brighterThan(ref *rgb, delta float64) *rgb {
primary := c.to_hsv()
secondary := ref.to_hsv()
if primary.v >= secondary.v+delta {
@ -187,7 +179,7 @@ func (c *RGB) brighterThan(ref *RGB, delta float64) *RGB {
return primary.to_rgb()
}
func (c *RGB) darkerThan(ref *RGB, delta float64) *RGB {
func (c *rgb) darkerThan(ref *rgb, delta float64) *rgb {
primary := c.to_hsv()
secondary := ref.to_hsv()
if primary.v <= secondary.v-delta {
@ -200,7 +192,7 @@ func (c *RGB) darkerThan(ref *RGB, delta float64) *RGB {
return primary.to_rgb()
}
func (c *RGB) brighterOrDarkerThan(ref *RGB, delta float64) *RGB {
func (c *rgb) brighterOrDarkerThan(ref *rgb, delta float64) *rgb {
primary := c.to_hsv()
secondary := ref.to_hsv()
if primary.v <= secondary.v {
@ -210,32 +202,31 @@ func (c *RGB) brighterOrDarkerThan(ref *RGB, delta float64) *RGB {
}
}
func (c *RGB) withAlpha(alpha float64) *RGB {
func (c *rgb) withAlpha(alpha float64) *rgb {
c.a = uint8(alpha * 255)
return c
}
func (c *RGB) html() string {
if (c.a == 255) {
func (c *rgb) html() string {
if c.a == 255 {
return fmt.Sprintf("#%02x%02x%02x", c.r, c.g, c.b)
}
return fmt.Sprintf("#%02x%02x%02x%02x", c.r, c.g, c.b, c.a)
}
func maleAvatar(seed uint64) string {
var g = LinearCongruentialGenerator(seed)
var skinColor = rgb(g.pickOne([]string{"#FFDBAC", "#F5CFA0", "#EAC393", "#E0B687", "#CB9E6E", "#B68655", "#A26D3D", "#8D5524"}))
var hairColor = rgb(g.pickOne([]string{"#090806", "#2c222b", "#71635a", "#b7a69e", "#b89778", "#a56b46", "#b55239", "#8d4a43",
func maleAvatar(seed uint64, mood string) string {
var g = linearCongruentialGenerator(seed)
var skinColor = to_rgb(g.pickOne([]string{"#FFDBAC", "#F5CFA0", "#EAC393", "#E0B687", "#CB9E6E", "#B68655", "#A26D3D", "#8D5524"}))
var hairColor = to_rgb(g.pickOne([]string{"#090806", "#2c222b", "#71635a", "#b7a69e", "#b89778", "#a56b46", "#b55239", "#8d4a43",
"#91553d", "#533d32", "#3b3024", "#554838", "#4e433f", "#504444", "#6a4e42", "#a7856a", "#977961"})).brighterOrDarkerThan(skinColor, 17)
var eyesColor = rgb(g.pickOne([]string{"#76778b", "#697b94", "#647b90", "#5b7c8b", "#588387"}))
var eyesColor = to_rgb(g.pickOne([]string{"#76778b", "#697b94", "#647b90", "#5b7c8b", "#588387"}))
var eyebrowsColor = hairColor.darkerThan(skinColor, 7).darkerThan(hairColor, 10)
var mustacheColor = hairColor.darkerThan(skinColor, 7).withAlpha(g.pickOneFloat([]float64{1, 0.75, 0.5}))
var mouthColor = rgb(g.pickOne([]string{"#eec1ad", "#dbac98", "#d29985"})).brighterOrDarkerThan(skinColor, 10)
var glassesColor = rgb(g.pickOne([]string{"#5f705c", "#43677d", "#5e172d", "#ffb67a", "#a04b5d", "#191919", "#323232", "#4b4b4b"}))
var clothesColor = rgb(g.pickOne([]string{"#5bc0de", "#5cb85c", "#428bca", "#03396c", "#005b96", "#6497b1", "#1b85b8", "#5a5255", "#559e83", "#ae5a41", "#c3cb71", "#666547", "#ffe28a"}))
var hatColor = rgb(g.pickOne([]string{"#18293b", "#2e1e05", "#989789", "#3d6ba7", "#517459", "#a62116"}))
var mouthColor = to_rgb(g.pickOne([]string{"#eec1ad", "#dbac98", "#d29985"})).brighterOrDarkerThan(skinColor, 10)
var glassesColor = to_rgb(g.pickOne([]string{"#5f705c", "#43677d", "#5e172d", "#ffb67a", "#a04b5d", "#191919", "#323232", "#4b4b4b"}))
var clothesColor = to_rgb(g.pickOne([]string{"#5bc0de", "#5cb85c", "#428bca", "#03396c", "#005b96", "#6497b1", "#1b85b8", "#5a5255", "#559e83", "#ae5a41", "#c3cb71", "#666547", "#ffe28a"}))
var hatColor = to_rgb(g.pickOne([]string{"#18293b", "#2e1e05", "#989789", "#3d6ba7", "#517459", "#a62116"}))
var mood = ""
if mood == "" {
mood = g.pickOne([]string{"sad", "happy", "surprised"})
}
@ -383,23 +374,22 @@ func maleAvatar(seed uint64) string {
return strings.NewReplacer(m...).Replace(s)
}
func femaleAvatar(seed uint64) string {
var g = LinearCongruentialGenerator(seed)
func femaleAvatar(seed uint64, mood string) string {
var g = linearCongruentialGenerator(seed)
var skinColor = rgb(g.pickOne([]string{"#FFDBAC", "#F5CFA0", "#EAC393", "#E0B687", "#CB9E6E", "#B68655", "#A26D3D", "#8D5524"}))
var hairColor = rgb(g.pickOne([]string{"#090806", "#2c222b", "#71635a", "#b7a69e", "#d6c4c2", "#cabfb1", "#dcd0ba", "#fff5e1",
var skinColor = to_rgb(g.pickOne([]string{"#FFDBAC", "#F5CFA0", "#EAC393", "#E0B687", "#CB9E6E", "#B68655", "#A26D3D", "#8D5524"}))
var hairColor = to_rgb(g.pickOne([]string{"#090806", "#2c222b", "#71635a", "#b7a69e", "#d6c4c2", "#cabfb1", "#dcd0ba", "#fff5e1",
"#e6cea8", "#e5c8a8", "#debc99", "#b89778", "#a56b46", "#b55239", "#8d4a43", "#91553d",
"#533d32", "#3b3024", "#554838", "#4e433f", "#504444", "#6a4e42", "#a7856a", "#977961"})).brighterOrDarkerThan(skinColor, 17)
var eyesColor = rgb(g.pickOne([]string{"#76778b", "#697b94", "#647b90", "#5b7c8b", "#588387"}))
var eyesColor = to_rgb(g.pickOne([]string{"#76778b", "#697b94", "#647b90", "#5b7c8b", "#588387"}))
var eyebrowsColor = hairColor.darkerThan(skinColor, 7).darkerThan(hairColor, 10)
var accessoriesColor = rgb(g.pickOne([]string{"#daa520", "#ffd700", "#eee8aa", "#fafad2", "#d3d3d3", "#a9a9a9"}))
var mouthColor = rgb(g.pickOne([]string{"#dbac98", "#d29985", "#c98276", "#e35d6a", "#e32153", "#de0f0d"})).brighterOrDarkerThan(skinColor, 10)
var glassesColor = rgb(g.pickOne([]string{"#5f705c", "#43677d", "#5e172d", "#ffb67a", "#a04b5d", "#191919", "#323232", "#4b4b4b"}))
var clothesColor = rgb(g.pickOne([]string{"#d11141", "#00b159", "#00aedb", "#f37735", "#ffc425", "#740001", "#ae0001", "#eeba30",
var accessoriesColor = to_rgb(g.pickOne([]string{"#daa520", "#ffd700", "#eee8aa", "#fafad2", "#d3d3d3", "#a9a9a9"}))
var mouthColor = to_rgb(g.pickOne([]string{"#dbac98", "#d29985", "#c98276", "#e35d6a", "#e32153", "#de0f0d"})).brighterOrDarkerThan(skinColor, 10)
var glassesColor = to_rgb(g.pickOne([]string{"#5f705c", "#43677d", "#5e172d", "#ffb67a", "#a04b5d", "#191919", "#323232", "#4b4b4b"}))
var clothesColor = to_rgb(g.pickOne([]string{"#d11141", "#00b159", "#00aedb", "#f37735", "#ffc425", "#740001", "#ae0001", "#eeba30",
"#96ceb4", "#ffeead", "#ff6f69", "#ffcc5c", "#88d8b0"}))
var hatColor = rgb(g.pickOne([]string{"#cc6192", "#2663a3", "#a62116", "#3d8a6b", "#614f8a"}))
var hatColor = to_rgb(g.pickOne([]string{"#cc6192", "#2663a3", "#a62116", "#3d8a6b", "#614f8a"}))
var mood = ""
if mood == "" {
mood = g.pickOne([]string{"sad", "happy", "surprised"})
}

16
example/main.go Normal file
View File

@ -0,0 +1,16 @@
package main
import (
"fmt"
"os"
"codeberg.org/Codeberg/avatars"
)
func main() {
if len(os.Args) < 2 {
fmt.Println("please specify a seed string")
return
}
fmt.Println(avatars.MakeAvatar(os.Args[1]))
}