[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
// 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/generic/dirdlgg.h"

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

/* 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 */
"                ",
"   @@@@@        ",
"  @$$$$$@       ",
" @$#+#+#$@@@@@@ ",
" @$+#+#+$$$$$$@.",
" @$#+#+#+#+#+#@.",
"@@@@@@@@@@@@@#@.",
"@$$$$$$$$$$@@+@.",
"@$#+#+#+#+##.@@.",
" @$#+#+#+#+#+.@.",
" @$+#+#+#+#+#.@.",
"  @$+#+#+#+##@..",
"  @@@@@@@@@@@@@.",
"   .............",
"                ",
"                "};

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
//-----------------------------------------------------------------------------

class wxDirItemData : public wxTreeItemData
{
public:
  wxDirItemData(wxString& path, wxString& name);
  ~wxDirItemData();
  bool HasSubDirs();
  wxString *m_path, *m_name;
  bool m_isHidden;
  bool m_hasSubDirs;
};

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

class wxDirCtrl: public wxTreeCtrl
{
  DECLARE_DYNAMIC_CLASS(wxDirCtrl)
  
  public:
    bool   m_showHidden;
    wxTreeItemId   m_rootId;
  
    wxDirCtrl(void);
    wxDirCtrl(wxWindow *parent, const wxWindowID id = -1, 
	      const wxString &dir = "/",
	      const wxPoint& pos = wxDefaultPosition,
	      const wxSize& size = wxDefaultSize,
	      const long style = wxTR_HAS_BUTTONS,
	      const wxString& name = "wxTreeCtrl" );
    void OnExpandItem(wxTreeEvent &event );
    void OnCollapseItem(wxTreeEvent &event );
    void ShowHidden( const bool yesno );
    DECLARE_EVENT_TABLE()
 protected:
    void CreateItems(const wxTreeItemId &parent);
    void SetupSections(void);
    wxArrayString m_paths, m_names;
};

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

wxDirItemData::wxDirItemData(wxString& path, wxString& name)
{
  m_path = new wxString(path);
  m_name = new wxString(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()
{
  delete m_path;
  delete m_name;
}

bool wxDirItemData::HasSubDirs()
{
  wxString search = *m_path + "/*";
  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)
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 )
{
  m_imageListNormal = new wxImageList(16, 16, TRUE);
  m_imageListNormal->Add(wxICON(icon1));
  m_imageListNormal->Add(wxICON(icon2));
  SetImageList(m_imageListNormal);
  
  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();
  ADD_SECTION(_T("/"), _("The Computer") )
  wxGetHomeDir(&home);
  ADD_SECTION(home, _("My Home") )
  ADD_SECTION(_T("/mnt"), _("Mounted Devices") )
  ADD_SECTION(_T("/usr"), _("User") )
  ADD_SECTION(_T("/usr/local"), _("User Local") )
  ADD_SECTION(_T("/var"), _("Variables") )
  ADD_SECTION(_T("/etc"), _("Etcetera") )
  ADD_SECTION(_T("/tmp"), _("Temporary") )
}
#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]);
    id = AppendItem( parent, m_names[i], 0, 1, dir_item);
    if (dir_item->m_hasSubDirs) SetItemHasChildren(id);
  }
}

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();
  search = *(data->m_path) + "/*";
  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());
  wxEndBusyCursor();
  
  SortChildren( event.GetItem() );
};


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|wxDIALOG_MODAL)
{
  m_message = message;
  m_dialogStyle = style;
  m_parent = parent;
  
  m_path = defaultPath;

  m_dir = new wxDirCtrl( this, ID_DIRCTRL, "/", wxDefaultPosition, wxDefaultSize, wxTR_HAS_BUTTONS | wxSUNKEN_BORDER ); 
  m_input = new wxTextCtrl( this, ID_TEXTCTRL, m_path, wxDefaultPosition ); 
  // m_check = new wxCheckBox( this, ID_CHECK, _("Show hidden") );
  m_ok = new wxButton( this, ID_OK, _("OK") );
  m_cancel = new wxButton( this, ID_CANCEL, _("Cancel"), wxDefaultPosition, wxSize(75,-1) );
  m_new = new wxButton( this, ID_NEW, _("New...") );

  // m_check->SetValue(TRUE);
  m_ok->SetDefault();
  m_dir->SetFocus();

  doSize();
}

void wxDirDialog::OnSize(wxSizeEvent& WXUNUSED(event))
{
  doSize();
}

void wxDirDialog::doSize()
{
  /* Figure out height of DirCtrl, which is what is left over by
   * the textctrl and the buttons. Manually, because I can't seem
   * to get the constraints stuff to do this */
  int w,h,h2;

  GetClientSize(&w, &h);
  m_input->GetSize(&w,&h2); h -= h2;
  m_ok->GetSize(&w, &h2); h -= h2;
  //m_check->GetSize(&w, &h2); h -= h2;
  h -= 30;

  wxLayoutConstraints *c = new wxLayoutConstraints;
  c->left.SameAs	(this, wxLeft,5);
  c->right.SameAs        (this, wxRight,5);
  c->height.Absolute    (h);
  c->top.SameAs		(this, wxTop,5);
  m_dir->SetConstraints(c);

  c = new wxLayoutConstraints;
  c->left.SameAs	(this, wxLeft,5);
  c->right.SameAs        (this, wxRight,5);
  c->height.AsIs	();
  c->top.Below		(m_dir,5);
  m_input->SetConstraints(c);

  /*  c = new wxLayoutConstraints;
  c->left.SameAs	(this, wxLeft,5);
  c->right.SameAs        (this, wxRight,5);
  c->height.AsIs	();
  c->top.Below		(m_input,5);
  m_check->SetConstraints(c); */

  c = new wxLayoutConstraints;
  c->width.SameAs	(m_cancel, wxWidth);
  c->height.AsIs	();
  c->top.Below		(m_input,10);
  c->centreX.PercentOf	(this, wxWidth, 20);
  m_ok->SetConstraints(c);

  c = new wxLayoutConstraints;
  c->width.SameAs	(m_cancel, wxWidth);
  c->height.AsIs	();
  c->top.Below		(m_input,10);
  c->bottom.SameAs      (this, wxBottom, 5);
  c->centreX.PercentOf	(this, wxWidth, 50);
  m_new->SetConstraints(c);

  c = new wxLayoutConstraints;
  c->width.AsIs	        ();
  c->height.AsIs	();
  c->top.Below		(m_input,10);
  c->centreX.PercentOf	(this, wxWidth, 80);
  m_cancel->SetConstraints(c);

  Layout();
}

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);
  if ( dialog.ShowModal() == wxID_YES ) {
    // Okay, let's make it
    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);
      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) )
{
  wxTextEntryDialog dialog(this, _("Enter the name of the directory to create"),
  	_("Create New Directory"), m_input->GetValue(), wxOK|wxCANCEL);

  while (dialog.ShowModal() == wxID_OK)
  {
    // Okay, let's make it
    if (wxMkdir(dialog.GetValue())) {
      // The new dir was created okay.
      m_path = dialog.GetValue();
      return;
    }
    wxString msg = _("Failed to create directory ")+dialog.GetValue()+
      _("\n(Do you have the required permissions?)") ;
    wxMessageDialog errmsg(this, msg, _("Error creating directory"), wxOK);
    errmsg.ShowModal();
    // Show the create dialog again, until user clicks cancel or enters
    // a valid dir.
  }
}

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