xref: /MusicPlayer2/MusicPlayer2/ListCtrlEx.cpp (revision 15db6719b6394c5944e4fa70a85b0947dcb0ae5a)

#include "stdafx.h"
#include "ListCtrlEx.h"
#include "MusicPlayer2.h"

IMPLEMENT_DYNAMIC(CListCtrlEx, CListCtrl)

CListCtrlEx::CListCtrlEx()
    : m_theme_color(theApp.m_app_setting_data.theme_color)
{

    m_drag_cursor = AfxGetApp()->LoadCursor(IDC_DRAG_CURSOR);
}


CListCtrlEx::~CListCtrlEx()
{
    DestroyCursor(m_drag_cursor);
}


//void CListCtrlEx::SetColor(const ColorTable & colors)
//{
//	m_theme_color = colors;
//	if (m_hWnd != NULL)
//		Invalidate();
//}

void CListCtrlEx::GetItemSelected(vector<int>& item_selected) const
{
    item_selected.clear();
    POSITION pos = GetFirstSelectedItemPosition();
    if (pos != NULL)
    {
        while (pos)
        {
            int nItem = GetNextSelectedItem(pos);
            item_selected.push_back(nItem);
        }
    }
}

int CListCtrlEx::GetCurSel() const
{
    //vector<int> item_selected;
    //GetItemSelected(item_selected);
    //if (!item_selected.empty())
    //	return item_selected[0];
    //else
    //	return -1;
    return GetSelectionMark();
}

void CListCtrlEx::SetCurSel(int select)
{
    int size = GetItemCount();
    for (int i{}; i < size; i++)
    {
        if (i == select)
        {
            SetItemState(select, LVIS_FOCUSED | LVIS_SELECTED, LVIS_FOCUSED | LVIS_SELECTED);	//选中行
            EnsureVisible(select, FALSE);		//使选中行保持可见
        }
        else
        {
            SetItemState(i, 0, LVIS_SELECTED);      //取消选中其他行
        }
    }
}

void CListCtrlEx::SetCurSel(int first, int last)
{
    int itemCnt = GetItemCount();
    if (first >= 0 && last < itemCnt && first <= last)
    {
        for (int i = 0; i < itemCnt; i++)
        {
            if(i>=first && i<=last)
                SetItemState(i, LVIS_FOCUSED | LVIS_SELECTED, LVIS_FOCUSED | LVIS_SELECTED);
            else
                SetItemState(i, 0, LVIS_SELECTED);
        }
    }
}

void CListCtrlEx::SetCurSel(const vector<int> indexes)
{
    int itemCnt = GetItemCount();
    for (int i = 0; i < itemCnt; i++)
    {
        auto iter = std::find(indexes.begin(), indexes.end(), i);
        if (iter != indexes.end())
            SetItemState(i, LVIS_FOCUSED | LVIS_SELECTED, LVIS_FOCUSED | LVIS_SELECTED);
        else
            SetItemState(i, 0, LVIS_SELECTED);
    }
}

void CListCtrlEx::SelectAll()
{
    int itemCnt = GetItemCount();
    for (int i = 0; i < itemCnt; i++)
    {
        SetItemState(i, LVIS_FOCUSED | LVIS_SELECTED, LVIS_FOCUSED | LVIS_SELECTED);
    }
}

void CListCtrlEx::SelectNone()
{
    int itemCnt = GetItemCount();
    for (int i = 0; i < itemCnt; i++)
    {
        SetItemState(i, 0, LVIS_SELECTED);
    }
}

void CListCtrlEx::SelectReverse()
{
    std::vector<int> selected_vect;
    GetItemSelected(selected_vect);
    std::set<int> selected_set;
    for (auto n : selected_vect)
        selected_set.insert(n);
    int size = GetItemCount();
    for (int i = 0; i < size; i++)
    {
        if(selected_set.find(i)==selected_set.end())
            SetItemState(i, LVIS_FOCUSED | LVIS_SELECTED, LVIS_FOCUSED | LVIS_SELECTED);
        else
            SetItemState(i, 0, LVIS_SELECTED);
    }
}

