Add Screenshot command

data/gui.xml

@@ -497,7 +497,18 @@
         <param name="byGrid" value="true" />
       </key>
       <key command="NewFrame">
-	<param name="content" value="cel" />
+        <param name="content" value="cel" />
+      </key>
+      <key command="Screenshot" />
+      <key command="Screenshot">
+        <param name="srgb" value="false" />
+      </key>
+      <key command="Screenshot">
+        <param name="save" value="true" />
+      </key>
+      <key command="Screenshot">
+        <param name="save" value="true" />
+        <param name="srgb" value="false" />
       </key>
     </commands>
 

data/strings/en.ini

@@ -396,6 +396,11 @@ SaveFileAs = Save File As
 SaveFileCopyAs = Export
 SaveMask = Save Selection
 SavePalette = Save Palette
+Screenshot = Screenshot
+Screenshot_Open = Take & Open Screenshot
+Screenshot_Save = Take & Save Screenshot
+Screenshot_sRGB = (sRGB Color Profile)
+Screenshot_DisplayCS = (Display Color Profile)
 Scroll = Scroll {0}
 ScrollCenter = Scroll to center of canvas
 SelectTile = Select Tile

src/app/CMakeLists.txt

@@ -294,6 +294,7 @@ if(ENABLE_UI)
     commands/filters/filter_preview.cpp
     commands/filters/filter_target_buttons.cpp
     commands/filters/filter_window.cpp
+    commands/screenshot.cpp
     file_selector.cpp
     modules/editors.cpp
     modules/gfx.cpp

src/app/commands/commands_list.h

@@ -127,6 +127,7 @@ FOR_EACH_COMMAND(ReverseFrames)
 FOR_EACH_COMMAND(Rotate)
 FOR_EACH_COMMAND(SaveMask)
 FOR_EACH_COMMAND(SavePalette)
+FOR_EACH_COMMAND(Screenshot)
 FOR_EACH_COMMAND(Scroll)
 FOR_EACH_COMMAND(ScrollCenter)
 FOR_EACH_COMMAND(SelectTile)

src/app/commands/screenshot.cpp

@@ -0,0 +1,138 @@
+// Aseprite
+// Copyright (C) 2019  Igara Studio S.A.
+//
+// This program is distributed under the terms of
+// the End-User License Agreement for Aseprite.
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include "app/app.h"
+#include "app/cmd/convert_color_profile.h"
+#include "app/commands/command.h"
+#include "app/commands/new_params.h"
+#include "app/context.h"
+#include "app/doc.h"
+#include "app/file/file.h"
+#include "app/i18n/strings.h"
+#include "app/resource_finder.h"
+#include "base/fs.h"
+#include "doc/cel.h"
+#include "doc/color.h"
+#include "doc/layer.h"
+#include "doc/sprite.h"
+#include "fmt/format.h"
+#include "os/display.h"
+#include "os/surface.h"
+#include "ui/manager.h"
+
+namespace app {
+
+using namespace ui;
+
+struct ScreenshotParams : public NewParams {
+  Param<bool> save { this, false, "save" };
+  Param<bool> srgb { this, true, "srgb" };
+};
+
+class ScreenshotCommand : public CommandWithNewParams<ScreenshotParams> {
+public:
+  ScreenshotCommand();
+
+protected:
+  void onExecute(Context* ctx) override;
+  std::string onGetFriendlyName() const override;
+};
+
+ScreenshotCommand::ScreenshotCommand()
+  : CommandWithNewParams<ScreenshotParams>(CommandId::Screenshot(), CmdUIOnlyFlag)
+{
+}
+
+void ScreenshotCommand::onExecute(Context* ctx)
+{
+  if (!ctx->isUIAvailable())
+    return;
+
+  app::ResourceFinder rf(false);
+  rf.includeDesktopDir("");
+
+  os::Display* display = ui::Manager::getDefault()->getDisplay();
+  os::Surface* surface = display->getSurface();
+  std::string fn;
+
+  if (params().save()) {
+    for (int i=0; i<10000; ++i) {
+      fn = base::join_path(rf.defaultFilename(),
+                           fmt::format("screenshot-{}.png", i));
+      if (!base::is_file(fn))
+        break;
+    }
+  }
+  else {
+    fn = "screenshot.png";
+  }
+
+  doc::ImageSpec spec(doc::ColorMode::RGB,
+                      surface->width(),
+                      surface->height(),
+                      0,    // Mask color
+                      display->colorSpace()->gfxColorSpace());
+
+  doc::Sprite* spr = doc::Sprite::MakeStdSprite(spec);
+  static_cast<doc::LayerImage*>(spr->firstLayer())->configureAsBackground();
+
+  doc::Cel* cel = spr->firstLayer()->cel(0);
+  doc::Image* img = cel->image();
+
+  for (int y=0; y<img->height(); ++y) {
+    for (int x=0; x<img->width(); ++x) {
+      gfx::Color c = surface->getPixel(x, y);
+
+      img->putPixel(x, y, doc::rgba(gfx::getr(c),
+                                    gfx::getg(c),
+                                    gfx::getb(c), 255));
+    }
+  }
+
+  std::unique_ptr<Doc> doc(new Doc(spr));
+  doc->setFilename(fn);
+
+  // Convert color profile to sRGB
+  if (params().srgb())
+    cmd::convert_color_profile(spr, gfx::ColorSpace::MakeSRGB());
+
+  if (params().save()) {
+    save_document(ctx, doc.get());
+  }
+  else {
+    doc->setContext(ctx);
+    doc.release();
+  }
+}
+
+std::string ScreenshotCommand::onGetFriendlyName() const
+{
+  std::string name;
+  if (params().save())
+    name = Strings::commands_Screenshot_Save();
+  else
+    name = Strings::commands_Screenshot_Open();
+  if (params().srgb()) {
+    name.push_back(' ');
+    name += Strings::commands_Screenshot_sRGB();
+  }
+  else {
+    name.push_back(' ');
+    name += Strings::commands_Screenshot_DisplayCS();
+  }
+  return name;
+}
+
+Command* CommandFactory::createScreenshotCommand()
+{
+  return new ScreenshotCommand;
+}
+
+} // namespace app