一、使用QTreeWidget
头文件：
/************************************************************************
树的搜索类（QTreeWidget版）
使用方法：
调用函数TreeSearch::SearchItem(para1, para2);
第一个参数是要搜索的树QTreeWidget或它的子类
第二个参数是搜索输入的关键字
************************************************************************/
#ifndef TREESEARCH_H
#define TREESEARCH_H
#include <QTreeWidget>
#include <Windows.h>
#define ITEM_SHOWN (Qt::UserRole + 17)
class TreeSearch
{
public:
TreeSearch();
~TreeSearch();
static void SearchItem(QTreeWidget *, const QString &);//搜索函数
private:
static void FuzzySearch(QString &); //模糊搜索
static void FuzzySearchChildren(QTreeWidgetItem *, QString &); //模糊搜索子节点（递归）
static void PreciseSearch(const QString &); //精确搜索
static void PreciseSearchChildren(QTreeWidgetItem *, const QString &);//精确搜索子节点（递归）
static void ShowTotalItem(); //显示所有节点
static void ShowTotalChildrenItem(QTreeWidgetItem *); //显示所有子节点（递归）
static void ShowTotalParentItem(QTreeWidgetItem *); //设置所有父节点可显示（递归）
static void GetPinYin(const QString &, QString &strInitial, QString &strQuanPin); //将汉字转为拼音（首字母以及全拼）
static void GetWString(const std::string &, std::wstring &wStrOut);//将string转为wstring
private:
static QTreeWidget *m_pCurTreeWidget; //当前的树
};
#endif // TREESEARCH_H
实现：
#include "TreeSearch.h"
QTreeWidget *TreeSearch::m_pCurTreeWidget;
TreeSearch::TreeSearch()
{
}
TreeSearch::~TreeSearch()
{
}
void TreeSearch::SearchItem(QTreeWidget *pTreeWidget, const QString &strKeyWord)
{
m_pCurTreeWidget = pTreeWidget;
if (m_pCurTreeWidget == NULL)
{
return;
}
QRegExp rx("([a-z]?[0-9]?)*");
QString strLower = strKeyWord.toLower().trimmed();
if (!strLower.isEmpty())
{
bool bIsPreciseSearch = true;
QRegExpValidator v(rx, 0);
int pos = 0;
if (v.validate(strLower, pos) == QValidator::Acceptable)
{
bIsPreciseSearch = false;
}
if (bIsPreciseSearch)
{
PreciseSearch(strLower);
}
else
{
FuzzySearch(strLower);
}
}
//搜索框为空，显示全部节点
else
{
ShowTotalItem();
}
}
void TreeSearch::FuzzySearch(QString &strKeyWord)
{
//QString strInitialOfInput; //输入
int iRootCount = m_pCurTreeWidget->topLevelItemCount();
for (int i = 0; i < iRootCount; ++i)
{
QTreeWidgetItem *pCurRootItem = m_pCurTreeWidget->topLevelItem(i);
if (pCurRootItem == NULL)
{
continue;
}
pCurRootItem->setData(0, ITEM_SHOWN, false);
QString strCurRootItem = pCurRootItem->text(0);
QString strItemInitial; //当前节点文本的拼音首字母
QString strItemQuanPin; //当前节点文本的全拼
GetPinYin(strCurRootItem, strItemInitial, strItemQuanPin);
QString strStart = "^";
QRegExp rx(strStart.append(strItemQuanPin));
QRegExpValidator v(rx, 0);
int pos = 0;
//如果输入的关键字匹配当前节点，则显示当前节点及其所有子节点
if (strItemInitial.contains(strKeyWord) || v.validate(strKeyWord, pos) != QValidator::Invalid)
{
pCurRootItem->setHidden(false);
ShowTotalChildrenItem(pCurRootItem);
continue;
}
//否则继续在其子节点中搜索
else
{
FuzzySearchChildren(pCurRootItem, strKeyWord);
}
}
}
void TreeSearch::FuzzySearchChildren(QTreeWidgetItem *pParentItem, QString &strKeyWord)
{
int iRootCount = pParentItem->childCount();
for (int i = 0; i < iRootCount; ++i)
{
QTreeWidgetItem *pCurChildItem = pParentItem->child(i);
if (pCurChildItem == NULL)
{
continue;
}
pCurChildItem->setData(0, ITEM_SHOWN, false);
QString strCurChildItem = pCurChildItem->text(0);
QString strItemInitial; //当前节点文本的拼音首字母
QString strItemQuanPin; //当前节点文本的全拼
GetPinYin(strCurChildItem, strItemInitial, strItemQuanPin);
QString strStart = "^";
QRegExp rx(strStart.append(strItemQuanPin));
QRegExpValidator v(rx, 0);
int pos = 0;
//如果输入的关键字匹配当前节点，则显示当前节点及其所有子节点
if (strItemInitial.contains(strKeyWord) || v.validate(strKeyWord, pos) != QValidator::Invalid)
{
pCurChildItem->setHidden(false);
ShowTotalParentItem(pCurChildItem);
ShowTotalChildrenItem(pCurChildItem);
continue;
}
//否则继续在其子节点中搜索
else
{
FuzzySearchChildren(pCurChildItem, strKeyWord);
}
}
bool bHasChildren = pParentItem->data(0, ITEM_SHOWN).toBool();
pParentItem->setHidden(!bHasChildren);
}
void TreeSearch::PreciseSearch(const QString &strKeyWord)
{
//QString strInitialOfInput; //输入
int iRootCount = m_pCurTreeWidget->topLevelItemCount();
for (int i = 0; i < iRootCount; ++i)
{
QTreeWidgetItem *pCurRootItem = m_pCurTreeWidget->topLevelItem(i);
if (pCurRootItem == NULL)
{
continue;
}
pCurRootItem->setData(0, ITEM_SHOWN, false);
QString strCurRootItem = pCurRootItem->text(0);
//如果输入的关键字匹配当前节点，则显示当前节点及其所有子节点
if (strCurRootItem.contains(strKeyWord))
{
pCurRootItem->setHidden(false);
ShowTotalChildrenItem(pCurRootItem);
continue;
}
//否则继续在其子节点中搜索
else
{
PreciseSearchChildren(pCurRootItem, strKeyWord);
}
}
}
void TreeSearch::PreciseSearchChildren(QTreeWidgetItem *pParentItem, const QString &strKeyWord)
{
int iRootCount = pParentItem->childCount();
for (int i = 0; i < iRootCount; ++i)
{
QTreeWidgetItem *pCurChildItem = pParentItem->child(i);
if (pCurChildItem == NULL)
{
continue;
}
pCurChildItem->setData(0, ITEM_SHOWN, false);
QString strCurChildItem = pCurChildItem->text(0);
//如果输入的关键字匹配当前节点，则显示当前节点及其所有子节点
if (strCurChildItem.contains(strKeyWord))
{
pCurChildItem->setHidden(false);
ShowTotalParentItem(pCurChildItem);
ShowTotalChildrenItem(pCurChildItem);
continue;
}
//否则继续在其子节点中搜索
else
{
PreciseSearchChildren(pCurChildItem, strKeyWord);
}
}
bool bHasChildren = pParentItem->data(0, ITEM_SHOWN).toBool();
pParentItem->setHidden(!bHasChildren);
}
void TreeSearch::ShowTotalItem()
{
int iRootCount = m_pCurTreeWidget->topLevelItemCount();
for (int i = 0; i < iRootCount; ++i)
{
QTreeWidgetItem *pRootItem = m_pCurTreeWidget->topLevelItem(i);
if (pRootItem == NULL)
{
continue;
}
pRootItem->setHidden(false);
ShowTotalChildrenItem(pRootItem);
}
}
void TreeSearch::ShowTotalChildrenItem(QTreeWidgetItem *pParentItem)
{
int iChildCount = pParentItem->childCount();
for (int i = 0; i < iChildCount; ++i)
{
QTreeWidgetItem *pChildItem = pParentItem->child(i);
if (pChildItem == NULL)
{
continue;
}
pChildItem->setHidden(false);
ShowTotalChildrenItem(pChildItem);
}
}
void TreeSearch::ShowTotalParentItem(QTreeWidgetItem *pChildItem)
{
QTreeWidgetItem *pParentItem = pChildItem->parent();
if (pParentItem != NULL)
{
pParentItem->setData(0, ITEM_SHOWN, true);
ShowTotalParentItem(pParentItem);
}
}
void TreeSearch::GetPinYin(const QString &strHanZi, QString &strInitial, QString &strQuanPin)
{
//保存拼音的ASCII值
const int pyAscii[] = { -20319, -20317, -20304, -20295, -20292, -20283, -20265, -20257, -20242, -20230, -20051, -20036, -20032, -20026,
-20002, -19990, -19986, -19982, -19976, -19805, -19784, -19775, -19774, -19763, -19756, -19751, -19746, -19741, -19739, -19728,
-19725, -19715, -19540, -19531, -19525, -19515, -19500, -19484, -19479, -19467, -19289, -19288, -19281, -19275, -19270, -19263,
-19261, -19249, -19243, -19242, -19238, -19235, -19227, -19224, -19218, -19212, -19038, -19023, -19018, -19006, -19003, -18996,
-18977, -18961, -18952, -18783, -18774, -18773, -18763, -18756, -18741, -18735, -18731, -18722, -18710, -18697, -18696, -18526,
-18518, -18501, -18490, -18478, -18463, -18448, -18447, -18446, -18239, -18237, -18231, -18220, -18211, -18201, -18184, -18183,
-18181, -18012, -17997, -17988, -17970, -17964, -17961, -17950, -17947, -17931, -17928, -17922, -17759, -17752, -17733, -17730,
-17721, -17703, -17701, -17697, -17692, -17683, -17676, -17496, -17487, -17482, -17468, -17454, -17433, -17427, -17417, -17202,
-17185, -16983, -16970, -16942, -16915, -16733, -16708, -16706, -16689, -16664, -16657, -16647, -16474, -16470, -16465, -16459,
-16452, -16448, -16433, -16429, -16427, -16423, -16419, -16412, -16407, -16403, -16401, -16393, -16220, -16216, -16212, -16205,
-16202, -16187, -16180, -16171, -16169, -16158, -16155, -15959, -15958, -15944, -15933, -15920, -15915, -15903, -15889, -15878,
-15707, -15701, -15681, -15667, -15661, -15659, -15652, -15640, -15631, -15625, -15454, -15448, -15436, -15435, -15419, -15416,
-15408, -15394, -15385, -15377, -15375, -15369, -15363, -15362, -15183, -15180, -15165, -15158, -15153, -15150, -15149, -15144,
-15143, -15141, -15140, -15139, -15128, -15121, -15119, -15117, -15110, -15109, -14941, -14937, -14933, -14930, -14929, -14928,
-14926, -14922, -14921, -14914, -14908, -14902, -14894, -14889, -14882, -14873, -14871, -14857, -14678, -14674, -14670, -14668,
-14663, -14654, -14645, -14630, -14594, -14429, -14407, -14399, -14384, -14379, -14368, -14355, -14353, -14345, -14170, -14159,
-14151, -14149, -14145, -14140, -14137, -14135, -14125, -14123, -14122, -14112, -14109, -14099, -14097, -14094, -14092, -14090,
-14087, -14083, -13917, -13914, -13910, -13907, -13906, -13905, -13896, -13894, -13878, -13870, -13859, -13847, -13831, -13658,
-13611, -13601, -13406, -13404, -13400, -13398, -13395, -13391, -13387, -13383, -13367, -13359, -13356, -13343, -13340, -13329,
-13326, -13318, -13147, -13138, -13120, -13107, -13096, -13095, -13091, -13076, -13068, -13063, -13060, -12888, -12875, -12871,
-12860, -12858, -12852, -12849, -12838, -12831, -12829, -12812, -12802, -12607, -12597, -12594, -12585, -12556, -12359, -12346,
-12320, -12300, -12120, -12099, -12089, -12074, -12067, -12058, -12039, -11867, -11861, -11847, -11831, -11798, -11781, -11604,
-11589, -11536, -11358, -11340, -11339, -11324, -11303, -11097, -11077, -11067, -11055, -11052, -11045, -11041, -11038, -11024,
-11020, -11019, -11018, -11014, -10838, -10832, -10815, -10800, -10790, -10780, -10764, -10587, -10544, -10533, -10519, -10331,
-10329, -10328, -10322, -10315, -10309, -10307, -10296, -10281, -10274, -10270, -10262, -10260, -10256, -10254 };
//保存拼音的音节
const char pyStr[396][7] = { "a", "ai", "an", "ang", "ao", "ba", "bai", "ban", "bang", "bao", "bei", "ben", "beng", "bi", "bian", "biao",
"bie", "bin", "bing", "bo", "bu", "ca", "cai", "can", "cang", "cao", "ce", "ceng", "cha", "chai", "chan", "chang", "chao", "che", "chen",
"cheng", "chi", "chong", "chou", "chu", "chuai", "chuan", "chuang", "chui", "chun", "chuo", "ci", "cong", "cou", "cu", "cuan", "cui",
"cun", "cuo", "da", "dai", "dan", "dang", "dao", "de", "deng", "di", "dian", "diao", "die", "ding", "diu", "dong", "dou", "du", "duan",
"dui", "dun", "duo", "e", "en", "er", "fa", "fan", "fang", "fei", "fen", "feng", "fo", "fou", "fu", "ga", "gai", "gan", "gang", "gao",
"ge", "gei", "gen", "geng", "gong", "gou", "gu", "gua", "guai", "guan", "guang", "gui", "gun", "guo", "ha", "hai", "han", "hang",
"hao", "he", "hei", "hen", "heng", "hong", "hou", "hu", "hua", "huai", "huan", "huang", "hui", "hun", "huo", "ji", "jia", "jian",
"jiang", "jiao", "jie", "jin", "jing", "jiong", "jiu", "ju", "juan", "jue", "jun", "ka", "kai", "kan", "kang", "kao", "ke", "ken",
"keng", "kong", "kou", "ku", "kua", "kuai", "kuan", "kuang", "kui", "kun", "kuo", "la", "lai", "lan", "lang", "lao", "le", "lei",
"leng", "li", "lia", "lian", "liang", "liao", "lie", "lin", "ling", "liu", "long", "lou", "lu", "lv", "luan", "lue", "lun", "luo",
"ma", "mai", "man", "mang", "mao", "me", "mei", "men", "meng", "mi", "mian", "miao", "mie", "min", "ming", "miu", "mo", "mou", "mu",
"na", "nai", "nan", "nang", "nao", "ne", "nei", "nen", "neng", "ni", "nian", "niang", "niao", "nie", "nin", "ning", "niu", "nong",
"nu", "nv", "nuan", "nue", "nuo", "o", "ou", "pa", "pai", "pan", "pang", "pao", "pei", "pen", "peng", "pi", "pian", "piao", "pie",
"pin", "ping", "po", "pu", "qi", "qia", "qian", "qiang", "qiao", "qie", "qin", "qing", "qiong", "qiu", "qu", "quan", "que", "qun",
"ran", "rang", "rao", "re", "ren", "reng", "ri", "rong", "rou", "ru", "ruan", "rui", "run", "ruo", "sa", "sai", "san", "sang",
"sao", "se", "sen", "seng", "sha", "shai", "shan", "shang", "shao", "she", "shen", "sheng", "shi", "shou", "shu", "shua",
"shuai", "shuan", "shuang", "shui", "shun", "shuo", "si", "song", "sou", "su", "suan", "sui", "sun", "suo", "ta", "tai",
"tan", "tang", "tao", "te", "teng", "ti", "tian", "tiao", "tie", "ting", "tong", "tou", "tu", "tuan", "tui", "tun", "tuo",
"wa", "wai", "wan", "wang", "wei", "wen", "weng", "wo", "wu", "xi", "xia", "xian", "xiang", "xiao", "xie", "xin", "xing",
"xiong", "xiu", "xu", "xuan", "xue", "xun", "ya", "yan", "yang", "yao", "ye", "yi", "yin", "ying", "yo", "yong", "you",
"yu", "yuan", "yue", "yun", "za", "zai", "zan", "zang", "zao", "ze", "zei", "zen", "zeng", "zha", "zhai", "zhan", "zhang",
"zhao", "zhe", "zhen", "zheng", "zhi", "zhong", "zhou", "zhu", "zhua", "zhuai", "zhuan", "zhuang", "zhui", "zhun", "zhuo",
"zi", "zong", "zou", "zu", "zuan", "zui", "zun", "zuo" };
int iChrasc = 0;
char *ret = NULL;
std::wstring wStr;
GetWString(std::string(strHanZi.toLower().toLocal8Bit()), wStr);
int iLength = WideCharToMultiByte(CP_ACP, 0, wStr.c_str(), -1, 0, 0, 0, 0);
char *nowChar = new char[iLength + 1];
memset(nowChar, 0, iLength + 1);
WideCharToMultiByte(CP_ACP, 0, wStr.c_str(), -1, nowChar, iLength, 0, 0);
//转换ASCII
int offset = 0;
char *returnstr = new char[6 * iLength + 1];
memset(returnstr, 0, 6 * iLength + 1);
strInitial.clear();
strQuanPin.clear();
for (int j = 0; j < iLength;)
{
if (nowChar[j] == 0)
{
break;
}
if (nowChar[j] >= 0 && nowChar[j] < 128)
{
returnstr[offset] = nowChar[j];
strInitial.append(nowChar[j]);
strInitial = strInitial.trimmed();
++offset;
++j;
continue;
}
// 汉字处理
iChrasc = nowChar[j] * 256 + nowChar[j + 1] + 256;
if (iChrasc > 0 && iChrasc < 160)
{
returnstr[offset] = nowChar[j];
strInitial.append(nowChar[j]);
strInitial = strInitial.trimmed();
++offset;
++j;
}
else
{
for (int i = (sizeof(pyAscii) / sizeof(pyAscii[0]) - 1); i >= 0; --i)
{
if (i == 396)
{
continue;
}
if (pyAscii[i] <= iChrasc)
{
strcpy(returnstr + offset, pyStr[i]);
strInitial.append(QChar(pyStr[i][0]));
strInitial = strInitial.trimmed();
offset += strlen(pyStr[i]);
break;
}
}
j += 2;
}
}
if (strlen(returnstr) > 0)
{
ret = new char[strlen(returnstr) + 1];
memset(ret, 0, strlen(returnstr) + 1);
strcpy(ret, returnstr);
}
delete[] returnstr;
delete[] nowChar;
strQuanPin = ret;
delete[] ret;
}
void TreeSearch::GetWString(const std::string &strIn, std::wstring &wStrOut)
{
int iLength = MultiByteToWideChar(CP_ACP, 0, (LPCSTR)strIn.c_str(), -1, NULL, 0);
wchar_t *wszUtf8 = new wchar_t[iLength + 1];
memset(wszUtf8, 0, iLength * 2 + 2);
MultiByteToWideChar(CP_ACP, 0, (LPCSTR)strIn.c_str(), -1, (LPWSTR)wszUtf8, iLength);
wStrOut = wszUtf8;
delete[] wszUtf8;
}
二、QTreeView + QStandardItemModel
头文件：
/************************************************************************
树的搜索类（QTreeView + QStandardItemModel版）
使用方法：
调用函数TreeSearch::SearchItem(para1, para2, para3);
第一个参数是要搜索的树QTreeView或它的子类
第二个参数是树的数据模型QStandardItemModel或它的子类
第三个参数是搜索输入的关键字
************************************************************************/
#ifndef TREESEARCH_H
#define TREESEARCH_H
#include <QTreeView>
#include <QStandardItemModel>
#include <Windows.h>
#define ITEM_SHOWN (Qt::UserRole + 17)
class TreeSearch
{
public:
TreeSearch();
~TreeSearch();
static void SearchItem(QTreeView *, QStandardItemModel *, const QString &);//搜索函数
private:
static void FuzzySearch(QString &); //模糊搜索
static void FuzzySearchChildren(QStandardItem *, QString &); //模糊搜索子节点（递归）
static void PreciseSearch(const QString &); //精确搜索
static void PreciseSearchChildren(QStandardItem *, const QString &);//精确搜索子节点（递归）
static void ShowTotalItem(); //显示所有节点
static void ShowTotalChildrenItem(QStandardItem *); //显示所有子节点（递归）
static void ShowTotalParentItem(QStandardItem *); //设置所有父节点可显示（递归）
static void GetPinYin(const QString &, QString &strInitial, QString &strQuanPin); //将汉字转为拼音（首字母以及全拼）
static void GetWString(const std::string &, std::wstring &wStrOut);//将string转为wstring
private:
static QTreeView *m_pCurTreeView; //当前的树
static QStandardItemModel *m_pCurModel; //当前树的数据模型
};
#endif // TREESEARCH_H
实现：
#include "TreeSearch.h"
QTreeView *TreeSearch::m_pCurTreeView;
QStandardItemModel *TreeSearch::m_pCurModel;
TreeSearch::TreeSearch()
{
}
TreeSearch::~TreeSearch()
{
}
void TreeSearch::SearchItem(QTreeView *pTreeView, QStandardItemModel *pModel, const QString &strKeyWord)
{
m_pCurTreeView = pTreeView;
if (m_pCurTreeView == NULL)
{
return;
}
m_pCurModel = pModel;
if (m_pCurModel == NULL)
{
return;
}
QRegExp rx("([a-z]?[0-9]?)*");
QString strLower = strKeyWord.toLower().trimmed();
if (!strLower.isEmpty())
{
bool bIsPreciseSearch = true;
QRegExpValidator v(rx, 0);
int pos = 0;
if (v.validate(strLower, pos) == QValidator::Acceptable)
{
bIsPreciseSearch = false;
}
if (bIsPreciseSearch)
{
PreciseSearch(strLower);
}
else
{
FuzzySearch(strLower);
}
}
//搜索框为空，显示全部节点
else
{
ShowTotalItem();
}
}
void TreeSearch::FuzzySearch(QString &strKeyWord)
{
//QString strInitialOfInput; //输入
int iRootCount = m_pCurModel->rowCount();
for (int i = 0; i < iRootCount; ++i)
{
QStandardItem *pCurRootItem = m_pCurModel->item(i);
if (pCurRootItem == NULL)
{
continue;
}
pCurRootItem->setData(false, ITEM_SHOWN);
QString strCurRootItem = pCurRootItem->text();
QString strItemInitial; //当前节点文本的拼音首字母
QString strItemQuanPin; //当前节点文本的全拼
GetPinYin(strCurRootItem, strItemInitial, strItemQuanPin);
QString strStart = "^";
QRegExp rx(strStart.append(strItemQuanPin));
QRegExpValidator v(rx, 0);
int pos = 0;
//如果输入的关键字匹配当前节点，则显示当前节点及其所有子节点
if (strItemInitial.contains(strKeyWord) || v.validate(strKeyWord, pos) != QValidator::Invalid)
{
m_pCurTreeView->setRowHidden(pCurRootItem->row(), m_pCurModel->indexFromItem(m_pCurModel->invisibleRootItem()), false);
ShowTotalChildrenItem(pCurRootItem);
continue;
}
//否则继续在其子节点中搜索
else
{
FuzzySearchChildren(pCurRootItem, strKeyWord);
}
}
}
void TreeSearch::FuzzySearchChildren(QStandardItem *pParentItem, QString &strKeyWord)
{
int iRootCount = pParentItem->rowCount();
for (int i = 0; i < iRootCount; ++i)
{
QStandardItem *pCurChildItem = pParentItem->child(i);
if (pCurChildItem == NULL)
{
continue;
}
pCurChildItem->setData(false, ITEM_SHOWN);
QString strCurChildItem = pCurChildItem->text();
QString strItemInitial; //当前节点文本的拼音首字母
QString strItemQuanPin; //当前节点文本的全拼
GetPinYin(strCurChildItem, strItemInitial, strItemQuanPin);
QString strStart = "^";
QRegExp rx(strStart.append(strItemQuanPin));
QRegExpValidator v(rx, 0);
int pos = 0;
//如果输入的关键字匹配当前节点，则显示当前节点及其所有子节点
if (strItemInitial.contains(strKeyWord) || v.validate(strKeyWord, pos) != QValidator::Invalid)
{
m_pCurTreeView->setRowHidden(pCurChildItem->row(), m_pCurModel->indexFromItem(pParentItem), false);
ShowTotalParentItem(pCurChildItem);
ShowTotalChildrenItem(pCurChildItem);
continue;
}
//否则继续在其子节点中搜索
else
{
FuzzySearchChildren(pCurChildItem, strKeyWord);
}
}
bool bHasChildren = pParentItem->data(ITEM_SHOWN).toBool();
if (pParentItem->parent() == NULL)
{
m_pCurTreeView->setRowHidden(pParentItem->row(), m_pCurModel->indexFromItem(m_pCurModel->invisibleRootItem()), !bHasChildren);
}
else
{
m_pCurTreeView->setRowHidden(pParentItem->row(), m_pCurModel->indexFromItem(pParentItem->parent()), !bHasChildren);
}
}
void TreeSearch::PreciseSearch(const QString &strKeyWord)
{
//QString strInitialOfInput; //输入
int iRootCount = m_pCurModel->rowCount();
for (int i = 0; i < iRootCount; ++i)
{
QStandardItem *pCurRootItem = m_pCurModel->item(i);
if (pCurRootItem == NULL)
{
continue;
}
pCurRootItem->setData(false, ITEM_SHOWN);
QString strCurRootItem = pCurRootItem->text();
//如果输入的关键字匹配当前节点，则显示当前节点及其所有子节点
if (strCurRootItem.contains(strKeyWord))
{
m_pCurTreeView->setRowHidden(pCurRootItem->row(), m_pCurModel->indexFromItem(m_pCurModel->invisibleRootItem()), false);
ShowTotalChildrenItem(pCurRootItem);
continue;
}
//否则继续在其子节点中搜索
else
{
PreciseSearchChildren(pCurRootItem, strKeyWord);
}
}
}
void TreeSearch::PreciseSearchChildren(QStandardItem *pParentItem, const QString &strKeyWord)
{
int iRootCount = pParentItem->rowCount();
for (int i = 0; i < iRootCount; ++i)
{
QStandardItem *pCurChildItem = pParentItem->child(i);
if (pCurChildItem == NULL)
{
continue;
}
pCurChildItem->setData(false, ITEM_SHOWN);
QString strCurChildItem = pCurChildItem->text();
//如果输入的关键字匹配当前节点，则显示当前节点及其所有子节点
if (strCurChildItem.contains(strKeyWord))
{
m_pCurTreeView->setRowHidden(pCurChildItem->row(), m_pCurModel->indexFromItem(pParentItem), false);
ShowTotalParentItem(pCurChildItem);
ShowTotalChildrenItem(pCurChildItem);
continue;
}
//否则继续在其子节点中搜索
else
{
PreciseSearchChildren(pCurChildItem, strKeyWord);
}
}
bool bHasChildren = pParentItem->data(ITEM_SHOWN).toBool();
if (pParentItem->parent() == NULL)
{
m_pCurTreeView->setRowHidden(pParentItem->row(), m_pCurModel->indexFromItem(m_pCurModel->invisibleRootItem()), !bHasChildren);
}
else
{
m_pCurTreeView->setRowHidden(pParentItem->row(), m_pCurModel->indexFromItem(pParentItem->parent()), !bHasChildren);
}
}
void TreeSearch::ShowTotalItem()
{
int iRootCount = m_pCurModel->rowCount();
for (int i = 0; i < iRootCount; ++i)
{
QStandardItem *pRootItem = m_pCurModel->item(i);
if (pRootItem == NULL)
{
continue;
}
m_pCurTreeView->setRowHidden(pRootItem->row(), m_pCurModel->indexFromItem(m_pCurModel->invisibleRootItem()), false);
ShowTotalChildrenItem(pRootItem);
}
}
void TreeSearch::ShowTotalChildrenItem(QStandardItem *pParentItem)
{
int iChildCount = pParentItem->rowCount();
for (int i = 0; i < iChildCount; ++i)
{
QStandardItem *pChildItem = pParentItem->child(i);
if (pChildItem == NULL)
{
continue;
}
m_pCurTreeView->setRowHidden(pChildItem->row(), m_pCurModel->indexFromItem(pParentItem), false);
ShowTotalChildrenItem(pChildItem);
}
}
void TreeSearch::ShowTotalParentItem(QStandardItem *pChildItem)
{
QStandardItem *pParentItem = pChildItem->parent();
if (pParentItem != NULL)
{
pParentItem->setData(true, ITEM_SHOWN);
ShowTotalParentItem(pParentItem);
}
}
void TreeSearch::GetPinYin(const QString &strHanZi, QString &strInitial, QString &strQuanPin)
{
//保存拼音的ASCII值
const int pyAscii[] = { -20319, -20317, -20304, -20295, -20292, -20283, -20265, -20257, -20242, -20230, -20051, -20036, -20032, -20026,
-20002, -19990, -19986, -19982, -19976, -19805, -19784, -19775, -19774, -19763, -19756, -19751, -19746, -19741, -19739, -19728,
-19725, -19715, -19540, -19531, -19525, -19515, -19500, -19484, -19479, -19467, -19289, -19288, -19281, -19275, -19270, -19263,
-19261, -19249, -19243, -19242, -19238, -19235, -19227, -19224, -19218, -19212, -19038, -19023, -19018, -19006, -19003, -18996,
-18977, -18961, -18952, -18783, -18774, -18773, -18763, -18756, -18741, -18735, -18731, -18722, -18710, -18697, -18696, -18526,
-18518, -18501, -18490, -18478, -18463, -18448, -18447, -18446, -18239, -18237, -18231, -18220, -18211, -18201, -18184, -18183,
-18181, -18012, -17997, -17988, -17970, -17964, -17961, -17950, -17947, -17931, -17928, -17922, -17759, -17752, -17733, -17730,
-17721, -17703, -17701, -17697, -17692, -17683, -17676, -17496, -17487, -17482, -17468, -17454, -17433, -17427, -17417, -17202,
-17185, -16983, -16970, -16942, -16915, -16733, -16708, -16706, -16689, -16664, -16657, -16647, -16474, -16470, -16465, -16459,
-16452, -16448, -16433, -16429, -16427, -16423, -16419, -16412, -16407, -16403, -16401, -16393, -16220, -16216, -16212, -16205,
-16202, -16187, -16180, -16171, -16169, -16158, -16155, -15959, -15958, -15944, -15933, -15920, -15915, -15903, -15889, -15878,
-15707, -15701, -15681, -15667, -15661, -15659, -15652, -15640, -15631, -15625, -15454, -15448, -15436, -15435, -15419, -15416,
-15408, -15394, -15385, -15377, -15375, -15369, -15363, -15362, -15183, -15180, -15165, -15158, -15153, -15150, -15149, -15144,
-15143, -15141, -15140, -15139, -15128, -15121, -15119, -15117, -15110, -15109, -14941, -14937, -14933, -14930, -14929, -14928,
-14926, -14922, -14921, -14914, -14908, -14902, -14894, -14889, -14882, -14873, -14871, -14857, -14678, -14674, -14670, -14668,
-14663, -14654, -14645, -14630, -14594, -14429, -14407, -14399, -14384, -14379, -14368, -14355, -14353, -14345, -14170, -14159,
-14151, -14149, -14145, -14140, -14137, -14135, -14125, -14123, -14122, -14112, -14109, -14099, -14097, -14094, -14092, -14090,
-14087, -14083, -13917, -13914, -13910, -13907, -13906, -13905, -13896, -13894, -13878, -13870, -13859, -13847, -13831, -13658,
-13611, -13601, -13406, -13404, -13400, -13398, -13395, -13391, -13387, -13383, -13367, -13359, -13356, -13343, -13340, -13329,
-13326, -13318, -13147, -13138, -13120, -13107, -13096, -13095, -13091, -13076, -13068, -13063, -13060, -12888, -12875, -12871,
-12860, -12858, -12852, -12849, -12838, -12831, -12829, -12812, -12802, -12607, -12597, -12594, -12585, -12556, -12359, -12346,
-12320, -12300, -12120, -12099, -12089, -12074, -12067, -12058, -12039, -11867, -11861, -11847, -11831, -11798, -11781, -11604,
-11589, -11536, -11358, -11340, -11339, -11324, -11303, -11097, -11077, -11067, -11055, -11052, -11045, -11041, -11038, -11024,
-11020, -11019, -11018, -11014, -10838, -10832, -10815, -10800, -10790, -10780, -10764, -10587, -10544, -10533, -10519, -10331,
-10329, -10328, -10322, -10315, -10309, -10307, -10296, -10281, -10274, -10270, -10262, -10260, -10256, -10254 };
//保存拼音的音节
const char pyStr[396][7] = { "a", "ai", "an", "ang", "ao", "ba", "bai", "ban", "bang", "bao", "bei", "ben", "beng", "bi", "bian", "biao",
"bie", "bin", "bing", "bo", "bu", "ca", "cai", "can", "cang", "cao", "ce", "ceng", "cha", "chai", "chan", "chang", "chao", "che", "chen",
"cheng", "chi", "chong", "chou", "chu", "chuai", "chuan", "chuang", "chui", "chun", "chuo", "ci", "cong", "cou", "cu", "cuan", "cui",
"cun", "cuo", "da", "dai", "dan", "dang", "dao", "de", "deng", "di", "dian", "diao", "die", "ding", "diu", "dong", "dou", "du", "duan",
"dui", "dun", "duo", "e", "en", "er", "fa", "fan", "fang", "fei", "fen", "feng", "fo", "fou", "fu", "ga", "gai", "gan", "gang", "gao",
"ge", "gei", "gen", "geng", "gong", "gou", "gu", "gua", "guai", "guan", "guang", "gui", "gun", "guo", "ha", "hai", "han", "hang",
"hao", "he", "hei", "hen", "heng", "hong", "hou", "hu", "hua", "huai", "huan", "huang", "hui", "hun", "huo", "ji", "jia", "jian",
"jiang", "jiao", "jie", "jin", "jing", "jiong", "jiu", "ju", "juan", "jue", "jun", "ka", "kai", "kan", "kang", "kao", "ke", "ken",
"keng", "kong", "kou", "ku", "kua", "kuai", "kuan", "kuang", "kui", "kun", "kuo", "la", "lai", "lan", "lang", "lao", "le", "lei",
"leng", "li", "lia", "lian", "liang", "liao", "lie", "lin", "ling", "liu", "long", "lou", "lu", "lv", "luan", "lue", "lun", "luo",
"ma", "mai", "man", "mang", "mao", "me", "mei", "men", "meng", "mi", "mian", "miao", "mie", "min", "ming", "miu", "mo", "mou", "mu",
"na", "nai", "nan", "nang", "nao", "ne", "nei", "nen", "neng", "ni", "nian", "niang", "niao", "nie", "nin", "ning", "niu", "nong",
"nu", "nv", "nuan", "nue", "nuo", "o", "ou", "pa", "pai", "pan", "pang", "pao", "pei", "pen", "peng", "pi", "pian", "piao", "pie",
"pin", "ping", "po", "pu", "qi", "qia", "qian", "qiang", "qiao", "qie", "qin", "qing", "qiong", "qiu", "qu", "quan", "que", "qun",
"ran", "rang", "rao", "re", "ren", "reng", "ri", "rong", "rou", "ru", "ruan", "rui", "run", "ruo", "sa", "sai", "san", "sang",
"sao", "se", "sen", "seng", "sha", "shai", "shan", "shang", "shao", "she", "shen", "sheng", "shi", "shou", "shu", "shua",
"shuai", "shuan", "shuang", "shui", "shun", "shuo", "si", "song", "sou", "su", "suan", "sui", "sun", "suo", "ta", "tai",
"tan", "tang", "tao", "te", "teng", "ti", "tian", "tiao", "tie", "ting", "tong", "tou", "tu", "tuan", "tui", "tun", "tuo",
"wa", "wai", "wan", "wang", "wei", "wen", "weng", "wo", "wu", "xi", "xia", "xian", "xiang", "xiao", "xie", "xin", "xing",
"xiong", "xiu", "xu", "xuan", "xue", "xun", "ya", "yan", "yang", "yao", "ye", "yi", "yin", "ying", "yo", "yong", "you",
"yu", "yuan", "yue", "yun", "za", "zai", "zan", "zang", "zao", "ze", "zei", "zen", "zeng", "zha", "zhai", "zhan", "zhang",
"zhao", "zhe", "zhen", "zheng", "zhi", "zhong", "zhou", "zhu", "zhua", "zhuai", "zhuan", "zhuang", "zhui", "zhun", "zhuo",
"zi", "zong", "zou", "zu", "zuan", "zui", "zun", "zuo" };
int iChrasc = 0;
char *ret = NULL;
std::wstring wStr;
GetWString(std::string(strHanZi.toLower().toLocal8Bit()), wStr);
int iLength = WideCharToMultiByte(CP_ACP, 0, wStr.c_str(), -1, 0, 0, 0, 0);
char *nowChar = new char[iLength + 1];
memset(nowChar, 0, iLength + 1);
WideCharToMultiByte(CP_ACP, 0, wStr.c_str(), -1, nowChar, iLength, 0, 0);
//转换ASCII
int offset = 0;
char *returnstr = new char[6 * iLength + 1];
memset(returnstr, 0, 6 * iLength + 1);
strInitial.clear();
strQuanPin.clear();
for (int j = 0; j < iLength;)
{
if (nowChar[j] == 0)
{
break;
}
if (nowChar[j] >= 0 && nowChar[j] < 128)
{
returnstr[offset] = nowChar[j];
strInitial.append(nowChar[j]);
strInitial = strInitial.trimmed();
++offset;
++j;
continue;
}
// 汉字处理
iChrasc = nowChar[j] * 256 + nowChar[j + 1] + 256;
if (iChrasc > 0 && iChrasc < 160)
{
returnstr[offset] = nowChar[j];
strInitial.append(nowChar[j]);
strInitial = strInitial.trimmed();
++offset;
++j;
}
else
{
for (int i = (sizeof(pyAscii) / sizeof(pyAscii[0]) - 1); i >= 0; --i)
{
if (i == 396)
{
continue;
}
if (pyAscii[i] <= iChrasc)
{
strcpy(returnstr + offset, pyStr[i]);
strInitial.append(QChar(pyStr[i][0]));
strInitial = strInitial.trimmed();
offset += strlen(pyStr[i]);
break;
}
}
j += 2;
}
}
if (strlen(returnstr) > 0)
{
ret = new char[strlen(returnstr) + 1];
memset(ret, 0, strlen(returnstr) + 1);
strcpy(ret, returnstr);
}
delete[] returnstr;
delete[] nowChar;
strQuanPin = ret;
delete[] ret;
}
void TreeSearch::GetWString(const std::string &strIn, std::wstring &wStrOut)
{
int iLength = MultiByteToWideChar(CP_ACP, 0, (LPCSTR)strIn.c_str(), -1, NULL, 0);
wchar_t *wszUtf8 = new wchar_t[iLength + 1];
memset(wszUtf8, 0, iLength * 2 + 2);
MultiByteToWideChar(CP_ACP, 0, (LPCSTR)strIn.c_str(), -1, (LPWSTR)wszUtf8, iLength);
wStrOut = wszUtf8;
delete[] wszUtf8;
}