Initial changes needed to convert XFLAIM over to FTK.

git-svn-id: https://svn.code.sf.net/p/flaim/code/trunk@368 0109f412-320b-0410-ab79-c3e0c5ffbbe6
2006-05-01 23:29:03 +00:00
parent e76ad4d5a8
commit bb1f3e4a80
13 changed files with 1391 additions and 405 deletions
--- a/ftk/src/ftkrset.cpp
+++ b/ftk/src/ftkrset.cpp
@@ -531,6 +531,105 @@ private:
 	friend class F_ResultSetBlk;
 };
 	
+/*****************************************************************************
+Desc:
+*****************************************************************************/
+class F_BtResultSet : public IF_BtResultSet, public F_Base
+{
+public:
+
+	F_BtResultSet( void)
+	{
+		m_pBtree = NULL;
+		m_pCompare = NULL;
+	}
+
+	virtual ~F_BtResultSet()
+	{
+		if( m_pBtree)
+		{
+			m_pBtree->Release();
+		}
+		
+		if( m_pCompare)
+		{
+			m_pCompare->Release();
+		}
+	}
+	
+	RCODE FLMAPI setupResultSet(
+		const char *				pszPath,
+		IF_ResultSetCompare *	pCompare);
+
+	RCODE FLMAPI addEntry(
+		FLMBYTE *	pucKey,
+		FLMUINT		uiKeyLength,
+		FLMBYTE *	pucEntry,
+		FLMUINT		uiEntryLength);
+
+	RCODE FLMAPI modifyEntry(
+		FLMBYTE *	pucKey,
+		FLMUINT		uiKeyLength,
+		FLMBYTE *	pucEntry,
+		FLMUINT		uiEntryLength);
+
+	RCODE FLMAPI getCurrent(
+		FLMBYTE *	pucKey,
+		FLMUINT		uiKeyLength,
+		FLMBYTE *	pucEntry,
+		FLMUINT		uiEntryLength,
+		FLMUINT *	puiReturnLength);
+
+	RCODE FLMAPI getNext(
+		FLMBYTE *	pucKey,
+		FLMUINT		uiKeyBufLen,
+		FLMUINT *	puiKeylen,
+		FLMBYTE *	pucBuffer,
+		FLMUINT		uiBufferLength,
+		FLMUINT *	puiReturnLength);
+
+	RCODE FLMAPI getPrev(
+		FLMBYTE *	pucKey,
+		FLMUINT		uiKeyBufLen,
+		FLMUINT *	puiKeylen,
+		FLMBYTE *	pucBuffer,
+		FLMUINT		uiBufferLength,
+		FLMUINT *	puiReturnLength);
+
+	RCODE FLMAPI getFirst(
+		FLMBYTE *	pucKey,
+		FLMUINT		uiKeyBufLen,
+		FLMUINT *	puiKeylen,
+		FLMBYTE *	pucBuffer,
+		FLMUINT		uiBufferLength,
+		FLMUINT *	puiReturnLength);
+
+	RCODE FLMAPI getLast(
+		FLMBYTE *	pucKey,
+		FLMUINT		uiKeyBufLen,
+		FLMUINT *	puiKeylen,
+		FLMBYTE *	pucBuffer,
+		FLMUINT		uiBufferLength,
+		FLMUINT *	puiReturnLength);
+
+	RCODE FLMAPI findEntry(
+		FLMBYTE *	pucKey,
+		FLMUINT		uiKeyBufLen,
+		FLMUINT *	puiKeylen,
+		FLMBYTE *	pucBuffer,
+		FLMUINT		uiBufferLength,
+		FLMUINT *	puiReturnLength);
+
+	RCODE FLMAPI deleteEntry(
+		FLMBYTE *	pucKey,
+		FLMUINT		uiKeyLength);
+
+private:
+
+	IF_Btree *						m_pBtree;
+	IF_ResultSetCompare *		m_pCompare;
+};
+
 /*****************************************************************************
 Desc:
 *****************************************************************************/
@@ -3763,3 +3862,286 @@ Exit:

 	return( rc);
 }
