jismesh 0.1.0

Utilities for handling and converting JIS X0410 area mesh codes.
Documentation 
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199

use super::*;
use crate::utils::meshcode::to_meshcode;
use ndarray::Array1;

/// Generate an envelope of mesh codes that cover the rectangular area
/// defined by the southwest and northeast mesh codes.
///
/// # Arguments
/// * `meshcode_sw` - Southwest mesh code
/// * `meshcode_ne` - Northeast mesh code
///
/// # Returns
/// * `Result<Array1<u64>>` - Array of mesh codes that cover the area
///
/// # Errors
/// * Returns an error if the mesh levels of the input codes don't match
pub fn to_envelope(meshcode_sw: u64, meshcode_ne: u64) -> Result<Array1<u64>> {
    // Get mesh levels for both codes
    let level_sw = to_meshlevel(&Array1::from_vec(vec![meshcode_sw]))?[0];
    let level_ne = to_meshlevel(&Array1::from_vec(vec![meshcode_ne]))?[0];

    // Check if the mesh levels match
    if level_sw != level_ne {
        return Err(JismeshError::MismatchedMeshLevels(level_sw, level_ne));
    }

    let margin_lat = 0.5;
    let margin_lon = 0.5;

    // Generate mesh points for southwest and northeast corners
    let sw_points = to_meshpoint(
        Array1::from_vec(vec![meshcode_sw]),
        Array1::from_vec(vec![margin_lat]),
        Array1::from_vec(vec![margin_lon]),
    )?;

    let ne_points = to_meshpoint(
        Array1::from_vec(vec![meshcode_ne]),
        Array1::from_vec(vec![1.0]),
        Array1::from_vec(vec![1.0]),
    )?;

    let lat_s = sw_points[[0, 0]];
    let lon_w = sw_points[[1, 0]];
    let lat_n = ne_points[[0, 0]];
    let lon_e = ne_points[[1, 0]];

    make_envelope(lat_s, lon_w, lat_n, lon_e, level_sw)
}

/// Generate mesh codes that intersect with the given mesh code at the specified level.
///
/// # Arguments
/// * `meshcode` - Mesh code to find intersections with
/// * `to_level` - Target mesh level for the intersection
///
/// # Returns
/// * `Result<Array1<u64>>` - Array of mesh codes that intersect with the input code
pub fn to_intersects(meshcode: u64, to_level: MeshLevel) -> Result<Array1<u64>> {
    // Get mesh level for the input code
    let from_level = to_meshlevel(&Array1::from_vec(vec![meshcode]))?[0];

    let from_unit_lat = unit_lat(from_level);
    let from_unit_lon = unit_lon(from_level);

    let to_unit_lat = unit_lat(to_level);
    let to_unit_lon = unit_lon(to_level);

    // Calculate margins based on the relative unit sizes
    let margin_lat = if to_unit_lat <= from_unit_lat {
        (to_unit_lat / from_unit_lat) / 2.0
    } else {
        0.5
    };

    let margin_lon = if to_unit_lon <= from_unit_lon {
        (to_unit_lon / from_unit_lon) / 2.0
    } else {
        0.5
    };

    // Generate mesh points for the original mesh code
    let from_points_sw = to_meshpoint(
        Array1::from_vec(vec![meshcode]),
        Array1::from_vec(vec![margin_lat]),
        Array1::from_vec(vec![margin_lon]),
    )?;

    let from_points_ne = to_meshpoint(
        Array1::from_vec(vec![meshcode]),
        Array1::from_vec(vec![1.0]),
        Array1::from_vec(vec![1.0]),
    )?;

    let from_lat_s = from_points_sw[[0, 0]];
    let from_lon_w = from_points_sw[[1, 0]];
    let from_lat_n = from_points_ne[[0, 0]];
    let from_lon_e = from_points_ne[[1, 0]];

    make_envelope(from_lat_s, from_lon_w, from_lat_n, from_lon_e, to_level)
}

/// Internal helper function to generate mesh codes within a bounding box
fn make_envelope(
    lat_s: f64,
    lon_w: f64,
    lat_n: f64,
    lon_e: f64,
    level: MeshLevel,
) -> Result<Array1<u64>> {
    let to_unit_lat = unit_lat(level);
    let to_unit_lon = unit_lon(level);

    // Calculate how many meshes we need in each direction
    let lat_count = ((lat_n - lat_s) / to_unit_lat).ceil() as usize;
    let lon_count = ((lon_e - lon_w) / to_unit_lon).ceil() as usize;
    let point_count = lat_count * lon_count;

    let mut lats = Vec::with_capacity(point_count);
    let mut lons = Vec::with_capacity(point_count);
    for i in 0..lat_count {
        let to_lat = lat_s + (i as f64 * to_unit_lat);

        // Generate all longitude points for this latitude
        for j in 0..lon_count {
            let to_lon = lon_w + (j as f64 * to_unit_lon);

            lats.push(to_lat);
            lons.push(to_lon);
        }
    }

    to_meshcode(&Array1::from_vec(lats), &Array1::from_vec(lons), level)
}

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

    #[test]
    fn test_to_envelope() {
        // Test with level 1 mesh codes for Tokyo area
        let meshcode_sw = 5339; // Southwest corner
        let meshcode_ne = 5339; // Same as SW for simple case

        let result = to_envelope(meshcode_sw, meshcode_ne).unwrap();
        assert_eq!(result.len(), 1);
        assert_eq!(result[0], 5339);

        // Test with level 2 mesh codes
        let meshcode_sw = 533900; // Southwest corner
        let meshcode_ne = 533901; // Northeast corner

        let result = to_envelope(meshcode_sw, meshcode_ne).unwrap();
        assert!(result.len() > 1);
        assert!(result.iter().any(|&x| x == 533900));
        assert!(result.iter().any(|&x| x == 533901));
    }

    #[test]
    fn test_to_intersects() {
        // Test conversion from level 1 to level 2
        let meshcode = 5339; // Level 1
        let to_level = MeshLevel::Lv2;

        let result = to_intersects(meshcode, to_level).unwrap();
        assert!(result.len() > 0);

        // All resulting codes should be level 2
        for code in result.iter() {
            let level = to_meshlevel(&array![*code]).unwrap();
            assert_eq!(level[0], MeshLevel::Lv2);
        }

        // Test conversion from level 2 to level 3
        let meshcode = 533900; // Level 2
        let to_level = MeshLevel::Lv3;

        let result = to_intersects(meshcode, to_level).unwrap();
        assert!(result.len() > 0);

        // All resulting codes should be level 3
        for code in result.iter() {
            let level = to_meshlevel(&array![*code]).unwrap();
            assert_eq!(level[0], MeshLevel::Lv3);
        }
    }

    #[test]
    fn test_error_mismatched_levels() {
        // Test with mismatched mesh levels
        let meshcode_sw = 5339; // Level 1
        let meshcode_ne = 533900; // Level 2

        let result = to_envelope(meshcode_sw, meshcode_ne);
        assert!(result.is_err());
    }
}