+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;
+}
+