[cpp] screenshot support

shish · Nov 18, 2022 · c08ff64 · c08ff64
1 parent f78b607
commit c08ff64
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 3 deletions.
diff --git a/cpp/src/args.cpp b/cpp/src/args.cpp
@@ -12,6 +12,7 @@ Args::Args(int argc, char *argv[]) {
     args::Flag debug_apu(parser, "debug-apu", "Debug APU", {'a', "debug-apu"});
     args::ValueFlag<int> frames(parser, "frames", "Exit after N frames", {'f', "frames"});
     args::ValueFlag<int> profile(parser, "profile", "Exit after N seconds", {'p', "profile"});
+    args::ValueFlag<std::string> screenshot(parser, "screenshot", "Write a BMP to this file on exit", {'s', "screenshot"});
     args::Flag turbo(parser, "turbo", "No sleep between frames", {'t', "turbo"});
     args::Positional<std::string> rom(parser, "rom", "Path to a .gb file");
     args::CompletionFlag completion(parser, {"complete"});
@@ -39,6 +40,7 @@ Args::Args(int argc, char *argv[]) {
     this->debug_gpu = debug_gpu;
     this->frames = frames ? args::get(frames) : 0;
     this->profile = profile ? args::get(profile) : 0;
+    this->screenshot = screenshot ? args::get(screenshot) : "";
     this->turbo = turbo;
     this->rom = args::get(rom);
 }
diff --git a/cpp/src/args.h b/cpp/src/args.h
@@ -14,6 +14,7 @@ class Args {
     bool debug_apu;
     int frames;
     int profile;
+    std::string screenshot;
     bool turbo;
     std::string rom;
 };

diff --git a/cpp/src/gameboy.cpp b/cpp/src/gameboy.cpp
@@ -4,7 +4,7 @@ GameBoy::GameBoy(Args *args) {
     this->cart = new Cart(args->rom);
     this->ram = new RAM(this->cart);
     this->cpu = new CPU(this->ram, args->debug_cpu);
-    this->gpu = new GPU(this->cpu, cart->name, args->headless, args->debug_gpu);
+    this->gpu = new GPU(this->cpu, cart->name, args->screenshot, args->headless, args->debug_gpu);
     this->buttons = new Buttons(this->cpu, args->headless);
     if(!args->silent) new APU(this->cpu, args->debug_apu);
     this->clock = new Clock(this->buttons, args->frames, args->profile, args->turbo);

diff --git a/cpp/src/gpu.cpp b/cpp/src/gpu.cpp
@@ -1,4 +1,5 @@
 #include <SDL2/SDL.h>
+#include <iostream>
 
 #include "consts.h"
 #include "gpu.h"
@@ -17,7 +18,8 @@ u32 bmask = 0x00ff0000;
 u32 amask = 0xff000000;
 #endif
 
-GPU::GPU(CPU *cpu, char *title, bool headless, bool debug) {
+GPU::GPU(CPU *cpu, char *title, std::string screenshot, bool headless, bool debug) {
+    this->screenshot = screenshot;
     this->cpu = cpu;
     this->debug = debug;
 
@@ -61,6 +63,10 @@ GPU::GPU(CPU *cpu, char *title, bool headless, bool debug) {
 };
 
 GPU::~GPU() {
+    std::cout << "Screenshot " << this->screenshot << "\n";
+    if(this->screenshot.length() > 0) {
+        SDL_SaveBMP(this->buffer, this->screenshot.c_str());
+    }
     SDL_FreeSurface(this->buffer);
     if(this->hw_window) SDL_DestroyWindow(this->hw_window);
     SDL_Quit();

diff --git a/cpp/src/gpu.h b/cpp/src/gpu.h
@@ -51,6 +51,7 @@ struct Sprite {
 class GPU {
 private:
     bool debug;
+    std::string screenshot;
     SDL_Window *hw_window;
     SDL_Texture *hw_buffer;
     SDL_Renderer *hw_renderer;
@@ -62,7 +63,7 @@ class GPU {
     CPU *cpu;
 
 public:
-    GPU(CPU *cpu, char *title, bool headless, bool debug);
+    GPU(CPU *cpu, char *title, std::string screenshot, bool headless, bool debug);
     ~GPU();
     void tick();