barnsley-fern-go/image.go

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", 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)
		}
	}
}