From 68b9d78386f74e14841ddc20df110591ac7fba23 Mon Sep 17 00:00:00 2001 From: Unknwon Date: Sun, 9 Aug 2015 12:06:08 +0800 Subject: [PATCH] remove sync.Pool and support go1.2 again --- .gopmfile | 1 - models/user.go | 3 +- modules/avatar/avatar.go | 2 +- modules/identicon/block.go | 405 ++++++++++++++++++++++++++++ modules/identicon/doc.go | 39 +++ modules/identicon/identicon.go | 147 ++++++++++ modules/identicon/identicon_test.go | 114 ++++++++ modules/identicon/polygon.go | 69 +++++ 8 files changed, 777 insertions(+), 3 deletions(-) create mode 100644 modules/identicon/block.go create mode 100644 modules/identicon/doc.go create mode 100644 modules/identicon/identicon.go create mode 100644 modules/identicon/identicon_test.go create mode 100644 modules/identicon/polygon.go diff --git a/.gopmfile b/.gopmfile index ad090603e..d59b474ee 100644 --- a/.gopmfile +++ b/.gopmfile @@ -14,7 +14,6 @@ github.com/go-xorm/core = github.com/go-xorm/xorm = github.com/gogits/chardet = commit:2404f77725 github.com/gogits/go-gogs-client = commit:92e76d616a -github.com/issue9/identicon = github.com/lib/pq = commit:0dad96c0b9 github.com/macaron-contrib/binding = commit:de6ed78668 github.com/macaron-contrib/cache = commit:cd824f6f2d diff --git a/models/user.go b/models/user.go index 65b23f791..3b28f5ed4 100644 --- a/models/user.go +++ b/models/user.go @@ -134,7 +134,7 @@ func (u *User) AvatarLink() string { return defaultImgUrl } if err = os.MkdirAll(path.Dir(imgPath), os.ModePerm); err != nil { - log.Error(3, "Create: %v", err) + log.Error(3, "MkdirAll: %v", err) return defaultImgUrl } fw, err := os.Create(imgPath) @@ -148,6 +148,7 @@ func (u *User) AvatarLink() string { log.Error(3, "Encode: %v", err) return defaultImgUrl } + log.Info("New random avatar created: %d", u.Id) } return setting.AppSubUrl + "/avatars/" + com.ToStr(u.Id) diff --git a/modules/avatar/avatar.go b/modules/avatar/avatar.go index 5aef6e02c..49a501bf9 100644 --- a/modules/avatar/avatar.go +++ b/modules/avatar/avatar.go @@ -32,9 +32,9 @@ import ( "sync" "time" - "github.com/issue9/identicon" "github.com/nfnt/resize" + "github.com/gogits/gogs/modules/identicon" "github.com/gogits/gogs/modules/log" "github.com/gogits/gogs/modules/setting" ) diff --git a/modules/identicon/block.go b/modules/identicon/block.go new file mode 100644 index 000000000..f8a6c6a34 --- /dev/null +++ b/modules/identicon/block.go @@ -0,0 +1,405 @@ +// Copyright 2015 by caixw, All rights reserved +// Use of this source code is governed by a MIT +// license that can be found in the LICENSE file. + +package identicon + +import ( + "image" +) + +var ( + // 可以出现在中间的方块，一般为了美观，都是对称图像。 + centerBlocks = []blockFunc{b0, b1, b2, b3} + + // 所有方块 + blocks = []blockFunc{b0, b1, b2, b3, b4, b5, b6, b7, b8, b9, b10, b11, b12, b13, b14, b15, b16} +) + +// 所有block函数的类型 +type blockFunc func(img *image.Paletted, x, y, size float64, angle int) + +// 将多边形points旋转angle个角度，然后输出到img上，起点为x,y坐标 +func drawBlock(img *image.Paletted, x, y, size float64, angle int, points []float64) { + if angle > 0 { // 0角度不需要转换 + // 中心坐标与x,y的距离，方便下面指定中心坐标(x+m,y+m)， + // 0.5的偏移值不能少，否则坐靠右，非正中央 + m := size/2 - 0.5 + rotate(points, x+m, y+m, angle) + } + + for i := x; i < x+size; i++ { + for j := y; j < y+size; j++ { + if pointInPolygon(i, j, points) { + img.SetColorIndex(int(i), int(j), 1) + } + } + } +} + +// 全空白 +// +// -------- +// | | +// | | +// | | +// -------- +func b0(img *image.Paletted, x, y, size float64, angle int) { +} + +// 全填充正方形 +// +// -------- +// |######| +// |######| +// |######| +// -------- +func b1(img *image.Paletted, x, y, size float64, angle int) { + isize := int(size) + ix := int(x) + iy := int(y) + for i := ix + 1; i < ix+isize; i++ { + for j := iy + 1; j < iy+isize; j++ { + img.SetColorIndex(i, j, 1) + } + } +} + +// 中间小方块 +// ---------- +// | | +// | #### | +// | #### | +// | | +// ---------- +func b2(img *image.Paletted, x, y, size float64, angle int) { + l := size / 4 + x = x + l + y = y + l + + for i := x; i < x+2*l; i++ { + for j := y; j < y+2*l; j++ { + img.SetColorIndex(int(i), int(j), 1) + } + } +} + +// 菱形 +// +// --------- +// | # | +// | ### | +// | ##### | +// |#######| +// | ##### | +// | ### | +// | # | +// --------- +func b3(img *image.Paletted, x, y, size float64, angle int) { + m := size / 2 + points := []float64{} + + drawBlock(img, x, y, size, 0, append(points, + x+m, y, + x+size, y+m, + x+m, y+size, + x, y+m, + x+m, y, + )) +} + +// b4 +// +// ------- +// |#####| +// |#### | +// |### | +// |## | +// |# | +// |------ +func b4(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + drawBlock(img, x, y, size, angle, append(points, + x, y, + x+size, y, + x, y+size, + x, y, + )) +} + +// b5 +// +// --------- +// | # | +// | ### | +// | ##### | +// |#######| +func b5(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x+m, y, + x+size, + y+size, + x, y+size, + x+m, y, + )) +} + +// b6 矩形 +// +// -------- +// |### | +// |### | +// |### | +// -------- +func b6(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x, y, + x+m, y, + x+m, y+size, + x, y+size, + x, y, + )) +} + +// b7 斜放的锥形 +// +// --------- +// | # | +// | ## | +// | #####| +// | ####| +// |-------- +func b7(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x, y, + x+size, y+m, + x+size, y+size, + x+m, y+size, + x, y, + )) +} + +// b8 三个堆叠的三角形 +// +// ----------- +// | # | +// | ### | +// | ##### | +// | # # | +// | ### ### | +// |#########| +// ----------- +func b8(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + mm := m / 2 + + // 顶部三角形 + drawBlock(img, x, y, size, angle, append(points, + x+m, y, + x+3*mm, y+m, + x+mm, y+m, + x+m, y, + )) + + // 底下左边 + drawBlock(img, x, y, size, angle, append(points[:0], + x+mm, y+m, + x+m, y+size, + x, y+size, + x+mm, y+m, + )) + + // 底下右边 + drawBlock(img, x, y, size, angle, append(points[:0], + x+3*mm, y+m, + x+size, y+size, + x+m, y+size, + x+3*mm, y+m, + )) +} + +// b9 斜靠的三角形 +// +// --------- +// |# | +// | #### | +// | #####| +// | #### | +// | # | +// --------- +func b9(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x, y, + x+size, y+m, + x+m, y+size, + x, y, + )) +} + +// b10 +// +// ---------- +// | ####| +// | ### | +// | ## | +// | # | +// |#### | +// |### | +// |## | +// |# | +// ---------- +func b10(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x+m, y, + x+size, y, + x+m, y+m, + x+m, y, + )) + + drawBlock(img, x, y, size, angle, append(points[:0], + x, y+m, + x+m, y+m, + x, y+size, + x, y+m, + )) +} + +// b11 左上角1/4大小的方块 +// +// ---------- +// |#### | +// |#### | +// |#### | +// | | +// | | +// ---------- +func b11(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x, y, + x+m, y, + x+m, y+m, + x, y+m, + x, y, + )) +} + +// b12 +// +// ----------- +// | | +// | | +// |#########| +// | ##### | +// | # | +// ----------- +func b12(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x, y+m, + x+size, y+m, + x+m, y+size, + x, y+m, + )) +} + +// b13 +// +// ----------- +// | | +// | | +// | # | +// | ##### | +// |#########| +// ----------- +func b13(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x+m, y+m, + x+size, y+size, + x, y+size, + x+m, y+m, + )) +} + +// b14 +// +// --------- +// | # | +// | ### | +// |#### | +// | | +// | | +// --------- +func b14(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x+m, y, + x+m, y+m, + x, y+m, + x+m, y, + )) +} + +// b15 +// +// ---------- +// |##### | +// |### | +// |# | +// | | +// | | +// ---------- +func b15(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x, y, + x+m, y, + x, y+m, + x, y, + )) +} + +// b16 +// +// --------- +// | # | +// | ##### | +// |#######| +// | # | +// | ##### | +// |#######| +// --------- +func b16(img *image.Paletted, x, y, size float64, angle int) { + points := []float64{} + m := size / 2 + drawBlock(img, x, y, size, angle, append(points, + x+m, y, + x+size, y+m, + x, y+m, + x+m, y, + )) + + drawBlock(img, x, y, size, angle, append(points[:0], + x+m, y+m, + x+size, y+size, + x, y+size, + x+m, y+m, + )) +} diff --git a/modules/identicon/doc.go b/modules/identicon/doc.go new file mode 100644 index 000000000..23a8e9b91 --- /dev/null +++ b/modules/identicon/doc.go @@ -0,0 +1,39 @@ +// Copyright 2015 by caixw, All rights reserved. +// Use of this source code is governed by a MIT +// license that can be found in the LICENSE file. + +// 一个基于hash值生成随机图像的包。 +// +// 关于identicon并没有统一的标准，一般用于在用户注册时， +// 取用户的邮箱或是访问IP等数据(也可以是其它任何数据)， +// 进行hash运算，之后根据hash数据，产生一张图像， +// 这样即可以为用户产生一张独特的头像，又不会泄漏用户的隐藏。 +// +// 在identicon中，把图像分成以下九个部分: +// ------------- +// | 1 | 2 | 3 | +// ------------- +// | 4 | 5 | 6 | +// ------------- +// | 7 | 8 | 9 | +// ------------- +// 其中1、3、9、7为不同角度(依次增加90度)的同一张图片， +// 2、6、8、4也是如此，这样可以保持图像是对称的，比较美观。 +// 5则单独使用一张图片。 +// +// // 根据用户访问的IP，为其生成一张头像 +// img, _ := identicon.Make(128, color.NRGBA{},color.NRGBA{}, []byte("192.168.1.1")) +// fi, _ := os.Create("/tmp/u1.png") +// png.Encode(fi, img) +// fi.Close() +// +// // 或者 +// ii, _ := identicon.New(128, color.NRGBA{}, color.NRGBA{}, color.NRGBA{}) +// img := ii.Make([]byte("192.168.1.1")) +// img = ii.Make([]byte("192.168.1.2")) +// +// NOTE: go test 会在当前目录的testdata文件夹下产生大量的随机图片。 +// 要运行测试，必须保证该文件夹是存在的，且有相应的写入权限。 +package identicon + +const Version = "0.2.6.150603" diff --git a/modules/identicon/identicon.go b/modules/identicon/identicon.go new file mode 100644 index 000000000..c0b1db3b0 --- /dev/null +++ b/modules/identicon/identicon.go @@ -0,0 +1,147 @@ +// Copyright 2015 by caixw, All rights reserved. +// Use of this source code is governed by a MIT +// license that can be found in the LICENSE file. + +package identicon + +import ( + "crypto/md5" + "fmt" + "image" + "image/color" +) + +const ( + minSize = 16 // 图片的最小尺寸 + maxForeColors = 32 // 在New()函数中可以指定的最大颜色数量 +) + +// Identicon 用于产生统一尺寸的头像。 +// 可以根据用户提供的数据，经过一定的算法，自动产生相应的图案和颜色。 +type Identicon struct { + foreColors []color.Color + backColor color.Color + size int + rect image.Rectangle +} + +// 声明一个Identicon实例。 +// size表示整个头像的大小。 +// back表示前景色。 +// fore表示所有可能的前景色，会为每个图像随机挑选一个作为其前景色。 +// NOTE:前景色不要与背景色太相近。 +func New(size int, back color.Color, fore ...color.Color) (*Identicon, error) { + if len(fore) == 0 || len(fore) > maxForeColors { + return nil, fmt.Errorf("前景色数量必须介于[1]~[%v]之间，当前为[%v]", maxForeColors, len(fore)) + } + + if size < minSize { + return nil, fmt.Errorf("参数size的值(%v)不能小于%v", size, minSize) + } + + return &Identicon{ + foreColors: fore, + backColor: back, + size: size, + + // 画布坐标从0开始，其长度应该是size-1 + rect: image.Rect(0, 0, size, size), + }, nil +} + +// 根据data数据产生一张唯一性的头像图片。 +func (i *Identicon) Make(data []byte) image.Image { + h := md5.New() + h.Write(data) + sum := h.Sum(nil) + + // 第一个方块 + index := int(sum[0]+sum[1]+sum[2]+sum[3]) % len(blocks) + b1 := blocks[index] + + // 第二个方块 + index = int(sum[4]+sum[5]+sum[6]+sum[7]) % len(blocks) + b2 := blocks[index] + + // 中间方块 + index = int(sum[8]+sum[9]+sum[10]+sum[11]) % len(centerBlocks) + c := centerBlocks[index] + + // 旋转角度 + angle := int(sum[12]+sum[13]+sum[14]) % 4 + + // 根据最后一个字段，获取前景颜色 + index = int(sum[15]) % len(i.foreColors) + + p := image.NewPaletted(i.rect, []color.Color{i.backColor, i.foreColors[index]}) + drawBlocks(p, i.size, c, b1, b2, angle) + return p +} + +// 根据data数据产生一张唯一性的头像图片。 +// size 头像的大小。 +// back, fore头像的背景和前景色。 +func Make(size int, back, fore color.Color, data []byte) (image.Image, error) { + if size < minSize { + return nil, fmt.Errorf("参数size的值(%v)不能小于%v", size, minSize) + } + + h := md5.New() + h.Write(data) + sum := h.Sum(nil) + + // 第一个方块 + index := int(sum[0]+sum[1]+sum[2]+sum[3]) % len(blocks) + b1 := blocks[index] + + // 第二个方块 + index = int(sum[4]+sum[5]+sum[6]+sum[7]) % len(blocks) + b2 := blocks[index] + + // 中间方块 + index = int(sum[8]+sum[9]+sum[10]+sum[11]) % len(centerBlocks) + c := centerBlocks[index] + + // 旋转角度 + angle := int(sum[12]+sum[13]+sum[14]+sum[15]) % 4 + + // 画布坐标从0开始，其长度应该是size-1 + p := image.NewPaletted(image.Rect(0, 0, size, size), []color.Color{back, fore}) + drawBlocks(p, size, c, b1, b2, angle) + return p, nil +} + +// 将九个方格都填上内容。 +// p为画板。 +// c为中间方格的填充函数。 +// b1,b2为边上8格的填充函数。 +// angle为b1,b2的起始旋转角度。 +func drawBlocks(p *image.Paletted, size int, c, b1, b2 blockFunc, angle int) { + // 每个格子的长宽。先转换成float，再计算！ + blockSize := float64(size) / 3 + twoBlockSize := 2 * blockSize + + incr := func() { // 增加angle的值，但不会大于3 + angle++ + if angle > 3 { + angle = 0 + } + } + + c(p, blockSize, blockSize, blockSize, 0) + + b1(p, 0, 0, blockSize, angle) + b2(p, blockSize, 0, blockSize, angle) + + incr() + b1(p, twoBlockSize, 0, blockSize, angle) + b2(p, twoBlockSize, blockSize, blockSize, angle) + + incr() + b1(p, twoBlockSize, twoBlockSize, blockSize, angle) + b2(p, blockSize, twoBlockSize, blockSize, angle) + + incr() + b1(p, 0, twoBlockSize, blockSize, angle) + b2(p, 0, blockSize, blockSize, angle) +} diff --git a/modules/identicon/identicon_test.go b/modules/identicon/identicon_test.go new file mode 100644 index 000000000..145a45bc0 --- /dev/null +++ b/modules/identicon/identicon_test.go @@ -0,0 +1,114 @@ +// Copyright 2015 by caixw, All rights reserved. +// Use of this source code is governed by a MIT +// license that can be found in the LICENSE file. + +package identicon + +import ( + "image" + "image/color" + "image/png" + "os" + "strconv" + "testing" + + "github.com/issue9/assert" +) + +var ( + back = color.RGBA{255, 0, 0, 100} + fore = color.RGBA{0, 255, 255, 100} + fores = []color.Color{color.Black, color.RGBA{200, 2, 5, 100}, color.RGBA{2, 200, 5, 100}} + size = 128 +) + +// 依次画出各个网络的图像。 +func TestBlocks(t *testing.T) { + p := []color.Color{back, fore} + + a := assert.New(t) + + for k, v := range blocks { + img := image.NewPaletted(image.Rect(0, 0, size*4, size), p) // 横向4张图片大小 + + for i := 0; i < 4; i++ { + v(img, float64(i*size), 0, float64(size), i) + } + + fi, err := os.Create("./testdata/block-" + strconv.Itoa(k) + ".png") + a.NotError(err).NotNil(fi) + a.NotError(png.Encode(fi, img)) + a.NotError(fi.Close()) // 关闭文件 + } +} + +// 产生一组测试图片 +func TestDrawBlocks(t *testing.T) { + a := assert.New(t) + + for i := 0; i < 20; i++ { + p := image.NewPaletted(image.Rect(0, 0, size, size), []color.Color{back, fore}) + c := (i + 1) % len(centerBlocks) + b1 := (i + 2) % len(blocks) + b2 := (i + 3) % len(blocks) + drawBlocks(p, size, centerBlocks[c], blocks[b1], blocks[b2], 0) + + fi, err := os.Create("./testdata/draw-" + strconv.Itoa(i) + ".png") + a.NotError(err).NotNil(fi) + a.NotError(png.Encode(fi, p)) + a.NotError(fi.Close()) // 关闭文件 + } +} + +func TestMake(t *testing.T) { + a := assert.New(t) + + for i := 0; i < 20; i++ { + img, err := Make(size, back, fore, []byte("make-"+strconv.Itoa(i))) + a.NotError(err).NotNil(img) + + fi, err := os.Create("./testdata/make-" + strconv.Itoa(i) + ".png") + a.NotError(err).NotNil(fi) + a.NotError(png.Encode(fi, img)) + a.NotError(fi.Close()) // 关闭文件 + } +} + +func TestIdenticon(t *testing.T) { + a := assert.New(t) + + ii, err := New(size, back, fores...) + a.NotError(err).NotNil(ii) + + for i := 0; i < 20; i++ { + img := ii.Make([]byte("identicon-" + strconv.Itoa(i))) + a.NotNil(img) + + fi, err := os.Create("./testdata/identicon-" + strconv.Itoa(i) + ".png") + a.NotError(err).NotNil(fi) + a.NotError(png.Encode(fi, img)) + a.NotError(fi.Close()) // 关闭文件 + } +} + +// BenchmarkMake 5000 229378 ns/op +func BenchmarkMake(b *testing.B) { + a := assert.New(b) + for i := 0; i < b.N; i++ { + img, err := Make(size, back, fore, []byte("Make")) + a.NotError(err).NotNil(img) + } +} + +// BenchmarkIdenticon_Make 10000 222127 ns/op +func BenchmarkIdenticon_Make(b *testing.B) { + a := assert.New(b) + + ii, err := New(size, back, fores...) + a.NotError(err).NotNil(ii) + + for i := 0; i < b.N; i++ { + img := ii.Make([]byte("Make")) + a.NotNil(img) + } +} diff --git a/modules/identicon/polygon.go b/modules/identicon/polygon.go new file mode 100644 index 000000000..d4cd1e9fd --- /dev/null +++ b/modules/identicon/polygon.go @@ -0,0 +1,69 @@ +// Copyright 2015 by caixw, All rights reserved. +// Use of this source code is governed by a MIT +// license that can be found in the LICENSE file. + +package identicon + +var ( + // 4个元素分别表示cos(0),cos(90),cos(180),cos(270) + cos = []float64{1, 0, -1, 0} + + // 4个元素分别表示sin(0),sin(90),sin(180),sin(270) + sin = []float64{0, 1, 0, -1} +) + +// 将points中的所有点，以x,y为原点旋转angle个角度。 +// angle取值只能是[0,1,2,3]，分别表示[0，90，180，270] +func rotate(points []float64, x, y float64, angle int) { + if angle > 3 { + panic("rotate:参数angle必须0,1,2,3三值之一") + } + + for i := 0; i < len(points); i += 2 { + px := points[i] - x + py := points[i+1] - y + points[i] = px*cos[angle] - py*sin[angle] + x + points[i+1] = px*sin[angle] + py*cos[angle] + y + } +} + +// 判断某个点是否在多边形之内，不包含构成多边形的线和点 +// x,y 需要判断的点坐标 +// points 组成多边形的所顶点，每两个元素表示一点顶点，其中最后一个顶点必须与第一个顶点相同。 +func pointInPolygon(x float64, y float64, points []float64) bool { + if len(points) < 8 { // 只有2个以上的点，才能组成闭合多边形 + return false + } + + // 大致算法如下： + // 把整个平面以给定的测试点为原点分两部分: + // - y>0，包含(x>0 && y==0) + // - y<0，包含(x<0 && y==0) + // 依次扫描每一个点，当该点与前一个点处于不同部分时（即一个在y>0区，一个在y<0区）， + // 则判断从前一点到当前点是顺时针还是逆时针（以给定的测试点为原点），如果是顺时针r++，否则r--。 + // 结果为：2==abs(r)。 + + r := 0 + x1, y1 := points[0], points[1] + prev := (y1 > y) || ((x1 > x) && (y1 == y)) + for i := 2; i < len(points); i += 2 { + x2, y2 := points[i], points[i+1] + curr := (y2 > y) || ((x2 > x) && (y2 == y)) + + if curr == prev { + x1, y1 = x2, y2 + continue + } + + mul := (x1-x)*(y2-y) - (x2-x)*(y1-y) + if mul > 0 { + r++ + } else if mul < 0 { + r-- + } + x1, y1 = x2, y2 + prev = curr + } + + return r == 2 || r == -2 +}