utils/screenshotgen/src/autocapture.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        autocapture.cpp
   3 // Purpose:     Implement wxCtrlMaskOut class
   4 // Author:      Utensil Candel (UtensilCandel@@gmail.com)
   5 // RCS-ID:      $Id$
   6 // Licence:     wxWindows license
   7 /////////////////////////////////////////////////////////////////////////////
   8
   9 // For compilers that support precompilation, includes "wx/wx.h".
  10 #include "wx/wxprec.h"
  11
  12 #ifdef __BORLANDC__
  13     #pragma hdrstop
  14 #endif
  15
  16 // for all others, include the necessary headers
  17 #ifndef WX_PRECOMP
  18     #include "wx/wx.h"
  19 #endif
  20
  21 #include <wx/filename.h>
  22 #include "autocapture.h"
  23
  24 #ifdef __WXMAC__
  25 #include <cstring>
  26 #endif
  27
  28
  29 // ----------------------------------------------------------------------------
  30 // AutoCaptureMechanism
  31 // ----------------------------------------------------------------------------
  32
  33 /* static */
  34 void AutoCaptureMechanism::Delay(int seconds)
  35 {
  36     using std::clock;
  37     using std::clock_t;
  38
  39     // Wait for 3 seconds
  40     clock_t start = clock();
  41     while (clock() - start < CLOCKS_PER_SEC * seconds)
  42         wxYieldIfNeeded();
  43 }
  44
  45 /* static */
  46 wxBitmap AutoCaptureMechanism::Capture(int x, int y, int width, int height, int delay)
  47 {
  48     // Somehow wxScreenDC.Blit() doesn't work under Mac for now. Here is a trick.
  49 #ifdef __WXMAC__
  50
  51     // wxExecute(_T("screencapture -x ") + tempfile, wxEXEC_SYNC);
  52
  53     char captureCommand[80] =""; // a reasonable max size is 80
  54
  55     sprintf(captureCommand, "sleep %d;%s", delay, "screencapture -x /tmp/wx_screen_capture.png");
  56
  57     system(captureCommand);
  58
  59     wxBitmap fullscreen;
  60
  61     if(delay) Delay(delay);
  62
  63     do
  64     {
  65         fullscreen = wxBitmap(_T("/tmp/wx_screen_capture.png"), wxBITMAP_TYPE_PNG);
  66     }
  67     while(!fullscreen.IsOk());
  68
  69     wxBitmap screenshot = fullscreen.GetSubBitmap(wxRect(x,y,width,height));
  70
  71     // to prevent loading the old screenshot next time
  72     system("rm /tmp/wx_screen_capture.png");
  73
  74 #else // Under other paltforms, take a real screenshot
  75
  76     // Create a DC for the whole screen area
  77     wxScreenDC dcScreen;
  78
  79     // Create a Bitmap that will later on hold the screenshot image
  80     // Note that the Bitmap must have a size big enough to hold the screenshot
  81     // -1 means using the current default colour depth
  82     wxBitmap screenshot(width, height, -1);
  83
  84     // Create a memory DC that will be used for actually taking the screenshot
  85     wxMemoryDC memDC;
  86
  87     // Tell the memory DC to use our Bitmap
  88     // all drawing action on the memory DC will go to the Bitmap now
  89     memDC.SelectObject(screenshot);
  90
  91     // Blit (in this case copy) the actual screen on the memory DC
  92     // and thus the Bitmap
  93     memDC.Blit( 0, // Copy to this X coordinate
  94                 0, // Copy to this Y coordinate
  95                 width, // Copy this width
  96                 height, // Copy this height
  97                 &dcScreen, // From where do we copy?
  98                 x, // What's the X offset in the original DC?
  99                 y  // What's the Y offset in the original DC?
 100               );
 101
 102     // Select the Bitmap out of the memory DC by selecting a new
 103     // uninitialized Bitmap
 104     memDC.SelectObject(wxNullBitmap);
 105 #endif // #ifdef __WXMAC__
 106
 107     return screenshot;
 108 }
 109
 110 /* static */
 111 wxBitmap AutoCaptureMechanism::Capture(wxRect rect, int delay)
 112 {
 113     wxPoint origin = rect.GetPosition();
 114     return Capture(origin.x, origin.y, rect.GetWidth(), rect.GetHeight(), delay);
 115 }
 116
 117 void AutoCaptureMechanism::CaptureAll()
 118 {
 119     // start from the first page
 120     m_notebook->SetSelection(0);
 121     wxYield();
 122
 123     for (ControlList::iterator it = m_controlList.begin();
 124          it != m_controlList.end();
 125          ++it)
 126     {
 127         Control &ctrl = *it;
 128
 129         if (ctrl.flag == AJ_TurnPage)    // Turn to next page
 130         {
 131             m_notebook->SetSelection(m_notebook->GetSelection() + 1);
 132             wxYield();
 133             continue;
 134         }
 135
 136         // create the screenshot
 137         wxBitmap screenshot = Capture(ctrl);
 138         if (ctrl.flag & AJ_Union)
 139             screenshot = Union(screenshot, Capture(*(++it)));
 140
 141         // and save it
 142         Save(screenshot, ctrl.name);
 143     }
 144 }
 145
 146 wxBitmap AutoCaptureMechanism::Capture(Control& ctrl)
 147 {
 148     if (ctrl.name == wxT(""))  // no manual specification for the control name
 149     {
 150         // Get its name from wxRTTI
 151         ctrl.name = ctrl.ctrl->GetClassInfo()->GetClassName();
 152     }
 153
 154     int choice = wxNO;
 155
 156     // for drop-down controls we need the help of the user
 157     if (ctrl.flag & AJ_Dropdown)
 158     {
 159         wxString caption = _("Drop-down screenshot...");
 160         wxString msg =
 161             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."),
 162                              ctrl.name, ctrl.name);
 163
 164         choice = wxMessageBox(msg, caption, wxYES_NO, m_notebook);
 165
 166         #ifndef __WXMAC__  //not __WXMAC__
 167         if (choice == wxYES)  Delay(3);
 168         #endif
 169     }
 170
 171     wxRect rect = GetRect(ctrl.ctrl, ctrl.flag);
 172
 173     // Do some rect adjust so it can include the dropdown list;
 174     // currently this only works well under MSW; not adjusted for Linux and Mac OS
 175     if (ctrl.flag & AJ_Dropdown && choice == wxYES)
 176     {
 177 //          #ifdef __WXMSW__
 178         int h = rect.GetHeight();
 179         rect.SetHeight(h * 4);
 180 //          #endif
 181     }
 182
 183     // cut off "wx" and change the name into lowercase.
 184     // e.g. wxButton will have a name of "button" at the end
 185     ctrl.name.StartsWith(_T("wx"), &(ctrl.name));
 186     ctrl.name.MakeLower();
 187
 188         // AD-HOC FIX for wxHyperlink
 189         if (ctrl.name == "generichyperlinkctrl")
 190                 ctrl.name = "hyperlinkctrl";
 191
 192     // take the screenshot
 193     wxBitmap screenshot = Capture(rect);
 194
 195     if (ctrl.flag & AJ_RegionAdjust)
 196         PutBack(ctrl.ctrl);
 197
 198     return screenshot;
 199 }
 200
 201 wxBitmap AutoCaptureMechanism::Union(wxBitmap pic1, wxBitmap pic2)
 202 {
 203     int w1, w2, h1, h2, w, h;
 204     w1 = pic1.GetWidth();
 205     w2 = pic2.GetWidth();
 206     h1 = pic1.GetHeight();
 207     h2 = pic2.GetHeight();
 208
 209     const int gap_between = 20;
 210
 211     w = (w1 >= w2) ? w1 : w2;
 212     h = h1 + h2 + gap_between;
 213
 214     wxBitmap result(w, h, -1);
 215
 216     wxMemoryDC dstDC;
 217     dstDC.SelectObject(result);
 218
 219     dstDC.DrawBitmap(pic1, 0, 0, false);
 220     dstDC.DrawBitmap(pic2, 0, h1 + gap_between, false);
 221
 222     dstDC.SelectObject(wxNullBitmap);
 223
 224     wxMemoryDC maskDC;
 225     wxBitmap mask(w, h, 1);
 226     maskDC.SelectObject(mask);
 227
 228     maskDC.SetPen(*wxTRANSPARENT_PEN);
 229     maskDC.SetBrush(*wxBLACK_BRUSH);
 230     maskDC.DrawRectangle(0, 0, w + 1, h + 1);
 231
 232     maskDC.SetBrush(*wxWHITE_BRUSH);
 233     maskDC.DrawRectangle(0, 0, w1, h1);
 234     maskDC.DrawRectangle(0, h1 + gap_between, w2, h2);
 235     maskDC.SelectObject(wxNullBitmap);
 236
 237     result.SetMask(new wxMask(mask));
 238
 239     return result;
 240 }
 241
 242 void AutoCaptureMechanism::Save(wxBitmap screenshot, wxString fileName)
 243 {
 244     // make sure m_dir exists
 245     if (!wxDirExists(m_dir))
 246         wxMkdir(m_dir);
 247
 248     wxFileName fullFileName(m_dir, fileName + ".png");
 249
 250     // do not overwrite already existing files with this name
 251     while (fullFileName.FileExists())
 252         fullFileName.SetName(fullFileName.GetName() + "_");
 253
 254     // save the screenshot as a PNG
 255     screenshot.SaveFile(fullFileName.GetFullPath(), wxBITMAP_TYPE_PNG);
 256 }
 257
 258 wxRect AutoCaptureMechanism::GetRect(wxWindow* ctrl, int flag)
 259 {
 260     if (flag & AJ_RegionAdjust)
 261     {
 262         wxWindow * parent = ctrl->GetParent();
 263         wxSizer * sizer = parent->GetSizer();
 264
 265         if (sizer)
 266         {
 267             sizer->Detach(ctrl);
 268
 269             /*
 270             +---------+-----------+---------+
 271             |    0    |   label   |    1    |
 272             +---------+-----------+---------+
 273             |  label  |    ctrl   |  label  |
 274             +---------+-----------+---------+
 275             |    2    |   label   |    3    |
 276             +---------+-----------+---------+
 277            */
 278
 279             m_grid = new wxFlexGridSizer(3, 3, m_border, m_border);
 280
 281             wxStaticText* l[4];
 282
 283             for (int i = 0; i < 4; ++i)
 284                 l[i] = new wxStaticText(parent, wxID_ANY, wxT(" "));
 285
 286             m_grid->Add(l[0]);
 287             m_grid->Add(new wxStaticText(parent, wxID_ANY, wxT(" ")));
 288             m_grid->Add(l[1]);
 289             m_grid->Add(new wxStaticText(parent, wxID_ANY, wxT(" ")));
 290             m_grid->Add(ctrl);
 291             m_grid->Add(new wxStaticText(parent, wxID_ANY, wxT(" ")));
 292             m_grid->Add(l[2]);
 293             m_grid->Add(new wxStaticText(parent, wxID_ANY, wxT(" ")));
 294             m_grid->Add(l[3]);
 295
 296             sizer->Add(m_grid);
 297             parent->SetSizer(sizer);
 298             parent->Layout();
 299
 300             parent->Refresh();
 301             wxYield();
 302
 303             return wxRect(l[0]->GetScreenRect().GetBottomRight(),
 304                     l[3]->GetScreenRect().GetTopLeft());
 305
 306         }
 307         else  // Actually it won't get here working with the current guiframe.h/guiframe.cpp
 308         {
 309             return ctrl->GetScreenRect().Inflate(m_border);
 310         }
 311     }
 312     else
 313     {
 314        return ctrl->GetScreenRect().Inflate(m_border);
 315     }
 316 }
 317
 318 void AutoCaptureMechanism::PutBack(wxWindow * ctrl)
 319 {
 320     m_grid->Detach(ctrl);
 321
 322     wxSizerItemList children = m_grid->GetChildren();
 323
 324     for (wxSizerItemList::iterator it = children.begin(); it != children.end(); ++it)
 325     {
 326         wxSizerItem* item = *it;
 327         if (item->IsWindow()) delete (*it)->GetWindow();
 328     }
 329
 330     wxSizer * sizer = ctrl->GetParent()->GetSizer();
 331     sizer->Detach(m_grid);
 332     delete m_grid;
 333     sizer->Add(ctrl);
 334 }
 335