summaryrefslogtreecommitdiff
path: root/jstests/core/geo_polygon3.js
blob: ed8f040fa8d089b7bb6aa03100dbe55226b80ce0 (plain)
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
//
// Tests for polygon querying with varying levels of accuracy
//

var numTests = 31;

for (var n = 0; n < numTests; n++) {
    t = db.geo_polygon3;
    t.drop();

    num = 0;
    for (x = 1; x < 9; x++) {
        for (y = 1; y < 9; y++) {
            o = {
                _id: num++,
                loc: [x, y]
            };
            t.save(o);
        }
    }

    t.ensureIndex({loc: "2d"}, {bits: 2 + n});

    triangle = [[0, 0], [1, 1], [0, 2]];

    // Look at only a small slice of the data within a triangle
    assert.eq(1, t.find({loc: {"$within": {"$polygon": triangle}}}).itcount(), "Triangle Test");

    boxBounds = [[0, 0], [0, 10], [10, 10], [10, 0]];

    assert.eq(
        num, t.find({loc: {"$within": {"$polygon": boxBounds}}}).itcount(), "Bounding Box Test");

    // Look in a box much bigger than the one we have data in
    boxBounds = [[-100, -100], [-100, 100], [100, 100], [100, -100]];
    assert.eq(num,
              t.find({loc: {"$within": {"$polygon": boxBounds}}}).itcount(),
              "Big Bounding Box Test");

    t.drop();

    pacman = [
        [0, 2],
        [0, 4],
        [2, 6],
        [4, 6],  // Head
        [6, 4],
        [4, 3],
        [6, 2],  // Mouth
        [4, 0],
        [2, 0]  // Bottom
    ];

    t.save({loc: [1, 3]});  // Add a point that's in
    t.ensureIndex({loc: "2d"}, {bits: 2 + t});

    assert.eq(1, t.find({loc: {$within: {$polygon: pacman}}}).itcount(), "Pacman single point");

    t.save({loc: [5, 3]});   // Add a point that's out right in the mouth opening
    t.save({loc: [3, 7]});   // Add a point above the center of the head
    t.save({loc: [3, -1]});  // Add a point below the center of the bottom

    assert.eq(1, t.find({loc: {$within: {$polygon: pacman}}}).itcount(), "Pacman double point");
}