]>
git.saurik.com Git - wxWidgets.git/blob - utils/screenshotgen/src/autocapture.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: autocapture.cpp
3 // Purpose: Implement wxCtrlMaskOut class
4 // Author: Utensil Candel (UtensilCandel@@gmail.com)
6 // Licence: wxWindows license
7 /////////////////////////////////////////////////////////////////////////////
9 // For compilers that support precompilation, includes "wx/wx.h".
10 #include "wx/wxprec.h"
16 // for all others, include the necessary headers
21 #include <wx/filename.h>
23 #include "autocapture.h"
26 wxBitmap
Capture(int x
, int y
, int width
, int height
)
28 //Somehow wxScreenDC.Blit() doesn't work under Mac for now. Here is a trick.
31 //wxExecute(_T("screencapture -x ") + tempfile, wxEXEC_SYNC);
33 system("screencapture -x /tmp/wx_screen_capture.png");
39 fullscreen
= wxBitmap(_T("/tmp/wx_screen_capture.png"), wxBITMAP_TYPE_PNG
);
41 while(!fullscreen
.IsOk());
43 wxBitmap screenshot
= fullscreen
.GetSubBitmap(wxRect(x
,y
,width
,height
));
45 #else //Under other paltforms, take a real screenshot
47 //Create a DC for the whole screen area
50 //Create a Bitmap that will later on hold the screenshot image
51 //Note that the Bitmap must have a size big enough to hold the screenshot
52 //-1 means using the current default colour depth
53 wxBitmap
screenshot(width
, height
, -1);
55 //Create a memory DC that will be used for actually taking the screenshot
57 //Tell the memory DC to use our Bitmap
58 //all drawing action on the memory DC will go to the Bitmap now
59 memDC
.SelectObject(screenshot
);
60 //Blit (in this case copy) the actual screen on the memory DC
62 memDC
.Blit( 0, //Copy to this X coordinate
63 0, //Copy to this Y coordinate
64 width
, //Copy this width
65 height
, //Copy this height
66 &dcScreen
, //From where do we copy?
67 x
, //What's the X offset in the original DC?
68 y
//What's the Y offset in the original DC?
70 //Select the Bitmap out of the memory DC by selecting a new
71 //uninitialized Bitmap
72 memDC
.SelectObject(wxNullBitmap
);
73 #endif //#ifdef __WXMAC__
75 // wxMessageBox(_(""),_(""));
81 wxBitmap
Capture(wxRect rect
)
83 wxPoint origin
= rect
.GetPosition();
84 return Capture(origin
.x
, origin
.y
, rect
.GetWidth(), rect
.GetHeight());
88 // ----------------------------------------------------------------------------
89 // AutoCaptureMechanism
90 // ----------------------------------------------------------------------------
92 wxBitmap
AutoCaptureMechanism::Capture(Control
& ctrl
)
94 if(ctrl
.name
== wxT("")) //no mannual specification for the control name
96 //Get name from wxRTTI
97 ctrl
.name
= ctrl
.ctrl
->GetClassInfo()->GetClassName();
102 if(ctrl
.flag
& AJ_Dropdown
)
104 wxString caption
= _("Do you wish to capture the dropdown list of ") + ctrl
.name
+ _("?");
105 wxString notice
= _("Click YES to capture it.\nAnd you MUST drop down the ") + ctrl
.name
+ _(" in 3 seconds after close me.\n");
106 notice
+= _("Click NO otherwise.");
108 choice
= wxMessageBox(notice
, caption
, wxYES_NO
, m_notebook
);
116 clock_t start
= clock();
117 while(clock() - start
< CLOCKS_PER_SEC
* 3)
124 wxRect rect
= GetRect(ctrl
.ctrl
, ctrl
.flag
);
126 //Do some rect adjust so it can include the dropdown list
127 //Currently it only works well under MSW, not adjusted for Linux and Mac OS
128 if(ctrl
.flag
& AJ_Dropdown
&& choice
== wxYES
)
131 int h
= rect
.GetHeight();
132 rect
.SetHeight(h
* 4);
136 //cut off "wx" and change them into lowercase.
137 // e.g. wxButton will have a name of "button" at the end
138 ctrl
.name
.StartsWith(_T("wx"), &(ctrl
.name
));
139 ctrl
.name
.MakeLower();
141 wxBitmap screenshot
= ::Capture(rect
);
143 if(ctrl
.flag
& AJ_RegionAdjust
)
151 //if AJ_RegionAdjust is specified, the following line will use the label trick to adjust
152 //the region position and size
153 wxRect
GetRect(wxWindow
* ctrl
, int flag
);
154 //put the control back after the label trick(Using reparent/resizer approach)
155 void PutBack(wxWindow
* ctrl
);
157 wxBitmap
AutoCaptureMechanism::Union(wxBitmap pic1
, wxBitmap pic2
)
159 int w1
, w2
, h1
, h2
, w
, h
;
160 w1
= pic1
.GetWidth();
161 w2
= pic2
.GetWidth();
162 h1
= pic1
.GetHeight();
163 h2
= pic2
.GetHeight();
165 const int gap_between
= 20;
167 w
= (w1
>= w2
) ? w1
: w2
;
168 h
= h1
+ h2
+ gap_between
;
170 wxBitmap
result(w
, h
, -1);
173 dstDC
.SelectObject(result
);
175 dstDC
.DrawBitmap(pic1
, 0, 0, false);
176 dstDC
.DrawBitmap(pic2
, 0, h1
+ gap_between
, false);
178 dstDC
.SelectObject(wxNullBitmap
);
181 wxBitmap
mask(w
, h
, 1);
182 maskDC
.SelectObject(mask
);
184 maskDC
.SetPen(*wxTRANSPARENT_PEN
);
185 maskDC
.SetBrush(*wxBLACK_BRUSH
);
186 maskDC
.DrawRectangle(0, 0, w
+ 1, h
+ 1);
188 maskDC
.SetBrush(*wxWHITE_BRUSH
);
189 maskDC
.DrawRectangle(0, 0, w1
, h1
);
190 maskDC
.DrawRectangle(0, h1
+ gap_between
, w2
, h2
);
191 maskDC
.SelectObject(wxNullBitmap
);
193 result
.SetMask(new wxMask(mask
));
198 void AutoCaptureMechanism::Save(wxBitmap screenshot
, wxString fileName
)
200 //Check if m_defaultDir already existed
201 if(!wxDirExists(m_dir
))
204 wxString fullFileName
= m_dir
+ wxFileName::GetPathSeparator() + fileName
;
206 //to prvent overwritten
207 while(wxFileName::FileExists(fullFileName
+ _T(".png"))) fullFileName
+= _T("_");
209 //Our Bitmap now has the screenshot, so let's save it as an png
210 //The filename itself is without extension.
211 screenshot
.SaveFile(fullFileName
+ _T(".png"), wxBITMAP_TYPE_PNG
);
214 wxRect
AutoCaptureMechanism::GetRect(wxWindow
* ctrl
, int flag
)
216 if(flag
& AJ_RegionAdjust
)
218 wxWindow
* parent
= ctrl
->GetParent();
219 wxSizer
* sizer
= parent
->GetSizer();
226 +---------+-----------+---------+
228 +---------+-----------+---------+
229 | label | ctrl | label |
230 +---------+-----------+---------+
232 +---------+-----------+---------+
235 m_grid
= new wxFlexGridSizer(3, 3, m_border
, m_border
);
239 for(int i
= 0; i
< 4; ++i
)
240 l
[i
] = new wxStaticText(parent
, wxID_ANY
, wxT(" "));
243 m_grid
->Add(new wxStaticText(parent
, wxID_ANY
, wxT(" ")));
245 m_grid
->Add(new wxStaticText(parent
, wxID_ANY
, wxT(" ")));
247 m_grid
->Add(new wxStaticText(parent
, wxID_ANY
, wxT(" ")));
249 m_grid
->Add(new wxStaticText(parent
, wxID_ANY
, wxT(" ")));
253 parent
->SetSizer(sizer
);
259 return wxRect(l
[0]->GetScreenRect().GetBottomRight(),
260 l
[3]->GetScreenRect().GetTopLeft());
263 else //Actually it won't get here working with the current guiframe.h/guiframe.cpp
265 return ctrl
->GetScreenRect().Inflate(m_border
);
270 return ctrl
->GetScreenRect().Inflate(m_border
);
274 void AutoCaptureMechanism::PutBack(wxWindow
* ctrl
)
276 m_grid
->Detach(ctrl
);
278 wxSizerItemList children
= m_grid
->GetChildren();
280 for(wxSizerItemList::iterator it
= children
.begin(); it
!= children
.end(); ++it
)
282 wxSizerItem
* item
= *it
;
283 if(item
->IsWindow()) delete (*it
)->GetWindow();
286 wxSizer
* sizer
= ctrl
->GetParent()->GetSizer();
287 sizer
->Detach(m_grid
);