utils/screenshotgen/src/screenshot_main.cpp

/////////////////////////////////////////////////////////////////////////////
// Name:        screenshot_main.cpp
// Purpose:     Implements the window containing all controls.
// Author:      Utensil Candel (UtensilCandel@@gmail.com)
// RCS-ID:      $Id$
// Licence:     wxWindows licence
/////////////////////////////////////////////////////////////////////////////

// For compilers that support precompilation, includes "wx/wx.h".
#include "wx/wxprec.h"

#ifdef __BORLANDC__
#pragma hdrstop
#endif

// for all others, include the necessary headers wxWidgets headers)
#ifndef WX_PRECOMP
#include "wx/wx.h"
#endif

#include "wx/aboutdlg.h"
#include "wx/dir.h"

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

// ----------------------------------------------------------------------------
// ScreenshotFrame
// ----------------------------------------------------------------------------

ScreenshotFrame::ScreenshotFrame(wxFrame *frame) : GUIFrame(frame)
{
#if wxUSE_STATUSBAR
    statusBar->SetStatusText(_("Welcome to the Automatic Screenshot Generator!"), 0);
#endif

    // set minimum size hints
    GetSizer()->SetSizeHints(this);
}

// ----------------------------------------------------------------------------
// ScreenshotFrame - event handlers
// ----------------------------------------------------------------------------

void ScreenshotFrame::OnClose(wxCloseEvent& WXUNUSED(event))
{
    Destroy();
}

void ScreenshotFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
{
    Destroy();
}

void ScreenshotFrame::OnSeeScreenshots(wxCommandEvent& WXUNUSED(event))
{
    wxString defaultDir = AutoCaptureMechanism::GetDefaultDirectoryAbsPath();

    if (wxFileName::DirExists(defaultDir))
        wxLaunchDefaultBrowser(defaultDir);
    else
        wxMessageBox(_("There isn't any screenshots yet."));
}

void ScreenshotFrame::OnAbout(wxCommandEvent& WXUNUSED(event))
{
    wxAboutDialogInfo info;
    info.SetName(_("Automatic Screenshot Generator"));
    info.SetVersion(_("1.0"));
    info.SetDescription(_("This utility automatically creates screenshots of wxWidgets controls for use in wxWidgets documentation."));
    info.SetCopyright(wxT("(C) 2008 Utensil Candel"));

    wxAboutBox(info);
}

void ScreenshotFrame::OnCaptureFullScreen(wxCommandEvent& WXUNUSED(event))
{
    // Create a DC for the whole screen area
    wxScreenDC dcScreen;

    // Get the size of the screenDC
    wxCoord screenWidth, screenHeight;
    dcScreen.GetSize(&screenWidth, &screenHeight);

    wxBitmap fullscreen(1, 1);
    AutoCaptureMechanism::Capture(&fullscreen, 0, 0, screenWidth, screenHeight);

    AutoCaptureMechanism::Save(&fullscreen, wxT("fullscreen"));

    wxMessageBox(_("A screenshot of the entire screen was saved as:\n\n  ")
                + AutoCaptureMechanism::GetDefaultDirectoryAbsPath() + wxT("fullscreen.png"),
                 _("Full screen capture"), wxICON_INFORMATION|wxOK, this);
}

