69 lines
1.2 KiB
Go
69 lines
1.2 KiB
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)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|