+void wxHeaderCtrlBase::SetColumnCount(unsigned int count)
+{
+ if ( count != GetColumnCount() )
+ OnColumnCountChanging(count);
+
+ // still call DoSetCount() even if the count didn't really change in order
+ // to update all the columns
+ DoSetCount(count);
+}
+
+// ----------------------------------------------------------------------------
+// wxHeaderCtrlBase event handling
+// ----------------------------------------------------------------------------
+
+void wxHeaderCtrlBase::OnSeparatorDClick(wxHeaderCtrlEvent& event)
+{
+ const unsigned col = event.GetColumn();
+
+ int w = wxWindowBase::GetTextExtent(GetColumn(col).GetTitle()).x;
+ w += 4*GetCharWidth(); // add some arbitrary margins around text
+
+ if ( !UpdateColumnWidthToFit(col, w) )
+ event.Skip();
+ else
+ UpdateColumn(col);
+}
+
+#if wxUSE_MENUS
+
+void wxHeaderCtrlBase::OnRClick(wxHeaderCtrlEvent& event)
+{
+ if ( !HasFlag(wxHD_ALLOW_HIDE) )
+ {
+ event.Skip();
+ return;
+ }
+
+ ShowColumnsMenu(ScreenToClient(wxGetMousePosition()));
+}
+
+#endif // wxUSE_MENUS
+
+// ----------------------------------------------------------------------------
+// wxHeaderCtrlBase column reordering
+// ----------------------------------------------------------------------------
+
+void wxHeaderCtrlBase::SetColumnsOrder(const wxArrayInt& order)
+{
+ const unsigned count = GetColumnCount();
+ wxCHECK_RET( order.size() == count, "wrong number of columns" );
+
+ // check the array validity
+ wxArrayInt seen(count, 0);
+ for ( unsigned n = 0; n < count; n++ )
+ {
+ const unsigned idx = order[n];
+ wxCHECK_RET( idx < count, "invalid column index" );
+ wxCHECK_RET( !seen[idx], "duplicate column index" );
+
+ seen[idx] = 1;
+ }
+
+ DoSetColumnsOrder(order);
+
+ // TODO-RTL: do we need to reverse the array?
+}
+
+void wxHeaderCtrlBase::ResetColumnsOrder()
+{
+ const unsigned count = GetColumnCount();
+ wxArrayInt order(count);
+ for ( unsigned n = 0; n < count; n++ )
+ order[n] = n;
+
+ DoSetColumnsOrder(order);
+}
+
+wxArrayInt wxHeaderCtrlBase::GetColumnsOrder() const
+{
+ const wxArrayInt order = DoGetColumnsOrder();
+
+ wxASSERT_MSG( order.size() == GetColumnCount(), "invalid order array" );
+
+ return order;
+}
+
+unsigned int wxHeaderCtrlBase::GetColumnAt(unsigned int pos) const
+{
+ wxCHECK_MSG( pos < GetColumnCount(), wxNO_COLUMN, "invalid position" );
+
+ return GetColumnsOrder()[pos];
+}
+
+unsigned int wxHeaderCtrlBase::GetColumnPos(unsigned int idx) const
+{
+ const unsigned count = GetColumnCount();
+
+ wxCHECK_MSG( idx < count, wxNO_COLUMN, "invalid index" );
+
+ const wxArrayInt order = GetColumnsOrder();
+ for ( unsigned n = 0; n < count; n++ )
+ {
+ if ( (unsigned)order[n] == idx )
+ return n;
+ }
+
+ wxFAIL_MSG( "column unexpectedly not displayed at all" );
+
+ return wxNO_COLUMN;
+}
+
+/* static */
+void wxHeaderCtrlBase::MoveColumnInOrderArray(wxArrayInt& order,
+ unsigned int idx,
+ unsigned int pos)
+{
+ const unsigned count = order.size();
+
+ wxArrayInt orderNew;
+ orderNew.reserve(count);
+ for ( unsigned n = 0; ; n++ )
+ {
+ // NB: order of checks is important for this to work when the new
+ // column position is the same as the old one
+
+ // insert the column at its new position
+ if ( orderNew.size() == pos )
+ orderNew.push_back(idx);
+
+ if ( n == count )
+ break;
+
+ // delete the column from its old position
+ const unsigned idxOld = order[n];
+ if ( idxOld == idx )
+ continue;
+
+ orderNew.push_back(idxOld);
+ }
+
+ order.swap(orderNew);
+}
+
+void
+wxHeaderCtrlBase::DoResizeColumnIndices(wxArrayInt& colIndices, unsigned int count)
+{
+ // update the column indices array if necessary
+ const unsigned countOld = colIndices.size();
+ if ( count > countOld )
+ {
+ // all new columns have default positions equal to their indices
+ for ( unsigned n = countOld; n < count; n++ )
+ colIndices.push_back(n);
+ }
+ else if ( count < countOld )
+ {
+ // filter out all the positions which are invalid now while keeping the
+ // order of the remaining ones
+ wxArrayInt colIndicesNew;
+ colIndicesNew.reserve(count);
+ for ( unsigned n = 0; n < countOld; n++ )
+ {
+ const unsigned idx = colIndices[n];
+ if ( idx < count )
+ colIndicesNew.push_back(idx);
+ }
+
+ colIndices.swap(colIndicesNew);
+ }
+ //else: count didn't really change, nothing to do
+
+ wxASSERT_MSG( colIndices.size() == count, "logic error" );
+}
+
+// ----------------------------------------------------------------------------
+// wxHeaderCtrl extra UI
+// ----------------------------------------------------------------------------
+
+#if wxUSE_MENUS
+
+void wxHeaderCtrlBase::AddColumnsItems(wxMenu& menu, int idColumnsBase)
+{
+ const unsigned count = GetColumnCount();
+ for ( unsigned n = 0; n < count; n++ )
+ {
+ const wxHeaderColumn& col = GetColumn(n);
+ menu.AppendCheckItem(idColumnsBase + n, col.GetTitle());
+ if ( col.IsShown() )
+ menu.Check(n, true);
+ }
+}
+
+bool wxHeaderCtrlBase::ShowColumnsMenu(const wxPoint& pt, const wxString& title)
+{
+ // construct the menu with the entries for all columns
+ wxMenu menu;
+ if ( !title.empty() )
+ menu.SetTitle(title);
+
+ AddColumnsItems(menu);
+
+ // ... and an extra one to show the customization dialog if the user is
+ // allowed to reorder the columns too
+ const unsigned count = GetColumnCount();
+ if ( HasFlag(wxHD_ALLOW_REORDER) )
+ {
+ menu.AppendSeparator();
+ menu.Append(count, _("&Customize..."));
+ }
+
+ // do show the menu and get the user selection
+ const int rc = GetPopupMenuSelectionFromUser(menu, pt);
+ if ( rc == wxID_NONE )
+ return false;
+
+ if ( static_cast<unsigned>(rc) == count )
+ {
+ return ShowCustomizeDialog();
+ }
+ else // a column selected from the menu
+ {
+ UpdateColumnVisibility(rc, !GetColumn(rc).IsShown());
+ }
+
+ return true;
+}
+
+#endif // wxUSE_MENUS
+
+bool wxHeaderCtrlBase::ShowCustomizeDialog()
+{
+#if wxUSE_REARRANGECTRL
+ // prepare the data for showing the dialog
+ wxArrayInt order = GetColumnsOrder();
+
+ const unsigned count = GetColumnCount();
+
+ // notice that titles are always in the index order, they will be shown
+ // rearranged according to the display order in the dialog
+ wxArrayString titles;
+ titles.reserve(count);
+ for ( unsigned n = 0; n < count; n++ )
+ titles.push_back(GetColumn(n).GetTitle());
+
+ // this loop is however over positions and not indices
+ unsigned pos;
+ for ( pos = 0; pos < count; pos++ )
+ {
+ int& idx = order[pos];
+ if ( GetColumn(idx).IsHidden() )
+ {
+ // indicate that this one is hidden
+ idx = ~idx;
+ }
+ }
+
+ // do show it
+ wxHeaderColumnsRearrangeDialog dlg(this, order, titles);
+ if ( dlg.ShowModal() == wxID_OK )
+ {
+ // and apply the changes
+ order = dlg.GetOrder();
+ for ( pos = 0; pos < count; pos++ )
+ {
+ int& idx = order[pos];
+ const bool show = idx >= 0;
+ if ( !show )
+ {
+ // make all indices positive for passing them to SetColumnsOrder()
+ idx = ~idx;
+ }
+
+ if ( show != GetColumn(idx).IsShown() )
+ UpdateColumnVisibility(idx, show);
+ }
+
+ UpdateColumnsOrder(order);
+ SetColumnsOrder(order);
+
+ return true;
+ }
+#endif // wxUSE_REARRANGECTRL
+
+ return false;
+}
+