void ScreenshotFrame::OnCaptureAllControls(wxCommandEvent& WXUNUSED(event))
{
    wxString dir = AutoCaptureMechanism::GetDefaultDirectoryAbsPath();

    // check if there are other screenshots taken before
    if (wxFileName::DirExists(dir))
    {
        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)
        {
            case wxYES:
            {
                wxArrayString files;
                wxDir::GetAllFiles(dir, &files, wxT("*.png"), wxDIR_FILES);

                // remove all PNG files from the screenshots folder
                int n = files.GetCount();
                for (int i = 0; i < n; ++i)
                    wxRemoveFile(files[i]);
            }
            break;

            case wxNO: break;
            case wxCANCEL: return;
        }
    }

    // proceed with the automatic screenshot capture

    this->Maximize();

    AutoCaptureMechanism auto_cap(m_notebook1);

    auto_cap.RegisterControl(m_button1);
    auto_cap.RegisterControl(m_staticText1);
    auto_cap.RegisterControl(m_checkBox1, AJ_Union);
    auto_cap.RegisterControl(m_checkBox2, AJ_UnionEnd);
    auto_cap.RegisterControl(m_radioBtn1, AJ_Union);
    auto_cap.RegisterControl(m_radioBtn2, AJ_UnionEnd);
    auto_cap.RegisterControl(m_bpButton1);
    auto_cap.RegisterControl(m_bitmap1);
    auto_cap.RegisterControl(m_gauge1, wxT("wxGauge"));
    auto_cap.RegisterControl(m_slider1);
    auto_cap.RegisterControl(m_toggleBtn1, AJ_Union);
    auto_cap.RegisterControl(m_toggleBtn2, AJ_UnionEnd);
    auto_cap.RegisterControl(m_hyperlink1, wxT("wxHyperlinkCtrl"));
    auto_cap.RegisterControl(m_spinCtrl1, AJ_RegionAdjust);
    auto_cap.RegisterControl(m_spinBtn1);
    auto_cap.RegisterControl(m_scrollBar1);

    auto_cap.RegisterPageTurn();

    auto_cap.RegisterControl(m_checkList1);
    auto_cap.RegisterControl(m_listBox1);
    auto_cap.RegisterControl(m_radioBox1);
    auto_cap.RegisterControl(m_staticBox1);
    auto_cap.RegisterControl(m_treeCtrl1);
    auto_cap.RegisterControl(m_listCtrl1, wxT("wxListCtrl"));

    auto_cap.RegisterControl(m_animationCtrl1);
    auto_cap.RegisterControl(m_collPane1, wxT("wxCollapsiblePane"), AJ_Union);
    auto_cap.RegisterControl(m_collPane2, AJ_UnionEnd);

    auto_cap.RegisterPageTurn();

    auto_cap.RegisterControl(m_textCtrl1, AJ_Union);
    auto_cap.RegisterControl(m_textCtrl2, AJ_UnionEnd);
    auto_cap.RegisterControl(m_richText1);

    auto_cap.RegisterPageTurn();

    auto_cap.RegisterControl(m_colourPicker1, wxT("wxColourPickerCtrl"));
    auto_cap.RegisterControl(m_fontPicker1, wxT("wxFontPickerCtrl"));
    auto_cap.RegisterControl(m_filePicker1, wxT("wxFilePickerCtrl"), AJ_RegionAdjust);
    auto_cap.RegisterControl(m_calendar1, wxT("wxCalendarCtrl"), AJ_RegionAdjust);
    auto_cap.RegisterControl(m_datePicker1, wxT("wxDatePickerCtrl"));
    auto_cap.RegisterControl(m_genericDirCtrl1, wxT("wxGenericDirCtrl"));
    auto_cap.RegisterControl(m_dirPicker1, wxT("wxDirPickerCtrl"), AJ_RegionAdjust);

    auto_cap.RegisterPageTurn();

    auto_cap.RegisterControl(m_choice1, AJ_Dropdown);
    auto_cap.RegisterControl(m_comboBox1, AJ_Dropdown);
    auto_cap.RegisterControl(m_bmpComboBox1, AJ_Dropdown);
    auto_cap.RegisterControl(m_ownerDrawnComboBox1, AJ_Dropdown);
    auto_cap.RegisterControl(m_comboCtrl1, AJ_Dropdown|AJ_Union);
    auto_cap.RegisterControl(m_comboCtrl2, AJ_Dropdown|AJ_UnionEnd);

    auto_cap.CaptureAll();

    wxMessageBox(_("All screenshots were generated successfully in the folder:\n  ") + dir,
                 _("Success"), wxOK|wxICON_INFORMATION, this);
}
Commit	Line	Data
	1	/////////////////////////////////////////////////////////////////////////////
	2	// Name: screenshot_main.cpp
	3	// Purpose: Implements the window containing all controls.
	4	// Author: Utensil Candel (UtensilCandel@@gmail.com)
	5	// RCS-ID: $Id$
	6	// Licence: wxWindows licence
	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 wxWidgets headers)
	17	#ifndef WX_PRECOMP
	18	#include "wx/wx.h"
	19	#endif
	20
	21	#include "wx/aboutdlg.h"
	22	#include "wx/dir.h"
	23
	24	#include "screenshot_main.h"
	25	#include "autocapture.h"
	26
	27	// ----------------------------------------------------------------------------
	28	// ScreenshotFrame
	29	// ----------------------------------------------------------------------------
	30
	31	ScreenshotFrame::ScreenshotFrame(wxFrame *frame) : GUIFrame(frame)
	32	{
	33	#if wxUSE_STATUSBAR
	34	statusBar->SetStatusText(_("Welcome to the Automatic Screenshot Generator!"), 0);
	35	#endif
	36
	37	// set minimum size hints
	38	GetSizer()->SetSizeHints(this);
	39	}
	40
	41	// ----------------------------------------------------------------------------
	42	// ScreenshotFrame - event handlers
	43	// ----------------------------------------------------------------------------
	44
	45	void ScreenshotFrame::OnClose(wxCloseEvent& WXUNUSED(event))
	46	{
	47	Destroy();
	48	}
	49
	50	void ScreenshotFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
	51	{
	52	Destroy();
	53	}
	54
	55	void ScreenshotFrame::OnSeeScreenshots(wxCommandEvent& WXUNUSED(event))
	56	{
	57	wxString defaultDir = AutoCaptureMechanism::GetDefaultDirectoryAbsPath();
	58
	59	if (wxFileName::DirExists(defaultDir))
	60	wxLaunchDefaultBrowser(defaultDir);
	61	else
	62	wxMessageBox(_("There isn't any screenshots yet."));
	63	}
	64
	65	void ScreenshotFrame::OnAbout(wxCommandEvent& WXUNUSED(event))
	66	{
	67	wxAboutDialogInfo info;
	68	info.SetName(_("Automatic Screenshot Generator"));
	69	info.SetVersion(_("1.0"));
	70	info.SetDescription(_("This utility automatically creates screenshots of wxWidgets controls for use in wxWidgets documentation."));
	71	info.SetCopyright(wxT("(C) 2008 Utensil Candel"));
	72
	73	wxAboutBox(info);
	74	}
	75
	76	void ScreenshotFrame::OnCaptureFullScreen(wxCommandEvent& WXUNUSED(event))
	77	{
	78	// Create a DC for the whole screen area
	79	wxScreenDC dcScreen;
	80
	81	// Get the size of the screenDC
	82	wxCoord screenWidth, screenHeight;
	83	dcScreen.GetSize(&screenWidth, &screenHeight);
	84
	85	wxBitmap fullscreen(1, 1);
	86	AutoCaptureMechanism::Capture(&fullscreen, 0, 0, screenWidth, screenHeight);
	87
	88	AutoCaptureMechanism::Save(&fullscreen, wxT("fullscreen"));
	89
	90	wxMessageBox(_("A screenshot of the entire screen was saved as:\n\n ")
	91	+ AutoCaptureMechanism::GetDefaultDirectoryAbsPath() + wxT("fullscreen.png"),
	92	_("Full screen capture"), wxICON_INFORMATION\|wxOK, this);
	93	}
	94
	95	void ScreenshotFrame::OnCaptureAllControls(wxCommandEvent& WXUNUSED(event))
	96	{
	97	wxString dir = AutoCaptureMechanism::GetDefaultDirectoryAbsPath();
	98
	99	// check if there are other screenshots taken before
	100	if (wxFileName::DirExists(dir))
	101	{
	102	int choice = wxMessageBox(
	103	_("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."),
	104	_("Delete existing screenshots?"),
	105	wxYES_NO \| wxCANCEL \| wxICON_QUESTION, this);
	106
	107	switch(choice)
	108	{
	109	case wxYES:
	110	{
	111	wxArrayString files;
	112	wxDir::GetAllFiles(dir, &files, wxT("*.png"), wxDIR_FILES);
	113
	114	// remove all PNG files from the screenshots folder
	115	int n = files.GetCount();
	116	for (int i = 0; i < n; ++i)
	117	wxRemoveFile(files[i]);
	118	}
	119	break;
	120
	121	case wxNO: break;
	122	case wxCANCEL: return;
	123	}
	124	}
	125
	126	// proceed with the automatic screenshot capture
	127
	128	this->Maximize();
	129
	130	AutoCaptureMechanism auto_cap(m_notebook1);
	131
	132	auto_cap.RegisterControl(m_button1);
	133	auto_cap.RegisterControl(m_staticText1);
	134	auto_cap.RegisterControl(m_checkBox1, AJ_Union);
	135	auto_cap.RegisterControl(m_checkBox2, AJ_UnionEnd);
	136	auto_cap.RegisterControl(m_radioBtn1, AJ_Union);
	137	auto_cap.RegisterControl(m_radioBtn2, AJ_UnionEnd);
	138	auto_cap.RegisterControl(m_bpButton1);
	139	auto_cap.RegisterControl(m_bitmap1);
	140	auto_cap.RegisterControl(m_gauge1, wxT("wxGauge"));
	141	auto_cap.RegisterControl(m_slider1);
	142	auto_cap.RegisterControl(m_toggleBtn1, AJ_Union);
	143	auto_cap.RegisterControl(m_toggleBtn2, AJ_UnionEnd);
	144	auto_cap.RegisterControl(m_hyperlink1, wxT("wxHyperlinkCtrl"));
	145	auto_cap.RegisterControl(m_spinCtrl1, AJ_RegionAdjust);
	146	auto_cap.RegisterControl(m_spinBtn1);
	147	auto_cap.RegisterControl(m_scrollBar1);
	148
	149	auto_cap.RegisterPageTurn();
	150
	151	auto_cap.RegisterControl(m_checkList1);
	152	auto_cap.RegisterControl(m_listBox1);
	153	auto_cap.RegisterControl(m_radioBox1);
	154	auto_cap.RegisterControl(m_staticBox1);
	155	auto_cap.RegisterControl(m_treeCtrl1);
	156	auto_cap.RegisterControl(m_listCtrl1, wxT("wxListCtrl"));
	157
	158	auto_cap.RegisterControl(m_animationCtrl1);
	159	auto_cap.RegisterControl(m_collPane1, wxT("wxCollapsiblePane"), AJ_Union);
	160	auto_cap.RegisterControl(m_collPane2, AJ_UnionEnd);
	161
	162	auto_cap.RegisterPageTurn();
	163
	164	auto_cap.RegisterControl(m_textCtrl1, AJ_Union);
	165	auto_cap.RegisterControl(m_textCtrl2, AJ_UnionEnd);
	166	auto_cap.RegisterControl(m_richText1);
	167
	168	auto_cap.RegisterPageTurn();
	169
	170	auto_cap.RegisterControl(m_colourPicker1, wxT("wxColourPickerCtrl"));
	171	auto_cap.RegisterControl(m_fontPicker1, wxT("wxFontPickerCtrl"));
	172	auto_cap.RegisterControl(m_filePicker1, wxT("wxFilePickerCtrl"), AJ_RegionAdjust);
	173	auto_cap.RegisterControl(m_calendar1, wxT("wxCalendarCtrl"), AJ_RegionAdjust);
	174	auto_cap.RegisterControl(m_datePicker1, wxT("wxDatePickerCtrl"));
	175	auto_cap.RegisterControl(m_genericDirCtrl1, wxT("wxGenericDirCtrl"));
	176	auto_cap.RegisterControl(m_dirPicker1, wxT("wxDirPickerCtrl"), AJ_RegionAdjust);
	177
	178	auto_cap.RegisterPageTurn();
	179
	180	auto_cap.RegisterControl(m_choice1, AJ_Dropdown);
	181	auto_cap.RegisterControl(m_comboBox1, AJ_Dropdown);
	182	auto_cap.RegisterControl(m_bmpComboBox1, AJ_Dropdown);
	183	auto_cap.RegisterControl(m_ownerDrawnComboBox1, AJ_Dropdown);
	184	auto_cap.RegisterControl(m_comboCtrl1, AJ_Dropdown\|AJ_Union);
	185	auto_cap.RegisterControl(m_comboCtrl2, AJ_Dropdown\|AJ_UnionEnd);
	186
	187	auto_cap.CaptureAll();
	188
	189	wxMessageBox(_("All screenshots were generated successfully in the folder:\n ") + dir,
	190	_("Success"), wxOK\|wxICON_INFORMATION, this);
	191	}