bool CListCtrlEx::SetRowHeight(int height, int left_space)
{
    if (height > 0 && height <= 512)
    {
        CImageList imgList;		//为ClistCtrl设置一个图像列表，以设置行高
        int width = left_space;
        if (width < 1)
            width = 1;
        BOOL rtn = imgList.Create(width, height, ILC_COLOR, 1, 1);
        if (rtn != FALSE)
        {
            m_row_height = height;
            SetImageList(&imgList, LVSIL_SMALL);
            return true;
        }
    }
    return false;
}


void CListCtrlEx::ShowPopupMenu(CMenu* pMenu, int item_index, CWnd* pWnd)
{
    CPoint point;			//定义一个用于确定光标位置的位置
    GetCursorPos(&point);	//获取当前光标的位置，以便使得菜单可以跟随光标
    if(item_index >= 0)
    {
        CRect item_rect;
        GetItemRect(item_index, item_rect, LVIR_BOUNDS);		//获取选中项目的矩形区域（以列表控件左上角为原点）
        CRect window_rect;
        GetWindowRect(window_rect);		//获取列表控件的矩形区域（以屏幕左上角为原点）
        point.y = window_rect.top + item_rect.bottom;	//设置鼠标要弹出的y坐标为选中项目的下边框位置，防止右键菜单挡住选中的项目
    }
    pMenu->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, point.x, point.y, pWnd); //在指定位置显示弹出菜单
}

void CListCtrlEx::FillLeftSpaceAfterPaint(bool fill)
{
    m_fill_left_space_after_paint = fill;
}

void CListCtrlEx::SetMouseWheelEnable(bool enable)
{
    m_mouse_wheel_enable = enable;
}

void CListCtrlEx::SetItemIcon(int item, HICON icon)
{
    m_icons[item] = icon;
    m_fill_left_space_after_paint = false;  //设置了图标后将m_fill_left_space_after_paint置为false，以确保图标部分背景显示正常
}

bool CListCtrlEx::DeleteItem(int nItem)
{
    bool rtn = (CListCtrl::DeleteItem(nItem) != FALSE);

    if (rtn)
    {
        //删除对应的图标
        m_icons.erase(nItem);

        //删除项目后要重新调整删除项目后面所有图标的序号
        for (int i = nItem + 1; i <= GetItemCount(); i++)
        {
            auto iter = m_icons.find(i);
            if (iter != m_icons.end())
            {
                HICON icon = iter->second;
                m_icons.erase(iter);
                m_icons[i - 1] = icon;
            }
        }

        Invalidate();       //控件重绘
    }

    return rtn;
}

void CListCtrlEx::SetListData(ListData* pListData)
{
    if (pListData == nullptr)
        return;
    m_pListData = pListData;
    SetItemCount(pListData->size());
}

void CListCtrlEx::SetListData(const ListData & list_data)
{
    m_pListData = nullptr;
    int item_num_before = GetItemCount();
    int item_num_after = list_data.size();
    //如果当前列表中项目的数量小于原来的，则直接清空原来列表中所有的项目，重新添加
    if (item_num_after < item_num_before)
    {
        DeleteAllItems();
        item_num_before = 0;
    }
    for (int i{}; i < item_num_after; i++)
    {
        const RowData& data_row = list_data[i];
        if (i >= item_num_before)	//如果当前列表中的项目数量大于之前的数量，则需要在不够时插入新的项目
        {
            auto iter = data_row.find(0);
            if (iter != data_row.end())
                InsertItem(i, iter->second.c_str());
            else
                InsertItem(i, _T(""));
        }
        for (const auto& item : data_row)
        {
            SetItemText(i, item.first, item.second.c_str());
        }
    }
}

