func mapToColor(o map[string]interface{}) color.Color { var c color.RGBA // Because JavaScript switch x := o["r"].(type) { case int64: c.R = uint8(colorF(float64(x))) case float64: c.R = uint8(colorF(x)) } switch x := o["g"].(type) { case int64: c.G = uint8(colorF(float64(x))) case float64: c.G = uint8(colorF(x)) } switch x := o["b"].(type) { case int64: c.B = uint8(colorF(float64(x))) case float64: c.B = uint8(colorF(x)) } c.A = 0xff return c }
// Convert a []float64 to an image, reading data with the format specified in // chans (e.g. RGBA, BGRA, BRG, RRR). If a component is not specified in the func FromSliceChans(img *Image, chans string, fill float64, data []float64) { surf := img.Surf cm := surf.ColorModel() b := surf.Bounds() minX, maxX := b.Min.X, b.Max.X minY, maxY := b.Min.Y, b.Max.Y k := 0 nc := len(chans) const mk = 0xff dfill := clamp8(fill) & mk for i := minY; i < maxY; i++ { for j := minX; j < maxX; j++ { outCol := color.RGBA{dfill, dfill, dfill, dfill} for c := 0; c < nc; c++ { switch chans[c] { case 'r', 'R': outCol.R = clamp8(data[k+c]) & mk case 'g', 'G': outCol.G = clamp8(data[k+c]) & mk case 'b', 'B': outCol.B = clamp8(data[k+c]) & mk case 'a', 'A': outCol.A = clamp8(data[k+c]) & mk default: // Just skip the channel } } k += nc img.Surf.Set(j, i, cm.Convert(outCol)) } } }
func superSampling(inputImage image.Image) image.Image { rect := inputImage.Bounds() width := rect.Size().X height := rect.Size().Y rect2 := image.Rect(rect.Min.X, rect.Min.Y, rect.Max.X-1, rect.Max.Y-1) rgba := image.NewRGBA(rect2) for x := 0; x < width-1; x++ { for y := 0; y < height-1; y++ { var col color.RGBA // 座標(x,y)のR, G, B, α の値を取得 r00, g00, b00, a00 := inputImage.At(x, y).RGBA() r01, g01, b01, a01 := inputImage.At(x, y+1).RGBA() r10, g10, b10, a10 := inputImage.At(x+1, y).RGBA() r11, g11, b11, a11 := inputImage.At(x+1, y+1).RGBA() col.R = uint8((uint(uint8(r00)) + uint(uint8(r01)) + uint(uint8(r10)) + uint(uint8(r11))) / 4) col.G = uint8((uint(uint8(g00)) + uint(uint8(g01)) + uint(uint8(g10)) + uint(uint8(g11))) / 4) col.B = uint8((uint(uint8(b00)) + uint(uint8(b01)) + uint(uint8(b10)) + uint(uint8(b11))) / 4) col.A = uint8((uint(uint8(a00)) + uint(uint8(a01)) + uint(uint8(a10)) + uint(uint8(a11))) / 4) rgba.Set(x, y, col) } } return rgba.SubImage(rect) }
func main() { img := shapes.FilledImage(500, 300, image.White) fill := color.RGBA{200, 200, 200, 255} for i := 0; i < 10; i++ { width, height := 50+(20*i), 30+(10*i) rect, err := shapes.New("rectangle", shapes.Option{Fill: fill, Rect: image.Rect(0, 0, width, height), Filled: true}) if err != nil { log.Fatal(err) } x := 10 + (20 * i) for j := i / 2; j >= 0; j-- { rect.Draw(img, x+j, (x/2)+j) } fill.R -= uint8(i * 5) fill.G = fill.R fill.B = fill.R } shapes.SaveImage(img, "rectangle.png") }
func (cf *ColorFinder) buildColorMap() *map[color.RGBA]colorStats { colorMap := make(map[color.RGBA]colorStats) bounds := cf.img.Bounds() for y := bounds.Min.Y; y < bounds.Max.Y; y++ { for x := bounds.Min.X; x < bounds.Max.X; x++ { r, g, b, a := cf.img.At(x, y).RGBA() rgb := color.RGBA{} rgb.R = uint8(r >> shiftRGB) rgb.G = uint8(g >> shiftRGB) rgb.B = uint8(b >> shiftRGB) rgb.A = uint8(a >> shiftRGB) colrStats, exist := colorMap[rgb] if exist { colrStats.count++ } else { colrStats := colorStats{count: 1, weight: weight(&rgb)} if colrStats.weight <= 0 { colrStats.weight = 1e-10 } colorMap[rgb] = colrStats } } } return &colorMap }
func (cf *ColorFinder) findMainColor(colorMap *map[color.RGBA]colorStats, shift uint, targetColor *shiftedRGBA) shiftedRGBA { colorWeights := make(map[shiftedRGBA]float64) bounds := cf.img.Bounds() stepLength := stepLength(bounds) for y := bounds.Min.Y; y < bounds.Max.Y; y += stepLength { for x := bounds.Min.X; x < bounds.Max.X; x += stepLength { r, g, b, a := cf.img.At(x, y).RGBA() color := color.RGBA{} color.R = uint8(r >> shiftRGB) color.G = uint8(g >> shiftRGB) color.B = uint8(b >> shiftRGB) color.A = uint8(a >> shiftRGB) if rgbMatchesTargetColor(targetColor, &color) { increaseColorWeight(&colorWeights, colorMap, &color, shift) } } } maxColor := shiftedRGBA{} maxWeight := 0.0 for sRGB, weight := range colorWeights { if weight > maxWeight { maxColor = sRGB maxWeight = weight } } return maxColor }
// Return the complementary color func Complementary(v color.RGBA) [2]color.RGBA { var result color.RGBA result.R = 255 - v.R result.G = 255 - v.G result.B = 255 - v.B result.A = v.A return [2]color.RGBA{v, result} }
// Apply the given color mapping to the specified image buffers. func Apply(from, to *Rule, src, dst draw.Image) { var x, y int var r, g, b, a uint32 var sc color.Color var dc color.RGBA var pix pixel rect := src.Bounds() for y = 0; y < rect.Dy(); y++ { for x = 0; x < rect.Dx(); x++ { sc = src.At(x, y) r, g, b, a = sc.RGBA() pix.r = uint8(r >> 8) pix.g = uint8(g >> 8) pix.b = uint8(b >> 8) pix.a = uint8(a >> 8) // Check if the pixel matches the filter rule. if !(match(pix.r, from.R) && match(pix.g, from.G) && match(pix.b, from.B) && match(pix.a, from.A)) { dst.Set(x, y, sc) continue } // Compute three different types of grayscale conversion. // These can be applied by named references. pix.average = uint8(((r + g + b) / 3) >> 8) pix.lightness = uint8(((min(min(r, g), b) + max(max(r, g), b)) / 2) >> 8) // For luminosity it is necessary to apply an inverse of the gamma // function for the color space before calculating the inner product. // Then you apply the gamma function to the reduced value. Failure to // incorporate the gamma function can result in errors of up to 20%. // // For typical computer stuff, the color space is sRGB. The right // numbers for sRGB are approx. 0.21, 0.72, 0.07. Gamma for sRGB // is a composite function that approximates exponentiation by 1/2.2 // // This is a rather expensive operation, but gives a much more accurate // and satisfactory result than the average and lightness versions. pix.luminosity = gammaSRGB( 0.212655*invGammaSRGB(pix.r) + 0.715158*invGammaSRGB(pix.g) + 0.072187*invGammaSRGB(pix.b)) // Transform color. dc.R = transform(&pix, pix.r, to.R) dc.G = transform(&pix, pix.g, to.G) dc.B = transform(&pix, pix.b, to.B) dc.A = transform(&pix, pix.a, to.A) // Set new pixel. dst.Set(x, y, dc) } } }
func setAlpha(c color.RGBA, a uint8) color.RGBA { if c.A == 0 { return c } c.R = mult(c.R, a, c.A) c.G = mult(c.G, a, c.A) c.B = mult(c.B, a, c.A) c.A = a return c }
// SetAll sets the given values for the channels func (d *Dioder) SetAll(colorSet color.RGBA) { d.ColorConfiguration = colorSet //Red colorSet.R = calculateOpacity(colorSet.R, colorSet.A) d.SetChannelInteger(colorSet.R, d.PinConfiguration.Red) //Green colorSet.G = calculateOpacity(colorSet.G, colorSet.A) d.SetChannelInteger(colorSet.G, d.PinConfiguration.Green) //Blue colorSet.B = calculateOpacity(colorSet.B, colorSet.A) d.SetChannelInteger(colorSet.B, d.PinConfiguration.Blue) }
func drawworld2(translate *noise.Translate, w *worldgen2.World, img draw.Image) { w.Log = false bounds := img.Bounds() for y := bounds.Min.Y; y < bounds.Max.Y; y++ { for x := bounds.Min.X; x < bounds.Max.X; x++ { var c color.RGBA c.A = 255 posx, posy := translate.Into(float64(x), float64(y)) per := w.GetHeight_Island(posx, posy) if per < 0 { v := uint8(255 - 127*(-per)) c.R = 0 c.G = 0 c.B = v } else if per < 0.00002 { v := uint8(255 - 2550*per) c.R = v c.G = v c.B = 0 } else if per > 0.34 { v := uint8(220 * (per)) c.R = v c.G = v c.B = v } else { //v := uint8(255-per*255) v := uint8(255 - per*300) c.R = 0 c.G = v c.B = 0 } img.Set(x, y, c) } } w.Log = true }
func parseColors(args []*ast.BasicLit) (color.RGBA, error) { ints := make([]uint8, 4) var ret color.RGBA var u uint8 for i := range args { v := args[i] switch v.Kind { case token.FLOAT: f, err := strconv.ParseFloat(args[i].Value, 8) if err != nil { return ret, err } // Has to be alpha, or bust u = uint8(f * 100) case token.INT: i, err := strconv.Atoi(v.Value) if err != nil { return ret, err } u = uint8(i) case token.COLOR: if i != 0 { return ret, fmt.Errorf("hex is only allowed as the first argumetn found: % #v", v) } var err error ret, err = ast.ColorFromHexString(v.Value) if err != nil { return ret, err } // This is only allowed as the first argument i = i + 2 default: log.Fatalf("unsupported kind %s % #v\n", v.Kind, v) } ints[i] = u } if ints[0] > 0 { ret.R = ints[0] } if ints[1] > 0 { ret.G = ints[1] } if ints[2] > 0 { ret.B = ints[2] } if ints[3] > 0 { ret.A = ints[3] } return ret, nil }
func blend(orig image.Image) (m image.Image) { // deviation range dev := uint16(50 << 8) c := new(color.RGBA) c.A = 255 bounds := orig.Bounds() // iterations for i := 0; i < 10; i++ { newm := image.NewRGBA(bounds) for y := bounds.Min.Y; y < bounds.Max.Y; y++ { for x := bounds.Min.X; x < bounds.Max.X; x++ { var r, g, b, ct uint32 _r, _g, _b, _ := orig.At(x, y).RGBA() for i := -1; i < 2; i++ { for j := -1; j < 2; j++ { rt, gt, bt, _ := orig.At(x+i, y+j).RGBA() if uint16(rt-_r) > dev || uint16(gt-_g) > dev || uint16(bt-_b) > dev { continue } r += rt g += gt b += bt ct++ } } c.R = uint8((r / ct) >> 8) c.G = uint8((g / ct) >> 8) c.B = uint8((b / ct) >> 8) newm.Set(x, y, c) } } m = newm } return }
func main() { img := shapes.FilledImage(420, 220, image.White) fill := color.RGBA{200, 200, 200, 0xFF} // light gray for i := 0; i < 10; i++ { width, height := 40+(20*i), 20+(10*i) rectangle := shapes.Rectangle{fill, image.Rect(0, 0, width, height), true} x := 10 + (20 * i) for j := i / 2; j >= 0; j-- { rectangle.Draw(img, x+j, (x/2)+j) } fill.R -= uint8(i * 5) fill.G = fill.R fill.B = fill.R } shapes.SaveImage(img, "rectangle.png") }
func DrawImage(src image.Image, dest draw.Image, tr MatrixTransform, op draw.Op, filter ImageFilter) { bounds := src.Bounds() x0, y0, x1, y1 := float64(bounds.Min.X), float64(bounds.Min.Y), float64(bounds.Max.X), float64(bounds.Max.Y) tr.TransformRectangle(&x0, &y0, &x1, &y1) var x, y, u, v float64 var c1, c2, cr color.Color var r, g, b, a, ia, r1, g1, b1, a1, r2, g2, b2, a2 uint32 var color color.RGBA for x = x0; x < x1; x++ { for y = y0; y < y1; y++ { u = x v = y tr.InverseTransform(&u, &v) if bounds.Min.X <= int(u) && bounds.Max.X > int(u) && bounds.Min.Y <= int(v) && bounds.Max.Y > int(v) { c1 = dest.At(int(x), int(y)) switch filter { case LinearFilter: c2 = src.At(int(u), int(v)) case BilinearFilter: c2 = getColorBilinear(src, u, v) case BicubicFilter: c2 = getColorBicubic(src, u, v) } switch op { case draw.Over: r1, g1, b1, a1 = c1.RGBA() r2, g2, b2, a2 = c2.RGBA() ia = M - a2 r = ((r1 * ia) / M) + r2 g = ((g1 * ia) / M) + g2 b = ((b1 * ia) / M) + b2 a = ((a1 * ia) / M) + a2 color.R = uint8(r >> 8) color.G = uint8(g >> 8) color.B = uint8(b >> 8) color.A = uint8(a >> 8) cr = color default: cr = c2 } dest.Set(int(x), int(y), cr) } } } }
// linterp performs linear interpolation between the first (p[0]) and // the last (pal[len(p) - 1]) colors in palette "pal". It fills the // remaining (pal[1:len(p) - 1]) palette slots with the intepolated // colors. Colors in slots pal[0] and pal[len(p) - 1] MUST be of type // color.RGBA. The remaining slots will also be filled with colors of // type color.RGBA func linterp(pal color.Palette) { n := len(pal) if n <= 2 { return } s, e := pal[0].(color.RGBA), pal[n-1].(color.RGBA) dr := int(e.R) - int(s.R) dg := int(e.G) - int(s.G) db := int(e.B) - int(s.B) da := int(e.A) - int(s.A) for i := 1; i < n-1; i++ { c := color.RGBA{} c.R = uint8(int(s.R) + i*dr/(n-1)) c.G = uint8(int(s.G) + i*dg/(n-1)) c.B = uint8(int(s.B) + i*db/(n-1)) c.A = uint8(int(s.A) + i*da/(n-1)) pal[i] = c } }
//Takes a list of words (i.e. from a book), finds words that match the search term, //and creates an image based on the usage of those words with each pixel representing //the specified number of words (sectionSize). The channel will be sent a value of 1 //upon the successful completion of the task func CreateImage(words []string, searchTerm string, sectionSize int, quit chan int) { searchTerms := strings.Fields(searchTerm) imageHeight := 20 //Setup the image totalWords := len(words) imageWidth := int(math.Ceil(float64(totalWords) / float64(sectionSize))) newImg := image.NewRGBA(image.Rect(0, 0, imageWidth, imageHeight)) usedInSection := make([]bool, imageWidth, imageWidth) usedCount := 0 for index, word := range words { section := int(math.Floor(float64(index) / float64(sectionSize))) for _, term := range searchTerms { if word == term { usedInSection[section] = true usedCount++ break } } } fmt.Println("The terms", searchTerm, "were found", usedCount, "times") for x := 0; x < imageWidth; x++ { color := color.RGBA{255, 255, 255, 255} if usedInSection[x] == true { color.R = 0 color.G = 0 color.B = 0 } for y := 0; y < imageHeight; y++ { newImg.SetRGBA(x, y, color) } } SaveImage(searchTerm+".png", newImg) quit <- 1 }
// colorOpColor combines two colors with the requested operation // applying appropriate overflows as Sass expects func colorOpColor(tok token.Token, x, y *BasicLit, combine bool) (*BasicLit, error) { colX, err := ColorFromHexString(x.Value) if err != nil { return nil, err } colY, err := ColorFromHexString(y.Value) if err != nil { return nil, err } var z color.RGBA z.R = overflowMath(tok, colX.R, colY.R) z.G = overflowMath(tok, colX.G, colY.G) z.B = overflowMath(tok, colX.B, colY.B) s := colorToHex(z) return &BasicLit{ Kind: token.COLOR, Value: LookupColor(s), }, nil }
// Compare compares a and b using binary comparison. func (d *binary) Compare(a, b image.Image) (image.Image, int, error) { ab, bb := a.Bounds(), b.Bounds() w, h := ab.Dx(), ab.Dy() if w != bb.Dx() || h != bb.Dy() { return nil, -1, ErrSize } diff := image.NewNRGBA(image.Rect(0, 0, w, h)) n := 0 for y := 0; y < h; y++ { for x := 0; x < w; x++ { d := diffColor(a.At(ab.Min.X+x, ab.Min.Y+y), b.At(bb.Min.X+x, bb.Min.Y+y)) c := color.RGBA{0, 0, 0, 0xff} if d > 0 { c.R = 0xff //c.A = uint8(100 + d*0xff/0xffff) n++ } diff.Set(x, y, c) } } return diff, n, nil }
func main() { start := time.Now() const pictureSize = 5000 img := image.NewRGBA(image.Rect(0, 0, pictureSize, pictureSize)) f, err := os.OpenFile("mandel.png", os.O_WRONLY|os.O_CREATE, 0666) if err != nil { fmt.Printf("Can't create picture file\n") } deltaX := math.Abs(float64(-2.0-1.0)) / float64(pictureSize) deltaY := math.Abs(float64(-1.0-1.0)) / float64(pictureSize) cx := float64(-2.0) for x := 0; x < pictureSize; x++ { cx += deltaX cy := float64(-1.0) for y := 0; y < pictureSize; y++ { cy += deltaY iter := PointIteration(cx, cy, 255.0, 255) col := color.RGBA{255, iter, iter, iter} col.A = 255 col.R = iter col.G = iter col.B = iter img.Set(x, y, col) } } w := bufio.NewWriter(f) png.Encode(w, img) w.Flush() fmt.Printf("Seconds needed %d\n", time.Now().Sub(start)) }
func (bilinear) RGBA(src *image.RGBA, x, y float64) color.RGBA { p := findLinearSrc(src.Bounds(), x, y) // Array offsets for the surrounding pixels. off00 := offRGBA(src, p.low.X, p.low.Y) off01 := offRGBA(src, p.high.X, p.low.Y) off10 := offRGBA(src, p.low.X, p.high.Y) off11 := offRGBA(src, p.high.X, p.high.Y) var fr, fg, fb, fa float64 fr += float64(src.Pix[off00+0]) * p.frac00 fg += float64(src.Pix[off00+1]) * p.frac00 fb += float64(src.Pix[off00+2]) * p.frac00 fa += float64(src.Pix[off00+3]) * p.frac00 fr += float64(src.Pix[off01+0]) * p.frac01 fg += float64(src.Pix[off01+1]) * p.frac01 fb += float64(src.Pix[off01+2]) * p.frac01 fa += float64(src.Pix[off01+3]) * p.frac01 fr += float64(src.Pix[off10+0]) * p.frac10 fg += float64(src.Pix[off10+1]) * p.frac10 fb += float64(src.Pix[off10+2]) * p.frac10 fa += float64(src.Pix[off10+3]) * p.frac10 fr += float64(src.Pix[off11+0]) * p.frac11 fg += float64(src.Pix[off11+1]) * p.frac11 fb += float64(src.Pix[off11+2]) * p.frac11 fa += float64(src.Pix[off11+3]) * p.frac11 var c color.RGBA c.R = uint8(fr + 0.5) c.G = uint8(fg + 0.5) c.B = uint8(fb + 0.5) c.A = uint8(fa + 0.5) return c }
func (me *renderTile) getSample(x, y, z int, col *color.RGBA) { if (x >= SceneSize) || (y >= SceneSize) || (z >= SceneSize) { col.R, col.G, col.B, col.A = 0, 0, 0, 0 } else if colorCube { if (num.IsEveni(x) && num.IsEveni(y)) || ((z > 0) && num.IsEveni(z)) { col.R, col.G, col.B, col.A = uint8(x/2), uint8(y/2), uint8(z/2), 255 } else { col.R, col.G, col.B, col.A = uint8(x), uint8(y), uint8(z), 255 } } else { me.voxel = Scene[x][y][z] if me.voxel.A >= 1 { if correctAlpha { me.mixCol(col, &me.voxel, col) } else { col.R = mixRGB(col.R, me.voxel.R, col.A, me.voxel.A) // num.Mixb(me.voxel.R, me.voxel.A, col.R) col.G = mixRGB(col.G, me.voxel.G, col.A, me.voxel.A) // num.Mixb(me.voxel.G, me.voxel.A, col.G) col.B = mixRGB(col.B, me.voxel.B, col.A, me.voxel.A) // num.Mixb(me.voxel.B, me.voxel.A, col.B) col.A = mixA(col.A, me.voxel.A) // num.Mixb(1, col.A, me.voxel.A) } } } }
func imfilterMerg(filter [][]float32, fX, fY, x, y, xoffset, yoffset int, rgbFunc func(xpos, ypos int) (r0, g0, b0, a0 uint8), mergFunc func(a, b float32) float32) color.RGBA { _, _, _, a := rgbFunc(x, y) newColor := color.RGBA{R: 0, G: 0, B: 0, A: a} for xx := 0; xx < fX; xx++ { for yy := 0; yy < fY; yy++ { r, g, b, _ := rgbFunc(x+xx-xoffset, y+yy-yoffset) newColor.R = oneColorCorrect(float32(newColor.R) + mergFunc(filter[xx][yy], float32(r))) if r != g { newColor.G = oneColorCorrect(float32(newColor.G) + mergFunc(filter[xx][yy], float32(g))) } else { newColor.G = newColor.R } switch b { case r: newColor.B = newColor.R case g: newColor.B = newColor.G default: newColor.B = oneColorCorrect(float32(newColor.B) + mergFunc(filter[xx][yy], float32(b))) } } } return newColor }
func main() { in := flag.String("i", "none", "input file") out := flag.String("o", "out.png", "output file") flag.Parse() if *in == "none" { return } bin := read(*in) binLength := len(bin) width := 384 height := len(bin) / width rect := image.Rect(0, 0, width, height) img := image.NewRGBA(rect) p := 0 for y := 0; y < height; y++ { for x := 0; x < width; x++ { var c color.RGBA for addr := p; addr < p+3 && addr < binLength; addr++ { if addr%3 == 0 { c.R = uint8(bin[addr]) } else if addr%3 == 1 { c.G = uint8(bin[addr]) } else if addr%3 == 2 { c.B = uint8(bin[addr]) } c.A = 0xff } p += 3 img.Set(x, y, c) } } f, _ := os.OpenFile(*out, os.O_CREATE|os.O_WRONLY, 0666) png.Encode(f, img) }
func featurize(orig image.Image) image.Image { // deviation range color_dev := uint16(20 << 8) feature_dev := 4 bounds := orig.Bounds() fmt.Println(bounds) // initialize the array of pixels traversed ex := make([][]bool, bounds.Max.X+1) for i := 0; i < bounds.Max.X+1; i++ { ex[i] = make([]bool, bounds.Max.Y+1) } // array of features // feature is an array of arrays [x,y] var features [][][2]int feature_i := 0 // recursively investigate all neighbors of supplied pixel // and build feature arrays // some shenanigans to make anonymous functions recurisively callable var discover func(x int, y int) discover = func(x int, y int) { // add this pixel to the current feature features[feature_i] = append(features[feature_i], [2]int{x, y}) // fmt.Println(x,y) ex[x][y] = true // grab the rgb for the supplied pixel _r, _g, _b, _ := orig.At(x, y).RGBA() for i := -1; i < 2; i++ { for j := -1; j < 2; j++ { xx, yy := x+i, y+j // check if it's within our bounds and if it's been processed already if xx < 0 || yy < 0 || xx > bounds.Max.X || yy > bounds.Max.Y || ex[xx][yy] == true { continue } rt, gt, bt, _ := orig.At(xx, yy).RGBA() // check the color range against our deviation spec if uint16(rt-_r) > color_dev || uint16(gt-_g) > color_dev || uint16(bt-_b) > color_dev { continue } discover(xx, yy) } } } // run through each pixel and build features for y := bounds.Min.Y; y < bounds.Max.Y; y++ { for x := bounds.Min.X; x < bounds.Max.X; x++ { // skip if already processed if ex[x][y] == true { continue } // fmt.Println(x,y) features = append(features, [][2]int{}) discover(x, y) feature_i++ } } newm := image.NewRGBA(bounds) c := new(color.RGBA) c.A = 255 for f := 0; f < len(features); f++ { // if the feature is large enough // average all pixel colors in it // and set all pixels to that color if len(features[f]) > feature_dev { var r, g, b, ct uint64 for p := 0; p < len(features[f]); p++ { rt, gt, bt, _ := orig.At(features[f][p][0], features[f][p][1]).RGBA() r += uint64(rt) g += uint64(gt) b += uint64(bt) ct++ } c.R = uint8((r / ct) >> 8) c.G = uint8((g / ct) >> 8) c.B = uint8((b / ct) >> 8) for p := 0; p < len(features[f]); p++ { newm.Set(features[f][p][0], features[f][p][1], c) } } else { // write the pixel out as-is for p := 0; p < len(features[f]); p++ { r, g, b, _ := orig.At(features[f][p][0], features[f][p][1]).RGBA() c.R = uint8(r >> 8) c.G = uint8(g >> 8) c.B = uint8(b >> 8) newm.Set(features[f][p][0], features[f][p][1], c) } } } return newm }
func main() { flag.Parse() // Set initial values. shades_per_iteration = float64(max_color) / float64(*max_iterations) var ( mset_x_total float64 = 3.5 mset_y_total float64 = 2.0 ) mset_x_min := *mset_x_ctr - (mset_x_total / 2) mset_x_max := *mset_x_ctr + (mset_x_total / 2) mset_y_min := *mset_y_ctr - (mset_y_total / 2) mset_y_max := *mset_y_ctr + (mset_y_total / 2) // Set coordinates according to xoom factor. mset_y_total = (mset_y_max - mset_y_min) / *zoom mset_x_total = (mset_x_max - mset_x_min) / *zoom mset_x_min = *mset_x_ctr - (mset_x_total / 2) mset_x_max = *mset_x_ctr + (mset_x_total / 2) mset_y_min = *mset_y_ctr - (mset_y_total / 2) mset_y_max = *mset_y_ctr + (mset_y_total / 2) mset_ratio := mset_x_total / mset_y_total img_ratio := float64(*img_width) / float64(*img_height) if mset_ratio > img_ratio { // The image is taller than the default mandelbrot set. // Therefore, the limiting dimension is the height of the mandelbrot set. // Don't touch the Y figures. // Reset the X limits. mset_y_total = mset_y_max - mset_y_min mset_x_total = mset_y_total * img_ratio mset_x_min = *mset_x_ctr - (mset_x_total / 2) mset_x_max = *mset_x_ctr + (mset_x_total / 2) } else { // The image is flatter than the default mandelbrot set. // Therefore, the limiting dimension is the width of the mandelbrot set. // Don't touch the X figures. // Reset the Y limits. mset_x_total = mset_x_max - mset_x_min mset_y_total = mset_x_total / img_ratio mset_y_min = *mset_y_ctr - (mset_y_total / 2) mset_y_max = *mset_y_ctr + (mset_y_total / 2) } // Set slope of conversion lines. // m = rise/run m_x = mset_x_total / float64(*img_width) m_y = mset_y_total / float64(*img_height) b_x = mset_x_min b_y = mset_y_min img := image.NewRGBA(image.Rect(0, 0, *img_width, *img_height)) // img := image.NewGray16(image.Rect(0, 0, *img_width, *img_height)) // img := image.NewGray(image.Rect(0, 0, *img_width, *img_height)) var ( c int64 c_max int64 = 0 clr color.RGBA // clr color.Gray16 // clr color.Gray // i int64 i_max int64 = 0 mpt MPT ) // mpt = new(MPT) for x := 0; x < *img_width; x++ { log.Printf("Percent Complete: %.1f", float64(x)/float64(*img_width)*100) for y := 0; y < *img_height; y++ { x_, y_ := getCoord(x, y) mpt.X0 = x_ mpt.Y0 = y_ mpt.X = 0 mpt.Y = 0 mpt.N = 0 mpt.iterate() factor_iterations := float64(mpt.N) / float64(*max_iterations) // This is a decimal revealing the scale based on iterations. factor_radius := math.Log2(mpt.R) / 2 c = int64((factor_iterations + factor_radius) * float64(*max_iterations) * shades_per_iteration) // c = max_color - c clr.G = uint8(c) clr.B = uint8(c >> 8) clr.R = uint8(c >> 16) clr.A = 255 // log.Printf("C: %b\n", c) // log.Printf("R: %b\n", color.R) // log.Printf("G: %b\n", color.G) // log.Printf("B: %b\n", color.B) // clr.Y = uint8(c) // The golange image is inverted on the y-axis compared to typical cartesian systems. img.Set(x, *img_height-y, clr) } } log.Printf("img_width: %d\n", *img_width) log.Printf("img_height: %d\n", *img_height) log.Printf("mset_x_min: %0.2f\n", mset_x_min) log.Printf("mset_x_max: %0.2f\n", mset_x_max) log.Printf("mset_x_total: %0.2f\n", mset_x_total) log.Printf("mset_y_min: %0.2f\n", mset_y_min) log.Printf("mset_y_max: %0.2f\n", mset_y_max) log.Printf("mset_y_total: %0.2f\n", mset_y_total) log.Printf("m_x: %0.8f\n", m_x) log.Printf("m_y: %0.8f\n", m_y) log.Printf("i_max: %d\n", i_max) log.Printf("c_max: %d\n", c_max) log.Println("Encoding image.") buffer := new(bytes.Buffer) if err := png.Encode(buffer, img); err != nil { log.Fatal(err.Error()) } log.Println("Writing image.") if err := ioutil.WriteFile(fmt.Sprintf("/Users/jackman/Pictures/mandelbrot(%.4f,%.4f)z%.1f[%dx%dz]x%d.png", *mset_x_ctr, *mset_y_ctr, *zoom, *img_width, *img_height, *max_iterations), buffer.Bytes(), 0644); err != nil { log.Fatal(err.Error()) } }
// Find returns the dominant color in img. func Find(img image.Image) color.RGBA { // Shrink image for faster processing. img = resize.Thumbnail(resizeTo, resizeTo, img, resize.NearestNeighbor) bounds := img.Bounds() width, height := bounds.Dx(), bounds.Dy() rnd := rand.New(rand.NewSource(0)) randomPoint := func() (x, y int) { x = bounds.Min.X + rnd.Intn(width) y = bounds.Min.Y + rnd.Intn(height) return } // Pick a starting point for each cluster. clusters := make(kMeanClusterGroup, 0, nCluster) for i := 0; i < nCluster; i++ { // Try up to 10 times to find a unique color. If no unique color can be // found, destroy this cluster. colorUnique := false for j := 0; j < maxSample; j++ { ri, gi, bi, a := img.At(randomPoint()).RGBA() // Ignore transparent pixels. if a == 0 { continue } r, g, b := uint8(ri/255), uint8(gi/255), uint8(bi/255) // Check to see if we have seen this color before. colorUnique = !clusters.ContainsCentroid(r, g, b) // If we have a unique color set the center of the cluster to // that color. if colorUnique { c := new(kMeanCluster) c.SetCentroid(r, g, b) clusters = append(clusters, c) break } } if !colorUnique { break } } convergence := false for i := 0; i < nIterations && !convergence && len(clusters) != 0; i++ { for x := bounds.Min.X; x < bounds.Max.X; x++ { for y := bounds.Min.Y; y < bounds.Max.Y; y++ { ri, gi, bi, a := img.At(x, y).RGBA() // Ignore transparent pixels. if a == 0 { continue } r, g, b := uint8(ri/255), uint8(gi/255), uint8(bi/255) // Figure out which cluster this color is closest to in RGB space. closest := clusters.Closest(r, g, b) closest.AddPoint(r, g, b) } } // Calculate the new cluster centers and see if we've converged or not. convergence = true for _, c := range clusters { convergence = convergence && c.CompareCentroidWithAggregate() c.RecomputeCentroid() } } // Sort the clusters by population so we can tell what the most popular // color is. sort.Sort(byWeight(clusters)) // Loop through the clusters to figure out which cluster has an appropriate // color. Skip any that are too bright/dark and go in order of weight. var col color.RGBA for i, c := range clusters { r, g, b := c.Centroid() // Sum the RGB components to determine if the color is too bright or too dark. var summedColor uint16 = uint16(r) + uint16(g) + uint16(b) if summedColor < maxBrightness && summedColor > minDarkness { // If we found a valid color just set it and break. We don't want to // check the other ones. col.R = r col.G = g col.B = b col.A = 0xFF break } else if i == 0 { // We haven't found a valid color, but we are at the first color so // set the color anyway to make sure we at least have a value here. col.R = r col.G = g col.B = b col.A = 0xFF } } return col }
func init_calc(x, y int) color.RGBA { //t_ray ray var x1, y1, z1 float32 var pixel color.RGBA pixel.R = 255 pixel.A = 255 x1 = float32(D) y1 = float32((WinX / 2) - x) z1 = float32((WinY / 2) - y) /* ray.Vx = x1 - eye->pos.X; ray.Vy = y1 - eye->pos.Y; ray.Vz = z1 - eye->pos.Z; ray.top_mesh = llist; ray.top_spot = spot; ray.bool = 1; rotate_x(&ray.Vx, &ray.Vy, &ray.Vz, -eye->rot.X); rotate_y(&ray.Vx, &ray.Vy, &ray.Vz, -eye->rot.Y); rotate_z(&ray.Vx, &ray.Vy, &ray.Vz, -eye->rot.Z); color = calc(eye, llist, spot, &ray); */ return pixel }