[wxWidgets.git] / src / generic / dirdlgg.cpp

/////////////////////////////////////////////////////////////////////////////
// Name:        dirdlg.cpp
// Purpose:     wxDirDialog
// Author:      Harm van der Heijden and Robert Roebling
// Modified by:
// Created:     12/12/98
// RCS-ID:      $Id$
// Copyright:   (c) Harm van der Heijden and Robert Roebling
// Licence:     wxWindows licence
/////////////////////////////////////////////////////////////////////////////

#ifdef __GNUG__
#pragma implementation "dirdlgg.h"
#endif

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

#ifdef __BORLANDC__
#pragma hdrstop
#endif

#include "wx/defs.h"

#if wxUSE_DIRDLG

#include "wx/utils.h"
#include "wx/dialog.h"
#include "wx/button.h"
#include "wx/layout.h"
#include "wx/msgdlg.h"
#include "wx/textdlg.h"
#include "wx/filefn.h"
#include "wx/cmndata.h"
#include "wx/gdicmn.h"
#include "wx/intl.h"
#include "wx/imaglist.h"
#include "wx/icon.h"
#include "wx/log.h"
#include "wx/sizer.h"

#if wxUSE_STATLINE
    #include "wx/statline.h"
#endif

#include "wx/generic/dirdlgg.h"

// If compiled under Windows, this macro can cause problems
#ifdef GetFirstChild
#undef GetFirstChild
#endif

#ifndef __WXMSW__
/* XPM */
static char * icon1_xpm[] = {
/* width height ncolors chars_per_pixel */
"16 16 6 1",
/* colors */
"   s None  c None",
".  c #000000",
"+  c #c0c0c0",
"@  c #808080",
"#  c #ffff00",
"$  c #ffffff",
/* pixels */
"                ",
"   @@@@@        ",
"  @#+#+#@       ",
" @#+#+#+#@@@@@@ ",
" @$$$$$$$$$$$$@.",
" @$#+#+#+#+#+#@.",
" @$+#+#+#+#+#+@.",
" @$#+#+#+#+#+#@.",
" @$+#+#+#+#+#+@.",
" @$#+#+#+#+#+#@.",
" @$+#+#+#+#+#+@.",
" @$#+#+#+#+#+#@.",
" @@@@@@@@@@@@@@.",
"  ..............",
"                ",
"                "};

/* XPM */
static char * icon2_xpm[] = {
/* width height ncolors chars_per_pixel */
"16 16 6 1",
/* colors */
"   s None  c None",
".  c #000000",
"+  c #c0c0c0",
"@  c #808080",
"#  c #ffff00",
"$  c #ffffff",
/* pixels */
"                ",
"   @@@@@        ",
"  @$$$$$@       ",
" @$#+#+#$@@@@@@ ",
" @$+#+#+$$$$$$@.",
" @$#+#+#+#+#+#@.",
"@@@@@@@@@@@@@#@.",
"@$$$$$$$$$$@@+@.",
"@$#+#+#+#+##.@@.",
" @$#+#+#+#+#+.@.",
" @$+#+#+#+#+#.@.",
"  @$+#+#+#+##@..",
"  @@@@@@@@@@@@@.",
"   .............",
"                ",
"                "};

#endif // !wxMSW

static const int ID_DIRCTRL = 1000;
static const int ID_TEXTCTRL = 1001;
static const int ID_OK = 1002;
static const int ID_CANCEL = 1003;
static const int ID_NEW = 1004;
//static const int ID_CHECK = 1005;

//-----------------------------------------------------------------------------
// wxDirItemData
//-----------------------------------------------------------------------------

wxDirItemData::wxDirItemData(wxString& path, wxString& name)
{
    m_path = path;
    m_name = name;
    /* Insert logic to detect hidden files here
     * In UnixLand we just check whether the first char is a dot
     * For FileNameFromPath read LastDirNameInThisPath ;-) */
    // m_isHidden = (bool)(wxFileNameFromPath(*m_path)[0] == '.');
    m_isHidden = FALSE;
    m_hasSubDirs = HasSubDirs();
}

wxDirItemData::~wxDirItemData()
{
}

void wxDirItemData::SetNewDirName( wxString path )
{
    m_path = path;
    m_name = wxFileNameFromPath( path );
}

bool wxDirItemData::HasSubDirs()
{
    wxString search = m_path + "/*";
    wxLogNull log;
    wxString path = wxFindFirstFile( search, wxDIR );
    return (bool)(!path.IsNull());
}

//-----------------------------------------------------------------------------
// wxDirCtrl
//-----------------------------------------------------------------------------

IMPLEMENT_DYNAMIC_CLASS(wxDirCtrl,wxTreeCtrl)

BEGIN_EVENT_TABLE(wxDirCtrl,wxTreeCtrl)
  EVT_TREE_ITEM_EXPANDING     (-1, wxDirCtrl::OnExpandItem)
  EVT_TREE_ITEM_COLLAPSED     (-1, wxDirCtrl::OnCollapseItem)
  EVT_TREE_BEGIN_LABEL_EDIT   (-1, wxDirCtrl::OnBeginEditItem)
  EVT_TREE_END_LABEL_EDIT     (-1, wxDirCtrl::OnEndEditItem)
END_EVENT_TABLE()

wxDirCtrl::wxDirCtrl(void)
{
    m_showHidden = FALSE;
}

