replace wx_{const,static,reinterpret}_cast with their standard C++ equivalents

[wxWidgets.git] / utils / screenshotgen / src / autocapture.cpp

diff --git a/utils/screenshotgen/src/autocapture.cpp b/utils/screenshotgen/src/autocapture.cpp
index 0d729281cf23823caa74cca0d797022e6e427b37..62ae1b88dce94d1d447635a712a6e6ef15a31eda 100644 (file)
--- a/utils/screenshotgen/src/autocapture.cpp
+++ b/utils/screenshotgen/src/autocapture.cpp
@@ -10,30 +10,56 @@
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__

-    #pragma hdrstop
+#pragma hdrstop

 #endif
 
 #endif
 

-// for all others, include the necessary headers
+// for all others, include the necessary headers wxWidgets headers)

 #ifndef WX_PRECOMP
 #ifndef WX_PRECOMP

-    #include "wx/wx.h"
+#include "wx/wx.h"

 #endif
 
 #endif
 

-#include <wx/filename.h>
+#include "wx/filename.h"

 
 #include "autocapture.h"
 
 
 #include "autocapture.h"
 

+#ifdef __WXMAC__
+#include <cstring>
+#endif
+
+
+// ----------------------------------------------------------------------------
+// AutoCaptureMechanism
+// ----------------------------------------------------------------------------
+
+/* static */
+void AutoCaptureMechanism::Delay(int seconds)
+{
+       // TODO: Switch this to use wxTimer.
+
+    // Wait for 3 seconds
+    clock_t start = clock();
+    while (clock() - start < CLOCKS_PER_SEC * seconds)
+        wxYieldIfNeeded();
+}

 
 

