feat: predictions

internal/pkg/ds/predictor.go

@@ -1,21 +1,95 @@
 package ds
 
-import "time"
+import (
+	"time"
+
+	api "git.intra.yksa.space/gsn/predictor/pkg/rest"
+)
 
 type PredictionParameters struct {
-	LaunchLatitude  float64
-	LaunchLongitude float64
-	LaunchDatetime  time.Time
-	LaunchAltitude  float64
+	LaunchLatitude  *float64
+	LaunchLongitude *float64
+	LaunchDatetime  *time.Time
+	LaunchAltitude  *float64
+	Profile         *string
+	AscentRate      *float64
+	BurstAltitude   *float64
+	DescentRate     *float64
+	FloatAltitude   *float64
+	StopDatetime    *time.Time
+	AscentCurve     *string // base64
+	DescentCurve    *string // base64
+	Interpolate     *bool
+	Format          *string
+	Dataset         *time.Time
 	// Add other parameters as needed
 }
 
 type PredicitonResult struct {
-	Latitude  float64
-	Longitude float64
-	Altitude  float64
-	Timestamp time.Time
-	WindU     float64
-	WindV     float64
+	Latitude  *float64
+	Longitude *float64
+	Altitude  *float64
+	Timestamp *time.Time
+	WindU     *float64
+	WindV     *float64
 	// Add other result fields as needed
 }
+
+// ConvertOptPredictionParameters converts ogen's OptPredictionParameters to the internal pointer-based model.
+// Returns nil if the input is not set.
+func ConvertOptPredictionParameters(opt api.OptPredictionParameters) *PredictionParameters {
+	if !opt.Set {
+		return nil
+	}
+	in := opt.Value
+	out := &PredictionParameters{}
+
+	if v, ok := in.LaunchLatitude.Get(); ok {
+		out.LaunchLatitude = &v
+	}
+	if v, ok := in.LaunchLongitude.Get(); ok {
+		out.LaunchLongitude = &v
+	}
+	if v, ok := in.LaunchDatetime.Get(); ok {
+		out.LaunchDatetime = &v
+	}
+	if v, ok := in.LaunchAltitude.Get(); ok {
+		out.LaunchAltitude = &v
+	}
+	if v, ok := in.Profile.Get(); ok {
+		s := string(v)
+		out.Profile = &s
+	}
+	if v, ok := in.AscentRate.Get(); ok {
+		out.AscentRate = &v
+	}
+	if v, ok := in.BurstAltitude.Get(); ok {
+		out.BurstAltitude = &v
+	}
+	if v, ok := in.DescentRate.Get(); ok {
+		out.DescentRate = &v
+	}
+	if v, ok := in.FloatAltitude.Get(); ok {
+		out.FloatAltitude = &v
+	}
+	if v, ok := in.StopDatetime.Get(); ok {
+		out.StopDatetime = &v
+	}
+	if v, ok := in.AscentCurve.Get(); ok {
+		out.AscentCurve = &v
+	}
+	if v, ok := in.DescentCurve.Get(); ok {
+		out.DescentCurve = &v
+	}
+	if v, ok := in.Interpolate.Get(); ok {
+		out.Interpolate = &v
+	}
+	if v, ok := in.Format.Get(); ok {
+		s := string(v)
+		out.Format = &s
+	}
+	if v, ok := in.Dataset.Get(); ok {
+		out.Dataset = &v
+	}
+	return out
+}