wxDirCtrl::wxDirCtrl(wxWindow *parent, const wxWindowID id, const wxString &WXUNUSED(dir),
            const wxPoint& pos, const wxSize& size,
            const long style, const wxString& name )
 :
  wxTreeCtrl( parent, id, pos, size, style, wxDefaultValidator, name )
{
 #ifndef __WXMSW__
    m_imageListNormal = new wxImageList(16, 16, TRUE);
    m_imageListNormal->Add(wxICON(icon1));
    m_imageListNormal->Add(wxICON(icon2));
    SetImageList(m_imageListNormal);
 #endif

    m_showHidden = FALSE;
    m_rootId = AddRoot( _("Sections") );
    SetItemHasChildren(m_rootId);
    Expand(m_rootId); // automatically expand first level
}

/* Quick macro. Don't worry, I'll #undef it later */
#define ADD_SECTION(a,b) \
  if (wxPathExists((a))) { m_paths.Add( (a) ); m_names.Add( (b) ); };

void wxDirCtrl::SetupSections()
{
  wxString home;

  m_paths.Clear();
  m_names.Clear();
#ifdef __WXMSW__
  // better than nothing
  ADD_SECTION(wxT("c:\\"), _("My Harddisk") )
#else
  ADD_SECTION(wxT("/"), _("The Computer") )
  wxGetHomeDir(&home);
  ADD_SECTION(home, _("My Home") )
  ADD_SECTION(wxT("/mnt"), _("Mounted Devices") )
  ADD_SECTION(wxT("/usr"), _("User") )
  ADD_SECTION(wxT("/usr/local"), _("User Local") )
  ADD_SECTION(wxT("/var"), _("Variables") )
  ADD_SECTION(wxT("/etc"), _("Etcetera") )
  ADD_SECTION(wxT("/tmp"), _("Temporary") )
#endif
}
#undef ADD_SECTION

void wxDirCtrl::CreateItems(const wxTreeItemId &parent)
{
    wxTreeItemId id;
    wxDirItemData *dir_item;

//  wxASSERT(m_paths.Count() == m_names.Count());  ?

    for (unsigned int i=0; i<m_paths.Count(); i++)
    {
  dir_item = new wxDirItemData(m_paths[i],m_names[i]);
#ifdef __WXMSW__
  id = AppendItem( parent, m_names[i], -1, -1, dir_item);
#else
  id = AppendItem( parent, m_names[i], 0, 1, dir_item);
#endif
        if (dir_item->m_hasSubDirs) SetItemHasChildren(id);
    }
}

void wxDirCtrl::OnBeginEditItem(wxTreeEvent &event)
{
    // don't rename the main entry "Sections"
    if (event.GetItem() == m_rootId)
    {
        event.Veto();
        return;
    }

    // don't rename the individual sections
    if (GetParent( event.GetItem() ) == m_rootId)
    {
        event.Veto();
        return;
    }
}

void wxDirCtrl::OnEndEditItem(wxTreeEvent &event)
{
    if ((event.GetLabel().IsEmpty()) ||
        (event.GetLabel() == _(".")) ||
        (event.GetLabel() == _("..")) ||
  (event.GetLabel().First( wxT("/") ) != wxNOT_FOUND))
    {
        wxMessageDialog dialog(this, _("Illegal directory name."), _("Error"), wxOK | wxICON_ERROR );
  dialog.ShowModal();
        event.Veto();
  return;
    }

    wxTreeItemId id = event.GetItem();
    wxDirItemData *data = (wxDirItemData*)GetItemData( id );
    wxASSERT( data );

    wxString new_name( wxPathOnly( data->m_path ) );
    new_name += wxT("/");
    new_name += event.GetLabel();

    wxLogNull log;

    if (wxFileExists(new_name))
    {
        wxMessageDialog dialog(this, _("File name exists already."), _("Error"), wxOK | wxICON_ERROR );
  dialog.ShowModal();
        event.Veto();
    }

    if (wxRenameFile(data->m_path,new_name))
    {
        data->SetNewDirName( new_name );
    }
    else
    {
        wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
  dialog.ShowModal();
        event.Veto();
    }
}

void wxDirCtrl::OnExpandItem(wxTreeEvent &event)
{
    if (event.GetItem() == m_rootId)
    {
        SetupSections();
        CreateItems(m_rootId);
        return;
    }

    // This may take a longish time. Go to busy cursor
    wxBeginBusyCursor();

    wxDirItemData *data = (wxDirItemData *)GetItemData(event.GetItem());
    wxASSERT(data);

    wxString search,path,filename;

    m_paths.Clear();
    m_names.Clear();
#ifdef __WXMSW__
    search = data->m_path + "\\*.*";
#else
    search = data->m_path + "/*";
#endif
    for (path = wxFindFirstFile( search, wxDIR ); !path.IsNull();
       path=wxFindNextFile() )
    {
        filename = wxFileNameFromPath( path );
        /* Don't add "." and ".." to the tree. I think wxFindNextFile
         * also checks this, but I don't quite understand what happens
         * there. Also wxFindNextFile seems to swallow hidden dirs */
        if ((filename != ".") && (filename != ".."))
        {
            m_paths.Add(path);
            m_names.Add(filename);
        }
    }

    CreateItems( event.GetItem() );
    SortChildren( event.GetItem() );

    wxEndBusyCursor();
}

void wxDirCtrl::OnCollapseItem(wxTreeEvent &event )
{
    wxTreeItemId child, parent = event.GetItem();
    long cookie;
    /* Workaround because DeleteChildren has disapeared (why?) and
     * CollapseAndReset doesn't work as advertised (deletes parent too) */
    child = GetFirstChild(parent, cookie);
    while (child.IsOk())
    {
        Delete(child);
        /* Not GetNextChild below, because the cookie mechanism can't
         * handle disappearing children! */
        child = GetFirstChild(parent, cookie);
    }
}

//-----------------------------------------------------------------------------
// wxDirDialog
//-----------------------------------------------------------------------------


