blob: 2082d6aec2b251d0622d245553f9c83afffec0da [file] [log] [blame]
drh75897232000-05-29 14:26:00 +00001/*
drhb19a2bc2001-09-16 00:13:26 +00002** 2001 September 15
drh75897232000-05-29 14:26:00 +00003**
drhb19a2bc2001-09-16 00:13:26 +00004** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
drh75897232000-05-29 14:26:00 +00006**
drhb19a2bc2001-09-16 00:13:26 +00007** May you do good and not evil.
8** May you find forgiveness for yourself and forgive others.
9** May you share freely, never taking more than you give.
drh75897232000-05-29 14:26:00 +000010**
11*************************************************************************
12** This module contains C code that generates VDBE code used to process
13** the WHERE clause of SQL statements. Also found here are subroutines
14** to generate VDBE code to evaluate expressions.
15**
drhad3cab52002-05-24 02:04:32 +000016** $Id: where.c,v 1.46 2002/05/24 02:04:34 drh Exp $
drh75897232000-05-29 14:26:00 +000017*/
18#include "sqliteInt.h"
19
20/*
21** The query generator uses an array of instances of this structure to
22** help it analyze the subexpressions of the WHERE clause. Each WHERE
23** clause subexpression is separated from the others by an AND operator.
24*/
25typedef struct ExprInfo ExprInfo;
26struct ExprInfo {
27 Expr *p; /* Pointer to the subexpression */
28 int indexable; /* True if this subexprssion is usable by an index */
drh967e8b72000-06-21 13:59:10 +000029 int idxLeft; /* p->pLeft is a column in this table number. -1 if
30 ** p->pLeft is not the column of any table */
31 int idxRight; /* p->pRight is a column in this table number. -1 if
32 ** p->pRight is not the column of any table */
drh75897232000-05-29 14:26:00 +000033 unsigned prereqLeft; /* Tables referenced by p->pLeft */
34 unsigned prereqRight; /* Tables referenced by p->pRight */
drh3f6b5482002-04-02 13:26:10 +000035 unsigned prereqAll; /* Tables referenced by this expression in any way */
drh75897232000-05-29 14:26:00 +000036};
37
38/*
39** Determine the number of elements in an array.
40*/
41#define ARRAYSIZE(X) (sizeof(X)/sizeof(X[0]))
42
43/*
44** This routine is used to divide the WHERE expression into subexpressions
45** separated by the AND operator.
46**
47** aSlot[] is an array of subexpressions structures.
48** There are nSlot spaces left in this array. This routine attempts to
49** split pExpr into subexpressions and fills aSlot[] with those subexpressions.
50** The return value is the number of slots filled.
51*/
52static int exprSplit(int nSlot, ExprInfo *aSlot, Expr *pExpr){
53 int cnt = 0;
54 if( pExpr==0 || nSlot<1 ) return 0;
55 if( nSlot==1 || pExpr->op!=TK_AND ){
56 aSlot[0].p = pExpr;
57 return 1;
58 }
59 if( pExpr->pLeft->op!=TK_AND ){
60 aSlot[0].p = pExpr->pLeft;
61 cnt = 1 + exprSplit(nSlot-1, &aSlot[1], pExpr->pRight);
62 }else{
63 cnt = exprSplit(nSlot, aSlot, pExpr->pRight);
64 cnt += exprSplit(nSlot-cnt, &aSlot[cnt], pExpr->pLeft);
65 }
66 return cnt;
67}
68
69/*
70** This routine walks (recursively) an expression tree and generates
71** a bitmask indicating which tables are used in that expression
72** tree. Bit 0 of the mask is set if table 0 is used. But 1 is set
73** if table 1 is used. And so forth.
74**
75** In order for this routine to work, the calling function must have
76** previously invoked sqliteExprResolveIds() on the expression. See
77** the header comment on that routine for additional information.
drh19a775c2000-06-05 18:54:46 +000078**
79** "base" is the cursor number (the value of the iTable field) that
drh832508b2002-03-02 17:04:07 +000080** corresponds to the first entry in the table list.
drh75897232000-05-29 14:26:00 +000081*/
drh19a775c2000-06-05 18:54:46 +000082static int exprTableUsage(int base, Expr *p){
drh75897232000-05-29 14:26:00 +000083 unsigned int mask = 0;
84 if( p==0 ) return 0;
drh967e8b72000-06-21 13:59:10 +000085 if( p->op==TK_COLUMN ){
drh19a775c2000-06-05 18:54:46 +000086 return 1<< (p->iTable - base);
drh75897232000-05-29 14:26:00 +000087 }
88 if( p->pRight ){
drh19a775c2000-06-05 18:54:46 +000089 mask = exprTableUsage(base, p->pRight);
drh75897232000-05-29 14:26:00 +000090 }
91 if( p->pLeft ){
drh19a775c2000-06-05 18:54:46 +000092 mask |= exprTableUsage(base, p->pLeft);
drh75897232000-05-29 14:26:00 +000093 }
drhdd579122002-04-02 01:58:57 +000094 if( p->pList ){
95 int i;
96 for(i=0; i<p->pList->nExpr; i++){
97 mask |= exprTableUsage(base, p->pList->a[i].pExpr);
98 }
99 }
drh75897232000-05-29 14:26:00 +0000100 return mask;
101}
102
103/*
drh487ab3c2001-11-08 00:45:21 +0000104** Return TRUE if the given operator is one of the operators that is
105** allowed for an indexable WHERE clause. The allowed operators are
106** "=", "<", ">", "<=", and ">=".
107*/
108static int allowedOp(int op){
109 switch( op ){
110 case TK_LT:
111 case TK_LE:
112 case TK_GT:
113 case TK_GE:
114 case TK_EQ:
115 return 1;
116 default:
117 return 0;
118 }
119}
120
121/*
drh75897232000-05-29 14:26:00 +0000122** The input to this routine is an ExprInfo structure with only the
123** "p" field filled in. The job of this routine is to analyze the
124** subexpression and populate all the other fields of the ExprInfo
125** structure.
drh19a775c2000-06-05 18:54:46 +0000126**
127** "base" is the cursor number (the value of the iTable field) that
drh832508b2002-03-02 17:04:07 +0000128** corresponds to the first entry in the table list.
drh75897232000-05-29 14:26:00 +0000129*/
drh19a775c2000-06-05 18:54:46 +0000130static void exprAnalyze(int base, ExprInfo *pInfo){
drh75897232000-05-29 14:26:00 +0000131 Expr *pExpr = pInfo->p;
drh19a775c2000-06-05 18:54:46 +0000132 pInfo->prereqLeft = exprTableUsage(base, pExpr->pLeft);
133 pInfo->prereqRight = exprTableUsage(base, pExpr->pRight);
drh3f6b5482002-04-02 13:26:10 +0000134 pInfo->prereqAll = exprTableUsage(base, pExpr);
drh75897232000-05-29 14:26:00 +0000135 pInfo->indexable = 0;
136 pInfo->idxLeft = -1;
137 pInfo->idxRight = -1;
drh487ab3c2001-11-08 00:45:21 +0000138 if( allowedOp(pExpr->op) && (pInfo->prereqRight & pInfo->prereqLeft)==0 ){
drh967e8b72000-06-21 13:59:10 +0000139 if( pExpr->pRight->op==TK_COLUMN ){
drh19a775c2000-06-05 18:54:46 +0000140 pInfo->idxRight = pExpr->pRight->iTable - base;
drh75897232000-05-29 14:26:00 +0000141 pInfo->indexable = 1;
142 }
drh967e8b72000-06-21 13:59:10 +0000143 if( pExpr->pLeft->op==TK_COLUMN ){
drh19a775c2000-06-05 18:54:46 +0000144 pInfo->idxLeft = pExpr->pLeft->iTable - base;
drh75897232000-05-29 14:26:00 +0000145 pInfo->indexable = 1;
146 }
147 }
148}
149
150/*
151** Generating the beginning of the loop used for WHERE clause processing.
152** The return value is a pointer to an (opaque) structure that contains
153** information needed to terminate the loop. Later, the calling routine
154** should invoke sqliteWhereEnd() with the return value of this function
155** in order to complete the WHERE clause processing.
156**
157** If an error occurs, this routine returns NULL.
158*/
159WhereInfo *sqliteWhereBegin(
160 Parse *pParse, /* The parser context */
drh832508b2002-03-02 17:04:07 +0000161 int base, /* VDBE cursor index for left-most table in pTabList */
drhad3cab52002-05-24 02:04:32 +0000162 SrcList *pTabList, /* A list of all tables to be scanned */
drh75897232000-05-29 14:26:00 +0000163 Expr *pWhere, /* The WHERE clause */
164 int pushKey /* If TRUE, leave the table key on the stack */
165){
166 int i; /* Loop counter */
167 WhereInfo *pWInfo; /* Will become the return value of this function */
168 Vdbe *v = pParse->pVdbe; /* The virtual database engine */
169 int brk, cont; /* Addresses used during code generation */
170 int *aOrder; /* Order in which pTabList entries are searched */
171 int nExpr; /* Number of subexpressions in the WHERE clause */
172 int loopMask; /* One bit set for each outer loop */
173 int haveKey; /* True if KEY is on the stack */
drhc4a3c772001-04-04 11:48:57 +0000174 int aDirect[32]; /* If TRUE, then index this table using ROWID */
drh8aff1012001-12-22 14:49:24 +0000175 int iDirectEq[32]; /* Term of the form ROWID==X for the N-th table */
176 int iDirectLt[32]; /* Term of the form ROWID<X or ROWID<=X */
177 int iDirectGt[32]; /* Term of the form ROWID>X or ROWID>=X */
drh75897232000-05-29 14:26:00 +0000178 ExprInfo aExpr[50]; /* The WHERE clause is divided into these expressions */
179
drh6b563442001-11-07 16:48:26 +0000180 /* Allocate space for aOrder[] and aiMem[]. */
drhad3cab52002-05-24 02:04:32 +0000181 aOrder = sqliteMalloc( sizeof(int) * pTabList->nSrc );
drh75897232000-05-29 14:26:00 +0000182
183 /* Allocate and initialize the WhereInfo structure that will become the
184 ** return value.
185 */
drhad3cab52002-05-24 02:04:32 +0000186 pWInfo = sqliteMalloc( sizeof(WhereInfo) + pTabList->nSrc*sizeof(WhereLevel));
drhdaffd0e2001-04-11 14:28:42 +0000187 if( sqlite_malloc_failed ){
drh75897232000-05-29 14:26:00 +0000188 sqliteFree(aOrder);
drhdaffd0e2001-04-11 14:28:42 +0000189 sqliteFree(pWInfo);
drh75897232000-05-29 14:26:00 +0000190 return 0;
191 }
192 pWInfo->pParse = pParse;
193 pWInfo->pTabList = pTabList;
drh832508b2002-03-02 17:04:07 +0000194 pWInfo->base = base;
195 pWInfo->peakNTab = pWInfo->savedNTab = pParse->nTab;
drh08192d52002-04-30 19:20:28 +0000196 pWInfo->iBreak = sqliteVdbeMakeLabel(v);
197
198 /* Special case: a WHERE clause that is constant. Evaluate the
199 ** expression and either jump over all of the code or fall thru.
200 */
201 if( pWhere && sqliteExprIsConstant(pWhere) ){
202 sqliteExprIfFalse(pParse, pWhere, pWInfo->iBreak);
203 }
drh75897232000-05-29 14:26:00 +0000204
205 /* Split the WHERE clause into as many as 32 separate subexpressions
206 ** where each subexpression is separated by an AND operator. Any additional
207 ** subexpressions are attached in the aExpr[32] and will not enter
208 ** into the query optimizer computations. 32 is chosen as the cutoff
209 ** since that is the number of bits in an integer that we use for an
210 ** expression-used mask.
211 */
212 memset(aExpr, 0, sizeof(aExpr));
213 nExpr = exprSplit(ARRAYSIZE(aExpr), aExpr, pWhere);
214
215 /* Analyze all of the subexpressions.
216 */
217 for(i=0; i<nExpr; i++){
drh22f70c32002-02-18 01:17:00 +0000218 exprAnalyze(base, &aExpr[i]);
drh1d1f3052002-05-21 13:18:25 +0000219
220 /* If we are executing a trigger body, remove all references to
221 ** new.* and old.* tables from the prerequisite masks.
222 */
223 if( pParse->trigStack ){
224 int x;
225 if( (x = pParse->trigStack->newIdx) >= 0 ){
226 int mask = ~(1 << (x - base));
227 aExpr[i].prereqRight &= mask;
228 aExpr[i].prereqLeft &= mask;
229 aExpr[i].prereqAll &= mask;
230 }
231 if( (x = pParse->trigStack->oldIdx) >= 0 ){
232 int mask = ~(1 << (x - base));
233 aExpr[i].prereqRight &= mask;
234 aExpr[i].prereqLeft &= mask;
235 aExpr[i].prereqAll &= mask;
236 }
danielk1977c3f9bad2002-05-15 08:30:12 +0000237 }
drh75897232000-05-29 14:26:00 +0000238 }
239
240 /* Figure out a good nesting order for the tables. aOrder[0] will
241 ** be the index in pTabList of the outermost table. aOrder[1] will
drhad3cab52002-05-24 02:04:32 +0000242 ** be the first nested loop and so on. aOrder[pTabList->nSrc-1] will
drh75897232000-05-29 14:26:00 +0000243 ** be the innermost loop.
244 **
drh1d1f3052002-05-21 13:18:25 +0000245 ** Someday we will put in a good algorithm here to reorder the loops
drh75897232000-05-29 14:26:00 +0000246 ** for an effiecient query. But for now, just use whatever order the
247 ** tables appear in in the pTabList.
248 */
drhad3cab52002-05-24 02:04:32 +0000249 for(i=0; i<pTabList->nSrc; i++){
drh75897232000-05-29 14:26:00 +0000250 aOrder[i] = i;
251 }
252
253 /* Figure out what index to use (if any) for each nested loop.
drh6b563442001-11-07 16:48:26 +0000254 ** Make pWInfo->a[i].pIdx point to the index to use for the i-th nested
drhad3cab52002-05-24 02:04:32 +0000255 ** loop where i==0 is the outer loop and i==pTabList->nSrc-1 is the inner
drh8aff1012001-12-22 14:49:24 +0000256 ** loop.
257 **
258 ** If terms exist that use the ROWID of any table, then set the
259 ** iDirectEq[], iDirectLt[], or iDirectGt[] elements for that table
260 ** to the index of the term containing the ROWID. We always prefer
261 ** to use a ROWID which can directly access a table rather than an
drh0a36c572002-02-18 22:49:59 +0000262 ** index which requires reading an index first to get the rowid then
263 ** doing a second read of the actual database table.
drh75897232000-05-29 14:26:00 +0000264 **
265 ** Actually, if there are more than 32 tables in the join, only the
drh0a36c572002-02-18 22:49:59 +0000266 ** first 32 tables are candidates for indices. This is (again) due
267 ** to the limit of 32 bits in an integer bitmask.
drh75897232000-05-29 14:26:00 +0000268 */
269 loopMask = 0;
drhad3cab52002-05-24 02:04:32 +0000270 for(i=0; i<pTabList->nSrc && i<ARRAYSIZE(aDirect); i++){
drhc4a3c772001-04-04 11:48:57 +0000271 int j;
drh75897232000-05-29 14:26:00 +0000272 int idx = aOrder[i];
273 Table *pTab = pTabList->a[idx].pTab;
274 Index *pIdx;
275 Index *pBestIdx = 0;
drh487ab3c2001-11-08 00:45:21 +0000276 int bestScore = 0;
drh75897232000-05-29 14:26:00 +0000277
drhc4a3c772001-04-04 11:48:57 +0000278 /* Check to see if there is an expression that uses only the
drh8aff1012001-12-22 14:49:24 +0000279 ** ROWID field of this table. For terms of the form ROWID==expr
280 ** set iDirectEq[i] to the index of the term. For terms of the
281 ** form ROWID<expr or ROWID<=expr set iDirectLt[i] to the term index.
282 ** For terms like ROWID>expr or ROWID>=expr set iDirectGt[i].
drhc4a3c772001-04-04 11:48:57 +0000283 */
drh8aff1012001-12-22 14:49:24 +0000284 iDirectEq[i] = -1;
285 iDirectLt[i] = -1;
286 iDirectGt[i] = -1;
drhc4a3c772001-04-04 11:48:57 +0000287 for(j=0; j<nExpr; j++){
288 if( aExpr[j].idxLeft==idx && aExpr[j].p->pLeft->iColumn<0
289 && (aExpr[j].prereqRight & loopMask)==aExpr[j].prereqRight ){
drh8aff1012001-12-22 14:49:24 +0000290 switch( aExpr[j].p->op ){
291 case TK_EQ: iDirectEq[i] = j; break;
292 case TK_LE:
293 case TK_LT: iDirectLt[i] = j; break;
294 case TK_GE:
295 case TK_GT: iDirectGt[i] = j; break;
296 }
drhc4a3c772001-04-04 11:48:57 +0000297 }
298 if( aExpr[j].idxRight==idx && aExpr[j].p->pRight->iColumn<0
299 && (aExpr[j].prereqLeft & loopMask)==aExpr[j].prereqLeft ){
drh8aff1012001-12-22 14:49:24 +0000300 switch( aExpr[j].p->op ){
301 case TK_EQ: iDirectEq[i] = j; break;
302 case TK_LE:
303 case TK_LT: iDirectGt[i] = j; break;
304 case TK_GE:
305 case TK_GT: iDirectLt[i] = j; break;
306 }
drhc4a3c772001-04-04 11:48:57 +0000307 }
308 }
drh8aff1012001-12-22 14:49:24 +0000309 if( iDirectEq[i]>=0 ){
drhc4a3c772001-04-04 11:48:57 +0000310 loopMask |= 1<<idx;
drh6b563442001-11-07 16:48:26 +0000311 pWInfo->a[i].pIdx = 0;
drhc4a3c772001-04-04 11:48:57 +0000312 continue;
313 }
314
drh75897232000-05-29 14:26:00 +0000315 /* Do a search for usable indices. Leave pBestIdx pointing to
drh487ab3c2001-11-08 00:45:21 +0000316 ** the "best" index. pBestIdx is left set to NULL if no indices
317 ** are usable.
drh75897232000-05-29 14:26:00 +0000318 **
drh487ab3c2001-11-08 00:45:21 +0000319 ** The best index is determined as follows. For each of the
320 ** left-most terms that is fixed by an equality operator, add
321 ** 4 to the score. The right-most term of the index may be
322 ** constrained by an inequality. Add 1 if for an "x<..." constraint
323 ** and add 2 for an "x>..." constraint. Chose the index that
324 ** gives the best score.
325 **
326 ** This scoring system is designed so that the score can later be
327 ** used to determine how the index is used. If the score&3 is 0
328 ** then all constraints are equalities. If score&1 is not 0 then
329 ** there is an inequality used as a termination key. (ex: "x<...")
330 ** If score&2 is not 0 then there is an inequality used as the
331 ** start key. (ex: "x>...");
drh75897232000-05-29 14:26:00 +0000332 */
333 for(pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext){
drh487ab3c2001-11-08 00:45:21 +0000334 int eqMask = 0; /* Index columns covered by an x=... constraint */
335 int ltMask = 0; /* Index columns covered by an x<... constraint */
336 int gtMask = 0; /* Index columns covered by an x>... constraing */
337 int nEq, m, score;
drh75897232000-05-29 14:26:00 +0000338
drh74e24cd2002-01-09 03:19:59 +0000339 if( pIdx->isDropped ) continue; /* Ignore dropped indices */
drh487ab3c2001-11-08 00:45:21 +0000340 if( pIdx->nColumn>32 ) continue; /* Ignore indices too many columns */
drh75897232000-05-29 14:26:00 +0000341 for(j=0; j<nExpr; j++){
342 if( aExpr[j].idxLeft==idx
343 && (aExpr[j].prereqRight & loopMask)==aExpr[j].prereqRight ){
drh967e8b72000-06-21 13:59:10 +0000344 int iColumn = aExpr[j].p->pLeft->iColumn;
drh75897232000-05-29 14:26:00 +0000345 int k;
drh967e8b72000-06-21 13:59:10 +0000346 for(k=0; k<pIdx->nColumn; k++){
347 if( pIdx->aiColumn[k]==iColumn ){
drh487ab3c2001-11-08 00:45:21 +0000348 switch( aExpr[j].p->op ){
349 case TK_EQ: {
350 eqMask |= 1<<k;
351 break;
352 }
353 case TK_LE:
354 case TK_LT: {
355 ltMask |= 1<<k;
356 break;
357 }
358 case TK_GE:
359 case TK_GT: {
360 gtMask |= 1<<k;
361 break;
362 }
363 default: {
364 /* CANT_HAPPEN */
365 assert( 0 );
366 break;
367 }
368 }
drh75897232000-05-29 14:26:00 +0000369 break;
370 }
371 }
372 }
373 if( aExpr[j].idxRight==idx
374 && (aExpr[j].prereqLeft & loopMask)==aExpr[j].prereqLeft ){
drh967e8b72000-06-21 13:59:10 +0000375 int iColumn = aExpr[j].p->pRight->iColumn;
drh75897232000-05-29 14:26:00 +0000376 int k;
drh967e8b72000-06-21 13:59:10 +0000377 for(k=0; k<pIdx->nColumn; k++){
378 if( pIdx->aiColumn[k]==iColumn ){
drh487ab3c2001-11-08 00:45:21 +0000379 switch( aExpr[j].p->op ){
380 case TK_EQ: {
381 eqMask |= 1<<k;
382 break;
383 }
384 case TK_LE:
385 case TK_LT: {
386 gtMask |= 1<<k;
387 break;
388 }
389 case TK_GE:
390 case TK_GT: {
391 ltMask |= 1<<k;
392 break;
393 }
394 default: {
395 /* CANT_HAPPEN */
396 assert( 0 );
397 break;
398 }
399 }
drh75897232000-05-29 14:26:00 +0000400 break;
401 }
402 }
403 }
404 }
drh487ab3c2001-11-08 00:45:21 +0000405 for(nEq=0; nEq<pIdx->nColumn; nEq++){
406 m = (1<<(nEq+1))-1;
407 if( (m & eqMask)!=m ) break;
408 }
409 score = nEq*4;
410 m = 1<<nEq;
411 if( m & ltMask ) score++;
412 if( m & gtMask ) score+=2;
413 if( score>bestScore ){
414 pBestIdx = pIdx;
415 bestScore = score;
drh75897232000-05-29 14:26:00 +0000416 }
417 }
drh6b563442001-11-07 16:48:26 +0000418 pWInfo->a[i].pIdx = pBestIdx;
drh487ab3c2001-11-08 00:45:21 +0000419 pWInfo->a[i].score = bestScore;
drh7e391e12000-05-30 20:17:49 +0000420 loopMask |= 1<<idx;
drh6b563442001-11-07 16:48:26 +0000421 if( pBestIdx ){
drh832508b2002-03-02 17:04:07 +0000422 pWInfo->a[i].iCur = pParse->nTab++;
423 pWInfo->peakNTab = pParse->nTab;
drh6b563442001-11-07 16:48:26 +0000424 }
drh75897232000-05-29 14:26:00 +0000425 }
426
drh6b563442001-11-07 16:48:26 +0000427 /* Open all tables in the pTabList and all indices used by those tables.
drh75897232000-05-29 14:26:00 +0000428 */
drhad3cab52002-05-24 02:04:32 +0000429 for(i=0; i<pTabList->nSrc; i++){
drhf57b3392001-10-08 13:22:32 +0000430 int openOp;
431 Table *pTab;
432
433 pTab = pTabList->a[i].pTab;
drha76b5df2002-02-23 02:32:10 +0000434 if( pTab->isTransient || pTab->pSelect ) continue;
drhf57b3392001-10-08 13:22:32 +0000435 openOp = pTab->isTemp ? OP_OpenAux : OP_Open;
drh99fcd712001-10-13 01:06:47 +0000436 sqliteVdbeAddOp(v, openOp, base+i, pTab->tnum);
437 sqliteVdbeChangeP3(v, -1, pTab->zName, P3_STATIC);
drh50e5dad2001-09-15 00:57:28 +0000438 if( i==0 && !pParse->schemaVerified &&
439 (pParse->db->flags & SQLITE_InTrans)==0 ){
drh99fcd712001-10-13 01:06:47 +0000440 sqliteVdbeAddOp(v, OP_VerifyCookie, pParse->db->schema_cookie, 0);
drh50e5dad2001-09-15 00:57:28 +0000441 pParse->schemaVerified = 1;
442 }
drh6b563442001-11-07 16:48:26 +0000443 if( pWInfo->a[i].pIdx!=0 ){
444 sqliteVdbeAddOp(v, openOp, pWInfo->a[i].iCur, pWInfo->a[i].pIdx->tnum);
445 sqliteVdbeChangeP3(v, -1, pWInfo->a[i].pIdx->zName, P3_STATIC);
drh75897232000-05-29 14:26:00 +0000446 }
447 }
448
449 /* Generate the code to do the search
450 */
drh75897232000-05-29 14:26:00 +0000451 loopMask = 0;
drhad3cab52002-05-24 02:04:32 +0000452 for(i=0; i<pTabList->nSrc; i++){
drh75897232000-05-29 14:26:00 +0000453 int j, k;
454 int idx = aOrder[i];
drhc4a3c772001-04-04 11:48:57 +0000455 Index *pIdx;
drh6b563442001-11-07 16:48:26 +0000456 WhereLevel *pLevel = &pWInfo->a[i];
drh75897232000-05-29 14:26:00 +0000457
drh8aff1012001-12-22 14:49:24 +0000458 pIdx = pLevel->pIdx;
459 if( i<ARRAYSIZE(iDirectEq) && iDirectEq[i]>=0 ){
460 /* Case 1: We can directly reference a single row using an
461 ** equality comparison against the ROWID field.
drhc4a3c772001-04-04 11:48:57 +0000462 */
drh8aff1012001-12-22 14:49:24 +0000463 k = iDirectEq[i];
464 assert( k<nExpr );
465 assert( aExpr[k].p!=0 );
466 assert( aExpr[k].idxLeft==idx || aExpr[k].idxRight==idx );
467 if( aExpr[k].idxLeft==idx ){
468 sqliteExprCode(pParse, aExpr[k].p->pRight);
469 }else{
470 sqliteExprCode(pParse, aExpr[k].p->pLeft);
drhc4a3c772001-04-04 11:48:57 +0000471 }
drh8aff1012001-12-22 14:49:24 +0000472 aExpr[k].p = 0;
drh6b563442001-11-07 16:48:26 +0000473 brk = pLevel->brk = sqliteVdbeMakeLabel(v);
474 cont = pLevel->cont = brk;
drh8aff1012001-12-22 14:49:24 +0000475 sqliteVdbeAddOp(v, OP_MustBeInt, 0, brk);
drhad3cab52002-05-24 02:04:32 +0000476 if( i==pTabList->nSrc-1 && pushKey ){
drh97665872002-02-13 23:22:53 +0000477 /* Note: The OP_Dup below will cause the recno to be left on the
478 ** stack if the record does not exists and the OP_NotExists jump is
drh6b125452002-01-28 15:53:03 +0000479 ** taken. This violates a general rule of the VDBE that you should
480 ** never leave values on the stack in order to avoid a stack overflow.
481 ** But in this case, the OP_Dup will never happen inside of a loop,
drh97665872002-02-13 23:22:53 +0000482 ** because the pushKey flag is only true for UPDATE and DELETE, not
483 ** for SELECT, and nested loops only occur on a SELECT.
484 ** So it is safe to leave the recno on the stack.
drh6b125452002-01-28 15:53:03 +0000485 */
drhc4a3c772001-04-04 11:48:57 +0000486 haveKey = 1;
drh6b125452002-01-28 15:53:03 +0000487 sqliteVdbeAddOp(v, OP_Dup, 0, 0);
drhc4a3c772001-04-04 11:48:57 +0000488 }else{
drhc4a3c772001-04-04 11:48:57 +0000489 haveKey = 0;
490 }
drh6b125452002-01-28 15:53:03 +0000491 sqliteVdbeAddOp(v, OP_NotExists, base+idx, brk);
drh6b563442001-11-07 16:48:26 +0000492 pLevel->op = OP_Noop;
drh8aff1012001-12-22 14:49:24 +0000493 }else if( pIdx!=0 && pLevel->score%4==0 ){
494 /* Case 2: All index constraints are equality operators.
drh75897232000-05-29 14:26:00 +0000495 */
drh6b563442001-11-07 16:48:26 +0000496 int start;
drh487ab3c2001-11-08 00:45:21 +0000497 int testOp;
498 int nColumn = pLevel->score/4;
499 for(j=0; j<nColumn; j++){
drh75897232000-05-29 14:26:00 +0000500 for(k=0; k<nExpr; k++){
501 if( aExpr[k].p==0 ) continue;
502 if( aExpr[k].idxLeft==idx
drh487ab3c2001-11-08 00:45:21 +0000503 && aExpr[k].p->op==TK_EQ
drh75897232000-05-29 14:26:00 +0000504 && (aExpr[k].prereqRight & loopMask)==aExpr[k].prereqRight
drh967e8b72000-06-21 13:59:10 +0000505 && aExpr[k].p->pLeft->iColumn==pIdx->aiColumn[j]
drh75897232000-05-29 14:26:00 +0000506 ){
507 sqliteExprCode(pParse, aExpr[k].p->pRight);
508 aExpr[k].p = 0;
509 break;
510 }
511 if( aExpr[k].idxRight==idx
drh487ab3c2001-11-08 00:45:21 +0000512 && aExpr[k].p->op==TK_EQ
drh75897232000-05-29 14:26:00 +0000513 && (aExpr[k].prereqLeft & loopMask)==aExpr[k].prereqLeft
drh967e8b72000-06-21 13:59:10 +0000514 && aExpr[k].p->pRight->iColumn==pIdx->aiColumn[j]
drh75897232000-05-29 14:26:00 +0000515 ){
516 sqliteExprCode(pParse, aExpr[k].p->pLeft);
517 aExpr[k].p = 0;
518 break;
519 }
520 }
521 }
drh6b563442001-11-07 16:48:26 +0000522 pLevel->iMem = pParse->nMem++;
523 brk = pLevel->brk = sqliteVdbeMakeLabel(v);
524 cont = pLevel->cont = sqliteVdbeMakeLabel(v);
drh487ab3c2001-11-08 00:45:21 +0000525 sqliteVdbeAddOp(v, OP_MakeKey, nColumn, 0);
526 if( nColumn==pIdx->nColumn ){
527 sqliteVdbeAddOp(v, OP_MemStore, pLevel->iMem, 0);
528 testOp = OP_IdxGT;
529 }else{
530 sqliteVdbeAddOp(v, OP_Dup, 0, 0);
531 sqliteVdbeAddOp(v, OP_IncrKey, 0, 0);
532 sqliteVdbeAddOp(v, OP_MemStore, pLevel->iMem, 1);
533 testOp = OP_IdxGE;
534 }
drh6b563442001-11-07 16:48:26 +0000535 sqliteVdbeAddOp(v, OP_MoveTo, pLevel->iCur, brk);
536 start = sqliteVdbeAddOp(v, OP_MemLoad, pLevel->iMem, 0);
drh487ab3c2001-11-08 00:45:21 +0000537 sqliteVdbeAddOp(v, testOp, pLevel->iCur, brk);
drh6b563442001-11-07 16:48:26 +0000538 sqliteVdbeAddOp(v, OP_IdxRecno, pLevel->iCur, 0);
drhad3cab52002-05-24 02:04:32 +0000539 if( i==pTabList->nSrc-1 && pushKey ){
drh75897232000-05-29 14:26:00 +0000540 haveKey = 1;
541 }else{
drh99fcd712001-10-13 01:06:47 +0000542 sqliteVdbeAddOp(v, OP_MoveTo, base+idx, 0);
drh75897232000-05-29 14:26:00 +0000543 haveKey = 0;
544 }
drh6b563442001-11-07 16:48:26 +0000545 pLevel->op = OP_Next;
546 pLevel->p1 = pLevel->iCur;
547 pLevel->p2 = start;
drh8aff1012001-12-22 14:49:24 +0000548 }else if( i<ARRAYSIZE(iDirectLt) && (iDirectLt[i]>=0 || iDirectGt[i]>=0) ){
549 /* Case 3: We have an inequality comparison against the ROWID field.
550 */
551 int testOp = OP_Noop;
552 int start;
553
554 brk = pLevel->brk = sqliteVdbeMakeLabel(v);
555 cont = pLevel->cont = sqliteVdbeMakeLabel(v);
556 if( iDirectGt[i]>=0 ){
557 k = iDirectGt[i];
558 assert( k<nExpr );
559 assert( aExpr[k].p!=0 );
560 assert( aExpr[k].idxLeft==idx || aExpr[k].idxRight==idx );
561 if( aExpr[k].idxLeft==idx ){
562 sqliteExprCode(pParse, aExpr[k].p->pRight);
563 }else{
564 sqliteExprCode(pParse, aExpr[k].p->pLeft);
565 }
566 sqliteVdbeAddOp(v, OP_MustBeInt, 0, brk);
567 if( aExpr[k].p->op==TK_LT || aExpr[k].p->op==TK_GT ){
568 sqliteVdbeAddOp(v, OP_AddImm, 1, 0);
569 }
570 sqliteVdbeAddOp(v, OP_MoveTo, base+idx, brk);
571 aExpr[k].p = 0;
572 }else{
573 sqliteVdbeAddOp(v, OP_Rewind, base+idx, brk);
574 }
575 if( iDirectLt[i]>=0 ){
576 k = iDirectLt[i];
577 assert( k<nExpr );
578 assert( aExpr[k].p!=0 );
579 assert( aExpr[k].idxLeft==idx || aExpr[k].idxRight==idx );
580 if( aExpr[k].idxLeft==idx ){
581 sqliteExprCode(pParse, aExpr[k].p->pRight);
582 }else{
583 sqliteExprCode(pParse, aExpr[k].p->pLeft);
584 }
585 sqliteVdbeAddOp(v, OP_MustBeInt, 0, sqliteVdbeCurrentAddr(v)+1);
586 pLevel->iMem = pParse->nMem++;
587 sqliteVdbeAddOp(v, OP_MemStore, pLevel->iMem, 0);
588 if( aExpr[k].p->op==TK_LT || aExpr[k].p->op==TK_GT ){
589 testOp = OP_Ge;
590 }else{
591 testOp = OP_Gt;
592 }
593 aExpr[k].p = 0;
594 }
595 start = sqliteVdbeCurrentAddr(v);
596 pLevel->op = OP_Next;
597 pLevel->p1 = base+idx;
598 pLevel->p2 = start;
599 if( testOp!=OP_Noop ){
600 sqliteVdbeAddOp(v, OP_Recno, base+idx, 0);
601 sqliteVdbeAddOp(v, OP_MemLoad, pLevel->iMem, 0);
602 sqliteVdbeAddOp(v, testOp, 0, brk);
603 }
604 haveKey = 0;
605 }else if( pIdx==0 ){
606 /* Case 4: There was no usable index. We must do a complete
607 ** scan of the entire database table.
608 */
609 int start;
610
611 brk = pLevel->brk = sqliteVdbeMakeLabel(v);
612 cont = pLevel->cont = sqliteVdbeMakeLabel(v);
613 sqliteVdbeAddOp(v, OP_Rewind, base+idx, brk);
614 start = sqliteVdbeCurrentAddr(v);
615 pLevel->op = OP_Next;
616 pLevel->p1 = base+idx;
617 pLevel->p2 = start;
618 haveKey = 0;
drh487ab3c2001-11-08 00:45:21 +0000619 }else{
drhaacc5432002-01-06 17:07:40 +0000620 /* Case 5: The contraint on the right-most index field is
621 ** an inequality.
drh487ab3c2001-11-08 00:45:21 +0000622 */
623 int score = pLevel->score;
624 int nEqColumn = score/4;
625 int start;
626 int leFlag, geFlag;
627 int testOp;
628
629 /* Evaluate the equality constraints
630 */
631 for(j=0; j<nEqColumn; j++){
632 for(k=0; k<nExpr; k++){
633 if( aExpr[k].p==0 ) continue;
634 if( aExpr[k].idxLeft==idx
635 && aExpr[k].p->op==TK_EQ
636 && (aExpr[k].prereqRight & loopMask)==aExpr[k].prereqRight
637 && aExpr[k].p->pLeft->iColumn==pIdx->aiColumn[j]
638 ){
639 sqliteExprCode(pParse, aExpr[k].p->pRight);
640 aExpr[k].p = 0;
641 break;
642 }
643 if( aExpr[k].idxRight==idx
644 && aExpr[k].p->op==TK_EQ
645 && (aExpr[k].prereqLeft & loopMask)==aExpr[k].prereqLeft
646 && aExpr[k].p->pRight->iColumn==pIdx->aiColumn[j]
647 ){
648 sqliteExprCode(pParse, aExpr[k].p->pLeft);
649 aExpr[k].p = 0;
650 break;
651 }
652 }
653 }
654
655 /* Duplicate the equality contraint values because they will all be
656 ** used twice: once to make the termination key and once to make the
657 ** start key.
658 */
659 for(j=0; j<nEqColumn; j++){
660 sqliteVdbeAddOp(v, OP_Dup, nEqColumn-1, 0);
661 }
662
663 /* Generate the termination key. This is the key value that
664 ** will end the search. There is no termination key if there
665 ** are no equality contraints and no "X<..." constraint.
666 */
667 if( (score & 1)!=0 ){
668 for(k=0; k<nExpr; k++){
669 Expr *pExpr = aExpr[k].p;
670 if( pExpr==0 ) continue;
671 if( aExpr[k].idxLeft==idx
672 && (pExpr->op==TK_LT || pExpr->op==TK_LE)
673 && (aExpr[k].prereqRight & loopMask)==aExpr[k].prereqRight
674 && pExpr->pLeft->iColumn==pIdx->aiColumn[j]
675 ){
676 sqliteExprCode(pParse, pExpr->pRight);
677 leFlag = pExpr->op==TK_LE;
678 aExpr[k].p = 0;
679 break;
680 }
681 if( aExpr[k].idxRight==idx
682 && (pExpr->op==TK_GT || pExpr->op==TK_GE)
683 && (aExpr[k].prereqLeft & loopMask)==aExpr[k].prereqLeft
684 && pExpr->pRight->iColumn==pIdx->aiColumn[j]
685 ){
686 sqliteExprCode(pParse, pExpr->pLeft);
687 leFlag = pExpr->op==TK_GE;
688 aExpr[k].p = 0;
689 break;
690 }
691 }
692 testOp = OP_IdxGE;
693 }else{
694 testOp = nEqColumn>0 ? OP_IdxGE : OP_Noop;
695 leFlag = 1;
696 }
697 if( testOp!=OP_Noop ){
698 pLevel->iMem = pParse->nMem++;
699 sqliteVdbeAddOp(v, OP_MakeKey, nEqColumn + (score & 1), 0);
700 if( leFlag ){
701 sqliteVdbeAddOp(v, OP_IncrKey, 0, 0);
702 }
703 sqliteVdbeAddOp(v, OP_MemStore, pLevel->iMem, 1);
704 }
705
706 /* Generate the start key. This is the key that defines the lower
707 ** bound on the search. There is no start key if there are not
708 ** equality constraints and if there is no "X>..." constraint. In
709 ** that case, generate a "Rewind" instruction in place of the
710 ** start key search.
711 */
712 if( (score & 2)!=0 ){
713 for(k=0; k<nExpr; k++){
714 Expr *pExpr = aExpr[k].p;
715 if( pExpr==0 ) continue;
716 if( aExpr[k].idxLeft==idx
717 && (pExpr->op==TK_GT || pExpr->op==TK_GE)
718 && (aExpr[k].prereqRight & loopMask)==aExpr[k].prereqRight
719 && pExpr->pLeft->iColumn==pIdx->aiColumn[j]
720 ){
721 sqliteExprCode(pParse, pExpr->pRight);
722 geFlag = pExpr->op==TK_GE;
723 aExpr[k].p = 0;
724 break;
725 }
726 if( aExpr[k].idxRight==idx
727 && (pExpr->op==TK_LT || pExpr->op==TK_LE)
728 && (aExpr[k].prereqLeft & loopMask)==aExpr[k].prereqLeft
729 && pExpr->pRight->iColumn==pIdx->aiColumn[j]
730 ){
731 sqliteExprCode(pParse, pExpr->pLeft);
732 geFlag = pExpr->op==TK_LE;
733 aExpr[k].p = 0;
734 break;
735 }
736 }
drh7900ead2001-11-12 13:51:43 +0000737 }else{
738 geFlag = 1;
drh487ab3c2001-11-08 00:45:21 +0000739 }
740 brk = pLevel->brk = sqliteVdbeMakeLabel(v);
741 cont = pLevel->cont = sqliteVdbeMakeLabel(v);
742 if( nEqColumn>0 || (score&2)!=0 ){
743 sqliteVdbeAddOp(v, OP_MakeKey, nEqColumn + ((score&2)!=0), 0);
744 if( !geFlag ){
745 sqliteVdbeAddOp(v, OP_IncrKey, 0, 0);
746 }
747 sqliteVdbeAddOp(v, OP_MoveTo, pLevel->iCur, brk);
748 }else{
749 sqliteVdbeAddOp(v, OP_Rewind, pLevel->iCur, brk);
750 }
751
752 /* Generate the the top of the loop. If there is a termination
753 ** key we have to test for that key and abort at the top of the
754 ** loop.
755 */
756 start = sqliteVdbeCurrentAddr(v);
757 if( testOp!=OP_Noop ){
758 sqliteVdbeAddOp(v, OP_MemLoad, pLevel->iMem, 0);
759 sqliteVdbeAddOp(v, testOp, pLevel->iCur, brk);
760 }
761 sqliteVdbeAddOp(v, OP_IdxRecno, pLevel->iCur, 0);
drhad3cab52002-05-24 02:04:32 +0000762 if( i==pTabList->nSrc-1 && pushKey ){
drh487ab3c2001-11-08 00:45:21 +0000763 haveKey = 1;
764 }else{
765 sqliteVdbeAddOp(v, OP_MoveTo, base+idx, 0);
766 haveKey = 0;
767 }
768
769 /* Record the instruction used to terminate the loop.
770 */
771 pLevel->op = OP_Next;
772 pLevel->p1 = pLevel->iCur;
773 pLevel->p2 = start;
drh75897232000-05-29 14:26:00 +0000774 }
775 loopMask |= 1<<idx;
776
777 /* Insert code to test every subexpression that can be completely
778 ** computed using the current set of tables.
779 */
780 for(j=0; j<nExpr; j++){
781 if( aExpr[j].p==0 ) continue;
drh3f6b5482002-04-02 13:26:10 +0000782 if( (aExpr[j].prereqAll & loopMask)!=aExpr[j].prereqAll ) continue;
drh75897232000-05-29 14:26:00 +0000783 if( haveKey ){
drh573bd272001-02-19 23:23:38 +0000784 haveKey = 0;
drh99fcd712001-10-13 01:06:47 +0000785 sqliteVdbeAddOp(v, OP_MoveTo, base+idx, 0);
drh75897232000-05-29 14:26:00 +0000786 }
787 sqliteExprIfFalse(pParse, aExpr[j].p, cont);
788 aExpr[j].p = 0;
789 }
790 brk = cont;
791 }
792 pWInfo->iContinue = cont;
793 if( pushKey && !haveKey ){
drh99fcd712001-10-13 01:06:47 +0000794 sqliteVdbeAddOp(v, OP_Recno, base, 0);
drh75897232000-05-29 14:26:00 +0000795 }
796 sqliteFree(aOrder);
797 return pWInfo;
798}
799
800/*
801** Generate the end of the WHERE loop.
802*/
803void sqliteWhereEnd(WhereInfo *pWInfo){
804 Vdbe *v = pWInfo->pParse->pVdbe;
drh19a775c2000-06-05 18:54:46 +0000805 int i;
drh19a775c2000-06-05 18:54:46 +0000806 int base = pWInfo->base;
drh6b563442001-11-07 16:48:26 +0000807 WhereLevel *pLevel;
drhad3cab52002-05-24 02:04:32 +0000808 SrcList *pTabList = pWInfo->pTabList;
drh19a775c2000-06-05 18:54:46 +0000809
drhad3cab52002-05-24 02:04:32 +0000810 for(i=pTabList->nSrc-1; i>=0; i--){
drh6b563442001-11-07 16:48:26 +0000811 pLevel = &pWInfo->a[i];
812 sqliteVdbeResolveLabel(v, pLevel->cont);
813 if( pLevel->op!=OP_Noop ){
814 sqliteVdbeAddOp(v, pLevel->op, pLevel->p1, pLevel->p2);
drh19a775c2000-06-05 18:54:46 +0000815 }
drh6b563442001-11-07 16:48:26 +0000816 sqliteVdbeResolveLabel(v, pLevel->brk);
drh19a775c2000-06-05 18:54:46 +0000817 }
drh6b563442001-11-07 16:48:26 +0000818 sqliteVdbeResolveLabel(v, pWInfo->iBreak);
drhad3cab52002-05-24 02:04:32 +0000819 for(i=0; i<pTabList->nSrc; i++){
drh22f70c32002-02-18 01:17:00 +0000820 if( pTabList->a[i].pTab->isTransient ) continue;
drh6b563442001-11-07 16:48:26 +0000821 pLevel = &pWInfo->a[i];
822 sqliteVdbeAddOp(v, OP_Close, base+i, 0);
823 if( pLevel->pIdx!=0 ){
824 sqliteVdbeAddOp(v, OP_Close, pLevel->iCur, 0);
825 }
drh19a775c2000-06-05 18:54:46 +0000826 }
drh832508b2002-03-02 17:04:07 +0000827 if( pWInfo->pParse->nTab==pWInfo->peakNTab ){
828 pWInfo->pParse->nTab = pWInfo->savedNTab;
829 }
drh75897232000-05-29 14:26:00 +0000830 sqliteFree(pWInfo);
831 return;
832}