File: geo_center_sphere1.js

package info (click to toggle)
mongodb 1%3A2.0.6-1.1
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 17,204 kB
  • sloc: cpp: 109,783; ansic: 101,073; python: 2,287; perl: 395; makefile: 370; sh: 242; asm: 46
file content (93 lines) | stat: -rw-r--r-- 2,774 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

t = db.geo_center_sphere1;
t.drop();

skip = 3 // lower for more rigor, higher for more speed (tested with .5, .678, 1, 2, 3, and 4)

searches = [ 
    //  x , y    rad
    [ [ 5 , 0 ] , 0.05 ] , // ~200 miles
    [ [ 135 , 0 ] , 0.05 ] ,

    [ [ 5 , 70 ] , 0.05 ] ,
    [ [ 135 , 70 ] , 0.05 ] ,
    [ [ 5 , 85 ] , 0.05 ] ,

    [ [ 20 , 0 ] , 0.25 ] , // ~1000 miles
    [ [ 20 , -45 ] , 0.25 ] ,
    [ [ -20 , 60 ] , 0.25 ] ,
    [ [ -20 , -70 ] , 0.25 ] ,
];
correct = searches.map( function(z){ return []; } );

num = 0;

for ( x=-179; x<=179; x += skip ){
    for ( y=-89; y<=89; y += skip ){
        o = { _id : num++ , loc : [ x , y ] } 
        t.save( o )
        for ( i=0; i<searches.length; i++ ){
            if ( Geo.sphereDistance( [ x , y ] , searches[i][0] ) <= searches[i][1])
                correct[i].push( o );
        }
    }
    gc(); // needed with low skip values
}

t.ensureIndex( { loc : "2d" } );

for ( i=0; i<searches.length; i++ ){
    print('------------');
    print( tojson( searches[i] ) + "\t" + correct[i].length )
    q = { loc : { $within : { $centerSphere : searches[i] } } }

    //correct[i].forEach( printjson )
    //printjson( q );
    //t.find( q ).forEach( printjson )
    
    //printjson(t.find( q ).explain())

    //printjson( Array.sort( correct[i].map( function(z){ return z._id; } ) ) )
    //printjson( Array.sort( t.find(q).map( function(z){ return z._id; } ) ) )
    
    var numExpected = correct[i].length
    var x = correct[i].map( function(z){ return z._id; } )
    var y = t.find(q).map( function(z){ return z._id; } )

    missing = [];
    epsilon = 0.001; // allow tenth of a percent error due to conversions
    for (var j=0; j<x.length; j++){
        if (!Array.contains(y, x[j])){
            missing.push(x[j]);
            var obj = t.findOne({_id: x[j]});
            var dist = Geo.sphereDistance(searches[i][0], obj.loc);
            print("missing: " + tojson(obj) + " " + dist)
            if ((Math.abs(dist - searches[i][1]) / dist) < epsilon)
                numExpected -= 1;
        }
    }
    for (var j=0; j<y.length; j++){
        if (!Array.contains(x, y[j])){
            missing.push(y[j]);
            var obj = t.findOne({_id: y[j]});
            var dist = Geo.sphereDistance(searches[i][0], obj.loc);
            print("extra: " + tojson(obj) + " " + dist)
            if ((Math.abs(dist - searches[i][1]) / dist) < epsilon)
                numExpected += 1;
        }
    }


    assert.eq( numExpected , t.find( q ).itcount() , "itcount : " + tojson( searches[i] ) );
    assert.eq( numExpected , t.find( q ).count() , "count : " + tojson( searches[i] ) );
    assert.gt( numExpected * 2 , t.find(q).explain().nscanned , "nscanned : " + tojson( searches[i] ) )
}