#if !USE_SHARED_LIBRARY
IMPLEMENT_CLASS(wxDirDialog, wxDialog)
#else
IMPLEMENT_DYNAMIC_CLASS( wxDirDialog, wxDialog )
#endif

BEGIN_EVENT_TABLE( wxDirDialog, wxDialog )
  EVT_TREE_KEY_DOWN        (ID_DIRCTRL,   wxDirDialog::OnTreeKeyDown)
  EVT_TREE_SEL_CHANGED     (ID_DIRCTRL,   wxDirDialog::OnTreeSelected)
  EVT_SIZE                 (              wxDirDialog::OnSize)
  EVT_BUTTON               (ID_OK,        wxDirDialog::OnOK)
  EVT_BUTTON               (ID_CANCEL,    wxDirDialog::OnCancel)
  EVT_BUTTON               (ID_NEW,       wxDirDialog::OnNew)
  EVT_TEXT_ENTER           (ID_TEXTCTRL,  wxDirDialog::OnOK)
  //  EVT_CHECKBOX             (ID_CHECK,     wxDirDialog::OnCheck)
END_EVENT_TABLE()

wxDirDialog::wxDirDialog(wxWindow *parent, const wxString& message,
       const wxString& defaultPath, long style,
       const wxPoint& pos) :
  wxDialog(parent, -1, message, pos, wxSize(300,300),
     wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER)
{
    m_message = message;
    m_dialogStyle = style;
    m_parent = parent;

    m_path = defaultPath;

    wxBeginBusyCursor();

    wxBoxSizer *topsizer = new wxBoxSizer( wxVERTICAL );

    // 1) dir ctrl
    m_dir = new wxDirCtrl( this, ID_DIRCTRL, "/", wxDefaultPosition, wxSize(200,200),
      wxTR_HAS_BUTTONS | wxSUNKEN_BORDER | wxTR_EDIT_LABELS);
    topsizer->Add( m_dir, 1, wxTOP|wxLEFT|wxRIGHT | wxEXPAND, 10 );

    // 2) text ctrl
    m_input = new wxTextCtrl( this, ID_TEXTCTRL, m_path, wxDefaultPosition );
    topsizer->Add( m_input, 0, wxTOP|wxLEFT|wxRIGHT | wxEXPAND, 10 );

    // m_check = new wxCheckBox( this, ID_CHECK, _("Show hidden") );
    // m_check->SetValue(TRUE);

#if wxUSE_STATLINE
    // 3) static line
    topsizer->Add( new wxStaticLine( this, -1 ), 0, wxEXPAND | wxLEFT|wxRIGHT|wxTOP, 10 );
#endif

    // 4) buttons
    wxSizer* buttonsizer = new wxBoxSizer( wxHORIZONTAL );
    m_ok = new wxButton( this, ID_OK, _("OK") );
    buttonsizer->Add( m_ok, 0, wxLEFT|wxRIGHT, 10 );
    m_cancel = new wxButton( this, ID_CANCEL, _("Cancel") );
    buttonsizer->Add( m_cancel, 0, wxLEFT|wxRIGHT, 10 );
    m_new = new wxButton( this, ID_NEW, _("New...") );
    buttonsizer->Add( m_new, 0, wxLEFT|wxRIGHT, 10 );

    topsizer->Add( buttonsizer, 0, wxALL | wxCENTER, 10 );

    m_ok->SetDefault();
    m_dir->SetFocus();

    SetAutoLayout( TRUE );
    SetSizer( topsizer );

    topsizer->SetSizeHints( this );
    topsizer->Fit( this );

    Centre( wxBOTH );

    wxEndBusyCursor();
}

int wxDirDialog::ShowModal()
{
    m_input->SetValue( m_path );
    return wxDialog::ShowModal();
}

void wxDirDialog::OnTreeSelected( wxTreeEvent &event )
{
    wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData(event.GetItem());
    if (data)
       m_input->SetValue( data->m_path );
};

void wxDirDialog::OnTreeKeyDown( wxTreeEvent &WXUNUSED(event) )
{
    wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData(m_dir->GetSelection());
    if (data)
        m_input->SetValue( data->m_path );
};

void wxDirDialog::OnOK( wxCommandEvent& WXUNUSED(event) )
{
  m_path = m_input->GetValue();
  // Does the path exist? (User may have typed anything in m_input)
  if (wxPathExists(m_path)) {
    // OK, path exists, we're done.
    EndModal(wxID_OK);
    return;
  }
  // Interact with user, find out if the dir is a typo or to be created
  wxString msg( _("The directory ") );
  msg = msg + m_path;
  msg = msg + _("\ndoes not exist\nCreate it now?") ;
  wxMessageDialog dialog(this, msg, _("Directory does not exist"), wxYES_NO | wxICON_WARNING );
  if ( dialog.ShowModal() == wxID_YES ) {
    // Okay, let's make it
    wxLogNull log;
    if (wxMkdir(m_path)) {
      // The new dir was created okay.
      EndModal(wxID_OK);
      return;
    }
    else {
      // Trouble...
      msg = _("Failed to create directory ")+m_path+
  _("\n(Do you have the required permissions?)");
      wxMessageDialog errmsg(this, msg, _("Error creating directory"), wxOK | wxICON_ERROR);
      errmsg.ShowModal();
      // We still don't have a valid dir. Back to the main dialog.
    }
  }
  // User has answered NO to create dir.
}

void wxDirDialog::OnCancel( wxCommandEvent& WXUNUSED(event) )
{
  EndModal(wxID_CANCEL);
}

