{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "888d089c-a9c8-4d2d-af74-dff1a8ccfefd",
   "metadata": {
    "pycharm": {
     "name": "#%%\n"
    }
   },
   "outputs": [],
   "source": [
    "import json\n",
    "import time\n",
    "from typing import List\n",
    "import requests\n",
    "import pandas as pd\n",
    "\n",
    "\n",
    "class GetElevation:\n",
    "\n",
    "    @classmethod\n",
    "    def __SendQuery(cls, latLngString: str) -> json:\n",
    "        query = ('https://api.opentopodata.org/v1/mapzen?locations={}&interpolation=bilinear'.format(latLngString))\n",
    "        res = requests.get(query).json()\n",
    "        if res[\"status\"] != \"OK\":\n",
    "            raise Exception(res[\"error\"])\n",
    "        return res\n",
    "\n",
    "    def GetSingleElevation(self, latitude: float, longitude: float) -> float:\n",
    "        \"\"\"\n",
    "        获取单个高程，输入经纬度格式为数值类型，返回值为高程float类型\n",
    "        :param latitude: 纬度\n",
    "        :param longitude: 经度\n",
    "        :return: 高程\n",
    "        \"\"\"\n",
    "        if latitude < -90 or latitude > 90:\n",
    "            raise Exception(\"纬度的范围应在-90-90之间！请检查数据源！\")\n",
    "        latLngString = str(latitude) + \",\" + str(longitude)\n",
    "        res = self.__SendQuery(latLngString)\n",
    "        elevation = res[\"results\"][0][\"elevation\"]\n",
    "        return elevation\n",
    "\n",
    "    def GetMultiElevation(self, latitude: List[float], longitude: List[float]) -> List[float]:\n",
    "        \"\"\"\n",
    "        获取数组类型的高程，输入经纬度格式为经度数组和纬度数组，返回值为高程数组\n",
    "        :param latitude:纬度数组\n",
    "        :param longitude:经度数组\n",
    "        :return:高程数组\n",
    "        \"\"\"\n",
    "        if len(latitude) != len(longitude):\n",
    "            raise Exception(\"纬度数组和经度数组长度不一致！请检查数据源！\")\n",
    "        for lat in latitude:\n",
    "            if lat < -90 or lat > 90:\n",
    "                raise Exception(\"纬度的范围应在-90-90之间！请检查数据源！\")\n",
    "        elevationList = []\n",
    "        hundredNums = len(latitude) // 100\n",
    "        # 查询整百的高程\n",
    "        for i in range(hundredNums):\n",
    "            latLngString = \"\"\n",
    "            for idx in range(100 * i, 100 * (i + 1)):\n",
    "                latLngString += (str(latitude[idx]) + \",\" + str(longitude[idx]) + \"|\")\n",
    "            res = self.__SendQuery(latLngString)\n",
    "            for idx in range(100):\n",
    "                elevationList.append(res[\"results\"][idx][\"elevation\"])\n",
    "            time.sleep(1)\n",
    "        # 查询剩余的不到100的高程\n",
    "        latLngString = \"\"\n",
    "        for i in range(hundredNums * 100, len(latitude)):\n",
    "            latLngString += (str(latitude[i]) + \",\" + str(longitude[i]) + \"|\")\n",
    "        res = self.__SendQuery(latLngString)\n",
    "        for i in range(len(latitude) - hundredNums * 100):\n",
    "            elevationList.append(res[\"results\"][i][\"elevation\"])\n",
    "        return elevationList\n",
    "\n",
    "    def ExportToXlsx(self, latLongDf: pd.DataFrame, elevationList: List[float], outputPath: str) -> None:\n",
    "        \"\"\"\n",
    "    如果用户可以传入一个DataFrame数据，可以将返回得到的高程拼接并输出\n",
    "    :param latLongDf: DataFrame数据\n",
    "    :param elevationList: 高程数组\n",
    "    :param outputPath: 输出路径\n",
    "    :return: 无返回值\n",
    "    \"\"\"\n",
    "        latLongDf[\"elevation\"] = elevationList\n",
    "        latLongDf.to_excel(outputPath, index=False)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "2a226b08-0c92-483e-b590-29a39dce6298",
   "metadata": {
    "pycharm": {
     "name": "#%%\n"
    }
   },
   "outputs": [],
   "source": [
    "data = pd.read_excel('./lat_lon.xlsx')\n",
    "data.columns = ['plant', 'longitude', 'latitude']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "91afb581-1994-47c4-85ca-d3ea3e13e95d",
   "metadata": {},
   "outputs": [],
   "source": [
    "multiEle = ele.GetMultiElevation(data[\"latitude\"], data[\"longitude\"])\n",
    "print(multiEle)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}