# pip install fastapi uvicorn pdal pyvista numpy from sanic import Blueprint, Request, json from sanic.response import text from typing import Dict, Any import logging from b3dm.earthwork_calculator_point_cloud import EarthworkCalculatorPointCloud # 导入计算模块 from b3dm.earthwork_calculator_3d_tiles import EarthworkCalculator3dTiles, AlgorithmType, EarthworkResult3dTiles from b3dm.tileset_data_source import TilesetDataSource earthwork_bp = Blueprint("earthwork", url_prefix="") # 配置日志 logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) # 初始化函数 def init_app(url, type = "3dtiles"): """初始化应用""" data_source = None calculator_3d_tiles = None calculator_point_cloud = None try: # 初始化数据源 data_source = TilesetDataSource(url) data_source.dowload_map_data(url) if type == "3dtiles" : # 初始化计算器-3dTiles calculator_3d_tiles = EarthworkCalculator3dTiles(data_source) elif type == "pointcloud" : # 初始化计算器-点云 calculator_point_cloud = EarthworkCalculatorPointCloud(data_source.tileset_path) else : logger.info(f"不支持的3d地图数据格式:{type}") raise logger.info("土方量计算器初始化完成") return { "data_source":data_source, "calculator_3d_tiles":calculator_3d_tiles, "calculator_point_cloud":calculator_point_cloud } except ImportError as e: logger.error(f"依赖库缺失: {str(e)}") raise except Exception as e: logger.error(f"初始化失败: {str(e)}") raise # 土方量计算接口-3dTiles @earthwork_bp.post("/api/v1/calc/earthwork3dTiles") async def calc_earthwork(request: Request): """ 土方量计算接口 请求参数示例: { "polygonCoords": [ [ 115.70440468338526, 30.77363140345639 ], [ 115.70443054007985, 30.773510462589584 ], [ 115.70459702429197, 30.77360789911405 ] ], "designElevation": 100, "algorithm": "tin", "resolution": 1, "crs": "EPSG:4326", "interpolationMethod": "linear", "url": "http://222.212.85.86:9000/300bdf2b-a150-406e-be63-d28bd29b409f/model/hbgldk/yzk/20260113/3D/terra_b3dms/tileset.json" } """ try: # 1. 接收前端传参 data = request.json if not data: return _error_response("请求参数不能为空", 400) # 2. 提取参数 polygon_coords = data.get("polygonCoords") design_elevation = data.get("designElevation") url = data.get("url") if not polygon_coords: return _error_response("多边形坐标不能为空", 400) if design_elevation is None: return _error_response("设计高程不能为空", 400) if url is None: return _error_response("地图不能为空", 400) # 3. 可选参数 algorithm = data.get("algorithm", "tin") resolution = data.get("resolution", 1.0) crs = data.get("crs", "EPSG:4326") interpolation_method = data.get("interpolationMethod", "linear") # 4. 参数验证 if len(polygon_coords) < 3: return _error_response("多边形至少需要3个点", 400) # 检查多边形是否闭合，如不闭合则自动闭合 if polygon_coords[0] != polygon_coords[-1]: polygon_coords.append(polygon_coords[0]) # 算法验证 if algorithm not in ["grid", "tin", "prism"]: return _error_response("算法必须是 grid, tin 或 prism", 400) # 分辨率验证 if resolution <= 0 or resolution > 100: return _error_response("分辨率必须在0-100米之间", 400) # 5. 确保计算器已初始化 app_info = init_app(url) calculator_3d_tiles = app_info.get("calculator_3d_tiles") # 6. 执行计算 algorithm_type = AlgorithmType(algorithm) result = await calculator_3d_tiles.calculate( polygon_coords=polygon_coords, design_elevation=design_elevation, algorithm=algorithm_type, resolution=resolution, target_crs=crs, interpolation_method=interpolation_method ) # 7. 返回成功响应 res_dict = result.to_dict() res_dict["calculation_details"] = None res_dict["elevation_statistics"] = None res_dict["volume_distribution"] = None return _success_response(res_dict) except ValueError as e: logger.warning(f"参数验证失败: {str(e)}") return _error_response(f"参数错误: {str(e)}", 400) except Exception as e: logger.error(f"计算失败: {str(e)}") return _error_response(f"服务器内部错误: {str(e)}", 500) # 两期对比接口-3dTiles @earthwork_bp.post("/api/v1/calc/twoPhaseComparison") async def two_phase_comparison(request: Request): """ 两期对比接口 请求参数示例: { "polygonCoords": [ [ 115.70440468338526, 30.77363140345639 ], [ 115.70443054007985, 30.773510462589584 ], [ 115.70459702429197, 30.77360789911405 ] ], "designElevation": 100, "algorithm": "grid", "resolution": 1, "crs": "EPSG:4326", "interpolationMethod": "linear", "urlA": "http://222.212.85.86:9000/300bdf2b-a150-406e-be63-d28bd29b409f/model/hbgldk/yzk/20260113/3D/terra_b3dms/tileset.json", "urlB": "http://222.212.85.86:9000/300bdf2b-a150-406e-be63-d28bd29b409f/model/hbgldk/yzk/20260113/3D/terra_b3dms/tileset.json" } """ try: # 1. 接收前端传参 data = request.json if not data: return _error_response("请求参数不能为空", 400) # 2. 提取参数 polygon_coords = data.get("polygonCoords") design_elevation = data.get("designElevation", 1000) urlA = data.get("urlA") urlB = data.get("urlB") if not polygon_coords: return _error_response("多边形坐标不能为空", 400) if design_elevation is None: return _error_response("设计高程不能为空", 400) if urlA is None or urlB is None : return _error_response("对比地图不能为空", 400) # 3. 可选参数 algorithm = data.get("algorithm", "tin") resolution = data.get("resolution", 1.0) crs = data.get("crs", "EPSG:4326") interpolation_method = data.get("interpolationMethod", "linear") # 4. 参数验证 if len(polygon_coords) < 3: return _error_response("多边形至少需要3个点", 400) # 检查多边形是否闭合，如不闭合则自动闭合 if polygon_coords[0] != polygon_coords[-1]: polygon_coords.append(polygon_coords[0]) # 算法验证 if algorithm not in ["grid", "tin", "prism"]: return _error_response("算法必须是 grid, tin 或 prism", 400) # 分辨率验证 if resolution <= 0 or resolution > 100: return _error_response("分辨率必须在0-100米之间", 400) # 5. 确保计算器已初始化 app_info_a = init_app(urlA) if not app_info_a.get('data_source').tileset_path : return _error_response(f"下载地图失败:{urlA}", 400) calculator_3d_tiles_a = app_info_a.get("calculator_3d_tiles") app_info_b = init_app(urlB) if not app_info_b.get('data_source').tileset_path : return _error_response(f"下载地图失败:{urlB}", 400) calculator_3d_tiles_b = app_info_b.get("calculator_3d_tiles") # 6. 执行计算 algorithm_type = AlgorithmType.GRID result_a = await calculator_3d_tiles_a.calculate( polygon_coords=polygon_coords, design_elevation=design_elevation, algorithm=algorithm_type, resolution=resolution, target_crs=crs, interpolation_method=interpolation_method ) result_b = await calculator_3d_tiles_b.calculate( polygon_coords=polygon_coords, design_elevation=design_elevation, algorithm=algorithm_type, resolution=resolution, target_crs=crs, interpolation_method=interpolation_method ) # 获取网格数据 grids_a = result_a.calculation_details grids_b = result_b.calculation_details # 比较网格数据 comparison_result = calculator_3d_tiles_a.compare_grid_cells(grids_a, grids_b) # 转换为字典 result_dict = comparison_result.to_dict() # 7. 返回成功响应 return _success_response(result_dict) except ValueError as e: logger.warning(f"参数验证失败: {str(e)}") return _error_response(f"参数错误: {str(e)}", 400) except Exception as e: logger.error(f"计算失败: {str(e)}") return _error_response(f"服务器内部错误: {str(e)}", 500) # 验证接口 @earthwork_bp.post("/api/v1/calc/earthwork3dTiles/validate") async def validate_earthwork(request: Request): """验证计算参数接口""" try: # 1. 接收前端传参 data = request.json if not data: return _error_response("请求参数不能为空", 400) # 2. 提取参数 polygon_coords = data.get("polygonCoords") if not polygon_coords: return _error_response("多边形坐标不能为空", 400) url = data.get("url") if url is None: return _error_response("地图不能为空", 400) # 3. 参数验证 if len(polygon_coords) < 3: return _error_response("多边形至少需要3个点", 400) # 检查多边形是否闭合，如不闭合则自动闭合 if polygon_coords[0] != polygon_coords[-1]: polygon_coords.append(polygon_coords[0]) # 4. 确保计算器已初始化 app_info = init_app(url) calculator_3d_tiles = app_info.get("calculator_3d_tiles") # 5. 执行验证 validation_result = await calculator_3d_tiles.validate(polygon_coords) # 6. 返回结果 return _success_response(validation_result) except Exception as e: logger.error(f"验证失败: {str(e)}") return _error_response(f"验证失败: {str(e)}", 400) # 获取算法列表接口 @earthwork_bp.get("/api/v1/calc/earthwork3dTiles/algorithms") async def get_algorithms(request: Request): """获取支持的算法列表接口""" try: algorithms = [ { "id": "grid", "name": "格网法", "description": "将计算区域划分为规则格网，通过插值计算每个格网的高程变化，适合平坦或规则地形", "accuracy": "中等", "performance": "快速", "parameters": { "resolution": { "name": "格网分辨率", "description": "格网大小（米），影响计算精度和性能", "default": 1.0, "range": [0.1, 10.0] } } }, { "id": "tin", "name": "三角网法", "description": "基于不规则三角网（TIN）构建地形表面，计算每个三角形的体积变化，适合复杂地形", "accuracy": "高", "performance": "中等", "parameters": { "resolution": { "name": "不适用", "description": "三角网法不使用固定的分辨率参数", "default": None } } }, { "id": "prism", "name": "三棱柱法", "description": "结合三角网和垂直棱柱的高精度算法，计算每个三棱柱的体积，精度最高", "accuracy": "最高", "performance": "较慢", "parameters": { "resolution": { "name": "棱柱宽度", "description": "棱柱宽度（米），影响计算精度", "default": 1.0, "range": [0.1, 5.0] } } } ] return _success_response(algorithms) except Exception as e: logger.error(f"获取算法列表失败: {str(e)}") return _error_response(f"获取算法列表失败: {str(e)}", 500) # 批量计算接口 @earthwork_bp.post("/api/v1/calc/earthwork3dTiles/batch") async def batch_calc_earthwork(request: Request): """批量土方量计算接口""" try: # 1. 接收前端传参 data = request.json if not data: return _error_response("请求参数不能为空", 400) calculations = data.get("calculations", []) if not calculations: return _error_response("计算任务列表不能为空", 400) if len(calculations) > 100: return _error_response("批量计算数量超过限制（最多100个）", 400) # 3. 执行批量计算 results = [] errors = [] for i, calc_data in enumerate(calculations): try: # 提取参数 polygon_coords = calc_data.get("polygonCoords") design_elevation = calc_data.get("designElevation") url = calc_data.get("url") if not polygon_coords or design_elevation is None or url is None: errors.append({ "index": i, "error": "缺少必要参数" }) continue # 参数验证 if len(polygon_coords) < 3: errors.append({ "index": i, "error": "多边形至少需要3个点" }) continue # 检查多边形是否闭合 if polygon_coords[0] != polygon_coords[-1]: polygon_coords.append(polygon_coords[0]) # 可选参数 algorithm = calc_data.get("algorithm", "tin") resolution = calc_data.get("resolution", 1.0) crs = calc_data.get("crs", "EPSG:4326") interpolation_method = calc_data.get("interpolationMethod", "linear") # 2. 确保计算器已初始化 app_info = init_app(url) calculator_3d_tiles = app_info.get("calculator_3d_tiles") # 执行计算 algorithm_type = AlgorithmType(algorithm) result = await calculator_3d_tiles.calculate( polygon_coords=polygon_coords, design_elevation=design_elevation, algorithm=algorithm_type, resolution=resolution, target_crs=crs, interpolation_method=interpolation_method ) results.append(result) except Exception as e: errors.append({ "index": i, "error": str(e), "polygon": polygon_coords if 'polygon_coords' in locals() else None }) continue # 4. 返回结果 batch_result = { "results": results, "errors": errors, "summary": { "total": len(calculations), "success": len(results), "failed": len(errors), "successRate": f"{(len(results)/len(calculations)*100):.1f}%" if calculations else "0%" } } message = f"批量计算完成，成功 {len(results)} 个，失败 {len(errors)} 个" return _success_response(batch_result) except Exception as e: logger.error(f"批量计算失败: {str(e)}") return _error_response(f"批量计算失败: {str(e)}", 500) # 核心接口：土方量计算-点云 @earthwork_bp.post("/api/v1/calc/earthworkPointCloud") async def calc_earthwork_point_cloud(request: Request): try: # 1. 接收前端传参 data = request.json if not data: return json({ "code": 400, "msg": "请求参数不能为空", "data": None }, status=400) polygon_coords = data.get("polygonCoords") # 计算区域多边形坐标 design_elev = data.get("designElevation") # 设计高程 crs = data.get("crs", "EPSG:4326") # 坐标系，默认WGS84 url = data.get("url") if url is None: return _error_response("地图不能为空", 400) # 2. 确保计算器已初始化 app_info = init_app(url) calculator_point_cloud = app_info.get("calculator_point_cloud") result = calculator_point_cloud.calculate_earthwork(polygon_coords=polygon_coords, design_elev=design_elev, crs=crs) # 3. 处理结果 if not result["success"]: return _error_response(result["error"], 400) # 4. 格式化结果 formatted_result = calculator_point_cloud.format_result(result) # 5. 返回成功响应 return _success_response(formatted_result) except Exception as e: logger.error(f"服务器错误: {str(e)}") return _error_response(f"服务器内部错误: {str(e)}", 500) def _success_response(data: Dict[str, Any]) -> json: """成功响应""" return json({ "code": 200, "msg": "计算成功", "data": data }) def _error_response(message: str, status_code: int = 400) -> json: """错误响应""" return json({ "code": status_code, "msg": message, "data": None }, status=status_code)