1 files changed, 286 insertions, 0 deletions
diff --git a/internal/ridetest/ride.go b/internal/ridetest/ride.go
new file mode 100644
index 0000000..a6413a9
--- /dev/null
+++ b/internal/ridetest/ride.go
@@ -0,0 +1,286 @@
+// Package ridetest builds ride.Ride test fixtures from a compact string DSL.
+//
+// Grammar: sections separated by "|", each "<distance>km at <slope>%" with
+// optional per-section metrics "hr=150 cad=80 w=200":
+//
+//	ridetest.MustRide("2km at 7% | 1km at -7%")
+//
+// Options adjust the generated columns: Sampling (step in meters, default
+// 100), StartCoord (first point; later points progress northeast by 0.01° of
+// latitude and longitude per kilometer), StartTime (default the zero time,
+// one second between points) and StartElevation (default 0).
+package ridetest
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/jftuga/geodist"
+	"github.com/martinlehoux/biking_home/ride"
+)
+
+const defaultSamplingM = 100
+
+// headingPerKm is the latitude and longitude offset applied to each point per
+// kilometer of ride, giving StartCoord rides a fixed northeast direction.
+const headingPerKm = 0.01
+
+type section struct {
+	distanceM float64
+	slope     float64
+	heartRate *float64
+	cadence   *float64
+	power     *float64
+}
+
+type point struct {
+	distanceM  float64
+	elevationM float64
+	coord      geodist.Coord
+	timestamp  time.Time
+	section    section
+}
+
+type options struct {
+	samplingM  float64
+	startCoord *geodist.Coord
+	startTime  time.Time
+	startElevM float64
+}
+
+// Option configures the ride generated by Ride.
+type Option func(*options)
+
+// Sampling sets the distance between consecutive points in meters
+// (default 100).
+func Sampling(m float64) Option {
+	return func(o *options) { o.samplingM = m }
+}
+
+// StartCoord sets the coordinate of the first point. Each following point
+// progresses northeast by 0.01° of latitude and longitude per kilometer.
+func StartCoord(lat, lon float64) Option {
+	return func(o *options) { o.startCoord = &geodist.Coord{Lat: lat, Lon: lon} }
+}
+
+// StartTime sets the timestamp of the first point; each following point is
+// one second later (default: the zero time).
+func StartTime(t time.Time) Option {
+	return func(o *options) { o.startTime = t }
+}
+
+// StartElevation sets the elevation of the first point, in meters (default 0).
+func StartElevation(m float64) Option {
+	return func(o *options) { o.startElevM = m }
+}
+
+// Ride parses a ride description into a ride.Ride.
+//
+// Example: "2km at 7% | 1km at -7%".
+func Ride(dsl string, opts ...Option) (ride.Ride, error) {
+	o := options{samplingM: defaultSamplingM}
+	for _, opt := range opts {
+		opt(&o)
+	}
+	if o.samplingM <= 0 {
+		return ride.Ride{}, fmt.Errorf("ridetest: sampling must be positive, got %v", o.samplingM)
+	}
+	sections, err := parseSections(dsl)
+	if err != nil {
+		return ride.Ride{}, err
+	}
+	if len(sections) == 0 {
+		return ride.Ride{}, fmt.Errorf("ridetest: empty ride description")
+	}
+	for _, metric := range []struct {
+		name string
+		get  func(section) *float64
+	}{
+		{"hr", func(s section) *float64 { return s.heartRate }},
+		{"cad", func(s section) *float64 { return s.cadence }},
+		{"w", func(s section) *float64 { return s.power }},
+	} {
+		present, complete := metricComplete(sections, metric.get)
+		if present && !complete {
+			return ride.Ride{}, fmt.Errorf("ridetest: %s= must be set on every section or none", metric.name)
+		}
+	}
+	return buildRide(sections, o), nil
+}
+
+// MustRide is like Ride but panics on a malformed description; use it inside
+// tests where the description is a static program constant.
+func MustRide(dsl string, opts ...Option) ride.Ride {
+	ride, err := Ride(dsl, opts...)
+	if err != nil {
+		panic(err)
+	}
+	return ride
+}
+
+func parseSections(dsl string) ([]section, error) {
+	rawSections := strings.Split(dsl, "|")
+	sections := make([]section, 0, len(rawSections))
+	for _, raw := range rawSections {
+		raw = strings.TrimSpace(raw)
+		if raw == "" {
+			return nil, fmt.Errorf("ridetest: empty section in %q", dsl)
+		}
+		sec, err := parseSection(raw)
+		if err != nil {
+			return nil, err
+		}
+		sections = append(sections, sec)
+	}
+	return sections, nil
+}
+
+// parseSection parses a single section like "2km at 7% hr=150 cad=80 w=200".
+func parseSection(raw string) (section, error) {
+	fields := strings.Fields(raw)
+	if len(fields) < 3 {
+		return section{}, fmt.Errorf("ridetest: section %q must look like \"2km at 7%%\"", raw)
+	}
+	if fields[1] != "at" {
+		return section{}, fmt.Errorf("ridetest: section %q: expected \"at\" between distance and slope, got %q", raw, fields[1])
+	}
+	distanceKm, err := parseUnit(fields[0], "km")
+	if err != nil {
+		return section{}, fmt.Errorf("ridetest: section %q: %w", raw, err)
+	}
+	slopePercent, err := parseUnit(fields[2], "%")
+	if err != nil {
+		return section{}, fmt.Errorf("ridetest: section %q: %w", raw, err)
+	}
+	if distanceKm <= 0 {
+		return section{}, fmt.Errorf("ridetest: section %q: distance must be positive", raw)
+	}
+	sec := section{
+		distanceM: distanceKm * 1000,
+		slope:     slopePercent / 100,
+	}
+	for _, token := range fields[3:] {
+		var value *float64
+		switch {
+		case strings.HasPrefix(token, "hr="):
+			value, err = parseMetric(token, "hr")
+			sec.heartRate = value
+		case strings.HasPrefix(token, "cad="):
+			value, err = parseMetric(token, "cad")
+			sec.cadence = value
+		case strings.HasPrefix(token, "w="):
+			value, err = parseMetric(token, "w")
+			sec.power = value
+		default:
+			return section{}, fmt.Errorf("ridetest: section %q: unknown metric %q (want hr=, cad= or w=)", raw, token)
+		}
+		if err != nil {
+			return section{}, fmt.Errorf("ridetest: section %q: %w", raw, err)
+		}
+	}
+	return sec, nil
+}
+
+func parseUnit(token, unit string) (float64, error) {
+	if !strings.HasSuffix(token, unit) {
+		return 0, fmt.Errorf("expected a value in %s, got %q", unit, token)
+	}
+	return strconv.ParseFloat(strings.TrimSuffix(token, unit), 64)
+}
+
+func parseMetric(token, name string) (*float64, error) {
+	value, err := strconv.ParseFloat(strings.TrimPrefix(token, name+"="), 64)
+	if err != nil {
+		return nil, fmt.Errorf("invalid %s value %q", name, token)
+	}
+	return &value, nil
+}
+
+// metricComplete reports whether a metric column is present (any section sets
+// it) and complete (every section sets it).
+func metricComplete(sections []section, get func(section) *float64) (present, complete bool) {
+	set := 0
+	for _, sec := range sections {
+		if get(sec) != nil {
+			set++
+		}
+	}
+	return set > 0, set == len(sections)
+}
+
+func buildRide(sections []section, o options) ride.Ride {
+	points := buildPoints(sections, o)
+	distances := make([]float64, len(points))
+	elevations := make([]float64, len(points))
+	coords := make([]geodist.Coord, len(points))
+	timestamps := make([]time.Time, len(points))
+	var heartRates, cadences, powers []float64
+	if sections[0].heartRate != nil {
+		heartRates = make([]float64, len(points))
+	}
+	if sections[0].cadence != nil {
+		cadences = make([]float64, len(points))
+	}
+	if sections[0].power != nil {
+		powers = make([]float64, len(points))
+	}
+	for i, p := range points {
+		distances[i] = p.distanceM
+		elevations[i] = p.elevationM
+		coords[i] = p.coord
+		timestamps[i] = p.timestamp
+		if heartRates != nil {
+			heartRates[i] = *p.section.heartRate
+		}
+		if cadences != nil {
+			cadences[i] = *p.section.cadence
+		}
+		if powers != nil {
+			powers[i] = *p.section.power
+		}
+	}
+	return ride.FromColumnsWithMetrics(distances, elevations, coords, timestamps, heartRates, cadences, powers)
+}
+
+func buildPoints(sections []section, o options) []point {
+	lat, lon := 0.0, 0.0
+	if o.startCoord != nil {
+		lat, lon = o.startCoord.Lat, o.startCoord.Lon
+	}
+	// Coordinates derive from the total distance so that exact multiples of
+	// the per-kilometer heading (0.01° x km as float64) stay exact; a running
+	// accumulation drifts after a few points.
+	coord := func(distanceM float64) geodist.Coord {
+		return geodist.Coord{Lat: lat + headingPerKm*(distanceM/1000), Lon: lon + headingPerKm*(distanceM/1000)}
+	}
+	points := make([]point, 0, len(sections)+1)
+	elevation := o.startElevM
+	timestamp := o.startTime
+	points = append(points, point{
+		distanceM:  0,
+		elevationM: elevation,
+		coord:      coord(0),
+		timestamp:  timestamp,
+		section:    sections[0],
+	})
+	distance := 0.0
+	for _, sec := range sections {
+		curSecDist := 0.0
+		for curSecDist < sec.distanceM {
+			curSecDist += o.samplingM
+			elevation += o.samplingM * sec.slope
+			timestamp = timestamp.Add(time.Second)
+			points = append(points, point{
+				distanceM:  distance + curSecDist,
+				elevationM: elevation,
+				coord:      coord(distance + curSecDist),
+				timestamp:  timestamp,
+				section:    sec,
+			})
+		}
+		distance += curSecDist
+	}
+	return points
+}