wstring CListCtrlEx::GetAllText(const wchar_t* sperator /*= L"\t"*/)
{
    wstring str_result;
    int item_count = GetItemCount();
    int col_count{};
    auto pHeader = GetHeaderCtrl();
    if (pHeader != nullptr)
        col_count = pHeader->GetItemCount();
    for (int i = 0; i < item_count; i++)
    {
        for (int j = 0; j < col_count; j++)
        {
            str_result += GetItemText(i, j).GetString();
            if (j < col_count - 1)
                str_result += sperator;
        }
        str_result += L"\r\n";
    }
    return str_result;
}

BEGIN_MESSAGE_MAP(CListCtrlEx, CListCtrl)
    ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, &CListCtrlEx::OnNMCustomdraw)
    ON_WM_LBUTTONDOWN()
    ON_WM_RBUTTONDOWN()
    ON_WM_SETFOCUS()
    ON_NOTIFY_REFLECT(LVN_BEGINDRAG, &CListCtrlEx::OnLvnBegindrag)
    ON_WM_LBUTTONUP()
    ON_WM_SETCURSOR()
    ON_WM_ERASEBKGND()
    ON_NOTIFY_REFLECT(LVN_GETDISPINFO, &CListCtrlEx::OnLvnGetdispinfo)
    ON_MESSAGE(WM_TABLET_QUERYSYSTEMGESTURESTATUS, &CListCtrlEx::OnTabletQuerysystemgesturestatus)
END_MESSAGE_MAP()


void CListCtrlEx::OnNMCustomdraw(NMHDR *pNMHDR, LRESULT *pResult)
{
    DWORD style = GetExtendedStyle();


    *pResult = CDRF_DODEFAULT;
    LPNMLVCUSTOMDRAW lplvdr = reinterpret_cast<LPNMLVCUSTOMDRAW>(pNMHDR);
    NMCUSTOMDRAW& nmcd = lplvdr->nmcd;
    static bool this_item_select = false;
    switch (lplvdr->nmcd.dwDrawStage)	//判断状态
    {
    case CDDS_PREPAINT:
        *pResult = CDRF_NOTIFYITEMDRAW;
        break;
    case CDDS_ITEMPREPAINT:			//如果为画ITEM之前就要进行颜色的改变
        this_item_select = false;
        if (IsWindowEnabled())
        {
            //当选中行又是高亮行时设置颜色
            if (GetItemState(nmcd.dwItemSpec, LVIS_SELECTED) == LVIS_SELECTED && nmcd.dwItemSpec == m_highlight_item)
            {
                this_item_select = true;
                //SetItemState(nmcd.dwItemSpec, 0, LVIS_SELECTED);
                lplvdr->clrText = m_theme_color.light3;
                lplvdr->clrTextBk = m_theme_color.dark1;
            }
            //设置选中行的颜色
            else if (GetItemState(nmcd.dwItemSpec, LVIS_SELECTED) == LVIS_SELECTED)
            {
                this_item_select = true;
                //注：当使用虚拟列表时，如果取消下面一行的注释，会导致当列表选中行处理可见状态时，窗口刷新不正常，甚至主窗口OnTimer也无法响应，原因暂时不明
                //SetItemState(nmcd.dwItemSpec, 0, LVIS_SELECTED);
                lplvdr->clrText = m_theme_color.dark3;
                lplvdr->clrTextBk = m_theme_color.light2;
            }
            //设置高亮行的颜色
            else if (nmcd.dwItemSpec == m_highlight_item)
            {
                lplvdr->clrText = m_theme_color.dark2;
                //lplvdr->clrText = 0;
                lplvdr->clrTextBk = m_theme_color.light3;
            }
            //设置偶数行的颜色
            else if (nmcd.dwItemSpec % 2 == 0)
            {
                lplvdr->clrText = CColorConvert::m_gray_color.dark3;
                lplvdr->clrTextBk = CColorConvert::m_gray_color.light3;
            }
            //设置奇数行的颜色
            else
            {
                lplvdr->clrText = CColorConvert::m_gray_color.dark3;
                lplvdr->clrTextBk = CColorConvert::m_gray_color.light4;
            }

            //用背景色填充单元格，以去掉每行前面的空白
            if(!m_fill_left_space_after_paint)
            {
                CRect rect = nmcd.rc;
                CDC* pDC = CDC::FromHandle(nmcd.hdc);		//获取绘图DC
                pDC->FillSolidRect(rect, lplvdr->clrTextBk);
            }
        }
        else		//当控件被禁用时，显示文本设为灰色
        {
            lplvdr->clrText = GRAY(140);
            lplvdr->clrTextBk = GRAY(240);
        }
        *pResult = CDRF_NOTIFYPOSTPAINT;
        break;
    case CDDS_ITEMPOSTPAINT:
        if (this_item_select)
            SetItemState(nmcd.dwItemSpec, 0xFF, LVIS_SELECTED);
        //用背景色填充单元格左侧的空白区域
        if(m_fill_left_space_after_paint)
        {
            CRect rect = nmcd.rc;
            rect.right = rect.left + 5;
            CDC* pDC = CDC::FromHandle(nmcd.hdc);		//获取绘图DC
            pDC->FillSolidRect(rect, lplvdr->clrTextBk);
        }
        //绘制图标
        CRect icon_rect = nmcd.rc;
        if (!icon_rect.IsRectEmpty())
        {
            auto iter = m_icons.find(nmcd.dwItemSpec);
            if (iter != m_icons.end())
            {
                icon_rect.left += 5;
                const int icon_size{ theApp.DPI(16) };
                icon_rect.right = icon_rect.left + icon_size;
                icon_rect.top = icon_rect.top + (icon_rect.Height() - icon_size) / 2;
                icon_rect.bottom = icon_rect.top + icon_size;
                CDC* pDC = CDC::FromHandle(nmcd.hdc);
                CDrawCommon drawer;
                drawer.Create(pDC);
                drawer.DrawIcon(iter->second, icon_rect);
            }
        }

        //*pResult = CDRF_DODEFAULT;
        break;
    }
}