void wxDirDialog::OnNew( wxCommandEvent& WXUNUSED(event) )
{
    wxTreeItemId id = m_dir->GetSelection();
    if ((id == m_dir->GetRootItem()) ||
        (m_dir->GetParent(id) == m_dir->GetRootItem()))
    {
        wxMessageDialog msg(this, _("You cannot add a new directory to this section."),
                            _("Create directory"), wxOK | wxICON_INFORMATION );
        msg.ShowModal();
        return;
    }

    wxTreeItemId parent = m_dir->GetParent( id );
    wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData( parent );
    wxASSERT( data );

    wxString new_name( wxT("NewName") );
    wxString path( data->m_path );
    path += wxT("/");
    path += new_name;
    if (wxFileExists(path))
    {
        // try NewName0, NewName1 etc.
        int i = 0;
  do {
            new_name = wxT("NewName");
      wxString num;
      num.Printf( "%d", i );
      new_name += num;

            path = data->m_path;
            path += wxT("/");
            path += new_name;
      i++;
  } while (wxFileExists(path));
    }

    wxLogNull log;
    if (!wxMkdir(path))
    {
        wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
  dialog.ShowModal();
        return;
    }

    wxDirItemData *new_data = new wxDirItemData( path, new_name );
    wxTreeItemId new_id = m_dir->AppendItem( parent, new_name, 0, 1, new_data );
    m_dir->EnsureVisible( new_id );
    m_dir->EditLabel( new_id );
}

/*
void wxDirDialog::OnCheck( wxCommandEvent& WXUNUSED(event) )
{
  printf("Checkbox clicked: %s\n", ( m_check->GetValue() ? "on" : "off" ) );
}
*/

#endif
Commit	Line	Data
d7a15103 JS	1	/////////////////////////////////////////////////////////////////////////////
	2	// Name: dirdlg.cpp
	3	// Purpose: wxDirDialog
	4	// Author: Harm van der Heijden and Robert Roebling
	5	// Modified by:
	6	// Created: 12/12/98
8b17ba72	7	// RCS-ID: $Id$
d7a15103	8	// Copyright: (c) Harm van der Heijden and Robert Roebling
e90c1d2a	9	// Licence: wxWindows licence
d7a15103 JS	10	/////////////////////////////////////////////////////////////////////////////
	11
	12	#ifdef __GNUG__
	13	#pragma implementation "dirdlgg.h"
	14	#endif
	15
1e6d9499 JS	16	// For compilers that support precompilation, includes "wx.h".
	17	#include "wx/wxprec.h"
	18
	19	#ifdef __BORLANDC__
	20	#pragma hdrstop
	21	#endif
	22
d7a15103	23	#include "wx/defs.h"
ce4169a4 RR	24
	25	#if wxUSE_DIRDLG
	26
d7a15103 JS	27	#include "wx/utils.h"
	28	#include "wx/dialog.h"
	29	#include "wx/button.h"
	30	#include "wx/layout.h"
	31	#include "wx/msgdlg.h"
	32	#include "wx/textdlg.h"
	33	#include "wx/filefn.h"
	34	#include "wx/cmndata.h"
	35	#include "wx/gdicmn.h"
	36	#include "wx/intl.h"
	37	#include "wx/imaglist.h"
25889d3c	38	#include "wx/icon.h"
dc6c62a9 RR	39	#include "wx/log.h"
	40	#include "wx/sizer.h"
	41
	42	#if wxUSE_STATLINE
	43	#include "wx/statline.h"
	44	#endif
d7a15103 JS	45
	46	#include "wx/generic/dirdlgg.h"
	47
4bf78aae JS	48	// If compiled under Windows, this macro can cause problems
	49	#ifdef GetFirstChild
	50	#undef GetFirstChild
	51	#endif
	52
a17e237f	53	#ifndef __WXMSW__
d7a15103 JS	54	/* XPM */
	55	static char * icon1_xpm[] = {
	56	/* width height ncolors chars_per_pixel */
	57	"16 16 6 1",
	58	/* colors */
e90c1d2a VZ	59	" s None c None",
	60	". c #000000",
	61	"+ c #c0c0c0",
	62	"@ c #808080",
	63	"# c #ffff00",
	64	"$ c #ffffff",
d7a15103 JS	65	/* pixels */
	66	" ",
	67	" @@@@@ ",
	68	" @#+#+#@ ",
	69	" @#+#+#+#@@@@@@ ",
	70	" @$$$$$$$$$$$$@.",
	71	" @$#+#+#+#+#+#@.",
	72	" @$+#+#+#+#+#+@.",
	73	" @$#+#+#+#+#+#@.",
	74	" @$+#+#+#+#+#+@.",
	75	" @$#+#+#+#+#+#@.",
	76	" @$+#+#+#+#+#+@.",
	77	" @$#+#+#+#+#+#@.",
	78	" @@@@@@@@@@@@@@.",
	79	" ..............",
	80	" ",
	81	" "};
	82
0659e7ee RR	83	/* XPM */
	84	static char * icon2_xpm[] = {
	85	/* width height ncolors chars_per_pixel */
	86	"16 16 6 1",
	87	/* colors */
e90c1d2a VZ	88	" s None c None",
	89	". c #000000",
	90	"+ c #c0c0c0",
	91	"@ c #808080",
	92	"# c #ffff00",
	93	"$ c #ffffff",
0659e7ee RR	94	/* pixels */
	95	" ",
	96	" @@@@@ ",
	97	" @$$$$$@ ",
	98	" @$#+#+#$@@@@@@ ",
	99	" @$+#+#+$$$$$$@.",
	100	" @$#+#+#+#+#+#@.",
	101	"@@@@@@@@@@@@@#@.",
	102	"@$$$$$$$$$$@@+@.",
	103	"@$#+#+#+#+##.@@.",
	104	" @$#+#+#+#+#+.@.",
	105	" @$+#+#+#+#+#.@.",
	106	" @$+#+#+#+##@..",
	107	" @@@@@@@@@@@@@.",
	108	" .............",
	109	" ",
	110	" "};
