change example inage

.gitignore

@@ -1,3 +1,3 @@
 barnsley-fern
 barnsley-fern-go
-barnsley-fern.png
+barnsley_fern.png

Makefile

@@ -1,8 +0,0 @@
-.PHONY: build, clean
-
-build:
-	go build -v 
-
-clean:
-	rm -f barnsley-fern-go barnsley-fern.png
-

README.md

@@ -1,7 +1,7 @@
 # barnsley-fern-go
 Implementation of Barnsley Fern in Go
 
-![Barnsley Fern](https://code.uint32.ru/dmitry/barnsley-fern-go/raw/branch/main/example/barnsley-fern.png "Fern")
+![Barnsley Fern](https://code.uint32.ru/dmitry/barnsley-fern-go/raw/branch/main/barnsley_fern.png "Fern")
 
 From the root of the repository
 ```bash
@@ -16,4 +16,4 @@ This will generate an image in current directory.
 
 See **./barnsley-fern --help** for available flags.
 
-Image on this page was generated with **./barnsley-fern-go -x 500 -y 500 -c 3 -out example/barnsley-fern.png**
+Image on this page was generated with **./barnsley-fern-go -h 500 -v 500 -rainbow**

barnsley-fern.go

@@ -1,12 +1,89 @@
 package main
 
 import (
+	"flag"
+	"fmt"
 	"image"
+	"image/color"
+	"image/png"
 	"math/rand"
+	"os"
 	"time"
 )
 
-func drawBarnsleyFern(img *image.RGBA, colorfunc colorFunc, dots int) {
+func createImage(h, v int) *image.RGBA {
+	rect := image.Rectangle{
+		image.Point{0, 0},
+		image.Point{h, v},
+	}
+	return image.NewRGBA(rect)
+}
+
+func fillBackground(img *image.RGBA, c color.Color) {
+	rect := img.Bounds()
+	for x := range rect.Max.X {
+		for y := range rect.Max.Y {
+			img.Set(x, y, c)
+		}
+	}
+}
+
+func newColorFunc(random bool, timed bool, rainbow bool) func(float64, float64) color.Color {
+	return func(x, y float64) color.Color {
+		var (
+			r, g, b, a uint8
+		)
+
+		switch {
+		case random:
+			// n := rand.Intn(3)
+
+			// switch n {
+			// case 0:
+			// 	r = 255
+			// case 1:
+			// 	g = 255
+			// case 2:
+			// 	b = 255
+			// }
+
+			r = uint8(rand.Intn(256))
+			g = uint8(rand.Intn(256))
+			b = uint8(rand.Intn(256))
+			a = 255
+
+		case timed:
+			n := time.Now().Nanosecond() % 256 // Nanosecond returns range [0, 999999999]
+
+			r = uint8((n + 256) % 256)
+			g = uint8((n + 128) % 256)
+			b = uint8(n % 256)
+
+			a = 255
+
+		case rainbow:
+			r = uint8(y * 256)
+			g = uint8(x * 256)
+			b = uint8(((y - x) - min(x, y)) * 256)
+
+			a = 255
+
+		default:
+			r, g, b, a = 0, 153, 0, 255
+		}
+
+		return color.RGBA{r, g, b, a}
+	}
+}
+
+func min(a, b float64) float64 {
+	if a > b {
+		a = b
+	}
+	return a
+}
+
+func drawBarnsleyFern(img *image.RGBA, colorfunc func(float64, float64) color.Color, dots int) {
 	var (
 		x, y, tmpx, tmpy, r, maxy, maxx, scale, yoffset, xoffset float64
 	)
@@ -41,15 +118,47 @@ func drawBarnsleyFern(img *image.RGBA, colorfunc colorFunc, dots int) {
 		// рисуем точку
 		x, y = tmpx, tmpy
 
-		color := colorfunc(x, y, time.Now())
+		color := colorfunc(x, y)
 
 		img.Set(int(xoffset+x*scale), int(maxy-yoffset-y*scale), color)
 	}
 }
 
-func min(a, b float64) float64 {
-	if a > b {
-		a = b
+func main() {
+	var (
+		filename string = "barnsley_fern.png"
+		h               = flag.Int("h", 1920, "размер картинки по горизонтали")
+		v               = flag.Int("v", 1080, "размер картинки по вертикали")
+		dots            = flag.Int("d", 100000, "сколько точек рисовать")
+		rand            = flag.Bool("rand", false, "использовать случайные цвета")
+		timed           = flag.Bool("timed", false, "привязка к наносекундам времени исполнения")
+		rainbow         = flag.Bool("rainbow", false, "привязка к координатам")
+		img      *image.RGBA
+		f        *os.File
+		err      error
+	)
+	flag.Parse()
+
+	img = createImage(*h, *v)
+
+	fillBackground(img, color.White)
+
+	colorfunc := newColorFunc(*rand, *timed, *rainbow)
+
+	drawBarnsleyFern(img, colorfunc, *dots)
+
+	f, err = os.Create(filename)
+	if err != nil {
+		fmt.Printf("не удалось создать файл %s: %v", filename, err)
+		return
 	}
-	return a
+
+	defer f.Close()
+
+	if err = png.Encode(f, img); err != nil {
+		fmt.Printf("не удалось сохранить изображение: %v", err)
+		return
+	}
+
+	fmt.Println("done...")
 }

color.go

@@ -1,47 +0,0 @@
-package main
-
-import (
-	"image/color"
-	"math/rand"
-	"time"
-)
-
-type colorFunc func(float64, float64, time.Time) color.Color
-
-func newColorFunc(c colormode) colorFunc {
-	start := time.Now()
-	return func(x, y float64, t time.Time) color.Color {
-		var (
-			r, g, b, a uint8
-		)
-
-		switch c {
-		case crandom:
-			r = uint8(rand.Intn(256))
-			g = uint8(rand.Intn(256))
-			b = uint8(rand.Intn(256))
-			a = 255
-
-		case ctimed:
-			n := time.Since(start).Nanoseconds()
-
-			r = uint8(n % 256)
-			g = uint8((n + 128) % 256)
-			//b = uint8((n + 1) % 256)
-
-			a = 255
-
-		case crainbow:
-			r = uint8(y * 256)
-			g = uint8(x * 256)
-			b = uint8(((y - x) - min(x, y)) * 256)
-
-			a = 255
-
-		default:
-			r, g, b, a = 0, 153, 0, 255
-		}
-
-		return color.RGBA{r, g, b, a}
-	}
-}

go.mod

@@ -1,5 +1,3 @@
 module code.uint32.ru/dmitry/barnsley-fern-go
 
 go 1.24.1
-
-require github.com/AlexEidt/Vidio v1.5.1 // indirect

go.sum

@@ -1,2 +0,0 @@
-github.com/AlexEidt/Vidio v1.5.1 h1:tovwvtgQagUz1vifiL9OeWkg1fP/XUzFazFKh7tFtaE=
-github.com/AlexEidt/Vidio v1.5.1/go.mod h1:djhIMnWMqPrC3X6nB6ymGX6uWWlgw+VayYGKE1bNwmI=

image.go

@@ -1,68 +0,0 @@
-package main
-
-import (
-	"errors"
-	"fmt"
-	"image"
-	"image/color"
-	"image/jpeg"
-	"image/png"
-	"io"
-)
-
-var (
-	errEncodeImage   = errors.New("не удалось создать изображение")
-	errUnknownFormat = errors.New("неизвестный формат изображения")
-)
-
-func createBarnsleyFernImage(s *settings, w io.Writer) error {
-	img := newImageWithBackGround(s.X, s.Y, s.BG)
-
-	drawBarnsleyFern(img, newColorFunc(s.CM), s.Dots)
-
-	switch s.OM {
-	case outputjpeg:
-		opts := &jpeg.Options{
-			Quality: 100,
-		}
-		if err := jpeg.Encode(w, img, opts); err != nil {
-			return errors.Join(errEncodeImage, err)
-		}
-
-	case outputpng:
-		if err := png.Encode(w, img); err != nil {
-			return errors.Join(errEncodeImage, err)
-		}
-
-	default:
-		return fmt.Errorf("%w: %d", errUnknownFormat, s.OM)
-	}
-
-	return nil
-}
-
-func newImageWithBackGround(x, y int, background color.Color) *image.RGBA {
-	img := createImage(x, y)
-
-	fillBackground(img, background)
-
-	return img
-}
-
-func createImage(x, y int) *image.RGBA {
-	rect := image.Rectangle{
-		image.Point{0, 0},
-		image.Point{x, y},
-	}
-
-	return image.NewRGBA(rect)
-}
-
-func fillBackground(img *image.RGBA, c color.Color) {
-	rect := img.Bounds()
-	for x := range rect.Max.X {
-		for y := range rect.Max.Y {
-			img.Set(x, y, c)
-		}
-	}
-}

main.go

@@ -1,35 +0,0 @@
-package main
-
-import (
-	"fmt"
-	"os"
-)
-
-func main() {
-	exitCode := run()
-	os.Exit(exitCode)
-}
-
-func run() int {
-	s := parseSettings()
-
-	file, err := os.Create(s.Fname)
-	if err != nil {
-		fmt.Println("ошибка открытия файла", err)
-		return 1
-	}
-
-	if err := createBarnsleyFernImage(s, file); err != nil {
-		fmt.Println("ошибка создания изображения:", err)
-		return 2
-	}
-
-	if err := file.Close(); err != nil {
-		fmt.Println("ошибка закрытия файла:", err)
-		return 3
-	}
-
-	fmt.Println("done")
-
-	return 0
-}

settings.go

@@ -1,63 +0,0 @@
-package main
-
-import (
-	"flag"
-	"fmt"
-	"image/color"
-)
-
-type colormode int
-
-const (
-	cdefault colormode = iota
-	crandom
-	ctimed
-	crainbow
-)
-
-type outputmode int
-
-const (
-	outputpng outputmode = iota
-	outputjpeg
-)
-
-type settings struct {
-	X, Y  int         // image size
-	Dots  int         // dots to draw for single image
-	BG    color.Color // background color
-	CM    colormode   // settings for colorFunc
-	OM    outputmode  // what to output
-	Fname string      // filename for output
-}
-
-func parseSettings() *settings {
-	var (
-		x, y   int
-		dots   int
-		cmode  int
-		output int
-		fname  string
-	)
-
-	flag.IntVar(&x, "x", 1920, "размер картинки по горизонтали")
-	flag.IntVar(&y, "y", 1080, "размер картинки по вертикали")
-	flag.IntVar(&dots, "d", 100000, "сколько точек рисовать")
-	flag.IntVar(&cmode, "c", 0, "color mode: 0 - default green, 1 - random, 2 - timed, 3 - rainbow")
-	flag.IntVar(&output, "o", 0, fmt.Sprintf("режим вывода: %d - png, %d - jpeg", outputpng, outputjpeg))
-	flag.StringVar(&fname, "out", "barnsley-fern.png", "полный путь файла для записи изображения")
-
-	flag.Parse()
-
-	s := &settings{
-		X:     x,
-		Y:     y,
-		Dots:  dots,
-		BG:    color.White,
-		CM:    colormode(cmode),
-		OM:    outputmode(output),
-		Fname: fname,
-	}
-
-	return s
-}