mirror of
https://github.com/FoggedLens/deflock.git
synced 2026-08-27 12:53:11 +02:00
slim down the data
This commit is contained in:
@@ -28,10 +28,17 @@ def terraform_rate_expression_to_minutes(rate_expression: str) -> Tuple[int, int
|
|||||||
else:
|
else:
|
||||||
raise ValueError(f"Unsupported time unit: {unit}")
|
raise ValueError(f"Unsupported time unit: {unit}")
|
||||||
|
|
||||||
UPDATE_RATE_MINS = terraform_rate_expression_to_minutes(os.getenv("UPDATE_RATE_MINS", "60"))
|
UPDATE_RATE_MINS = terraform_rate_expression_to_minutes(os.getenv("UPDATE_RATE_MINS", "rate(60 minutes)"))
|
||||||
GRACE_PERIOD_MINS = int(2) # XXX: set expiration a few minutes after in case put object takes a while
|
GRACE_PERIOD_MINS = int(2) # XXX: set expiration a few minutes after in case put object takes a while
|
||||||
TILE_SIZE_DEGREES = int(5)
|
TILE_SIZE_DEGREES = int(5)
|
||||||
|
|
||||||
|
WHITELISTED_TAGS = [
|
||||||
|
"operator",
|
||||||
|
"manufacturer",
|
||||||
|
"direction",
|
||||||
|
"brand",
|
||||||
|
]
|
||||||
|
|
||||||
def get_all_nodes():
|
def get_all_nodes():
|
||||||
# Set up the Overpass API query
|
# Set up the Overpass API query
|
||||||
query = """
|
query = """
|
||||||
@@ -98,7 +105,13 @@ def segment_regions(nodes: Any, tile_size_degrees: int) -> dict[Any]:
|
|||||||
lat, lon = node["lat"], node["lon"]
|
lat, lon = node["lat"], node["lon"]
|
||||||
tile_lat = int(np.floor(lat / tile_size_degrees)) * tile_size_degrees
|
tile_lat = int(np.floor(lat / tile_size_degrees)) * tile_size_degrees
|
||||||
tile_lon = int(np.floor(lon / tile_size_degrees)) * tile_size_degrees
|
tile_lon = int(np.floor(lon / tile_size_degrees)) * tile_size_degrees
|
||||||
tile_dict[f"{tile_lat}/{tile_lon}"].append(node)
|
bare_node = {
|
||||||
|
"id": node["id"],
|
||||||
|
"lat": lat,
|
||||||
|
"lon": lon,
|
||||||
|
"tags": {k: v for k, v in node["tags"].items() if k in WHITELISTED_TAGS},
|
||||||
|
}
|
||||||
|
tile_dict[f"{tile_lat}/{tile_lon}"].append(bare_node)
|
||||||
print("Region segmentation complete.")
|
print("Region segmentation complete.")
|
||||||
|
|
||||||
return tile_dict
|
return tile_dict
|
||||||
|
|||||||
Reference in New Issue
Block a user