src/TortoiseProc/ResizableColumnsListCtrl.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2016 - TortoiseGit
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 #pragma once
  20 #include "ColumnManager.h"
  21
  22 template <typename BaseType> class CResizableColumnsListCtrl : public BaseType
  23 {
  24 public:
  25         CResizableColumnsListCtrl()
  26                 : BaseType()
  27                 , m_ColumnManager(this)
  28                 , m_bAllowHiding(true)
  29         {}
  30
  31         DECLARE_MESSAGE_MAP()
  32
  33 protected:
  34         void OnDestroy()
  35         {
  36                 SaveColumnWidths();
  37                 BaseType::OnDestroy();
  38         }
  39
  40         void OnHeaderDblClick(NMHDR* pNMHDR, LRESULT* pResult)
  41         {
  42                 m_ColumnManager.OnHeaderDblClick(pNMHDR, pResult);
  43
  44                 *pResult = FALSE;
  45         }
  46
  47         void OnColumnResized(NMHDR* pNMHDR, LRESULT* pResult)
  48         {
  49                 m_ColumnManager.OnColumnResized(pNMHDR, pResult);
  50
  51                 *pResult = FALSE;
  52         }
  53
  54         void OnColumnMoved(NMHDR* pNMHDR, LRESULT* pResult)
  55         {
  56                 m_ColumnManager.OnColumnMoved(pNMHDR, pResult);
  57
  58                 Invalidate(FALSE);
  59         }
  60
  61         void OnContextMenuHeader(CWnd* pWnd, CPoint point)
  62         {
  63                 m_ColumnManager.OnContextMenuHeader(pWnd, point, !!IsGroupViewEnabled());
  64         }
  65
  66         void OnContextMenu(CWnd* pWnd, CPoint point)
  67         {
  68                 if (pWnd == GetHeaderCtrl() && m_bAllowHiding)
  69                         OnContextMenuHeader(pWnd, point);
  70                 else if (pWnd == this && m_ContextMenuHandler)
  71                         m_ContextMenuHandler(point);
  72         }
  73
  74         // prevent users from extending our hidden (size 0) columns
  75         void OnHdnBegintrack(NMHDR* pNMHDR, LRESULT* pResult)
  76         {
  77                 m_ColumnManager.OnHdnBegintrack(pNMHDR, pResult);
  78         }
  79
  80         // prevent any function from extending our hidden (size 0) columns
  81         void OnHdnItemchanging(NMHDR* pNMHDR, LRESULT* pResult)
  82         {
  83                 if (!m_ColumnManager.OnHdnItemchanging(pNMHDR, pResult))
  84                         Default();
  85         }
  86
  87         typedef std::function<void(CPoint point)> ContextMenuHandler;
  88         ContextMenuHandler m_ContextMenuHandler;
  89
  90 public:
  91         void Init()
  92         {
  93                 CRegDWORD regFullRowSelect(L"Software\\TortoiseGit\\FullRowSelect", TRUE);
  94                 DWORD exStyle = LVS_EX_HEADERDRAGDROP;
  95                 if (DWORD(regFullRowSelect))
  96                         exStyle |= LVS_EX_FULLROWSELECT;
  97                 SetExtendedStyle(GetExtendedStyle() | exStyle);
  98         }
  99
 100         void SetListContextMenuHandler(ContextMenuHandler pContextMenuHandler)
 101         {
 102                 m_ContextMenuHandler = pContextMenuHandler;
 103         }
 104
 105         void AdjustColumnWidths()
 106         {
 107                 auto header = GetHeaderCtrl();
 108                 if (!header)
 109                         return;
 110                 int maxcol = header->GetItemCount() - 1;
 111                 for (int col = 0; col <= maxcol; col++)
 112                         SetColumnWidth(col, m_ColumnManager.GetWidth(col, true));
 113         }
 114         virtual void SaveColumnWidths()
 115         {
 116                 auto header = GetHeaderCtrl();
 117                 if (!header)
 118                         return;
 119                 int maxcol = header->GetItemCount() - 1;
 120                 for (int col = 0; col <= maxcol; col++)
 121                         if (m_ColumnManager.IsVisible(col))
 122                                 m_ColumnManager.ColumnResized(col);
 123
 124                 m_ColumnManager.WriteSettings();
 125         }
 126
 127         bool                    m_bAllowHiding;
 128         ColumnManager   m_ColumnManager;
 129 };
 130
 131 BEGIN_TEMPLATE_MESSAGE_MAP(CResizableColumnsListCtrl, BaseType, BaseType)
 132         ON_NOTIFY(HDN_BEGINTRACKA, 0, OnHdnBegintrack)
 133         ON_NOTIFY(HDN_BEGINTRACKW, 0, OnHdnBegintrack)
 134         ON_NOTIFY(HDN_ENDTRACK, 0, OnColumnResized)
 135         ON_NOTIFY(HDN_ENDDRAG, 0, OnColumnMoved)
 136         ON_NOTIFY(HDN_DIVIDERDBLCLICK, 0, OnHeaderDblClick)
 137         ON_NOTIFY(HDN_ITEMCHANGINGA, 0, OnHdnItemchanging)
 138         ON_NOTIFY(HDN_ITEMCHANGINGW, 0, OnHdnItemchanging)
 139         ON_WM_DESTROY()
 140         ON_WM_CONTEXTMENU()
 141 END_MESSAGE_MAP()