Cleaning cleaning cleaning

This commit is contained in:
Michał Gdula 2024-01-25 20:02:20 +00:00
parent 105785e6c0
commit d0ebc1ee54
4 changed files with 159 additions and 143 deletions

86
canvas.go Normal file
View file

@ -0,0 +1,86 @@
package main
import (
raylib "github.com/gen2brain/raylib-go/raylib"
)
type Canvas struct {
Name string
Size raylib.Vector2
Offset raylib.Vector2
Target raylib.RenderTexture2D
Strokes []penTool
UndoneStrokes []penTool
Refresh bool
}
func (c *Canvas) Update() {
if c.Refresh {
raylib.BeginTextureMode(c.Target)
raylib.ClearBackground(raylib.White)
for _, mark := range c.Strokes {
mark.Draw(raylib.Vector2Scale(c.Offset, -1))
}
raylib.EndTextureMode()
c.Refresh = false
}
}
func (c *Canvas) Undo() {
if len(c.Strokes) > 0 {
c.UndoneStrokes = append(c.UndoneStrokes, c.Strokes[len(c.Strokes)-1])
c.Strokes = c.Strokes[:len(c.Strokes)-1]
c.Refresh = true
AddToast("Undo")
}
}
func (c *Canvas) Redo() {
if len(c.UndoneStrokes) > 0 {
c.Strokes = append(c.Strokes, c.UndoneStrokes[len(c.UndoneStrokes)-1])
c.UndoneStrokes = c.UndoneStrokes[:len(c.UndoneStrokes)-1]
c.Refresh = true
AddToast("Redo")
}
}
func (c *Canvas) Draw() {
raylib.DrawTexturePro(
c.Target.Texture,
raylib.NewRectangle(0, 0, c.Size.X, -c.Size.Y),
raylib.NewRectangle(c.Offset.X, c.Offset.Y, c.Size.X, c.Size.Y),
raylib.Vector2Zero(),
0,
raylib.White,
)
}
func (c *Canvas) Save() {
if c.Name == "" {
AddToast("Please enter a file name!")
} else {
image := raylib.LoadImageFromTexture(c.Target.Texture)
raylib.ImageRotate(image, 180)
raylib.ImageFlipHorizontal(image)
raylib.ExportImage(*image, DirUserData+c.Name+".png")
AddToast("Drawing saved as " + c.Name + ".png")
}
}
func NewCanvas(name string, size, offset raylib.Vector2) *Canvas {
return &Canvas{
Name: name,
Size: size,
Offset: offset,
Target: raylib.LoadRenderTexture(int32(size.X), int32(size.Y)),
Strokes: []penTool{},
UndoneStrokes: []penTool{},
Refresh: true,
}
}

180
main.go
View file

