added screenshot capture

app_main.go

@@ -137,6 +137,7 @@ func RunApplication() {
 	InitUnitext()
 	InitAlert()
 	InitTransition()
+	InitScreenshot()
 	defer FreeTransition()
 	InitPopupDialog()
 	defer FreePopupDialog()
@@ -300,6 +301,7 @@ func RunApplication() {
 			UpdateTransition()
 			UpdateMenuManager(updateDelta)
 			UpdateAlert(updateDelta)
+			UpdateScreenshot()
 
 			//update screen
 			if !TheTransitionManager.ShowTransition {

game_screen.go

@@ -2965,7 +2965,7 @@ func (gs *GameScreen) BeforeScreenEnd() {
 	}
 
 	if err := SaveSettings(); err != nil {
-		ErrorLogger.Printf("failed to save settings %v", err)
+		ErrorLogger.Printf("failed to save settings: %v", err)
 		DisplayAlert("failed to save settings")
 	}
 

options.go

@@ -41,6 +41,8 @@ const (
 	ZoomOutKey
 	ZoomInKey
 
+	ScreenshotKey
+
 	ToggleDebugMsg
 	ToggleLogNoteEvent
 	ToggleDebugGraphics
@@ -93,6 +95,8 @@ func init() {
 	DefaultKM[ZoomOutKey] = rl.KeyLeftBracket
 	DefaultKM[ZoomInKey] = rl.KeyRightBracket
 
+	DefaultKM[ScreenshotKey] = rl.KeyF12
+
 	DefaultKM[ToggleDebugMsg] = rl.KeyF1
 	DefaultKM[ToggleLogNoteEvent] = rl.KeyF2
 	DefaultKM[ToggleDebugGraphics] = rl.KeyF3
@@ -141,6 +145,8 @@ func init() {
 	KeyHumanName[ZoomOutKey] = "note spacing up"
 	KeyHumanName[ZoomInKey] = "note spacing down"
 
+	KeyHumanName[ScreenshotKey] = "screenshot"
+
 	KeyHumanName[ToggleDebugMsg] = "toggle debug message"
 	KeyHumanName[ToggleLogNoteEvent] = "toggle note event"
 	KeyHumanName[ToggleDebugGraphics] = "toggle debug graphics"

options_screen.go

@@ -658,7 +658,13 @@ func NewOptionsControlsScreen() *BaseOptionsScreen {
 
 			// add extra bottom margin
 			switch key {
-			case SongResetKey, NoteScrollDownKey, AudioSpeedDownKey, AudioOffsetDownKey, JumpToBookMarkKey:
+			case SongResetKey,
+				NoteScrollDownKey,
+				AudioSpeedDownKey,
+				AudioOffsetDownKey,
+				JumpToBookMarkKey,
+				ZoomInKey:
+
 				item.BottomMargin += extraBottomMargin
 			}
 

screenshot.go

@@ -0,0 +1,84 @@
+package fnf
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+
+	rl "github.com/gen2brain/raylib-go/raylib"
+)
+
+var TheScreenshotManager struct {
+	screenshotCounter int
+	InputId           InputGroupId
+}
+
+func InitScreenshot() {
+	ts := &TheScreenshotManager
+
+	ts.InputId = NewInputGroupId()
+}
+
+func takeScreenshot() {
+	ts := &TheScreenshotManager
+
+	dirPath, err := RelativePath("./")
+	if err != nil {
+		ErrorLogger.Printf("failed to take screenshot: %v", err)
+		DisplayAlert("failed to take screenshot")
+		return
+	}
+
+	entries, err := os.ReadDir(dirPath)
+	if err != nil {
+		ErrorLogger.Printf("failed to take screenshot: %v", err)
+		DisplayAlert("failed to take screenshot")
+		return
+	}
+
+	const fmtStr = "screenshot-%03d.png"
+
+	screenshotName := fmt.Sprintf(fmtStr, ts.screenshotCounter)
+
+	for _, entry := range entries {
+		if entry.Name() == screenshotName {
+			ts.screenshotCounter += 1
+			screenshotName = fmt.Sprintf(fmtStr, ts.screenshotCounter)
+		}
+	}
+
+	// actually take screenshot
+	img := rl.LoadImageFromTexture(TheRenderTexture.Texture)
+	if !rl.IsImageReady(img) {
+		ErrorLogger.Printf("failed to take screenshot: failed to load image from render texture")
+		DisplayAlert("failed to take screenshot")
+		return
+	}
+	defer rl.UnloadImage(img)
+
+	rl.ImageFlipVertical(img)
+
+	data := rl.ExportImageToMemory(*img, ".png")
+
+	fullPath := filepath.Join(dirPath, screenshotName)
+
+	err = os.WriteFile(fullPath, data, 0664)
+	if err != nil {
+		ErrorLogger.Printf("failed to take screenshot: %v", err)
+		DisplayAlert("failed to take screenshot")
+		return
+	}
+
+	FnfLogger.Printf("saved screenshot as %s", fullPath)
+	DisplayAlert(fmt.Sprintf("saved screenshot as %s", screenshotName))
+
+	ts.screenshotCounter += 1
+}
+
+func UpdateScreenshot() {
+	ts := &TheScreenshotManager
+
+	if AreKeysPressed(ts.InputId, TheKM[ScreenshotKey]) {
+		takeScreenshot()
+	}
+}