wx

type AtmosByPoint struct {
	// Lat-longs to stack of levels
	SampleStacks map[math.Point2LL]*AtmosSampleStack
}

type AtmosByPointSOA struct {
	Lat, Long []float32
	Levels    [NumSampleLevels]AtmosLevelsSOA
}

type AtmosByTime struct {
	SampleStacks map[time.Time]*AtmosSampleStack
}

type AtmosByTimeSOA struct {
	Times  []int64 // Delta-encoded Unix timestamps
	Levels [NumSampleLevels]AtmosLevelsSOA
}

type AtmosGrid struct {
	Extent   math.Extent2D
	Res      [3]int
	AltRange [2]float32
	Points   []Sample
}

type AtmosLevelsSOA struct {
	// All of the following are delta encoded
	Heading     []uint8 // degrees/2
	Speed       []uint8 // knots
	Temperature []int8  // Temperature in Celsius
	Dewpoint    []int8  // Dewpoint in Celsius
	Height      []uint8 // geopotential height (MSL) + windHeightOffset in meters
}

type AtmosResult struct {
	AtmosByPointSOA *AtmosByPointSOA
	Time            time.Time
	NextTime        time.Time
	Err             error
}

type AtmosSample struct {
	UComponent  float32 // eastward velocity m/s
	VComponent  float32 // northward velocity m/s
	Temperature float32 // Kelvin
	Dewpoint    float32 // Kelvin
	Height      float32 // geopotential height (meters)
}

type AtmosSampleStack struct {
	Levels [NumSampleLevels]AtmosSample
}

type METAR struct {
	ICAO        string `json:"icaoId"`
	Time        time.Time
	Temperature float32 `json:"temp"`
	Dewpoint    float32 `json:"dewp"`
	Altimeter   float32 `json:"altim"`
	WindDir     *int    `json:"-"` // nil for variable winds, otherwise heading 0-360
	WindSpeed   int     `json:"wspd"`
	WindGust    *int    `json:"wgst"`
	Raw         string  `json:"rawOb"`

	// WindDirRaw and ReportTime are used for JSON unmarshaling only
	WindDirRaw any    `json:"wdir"` // nil or string "VRB" for variable, else number for heading
	ReportTime string `json:"reportTime"`
}

type METARSOA struct {
	// These are all delta coded
	ReportTime  [][]byte
	Temperature []int16 // fixed point, one decimal digit
	Dewpoint    []int16 // fixed point, one decimal digit
	Altimeter   []int16 // fixed point, one decimal digit
	WindDir     []int16
	WindSpeed   []int16
	WindGust    []int16

	// This is not; it's not worth delta encoding the raw METAR reports
	// since there's generally not much character alignment between
	// successive reports.
	Raw []string
}

type Model struct {
	// contains filtered or unexported fields
}

type Precip struct {
	DBZ        []byte
	Resolution int
	Latitude   float32
	Longitude  float32
}

type Provider interface {
	GetAvailableTimeIntervals() []util.TimeInterval

	// Best effort, may not have it for all airports, but no error is returned for that.
	GetMETAR(airports []string) (map[string]METARSOA, error)

	// Returns the item at-or-before the given time
	GetPrecipURL(tracon string, t time.Time) (string, time.Time, error)
	// Returns atmos, it's time, the time for the next one in the series.
	GetAtmosGrid(tracon string, t time.Time) (*AtmosByPointSOA, time.Time, time.Time, error)
}

type Sample struct {
	WindSample
	Temperature float32 // Celsius
	Dewpoint    float32 // Celsius
	Pressure    float32 // millibars
}

type WindSample struct {
	// WindVec is the wind velocity vector (direction air mass is moving) in nm/s.
	// When added to aircraft velocity, it represents wind's effect on the aircraft.
	WindVec [2]float32 // nm / s
}

type WindSpecifier struct {
	Direction   string `json:"direction,omitempty"`    // e.g., "30-90", "270", empty for any
	Speed       string `json:"speed,omitempty"`        // e.g., "5+", "5-", "5-15", empty for any
	FlightRules string `json:"flight_rules,omitempty"` // "VMC" or "IMC", empty defaults to "VMC"
}