@ -2,17 +2,52 @@ package main
import ( import (
"fmt" "fmt"
"os"
gui "github.com/gen2brain/raylib-go/raygui" gui "github.com/gen2brain/raylib-go/raygui"
raylib "github.com/gen2brain/raylib-go/raylib" raylib "github.com/gen2brain/raylib-go/raylib"
"os"
) )
const (
WindowTitle = "Colouring App"
WindowMinWidth = int32(800)
WindowMinHeight = int32(600)
WindowFPS = int32(144)
)
var (
WindowWidth = WindowMinWidth
WindowHeight = WindowMinHeight
)
const (
DirAssets = "./assets/"
DirUserData = "./userData/"
)
const (
StateNone = iota
StateDrawing
StateFileMenu
)
func checkDirs() {
if _, err := os.Stat(DirUserData); os.IsNotExist(err) {
err := os.Mkdir(DirUserData, 0755)
if err != nil {
panic(err)
}
}
}
func main() { func main() {
raylib.SetConfigFlags(raylib.FlagWindowResizable) raylib.SetConfigFlags(raylib.FlagWindowResizable)
raylib.SetTraceLogLevel(raylib.LogTrace) raylib.SetTraceLogLevel(raylib.LogTrace)
raylib.SetConfigFlags(raylib.FlagMsaa4xHint) raylib.SetConfigFlags(raylib.FlagMsaa4xHint)
raylib.InitWindow(WindowWidth, WindowHeight, WindowTitle) raylib.InitWindow(WindowWidth, WindowHeight, WindowTitle)
raylib.SetWindowMinSize(int(WindowMinWidth), int(WindowMinHeight))
raylib.InitAudioDevice() raylib.InitAudioDevice()
raylib.SetTargetFPS(WindowFPS) raylib.SetTargetFPS(WindowFPS)
@ -20,115 +55,41 @@ func main() {
var ( var (
camera = raylib.NewCamera2D(raylib.NewVector2(0, 0), raylib.NewVector2(0, 0), 0, 1) camera = raylib.NewCamera2D(raylib.NewVector2(0, 0), raylib.NewVector2(0, 0), 0, 1)
//cameraMoveOffset = raylib.NewVector2(10, 10)
canvasSize = raylib.NewVector2(500, 430) canvas = NewCanvas("NewProject", raylib.NewVector2(600, 530), raylib.NewVector2(15, 15))
canvasScale = float32(1) currentStroke = penTool{}
canvas = raylib.LoadRenderTexture(int32(canvasSize.X), int32(canvasSize.Y)) drawing = false
canvasRefresh = true
sidePanelWidth = float32(350) sidePanelWidth = float32(350)
sidePanelRelativeX = WindowWidth - int32(sidePanelWidth) sidePanelRelativeX = WindowWidth - int32(sidePanelWidth)
drawing = false
//drawingMode = ModeDrawing
currentStroke = penTool{}
strokes = []penTool{currentStroke}
undoneStrokes = []penTool{}
colourPickerVal = raylib.Orange colourPickerVal = raylib.Orange
colourPickerHeight = float32(200) colourPickerHeight = float32(250)
brushSize = float32(10) brushSize = float32(10)
brushOpacity = float32(1)
fileName = "NewProject"
fileNameEditing = false fileNameEditing = false
menu = StateNone menu = StateNone
appShouldQuit = false
) )
// check if userData exists // check if userData exists
if _, err := os.Stat(DirUserData); os.IsNotExist(err) { checkDirs()
err := os.Mkdir(DirUserData, 0755)
if err != nil {
panic(err)
}
}
WindowWidth = int32(raylib.GetScreenWidth()) for !appShouldQuit {
WindowHeight = int32(raylib.GetScreenHeight())
refreshCanvas := func() {
raylib.BeginTextureMode(canvas)
raylib.ClearBackground(raylib.White)
for i := 1; i < len(strokes); i++ {
strokes[i].Draw(raylib.NewVector2(-10, -10))
}
raylib.EndTextureMode()
}
redoStroke := func() {
if len(undoneStrokes) > 0 {
strokes = append(strokes, undoneStrokes[len(undoneStrokes)-1])
undoneStrokes = undoneStrokes[:len(undoneStrokes)-1]
AddToast("Redo")
canvasRefresh = true
}
}
undoStroke := func() {
// 1 because I dont know why
if len(strokes) > 1 {
undoneStrokes = append(undoneStrokes, strokes[len(strokes)-1])
strokes = strokes[:len(strokes)-1]
canvasRefresh = true
AddToast("Undo")
}
}
saveImage := func() {
if fileName == "" {
AddToast("Please enter a file name")
} else {
image := raylib.LoadImageFromTexture(canvas.Texture)
raylib.ImageRotate(image, 180)
raylib.ImageFlipHorizontal(image)
raylib.ExportImage(*image, DirUserData+fileName+".png")
AddToast("Drawing saved at " + DirUserData + fileName + ".png")
}
}
for !ShouldQuit {
// LOOP // LOOP
ShouldQuit = raylib.WindowShouldClose() appShouldQuit = raylib.WindowShouldClose()
if raylib.IsWindowResized() { if raylib.IsWindowResized() {
WindowWidth = int32(raylib.GetScreenWidth()) WindowWidth = int32(raylib.GetScreenWidth())
WindowHeight = int32(raylib.GetScreenHeight()) WindowHeight = int32(raylib.GetScreenHeight())
sidePanelRelativeX = WindowWidth - int32(sidePanelWidth) sidePanelRelativeX = WindowWidth - int32(sidePanelWidth)
} }
// INPUT // INPUT
{ {
if raylib.GetMouseWheelMove() != 0 && !drawing {
canvasScale += float32(raylib.GetMouseWheelMove()) * 0.05
}
//if raylib.IsMouseButtonPressed(raylib.MouseMiddleButton) {
// cameraMoveOffset = raylib.Vector2Subtract(camera.Target, raylib.GetMousePosition())
//}
//if raylib.IsMouseButtonDown(raylib.MouseMiddleButton) {
// camera.Target = raylib.Vector2Subtract(raylib.GetMousePosition(), raylib.Vector2Scale(cameraMoveOffset, -1))
//}
if raylib.IsKeyPressed(raylib.KeyF8) { if raylib.IsKeyPressed(raylib.KeyF8) {
AddToast("This is a toast message!") AddToast("This is a toast message!")
} }
@ -136,12 +97,11 @@ func main() {
if raylib.IsMouseButtonPressed(raylib.MouseLeftButton) { if raylib.IsMouseButtonPressed(raylib.MouseLeftButton) {
if raylib.CheckCollisionPointRec(raylib.GetMousePosition(), raylib.NewRectangle(float32(WindowWidth-int32(sidePanelWidth)), 0, sidePanelWidth, float32(WindowHeight))) { if raylib.CheckCollisionPointRec(raylib.GetMousePosition(), raylib.NewRectangle(float32(WindowWidth-int32(sidePanelWidth)), 0, sidePanelWidth, float32(WindowHeight))) {
drawing = false drawing = false
} else if raylib.CheckCollisionPointRec(raylib.GetMousePosition(), raylib.NewRectangle(10, 10, canvasSize.X, canvasSize.Y)) { } else if raylib.CheckCollisionPointRec(raylib.GetMousePosition(), raylib.NewRectangle(10, 10, canvas.Size.X, canvas.Size.Y)) {
drawing = true drawing = true
currentStroke = penTool{ currentStroke = penTool{
Color: colourPickerVal, Color: colourPickerVal,
Size: brushSize, Size: brushSize,
Opacity: 1,
} }
} }
} }
@ -155,20 +115,20 @@ func main() {
} }
} }
if raylib.IsMouseButtonReleased(raylib.MouseLeftButton) && currentStroke.Points != nil { if raylib.IsMouseButtonReleased(raylib.MouseLeftButton) && currentStroke.Points != nil {
strokes = append(strokes, currentStroke) canvas.Strokes = append(canvas.Strokes, currentStroke)
currentStroke = penTool{} canvas.UndoneStrokes = []penTool{}
undoneStrokes = []penTool{} canvas.Refresh = true
currentStroke = penTool{}
drawing = false drawing = false
canvasRefresh = true
} }
if raylib.IsKeyDown(raylib.KeyLeftControl) && raylib.IsKeyDown(raylib.KeyLeftShift) && raylib.IsKeyPressed(raylib.KeyZ) { if raylib.IsKeyDown(raylib.KeyLeftControl) && raylib.IsKeyDown(raylib.KeyLeftShift) && raylib.IsKeyPressed(raylib.KeyZ) {
redoStroke() canvas.Redo()
} else if raylib.IsKeyDown(raylib.KeyLeftControl) && raylib.IsKeyPressed(raylib.KeyZ) { } else if raylib.IsKeyDown(raylib.KeyLeftControl) && raylib.IsKeyPressed(raylib.KeyZ) {
undoStroke() canvas.Undo()
} else if raylib.IsKeyDown(raylib.KeyLeftControl) && raylib.IsKeyPressed(raylib.KeyS) { } else if raylib.IsKeyDown(raylib.KeyLeftControl) && raylib.IsKeyPressed(raylib.KeyS) {
saveImage() canvas.Save()
} }
} }
@ -180,10 +140,7 @@ func main() {
gui.SetState(gui.STATE_NORMAL) gui.SetState(gui.STATE_NORMAL)
} }
if canvasRefresh { canvas.Update()
refreshCanvas()
canvasRefresh = false
}
UpdateToasts() UpdateToasts()
} }
@ -197,18 +154,18 @@ func main() {
// Canvas stuff // Canvas stuff
raylib.BeginMode2D(camera) raylib.BeginMode2D(camera)
{ {
raylib.DrawRectangle(20, 20, int32(canvasSize.X), int32(canvasSize.Y), raylib.Fade(raylib.Black, 0.3)) raylib.DrawRectangle(int32(canvas.Offset.X)+10, int32(canvas.Offset.Y)+10, int32(canvas.Size.X), int32(canvas.Size.Y), raylib.Fade(raylib.Black, 0.3))
raylib.DrawTexturePro(canvas.Texture, raylib.NewRectangle(0, 0, float32(canvas.Texture.Width), float32(-canvas.Texture.Height)), raylib.NewRectangle(10, 10, canvasSize.X, canvasSize.Y), raylib.Vector2{}, 0, raylib.White) canvas.Draw()
//raylib.BeginScissorMode(10, 10, int32(canvasSize.X), int32(canvasSize.Y)) raylib.BeginScissorMode(int32(canvas.Offset.X), int32(canvas.Offset.Y), int32(canvas.Size.X), int32(canvas.Size.Y))
currentStroke.Draw(raylib.NewVector2(0, 0)) currentStroke.Draw(raylib.NewVector2(0, 0))
//raylib.EndScissorMode() raylib.EndScissorMode()
if drawing { if drawing {
raylib.DrawRectangleLines(10, 10, int32(canvasSize.X), int32(canvasSize.Y), raylib.DarkGray) raylib.DrawRectangleLines(int32(canvas.Offset.X), int32(canvas.Offset.Y), int32(canvas.Size.X), int32(canvas.Size.Y), raylib.DarkGray)
raylib.DrawCircleLines(int32(raylib.GetMousePosition().X), int32(raylib.GetMousePosition().Y), brushSize/2, raylib.Black) raylib.DrawCircleLines(int32(raylib.GetMousePosition().X), int32(raylib.GetMousePosition().Y), brushSize/2, raylib.Black)
} else { } else {
raylib.DrawRectangleLines(10, 10, int32(canvasSize.X), int32(canvasSize.Y), raylib.Gray) raylib.DrawRectangleLines(int32(canvas.Offset.X), int32(canvas.Offset.Y), int32(canvas.Size.X), int32(canvas.Size.Y), raylib.Gray)
raylib.DrawCircleLines(int32(raylib.GetMousePosition().X), int32(raylib.GetMousePosition().Y), brushSize/2, raylib.Black) raylib.DrawCircleLines(int32(raylib.GetMousePosition().X), int32(raylib.GetMousePosition().Y), brushSize/2, raylib.Black)
} }
} }
@ -223,14 +180,14 @@ func main() {
menu = StateFileMenu menu = StateFileMenu
} }
if gui.Button(raylib.NewRectangle(float32(sidePanelRelativeX+20+25), 10, 25, 25), gui.IconText(gui.ICON_FOLDER_SAVE, "")) { if gui.Button(raylib.NewRectangle(float32(sidePanelRelativeX+20+25), 10, 25, 25), gui.IconText(gui.ICON_FOLDER_SAVE, "")) {
saveImage() canvas.Save()
} }
if gui.Button(raylib.NewRectangle(float32(WindowWidth-70), 10, 25, 25), gui.IconText(gui.ICON_UNDO, "")) { if gui.Button(raylib.NewRectangle(float32(WindowWidth-70), 10, 25, 25), gui.IconText(gui.ICON_UNDO, "")) {
undoStroke() canvas.Undo()
} }
if gui.Button(raylib.NewRectangle(float32(WindowWidth-35), 10, 25, 25), gui.IconText(gui.ICON_REDO, "")) { if gui.Button(raylib.NewRectangle(float32(WindowWidth-35), 10, 25, 25), gui.IconText(gui.ICON_REDO, "")) {
redoStroke() canvas.Redo()
} }
colourPickerVal = gui.ColorPicker(raylib.NewRectangle(float32(sidePanelRelativeX+10), 45, sidePanelWidth-45, colourPickerHeight), "Color", colourPickerVal) colourPickerVal = gui.ColorPicker(raylib.NewRectangle(float32(sidePanelRelativeX+10), 45, sidePanelWidth-45, colourPickerHeight), "Color", colourPickerVal)
@ -238,11 +195,8 @@ func main() {
gui.Label(raylib.NewRectangle(float32(sidePanelRelativeX+10), 55+colourPickerHeight, 60, 20), "Brush Size") gui.Label(raylib.NewRectangle(float32(sidePanelRelativeX+10), 55+colourPickerHeight, 60, 20), "Brush Size")
brushSize = gui.Slider(raylib.NewRectangle(float32(sidePanelRelativeX+80), 55+colourPickerHeight, sidePanelWidth-90, 20), "", "", brushSize, 1, 100) brushSize = gui.Slider(raylib.NewRectangle(float32(sidePanelRelativeX+80), 55+colourPickerHeight, sidePanelWidth-90, 20), "", "", brushSize, 1, 100)
gui.Label(raylib.NewRectangle(float32(sidePanelRelativeX+10), 85+colourPickerHeight, 60, 20), "Brush Opacity")
brushOpacity = gui.Slider(raylib.NewRectangle(float32(sidePanelRelativeX+80), 85+colourPickerHeight, sidePanelWidth-90, 20), "", "", brushOpacity, 0, 1)
gui.Label(raylib.NewRectangle(float32(sidePanelRelativeX+10), 115+colourPickerHeight, 60, 20), "File Name") gui.Label(raylib.NewRectangle(float32(sidePanelRelativeX+10), 115+colourPickerHeight, 60, 20), "File Name")
if gui.TextBox(raylib.NewRectangle(float32(sidePanelRelativeX+80), 115+colourPickerHeight, sidePanelWidth-90, 20), &fileName, 40, fileNameEditing) { if gui.TextBox(raylib.NewRectangle(float32(sidePanelRelativeX+80), 115+colourPickerHeight, sidePanelWidth-90, 20), &canvas.Name, 40, fileNameEditing) {
fileNameEditing = !fileNameEditing fileNameEditing = !fileNameEditing
} }
} }
@ -253,17 +207,17 @@ func main() {
{ {
var text string var text string
text = fmt.Sprintf("Strokes: %d | Points: %d", len(strokes), len(currentStroke.Points)) text = fmt.Sprintf("Strokes: %d | Points: %d", len(canvas.Strokes), len(currentStroke.Points))
gui.StatusBar(raylib.NewRectangle(0, float32(WindowHeight-20), 200, 20), text) gui.StatusBar(raylib.NewRectangle(0, float32(WindowHeight-20), 200, 20), text)
text = fmt.Sprintf("Canvas Size: %dx%d | Scale: %v", int32(canvasSize.X), int32(canvasSize.Y), canvasScale) text = fmt.Sprintf("Canvas Size: %dx%d", int32(canvas.Size.X), int32(canvas.Size.Y))
gui.StatusBar(raylib.NewRectangle(199, float32(WindowHeight-20), 200, 20), text) gui.StatusBar(raylib.NewRectangle(199, float32(WindowHeight-20), 200, 20), text)
} }
switch menu { switch menu {
case StateFileMenu: case StateFileMenu:
raylib.DrawRectangle(0, 0, WindowWidth, WindowHeight, raylib.Fade(raylib.Black, 0.5)) raylib.DrawRectangle(0, 0, WindowWidth, WindowHeight, raylib.Fade(raylib.Black, 0.5))
choice := gui.MessageBox(raylib.NewRectangle(float32(WindowWidth/2-200), float32(WindowHeight/2-100), 400, 200), "File", "This is a message box", "OK") choice := gui.MessageBox(raylib.NewRectangle(float32(WindowWidth/2-200), float32(WindowHeight/2-100), 400, 200), "File", "This is a message box", "Ok")
if choice == 0 || choice == 1 { if choice == 0 || choice == 1 {
menu = StateNone menu = StateNone
} }

View file

@ -7,7 +7,6 @@ import (
type penTool struct { type penTool struct {
Color raylib.Color Color raylib.Color
Size float32 Size float32
Opacity float32
Points []raylib.Vector2 Points []raylib.Vector2
} }

23
vars.go
View file

@ -1,23 +0,0 @@
package main
var (
WindowTitle = "Colouring App"
WindowWidth int32 = 800
WindowHeight int32 = 600
WindowFPS int32 = 144
)
const (
StateNone = iota
StateFileMenu
StateDrawing
)
const (
DirAssets = "./assets/"
DirUserData = "./userData/"
)
var (
ShouldQuit = false
)