+
+/****************************************************************************
+Desc:
+****************************************************************************/
+RCODE F_BtResultSet::addEntry(
+	FLMBYTE *	pucKey,
+	FLMUINT		uiKeyLength,
+	FLMBYTE *	pucEntry,
+	FLMUINT		uiEntryLength)
+{
+	RCODE			rc = NE_FLM_OK;
+
+	f_assert( uiKeyLength <= FLM_MAX_KEY_SIZE);
+
+	if( RC_BAD( rc = m_pBtree->btInsertEntry( pucKey, uiKeyLength,
+		uiKeyLength, pucEntry, uiEntryLength, TRUE, TRUE)))
+	{
+		if (rc == NE_FLM_NOT_UNIQUE)
+		{
+			rc = NE_FLM_OK;
+		}
+		else
+		{
+			goto Exit;
+		}
+	}
+
+Exit:
+
+	return( rc);
+}
+
+/****************************************************************************
+Desc:
+****************************************************************************/
+RCODE F_BtResultSet::modifyEntry(
+	FLMBYTE *	pucKey,
+	FLMUINT		uiKeyLength,
+	FLMBYTE *	pucEntry,
+	FLMUINT		uiEntryLength)
+{
+	RCODE			rc = NE_FLM_OK;
+
+	f_assert( uiKeyLength <= FLM_MAX_KEY_SIZE);
+
+	if( RC_BAD( rc = m_pBtree->btReplaceEntry( pucKey, uiKeyLength,
+		uiKeyLength, pucEntry, uiEntryLength, TRUE, TRUE)))
+	{
+		goto Exit;
+	}
+
+Exit:
+
+	return( rc);
+}
+
+/****************************************************************************
+Desc:
+****************************************************************************/
+RCODE F_BtResultSet::deleteEntry(
+	FLMBYTE *	pucKey,
+	FLMUINT		uiKeyLength)
+{
+	RCODE			rc = NE_FLM_OK;
+
+	f_assert( uiKeyLength <= FLM_MAX_KEY_SIZE);
+
+	if (RC_BAD( rc = m_pBtree->btRemoveEntry( 
+		pucKey, uiKeyLength, uiKeyLength)))
+	{
+		goto Exit;
+	}
+
+Exit:
+
+	return( rc);
+}
+
+/****************************************************************************
+Desc:
+****************************************************************************/
+RCODE F_BtResultSet::findEntry(
+	FLMBYTE *	pucKey,
+	FLMUINT		uiKeyBufLen,
+	FLMUINT *	puiKeyLen,
+	FLMBYTE *	pucBuffer,
+	FLMUINT		uiBufferLength,
+	FLMUINT *	puiReturnLength)
+{
+	RCODE			rc = NE_FLM_OK;
+	FLMUINT		uiLengthRV;
+
+	f_assert( uiKeyBufLen <= FLM_MAX_KEY_SIZE);
+
+	if( RC_BAD( rc = m_pBtree->btLocateEntry( pucKey, uiKeyBufLen, puiKeyLen,
+		FLM_EXACT, NULL, &uiLengthRV)))
+	{
+		goto Exit;
+	}
+
+	if( pucBuffer)
+	{
+		if( RC_BAD( rc = m_pBtree->btGetEntry( pucKey, *puiKeyLen,
+			pucBuffer, uiBufferLength, puiReturnLength)))
+		{
+			goto Exit;
+		}
+	}
+	else if( puiReturnLength)
+	{
+		*puiReturnLength = uiLengthRV;
+	}
+
+Exit:
+
+	return( rc);
+}
+
+/****************************************************************************
+Desc:
+****************************************************************************/
+RCODE F_BtResultSet::getCurrent(
+	FLMBYTE *	pucKey,
+	FLMUINT		uiKeyLength,
+	FLMBYTE *	pucEntry,
+	FLMUINT		uiEntryLength,
+	FLMUINT *	puiReturnLength)
+{
+	RCODE			rc = NE_FLM_OK;
+
+	f_assert( uiKeyLength <= FLM_MAX_KEY_SIZE);
+
+	if( RC_BAD( rc = m_pBtree->btGetEntry( pucKey, uiKeyLength,
+		pucEntry, uiEntryLength, puiReturnLength)))
+	{
+		goto Exit;
+	}
+
+Exit:
+
+	return( rc);
+}
+
+/****************************************************************************
+Desc:
+****************************************************************************/
+RCODE F_BtResultSet::getNext(
+	FLMBYTE *	pucKey,
+	FLMUINT		uiKeyBufLen,
+	FLMUINT *	puiKeyLen,
+	FLMBYTE *	pucEntry,
+	FLMUINT		uiEntryLength,
+	FLMUINT *	puiReturnLength)
+{
+	RCODE			rc = NE_FLM_OK;
+
+	f_assert( uiKeyBufLen <= FLM_MAX_KEY_SIZE);
+
+	if( RC_BAD( rc = m_pBtree->btNextEntry( pucKey, uiKeyBufLen, puiKeyLen,
+		puiReturnLength)))
+	{
+		goto Exit;
+	}
+
+	if( pucEntry)
+	{
+		if( RC_BAD( rc = m_pBtree->btGetEntry( pucKey, *puiKeyLen,
+			pucEntry, uiEntryLength, puiReturnLength)))
+		{
+			goto Exit;
+		}
+	}
+
+Exit:
+
+	return( rc);
+}
+
+/****************************************************************************
+Desc:
+****************************************************************************/
+RCODE F_BtResultSet::getPrev(
+	FLMBYTE *	pucKey,
+	FLMUINT		uiKeyBufLen,
+	FLMUINT *	puiKeyLen,
+	FLMBYTE *	pucEntry,
+	FLMUINT		uiEntryLength,
+	FLMUINT *	puiReturnLength)
+{
+	RCODE			rc = NE_FLM_OK;
+
+	f_assert( uiKeyBufLen <= FLM_MAX_KEY_SIZE);
+
+	if( RC_BAD( rc = m_pBtree->btPrevEntry( pucKey, uiKeyBufLen, puiKeyLen,
+		puiReturnLength)))
+	{
+		goto Exit;
+	}
+
+	if( pucEntry)
+	{
+		if( RC_BAD( rc = m_pBtree->btGetEntry( pucKey, *puiKeyLen,
+			pucEntry, uiEntryLength, puiReturnLength)))
+		{
+			goto Exit;
+		}
+	}
+
+Exit:
+
+	return( rc);
+}
+
+/****************************************************************************
+Desc:
+****************************************************************************/
+RCODE F_BtResultSet::getFirst(
+	FLMBYTE *	pucKey,
+	FLMUINT		uiKeyBufLen,
+	FLMUINT *	puiKeyLen,
+	FLMBYTE *	pucEntry,
+	FLMUINT		uiEntryLength,
+	FLMUINT *	puiReturnLength)
+{
+	RCODE			rc = NE_FLM_OK;
+
+	f_assert( uiKeyBufLen <= FLM_MAX_KEY_SIZE);
+
+	m_pBtree->btResetBtree();
+	if( RC_BAD( rc = m_pBtree->btFirstEntry( pucKey, uiKeyBufLen, puiKeyLen,
+		puiReturnLength)))
+	{
+		goto Exit;
+	}
+
+	if( pucEntry)
+	{
+		if( RC_BAD( rc = m_pBtree->btGetEntry( pucKey, *puiKeyLen,
+			pucEntry, uiEntryLength, puiReturnLength)))
+		{
+			goto Exit;
+		}
+	}
+
+Exit:
+
+	return( rc);
+}
+
+/****************************************************************************
+Desc:
+****************************************************************************/
+RCODE F_BtResultSet::getLast(
+	FLMBYTE *	pucKey,
+	FLMUINT		uiKeyBufLen,
+	FLMUINT *	puiKeyLen,
+	FLMBYTE *	pucEntry,
+	FLMUINT		uiEntryLength,
+	FLMUINT *	puiReturnLength)
+{
+	RCODE			rc = NE_FLM_OK;
+
+	f_assert( uiKeyBufLen <= FLM_MAX_KEY_SIZE);
+
+	if( RC_BAD( rc = m_pBtree->btLastEntry( pucKey, uiKeyBufLen, puiKeyLen,
+		puiReturnLength)))
+	{
+		goto Exit;
+	}
+
+	if( pucEntry)
+	{
+		if( RC_BAD( rc = m_pBtree->btGetEntry( pucKey, *puiKeyLen,
+			pucEntry, uiEntryLength, puiReturnLength)))
+		{
+			goto Exit;
+		}
+	}
+
+Exit:
+
+	return( rc);
+}