HeightMap/backend_fastapi.py at master · jaluebbe/HeightMap · GitHub

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
from fastapi import FastAPI, Query, HTTPException
from starlette.staticfiles import StaticFiles
from starlette.responses import FileResponse
import uvicorn
import geojson
from height_map.height_info import HeightInfo
from height_map.cci_land_cover import LandCover
import height_map.track_methods as track_methods

hi = HeightInfo()
lc = LandCover()

app = FastAPI(
    openapi_prefix="",
    title="HeightMap",
    description="Combination of several elevation data sources.",
)

app.mount("/static", StaticFiles(directory="static"), name="static")
app.include_router(track_methods.router)


@app.get("/", include_in_schema=False)
async def root():
    return FileResponse("static/heightmap.html")


@app.get("/gps", include_in_schema=False)
async def gps():
    return FileResponse("static/heightmap_gps.html")


@app.get("/railway", include_in_schema=False)
async def railway():
    return FileResponse("static/railwaymap.html")


@app.get("/api/get_height")
def get_height(
    lat: float = Query(..., ge=-90, le=90),
    lon: float = Query(..., ge=-180, le=180),
):
    response = hi.get_height(lat, lon)
    if response["source"] == "NODATA":
        raise HTTPException(status_code=404, detail="no data available")
    return response


@app.get("/api/get_max_height")
def get_max_height(
    lat_ll: float = Query(..., ge=-90, le=90),
    lon_ll: float = Query(..., ge=-180, le=180),
    lat_ur: float = Query(..., ge=-90, le=90),
    lon_ur: float = Query(..., ge=-180, le=180),
):
    return hi.get_max_height(lat_ll, lon_ll, lat_ur, lon_ur)


@app.get("/api/get_min_height")
def get_min_height(
    lat_ll: float = Query(..., ge=-90, le=90),
    lon_ll: float = Query(..., ge=-180, le=180),
    lat_ur: float = Query(..., ge=-90, le=90),
    lon_ur: float = Query(..., ge=-180, le=180),
):
    return hi.get_min_height(lat_ll, lon_ll, lat_ur, lon_ur)


@app.get("/api/get_min_max_height")
def get_min_max_height(
    lat_ll: float = Query(..., ge=-90, le=90),
    lon_ll: float = Query(..., ge=-180, le=180),
    lat_ur: float = Query(..., ge=-90, le=90),
    lon_ur: float = Query(..., ge=-180, le=180),
):
    extreme_locations = []
    result = hi.get_min_max_height(lat_ll, lon_ll, lat_ur, lon_ur)
    for _location in result["location_min"]:
        extreme_locations.append(
            geojson.Feature(
                geometry=geojson.Point(_location[::-1]),
                properties={
                    "type": "minimum",
                    "elevation_m": round(result["h_min"], 1),
                    "source": result["source_min"],
                },
            )
        )
    for _location in result["location_max"]:
        extreme_locations.append(
            geojson.Feature(
                geometry=geojson.Point(_location[::-1]),
                properties={
                    "type": "maximum",
                    "elevation_m": round(result["h_max"], 1),
                    "source": result["source_max"],
                },
            )
        )
    return geojson.FeatureCollection(
        extreme_locations, properties={"source": result["source"]}
    )


@app.get("/api/get_land_cover")
def get_surface_cover(
    lat: float = Query(..., ge=-90, le=90),
    lon: float = Query(..., ge=-180, le=180),
):
    return lc.get_data_at_position(lat, lon)


if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)