如何使用Golang在图片上绘制图形和形状

package main

import (
    "image"
    "image/color"
    "image/draw"
    "image/png"
    "log"
    "os"
)

func main() {
    // 创建一个新的图片，并设置宽高
    width := 300
    height := 200
    img := image.NewRGBA(image.Rect(0, 0, width, height))

    // 设置绘制的颜色和形状
    red := color.RGBA{255, 0, 0, 255}    // 设置为红色
    blue := color.RGBA{0, 0, 255, 255}   // 设置为蓝色
    green := color.RGBA{0, 255, 0, 255}  // 设置为绿色
    yellow := color.RGBA{255, 255, 0, 255} // 设置为黄色

    // 在图片上绘制直线
    drawLine(img, 10, 10, width-10, height-10, red)

    // 在图片上绘制矩形
    drawRectangle(img, 20, 20, width-20, height-20, blue)

    // 在图片上绘制圆形
    drawCircle(img, width/2, height/2, 50, green)

    // 在图片上绘制椭圆形
    drawEllipse(img, width/2, height/2, 100, 50, yellow)

    // 将图片保存为PNG文件
    saveToPNG(img, "shapes.png")
}

// 在图片上绘制直线
func drawLine(img *image.RGBA, x1, y1, x2, y2 int, color color.RGBA) {
    drawLine := func(img *image.RGBA, x0, y0, x1, y1 int, color color.RGBA) {
        dx := abs(x1 - x0)
        dy := abs(y1 - y0)
        sx, sy := 0, 0

        if x0 < x1 {
            sx = 1
        } else {
            sx = -1
        }

        if y0 < y1 {
            sy = 1
        } else {
            sy = -1
        }

        err := dx - dy

        for {
            img.Set(x0, y0, color)

            if x0 == x1 && y0 == y1 {
                break
            }

            e2 := 2 * err

            if e2 > -dy {
                err -= dy
                x0 += sx
            }

            if e2 < dx {
                err += dx
                y0 += sy
            }
        }
    }

    drawLine(img, x1, y1, x2, y2, color)
}

// 在图片上绘制矩形
func drawRectangle(img *image.RGBA, x1, y1, x2, y2 int, color color.RGBA) {
    draw.Draw(img, image.Rect(x1, y1, x2, y2), &image.Uniform{color}, image.ZP, draw.Src)
}

// 在图片上绘制圆形
func drawCircle(img *image.RGBA, x, y, r int, color color.RGBA) {
    for px := -r; px <= r; px++ {
        for py := -r; py <= r; py++ {
            if px*px+py*py <= r*r {
                img.Set(x+px, y+py, color)
            }
        }
    }
}

// 在图片上绘制椭圆形
func drawEllipse(img *image.RGBA, x, y, rx, ry int, color color.RGBA) {
    for px := -rx; px <= rx; px++ {
        for py := -ry; py <= ry; py++ {
            if (px*px)/(rx*rx)+(py*py)/(ry*ry) <= 1 {
                img.Set(x+px, y+py, color)
            }
        }
    }
}

// 将图片保存为PNG文件
func saveToPNG(img image.Image, filename string) {
    f, err := os.Create(filename)
    if err != nil {
        log.Fatal(err)
    }
    defer f.Close()

    err = png.Encode(f, img)
    if err != nil {
        log.Fatal(err)
    }
}

// 辅助函数：获取值的绝对值
func abs(x int) int {
    if x < 0 {
        return -x
    }
    return x
}