package ride

import (
	"fmt"
	"log/slog"
	"math"
	"slices"
	"time"

	"github.com/jftuga/geodist"
	"github.com/martinlehoux/kagamigo/kcore"
)

const ClimbDistanceMinimum = 500

// Category is the difficulty class of a climb, ordered from easiest to
// hardest.
type Category int

const (
	CategoryNo Category = iota
	CategoryFour
	CategoryThree
	CategoryTwo
	CategoryOne
	CategoryHC
)

type categoryInfo struct {
	label    string
	minScore float64
}

// categories describes each Category: its display label and the inclusive
// Cotacol score floor at which it starts applying. A score at or above
// CategoryOne's floor is Hors Catégorie; CategoryNo has no floor.
var categories = map[Category]categoryInfo{
	CategoryNo:    {label: "NO", minScore: math.Inf(-1)},
	CategoryFour:  {label: "Cat 4", minScore: 35},
	CategoryThree: {label: "Cat 3", minScore: 80},
	CategoryTwo:   {label: "Cat 2", minScore: 180},
	CategoryOne:   {label: "Cat 1", minScore: 250},
	CategoryHC:    {label: "HC", minScore: 600},
}

// climbDetectionMinimumScore is the climb-detection score below which a
// candidate segment is not reported as a climb.
const climbDetectionMinimumScore = 35

type Climb struct {
	ride      Ride
	rideStart int
	rideEnd   int
	Name      string
}

func (climb Climb) Duration() time.Duration {
	return climb.ride.Timestamp(climb.rideEnd).Sub(climb.ride.Timestamp(climb.rideStart))
}

func (climb Climb) Speed() float64 {
	return (climb.EndDistanceM() - climb.StartDistanceM()) / climb.Duration().Seconds()
}

func (climb Climb) StartIndex() int { return climb.rideStart }

func (climb Climb) EndIndex() int { return climb.rideEnd }

func (climb Climb) StartDistanceM() float64 {
	return climb.ride.DistanceM(climb.rideStart)
}

func (climb Climb) EndDistanceM() float64 {
	return climb.ride.DistanceM(climb.rideEnd)
}

func (climb Climb) StartCoord() geodist.Coord {
	return climb.ride.Coord(climb.rideStart)
}

func (climb Climb) EndCoord() geodist.Coord {
	return climb.ride.Coord(climb.rideEnd)
}

func (climb Climb) PointCoord(index int) geodist.Coord {
	kcore.Assert(index >= climb.rideStart && index <= climb.rideEnd, "point is outside climb")
	return climb.ride.Coord(index)
}

// TopIndex returns the highest point inside the climb, used as the crest anchor.
func (climb Climb) TopIndex() int {
	top := climb.rideStart
	for i := climb.rideStart; i <= climb.rideEnd; i++ {
		if climb.ride.ElevationM(i) > climb.ride.ElevationM(top) {
			top = i
		}
	}
	return top
}

func (climb Climb) TopDistanceM() float64 {
	return climb.ride.DistanceM(climb.TopIndex())
}

func (climb Climb) TopElevationM() float64 {
	return climb.ride.ElevationM(climb.TopIndex())
}

func (climb Climb) TopCoord() geodist.Coord {
	return climb.ride.Coord(climb.TopIndex())
}

func (climb Climb) String() string {
	startDistance := climb.StartDistanceM()
	endDistance := climb.EndDistanceM()
	cotacol := climb.DifficultyScore()
	body := fmt.Sprintf("%.1fkm-%.1fkm: %.1fkm at %.1f%% (%d pts - %s)", startDistance/1000, endDistance/1000, (endDistance-startDistance)/1000, Slope(climb.ride, climb.rideStart, climb.rideEnd)*100, int(cotacol), CategoryOf(cotacol))
	if climb.Name == "" {
		return body
	}
	return climb.Name + ": " + body
}

func (climb Climb) DifficultyScore() float64 {
	return difficultyScore(climb.ride, climb.rideStart, climb.rideEnd)
}