-wxBitmap Capture(int x, int y, int width, int height)
+/* static */
+wxBitmap AutoCaptureMechanism::Capture(int x, int y, int width, int height, int delay)

 {
 {

-    //Somehow wxScreenDC.Blit() doesn't work under Mac for now. Here is a trick.
-    #ifdef __WXMAC__
+    // Somehow wxScreenDC.Blit() doesn't work under Mac for now. Here is a trick.
+#ifdef __WXMAC__
+
+    // wxExecute(_T("screencapture -x ") + tempfile, wxEXEC_SYNC);

 
 

-    //wxExecute(_T("screencapture -x ") + tempfile, wxEXEC_SYNC);
+    char captureCommand[80] =""; // a reasonable max size is 80

 
 

-    system("screencapture -x /tmp/wx_screen_capture.png");
+    sprintf(captureCommand, "sleep %d;%s", delay, "screencapture -x /tmp/wx_screen_capture.png");
+
+    system(captureCommand);

 
     wxBitmap fullscreen;
 
 
     wxBitmap fullscreen;
 

+    if(delay) Delay(delay);
+

     do
     {
         fullscreen = wxBitmap(_T("/tmp/wx_screen_capture.png"), wxBITMAP_TYPE_PNG);
     do
     {
         fullscreen = wxBitmap(_T("/tmp/wx_screen_capture.png"), wxBITMAP_TYPE_PNG);


@@ -42,72 +68,197 @@ wxBitmap Capture(int x, int y, int width, int height)
 
     wxBitmap screenshot = fullscreen.GetSubBitmap(wxRect(x,y,width,height));
 
 
     wxBitmap screenshot = fullscreen.GetSubBitmap(wxRect(x,y,width,height));
 

-    #else //Under other paltforms, take a real screenshot
+    // to prevent loading the old screenshot next time
+    system("rm /tmp/wx_screen_capture.png");

 
 

-    //Create a DC for the whole screen area
+#else // Under other paltforms, take a real screenshot
+
+    // Create a DC for the whole screen area

     wxScreenDC dcScreen;
 
     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);
 
     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;
     wxMemoryDC memDC;

-    //Tell the memory DC to use our Bitmap
-    //all drawing action on the memory DC will go to the Bitmap now
+
+    // Tell the memory DC to use our Bitmap
+    // all drawing action on the memory DC will go to the Bitmap now

     memDC.SelectObject(screenshot);
     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(_(""),_(""));
+    // 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;
 
     return screenshot;

-

 }
 
 }
 

-wxBitmap Capture(wxRect rect)
+/* static */
+wxBitmap AutoCaptureMechanism::Capture(wxRect rect, int delay)

 {
     wxPoint origin = rect.GetPosition();
 {
     wxPoint origin = rect.GetPosition();

-    return Capture(origin.x, origin.y, rect.GetWidth(), rect.GetHeight());
+    return Capture(origin.x, origin.y, rect.GetWidth(), rect.GetHeight(), delay);
+}
+
+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)
+    {
+        Control &ctrl = *it;
+
+        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)
+            screenshot = Union(screenshot, Capture(*(++it)));
+
+        // and save it
+        Save(screenshot, ctrl.name);
+    }
+}
+
+wxBitmap AutoCaptureMechanism::Capture(Control& ctrl)
+{
+    if (ctrl.name == wxT(""))  // no manual specification for the control name
+    {
+        // Get its name from wxRTTI
+        ctrl.name = ctrl.ctrl->GetClassInfo()->GetClassName();
+    }
+
+    int choice = wxNO;
+
+    // for drop-down controls we need the help of the user
+    if (ctrl.flag & AJ_Dropdown)
+    {
+        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(msg, caption, wxYES_NO, m_notebook);
+
+        #ifndef __WXMAC__  //not __WXMAC__
+        if (choice == wxYES)  Delay(3);
+        #endif
+    }
+
+    wxRect rect = GetRect(ctrl.ctrl, ctrl.flag);
+
+    // 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();
+        rect.SetHeight(h * 4);
+//          #endif
+    }
+
+    // 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();
+
+    // take the screenshot
+    wxBitmap screenshot = Capture(rect);
+
+    if (ctrl.flag & AJ_RegionAdjust)
+        PutBack(ctrl.ctrl);
+
+    return screenshot;
+}
+
+wxBitmap AutoCaptureMechanism::Union(wxBitmap pic1, wxBitmap pic2)
+{
+    int w1, w2, h1, h2, w, h;
+    w1 = pic1.GetWidth();
+    w2 = pic2.GetWidth();
+    h1 = pic1.GetHeight();
+    h2 = pic2.GetHeight();
+
+    const int gap_between = 20;
+
+    w = (w1 >= w2) ? w1 : w2;
+    h = h1 + h2 + gap_between;
+
+    wxBitmap result(w, h, -1);
+
+    wxMemoryDC dstDC;
+    dstDC.SelectObject(result);
+
+    dstDC.DrawBitmap(pic1, 0, 0, false);
+    dstDC.DrawBitmap(pic2, 0, h1 + gap_between, false);
+
+    dstDC.SelectObject(wxNullBitmap);
+
+    wxMemoryDC maskDC;
+    wxBitmap mask(w, h, 1);
+    maskDC.SelectObject(mask);
+
+    maskDC.SetPen(*wxTRANSPARENT_PEN);
+    maskDC.SetBrush(*wxBLACK_BRUSH);
+    maskDC.DrawRectangle(0, 0, w + 1, h + 1);
+
+    maskDC.SetBrush(*wxWHITE_BRUSH);
+    maskDC.DrawRectangle(0, 0, w1, h1);
+    maskDC.DrawRectangle(0, h1 + gap_between, w2, h2);
+    maskDC.SelectObject(wxNullBitmap);
+
+    result.SetMask(new wxMask(mask));
+
+    return result;

 }
 
 void AutoCaptureMechanism::Save(wxBitmap screenshot, wxString fileName)
 {
 }
 
 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);
 
         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)
 {
 }
 
 wxRect AutoCaptureMechanism::GetRect(wxWindow* ctrl, int flag)
 {

-    if(flag & AJ_RegionAdjust)
+    if (flag & AJ_RegionAdjust)

     {
         wxWindow * parent = ctrl->GetParent();
         wxSizer * sizer = parent->GetSizer();
 
     {
         wxWindow * parent = ctrl->GetParent();
         wxSizer * sizer = parent->GetSizer();
 

-        if(sizer)
+        if (sizer)

         {
             sizer->Detach(ctrl);
 
         {
             sizer->Detach(ctrl);
 


@@ -125,7 +276,7 @@ wxRect AutoCaptureMechanism::GetRect(wxWindow* ctrl, int flag)
 
             wxStaticText* l[4];
 
 
             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]);
                 l[i] = new wxStaticText(parent, wxID_ANY, wxT(" "));
 
             m_grid->Add(l[0]);


@@ -149,7 +300,7 @@ wxRect AutoCaptureMechanism::GetRect(wxWindow* ctrl, int flag)
                     l[3]->GetScreenRect().GetTopLeft());
 
         }
                     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);
         }
         {
             return ctrl->GetScreenRect().Inflate(m_border);
         }


@@ -166,10 +317,10 @@ void AutoCaptureMechanism::PutBack(wxWindow * ctrl)
 
     wxSizerItemList children = m_grid->GetChildren();
 
 
     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;
     {
         wxSizerItem* item = *it;

-        if(item->IsWindow()) delete (*it)->GetWindow();
+        if (item->IsWindow()) delete (*it)->GetWindow();

     }
 
     wxSizer * sizer = ctrl->GetParent()->GetSizer();
     }
 
     wxSizer * sizer = ctrl->GetParent()->GetSizer();