feat(selection): Use circle for units

1 files changed, 20 insertions, 1 deletions

diff --git a/image.go b/image.go
index 0823576..5045d90 100644
--- a/image.go
+++ b/image.go
@@ -4,10 +4,29 @@ import (
 	"image/color"
 
 	"github.com/hajimehoshi/ebiten/v2"
+	"github.com/hajimehoshi/ebiten/v2/vector"
 )
 
-func NewColorImage(size Point, color color.Color) *ebiten.Image {
+func NewFilledRectangleImage(size Point, color color.Color) *ebiten.Image {
 	image := ebiten.NewImage(size.X, size.Y)
 	image.Fill(color)
 	return image
 }
+
+func NewStrokeRectangleImage(size Point, strokeWidth int, color color.Color) *ebiten.Image {
+	image := ebiten.NewImage(size.X, size.Y)
+	vector.StrokeRect(image, 0, 0, float32(size.X), float32(size.Y), float32(strokeWidth), color, true)
+	return image
+}
+
+func NewFilledCircleImage(width int, color color.Color) *ebiten.Image {
+	image := ebiten.NewImage((width), (width))
+	vector.DrawFilledCircle(image, float32(width/2), float32(width/2), float32(width/2), color, true)
+	return image
+}
+
+func NewStrokeCircleImage(width int, strokeWidth int, color color.Color) *ebiten.Image {
+	image := ebiten.NewImage((width), (width))
+	vector.StrokeCircle(image, float32(width/2), float32(width/2), float32(width/2), float32(strokeWidth), color, true)
+	return image
+}