stats.go

package tok

import "github.com/GrayCodeAI/tok/internal/filter"

// Stats contains compression statistics.
type Stats struct {
	OriginalTokens   int
	FinalTokens      int
	TokensSaved      int
	ReductionPercent float64
	Layers           map[string]LayerStat
}

// LayerStat contains per-layer statistics.
type LayerStat struct {
	TokensSaved int
	DurationMs  int64
}

func newStats(ps *filter.PipelineStats) Stats {
	if ps == nil {
		return Stats{}
	}

	layers := make(map[string]LayerStat, len(ps.LayerStats))
	for name, ls := range ps.LayerStats {
		layers[name] = LayerStat{
			TokensSaved: ls.TokensSaved,
			DurationMs:  ls.Duration,
		}
	}

	return Stats{
		OriginalTokens:   ps.OriginalTokens,
		FinalTokens:      ps.FinalTokens,
		TokensSaved:      ps.TotalSaved,
		ReductionPercent: ps.ReductionPercent,
		Layers:           layers,
	}
}