mirror of
https://github.com/Fluffy-Bean/ColouringApp.git
synced 2025-01-14 17:35:12 +00:00
375 lines
14 KiB
Go
375 lines
14 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
gui "github.com/gen2brain/raylib-go/raygui"
|
|
raylib "github.com/gen2brain/raylib-go/raylib"
|
|
)
|
|
|
|
func main() {
|
|
// Initialize raylib
|
|
raylib.SetConfigFlags(raylib.FlagWindowResizable | raylib.FlagWindowHighdpi | raylib.FlagMsaa4xHint)
|
|
// raylib.SetConfigFlags(raylib.FlagWindowResizable | raylib.FlagMsaa4xHint)
|
|
raylib.InitWindow(applicationWindowWidth, applicationWindowHeight, applicationTitle)
|
|
raylib.SetWindowMinSize(int(applicationMinWindowWidth), int(applicationMinWindowHeight)) // Set a minimum window size
|
|
raylib.SetTargetFPS(int32(raylib.GetMonitorRefreshRate(raylib.GetCurrentMonitor()))) // Match monitor refresh rate
|
|
raylib.SetExitKey(raylib.KeyNull) // disable exit key
|
|
raylib.HideCursor() // Hide cursor
|
|
|
|
// Make sure both assets and userData directories exist
|
|
if _, err := os.Stat(dirAssets); os.IsNotExist(err) {
|
|
panic("Assets directory not found")
|
|
}
|
|
if _, err := os.Stat(dirUserData); os.IsNotExist(err) {
|
|
if err := os.Mkdir(dirUserData, 0755); err != nil {
|
|
panic("Could not create userData directory")
|
|
}
|
|
}
|
|
|
|
// Load all user data projects
|
|
var userDataProjects []string
|
|
if files, err := os.ReadDir(dirUserData); err == nil {
|
|
for _, file := range files {
|
|
if strings.HasSuffix(file.Name(), ".png") {
|
|
userDataProjects = append(userDataProjects, file.Name())
|
|
}
|
|
}
|
|
}
|
|
|
|
// LOOP
|
|
for !applicationShouldQuit {
|
|
// Update default loop values
|
|
if raylib.WindowShouldClose() {
|
|
applicationState = StateWindowWantsToDie
|
|
}
|
|
if raylib.IsWindowResized() {
|
|
applicationWindowWidth = int32(raylib.GetScreenWidth())
|
|
applicationWindowHeight = int32(raylib.GetScreenHeight())
|
|
toolPanelOffset = applicationWindowWidth - int32(toolPanelWidth)
|
|
}
|
|
|
|
// Create new canvas if needed
|
|
if shouldCreateNewCanvas {
|
|
var canvasBackground raylib.Texture2D
|
|
|
|
if newCanvasImagePath != "" {
|
|
canvasBackground = NewBackgroundImage(newCanvasImagePath)
|
|
newCanvasWidth = int(canvasBackground.Width)
|
|
newCanvasHeight = int(canvasBackground.Height)
|
|
} else {
|
|
canvasBackground = NewBackgroundColour(raylib.NewVector2(float32(newCanvasWidth), float32(newCanvasHeight)), newCanvasColor)
|
|
}
|
|
|
|
canvas = NewCanvas(newCanvasName, raylib.NewVector2(float32(newCanvasWidth), float32(newCanvasHeight)), raylib.NewVector2(15, 15), canvasBackground)
|
|
|
|
// Reset all values
|
|
shouldCreateNewCanvas = false
|
|
newCanvasName = defaultProjectName
|
|
newCanvasWidth = defaultProjectWidth
|
|
newCanvasHeight = defaultProjectHeight
|
|
newCanvasColor = raylib.White
|
|
newCanvasImagePath = ""
|
|
}
|
|
|
|
// INPUT
|
|
{
|
|
if applicationState == StateNormal {
|
|
if raylib.IsKeyPressed(raylib.KeyF1) {
|
|
newStrokeType = toolNone
|
|
addToast("Tool: None")
|
|
}
|
|
if raylib.IsKeyPressed(raylib.KeyF2) {
|
|
newStrokeType = toolPen
|
|
addToast("Tool: Pen")
|
|
}
|
|
}
|
|
|
|
if raylib.IsKeyPressed(raylib.KeyF7) {
|
|
addToast("This is a test toast")
|
|
}
|
|
if raylib.IsKeyPressed(raylib.KeyF8) {
|
|
applicationShowDebugValues = !applicationShowDebugValues
|
|
}
|
|
if raylib.IsKeyPressed(raylib.KeyF12) {
|
|
addToast("Screenshot saved!")
|
|
}
|
|
|
|
switch newStrokeType {
|
|
case toolPen:
|
|
if raylib.IsMouseButtonPressed(raylib.MouseLeftButton) && applicationState == StateNormal {
|
|
if !raylib.CheckCollisionPointRec(raylib.GetMousePosition(), raylib.NewRectangle(float32(applicationWindowWidth-int32(toolPanelWidth)), 0, toolPanelWidth, float32(applicationWindowHeight))) &&
|
|
raylib.CheckCollisionPointRec(raylib.GetMousePosition(), raylib.NewRectangle(10, 10, canvas.Size.X, canvas.Size.Y)) {
|
|
applicationState = StateDrawing
|
|
newPenStroke = penTool{
|
|
Size: toolPanelBrushSize,
|
|
Color: toolPanelColourPicker,
|
|
}
|
|
}
|
|
}
|
|
|
|
if raylib.IsMouseButtonDown(raylib.MouseLeftButton) && applicationState == StateDrawing {
|
|
if len(newPenStroke.Points) <= 1 {
|
|
newPenStroke.Points = append(newPenStroke.Points, raylib.GetMousePosition())
|
|
} else if raylib.Vector2Distance(newPenStroke.Points[len(newPenStroke.Points)-1], raylib.GetMousePosition()) > float32(newStrokeSafeZone) {
|
|
newPenStroke.Points = append(newPenStroke.Points, raylib.GetMousePosition())
|
|
}
|
|
|
|
applicationState = StateDrawing
|
|
}
|
|
|
|
if raylib.IsMouseButtonReleased(raylib.MouseLeftButton) && newPenStroke.Points != nil {
|
|
canvas.AddStroke(newPenStroke.Render())
|
|
newPenStroke = penTool{}
|
|
applicationState = StateNormal
|
|
}
|
|
case toolNone:
|
|
fallthrough
|
|
default:
|
|
// yyeeeeet
|
|
}
|
|
|
|
if raylib.IsKeyDown(raylib.KeyLeftControl) && raylib.IsKeyDown(raylib.KeyLeftShift) && raylib.IsKeyPressed(raylib.KeyZ) {
|
|
canvas.Redo()
|
|
} else if raylib.IsKeyDown(raylib.KeyLeftControl) && raylib.IsKeyPressed(raylib.KeyZ) {
|
|
canvas.Undo()
|
|
} else if raylib.IsKeyDown(raylib.KeyLeftControl) && raylib.IsKeyPressed(raylib.KeyS) {
|
|
canvas.Save()
|
|
}
|
|
}
|
|
|
|
// UPDATE
|
|
{
|
|
applicationRuntime += 1
|
|
if math.Mod(float64(applicationRuntime), 1) == 0 {
|
|
// ToDo: check cursor color contrast with background
|
|
}
|
|
|
|
if applicationState != StateNormal {
|
|
gui.Lock()
|
|
} else {
|
|
gui.Unlock()
|
|
}
|
|
|
|
canvas.Update()
|
|
updateToasts()
|
|
}
|
|
|
|
// DRAW
|
|
raylib.BeginDrawing()
|
|
{
|
|
raylib.ClearBackground(raylib.White)
|
|
gui.Grid(raylib.NewRectangle(0, 0, float32(applicationWindowWidth), float32(applicationWindowHeight)), "", 30, 1, &raylib.Vector2{})
|
|
|
|
// Canvas
|
|
{
|
|
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))
|
|
canvas.Draw()
|
|
|
|
raylib.BeginScissorMode(int32(canvas.Offset.X), int32(canvas.Offset.Y), int32(canvas.Size.X), int32(canvas.Size.Y))
|
|
newPenStroke.Draw()
|
|
raylib.EndScissorMode()
|
|
|
|
raylib.DrawRectangleLines(int32(canvas.Offset.X), int32(canvas.Offset.Y), int32(canvas.Size.X), int32(canvas.Size.Y), raylib.DarkGray)
|
|
}
|
|
|
|
// Tool Panel
|
|
raylib.BeginScissorMode(toolPanelOffset, 0, int32(toolPanelWidth), applicationWindowHeight)
|
|
{
|
|
raylib.DrawRectangle(toolPanelOffset, 0, int32(toolPanelWidth), applicationWindowHeight, raylib.Fade(raylib.White, 0.9))
|
|
|
|
if gui.Button(raylib.NewRectangle(float32(toolPanelOffset+10), 10, 25, 25), gui.IconText(gui.ICON_FOLDER_OPEN, "")) {
|
|
applicationState = StateFileMenu
|
|
}
|
|
if gui.Button(raylib.NewRectangle(float32(toolPanelOffset+20+25), 10, 25, 25), gui.IconText(gui.ICON_FOLDER_SAVE, "")) {
|
|
canvas.Save()
|
|
}
|
|
|
|
if gui.Button(raylib.NewRectangle(float32(applicationWindowWidth-70), 10, 25, 25), gui.IconText(gui.ICON_UNDO, "")) {
|
|
canvas.Undo()
|
|
}
|
|
if gui.Button(raylib.NewRectangle(float32(applicationWindowWidth-35), 10, 25, 25), gui.IconText(gui.ICON_REDO, "")) {
|
|
canvas.Redo()
|
|
}
|
|
|
|
toolPanelColourPicker = gui.ColorPicker(raylib.NewRectangle(float32(toolPanelOffset+10), 45, toolPanelWidth-45, toolPanelColourPickerHeight), "Color", toolPanelColourPicker)
|
|
|
|
gui.Label(raylib.NewRectangle(float32(toolPanelOffset+10), 55+toolPanelColourPickerHeight, 60, 20), "Brush Size")
|
|
toolPanelBrushSize = gui.Slider(raylib.NewRectangle(float32(toolPanelOffset+80), 55+toolPanelColourPickerHeight, toolPanelWidth-90, 20), "", "", toolPanelBrushSize, 1, 100)
|
|
|
|
gui.Label(raylib.NewRectangle(float32(toolPanelOffset+10), 115+toolPanelColourPickerHeight, 60, 20), "File Name")
|
|
if gui.TextBox(raylib.NewRectangle(float32(toolPanelOffset+80), 115+toolPanelColourPickerHeight, toolPanelWidth-90, 20), &canvas.Name, 40, isEditingCanvasName) {
|
|
isEditingCanvasName = !isEditingCanvasName
|
|
}
|
|
raylib.DrawLine(toolPanelOffset, 0, toolPanelOffset, applicationWindowHeight, raylib.Black)
|
|
}
|
|
raylib.EndScissorMode()
|
|
|
|
// Debug Values
|
|
if applicationShowDebugValues {
|
|
var text string
|
|
|
|
text = fmt.Sprintf("Strokes: %d | Points: %d", len(canvas.Strokes), len(newPenStroke.Points))
|
|
gui.StatusBar(raylib.NewRectangle(0, float32(applicationWindowHeight-20), 150, 20), text)
|
|
|
|
text = fmt.Sprintf("Canvas Size: %dx%d", int32(canvas.Size.X), int32(canvas.Size.Y))
|
|
gui.StatusBar(raylib.NewRectangle(150, float32(applicationWindowHeight-20), 150, 20), text)
|
|
|
|
text = fmt.Sprintf("FPS: %d | DT: %f", raylib.GetFPS(), raylib.GetFrameTime())
|
|
gui.StatusBar(raylib.NewRectangle(300, float32(applicationWindowHeight-20), 170, 20), text)
|
|
}
|
|
|
|
// Menus
|
|
switch applicationState {
|
|
case StateFileMenu:
|
|
gui.Unlock()
|
|
raylib.DrawRectangle(0, 0, applicationWindowWidth, applicationWindowHeight, raylib.Fade(raylib.Black, 0.5))
|
|
windowPos := raylib.NewRectangle(float32((applicationWindowWidth/2)-200), float32((applicationWindowHeight/2)-200), 400, 400)
|
|
if gui.WindowBox(windowPos, "Open or New File") {
|
|
applicationState = StateNormal
|
|
}
|
|
|
|
// Magic numbers
|
|
raylib.BeginScissorMode(int32(windowPos.X)+1, int32(windowPos.Y)+24, int32(windowPos.Width)-2, int32(windowPos.Height)-25)
|
|
gui.GroupBox(raylib.NewRectangle(windowPos.X+11, windowPos.Y+34, windowPos.Width-22, 200), "Create New")
|
|
{
|
|
var err error
|
|
|
|
gui.Label(raylib.NewRectangle(windowPos.X+21, windowPos.Y+44, 100, 20), "File Name")
|
|
if gui.TextBox(raylib.NewRectangle(windowPos.X+131, windowPos.Y+44, windowPos.Width-152, 20), &newCanvasName, 40, isEditingNewCanvasName) {
|
|
isEditingNewCanvasName = !isEditingNewCanvasName
|
|
}
|
|
|
|
gui.Label(raylib.NewRectangle(windowPos.X+21, windowPos.Y+74, 100, 20), "Canvas Width")
|
|
width := fmt.Sprintf("%d", newCanvasWidth)
|
|
lastWidth := newCanvasWidth
|
|
if gui.TextBox(raylib.NewRectangle(windowPos.X+131, windowPos.Y+74, windowPos.Width-152, 20), &width, 6, isEditingNewCanvasWidth) {
|
|
isEditingNewCanvasWidth = !isEditingNewCanvasWidth
|
|
}
|
|
if newCanvasWidth, err = strconv.Atoi(width); err != nil {
|
|
newCanvasWidth = lastWidth
|
|
}
|
|
|
|
gui.Label(raylib.NewRectangle(windowPos.X+21, windowPos.Y+104, 100, 20), "Canvas Height")
|
|
height := fmt.Sprintf("%d", newCanvasHeight)
|
|
lastHeight := newCanvasHeight
|
|
if gui.TextBox(raylib.NewRectangle(windowPos.X+131, windowPos.Y+104, windowPos.Width-152, 20), &height, 6, isEditingNewCanvasHeight) {
|
|
isEditingNewCanvasHeight = !isEditingNewCanvasHeight
|
|
}
|
|
if newCanvasHeight, err = strconv.Atoi(height); err != nil {
|
|
newCanvasHeight = lastHeight
|
|
}
|
|
|
|
colors := []raylib.Color{raylib.Red, raylib.Orange, raylib.Yellow, raylib.Green, raylib.Blue, raylib.Purple, raylib.Pink, raylib.Brown, raylib.Black, raylib.White}
|
|
for i := 0; i < len(colors); i++ {
|
|
posX := windowPos.X + 21 + ((float32(i) * 26) + float32(i)*11)
|
|
posY := windowPos.Y + 149
|
|
|
|
if gui.Button(raylib.NewRectangle(posX, posY, 26, 26), "") {
|
|
newCanvasColor = colors[i]
|
|
}
|
|
|
|
if newCanvasColor == colors[i] {
|
|
raylib.DrawRectangle(int32(posX)-4, int32(posY)-4, 26+8, 26+8, raylib.Fade(raylib.Black, 0.2))
|
|
raylib.DrawRectangle(int32(posX), int32(posY), 26, 26, colors[i])
|
|
} else {
|
|
raylib.DrawRectangle(int32(posX), int32(posY), 26, 26, colors[i])
|
|
raylib.DrawRectangleLines(int32(posX), int32(posY), 26, 26, raylib.Black)
|
|
}
|
|
}
|
|
|
|
if gui.Button(raylib.NewRectangle(windowPos.X+windowPos.Width-140, windowPos.Y+204, 120, 20), "Create") {
|
|
applicationState = StateNewCanvas
|
|
}
|
|
}
|
|
|
|
gui.GroupBox(raylib.NewRectangle(windowPos.X+11, windowPos.Y+244, windowPos.Width-22, float32(len(userDataProjects)*20)+10), "Open Existing")
|
|
{
|
|
if gui.Button(raylib.NewRectangle(windowPos.X+21, windowPos.Y+254, windowPos.Width-42, 20), "Maned Wolf") {
|
|
newCanvasImagePath = filepath.Join(dirAssets, "manedWolf.jpg")
|
|
newCanvasName = "ManedWolf"
|
|
applicationState = StateNewCanvas
|
|
}
|
|
|
|
for i := 0; i < len(userDataProjects); i++ {
|
|
if gui.Button(raylib.NewRectangle(windowPos.X+21, windowPos.Y+274+float32(i*20), windowPos.Width-42, 20), userDataProjects[i]) {
|
|
newCanvasImagePath = filepath.Join(dirAssets, userDataProjects[i])
|
|
splitName := strings.Split(userDataProjects[i], ".")
|
|
newCanvasName = filepath.Base(splitName[0])
|
|
applicationState = StateNewCanvas
|
|
}
|
|
}
|
|
}
|
|
raylib.EndScissorMode()
|
|
case StateNewCanvas:
|
|
if !canvas.UnsavedChanges {
|
|
applicationState = StateNormal
|
|
shouldCreateNewCanvas = true
|
|
addToast("Created New Canvas: " + canvas.Name)
|
|
break
|
|
}
|
|
|
|
gui.Unlock()
|
|
raylib.DrawRectangle(0, 0, applicationWindowWidth, applicationWindowHeight, raylib.Fade(raylib.Black, 0.5))
|
|
windowPos := raylib.NewRectangle(float32((applicationWindowWidth/2)-200), float32((applicationWindowHeight/2)-75), 400, 150)
|
|
choice := gui.MessageBox(windowPos, "New Canvas", "Are you sure you want to create a new canvas?", "No;Yes")
|
|
|
|
if choice == 0 || choice == 1 {
|
|
applicationState = StateNormal
|
|
shouldCreateNewCanvas = false
|
|
} else if choice == 2 {
|
|
applicationState = StateNormal
|
|
shouldCreateNewCanvas = true
|
|
addToast("Created New Canvas: " + canvas.Name)
|
|
}
|
|
case StateWindowWantsToDie:
|
|
if !canvas.UnsavedChanges {
|
|
applicationShouldQuit = true
|
|
break
|
|
}
|
|
if !canvas.UnsavedChanges {
|
|
applicationState = StateNormal
|
|
shouldCreateNewCanvas = true
|
|
addToast("Created New Canvas: " + canvas.Name)
|
|
break
|
|
}
|
|
gui.Unlock()
|
|
raylib.DrawRectangle(0, 0, applicationWindowWidth, applicationWindowHeight, raylib.Fade(raylib.Black, 0.5))
|
|
windowPos := raylib.NewRectangle(float32((applicationWindowWidth/2)-200), float32((applicationWindowHeight/2)-75), 400, 150)
|
|
choice := gui.MessageBox(windowPos, "Unsaved Changes", "You have unsaved changes, are you sure you want to exit?", "No;Yes")
|
|
|
|
if choice == 0 || choice == 1 {
|
|
applicationState = StateNormal
|
|
} else if choice == 2 {
|
|
applicationShouldQuit = true
|
|
}
|
|
default:
|
|
}
|
|
|
|
// Cursor
|
|
switch newStrokeType {
|
|
case toolNone:
|
|
raylib.DrawTriangleLines(
|
|
raylib.NewVector2(raylib.GetMousePosition().X, raylib.GetMousePosition().Y),
|
|
raylib.NewVector2(raylib.GetMousePosition().X+10, raylib.GetMousePosition().Y+10),
|
|
raylib.NewVector2(raylib.GetMousePosition().X, raylib.GetMousePosition().Y+14),
|
|
cursorColor,
|
|
)
|
|
case toolPen:
|
|
raylib.DrawCircleLines(int32(raylib.GetMousePosition().X), int32(raylib.GetMousePosition().Y), toolPanelBrushSize/2, cursorColor)
|
|
}
|
|
|
|
drawToasts()
|
|
}
|
|
raylib.EndDrawing()
|
|
}
|
|
|
|
// GOODBYE
|
|
raylib.CloseWindow()
|
|
}
|