feat(AveragePosition/GetNTRIPMountPoint): change return type, update cmd average_position

This commit is contained in:
2025-06-12 14:43:41 +02:00
parent 2c5d1442fe
commit 1fcf748310
11 changed files with 148 additions and 187 deletions

View File

@ -2,12 +2,10 @@ package main
import (
"context"
"encoding/json"
"flag"
"fmt"
"log/slog"
"os"
"time"
reach "forge.cadoles.com/cadoles/go-emlid/reach/client"
"forge.cadoles.com/cadoles/go-emlid/reach/client/logger"
@ -18,24 +16,15 @@ import (
)
var (
host string = "192.168.42.1"
rawLogLevel string = "ERROR"
host string = "192.168.42.1"
rawLogLevel string = "ERROR"
baseAccuracy string = "fix"
)
func init() {
flag.StringVar(&rawLogLevel, "log-level", rawLogLevel, "log level")
flag.StringVar(&host, "host", host, "the reachrs module host")
}
type Coordinates struct {
Latitude float64 `json:"latitude"`
Longitude float64 `json:"longitude"`
Height float64 `json:"height"`
}
type Payload struct {
Coordinates Coordinates `json:"coordinates"`
AntennaOffset float64 `json:"antenna_offset"`
flag.StringVar(&baseAccuracy, "baseAccuracy", baseAccuracy, "precision required for average position")
}
func main() {
@ -64,25 +53,17 @@ func main() {
fmt.Printf("Configuration actuelle de la base :\n lat=%v\n lon=%v\n height=%v\n antenna_offset=%v\n accumulation=%v\n\n NTRIPSettings: \n address:%s\n port:%d \n username: %s\n password:%s\n sendPositionToBase:%t\n",
config.BaseMode.BaseCoordinates.Coordinates.Latitude, config.BaseMode.BaseCoordinates.Coordinates.Longitude, config.BaseMode.BaseCoordinates.Coordinates.Height, config.BaseMode.BaseCoordinates.AntennaOffset, config.BaseMode.BaseCoordinates.Accumulation, config.CorrectionInput.BaseCorrections.Settings.Ntripcli.Address, config.CorrectionInput.BaseCorrections.Settings.Ntripcli.Port, config.CorrectionInput.BaseCorrections.Settings.Ntripcli.Username, config.CorrectionInput.BaseCorrections.Settings.Ntripcli.Password, config.CorrectionInput.BaseCorrections.Settings.Ntripcli.SendPositionToBase)
// Configuration des corrections NTRIP
if err := setupNTRIPCorrections(ctx, client, config); err != nil {
//NTRIP Correction
processNTRIPCorrection(ctx, client, config)
//Base configuration
if err := setBaseToModeAndHold(ctx, client, config, baseAccuracy, 30); err != nil {
fmt.Printf("[FATAL] %+v", err)
os.Exit(1)
}
//Configuration de la base
if err := setBaseToModeAndHold(ctx, client, config, "float", 30); err != nil {
fmt.Printf("[FATAL] %+v", err)
os.Exit(1)
}
// Collecte des données de positions
if err := averagePositionAndSave(ctx, client); err != nil {
fmt.Printf("[FATAL] %+v", err)
os.Exit(1)
}
fmt.Println("Configuration terminée avec succès")
// AveragePosition
processAveragePosition(ctx, client)
}
func initializeClient(ctx context.Context) (*reach.Client, error) {
@ -121,58 +102,32 @@ func retrieveAndProcessConfig(ctx context.Context, client *reach.Client) (*model
return &config, nil
}
func setupNTRIPCorrections(ctx context.Context, client *reach.Client, config *model.Configuration) error {
func processNTRIPCorrection(ctx context.Context, client *reach.Client, config *model.Configuration) {
// Configuration des corrections NTRIP
ntripMsg, err := setupNTRIPCorrections(ctx, client)
if err != nil {
fmt.Printf("[FATAL] %+v", err)
os.Exit(1)
}
displayAvailableMountPoint(ntripMsg)
updateNTRIPMountPoint(ctx, client, config)
}
func setupNTRIPCorrections(ctx context.Context, client *reach.Client) (*protocol.TaskMessage[protocol.NTRIPPayload], error) {
fmt.Println("\nConfiguration des corrections NTRIP...")
// Recherche de points de montage
if err := client.GetNTRIPMountPoint(ctx); err != nil {
return errors.WithStack(err)
}
return processNTRIPStreams(ctx, client, config)
}
func processNTRIPStreams(ctx context.Context, client *reach.Client, config *model.Configuration) error {
messages, err := reach.OnMessageType(ctx, client, "task_status")
ntripMsg, err := client.GetNTRIPMountPoint(ctx)
if err != nil {
return errors.WithStack(err)
}
timeout := time.NewTimer(30 * time.Second)
defer timeout.Stop()
for {
select {
case b := <-messages:
if b.State == "completed" {
return handleNTRIPResponse(ctx, client, config, b)
}
case <-timeout.C:
return errors.New("timeout lors de la récupération des streams NTRIP")
case <-ctx.Done():
return ctx.Err()
}
return nil, errors.WithStack(err)
}
return ntripMsg, nil
}
func handleNTRIPResponse(ctx context.Context, client *reach.Client, config *model.Configuration, message reach.Broadcast) error {
var response model.NTRIPResponse
if err := mapstructure.Decode(message, &response); err != nil {
return errors.WithStack(err)
}
displayAvailableStreams(response)
return updateNTRIPMountPoint(ctx, client, config)
}
func displayAvailableStreams(response model.NTRIPResponse) {
func displayAvailableMountPoint(response *protocol.TaskMessage[protocol.NTRIPPayload]) {
streams := response.Payload.Str
fmt.Printf("=== %d Streams disponibles ===\n", len(streams))
fmt.Printf("=== Base < 50 km disponibles ===\n")
for i, stream := range streams {
if stream.Distance < 50 {
fmt.Printf("%d. %-15s | %s (%s) | %.1fkm\n",
@ -205,7 +160,8 @@ func updateNTRIPMountPoint(ctx context.Context, client *reach.Client, config *mo
}
func setBaseToModeAndHold(ctx context.Context, client *reach.Client, baseConfig *model.Configuration, mode string, accumulation int) error {
fmt.Println("Configuration de la base en mode float-and-hold...")
fmt.Printf("Configuration de la base en mode %s-and-hold...\n", baseAccuracy)
opts := []protocol.SetBaseOptionFunc{
protocol.WithBaseLatitude(baseConfig.BaseMode.BaseCoordinates.Coordinates.Latitude),
protocol.WithBaseLongitude(baseConfig.BaseMode.BaseCoordinates.Coordinates.Longitude),
@ -219,50 +175,46 @@ func setBaseToModeAndHold(ctx context.Context, client *reach.Client, baseConfig
return errors.WithStack(err)
}
fmt.Println("Base configurée en mode float-and-hold")
fmt.Printf("Base configurée en mode %s-and-hold\n", baseAccuracy)
return nil
}
func averagePositionAndSave(ctx context.Context, client *reach.Client) error {
fmt.Println("Démarrage de la moyenne des position...")
func processAveragePosition(ctx context.Context, client *reach.Client) {
// Collecte des données de positions
messageTask, err := averagePosition(ctx, client)
if err != nil {
fmt.Printf("[FATAL] %+v", err)
os.Exit(1)
}
displayAveragePositionMessage(messageTask)
saveAveragedPosition(ctx, client, messageTask)
fmt.Println("Configuration terminée avec succès")
}
func averagePosition(ctx context.Context, client *reach.Client) (*protocol.TaskMessage[protocol.AveragePositionPayload], error) {
fmt.Println("Démarrage de la moyenne des position...")
messageTask, err := client.AveragePosition(ctx)
if err != nil {
return errors.WithStack(err)
return nil, errors.WithStack(err)
}
logTaskProgress(messageTask)
return handleAverageCompletion(ctx, client, messageTask)
return messageTask, nil
}
func logTaskProgress(message *protocol.TaskMessage) error {
data, err := json.MarshalIndent(message, "", " ")
if err != nil {
return errors.WithStack(err)
}
fmt.Println(string(data))
return nil
}
func handleAverageCompletion(ctx context.Context, client *reach.Client, message *protocol.TaskMessage) error {
var payload Payload
if err := mapstructure.Decode(message.Payload, &payload); err != nil {
return errors.WithStack(err)
}
coords := payload.Coordinates
func displayAveragePositionMessage(message *protocol.TaskMessage[protocol.AveragePositionPayload]) error {
fmt.Printf("Position moyennée: lat=%g, lon=%g, altitude=%g\n",
coords.Latitude, coords.Longitude, coords.Height)
message.Payload.Coordinates.Latitude, message.Payload.Coordinates.Longitude, message.Payload.Coordinates.Height)
return nil
return saveAveragedPosition(ctx, client, payload)
}
func saveAveragedPosition(ctx context.Context, client *reach.Client, payload Payload) error {
func saveAveragedPosition(ctx context.Context, client *reach.Client, message *protocol.TaskMessage[protocol.AveragePositionPayload]) error {
opts := []protocol.SetBaseOptionFunc{
protocol.WithBaseLatitude(payload.Coordinates.Latitude),
protocol.WithBaseLongitude(payload.Coordinates.Longitude),
protocol.WithBaseHeight(payload.Coordinates.Height),
protocol.WithBaseAntennaOffset(payload.AntennaOffset),
protocol.WithBaseLatitude(message.Payload.Coordinates.Latitude),
protocol.WithBaseLongitude(message.Payload.Coordinates.Longitude),
protocol.WithBaseHeight(message.Payload.Coordinates.Height),
protocol.WithBaseAntennaOffset(message.Payload.AntennaOffset),
protocol.WithBaseMode("manual"),
}