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
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110

use crate::errors::*;
use crate::hlua::AnyLuaValue;
use crate::json::LuaJsonValue;
use geo::{LineString, Polygon, Coordinate};
use geo::prelude::*;
use serde::Deserialize;

#[derive(Debug, Deserialize)]
pub struct Point {
    lon: f64,
    lat: f64,
}

impl Point {
    pub fn try_from(x: AnyLuaValue) -> Result<Point> {
        let x = LuaJsonValue::from(x);
        let x = serde_json::from_value(x.into())?;
        Ok(x)
    }
}

pub fn polygon_contains(ring: &[Point], p: &Point) -> bool {
    let ring = ring.iter()
        .map(|p| Coordinate { x: p.lon, y: p.lat })
        .collect::<Vec<_>>();

    let polygon = Polygon::new(LineString::from(ring), vec![]);
    let point = geo::Point::new(p.lon, p.lat);

    polygon.contains(&point)
}

#[cfg(test)]
mod tests {
    use super::*;

    fn hamburg_polygon() -> &'static [Point] {
        &[
            Point { lat: 53.63975308945899,  lon: 9.764785766601562  },
            Point { lat: 53.59494998253459,  lon: 9.827270507812     },
            Point { lat: 53.663153974456456, lon: 9.9151611328125    },
            Point { lat: 53.65582987649682,  lon: 9.976272583007812  },
            Point { lat: 53.68613523817129,  lon: 9.992752075195312  },
            Point { lat: 53.68674518938816,  lon: 10.051460266113281 },
            Point { lat: 53.72495117617815,  lon: 10.075492858886719 },
            Point { lat: 53.71946627930625,  lon: 10.118408203125    },
            Point { lat: 53.743635083157756, lon: 10.164413452148438 },
            Point { lat: 53.73104466704585,  lon: 10.202865600585938 },
            Point { lat: 53.676781546441546, lon: 10.16304016113281  },
            Point { lat: 53.632832079199474, lon: 10.235824584960938 },
            Point { lat: 53.608803292930894, lon: 10.2008056640625   },
            Point { lat: 53.578646152866504, lon: 10.208358764648438 },
            Point { lat: 53.57212285981298,  lon: 10.163726806640625 },
            Point { lat: 53.52071674896369,  lon: 10.18707275390625  },
            Point { lat: 53.52643162253097,  lon: 10.224151611328125 },
            Point { lat: 53.44062753992289,  lon: 10.347747802734375 },
            Point { lat: 53.38824275010831,  lon: 10.248870849609375 },
            Point { lat: 53.38824275010831,  lon: 10.15960693359375  },
            Point { lat: 53.44635321212876,  lon: 10.064849853515625 },
            Point { lat: 53.40595029739904,  lon: 9.985198974609375  },
            Point { lat: 53.42385506057106,  lon: 9.951210021972656  },
            Point { lat: 53.41843327091211,  lon: 9.944171905517578  },
            Point { lat: 53.41812635648326,  lon: 9.927349090576172  },
            Point { lat: 53.412294561442884, lon: 9.917736053466797  },
            Point { lat: 53.41464783813818,  lon: 9.901256561279297  },
            Point { lat: 53.443490472483326, lon: 9.912586212158201  },
            Point { lat: 53.45177144115704,  lon: 9.897651672363281  },
            Point { lat: 53.43633277935392,  lon: 9.866924285888672  },
            Point { lat: 53.427639673754776, lon: 9.866409301757812  },
            Point { lat: 53.427639673754776, lon: 9.858856201171875  },
            Point { lat: 53.46710230573499,  lon: 9.795513153076172  },
            Point { lat: 53.49039461941655,  lon: 9.795341491699219  },
            Point { lat: 53.49029248806277,  lon: 9.77903366088867   },
            Point { lat: 53.49856433088649,  lon: 9.780235290527344  },
            Point { lat: 53.5078554643033,   lon: 9.758434295654297  },
            Point { lat: 53.545407634092975, lon: 9.759807586669922  },
            Point { lat: 53.568147234570084, lon: 9.633293151855469  },
            Point { lat: 53.58802162343514,  lon: 9.655780792236328  },
            Point { lat: 53.568351121879815, lon: 9.727706909179688  },
            Point { lat: 53.60921067445695,  lon: 9.737663269042969  },
        ]
    }

    #[test]
    fn test_polygon_hamburg_contains_hamburg() {
        let contains = polygon_contains(hamburg_polygon(), &Point {
            lat: 53.551085,
            lon: 9.993682,
        });
        assert!(contains);
    }

    #[test]
    fn test_polygon_hamburg_not_contains_berlin() {
        let contains = polygon_contains(hamburg_polygon(), &Point {
            lat: 52.52437,
            lon: 13.41053,
        });
        assert!(!contains);
    }

    #[test]
    fn test_polygon_hamburg_not_contains_ny() {
        let contains = polygon_contains(hamburg_polygon(), &Point {
            lat: 40.726662,
            lon: -74.036677,
        });
        assert!(!contains);
    }
}