void CListCtrlEx::PreSubclassWindow()
{
    // TODO: 在此添加专用代码和/或调用基类
    CListCtrl::PreSubclassWindow();

    DWORD style = GetExtendedStyle();
    SetExtendedStyle(style | LVS_EX_DOUBLEBUFFER | LVS_EX_FULLROWSELECT | LVS_EX_LABELTIP);      //设置扩展样式（双缓冲绘图、整行选中，鼠标提示）

    SetBkColor(m_background_color);
    //SetHightItem(-1);
    SetRowHeight(theApp.DPI(22));
}


void CListCtrlEx::OnLButtonDown(UINT nFlags, CPoint point)
{
    // TODO: 在此添加消息处理程序代码和/或调用默认值
    this->SetFocus();
    m_dragging = false;
    CListCtrl::OnLButtonDown(nFlags, point);
}


void CListCtrlEx::OnRButtonDown(UINT nFlags, CPoint point)
{
    // TODO: 在此添加消息处理程序代码和/或调用默认值
    this->SetFocus();
    m_dragging = false;
    CListCtrl::OnRButtonDown(nFlags, point);
}


BOOL CListCtrlEx::PreTranslateMessage(MSG* pMsg)
{
    // TODO: 在此添加专用代码和/或调用基类

    //if (pMsg->message == WM_KEYDOWN || pMsg->message == WM_CHAR)		//屏蔽列表控件的键盘消息，防止每次按下一个键时列表选中行会出现讨厌的、难看的虚线框
    //	return TRUE;

    // 按Ctrl+A全选，向父窗口发送选中项更新消息
    if(m_enable_ctrl_a)
    {
        if ((GetKeyState(VK_CONTROL) & 0x80) && (pMsg->wParam == 'A'))
        {
            SelectAll();
            CWnd* pParent{ GetParent() };
            if (pParent != nullptr)
            {
                pParent->SendMessage(WM_COMMAND, ID_PLAYLIST_SELECT_CHANGE);
            }
            return TRUE;
        }
    }

    if (pMsg->message == WM_MOUSEWHEEL && !m_mouse_wheel_enable)
    {
        CRect rect;
        GetWindowRect(rect);
        //根据行高*行数小于控件高度来判断是否显示垂直滚动条
        if (m_row_height > 0 && (m_row_height + 1) * GetItemCount() < rect.Height())
        {
            //如果不显示滚动条，则将鼠标滚轮消息发送给父窗口
            CWnd* pParent = GetParent();
            pParent->SendMessage(WM_MOUSEWHEEL, pMsg->wParam, pMsg->lParam);
            return TRUE;
        }
    }

    return CListCtrl::PreTranslateMessage(pMsg);
}


