1
0
Fork 0
mirror of https://github.com/DBD-SQLite/DBD-SQLite synced 2025-06-07 14:19:10 -04:00

cleanup tokenizer code and test

This commit is contained in:
dami 2021-05-13 12:27:57 +02:00
parent 80c7c91ae6
commit 6669dbc332
2 changed files with 23 additions and 43 deletions

View file

@ -13,7 +13,7 @@ typedef struct perl_tokenizer_cursor {
/* members below are only used if the input string is in utf8 */ /* members below are only used if the input string is in utf8 */
const char *pInput; /* input we are tokenizing */ const char *pInput; /* input we are tokenizing */
const char *currentByte; /* pointer into pInput */ const char *currentByte; /* pointer into pInput */
int currentChar; /* char corresponding to currentByte */ int currentChar; /* char position corresponding to currentByte */
} perl_tokenizer_cursor; } perl_tokenizer_cursor;
/* /*
@ -134,7 +134,7 @@ static int perl_tokenizer_Open(
/* store the cursor coderef returned by the tokenizer */ /* store the cursor coderef returned by the tokenizer */
if (n_retval != 1) { if (n_retval != 1) {
warn("tokenizer returned %d arguments", n_retval); warn("tokenizer returned %d arguments, expected 1", n_retval);
} }
c->coderef = newSVsv(POPs); c->coderef = newSVsv(POPs);
@ -164,11 +164,11 @@ static int perl_tokenizer_Close(sqlite3_tokenizer_cursor *pCursor){
*/ */
static int perl_tokenizer_Next( static int perl_tokenizer_Next(
sqlite3_tokenizer_cursor *pCursor, /* Cursor returned by perl_tokenizer_Open */ sqlite3_tokenizer_cursor *pCursor, /* Cursor returned by perl_tokenizer_Open */
const char **ppToken, /* OUT: *ppToken is the token text */ const char **ppToken, /* OUT: Normalized text for token */
int *pnBytes, /* OUT: Number of bytes in token */ int *pnBytes, /* OUT: Number of bytes in normalized text */
int *piStartOffset, /* OUT: Starting offset of token */ int *piStartOffset, /* Starting offset of token. IN : char offset; OUT : byte offset */
int *piEndOffset, /* OUT: Ending offset of token */ int *piEndOffset, /* Ending offset of token. IN : char offset; OUT : byte offset */
int *piPosition /* OUT: Position integer of token */ int *piPosition /* OUT: Number of tokens returned before this one */
){ ){
perl_tokenizer_cursor *c = (perl_tokenizer_cursor *) pCursor; perl_tokenizer_cursor *c = (perl_tokenizer_cursor *) pCursor;
int result; int result;
@ -197,7 +197,7 @@ static int perl_tokenizer_Next(
/* otherwise, get token details from the return list */ /* otherwise, get token details from the return list */
else { else {
if (n_retval != 5) { if (n_retval != 5) {
warn("tokenizer cursor returned %d arguments", n_retval); warn("tokenizer cursor returned %d arguments, expected 5", n_retval);
} }
*piPosition = POPi; *piPosition = POPi;
*piEndOffset = POPi; *piEndOffset = POPi;
@ -205,41 +205,31 @@ static int perl_tokenizer_Next(
*pnBytes = POPi; *pnBytes = POPi;
token = POPpx; token = POPpx;
if (c->pInput) { /* if working with utf8 data */ if (c->pInput) { /* if working with utf8 data */
/* compute first hop : nb of chars from last position to the start of the token */
#ifdef DEBUG_OFFSETS
warn("INI: token: %s, start=%d, end=%d, nBytes=%d\n", token, *piStartOffset, *piEndOffset, *pnBytes);
#endif
/* recompute *pnBytes in bytes, not in chars */
*pnBytes = strlen(token);
/* nb of chars from last position to the start of the token */
hop = *piStartOffset - c->currentChar; hop = *piStartOffset - c->currentChar;
/* advance to the first byte in token */ /* hop: advance to the first byte in token */
nextByte = (char*)utf8_hop((U8*)c->currentByte, hop); nextByte = (char*)utf8_hop((U8*)c->currentByte, hop);
/* nb of chars in token */ /* compute 2nd hop : nb of chars from start of the token to end of token */
hop = *piEndOffset - *piStartOffset; hop = *piEndOffset - *piStartOffset;
/* recompute start offset in bytes, not in chars */ /* now recompute the start offset in bytes, not in chars */
*piStartOffset = nextByte - c->pInput; *piStartOffset = nextByte - c->pInput;
/* advance past to the last byte in token */ /* 2nd hop: advance past to the last byte in token */
nextByte = (char*)utf8_hop((U8*)nextByte, hop); nextByte = (char*)utf8_hop((U8*)nextByte, hop);
/* remember where we are for next round */ /* remember current position (useful for the next invocation) */
c->currentChar = *piEndOffset; c->currentChar = *piEndOffset;
c->currentByte = nextByte; c->currentByte = nextByte;
/* recompute end offset in bytes, not in chars */ /* now recompute the end offset in bytes, not in chars */
*piEndOffset = nextByte - c->pInput; *piEndOffset = nextByte - c->pInput;
#ifdef DEBUG_OFFSETS /* compute the size of the normalized token in bytes, not in chars */
warn("FIX: token: %s, start=%d, end=%d, nBytes=%d\n", token, *piStartOffset, *piEndOffset, *pnBytes); *pnBytes = strlen(token);
#endif
} }
/* make sure we have enough storage for copying the token */ /* make sure we have enough storage for copying the token */
@ -251,8 +241,7 @@ static int perl_tokenizer_Next(
c->pToken = pNew; c->pToken = pNew;
} }
/* need to copy the token into the C cursor before perl frees that /* need to copy the token into the C cursor before perl frees that memory */
memory */
memcpy(c->pToken, token, *pnBytes); memcpy(c->pToken, token, *pnBytes);
*ppToken = c->pToken; *ppToken = c->pToken;

View file

@ -1,13 +1,6 @@
use strict; use strict;
use warnings; use warnings;
no if $] >= 5.022, "warnings", "locale";
use lib "t/lib"; use lib "t/lib";
# TMP for running tests from Emacs
use lib "lib";
use lib "../blib/lib";
use lib "../blib/arch";
use Time::HiRes qw/time/; use Time::HiRes qw/time/;
use SQLiteTest; use SQLiteTest;
use Test::More; use Test::More;
@ -32,13 +25,9 @@ my @tests = (
["(il OR elle) AND un*" => 0, 2 ], ["(il OR elle) AND un*" => 0, 2 ],
); );
my $ix_une_native = index($texts[0], "une"); my $ix_une_native = index($texts[0], "une");
my $ix_une_utf8 = do {use bytes; utf8::upgrade(my $bergere_utf8 = $texts[0]); index($bergere_utf8, "une");}; my $ix_une_utf8 = do {use bytes; utf8::upgrade(my $bergere_utf8 = $texts[0]); index($bergere_utf8, "une");};
BEGIN { BEGIN {
requires_unicode_support(); requires_unicode_support();
@ -59,8 +48,10 @@ sub Unicode_Word_tokenizer { # see also: Search::Tokenizer
return sub { return sub {
$string =~ /$regex/g or return; # either match, or no more token $string =~ /$regex/g or return; # either match, or no more token
my ($start, $end) = ($-[0], $+[0]); my $term = $&;
my $term = substr($string, $start, my $len = $end-$start); my $end = pos $string; # $+[0] is much slower
my $len = length($term);
my $start = $end - $len;
return ($term, $len, $start, $end, $term_index++); return ($term, $len, $start, $end, $term_index++);
}; };
}; };
@ -129,7 +120,7 @@ for my $use_unicode (0, 1) {
# simulated large document # simulated large document
open my $fh, "<", $INC{'DBD/SQLite.pm'} or die $!; open my $fh, "<", $INC{'DBD/SQLite.pm'} or die $!;
my $source_code = do {local $/; <$fh>}; my $source_code = do {local $/; <$fh>};
my $long_doc = $source_code x 1; my $long_doc = $source_code x 5;
my $t0 = time; my $t0 = time;
$insert_sth->execute($long_doc); $insert_sth->execute($long_doc);