Files
harbour-core/harbour/contrib/hbmisc/irm.c
Viktor Szakats 744655e0f1 2010-11-25 03:20 UTC+0100 Viktor Szakats (harbour.01 syenar.hu)
+ contrib/hbmisc/irm.c
  * contrib/hbmisc/hbmisc.hbp
    + Added IRM (Independent Record Map) API contributed
      and donated to public domain by Mindaugas.
        - IRMEXECUTE( aFilterTree ) --> pMap
              Condition array syntax:
                 { operation, tag, bag, value }
              F.e.:
                 IrmExecute( { "&", { "<=", "date",, 0d20100101 }, { "=", "place",, 12 } } )
        - IRMMAPNEXT( pMap, @ulRecNo ) --> lFound
        - IRMMAPSKIP( pMap, @ulRecNo ) --> lFound
        - IRMMAPCOUNT( pMap ) --> nRecCount
      (function names may change in the recent future)
    + Added 'NO GUARANTEES" disclaimer on top. (pls refine it if needed)
    ! Some minor tweaks to compile with current Harbour.
    ! Casts for win64.
    * Changed to use Harbour types, and minor cleanups in variable scopes.
    % Added optimized bit counting to hb_irmMapCount(). (it was a TODO)
      Mindaugas, pls take a look at it and/or test it with real code.
2010-11-25 02:22:52 +00:00

344 lines
9.3 KiB
C

/*
* $Id$
*/
/*
* Harbour Project source code:
* IRM (Independent Record Map) API
*
* Released to Public Domain in 2010 by Mindaugas Kavaliauskas <dbtopas at dbtopas.lt>
* www - http://harbour-project.org
*
* DISCLAIMER: USE AT YOUR OWN RISK. NO GUARANTEES.
*
*/
#include "hbapi.h"
#include "hbapiitm.h"
#include "hbapierr.h"
#include "hbapirdd.h"
typedef struct
{
HB_BYTE * pBits;
HB_ULONG ulSize; /* in bits */
HB_ULONG ulAlloc; /* in bits */
} HB_IRMMAP, * PHB_IRMMAP;
PHB_IRMMAP hb_irmMapAlloc( HB_ULONG ulSize )
{
PHB_IRMMAP pMap = ( PHB_IRMMAP ) hb_xgrab( sizeof( HB_IRMMAP ) );
if( ulSize == 0 )
ulSize = 256;
pMap->ulSize = ulSize;
pMap->ulAlloc = ( ulSize + 7 ) & ~7UL;
pMap->pBits = ( HB_BYTE * ) hb_xgrab( pMap->ulAlloc >> 3 );
memset( pMap->pBits, 0, pMap->ulAlloc >> 3 );
return pMap;
}
void hb_irmMapFree( PHB_IRMMAP pMap )
{
hb_xfree( pMap->pBits );
hb_xfree( pMap );
}
int hb_irmMapBitGet( PHB_IRMMAP pMap, HB_ULONG ulRecNo )
{
--ulRecNo;
return ( pMap->pBits[ ulRecNo >> 3 ] >> ( ulRecNo & 7 ) ) & 1;
}
void hb_irmMapBitSet( PHB_IRMMAP pMap, HB_ULONG ulRecNo )
{
--ulRecNo;
pMap->pBits[ ulRecNo >> 3 ] |= 1 << ( ulRecNo & 7 );
}
void hb_irmMapBitClear( PHB_IRMMAP pMap, HB_ULONG ulRecNo )
{
--ulRecNo;
pMap->pBits[ ulRecNo >> 3 ] &= ~( 1 << ( ulRecNo & 7 ) );
}
HB_ULONG hb_irmMapNext( PHB_IRMMAP pMap, HB_ULONG ulRecNo )
{
HB_ULONG ulNext;
if( ulRecNo >= pMap->ulSize )
return 0;
ulNext = ( ulRecNo + 7 ) & ~7UL;
if( ulNext > pMap->ulSize )
ulNext = pMap->ulSize;
/* test the first byte */
while( ulRecNo < ulNext )
{
if( ( pMap->pBits[ ulRecNo >> 3 ] >> ( ulRecNo & 7 ) ) & 1 )
return ulRecNo + 1;
ulRecNo++;
}
while( ulRecNo < pMap->ulSize )
{
if( pMap->pBits[ ulRecNo >> 3 ] != 0 )
{
while( ulRecNo < pMap->ulSize )
{
if( ( pMap->pBits[ ulRecNo >> 3 ] >> ( ulRecNo & 7 ) ) & 1 )
return ulRecNo + 1;
ulRecNo++;
}
break;
}
ulRecNo += 8;
}
return 0;
}
HB_ULONG hb_irmMapCount( PHB_IRMMAP pMap )
{
static const HB_BYTE s_bitcnt[] =
{
0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4,
1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5,
1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5,
2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5,
2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5,
2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
4, 5, 5, 6, 5, 6, 6, 7, 5, 6, 6, 7, 6, 7, 7, 8
};
HB_ULONG ul;
HB_ULONG ulCount;
HB_ULONG ulSize = pMap->ulSize >> 3;
ulCount = 0;
for( ul = 0; ul < ulSize; ++ul )
ulCount += s_bitcnt[ pMap->pBits[ ul ] ];
return ulCount;
}
static HB_GARBAGE_FUNC( hb_irmMapDestroy )
{
PHB_IRMMAP * ppMap = ( PHB_IRMMAP * ) Cargo;
hb_irmMapFree( * ppMap );
}
static const HB_GC_FUNCS s_irmMapFuncs =
{
hb_irmMapDestroy,
hb_gcDummyMark
};
static PHB_IRMMAP hb_irmMapParam( int iParam )
{
PHB_IRMMAP * ppMap = ( PHB_IRMMAP * ) hb_parptrGC( &s_irmMapFuncs, iParam );
if( ppMap && * ppMap )
return * ppMap;
hb_errRT_BASE_SubstR( EG_ARG, 3012, NULL, HB_ERR_FUNCNAME, HB_ERR_ARGS_BASEPARAMS );
return NULL;
}
static void hb_irmMapMarkCallback( HB_ULONG ulRecNo, unsigned char * pKey, unsigned short uiLen, void * pMap )
{
HB_SYMBOL_UNUSED( pKey );
HB_SYMBOL_UNUSED( uiLen );
hb_irmMapBitSet( ( PHB_IRMMAP ) pMap, ulRecNo );
}
/*
* Expresion operators:
* "&", expr1, expr2, ...
* "|", expr1, expr2, ...
* Filter operators:
* "=", tag, bag, value ORKEYVAL() = value
* "<=", tag, bag, value ORKEYVAL() <= value
* ">=", tag, bag, value ORKEYVAL() >= value
* "<=<=", tag, bag, value1, value2 value1 <= ORKEYVAL() <= value2
*/
PHB_IRMMAP hb_irmExecute( PHB_ITEM pItem )
{
PHB_IRMMAP pMap, * pMapArray;
const char * szOper;
HB_ULONG ulLen, ulSize, ul, ul2;
AREAP pArea;
DBORDERINFO dboi;
if( HB_IS_ARRAY( pItem ) && ( szOper = hb_arrayGetCPtr( pItem, 1 ) ) != NULL )
{
/* Expression operators */
if( ( ! strcmp( szOper, "&" ) || ! strcmp( szOper, "|" ) ) && ( ulLen = ( HB_ULONG ) hb_arrayLen( pItem ) ) > 1 )
{
--ulLen;
pMapArray = ( PHB_IRMMAP * ) hb_xgrab( sizeof( PHB_IRMMAP ) * ulLen );
for( ul = 0; ul < ulLen; ++ul )
{
pMapArray[ ul ] = hb_irmExecute( hb_arrayGetItemPtr( pItem, ul + 2 ) );
}
ulSize = ( pMapArray[ 0 ]->ulSize + 7 ) >> 3;
if( ! strcmp( szOper, "&" ) )
{
for( ul = 0; ul < ulSize; ++ul )
for( ul2 = 1; ul2 < ulLen; ++ul2 )
pMapArray[ 0 ]->pBits[ ul ] &= pMapArray[ ul2 ]->pBits[ ul ];
}
else
{
for( ul = 0; ul < ulSize; ++ul )
for( ul2 = 1; ul2 < ulLen; ++ul2 )
pMapArray[ 0 ]->pBits[ ul ] |= pMapArray[ ul2 ]->pBits[ ul ];
}
pMap = pMapArray[ 0 ];
for( ul = 1; ul < ulLen; ++ul )
{
hb_irmMapFree( pMapArray[ ul ] );
}
hb_xfree( pMapArray );
return pMap;
}
/* Filter operators */
else if( ( ! strcmp( szOper, "=" ) && ( hb_arrayLen( pItem ) == 4 ) ) ||
( ! strcmp( szOper, "<=" ) && ( hb_arrayLen( pItem ) == 4 ) ) ||
( ! strcmp( szOper, ">=" ) && ( hb_arrayLen( pItem ) == 4 ) ) ||
( ! strcmp( szOper, "<=<=" ) && ( hb_arrayLen( pItem ) == 5 ) ) )
{
pArea = ( AREAP ) hb_rddGetCurrentWorkAreaPointer();
if( pArea )
{
SELF_RECCOUNT( pArea, &ulSize );
pMap = hb_irmMapAlloc( ulSize );
dboi.itmOrder = hb_arrayGetItemPtr( pItem, 2 );
dboi.atomBagName = hb_arrayGetItemPtr( pItem, 3 );
dboi.itmResult = hb_itemNew( NULL );
dboi.itmNewVal = hb_itemArrayNew( DBRMI_SIZE );
hb_arraySetPtr( dboi.itmNewVal, DBRMI_FUNCTION, ( void * ) hb_irmMapMarkCallback );
hb_arraySetPtr( dboi.itmNewVal, DBRMI_PARAM, ( void * ) pMap );
if( ! strcmp( szOper, "=" ) )
{
hb_arraySet( dboi.itmNewVal, DBRMI_LOVAL, hb_arrayGetItemPtr( pItem, 4 ) );
hb_arraySet( dboi.itmNewVal, DBRMI_HIVAL, hb_arrayGetItemPtr( pItem, 4 ) );
}
else if( ! strcmp( szOper, "<=" ) )
{
hb_arraySet( dboi.itmNewVal, DBRMI_HIVAL, hb_arrayGetItemPtr( pItem, 4 ) );
}
else if( ! strcmp( szOper, ">=" ) )
{
hb_arraySet( dboi.itmNewVal, DBRMI_LOVAL, hb_arrayGetItemPtr( pItem, 4 ) );
}
else if( ! strcmp( szOper, "<=<=" ) )
{
hb_arraySet( dboi.itmNewVal, DBRMI_LOVAL, hb_arrayGetItemPtr( pItem, 4 ) );
hb_arraySet( dboi.itmNewVal, DBRMI_HIVAL, hb_arrayGetItemPtr( pItem, 5 ) );
}
SELF_ORDINFO( pArea, DBOI_SCOPEEVAL, &dboi );
/* bitcount ulSize = hb_itemGetNL( dboi.itmResult ); */
hb_itemRelease( dboi.itmNewVal );
hb_itemRelease( dboi.itmResult );
return pMap;
}
else
hb_errRT_BASE_SubstR( EG_NOTABLE, 0, NULL, HB_ERR_FUNCNAME, HB_ERR_ARGS_BASEPARAMS );
}
else
hb_errRT_BASE_SubstR( EG_ARG, 3012, NULL, HB_ERR_FUNCNAME, HB_ERR_ARGS_BASEPARAMS );
}
else
hb_errRT_BASE_SubstR( EG_ARG, 3012, NULL, HB_ERR_FUNCNAME, HB_ERR_ARGS_BASEPARAMS );
return NULL;
}
/* IRMEXECUTE( aFilterTree ) --> pMap */
HB_FUNC( IRMEXECUTE )
{
PHB_IRMMAP pMap = hb_irmExecute( hb_param( 1, HB_IT_ANY ) );
if( pMap )
{
PHB_IRMMAP * ppMap = ( PHB_IRMMAP * ) hb_gcAllocate( sizeof( PHB_IRMMAP ), &s_irmMapFuncs );
* ppMap = pMap;
hb_retptrGC( ppMap );
}
}
/*
* ulRecNo := 0
* DO WHILE IRMMAPNEXT( pMap, @ulRecNo )
* dbGoto( nRecNo )
* ...
* ENDDO
*/
HB_FUNC( IRMMAPNEXT )
{
PHB_IRMMAP pMap = hb_irmMapParam( 1 );
if( pMap )
{
HB_ULONG ulRecNo = hb_parnl( 2 );
ulRecNo = hb_irmMapNext( pMap, ulRecNo );
hb_stornl( ulRecNo, 2 );
hb_retl( ulRecNo != 0 );
}
}
/*
* ulRecNo := 0
* DO WHILE IRMMAPSKIP( pMap, @ulRecNo )
* ...
* ENDDO
*/
HB_FUNC( IRMMAPSKIP )
{
PHB_IRMMAP pMap = hb_irmMapParam( 1 );
if( pMap )
{
HB_ULONG ulRecNo = hb_parnl( 2 );
ulRecNo = hb_irmMapNext( pMap, ulRecNo );
hb_stornl( ulRecNo, 2 );
if( ulRecNo != 0 )
hb_retl( SELF_GOTO( ( AREAP ) hb_rddGetCurrentWorkAreaPointer(), ulRecNo ) == HB_SUCCESS );
else
hb_retl( HB_FALSE );
}
}
/* IRMMAPCOUNT( pMap ) --> nRecCount */
HB_FUNC( IRMMAPCOUNT )
{
PHB_IRMMAP pMap = hb_irmMapParam( 1 );
if( pMap )
hb_retnl( hb_irmMapCount( pMap ) );
}