OLD | NEW |
1 /* | 1 /* |
2 ** 2010 August 28 | 2 ** 2010 August 28 |
3 ** | 3 ** |
4 ** The author disclaims copyright to this source code. In place of | 4 ** The author disclaims copyright to this source code. In place of |
5 ** a legal notice, here is a blessing: | 5 ** a legal notice, here is a blessing: |
6 ** | 6 ** |
7 ** May you do good and not evil. | 7 ** May you do good and not evil. |
8 ** May you find forgiveness for yourself and forgive others. | 8 ** May you find forgiveness for yourself and forgive others. |
9 ** May you share freely, never taking more than you give. | 9 ** May you share freely, never taking more than you give. |
10 ** | 10 ** |
11 ************************************************************************* | 11 ************************************************************************* |
12 ** Code for testing all sorts of SQLite interfaces. This code | 12 ** Code for testing all sorts of SQLite interfaces. This code |
13 ** is not included in the SQLite library. | 13 ** is not included in the SQLite library. |
14 */ | 14 */ |
15 | 15 |
16 #include <sqlite3.h> | 16 #include "sqlite3.h" |
17 #include <tcl.h> | 17 #include <tcl.h> |
18 | 18 |
19 /* Solely for the UNUSED_PARAMETER() macro. */ | 19 /* Solely for the UNUSED_PARAMETER() macro. */ |
20 #include "sqliteInt.h" | 20 #include "sqliteInt.h" |
21 | 21 |
22 #ifdef SQLITE_ENABLE_RTREE | 22 #ifdef SQLITE_ENABLE_RTREE |
23 /* | 23 /* |
24 ** Type used to cache parameter information for the "circle" r-tree geometry | 24 ** Type used to cache parameter information for the "circle" r-tree geometry |
25 ** callback. | 25 ** callback. |
26 */ | 26 */ |
(...skipping 121 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
148 | 148 |
149 /* The specified bounding box does not intersect the circular region. Set | 149 /* The specified bounding box does not intersect the circular region. Set |
150 ** the output variable to zero and return SQLITE_OK. */ | 150 ** the output variable to zero and return SQLITE_OK. */ |
151 *pRes = 0; | 151 *pRes = 0; |
152 return SQLITE_OK; | 152 return SQLITE_OK; |
153 } | 153 } |
154 | 154 |
155 /* | 155 /* |
156 ** Implementation of "circle" r-tree geometry callback using the | 156 ** Implementation of "circle" r-tree geometry callback using the |
157 ** 2nd-generation interface that allows scoring. | 157 ** 2nd-generation interface that allows scoring. |
| 158 ** |
| 159 ** Two calling forms: |
| 160 ** |
| 161 ** Qcircle(X,Y,Radius,eType) -- All values are doubles |
| 162 ** Qcircle('x:X y:Y r:R e:ETYPE') -- Single string parameter |
158 */ | 163 */ |
159 static int circle_query_func(sqlite3_rtree_query_info *p){ | 164 static int circle_query_func(sqlite3_rtree_query_info *p){ |
160 int i; /* Iterator variable */ | 165 int i; /* Iterator variable */ |
161 Circle *pCircle; /* Structure defining circular region */ | 166 Circle *pCircle; /* Structure defining circular region */ |
162 double xmin, xmax; /* X dimensions of box being tested */ | 167 double xmin, xmax; /* X dimensions of box being tested */ |
163 double ymin, ymax; /* X dimensions of box being tested */ | 168 double ymin, ymax; /* X dimensions of box being tested */ |
164 int nWithin = 0; /* Number of corners inside the circle */ | 169 int nWithin = 0; /* Number of corners inside the circle */ |
165 | 170 |
166 xmin = p->aCoord[0]; | 171 xmin = p->aCoord[0]; |
167 xmax = p->aCoord[1]; | 172 xmax = p->aCoord[1]; |
168 ymin = p->aCoord[2]; | 173 ymin = p->aCoord[2]; |
169 ymax = p->aCoord[3]; | 174 ymax = p->aCoord[3]; |
170 pCircle = (Circle *)p->pUser; | 175 pCircle = (Circle *)p->pUser; |
171 if( pCircle==0 ){ | 176 if( pCircle==0 ){ |
172 /* If pUser is still 0, then the parameter values have not been tested | 177 /* If pUser is still 0, then the parameter values have not been tested |
173 ** for correctness or stored into a Circle structure yet. Do this now. */ | 178 ** for correctness or stored into a Circle structure yet. Do this now. */ |
174 | 179 |
175 /* This geometry callback is for use with a 2-dimensional r-tree table. | 180 /* This geometry callback is for use with a 2-dimensional r-tree table. |
176 ** Return an error if the table does not have exactly 2 dimensions. */ | 181 ** Return an error if the table does not have exactly 2 dimensions. */ |
177 if( p->nCoord!=4 ) return SQLITE_ERROR; | 182 if( p->nCoord!=4 ) return SQLITE_ERROR; |
178 | 183 |
179 /* Test that the correct number of parameters (4) have been supplied, | 184 /* Test that the correct number of parameters (1 or 4) have been supplied. |
180 ** and that the parameters are in range (that the radius of the circle | 185 */ |
181 ** radius is greater than zero). */ | 186 if( p->nParam!=4 && p->nParam!=1 ) return SQLITE_ERROR; |
182 if( p->nParam!=4 || p->aParam[2]<0.0 ) return SQLITE_ERROR; | |
183 | 187 |
184 /* Allocate a structure to cache parameter data in. Return SQLITE_NOMEM | 188 /* Allocate a structure to cache parameter data in. Return SQLITE_NOMEM |
185 ** if the allocation fails. */ | 189 ** if the allocation fails. */ |
186 pCircle = (Circle *)(p->pUser = sqlite3_malloc(sizeof(Circle))); | 190 pCircle = (Circle *)(p->pUser = sqlite3_malloc(sizeof(Circle))); |
187 if( !pCircle ) return SQLITE_NOMEM; | 191 if( !pCircle ) return SQLITE_NOMEM; |
188 p->xDelUser = circle_del; | 192 p->xDelUser = circle_del; |
189 | 193 |
190 /* Record the center and radius of the circular region. One way that | 194 /* Record the center and radius of the circular region. One way that |
191 ** tested bounding boxes that intersect the circular region are detected | 195 ** tested bounding boxes that intersect the circular region are detected |
192 ** is by testing if each corner of the bounding box lies within radius | 196 ** is by testing if each corner of the bounding box lies within radius |
193 ** units of the center of the circle. */ | 197 ** units of the center of the circle. */ |
194 pCircle->centerx = p->aParam[0]; | 198 if( p->nParam==4 ){ |
195 pCircle->centery = p->aParam[1]; | 199 pCircle->centerx = p->aParam[0]; |
196 pCircle->radius = p->aParam[2]; | 200 pCircle->centery = p->aParam[1]; |
197 pCircle->eScoreType = (int)p->aParam[3]; | 201 pCircle->radius = p->aParam[2]; |
| 202 pCircle->eScoreType = (int)p->aParam[3]; |
| 203 }else{ |
| 204 const char *z = (const char*)sqlite3_value_text(p->apSqlParam[0]); |
| 205 pCircle->centerx = 0.0; |
| 206 pCircle->centery = 0.0; |
| 207 pCircle->radius = 0.0; |
| 208 pCircle->eScoreType = 0; |
| 209 while( z && z[0] ){ |
| 210 if( z[0]=='r' && z[1]==':' ){ |
| 211 pCircle->radius = atof(&z[2]); |
| 212 }else if( z[0]=='x' && z[1]==':' ){ |
| 213 pCircle->centerx = atof(&z[2]); |
| 214 }else if( z[0]=='y' && z[1]==':' ){ |
| 215 pCircle->centery = atof(&z[2]); |
| 216 }else if( z[0]=='e' && z[1]==':' ){ |
| 217 pCircle->eScoreType = (int)atof(&z[2]); |
| 218 }else if( z[0]==' ' ){ |
| 219 z++; |
| 220 continue; |
| 221 } |
| 222 while( z[0]!=0 && z[0]!=' ' ) z++; |
| 223 while( z[0]==' ' ) z++; |
| 224 } |
| 225 } |
| 226 if( pCircle->radius<0.0 ){ |
| 227 sqlite3_free(pCircle); |
| 228 return SQLITE_NOMEM; |
| 229 } |
198 | 230 |
199 /* Define two bounding box regions. The first, aBox[0], extends to | 231 /* Define two bounding box regions. The first, aBox[0], extends to |
200 ** infinity in the X dimension. It covers the same range of the Y dimension | 232 ** infinity in the X dimension. It covers the same range of the Y dimension |
201 ** as the circular region. The second, aBox[1], extends to infinity in | 233 ** as the circular region. The second, aBox[1], extends to infinity in |
202 ** the Y dimension and is constrained to the range of the circle in the | 234 ** the Y dimension and is constrained to the range of the circle in the |
203 ** X dimension. | 235 ** X dimension. |
204 ** | 236 ** |
205 ** Then imagine each box is split in half along its short axis by a line | 237 ** Then imagine each box is split in half along its short axis by a line |
206 ** that intersects the center of the circular region. A bounding box | 238 ** that intersects the center of the circular region. A bounding box |
207 ** being tested can be said to intersect the circular region if it contains | 239 ** being tested can be said to intersect the circular region if it contains |
(...skipping 254 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
462 Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_STATIC); | 494 Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_STATIC); |
463 #endif | 495 #endif |
464 return TCL_OK; | 496 return TCL_OK; |
465 } | 497 } |
466 | 498 |
467 int Sqlitetestrtree_Init(Tcl_Interp *interp){ | 499 int Sqlitetestrtree_Init(Tcl_Interp *interp){ |
468 Tcl_CreateObjCommand(interp, "register_cube_geom", register_cube_geom, 0, 0); | 500 Tcl_CreateObjCommand(interp, "register_cube_geom", register_cube_geom, 0, 0); |
469 Tcl_CreateObjCommand(interp, "register_circle_geom",register_circle_geom,0,0); | 501 Tcl_CreateObjCommand(interp, "register_circle_geom",register_circle_geom,0,0); |
470 return TCL_OK; | 502 return TCL_OK; |
471 } | 503 } |
OLD | NEW |