为古城墙（堡）“把脉”：基于二哈识图2墙砖病害AI智能巡诊系统 简单

#include <DFRobot_Iot.h>
#include "DFRobot_GNSS.h"
#include "unihiker_k10.h"
#include "DFRobot_HuskylensV2.h"
#include <SD.h>
#include <base64.h>

// 静态常量
const String topics[5] = {"","","","",""};
// 创建对象
DFRobot_GNSS_UART gnss;
HuskylensV2       huskylens;
DFRobot_Iot       myIot;
UNIHIKER_K10      k10;
uint8_t           screen_dir=2;

// ID到中文病害类别的映射函数
String getDamageCategory(int id) {
  switch(id) {
    case 1: return "裂缝";
    case 2: return "风化";
    case 3: return "碱蚀";
    case 4: return "缺损";
    case 5: return "植被生长";
    default: return "UNKNOWN";
  }
}

// 辅助函数：读取图片文件并转换为base64编码
String imageToBase64(const char* filename) {
    File file = SD.open(filename, FILE_READ);
    if (!file) {
        return "";
    }
    
    // 获取文件大小
    size_t fileSize = file.size();
    
    // 读取文件内容
    uint8_t* buffer = new uint8_t[fileSize];
    if (!buffer) {
        file.close();
        return "";
    }
    
    file.read(buffer, fileSize);
    file.close();
    
    // Base64编码
    String base64Str = base64::encode(buffer, fileSize);
    
    // 清理内存
    delete[] buffer;
    
    return base64Str;
}

// 辅助函数：构建JSON字符串
String buildJsonMessage(String timestamp, double longitude, double latitude, 
                       String photoBase64, String damageCategory) {
    String json = "{";
    json += "\"time\":\"" + timestamp + "\",";
    json += "\"lng\":" + String(longitude, 6) + ",";
    json += "\"lat\":" + String(latitude, 6) + ",";
    json += "\"photo\":\"" + photoBase64 + "\",";
    json += "\"damageCategory\":\"" + damageCategory + "\"";
    json += "}";
    return json;
}

// 辅助函数：格式化时间字符串
String formatDateTime(sTim_t date, sTim_t utc) {
    String year = String(date.year);
    String month = (date.month < 10) ? "0" + String(date.month) : String(date.month);
    String day = (date.date < 10) ? "0" + String(date.date) : String(date.date);
    String hour = (utc.hour < 10) ? "0" + String(utc.hour) : String(utc.hour);
    String minute = (utc.minute < 10) ? "0" + String(utc.minute) : String(utc.minute);
    String second = (utc.second < 10) ? "0" + String(utc.second) : String(utc.second);
    
    return year + "-" + month + "-" + day + " " + hour + ":" + minute + ":" + second;
}

// 主程序开始
void setup() {
    k10.begin();
    gnss.begin(&Serial1, P0, P1);
    k10.initScreen(screen_dir);
    k10.initBgCamerImage();
    k10.setBgCamerImage(false);
    k10.creatCanvas();
    k10.initSDFile();
    Wire.begin();
    
    while (!huskylens.begin(Wire)) {
        delay(100);
    }
    huskylens.switchAlgorithm((eAlgorithm_t)128);
    delay(5000);
    
    myIot.wifiConnect("yourSSID", "yourPASSWD");
    while (!myIot.wifiStatus()) {}
    
    k10.canvas->canvasText(myIot.getWiFiLocalIP(), 11, 0x0000FF);
    k10.canvas->updateCanvas();
    
    myIot.init("127.0.0.1", "siot", "9455149370301812", "dfrobot", topics, 1883);
    myIot.connect();
    while (!myIot.connected()) {}
    
    k10.canvas->canvasText("MQTT已连接", 13, 0x0000FF);
    k10.canvas->updateCanvas();
}

void loop() {
    if ((huskylens.available((eAlgorithm_t)128))) {
        huskylens.getResult(ALGORITHM_FALLDOWN_RECOGNITION);
        
        // 1. 保存图片到SD卡
        k10.photoSaveToTFCard("S:/photo.bmp");
        
        // 2. 获取GNSS数据
        sTim_t utc = gnss.getUTC();
        sTim_t date = gnss.getDate();
        sLonLat_t lat = gnss.getLat();
        sLonLat_t lon = gnss.getLon();
        
        // 3. 获取识别的ID并映射到中文病害类别
        int detectedId = RET_ITEM_NUM(huskylens.getCachedCenterResult((eAlgorithm_t)128), Result, ID);
        String damageCategory = getDamageCategory(detectedId);
        
        // 4. 读取图片并转换为base64
        String photoBase64 = imageToBase64("S:/photo.bmp");
        
        // 5. 格式化时间
        String timestamp = formatDateTime(date, utc);
        
        // 6. 构建JSON消息
        String jsonMessage = buildJsonMessage(
            timestamp, 
            lon.lonitudeDegree, 
            lat.latitudeDegree, 
            photoBase64,
            damageCategory
        );
        
        // 7. 在屏幕上显示信息
        k10.canvas->canvasText(timestamp, 1, 0x0000FF);
        k10.canvas->canvasText("Lng: " + String(lon.lonitudeDegree, 6), 3, 0x0000FF);
        k10.canvas->canvasText("Lat: " + String(lat.latitudeDegree, 6), 5, 0x0000FF);
        k10.canvas->canvasText("ID: " + String(detectedId), 7, 0x0000FF);
        k10.canvas->canvasText("Damage: " + damageCategory, 9, 0x0000FF);
        k10.canvas->canvasText("Photo: " + String(photoBase64.length()) + " chars", 11, 0x0000FF);
        k10.canvas->updateCanvas();
        
        // 8. 发送MQTT消息
        myIot.publish("siot/inspectionLog", jsonMessage.c_str(), 1);
        
        delay(1000); // 防止过于频繁发送
    }
    
    // 可以添加一些延迟，减少CPU使用率
    delay(100);
}