初始

third/solaranalyzer/analyzer.go

@@ -0,0 +1,88 @@
+package solaranalyzer
+
+import (
+	"encoding/json"
+	"os/exec"
+	"path/filepath"
+	"strings"
+
+	"github.com/tidwall/gjson"
+)
+
+const (
+	drillingHole = "hole"    // 钻孔
+	pillar       = "pile"    // 桩基
+	bracket      = "shelves" // 支架
+	solarPanel   = "pho"     // 光伏板
+)
+
+type Point struct {
+	Points []XYPosition // 光伏板的坐标点
+	ID     int          // pv_module 表中的自增ID
+}
+
+type XYPosition struct {
+	X float64 `json:"x"`
+	Y float64 `json:"y"`
+}
+
+// getSolarPanelRanges 解析光伏板 JSON
+func getSolarPanelRanges(file string) []Point {
+	var points []Point
+
+	gjson.Parse(file).ForEach(func(_, record gjson.Result) bool {
+		record.Get("detail").ForEach(func(_, detail gjson.Result) bool {
+			id := record.Get("id").Int()
+
+			var longitudeAndLatitude []XYPosition
+			gjson.Get(detail.String(), "positions").ForEach(func(_, position gjson.Result) bool {
+				longitude := position.Get("lng").Float()
+				latitude := position.Get("lat").Float()
+
+				longitudeAndLatitude = append(longitudeAndLatitude, XYPosition{X: longitude, Y: latitude})
+				return true
+			})
+
+			points = append(points, Point{ID: int(id), Points: longitudeAndLatitude})
+
+			return true
+		})
+
+		return true
+	})
+
+	return points
+}
+
+// convertToLongitudeAndLatitude 根据 tif 文件将坐标转换为经纬度
+func convertToLongitudeAndLatitude(tifFilePath string, coordinatePoints []XYPosition) []XYPosition {
+	positionsJSON, err := json.Marshal(coordinatePoints)
+	if err != nil {
+		panic(err)
+	}
+
+	convertPath, err := filepath.Abs("./convert")
+	if err != nil {
+		panic(err)
+	}
+
+	output, err := exec.Command(convertPath, filepath.Clean(tifFilePath), string(positionsJSON)).CombinedOutput()
+	if err != nil {
+		panic(err)
+	}
+
+	outputLines := strings.Split(string(output), "\n")
+	lastLine := outputLines[len(outputLines)-2] // 获取最后一行
+
+	// 解析二进制返回的 JSON, 并将其转换为 XYPosition 结构
+	var longitudeAndLatitude []XYPosition
+	gjson.Get(lastLine, "data").ForEach(func(_, position gjson.Result) bool {
+		longitudeAndLatitude = append(longitudeAndLatitude, XYPosition{
+			X: position.Get("y").Float(),
+			Y: position.Get("x").Float(),
+		})
+		return true
+	})
+
+	return longitudeAndLatitude
+}