d7a15103	111
a17e237f VZ	112	#endif // !wxMSW
a17e237f VZ	113
d7a15103 JS	114	static const int ID_DIRCTRL = 1000;
	115	static const int ID_TEXTCTRL = 1001;
	116	static const int ID_OK = 1002;
	117	static const int ID_CANCEL = 1003;
	118	static const int ID_NEW = 1004;
	119	//static const int ID_CHECK = 1005;
	120
	121	//-----------------------------------------------------------------------------
	122	// wxDirItemData
	123	//-----------------------------------------------------------------------------
	124
dc6c62a9	125	wxDirItemData::wxDirItemData(wxString& path, wxString& name)
0659e7ee	126	{
dc6c62a9 RR	127	m_path = path;
dc6c62a9 RR	128	m_name = name;
e90c1d2a VZ	129	/* Insert logic to detect hidden files here
e90c1d2a VZ	130	* In UnixLand we just check whether the first char is a dot
dc6c62a9 RR	131	* For FileNameFromPath read LastDirNameInThisPath ;-) */
	132	// m_isHidden = (bool)(wxFileNameFromPath(*m_path)[0] == '.');
	133	m_isHidden = FALSE;
	134	m_hasSubDirs = HasSubDirs();
	135	}
0659e7ee	136
dc6c62a9	137	wxDirItemData::~wxDirItemData()
d7a15103	138	{
d7a15103 JS	139	}
d7a15103 JS	140
dc6c62a9	141	void wxDirItemData::SetNewDirName( wxString path )
d7a15103	142	{
dc6c62a9 RR	143	m_path = path;
dc6c62a9 RR	144	m_name = wxFileNameFromPath( path );
d7a15103 JS	145	}
	146
	147	bool wxDirItemData::HasSubDirs()
	148	{
dc6c62a9 RR	149	wxString search = m_path + "/*";
	150	wxLogNull log;
	151	wxString path = wxFindFirstFile( search, wxDIR );
	152	return (bool)(!path.IsNull());
d7a15103 JS	153	}
	154
	155	//-----------------------------------------------------------------------------
	156	// wxDirCtrl
	157	//-----------------------------------------------------------------------------
	158
	159	IMPLEMENT_DYNAMIC_CLASS(wxDirCtrl,wxTreeCtrl)
	160
	161	BEGIN_EVENT_TABLE(wxDirCtrl,wxTreeCtrl)
	162	EVT_TREE_ITEM_EXPANDING (-1, wxDirCtrl::OnExpandItem)
	163	EVT_TREE_ITEM_COLLAPSED (-1, wxDirCtrl::OnCollapseItem)
dc6c62a9 RR	164	EVT_TREE_BEGIN_LABEL_EDIT (-1, wxDirCtrl::OnBeginEditItem)
dc6c62a9 RR	165	EVT_TREE_END_LABEL_EDIT (-1, wxDirCtrl::OnEndEditItem)
d7a15103 JS	166	END_EVENT_TABLE()
	167
	168	wxDirCtrl::wxDirCtrl(void)
	169	{
dc6c62a9 RR	170	m_showHidden = FALSE;
dc6c62a9 RR	171	}
d7a15103 JS	172
	173	wxDirCtrl::wxDirCtrl(wxWindow *parent, const wxWindowID id, const wxString &WXUNUSED(dir),
	174	const wxPoint& pos, const wxSize& size,
	175	const long style, const wxString& name )
	176	:
	177	wxTreeCtrl( parent, id, pos, size, style, wxDefaultValidator, name )
	178	{
e951f847 HH	179	#ifndef __WXMSW__
e951f847 HH	180	m_imageListNormal = new wxImageList(16, 16, TRUE);
dc6c62a9 RR	181	m_imageListNormal->Add(wxICON(icon1));
	182	m_imageListNormal->Add(wxICON(icon2));
	183	SetImageList(m_imageListNormal);
3f3cec48	184	#endif
e90c1d2a	185
dc6c62a9 RR	186	m_showHidden = FALSE;
	187	m_rootId = AddRoot( _("Sections") );
	188	SetItemHasChildren(m_rootId);
	189	Expand(m_rootId); // automatically expand first level
	190	}
d7a15103 JS	191
	192	/* Quick macro. Don't worry, I'll #undef it later */
	193	#define ADD_SECTION(a,b) \
	194	if (wxPathExists((a))) { m_paths.Add( (a) ); m_names.Add( (b) ); };
	195
