Amebis/wxWidgets
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make Capture() functions static members of AutoCaptureMechanism; adjusted slightly question boxes to fit the default wxGTK width of wxMessageBox

Browse Source 
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@56123 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
This commit is contained in:
Francesco Montorsi
2008-10-06 14:48:48 +00:00
parent 60a2264d5d
commit 6cd1aa9d38
5 changed files with 104 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

168
utils/screenshotgen/src/autocapture.cpp
View File

@@ -19,16 +19,20 @@
#endif
#include <wx/filename.h>
#include "autocapture.h"
wxBitmap Capture(int x, int y, int width, int height)
{
//Somehow wxScreenDC.Blit() doesn't work under Mac for now. Here is a trick.
#ifdef __WXMAC__
// ----------------------------------------------------------------------------
// AutoCaptureMechanism
// ----------------------------------------------------------------------------
//wxExecute(_T("screencapture -x ") + tempfile, wxEXEC_SYNC);
/* static */
wxBitmap AutoCaptureMechanism::Capture(int x, int y, int width, int height)
{
// Somehow wxScreenDC.Blit() doesn't work under Mac for now. Here is a trick.
#ifdef __WXMAC__
// wxExecute(_T("screencapture -x ") + tempfile, wxEXEC_SYNC);
system("screencapture -x /tmp/wx_screen_capture.png");
@@ -42,119 +46,114 @@ wxBitmap Capture(int x, int y, int width, int height)
wxBitmap screenshot = fullscreen.GetSubBitmap(wxRect(x,y,width,height));
#else //Under other paltforms, take a real screenshot
#else // Under other paltforms, take a real screenshot
//Create a DC for the whole screen area
// Create a DC for the whole screen area
wxScreenDC dcScreen;
//Create a Bitmap that will later on hold the screenshot image
//Note that the Bitmap must have a size big enough to hold the screenshot
//-1 means using the current default colour depth
// Create a Bitmap that will later on hold the screenshot image
// Note that the Bitmap must have a size big enough to hold the screenshot
// -1 means using the current default colour depth
wxBitmap screenshot(width, height, -1);
//Create a memory DC that will be used for actually taking the screenshot
// Create a memory DC that will be used for actually taking the screenshot
wxMemoryDC memDC;
//Tell the memory DC to use our Bitmap
//all drawing action on the memory DC will go to the Bitmap now
memDC.SelectObject(screenshot);
//Blit (in this case copy) the actual screen on the memory DC
//and thus the Bitmap
memDC.Blit( 0, //Copy to this X coordinate
0, //Copy to this Y coordinate
width, //Copy this width
height, //Copy this height
&dcScreen, //From where do we copy?
x, //What's the X offset in the original DC?
y //What's the Y offset in the original DC?
);
//Select the Bitmap out of the memory DC by selecting a new
//uninitialized Bitmap
memDC.SelectObject(wxNullBitmap);
#endif //#ifdef __WXMAC__
// wxMessageBox(_(""),_(""));
// Tell the memory DC to use our Bitmap
// all drawing action on the memory DC will go to the Bitmap now
memDC.SelectObject(screenshot);
// Blit (in this case copy) the actual screen on the memory DC
// and thus the Bitmap
memDC.Blit( 0, // Copy to this X coordinate
0, // Copy to this Y coordinate
width, // Copy this width
height, // Copy this height
&dcScreen, // From where do we copy?
x, // What's the X offset in the original DC?
y // What's the Y offset in the original DC?
);
// Select the Bitmap out of the memory DC by selecting a new
// uninitialized Bitmap
memDC.SelectObject(wxNullBitmap);
#endif // #ifdef __WXMAC__
return screenshot;
}
wxBitmap Capture(wxRect rect)
/* static */
wxBitmap AutoCaptureMechanism::Capture(wxRect rect)
{
wxPoint origin = rect.GetPosition();
return Capture(origin.x, origin.y, rect.GetWidth(), rect.GetHeight());
}
// ----------------------------------------------------------------------------
// AutoCaptureMechanism
// ----------------------------------------------------------------------------
void AutoCaptureMechanism::CaptureAll()
{
// start from the first page
m_notebook->SetSelection(0);
wxYield();
for(ControlList::iterator it = m_controlList.begin();
it != m_controlList.end();
++it)
for (ControlList::iterator it = m_controlList.begin();
it != m_controlList.end();
++it)
{
Control & ctrl = *it;
Control &ctrl = *it;
if(ctrl.flag == AJ_TurnPage) // Turn to next page
if (ctrl.flag == AJ_TurnPage) // Turn to next page
{
m_notebook->SetSelection(m_notebook->GetSelection() + 1);
wxYield();
continue;
}
// create the screenshot
wxBitmap screenshot = Capture(ctrl);
if(ctrl.flag & AJ_Union)
{
if (ctrl.flag & AJ_Union)
screenshot = Union(screenshot, Capture(*(++it)));
}
// and save it
Save(screenshot, ctrl.name);
}
}
wxBitmap AutoCaptureMechanism::Capture(Control & ctrl)
wxBitmap AutoCaptureMechanism::Capture(Control& ctrl)
{
if(ctrl.name == wxT("")) //no mannual specification for the control name
if (ctrl.name == wxT("")) // no manual specification for the control name
{
//Get name from wxRTTI
// Get its name from wxRTTI
ctrl.name = ctrl.ctrl->GetClassInfo()->GetClassName();
}
int choice = wxNO;
if(ctrl.flag & AJ_Dropdown)
// for drop-down controls we need the help of the user
if (ctrl.flag & AJ_Dropdown)
{
wxString caption = _("Do you wish to capture the dropdown list of ") + ctrl.name + _("?");
wxString notice = _("Click YES to capture it.\nAnd you MUST drop down the ") + ctrl.name + _(" in 3 seconds after close me.\n");
notice += _("Click NO otherwise.");
wxString caption = _("Drop-down screenshot...");
wxString msg =
wxString::Format(_("Do you wish to capture the drop-down list of '%s' ?\n\nIf you click YES you must drop-down the list of '%s' in 3 seconds after closing this message box.\nIf you click NO the screenshot for this control won't contain its drop-down list."),
ctrl.name, ctrl.name);
choice = wxMessageBox(notice, caption, wxYES_NO, m_notebook);
if(choice == wxYES)
choice = wxMessageBox(msg, caption, wxYES_NO, m_notebook);
if (choice == wxYES)
{
//Wait for 3 seconds
using std::clock;
using std::clock_t;
// Wait for 3 seconds
clock_t start = clock();
while(clock() - start < CLOCKS_PER_SEC * 3)
{
while (clock() - start < CLOCKS_PER_SEC * 3)
wxYieldIfNeeded();
}
}
}
wxRect rect = GetRect(ctrl.ctrl, ctrl.flag);
//Do some rect adjust so it can include the dropdown list
//Currently it only works well under MSW, not adjusted for Linux and Mac OS
if(ctrl.flag & AJ_Dropdown && choice == wxYES)
// Do some rect adjust so it can include the dropdown list;
// currently this only works well under MSW; not adjusted for Linux and Mac OS
if (ctrl.flag & AJ_Dropdown && choice == wxYES)
{
// #ifdef __WXMSW__
int h = rect.GetHeight();
@@ -162,27 +161,20 @@ wxBitmap AutoCaptureMechanism::Capture(Control & ctrl)
// #endif
}
//cut off "wx" and change them into lowercase.
// cut off "wx" and change the name into lowercase.
// e.g. wxButton will have a name of "button" at the end
ctrl.name.StartsWith(_T("wx"), &(ctrl.name));
ctrl.name.MakeLower();
wxBitmap screenshot = ::Capture(rect);
// take the screenshot
wxBitmap screenshot = Capture(rect);
if(ctrl.flag & AJ_RegionAdjust)
{
if (ctrl.flag & AJ_RegionAdjust)
PutBack(ctrl.ctrl);
}
return screenshot;
}
//if AJ_RegionAdjust is specified, the following line will use the label trick to adjust
//the region position and size
wxRect GetRect(wxWindow* ctrl, int flag);
//put the control back after the label trick(Using reparent/resizer approach)
void PutBack(wxWindow * ctrl);
wxBitmap AutoCaptureMechanism::Union(wxBitmap pic1, wxBitmap pic2)
{
int w1, w2, h1, h2, w, h;
@@ -226,28 +218,28 @@ wxBitmap AutoCaptureMechanism::Union(wxBitmap pic1, wxBitmap pic2)
void AutoCaptureMechanism::Save(wxBitmap screenshot, wxString fileName)
{
//Check if m_defaultDir already existed
if(!wxDirExists(m_dir))
// make sure m_dir exists
if (!wxDirExists(m_dir))
wxMkdir(m_dir);
wxString fullFileName = m_dir + wxFileName::GetPathSeparator() + fileName;
wxFileName fullFileName(m_dir, fileName + ".png");
//to prvent overwritten
while(wxFileName::FileExists(fullFileName + _T(".png"))) fullFileName += _T("_");
// do not overwrite already existing files with this name
while (fullFileName.FileExists())
fullFileName.SetName(fullFileName.GetName() + "_");
//Our Bitmap now has the screenshot, so let's save it as an png
//The filename itself is without extension.
screenshot.SaveFile(fullFileName + _T(".png"), wxBITMAP_TYPE_PNG);
// save the screenshot as a PNG
screenshot.SaveFile(fullFileName.GetFullPath(), wxBITMAP_TYPE_PNG);
}
wxRect AutoCaptureMechanism::GetRect(wxWindow* ctrl, int flag)
{
if(flag & AJ_RegionAdjust)
if (flag & AJ_RegionAdjust)
{
wxWindow * parent = ctrl->GetParent();
wxSizer * sizer = parent->GetSizer();
if(sizer)
if (sizer)
{
sizer->Detach(ctrl);
@@ -265,7 +257,7 @@ wxRect AutoCaptureMechanism::GetRect(wxWindow* ctrl, int flag)
wxStaticText* l[4];
for(int i = 0; i < 4; ++i)
for (int i = 0; i < 4; ++i)
l[i] = new wxStaticText(parent, wxID_ANY, wxT(" "));
m_grid->Add(l[0]);
@@ -289,7 +281,7 @@ wxRect AutoCaptureMechanism::GetRect(wxWindow* ctrl, int flag)
l[3]->GetScreenRect().GetTopLeft());
}
else //Actually it won't get here working with the current guiframe.h/guiframe.cpp
else // Actually it won't get here working with the current guiframe.h/guiframe.cpp
{
return ctrl->GetScreenRect().Inflate(m_border);
}
@@ -306,10 +298,10 @@ void AutoCaptureMechanism::PutBack(wxWindow * ctrl)
wxSizerItemList children = m_grid->GetChildren();
for(wxSizerItemList::iterator it = children.begin(); it != children.end(); ++it)
for (wxSizerItemList::iterator it = children.begin(); it != children.end(); ++it)
{
wxSizerItem* item = *it;
if(item->IsWindow()) delete (*it)->GetWindow();
if (item->IsWindow()) delete (*it)->GetWindow();
}
wxSizer * sizer = ctrl->GetParent()->GetSizer();

18
utils/screenshotgen/src/autocapture.h
View File

@@ -14,10 +14,8 @@
#include <vector>
#include <ctime>
// Global helper functions. to take screenshot for a rect region
wxBitmap Capture(wxRect rect);
wxBitmap Capture(int x, int y, int width, int height);
// TODO: document what these flags mean
enum AdjustFlags
{
AJ_Normal = 0,
@@ -36,7 +34,7 @@ enum AdjustFlags
class AutoCaptureMechanism
{
public:
AutoCaptureMechanism(wxNotebook * notebook,
AutoCaptureMechanism(wxNotebook *notebook,
wxString directory = wxT("screenshots"),
int border = 5)
: m_notebook(notebook), m_dir(directory), m_border(border) {}
@@ -63,8 +61,14 @@ public:
m_controlList.push_back(Control(0, wxT(""), AJ_TurnPage));
}
// capture all controls of the associated notebook
void CaptureAll();
// take a screenshot only of the given rect
static wxBitmap Capture(wxRect rect);
static wxBitmap Capture(int x, int y, int width, int height);
protected: // internal utils
struct Control
{
@@ -80,11 +84,11 @@ protected: // internal utils
wxBitmap Capture(Control & ctrl);
//if AJ_RegionAdjust is specified, the following line will use the label trick to adjust
//the region position and size
// if AJ_RegionAdjust is specified, the following line will use the label
// trick to adjust the region position and size
wxRect GetRect(wxWindow* ctrl, int flag);
//put the control back after the label trick(Using reparent/resizer approach)
// put the control back after the label trick(Using reparent/resizer approach)
void PutBack(wxWindow * ctrl);
wxBitmap Union(wxBitmap pic1, wxBitmap pic2);

10
utils/screenshotgen/src/guidesign.fbp
View File

@@ -238,7 +238,7 @@
</object>
</object>
</object>
<object class="wxBoxSizer" expanded="0">
<object class="wxBoxSizer" expanded="1">
<property name="minimum_size"></property>
<property name="name">bSizer0</property>
<property name="orient">wxVERTICAL</property>
@@ -1348,7 +1348,7 @@
<object class="notebookpage" expanded="1">
<property name="bitmap"></property>
<property name="label">Multiple choice Controls</property>
<property name="select">0</property>
<property name="select">1</property>
<object class="wxPanel" expanded="1">
<property name="bg"></property>
<property name="context_help"></property>
@@ -1638,7 +1638,7 @@
<property name="tooltip"></property>
<property name="window_extra_style"></property>
<property name="window_name"></property>
<property name="window_style"></property>
<property name="window_style">wxSUNKEN_BORDER</property>
<event name="OnChar"></event>
<event name="OnEnterWindow"></event>
<event name="OnEraseBackground"></event>
@@ -1708,7 +1708,7 @@
<property name="tooltip">wxListCtrl</property>
<property name="window_extra_style"></property>
<property name="window_name"></property>
<property name="window_style"></property>
<property name="window_style">wxSUNKEN_BORDER</property>
<event name="OnChar"></event>
<event name="OnEnterWindow"></event>
<event name="OnEraseBackground"></event>
@@ -2155,7 +2155,7 @@
<object class="notebookpage" expanded="1">
<property name="bitmap"></property>
<property name="label">Picker Controls</property>
<property name="select">1</property>
<property name="select">0</property>
<object class="wxPanel" expanded="1">
<property name="bg"></property>
<property name="context_help"></property>

8
utils/screenshotgen/src/guiframe.cpp
View File

@@ -227,10 +227,10 @@ GUIFrame::GUIFrame( wxWindow* parent, wxWindowID id, const wxString& title, cons
fgSizer2->Add( m_staticBox1, 1, wxALL|wxALIGN_CENTER_VERTICAL|wxALIGN_CENTER_HORIZONTAL|wxEXPAND, 20 );
m_treeCtrl1 = new wxTreeCtrl( m_panel2, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTR_DEFAULT_STYLE );
m_treeCtrl1 = new wxTreeCtrl( m_panel2, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTR_DEFAULT_STYLE|wxSUNKEN_BORDER );
fgSizer2->Add( m_treeCtrl1, 1, wxALL|wxEXPAND|wxALIGN_CENTER_HORIZONTAL|wxALIGN_CENTER_VERTICAL, 20 );
m_listCtrl1 = new wxListCtrl( m_panel2, wxID_ANY, wxDefaultPosition, wxDefaultSize, 0 );
m_listCtrl1 = new wxListCtrl( m_panel2, wxID_ANY, wxDefaultPosition, wxDefaultSize, 0|wxSUNKEN_BORDER );
m_listCtrl1->SetToolTip( _("wxListCtrl") );
fgSizer2->Add( m_listCtrl1, 1, wxALL|wxALIGN_CENTER_HORIZONTAL|wxALIGN_CENTER_VERTICAL|wxEXPAND, 20 );
@@ -274,7 +274,7 @@ GUIFrame::GUIFrame( wxWindow* parent, wxWindowID id, const wxString& title, cons
m_panel2->SetSizer( fgSizer2 );
m_panel2->Layout();
fgSizer2->Fit( m_panel2 );
m_notebook1->AddPage( m_panel2, _("Multiple choice Controls"), false );
m_notebook1->AddPage( m_panel2, _("Multiple choice Controls"), true );
m_panel3 = new wxPanel( m_notebook1, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL );
wxBoxSizer* bSizer2;
bSizer2 = new wxBoxSizer( wxVERTICAL );
@@ -353,7 +353,7 @@ GUIFrame::GUIFrame( wxWindow* parent, wxWindowID id, const wxString& title, cons
m_panel4->SetSizer( fgSizer5 );
m_panel4->Layout();
fgSizer5->Fit( m_panel4 );
m_notebook1->AddPage( m_panel4, _("Picker Controls"), true );
m_notebook1->AddPage( m_panel4, _("Picker Controls"), false );
m_panel5 = new wxPanel( m_notebook1, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL );
wxFlexGridSizer* fgSizer4;
fgSizer4 = new wxFlexGridSizer( 2, 2, 0, 0 );

5
utils/screenshotgen/src/screenshot_main.cpp
View File

@@ -80,6 +80,9 @@ ScreenshotFrame::~ScreenshotFrame()
*/
void ScreenshotFrame::InitFBControls()
{
// For some reason, wxFormBuilder does not set the scrollbar range
m_scrollBar1->SetScrollbar(50, 1, 100, 1);
// Do the default selection for wxComboBox
m_comboBox1->Select(0);
@@ -250,7 +253,7 @@ void ScreenshotFrame::OnCaptureAllControls(wxCommandEvent& WXUNUSED(event))
// check if there are other screenshots taken before
if (wxFileName::DirExists(dir))
{
int choice = wxMessageBox(_("It seems that you have already generated some screenshots.\nClick YES to delete them all (recommended) or NO to preserve them.\nClick CANCEL to cancel this auto-capture operation."),
int choice = wxMessageBox(_("It seems that you have already generated some screenshots.\n\nClick YES to delete them all (recommended) or NO to preserve them.\nClick CANCEL to cancel this auto-capture operation."),
_("Delete existing screenshots?"),
wxYES_NO|wxCANCEL|wxICON_QUESTION, this);
switch(choice)