func Slope(r Ride, start, end int) float64 {
	return (r.ElevationM(end) - r.ElevationM(start)) / (r.DistanceM(end) - r.DistanceM(start))
}

func climbDetectionScore(r Ride, start, end int) float64 {
	kcore.Assert(end > start, "no points for climb detection")
	distance := r.DistanceM(end) - r.DistanceM(start)
	if distance == 0 {
		return 0
	}
	dElevation := r.ElevationM(end) - r.ElevationM(start)
	return math.Abs(dElevation) * dElevation / distance * 100.0 * 100.0 / 1000.0
}

func (c Category) String() string {
	return categories[c].label
}

// CategoryOf returns the difficulty class of a Cotacol score: the category
// with the highest floor the score reaches.
func CategoryOf(cotacol float64) Category {
	best := CategoryNo
	bestMin := math.Inf(-1)
	found := false
	for category, info := range categories {
		if cotacol >= info.minScore && (!found || info.minScore > bestMin) {
			best = category
			bestMin = info.minScore
			found = true
		}
	}
	if !found {
		// NaN reaches no floor; preserve the original switch default (HC).
		return CategoryHC
	}
	return best
}

func bestClimbBetween(r Ride, start, end int) Climb {
	kcore.Assert(end > start, "empty points")

	bestDetectionScore := climbDetectionScore(r, start, end)
	bestStart := start
	for i := start; i < end; i++ {
		detectionScore := climbDetectionScore(r, i, end)
		if detectionScore > bestDetectionScore {
			bestStart = i
			bestDetectionScore = detectionScore
		}
	}
	bestEnd := end
	for i := end; i > bestStart; i-- {
		detectionScore := climbDetectionScore(r, bestStart, i)
		if detectionScore > bestDetectionScore {
			bestEnd = i
			bestDetectionScore = detectionScore
		}
	}
	for i := bestStart; i < bestEnd; i++ {
		detectionScore := climbDetectionScore(r, i, bestEnd)
		if detectionScore > bestDetectionScore {
			bestStart = i
			bestDetectionScore = detectionScore
		}
	}
	kcore.Assert(bestStart < bestEnd, "empty climb")
	return Climb{ride: r, rideStart: bestStart, rideEnd: bestEnd}
}

func climbsBetween(r Ride, start, end int) []Climb {
	climbs := []Climb{}
	if r.DistanceM(end)-r.DistanceM(start) < ClimbDistanceMinimum {
		return climbs
	}
	slog.Debug("Searching climbs between", slog.Int("start", int(r.DistanceM(start))), slog.Int("end", int(r.DistanceM(end))))
	highest := start
	for i := start; i <= end; i++ {
		if r.ElevationM(i) > r.ElevationM(highest) {
			highest = i
		}
	}
	// TODO: Use descent to reduce recursion
	if r.DistanceM(highest)-r.DistanceM(start) < ClimbDistanceMinimum {
		return climbsBetween(r, start+1, end)
	}
	climb := bestClimbBetween(r, start, highest)
	if climbDetectionScore(r, climb.rideStart, climb.rideEnd) >= climbDetectionMinimumScore && climb.EndDistanceM()-climb.StartDistanceM() >= ClimbDistanceMinimum {
		slog.Debug("Found climb between", slog.Int("start", int(climb.StartDistanceM())), slog.Int("end", int(climb.EndDistanceM())))
		climbs = append(climbs, climb)
	}
	climbs = append(climbs, climbsBetween(r, start, climb.rideStart)...)
	climbs = append(climbs, climbsBetween(r, climb.rideEnd, end)...)

	return climbs
}

func (ride *Ride) AllClimbs() []Climb {
	climbs := climbsBetween(*ride, 0, ride.Len()-1)
	slices.SortFunc(climbs, climbCmpStart)
	return climbs
}

func climbCmpStart(a, b Climb) int { return a.rideStart - b.rideStart }