OLD | NEW |
| (Empty) |
1 # 2010 August 28 | |
2 # | |
3 # The author disclaims copyright to this source code. In place of | |
4 # a legal notice, here is a blessing: | |
5 # | |
6 # May you do good and not evil. | |
7 # May you find forgiveness for yourself and forgive others. | |
8 # May you share freely, never taking more than you give. | |
9 # | |
10 #*********************************************************************** | |
11 # This file contains tests for the r-tree module. Specifically, it tests | |
12 # that new-style custom r-tree queries (geometry callbacks) work. | |
13 # | |
14 | |
15 if {![info exists testdir]} { | |
16 set testdir [file join [file dirname [info script]] .. .. test] | |
17 } | |
18 source $testdir/tester.tcl | |
19 ifcapable !rtree { finish_test ; return } | |
20 ifcapable rtree_int_only { finish_test; return } | |
21 | |
22 | |
23 #------------------------------------------------------------------------- | |
24 # Test the example 2d "circle" geometry callback. | |
25 # | |
26 register_circle_geom db | |
27 | |
28 do_execsql_test rtreeE-1.1 { | |
29 PRAGMA page_size=512; | |
30 CREATE VIRTUAL TABLE rt1 USING rtree(id,x0,x1,y0,y1); | |
31 | |
32 /* A tight pattern of small boxes near 0,0 */ | |
33 WITH RECURSIVE | |
34 x(x) AS (VALUES(0) UNION ALL SELECT x+1 FROM x WHERE x<4), | |
35 y(y) AS (VALUES(0) UNION ALL SELECT y+1 FROM y WHERE y<4) | |
36 INSERT INTO rt1 SELECT x+5*y, x, x+2, y, y+2 FROM x, y; | |
37 | |
38 /* A looser pattern of small boxes near 100, 0 */ | |
39 WITH RECURSIVE | |
40 x(x) AS (VALUES(0) UNION ALL SELECT x+1 FROM x WHERE x<4), | |
41 y(y) AS (VALUES(0) UNION ALL SELECT y+1 FROM y WHERE y<4) | |
42 INSERT INTO rt1 SELECT 100+x+5*y, x*3+100, x*3+102, y*3, y*3+2 FROM x, y; | |
43 | |
44 /* A looser pattern of larger boxes near 0, 200 */ | |
45 WITH RECURSIVE | |
46 x(x) AS (VALUES(0) UNION ALL SELECT x+1 FROM x WHERE x<4), | |
47 y(y) AS (VALUES(0) UNION ALL SELECT y+1 FROM y WHERE y<4) | |
48 INSERT INTO rt1 SELECT 200+x+5*y, x*7, x*7+15, y*7+200, y*7+215 FROM x, y; | |
49 } {} | |
50 | |
51 # Queries against each of the three clusters */ | |
52 do_execsql_test rtreeE-1.1 { | |
53 SELECT id FROM rt1 WHERE id MATCH Qcircle(0.0, 0.0, 50.0, 3) ORDER BY id; | |
54 } {0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24} | |
55 do_execsql_test rtreeE-1.1x { | |
56 SELECT id FROM rt1 WHERE id MATCH Qcircle('x:0 y:0 r:50.0 e:3') ORDER BY id; | |
57 } {0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24} | |
58 do_execsql_test rtreeE-1.2 { | |
59 SELECT id FROM rt1 WHERE id MATCH Qcircle(100.0, 0.0, 50.0, 3) ORDER BY id; | |
60 } {100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 1
19 120 121 122 123 124} | |
61 do_execsql_test rtreeE-1.3 { | |
62 SELECT id FROM rt1 WHERE id MATCH Qcircle(0.0, 200.0, 50.0, 3) ORDER BY id; | |
63 } {200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 2
19 220 221 222 223 224} | |
64 | |
65 # The Qcircle geometry function gives a lower score to larger leaf-nodes. | |
66 # This causes the 200s to sort before the 100s and the 0s to sort before | |
67 # last. | |
68 # | |
69 do_execsql_test rtreeE-1.4 { | |
70 SELECT id FROM rt1 WHERE id MATCH Qcircle('r:1000 e:3') AND id%100==0 | |
71 } {200 100 0} | |
72 | |
73 # Exclude odd rowids on a depth-first search | |
74 do_execsql_test rtreeE-1.5 { | |
75 SELECT id FROM rt1 WHERE id MATCH Qcircle('r:1000 e:4') ORDER BY +id | |
76 } {0 2 4 6 8 10 12 14 16 18 20 22 24 100 102 104 106 108 110 112 114 116 118 120
122 124 200 202 204 206 208 210 212 214 216 218 220 222 224} | |
77 | |
78 # Exclude odd rowids on a breadth-first search. | |
79 do_execsql_test rtreeE-1.6 { | |
80 SELECT id FROM rt1 WHERE id MATCH Qcircle(0,0,1000,5) ORDER BY +id | |
81 } {0 2 4 6 8 10 12 14 16 18 20 22 24 100 102 104 106 108 110 112 114 116 118 120
122 124 200 202 204 206 208 210 212 214 216 218 220 222 224} | |
82 | |
83 # Test that rtree prefers MATCH to lookup-by-rowid. | |
84 # | |
85 do_execsql_test rtreeE-1.7 { | |
86 SELECT id FROM rt1 WHERE id=18 AND id MATCH Qcircle(0,0,1000,5) | |
87 } {18} | |
88 | |
89 | |
90 # Construct a large 2-D RTree with thousands of random entries. | |
91 # | |
92 do_test rtreeE-2.1 { | |
93 db eval { | |
94 CREATE TABLE t2(id,x0,x1,y0,y1); | |
95 CREATE VIRTUAL TABLE rt2 USING rtree(id,x0,x1,y0,y1); | |
96 BEGIN; | |
97 } | |
98 expr srand(0) | |
99 for {set i 1} {$i<=10000} {incr i} { | |
100 set dx [expr {int(rand()*40)+1}] | |
101 set dy [expr {int(rand()*40)+1}] | |
102 set x0 [expr {int(rand()*(10000 - $dx))}] | |
103 set x1 [expr {$x0+$dx}] | |
104 set y0 [expr {int(rand()*(10000 - $dy))}] | |
105 set y1 [expr {$y0+$dy}] | |
106 set id [expr {$i+10000}] | |
107 db eval {INSERT INTO t2 VALUES($id,$x0,$x1,$y0,$y1)} | |
108 } | |
109 db eval { | |
110 INSERT INTO rt2 SELECT * FROM t2; | |
111 COMMIT; | |
112 } | |
113 } {} | |
114 | |
115 for {set i 1} {$i<=200} {incr i} { | |
116 set dx [expr {int(rand()*100)}] | |
117 set dy [expr {int(rand()*100)}] | |
118 set x0 [expr {int(rand()*(10000 - $dx))}] | |
119 set x1 [expr {$x0+$dx}] | |
120 set y0 [expr {int(rand()*(10000 - $dy))}] | |
121 set y1 [expr {$y0+$dy}] | |
122 set ans [db eval {SELECT id FROM t2 WHERE x1>=$x0 AND x0<=$x1 AND y1>=$y0 AND
y0<=$y1 ORDER BY id}] | |
123 do_execsql_test rtreeE-2.2.$i { | |
124 SELECT id FROM rt2 WHERE id MATCH breadthfirstsearch($x0,$x1,$y0,$y1) ORDER
BY id | |
125 } $ans | |
126 } | |
127 | |
128 # Run query that have very deep priority queues | |
129 # | |
130 set ans [db eval {SELECT id FROM t2 WHERE x1>=0 AND x0<=5000 AND y1>=0 AND y0<=5
000 ORDER BY id}] | |
131 do_execsql_test rtreeE-2.3 { | |
132 SELECT id FROM rt2 WHERE id MATCH breadthfirstsearch(0,5000,0,5000) ORDER BY i
d | |
133 } $ans | |
134 set ans [db eval {SELECT id FROM t2 WHERE x1>=0 AND x0<=10000 AND y1>=0 AND y0<=
10000 ORDER BY id}] | |
135 do_execsql_test rtreeE-2.4 { | |
136 SELECT id FROM rt2 WHERE id MATCH breadthfirstsearch(0,10000,0,10000) ORDER BY
id | |
137 } $ans | |
138 | |
139 | |
140 finish_test | |
OLD | NEW |