e90c1d2a	196	void wxDirCtrl::SetupSections()
d7a15103 JS	197	{
	198	wxString home;
	199
	200	m_paths.Clear();
	201	m_names.Clear();
3f3cec48 RR	202	#ifdef __WXMSW__
3f3cec48 RR	203	// better than nothing
223d09f6	204	ADD_SECTION(wxT("c:\\"), _("My Harddisk") )
3f3cec48	205	#else
223d09f6	206	ADD_SECTION(wxT("/"), _("The Computer") )
d7a15103 JS	207	wxGetHomeDir(&home);
d7a15103 JS	208	ADD_SECTION(home, _("My Home") )
223d09f6 KB	209	ADD_SECTION(wxT("/mnt"), _("Mounted Devices") )
	210	ADD_SECTION(wxT("/usr"), _("User") )
	211	ADD_SECTION(wxT("/usr/local"), _("User Local") )
	212	ADD_SECTION(wxT("/var"), _("Variables") )
	213	ADD_SECTION(wxT("/etc"), _("Etcetera") )
	214	ADD_SECTION(wxT("/tmp"), _("Temporary") )
3f3cec48	215	#endif
d7a15103 JS	216	}
	217	#undef ADD_SECTION
	218
	219	void wxDirCtrl::CreateItems(const wxTreeItemId &parent)
	220	{
dc6c62a9 RR	221	wxTreeItemId id;
dc6c62a9 RR	222	wxDirItemData *dir_item;
d7a15103 JS	223
d7a15103 JS	224	// wxASSERT(m_paths.Count() == m_names.Count()); ?
e90c1d2a VZ	225
e90c1d2a VZ	226	for (unsigned int i=0; i<m_paths.Count(); i++)
dc6c62a9	227	{
e90c1d2a	228	dir_item = new wxDirItemData(m_paths[i],m_names[i]);
3f3cec48	229	#ifdef __WXMSW__
e90c1d2a	230	id = AppendItem( parent, m_names[i], -1, -1, dir_item);
3f3cec48	231	#else
e90c1d2a	232	id = AppendItem( parent, m_names[i], 0, 1, dir_item);
3f3cec48	233	#endif
dc6c62a9 RR	234	if (dir_item->m_hasSubDirs) SetItemHasChildren(id);
dc6c62a9 RR	235	}
d7a15103 JS	236	}
d7a15103 JS	237
dc6c62a9	238	void wxDirCtrl::OnBeginEditItem(wxTreeEvent &event)
d7a15103	239	{
dc6c62a9 RR	240	// don't rename the main entry "Sections"
	241	if (event.GetItem() == m_rootId)
	242	{
	243	event.Veto();
	244	return;
	245	}
e90c1d2a	246
dc6c62a9 RR	247	// don't rename the individual sections
	248	if (GetParent( event.GetItem() ) == m_rootId)
	249	{
	250	event.Veto();
	251	return;
	252	}
	253	}
d7a15103	254
dc6c62a9 RR	255	void wxDirCtrl::OnEndEditItem(wxTreeEvent &event)
	256	{
	257	if ((event.GetLabel().IsEmpty()) \|\|
	258	(event.GetLabel() == _(".")) \|\|
	259	(event.GetLabel() == _("..")) \|\|
223d09f6	260	(event.GetLabel().First( wxT("/") ) != wxNOT_FOUND))
dc6c62a9 RR	261	{
dc6c62a9 RR	262	wxMessageDialog dialog(this, _("Illegal directory name."), _("Error"), wxOK \| wxICON_ERROR );
e90c1d2a	263	dialog.ShowModal();
dc6c62a9	264	event.Veto();
e90c1d2a	265	return;
dc6c62a9	266	}
d7a15103	267
dc6c62a9 RR	268	wxTreeItemId id = event.GetItem();
	269	wxDirItemData data = (wxDirItemData)GetItemData( id );
	270	wxASSERT( data );
e90c1d2a	271
dc6c62a9	272	wxString new_name( wxPathOnly( data->m_path ) );
223d09f6	273	new_name += wxT("/");
dc6c62a9	274	new_name += event.GetLabel();
e90c1d2a	275
dc6c62a9	276	wxLogNull log;
e90c1d2a	277
dc6c62a9 RR	278	if (wxFileExists(new_name))
	279	{
	280	wxMessageDialog dialog(this, _("File name exists already."), _("Error"), wxOK \| wxICON_ERROR );
e90c1d2a	281	dialog.ShowModal();
dc6c62a9 RR	282	event.Veto();
dc6c62a9 RR	283	}
e90c1d2a	284
dc6c62a9 RR	285	if (wxRenameFile(data->m_path,new_name))
	286	{
	287	data->SetNewDirName( new_name );
	288	}
	289	else
	290	{
	291	wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK \| wxICON_ERROR );
e90c1d2a	292	dialog.ShowModal();
dc6c62a9 RR	293	event.Veto();
	294	}
	295	}
d7a15103	296
dc6c62a9 RR	297	void wxDirCtrl::OnExpandItem(wxTreeEvent &event)
	298	{
	299	if (event.GetItem() == m_rootId)
	300	{
	301	SetupSections();
	302	CreateItems(m_rootId);
	303	return;
d7a15103	304	}
d7a15103	305
dc6c62a9 RR	306	// This may take a longish time. Go to busy cursor
	307	wxBeginBusyCursor();
	308
	309	wxDirItemData data = (wxDirItemData )GetItemData(event.GetItem());
	310	wxASSERT(data);
	311
	312	wxString search,path,filename;
	313
	314	m_paths.Clear();
	315	m_names.Clear();
5e4ff78a RR	316	#ifdef __WXMSW__
	317	search = data->m_path + "\\.";
	318	#else
dc6c62a9	319	search = data->m_path + "/*";
5e4ff78a	320	#endif
e90c1d2a VZ	321	for (path = wxFindFirstFile( search, wxDIR ); !path.IsNull();
e90c1d2a VZ	322	path=wxFindNextFile() )
dc6c62a9 RR	323	{
	324	filename = wxFileNameFromPath( path );
	325	/* Don't add "." and ".." to the tree. I think wxFindNextFile
	326	* also checks this, but I don't quite understand what happens
	327	* there. Also wxFindNextFile seems to swallow hidden dirs */
e90c1d2a	328	if ((filename != ".") && (filename != ".."))
dc6c62a9 RR	329	{
	330	m_paths.Add(path);
	331	m_names.Add(filename);
	332	}
	333	}
e90c1d2a	334
dc6c62a9 RR	335	CreateItems( event.GetItem() );
dc6c62a9 RR	336	SortChildren( event.GetItem() );
e90c1d2a	337
dc6c62a9 RR	338	wxEndBusyCursor();
dc6c62a9 RR	339	}
d7a15103	340
1e6d9499	341	void wxDirCtrl::OnCollapseItem(wxTreeEvent &event )
d7a15103	342	{
dc6c62a9 RR	343	wxTreeItemId child, parent = event.GetItem();
	344	long cookie;
	345	/* Workaround because DeleteChildren has disapeared (why?) and
	346	* CollapseAndReset doesn't work as advertised (deletes parent too) */
d7a15103	347	child = GetFirstChild(parent, cookie);
e90c1d2a	348	while (child.IsOk())
dc6c62a9 RR	349	{
dc6c62a9 RR	350	Delete(child);
e90c1d2a	351	/* Not GetNextChild below, because the cookie mechanism can't
dc6c62a9 RR	352	* handle disappearing children! */
	353	child = GetFirstChild(parent, cookie);
	354	}
	355	}
