Destroyed screenshot_tool (markdown)
omar
parent
65cc82929e
commit
55e3511ff7
@ -1,286 +0,0 @@
|
||||
## Screenshot tool
|
||||
|
||||
Simple tool to help creating screenshot of ImGui windows (requires OpenGL + stb_image_write.h)
|
||||
|
||||
|
||||
|
||||
It can also create big screenshot of a large window by stitching multiple shots together.
|
||||
|
||||
### Source (V2)
|
||||
|
||||
```
|
||||
// ImGui Screenshot Maker (OpenGL only)
|
||||
|
||||
#define _CRT_SECURE_NO_WARNINGS
|
||||
#include "imgui.h"
|
||||
#define IMGUI_DEFINE_MATH_OPERATORS
|
||||
#include "imgui_internal.h"
|
||||
|
||||
#include <examples/libs/glfw/include/GLFW/glfw3.h>
|
||||
#define STB_IMAGE_WRITE_IMPLEMENTATION
|
||||
#include "stb_image_write.h"
|
||||
|
||||
// Helper class for simple bitmap manipulation (not particularly efficient!)
|
||||
struct ImageBuf
|
||||
{
|
||||
typedef unsigned int u32;
|
||||
|
||||
int Width, Height;
|
||||
u32* Data;
|
||||
|
||||
ImageBuf() { Width = Height = 0; Data = NULL; }
|
||||
~ImageBuf() { Clear(); }
|
||||
void Clear() { if (Data) free(Data); Data = NULL; }
|
||||
|
||||
void CreateEmpty(int w, int h)
|
||||
{
|
||||
Clear();
|
||||
Width = w;
|
||||
Height = h;
|
||||
Data = (u32*)malloc(Width * Height * 4);
|
||||
memset(Data, 0, Width * Height * 4);
|
||||
}
|
||||
|
||||
void CreateFromCaptureGL(int x, int y, int w, int h)
|
||||
{
|
||||
Clear();
|
||||
Width = w;
|
||||
Height = h;
|
||||
Data = (u32*)malloc(Width * Height * 4);
|
||||
glPixelStorei(GL_PACK_ALIGNMENT, 1);
|
||||
glReadPixels(x, y, w, h, GL_RGBA, GL_UNSIGNED_BYTE, Data);
|
||||
RemoveAlpha();
|
||||
}
|
||||
|
||||
void SaveFile(const char* filename)
|
||||
{
|
||||
stbi_write_png(filename, Width, Height, 4, Data, Width * 4);
|
||||
}
|
||||
|
||||
void RemoveAlpha()
|
||||
{
|
||||
u32* p = Data;
|
||||
int n = Width * Height;
|
||||
while (n-- > 0)
|
||||
{
|
||||
*p |= 0xFF000000;
|
||||
p++;
|
||||
}
|
||||
}
|
||||
|
||||
void BlitTo(ImageBuf* dst, int src_x, int src_y, int dst_x, int dst_y, int w, int h)
|
||||
{
|
||||
ImageBuf* src = this;
|
||||
IM_ASSERT(dst != src);
|
||||
IM_ASSERT(dst != NULL);
|
||||
IM_ASSERT(src_x >= 0 && src_y >= 0);
|
||||
IM_ASSERT(src_x + w <= src->Width);
|
||||
IM_ASSERT(src_y + h <= src->Height);
|
||||
IM_ASSERT(dst_x >= 0 && dst_y >= 0);
|
||||
IM_ASSERT(dst_x + w <= dst->Width);
|
||||
IM_ASSERT(dst_y + h <= dst->Height);
|
||||
for (int y = 0; y < h; y++)
|
||||
memcpy(dst->Data + dst_x + (dst_y + y) * dst->Width, src->Data + src_x + (src_y + y) * src->Width, w * 4);
|
||||
}
|
||||
|
||||
void FlipVertical()
|
||||
{
|
||||
int comp = 4;
|
||||
int stride = Width * comp;
|
||||
unsigned char* line_tmp = new unsigned char[stride];
|
||||
unsigned char* line_a = (unsigned char*)Data;
|
||||
unsigned char* line_b = (unsigned char*)Data + (stride * (Height - 1));
|
||||
while (line_a < line_b)
|
||||
{
|
||||
memcpy(line_tmp, line_a, stride);
|
||||
memcpy(line_a, line_b, stride);
|
||||
memcpy(line_b, line_tmp, stride);
|
||||
line_a += stride;
|
||||
line_b -= stride;
|
||||
}
|
||||
delete[] line_tmp;
|
||||
}
|
||||
u32* GetPtr(int x, int y)
|
||||
{
|
||||
return &Data[x + y * Width];
|
||||
}
|
||||
u32 GetPixel(int x, int y) const
|
||||
{
|
||||
return Data[x + y * Width];
|
||||
}
|
||||
};
|
||||
|
||||
struct ScreenshotMaker
|
||||
{
|
||||
ImGuiStorage Storage;
|
||||
float BorderSize;
|
||||
bool LargePictureMode;
|
||||
ImRect VisibleRect;
|
||||
bool CaptureActive;
|
||||
int CaptureFrame;
|
||||
ImRect CaptureRect;
|
||||
int CaptureLargeOffY;
|
||||
ImGuiWindow* CaptureWindow;
|
||||
ImageBuf Output;
|
||||
|
||||
ScreenshotMaker()
|
||||
{
|
||||
BorderSize = 16.0f;
|
||||
LargePictureMode = false;
|
||||
CaptureActive = false;
|
||||
CaptureFrame = 0;
|
||||
CaptureLargeOffY = 0;
|
||||
CaptureWindow = NULL;
|
||||
}
|
||||
|
||||
void CaptureStart(ImGuiWindow* window)
|
||||
{
|
||||
CaptureActive = true;
|
||||
CaptureFrame = 0;
|
||||
Output.Clear();
|
||||
CaptureWindow = window;
|
||||
if (!CaptureWindow)
|
||||
LargePictureMode = false;
|
||||
|
||||
if (LargePictureMode)
|
||||
{
|
||||
ImGui::SetWindowSize(CaptureWindow->Name, ImVec2(CaptureWindow->SizeFull.x, CaptureWindow->SizeContents.y));
|
||||
Output.CreateEmpty((int)(CaptureWindow->Size.x + BorderSize * 2), (int)(CaptureWindow->Size.y + BorderSize * 2));
|
||||
}
|
||||
else
|
||||
{
|
||||
// Capture immediately
|
||||
ImGuiIO& io = ImGui::GetIO();
|
||||
Output.CreateFromCaptureGL((int)CaptureRect.Min.x, (int)io.DisplaySize.y - (int)CaptureRect.Max.y, (int)CaptureRect.GetWidth(), (int)CaptureRect.GetHeight());
|
||||
Output.FlipVertical();
|
||||
CaptureEnd();
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
void CaptureUpdate()
|
||||
{
|
||||
if (!LargePictureMode)
|
||||
return;
|
||||
|
||||
int TIME_STEP = 15;
|
||||
IM_ASSERT(TIME_STEP > 5);
|
||||
|
||||
int chunk_no = CaptureFrame / TIME_STEP;
|
||||
int chunk_w = (int)CaptureRect.GetWidth();
|
||||
int chunk_h = (int)CaptureRect.GetHeight();
|
||||
|
||||
if ((CaptureFrame % TIME_STEP) == 2)
|
||||
{
|
||||
// Move window
|
||||
ImVec2 p = CaptureRect.Min + ImVec2(BorderSize, BorderSize);
|
||||
p.y -= chunk_no * CaptureRect.GetHeight();
|
||||
ImGui::SetWindowPos(CaptureWindow->Name, p);
|
||||
}
|
||||
|
||||
if ((CaptureFrame % TIME_STEP) == 4)
|
||||
{
|
||||
// Capture, stitch into output
|
||||
ImGuiIO& io = ImGui::GetIO();
|
||||
ImageBuf img;
|
||||
img.CreateFromCaptureGL((int)CaptureRect.Min.x, (int)io.DisplaySize.y - (int)CaptureRect.Max.y, chunk_w, chunk_h);
|
||||
img.FlipVertical();
|
||||
int chunk_y = chunk_no * chunk_h;
|
||||
img.BlitTo(&Output, 0, 0, 0, chunk_y, chunk_w, ImMin(chunk_h, Output.Height - chunk_y));
|
||||
char buf[128];
|
||||
sprintf(buf, "out%03d.png", chunk_no);
|
||||
//img.SaveFile(buf);
|
||||
if (CaptureWindow->Pos.y + CaptureWindow->SizeFull.y < CaptureRect.Max.y)
|
||||
CaptureEnd();
|
||||
}
|
||||
|
||||
CaptureFrame++;
|
||||
}
|
||||
|
||||
void CaptureEnd()
|
||||
{
|
||||
if (Output.Data)
|
||||
Output.SaveFile("out.png");
|
||||
CaptureActive = false;
|
||||
}
|
||||
|
||||
void Draw(bool* p_open = NULL)
|
||||
{
|
||||
if (!ImGui::Begin("Screenshot Maker", p_open))
|
||||
{
|
||||
ImGui::End();
|
||||
return;
|
||||
}
|
||||
|
||||
ImGuiContext& g = *GImGui;
|
||||
ImGuiIO& io = ImGui::GetIO();
|
||||
|
||||
bool capture = ImGui::Button("ALT+C: Capture");
|
||||
if (CaptureActive)
|
||||
{
|
||||
ImGui::SameLine();
|
||||
ImGui::Text("Capturing...");
|
||||
}
|
||||
ImGui::SliderFloat("Border", &BorderSize, 0.0f, 32.0f, "%.0f");
|
||||
ImGui::Checkbox("Large Window Capture Mode (1 window)", &LargePictureMode);
|
||||
ImGui::Separator();
|
||||
|
||||
ImVector<ImGuiWindow*> windows_selected;
|
||||
VisibleRect = ImRect(ImVec2(0,0), io.DisplaySize);
|
||||
ImRect capture_rect;
|
||||
|
||||
ImGui::Text("Windows:");
|
||||
for (int i = 0; i < g.Windows.size(); i++)
|
||||
{
|
||||
ImGuiWindow* window = g.Windows[i];
|
||||
if (!window->Active && !window->WasActive)
|
||||
continue;
|
||||
if (window->Flags & ImGuiWindowFlags_Popup)
|
||||
{
|
||||
CaptureRect.Add(window->Rect());
|
||||
continue;
|
||||
}
|
||||
if (window->Flags & ImGuiWindowFlags_ChildWindow)
|
||||
continue;
|
||||
|
||||
ImGui::PushID(window);
|
||||
bool* p_selected = (bool*)Storage.GetIntRef(window->RootWindow->ID, 0);
|
||||
ImGui::Checkbox("##checkbox", p_selected);
|
||||
ImGui::SameLine();
|
||||
ImGui::TextUnformatted(window->Name);
|
||||
if (*p_selected)
|
||||
{
|
||||
capture_rect.Add(window->Rect());
|
||||
windows_selected.push_back(window);
|
||||
}
|
||||
ImGui::SameLine(160);
|
||||
ImGui::PushItemWidth(120);
|
||||
ImGui::DragFloat2("Pos", &window->PosFloat.x, 0.05f, 0.0f, 0.0f, "%.0f");
|
||||
ImGui::SameLine();
|
||||
ImGui::DragFloat2("Size", &window->SizeFull.x, 0.05f, 0.0f, 0.0f, "%.0f");
|
||||
ImGui::PopItemWidth();
|
||||
ImGui::PopID();
|
||||
}
|
||||
|
||||
capture_rect.Expand(BorderSize);
|
||||
capture_rect.Clip(VisibleRect);
|
||||
|
||||
if (!CaptureActive)
|
||||
CaptureRect = capture_rect;
|
||||
|
||||
char buf[128];
|
||||
sprintf(buf, "%.0f x %.0f", CaptureRect.GetWidth(), capture_rect.GetHeight());
|
||||
g.OverlayDrawList.AddText(ImGui::GetWindowFont(), ImGui::GetWindowFontSize(), ImVec2(1,1), IM_COL32_WHITE, buf);
|
||||
g.OverlayDrawList.AddRect(CaptureRect.Min-ImVec2(1,1), CaptureRect.Max+ImVec2(1,1), IM_COL32_WHITE);
|
||||
|
||||
capture |= io.KeyAlt && ImGui::IsKeyPressed(ImGui::GetKeyIndex(ImGuiKey_C));
|
||||
if (capture)
|
||||
CaptureStart(windows_selected.empty() ? NULL : windows_selected[0]);
|
||||
|
||||
if (CaptureActive)
|
||||
CaptureUpdate();
|
||||
|
||||
ImGui::End();
|
||||
}
|
||||
};
|
||||
```
|
||||
Loading…
Reference in New Issue
Block a user