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

(function() {
"use strict";

const numTests = 31;

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

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

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

    const 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");

    let 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");

    assert(t.drop());

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

    assert.writeOK(t.insert({loc: [1, 3]}));  // Add a point that's in
    assert.commandWorked(t.createIndex({loc: "2d"}, {bits: 2 + n}));

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

    assert.writeOK(t.insert({loc: [5, 3]}));   // Add a point that's out right in the mouth opening
    assert.writeOK(t.insert({loc: [3, 7]}));   // Add a point above the center of the head
    assert.writeOK(t.insert({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");
}
})();