|
|
# 这是一个示例 Python 脚本。
|
|
|
import json
|
|
|
import os
|
|
|
# 按 Shift+F10 执行或将其替换为您的代码。
|
|
|
# 按 双击 Shift 在所有地方搜索类、文件、工具窗口、操作和设置。
|
|
|
import SimpleView_SaveImage as simple
|
|
|
import config
|
|
|
import image
|
|
|
import time # 在文件顶部导入 time 模块
|
|
|
|
|
|
from flask import Flask, jsonify, request
|
|
|
|
|
|
from logConfig import get_logger
|
|
|
|
|
|
app = Flask(__name__)
|
|
|
import concurrent.futures
|
|
|
from functools import partial
|
|
|
|
|
|
# 使用示例
|
|
|
logger = get_logger()
|
|
|
# 示例接口：GET 请求
|
|
|
@app.route("/config/update")
|
|
|
def home():
|
|
|
logger.info("配置更新请求")
|
|
|
config.load_configs()
|
|
|
logger.info("配置更新完成")
|
|
|
return {"message": "Hello, this service config update!", "status": "OK"}
|
|
|
|
|
|
|
|
|
@app.route("/api/addTemplate", methods=["POST"])
|
|
|
def save_template():
|
|
|
logger.info("接收到添加模板请求")
|
|
|
data = request.get_json()
|
|
|
|
|
|
# 验证必要字段
|
|
|
if not data:
|
|
|
logger.error("添加模板失败：未提供数据")
|
|
|
return jsonify({"status": "ERROR", "message": "No data provided"}), 400
|
|
|
|
|
|
template_type = data.get("type")
|
|
|
if not template_type:
|
|
|
logger.error("添加模板失败：缺少'type'字段")
|
|
|
return jsonify({"status": "ERROR", "message": "Missing 'type' field"}), 400
|
|
|
CONFIG_DIR = "./config/template"
|
|
|
# 构造文件路径
|
|
|
filename = f"{template_type}.json"
|
|
|
file_path = os.path.join(CONFIG_DIR, filename)
|
|
|
|
|
|
# 写入文件
|
|
|
try:
|
|
|
with open(file_path, "w", encoding="utf-8") as f:
|
|
|
json.dump(data, f, ensure_ascii=False, indent=4)
|
|
|
config.load_configs()
|
|
|
logger.info(f"模板 '{template_type}' 保存成功")
|
|
|
return jsonify({"status": "OK", "message": f"Template '{template_type}' saved successfully."}), 200
|
|
|
except Exception as e:
|
|
|
logger.error(f"保存模板失败: {str(e)}")
|
|
|
return jsonify({"status": "ERROR", "message": f"Failed to save template: {str(e)}"}), 500
|
|
|
|
|
|
|
|
|
@app.route("/api/initDevice", methods=['POST'])
|
|
|
def initDevice():
|
|
|
logger.info("设备初始化请求")
|
|
|
simple.initialize_devices()
|
|
|
logger.info("设备初始化完成")
|
|
|
|
|
|
#
|
|
|
# @app.route("/api/picComputeAll", methods=['POST'])
|
|
|
# def picComputeAll():
|
|
|
# logger.info("接收到全设备图片计算请求")
|
|
|
# start_time_total = time.time()
|
|
|
#
|
|
|
# # 输入参数
|
|
|
# # sn:设备编号
|
|
|
# # type:类型
|
|
|
#
|
|
|
# # 获取 JSON 数据
|
|
|
# data = request.get_json()
|
|
|
# type = data['type']
|
|
|
# template = config.TEMPLATE_CONFIG_MAP.get(type)
|
|
|
# if not template:
|
|
|
# logger.error("缺少模板配置: 'type'")
|
|
|
# return jsonify({
|
|
|
# "message": "",
|
|
|
# "status": "ERROR",
|
|
|
# "error": "Missing required parameter: 'type'"
|
|
|
# }), 400
|
|
|
# all_sn_list = list(config.CAMERA_CONFIG_MAP.keys())
|
|
|
#
|
|
|
# # 遍历每个 sn 并判断是否都有 hole
|
|
|
# all_have_holes = False
|
|
|
# color_list = [] # 新增：用于保存所有 sn 的 color
|
|
|
#
|
|
|
# count_assemble = 0
|
|
|
# for sn in all_sn_list:
|
|
|
# start_time = time.time() # 开始计时
|
|
|
# logger.info(f"开始处理设备 {sn} 的图片计算")
|
|
|
# pic_info = simple.enhanced_pic(sn, type)
|
|
|
# points = pic_info.get("point")
|
|
|
# if points is None:
|
|
|
# logger.warning(f"设备 {sn} 未返回点云数据")
|
|
|
# all_have_holes = True
|
|
|
# break
|
|
|
#
|
|
|
# # 提取 color 并加入列表
|
|
|
# color = pic_info.get("color")
|
|
|
# if color is not None:
|
|
|
# color_list.append(color)
|
|
|
#
|
|
|
# try:
|
|
|
# real_holes, count = image.detect_large_holes(points, sn, type)
|
|
|
# # count_assemble += count
|
|
|
#
|
|
|
# if len(real_holes) != 0:
|
|
|
# all_have_holes = True
|
|
|
# # 计算耗时
|
|
|
# elapsed_time = round(time.time() - start_time, 3)
|
|
|
# # 输出日志
|
|
|
# logger.info(f"[INFO] picComputeAll executed in {elapsed_time} seconds {sn}")
|
|
|
# break # 判断到空洞就直接退出
|
|
|
# except Exception as e:
|
|
|
# all_have_holes = True
|
|
|
# logger.error(f"Failed to process request: {str(e)} sn：{sn}")
|
|
|
#
|
|
|
# # 计算总耗时
|
|
|
# total_elapsed_time = round(time.time() - start_time_total, 3)
|
|
|
# logger.info(f"全设备图片计算完成，总耗时 {total_elapsed_time} 秒，结果: {all_have_holes}")
|
|
|
#
|
|
|
# # 返回结果中包含 color 列表
|
|
|
# return jsonify({
|
|
|
# "message": all_have_holes,
|
|
|
# "lack": all_have_holes,
|
|
|
# "type": type,
|
|
|
# "count": count_assemble,
|
|
|
# "colors": color_list, # 添加 color 列表
|
|
|
# "status": "OK"
|
|
|
# })
|
|
|
#
|
|
|
@app.route("/api/picComputeAll", methods=['POST'])
|
|
|
def picComputeAll():
|
|
|
logger.info("接收到全设备图片计算请求")
|
|
|
start_time_total = time.time()
|
|
|
|
|
|
# 获取 JSON 数据
|
|
|
data = request.get_json()
|
|
|
type = data['type']
|
|
|
template = config.TEMPLATE_CONFIG_MAP.get(type)
|
|
|
if not template:
|
|
|
logger.error("缺少模板配置: 'type'")
|
|
|
return jsonify({
|
|
|
"message": "",
|
|
|
"status": "ERROR",
|
|
|
"error": "Missing required parameter: 'type'"
|
|
|
}), 400
|
|
|
all_sn_list = list(config.CAMERA_CONFIG_MAP.keys())
|
|
|
|
|
|
# 同步处理每个设备
|
|
|
all_have_holes = False
|
|
|
color_list = []
|
|
|
count_assemble = 0
|
|
|
|
|
|
# 逐个同步处理设备
|
|
|
for sn in all_sn_list:
|
|
|
try:
|
|
|
start_time = time.time()
|
|
|
logger.info(f"开始处理设备 {sn} 的图片计算")
|
|
|
|
|
|
# 同步采集图片信息
|
|
|
pic_info = simple.enhanced_pic(sn, type)
|
|
|
points = pic_info.get("point")
|
|
|
|
|
|
if points is None:
|
|
|
logger.warning(f"设备 {sn} 未返回点云数据")
|
|
|
all_have_holes = True
|
|
|
break
|
|
|
|
|
|
# 提取 color 并加入列表
|
|
|
if pic_info.get("color") is not None:
|
|
|
color_list.append(pic_info["color"])
|
|
|
|
|
|
# 检测空洞
|
|
|
real_holes, count = image.detect_large_holes(points, sn, type)
|
|
|
|
|
|
elapsed_time = round(time.time() - start_time, 3)
|
|
|
logger.info(f"设备 {sn} 处理完成，耗时 {elapsed_time} 秒，发现空洞: {len(real_holes) != 0}")
|
|
|
|
|
|
# 检查是否发现空洞
|
|
|
if len(real_holes) != 0:
|
|
|
all_have_holes = True
|
|
|
count_assemble += count
|
|
|
logger.info(f"设备 {sn} 发现空洞，停止其他任务")
|
|
|
break # 发现空洞就直接退出
|
|
|
|
|
|
except Exception as e:
|
|
|
logger.error(f"处理设备 {sn} 时发生异常: {str(e)}")
|
|
|
all_have_holes = True
|
|
|
break
|
|
|
|
|
|
# 计算总耗时
|
|
|
total_elapsed_time = round(time.time() - start_time_total, 3)
|
|
|
logger.info(f"全设备图片计算完成，总耗时 {total_elapsed_time} 秒，结果: {all_have_holes}")
|
|
|
|
|
|
# 返回结果中包含 color 列表
|
|
|
return jsonify({
|
|
|
"message": all_have_holes,
|
|
|
"lack": all_have_holes,
|
|
|
"type": type,
|
|
|
"count": count_assemble,
|
|
|
"colors": color_list,
|
|
|
"status": "OK"
|
|
|
})
|
|
|
|
|
|
|
|
|
|
|
|
@app.route("/api/picBitCloudy", methods=['POST'])
|
|
|
def picBitCloudy():
|
|
|
logger.info("接收到picBitCloudy请求")
|
|
|
|
|
|
# 输入参数
|
|
|
# sn:设备编号
|
|
|
# type:类型
|
|
|
|
|
|
# 获取 JSON 数据
|
|
|
data = request.get_json()
|
|
|
direction = data.get('direction')
|
|
|
type = data.get('type')
|
|
|
all_sn_list = list(config.CAMERA_CONFIG_MAP.keys())
|
|
|
|
|
|
# 遍历每个 sn 并判断是否都有 hole
|
|
|
all_have_holes = False
|
|
|
color_list = [] # 新增：用于保存所有 sn 的 color
|
|
|
|
|
|
count_assemble = 0
|
|
|
for sn in all_sn_list:
|
|
|
if direction and config.CAMERA_CONFIG_MAP[sn].get("direction") != direction:
|
|
|
logger.debug(f"设备 {sn} 方向不匹配，跳过")
|
|
|
continue
|
|
|
|
|
|
start_time = time.time() # 开始计时
|
|
|
logger.info(f"开始处理设备 {sn} 的图片计算")
|
|
|
pic_info = simple.enhanced_pic(sn, type)
|
|
|
points = pic_info.get("point")
|
|
|
if points is None or len(points) < 300:
|
|
|
logger.warning(f"设备 {sn} 点云数据不足")
|
|
|
all_have_holes = True
|
|
|
break
|
|
|
|
|
|
# 提取 color 并加入列表
|
|
|
color = pic_info.get("color")
|
|
|
if color is not None:
|
|
|
color_list.append(color)
|
|
|
|
|
|
logger.info(f"picBitCloudy处理完成，结果: {all_have_holes}")
|
|
|
# 返回结果中包含 color 列表
|
|
|
return jsonify({
|
|
|
"message": all_have_holes,
|
|
|
"lack": all_have_holes,
|
|
|
"colors": color_list, # 添加 color 列表
|
|
|
"status": "OK"
|
|
|
})
|
|
|
|
|
|
|
|
|
@app.route("/api/picCompute", methods=['POST'])
|
|
|
def picCompute():
|
|
|
logger.info("接收到单设备图片计算请求")
|
|
|
|
|
|
# 输入参数
|
|
|
# sn:设备编号
|
|
|
# type:类型
|
|
|
try:
|
|
|
# 获取 JSON 数据
|
|
|
data = request.get_json()
|
|
|
if not data or 'sn' not in data:
|
|
|
logger.error("缺少必需参数: 'sn'")
|
|
|
return jsonify({
|
|
|
"message": "",
|
|
|
"status": "ERROR",
|
|
|
"error": "Missing required parameter: 'sn'"
|
|
|
}), 400
|
|
|
|
|
|
sn = data['sn']
|
|
|
type = data['type']
|
|
|
logger.info(f"开始处理设备 {sn} 类型 {type} 的图片计算")
|
|
|
|
|
|
template = config.TEMPLATE_CONFIG_MAP.get(type)
|
|
|
if not template:
|
|
|
logger.error(f"缺少模板配置: {type}")
|
|
|
return jsonify({
|
|
|
"message": "",
|
|
|
"status": "ERROR",
|
|
|
"error": "Missing required parameter: 'type'"
|
|
|
}), 400
|
|
|
|
|
|
pic_info = simple.enhanced_pic(sn, type)
|
|
|
real_holes, count = image.detect_large_holes(pic_info["point"], sn, type)
|
|
|
|
|
|
result = len(real_holes) != 0
|
|
|
logger.info(f"设备 {sn} 图片计算完成，检测到空洞: {result}")
|
|
|
|
|
|
return jsonify({"message": result,
|
|
|
"lack": result, "status": "OK"})
|
|
|
|
|
|
except Exception as e:
|
|
|
logger.error(f"图片计算失败: {str(e)}")
|
|
|
return jsonify({"message": "", "status": "ERROR", "error": f"Failed to get TIFF paths: {str(e)}"}), 500
|
|
|
|
|
|
return jsonify({"message": True, "lack": True, "status": "OK"})
|
|
|
|
|
|
|
|
|
@app.errorhandler(Exception)
|
|
|
def handle_exception(e):
|
|
|
# 处理所有未被捕获的异常
|
|
|
logger.error(f"未捕获的异常: {str(e)}")
|
|
|
return jsonify({
|
|
|
"message": "",
|
|
|
"status": "ERROR",
|
|
|
"error": str(e)
|
|
|
}), 500
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
from waitress import serve
|
|
|
|
|
|
logger.info("服务启动中...")
|
|
|
print("Serving on http://0.0.0.0:5000")
|
|
|
logger.info("服务启动成功，监听地址: http://0.0.0.0:5000")
|
|
|
serve(app, host='0.0.0.0', port=5000)
|