Skip to content

Instantly share code, notes, and snippets.

@bohwaz
Created November 10, 2011 16:09
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save bohwaz/1355232 to your computer and use it in GitHub Desktop.
Save bohwaz/1355232 to your computer and use it in GitHub Desktop.
Pure PHP rank function for SQLite FTS4 (adapted from SQLite doc)
<?php
/*
Adapted from C function available at http://www.sqlite.org/fts3.html#appendix_a
Use like this:
$db = new SQLite3('database.db');
$db->createFunction('rank', 'sql_rank');
$db->query('CREATE VIRTUAL TABLE products USING fts4 (id INTEGER, title TEXT, description TEXT);');
$db->query('SELECT * FROM products WHERE products MATCH \'Computer\' ORDER BY rank(matchinfo(products), 0, 1.0, 0.5) DESC;');
*/
function sql_rank($aMatchInfo)
{
$iSize = 4;
$iPhrase = (int) 0; // Current phrase //
$score = (double)0.0; // Value to return //
/* Check that the number of arguments passed to this function is correct.
** If not, jump to wrong_number_args. Set aMatchinfo to point to the array
** of unsigned integer values returned by FTS function matchinfo. Set
** nPhrase to contain the number of reportable phrases in the users full-text
** query, and nCol to the number of columns in the table.
*/
$aMatchInfo = (string) func_get_arg(0);
$nPhrase = ord(substr($aMatchInfo, 0, $iSize));
$nCol = ord(substr($aMatchInfo, $iSize, $iSize));
if (func_num_args() > (1 + $nCol))
{
throw new Exception("Invalid number of arguments : ".$nCol);
}
// Iterate through each phrase in the users query. //
for ($iPhrase = 0; $iPhrase < $nPhrase; $iPhrase++)
{
$iCol = (int) 0; // Current column //
/* Now iterate through each column in the users query. For each column,
** increment the relevancy score by:
**
** (<hit count> / <global hit count>) * <column weight>
**
** aPhraseinfo[] points to the start of the data for phrase iPhrase. So
** the hit count and global hit counts for each column are found in
** aPhraseinfo[iCol*3] and aPhraseinfo[iCol*3+1], respectively.
*/
$aPhraseinfo = substr($aMatchInfo, (2 + $iPhrase * $nCol * 3) * $iSize);
for ($iCol = 0; $iCol < $nCol; $iCol++)
{
$nHitCount = ord(substr($aPhraseinfo, 3 * $iCol * $iSize, $iSize));
$nGlobalHitCount = ord(substr($aPhraseinfo, (3 * $iCol + 1) * $iSize, $iSize));
$weight = ($iCol < func_num_args() - 1) ? (double) func_get_arg($iCol + 1) : 0;
if ($nHitCount > 0)
{
$score += ((double)$nHitCount / (double)$nGlobalHitCount) * $weight;
}
}
}
return $score;
}
?>
@shimikano
Copy link

Thank you for this!

I sometimes ended up with errors due to $nGlobalHitCount being zero. A little debugging revealed that the ord() function probably isn't the right way to convert the binary data to an unsigned 32-bit integer because it only looks at a single char.

Instead, I used the upnack() function. This should take care of machine-dependent endianness as well:

function toInt($binaryData) {
  return unpack('L', $binaryData)[1]; // 'L' is for: unsigned long (always 32 bit, machine byte order)
}

function extractInt($binaryData, $position) {
  $iSize = 4;
  $extracted = substr($binaryData, $position * $iSize, $iSize);

  return toInt($extracted);
}

// example usage in the rank() function
$nGlobalHitCount = extractInt($aPhraseinfo, 3 * $iCol + 1)

@speich
Copy link

speich commented Jul 3, 2020

Note: The asterisk character can be used as an repeater argument to read the whole binary into an array, e.g.:
$arrInt = unpack('L*', $binaryData);

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment