Finding an item (matching any member of TV_ITEM)

FindNextItem traverses a tree searching for an item matching all the item
attributes set in a TV_ITEM structure. In the TV_ITEM structure, the mask
member specifies which attributes make up the search criteria. If a match
is found, the function returns the handle otherwise NULL. This function
uses the function
GetNextItem,
seen previously. The function only searches in one direction (down) and if
hItem is NULL starts from the root of the tree.

HTREEITEM CTreeCtrlEx::FindNextItem(TV_ITEM* pItem, HTREEITEM hItem)
{
ASSERT(::IsWindow(m_hWnd));

TV_ITEM hNextItem;

//Clear Item data
ZeroMemory(&hNextItem, sizeof(hNextItem));

//The mask is used to retrieve the data to compare
hNextItem.mask = pItem->mask;
hNextItem.hItem = (hItem) ? GetNextItem(hItem) : GetRootItem();

//Prepare to compare pszText
//Testing pItem->pszText protects the code from a client setting the
//TVIF_TEXT bit but passing in a NULL pointer.
if((pItem->mask & TVIF_TEXT) && pItem->pszText)
{
hNextItem.cchTextMax = strlen(pItem->pszText);

if(hNextItem.cchTextMax)
hNextItem.pszText = new char[++hNextItem.cchTextMax];
}

while(hNextItem.hItem)
{
if(Compare(pItem, hNextItem))
{
//Copy all the information into pItem and return
memcpy(pItem, &hNextItem, sizeof(TV_ITEM));

//Free resources
if(hNextItem.pszText)
delete hNextItem.pszText;

return pItem->hItem;
}

//The mask is used to retrieve the data to compare and must be
//reset before calling Compare
hNextItem.mask = pItem->mask;
hNextItem.hItem = GetNextItem(hNextItem.hItem);
}

//Set hItem in pItem
pItem->hItem = NULL;

//Free resources
if(hNextItem.pszText)
delete hNextItem.pszText;

return NULL;
}

BOOL CTreeCtrlEx::Compare(TV_ITEM* pItem, TV_ITEM& tvTempItem)
{
//This call uses the .mask setting to just retrieve the values
//that the client wants to compare.
//Get all the data passed in by pItem
GetItem(&tvTempItem);

//Reset the mask so I can keep track of the matching attributes
tvTempItem.mask = 0;

if((pItem->mask & TVIF_STATE) &&
(pItem->state == tvTempItem.state))
tvTempItem.mask |= TVIF_STATE;

if((pItem->mask & TVIF_IMAGE) &&
(pItem->iImage == tvTempItem.iImage))
tvTempItem.mask |= TVIF_IMAGE;

if((pItem->mask & TVIF_PARAM) &&
(pItem->lParam == tvTempItem.lParam))
tvTempItem.mask |= TVIF_PARAM;

if((pItem->mask & TVIF_TEXT) &&
pItem->pszText && tvTempItem.pszText && //Don’t compare if either is NULL
!strcmp(pItem->pszText, tvTempItem.pszText))
tvTempItem.mask |= TVIF_TEXT;

if((pItem->mask & TVIF_CHILDREN) &&
(pItem->cChildren == tvTempItem.cChildren))
tvTempItem.mask |= TVIF_CHILDREN;

if((pItem->mask & TVIF_SELECTEDIMAGE) &&
(pItem->iSelectedImage == tvTempItem.iSelectedImage))
tvTempItem.mask |= TVIF_SELECTEDIMAGE;

//If by this point these two values are the same.
//tvTempItem.hItem is the desired item
return (pItem->mask == tvTempItem.mask);
}

Date Last Updated: April 24, 1999

More by Author

Must Read