d7a15103 JS	356
	357	//-----------------------------------------------------------------------------
	358	// wxDirDialog
	359	//-----------------------------------------------------------------------------
	360
	361
	362	#if !USE_SHARED_LIBRARY
	363	IMPLEMENT_CLASS(wxDirDialog, wxDialog)
	364	#else
	365	IMPLEMENT_DYNAMIC_CLASS( wxDirDialog, wxDialog )
	366	#endif
	367
	368	BEGIN_EVENT_TABLE( wxDirDialog, wxDialog )
	369	EVT_TREE_KEY_DOWN (ID_DIRCTRL, wxDirDialog::OnTreeKeyDown)
	370	EVT_TREE_SEL_CHANGED (ID_DIRCTRL, wxDirDialog::OnTreeSelected)
	371	EVT_SIZE ( wxDirDialog::OnSize)
	372	EVT_BUTTON (ID_OK, wxDirDialog::OnOK)
	373	EVT_BUTTON (ID_CANCEL, wxDirDialog::OnCancel)
	374	EVT_BUTTON (ID_NEW, wxDirDialog::OnNew)
	375	EVT_TEXT_ENTER (ID_TEXTCTRL, wxDirDialog::OnOK)
	376	// EVT_CHECKBOX (ID_CHECK, wxDirDialog::OnCheck)
	377	END_EVENT_TABLE()
	378
	379	wxDirDialog::wxDirDialog(wxWindow *parent, const wxString& message,
e90c1d2a VZ	380	const wxString& defaultPath, long style,
e90c1d2a VZ	381	const wxPoint& pos) :
d7a15103	382	wxDialog(parent, -1, message, pos, wxSize(300,300),
e90c1d2a	383	wxDEFAULT_DIALOG_STYLE \| wxRESIZE_BORDER)
d7a15103	384	{
dc6c62a9 RR	385	m_message = message;
	386	m_dialogStyle = style;
	387	m_parent = parent;
e90c1d2a	388
dc6c62a9	389	m_path = defaultPath;
e90c1d2a	390
dc6c62a9	391	wxBeginBusyCursor();
e90c1d2a	392
dc6c62a9 RR	393	wxBoxSizer *topsizer = new wxBoxSizer( wxVERTICAL );
	394
	395	// 1) dir ctrl
e90c1d2a VZ	396	m_dir = new wxDirCtrl( this, ID_DIRCTRL, "/", wxDefaultPosition, wxSize(200,200),
e90c1d2a VZ	397	wxTR_HAS_BUTTONS \| wxSUNKEN_BORDER \| wxTR_EDIT_LABELS);
dc6c62a9	398	topsizer->Add( m_dir, 1, wxTOP\|wxLEFT\|wxRIGHT \| wxEXPAND, 10 );
e90c1d2a	399
dc6c62a9	400	// 2) text ctrl
e90c1d2a	401	m_input = new wxTextCtrl( this, ID_TEXTCTRL, m_path, wxDefaultPosition );
dc6c62a9	402	topsizer->Add( m_input, 0, wxTOP\|wxLEFT\|wxRIGHT \| wxEXPAND, 10 );
e90c1d2a	403
dc6c62a9 RR	404	// m_check = new wxCheckBox( this, ID_CHECK, _("Show hidden") );
dc6c62a9 RR	405	// m_check->SetValue(TRUE);
e90c1d2a	406
dc6c62a9 RR	407	#if wxUSE_STATLINE
	408	// 3) static line
	409	topsizer->Add( new wxStaticLine( this, -1 ), 0, wxEXPAND \| wxLEFT\|wxRIGHT\|wxTOP, 10 );
	410	#endif
d7a15103	411
dc6c62a9 RR	412	// 4) buttons
	413	wxSizer* buttonsizer = new wxBoxSizer( wxHORIZONTAL );
	414	m_ok = new wxButton( this, ID_OK, _("OK") );
	415	buttonsizer->Add( m_ok, 0, wxLEFT\|wxRIGHT, 10 );
	416	m_cancel = new wxButton( this, ID_CANCEL, _("Cancel") );
	417	buttonsizer->Add( m_cancel, 0, wxLEFT\|wxRIGHT, 10 );
	418	m_new = new wxButton( this, ID_NEW, _("New...") );
	419	buttonsizer->Add( m_new, 0, wxLEFT\|wxRIGHT, 10 );
e90c1d2a	420
dc6c62a9 RR	421	topsizer->Add( buttonsizer, 0, wxALL \| wxCENTER, 10 );
	422
	423	m_ok->SetDefault();
	424	m_dir->SetFocus();
e90c1d2a	425
dc6c62a9 RR	426	SetAutoLayout( TRUE );
dc6c62a9 RR	427	SetSizer( topsizer );
e90c1d2a	428
dc6c62a9 RR	429	topsizer->SetSizeHints( this );
	430	topsizer->Fit( this );
	431
	432	Centre( wxBOTH );
	433
	434	wxEndBusyCursor();
d7a15103 JS	435	}
	436
	437	int wxDirDialog::ShowModal()
	438	{
dc6c62a9 RR	439	m_input->SetValue( m_path );
dc6c62a9 RR	440	return wxDialog::ShowModal();
d7a15103 JS	441	}
	442
	443	void wxDirDialog::OnTreeSelected( wxTreeEvent &event )
	444	{
dc6c62a9	445	wxDirItemData data = (wxDirItemData)m_dir->GetItemData(event.GetItem());
e90c1d2a	446	if (data)
dc6c62a9	447	m_input->SetValue( data->m_path );
d7a15103 JS	448	};
d7a15103 JS	449
1e6d9499	450	void wxDirDialog::OnTreeKeyDown( wxTreeEvent &WXUNUSED(event) )
d7a15103	451	{
dc6c62a9	452	wxDirItemData data = (wxDirItemData)m_dir->GetItemData(m_dir->GetSelection());
e90c1d2a	453	if (data)
dc6c62a9	454	m_input->SetValue( data->m_path );
d7a15103 JS	455	};
	456
	457	void wxDirDialog::OnOK( wxCommandEvent& WXUNUSED(event) )
	458	{
	459	m_path = m_input->GetValue();
	460	// Does the path exist? (User may have typed anything in m_input)
	461	if (wxPathExists(m_path)) {
	462	// OK, path exists, we're done.
	463	EndModal(wxID_OK);
	464	return;
	465	}
	466	// Interact with user, find out if the dir is a typo or to be created
	467	wxString msg( _("The directory ") );
	468	msg = msg + m_path;
	469	msg = msg + _("\ndoes not exist\nCreate it now?") ;
dc6c62a9	470	wxMessageDialog dialog(this, msg, _("Directory does not exist"), wxYES_NO \| wxICON_WARNING );
d7a15103 JS	471	if ( dialog.ShowModal() == wxID_YES ) {
d7a15103 JS	472	// Okay, let's make it
dc6c62a9	473	wxLogNull log;
d7a15103 JS	474	if (wxMkdir(m_path)) {
	475	// The new dir was created okay.
	476	EndModal(wxID_OK);
	477	return;
	478	}
	479	else {
	480	// Trouble...
	481	msg = _("Failed to create directory ")+m_path+
e90c1d2a	482	_("\n(Do you have the required permissions?)");
dc6c62a9	483	wxMessageDialog errmsg(this, msg, _("Error creating directory"), wxOK \| wxICON_ERROR);
d7a15103 JS	484	errmsg.ShowModal();
	485	// We still don't have a valid dir. Back to the main dialog.
	486	}
	487	}
	488	// User has answered NO to create dir.
	489	}
	490
	491	void wxDirDialog::OnCancel( wxCommandEvent& WXUNUSED(event) )
	492	{
	493	EndModal(wxID_CANCEL);
	494	}
	495
	496	void wxDirDialog::OnNew( wxCommandEvent& WXUNUSED(event) )
	497	{
dc6c62a9 RR	498	wxTreeItemId id = m_dir->GetSelection();
	499	if ((id == m_dir->GetRootItem()) \|\|
	500	(m_dir->GetParent(id) == m_dir->GetRootItem()))
	501	{
e90c1d2a VZ	502	wxMessageDialog msg(this, _("You cannot add a new directory to this section."),
e90c1d2a VZ	503	_("Create directory"), wxOK \| wxICON_INFORMATION );
dc6c62a9 RR	504	msg.ShowModal();
	505	return;
	506	}
d7a15103	507
dc6c62a9 RR	508	wxTreeItemId parent = m_dir->GetParent( id );
	509	wxDirItemData data = (wxDirItemData)m_dir->GetItemData( parent );
	510	wxASSERT( data );
e90c1d2a	511
223d09f6	512	wxString new_name( wxT("NewName") );
dc6c62a9	513	wxString path( data->m_path );
223d09f6	514	path += wxT("/");
dc6c62a9 RR	515	path += new_name;
	516	if (wxFileExists(path))
	517	{
	518	// try NewName0, NewName1 etc.
	519	int i = 0;
e90c1d2a	520	do {
223d09f6	521	new_name = wxT("NewName");
e90c1d2a VZ	522	wxString num;
	523	num.Printf( "%d", i );
	524	new_name += num;
	525
dc6c62a9	526	path = data->m_path;
223d09f6	527	path += wxT("/");
dc6c62a9	528	path += new_name;
e90c1d2a VZ	529	i++;
e90c1d2a VZ	530	} while (wxFileExists(path));
d7a15103	531	}
e90c1d2a	532
dc6c62a9	533	wxLogNull log;
e90c1d2a	534	if (!wxMkdir(path))
dc6c62a9 RR	535	{
dc6c62a9 RR	536	wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK \| wxICON_ERROR );
e90c1d2a	537	dialog.ShowModal();
dc6c62a9 RR	538	return;
	539	}
	540
	541	wxDirItemData *new_data = new wxDirItemData( path, new_name );
	542	wxTreeItemId new_id = m_dir->AppendItem( parent, new_name, 0, 1, new_data );
	543	m_dir->EnsureVisible( new_id );
	544	m_dir->EditLabel( new_id );
d7a15103 JS	545	}
	546
	547	/*
	548	void wxDirDialog::OnCheck( wxCommandEvent& WXUNUSED(event) )
	549	{
	550	printf("Checkbox clicked: %s\n", ( m_check->GetValue() ? "on" : "off" ) );
	551	}
	552	*/
ce4169a4 RR	553
ce4169a4 RR	554	#endif