mirrors/electron
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

Return screenshot's data buffer instead of directly saving to file.

Browse source
This commit is contained in:
Cheng Zhao 2013-11-21 22:03:29 +08:00
parent 36ecb35cb1
commit 8d6764e0a0
2 changed files with 10 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
browser/native_window.cc
View file

@ -200,14 +200,12 @@ bool NativeWindow::SetIcon(const std::string& str_path) {
}
void NativeWindow::CapturePage(const gfx::Rect& rect,
const base::FilePath& path,
const CapturePageCallback& callback) {
GetWebContents()->GetRenderViewHost()->CopyFromBackingStore(
rect,
gfx::Size(),
base::Bind(&NativeWindow::OnCapturePageDone,
base::Unretained(this),
path,
callback));
}
@ -387,27 +385,13 @@ void NativeWindow::Observe(int type,
}
}
void NativeWindow::OnCapturePageDone(const base::FilePath& filename,
const CapturePageCallback& callback,
void NativeWindow::OnCapturePageDone(const CapturePageCallback& callback,
bool succeed,
const SkBitmap& bitmap) {
if (!succeed) {
callback.Run(false);
return;
}
std::vector<unsigned char> data;
bool encoded = gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, true, &data);
if (!encoded) {
callback.Run(false);
return;
}
int written = file_util::WriteFile(
filename,
reinterpret_cast<const char*>(&data[0]),
data.size());
callback.Run(written > 0);
if (succeed)
gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, true, &data);
callback.Run(data);
}
void NativeWindow::OnRendererMessage(const string16& channel,

12
browser/native_window.h
View file

@ -49,6 +49,9 @@ class NativeWindow : public brightray::DefaultWebContentsDelegate,
public content::WebContentsObserver,
public content::NotificationObserver {
public:
typedef base::Callback<void(const std::vector<unsigned char>& buffer)>
CapturePageCallback;
virtual ~NativeWindow();
// Create window with existing WebContents.
@ -108,11 +111,9 @@ class NativeWindow : public brightray::DefaultWebContentsDelegate,
virtual void RestartHangMonitorTimeout();
virtual bool SetIcon(const std::string& path);
// Captures the page with |rect| and saves the image to |path|, |callback|
// would be called when capturing is done.
typedef base::Callback<void(bool succeed)> CapturePageCallback;
// Captures the page with |rect|, |callback| would be called when capturing is
// done.
virtual void CapturePage(const gfx::Rect& rect,
const base::FilePath& path,
const CapturePageCallback& callback);
// The same with closing a tab in a real browser.
@ -192,8 +193,7 @@ class NativeWindow : public brightray::DefaultWebContentsDelegate,
void RendererUnresponsiveDelayed();
// Called when CapturePage has done.
void OnCapturePageDone(const base::FilePath& filename,
const CapturePageCallback& callback,
void OnCapturePageDone(const CapturePageCallback& callback,
bool succeed,
const SkBitmap& bitmap);