void CListCtrlEx::OnSetFocus(CWnd* pOldWnd)
{
    CListCtrl::OnSetFocus(pOldWnd);

    // TODO: 在此处添加消息处理程序代码

    SendMessage(WM_KILLFOCUS);				//禁止列表控件获取焦点，防止选中行会出现难看的虚线框
}


void CListCtrlEx::OnLvnBegindrag(NMHDR *pNMHDR, LRESULT *pResult)
{
    LPNMLISTVIEW pNMLV = reinterpret_cast<LPNMLISTVIEW>(pNMHDR);
    // TODO: 在此添加控件通知处理程序代码
    if (m_drag_enable)
    {
        m_dragging = true;
    }

    *pResult = 0;
}


void CListCtrlEx::OnLButtonUp(UINT nFlags, CPoint point)
{
    // TODO: 在此添加消息处理程序代码和/或调用默认值
    if (m_dragging)
    {
        CPoint pt(point);
        int drop_index = this->HitTest(pt);     //鼠标松开时的项目序号
        CWnd* pParent{ GetParent() };
        if (pParent != nullptr)
        {
            pParent->SendMessage(WM_LIST_ITEM_DRAGGED, drop_index, 0);       //结束拖放时向父窗口发送消息，传递拖放结束位置索引
        }
        m_dragging = false;
        SendMessage(WM_SETCURSOR);      //鼠标松开时刷新光标
    }

    CListCtrl::OnLButtonUp(nFlags, point);
}


BOOL CListCtrlEx::OnSetCursor(CWnd* pWnd, UINT nHitTest, UINT message)
{
    // TODO: 在此添加消息处理程序代码和/或调用默认值
    if (m_dragging)
    {
        ::SetCursor(m_drag_cursor);
        return TRUE;
    }

    return CListCtrl::OnSetCursor(pWnd, nHitTest, message);
}


BOOL CListCtrlEx::OnEraseBkgnd(CDC* pDC)
{
    // TODO: 在此添加消息处理程序代码和/或调用默认值

    //return CListCtrl::OnEraseBkgnd(pDC);
    return TRUE;
}


void CListCtrlEx::OnLvnGetdispinfo(NMHDR *pNMHDR, LRESULT *pResult)
{
    NMLVDISPINFO *pDispInfo = reinterpret_cast<NMLVDISPINFO*>(pNMHDR);
    // TODO: 在此添加控件通知处理程序代码
    if (pDispInfo->hdr.hwndFrom == m_hWnd && m_pListData != nullptr)
    {
        if (pDispInfo->item.iItem >= 0 && pDispInfo->item.iItem < static_cast<int>(m_pListData->size()))
        {
            auto& row_data{ m_pListData->at(pDispInfo->item.iItem) };
            auto iter = row_data.find(pDispInfo->item.iSubItem);
            if (iter != row_data.end())
                CCommon::WStringCopy(pDispInfo->item.pszText, pDispInfo->item.cchTextMax, iter->second.c_str());
        }
    }
    *pResult = 0;
}


afx_msg LRESULT CListCtrlEx::OnTabletQuerysystemgesturestatus(WPARAM wParam, LPARAM lParam)
{
    return 0;
}