initial commit

row.go

@@ -0,0 +1,161 @@
+package main
+
+import (
+	"iter"
+	"log/slog"
+	"reflect"
+	"strconv"
+)
+
+type Row struct {
+	File               string
+	GainSource         string
+	Speed              int
+	Stamina            int
+	Power              int
+	Guts               int
+	Wit                int
+	SP                 int
+	NumSkills          int
+	Deck1              int
+	Deck2              int
+	Deck3              int
+	Deck4              int
+	Deck5              int
+	Deck6              int
+	Races              int
+	Wins               int
+	NightOwl           bool // chara effect id 1
+	Slacker            bool // 2
+	SkinOutbreak       bool // 3
+	SlowMetabolism     bool // 4
+	Migraine           bool // 5
+	PracticePoor       bool // 6
+	FastLearner        bool // 7
+	Charming           bool // 8
+	HotTopic           bool // 9
+	PracticePerfect    bool // 10
+	VeryGoodAtTraining bool // 11
+}
+
+var headers []string
+
+func init() {
+	t := reflect.TypeFor[Row]()
+	for f := range t.Fields() {
+		headers = append(headers, f.Name)
+	}
+}
+
+func ismRows(key string, ism IdleSingleModeProgressLog) iter.Seq[Row] {
+	return func(yield func(Row) bool) {
+		races := len(ism.RaceHistory)
+		var wins int
+		for _, r := range ism.RaceHistory {
+			if r.RaceHistory.ResultRank == 1 {
+				wins++
+			}
+		}
+		var conditions [11]bool
+		for _, v := range ism.CharaEffectLog {
+			if v.CharaEffectID < 1 || v.CharaEffectID > 11 {
+				slog.Warn("skipping unknown charaEffectID", slog.Int("id", int(v.CharaEffectID)), slog.Bool("active", v.IsActive))
+				continue
+			}
+			conditions[v.CharaEffectID-1] = true
+		}
+		row := Row{
+			File:               key,
+			Deck1:              int(ism.SupportCardGains[0].SupportCardID),
+			Deck2:              int(ism.SupportCardGains[1].SupportCardID),
+			Deck3:              int(ism.SupportCardGains[2].SupportCardID),
+			Deck4:              int(ism.SupportCardGains[3].SupportCardID),
+			Deck5:              int(ism.SupportCardGains[4].SupportCardID),
+			Deck6:              int(ism.SupportCardGains[5].SupportCardID),
+			Races:              races,
+			Wins:               wins,
+			NightOwl:           conditions[0],
+			Slacker:            conditions[1],
+			SkinOutbreak:       conditions[2],
+			SlowMetabolism:     conditions[3],
+			Migraine:           conditions[4],
+			PracticePoor:       conditions[5],
+			FastLearner:        conditions[6],
+			Charming:           conditions[7],
+			HotTopic:           conditions[8],
+			PracticePerfect:    conditions[9],
+			VeryGoodAtTraining: conditions[10],
+		}
+
+		row.GainSource = "Events"
+		row.Speed = int(ism.EventGain.Speed.Value)
+		row.Stamina = int(ism.EventGain.Stamina.Value)
+		row.Power = int(ism.EventGain.Power.Value)
+		row.Guts = int(ism.EventGain.Guts.Value)
+		row.Wit = int(ism.EventGain.Wiz.Value)
+		row.SP = int(ism.EventGain.SkillPoint)
+		row.NumSkills = len(ism.EventGain.SkillTips)
+		if !yield(row) {
+			return
+		}
+
+		row.GainSource = "Inspiration"
+		row.Speed = int(ism.SuccessionGain.Speed.Value)
+		row.Stamina = int(ism.SuccessionGain.Stamina.Value)
+		row.Power = int(ism.SuccessionGain.Power.Value)
+		row.Guts = int(ism.SuccessionGain.Guts.Value)
+		row.Wit = int(ism.SuccessionGain.Wiz.Value)
+		row.SP = int(ism.SuccessionGain.SkillPoint)
+		row.NumSkills = len(ism.SuccessionGain.SkillTips)
+		if !yield(row) {
+			return
+		}
+
+		for _, sup := range ism.SupportCardGains {
+			row.GainSource = strconv.Itoa(int(sup.SupportCardID))
+			row.Speed = int(sup.GainInfo.Speed.Value)
+			row.Stamina = int(sup.GainInfo.Stamina.Value)
+			row.Power = int(sup.GainInfo.Power.Value)
+			row.Guts = int(sup.GainInfo.Guts.Value)
+			row.Wit = int(sup.GainInfo.Wiz.Value)
+			row.SP = int(sup.GainInfo.SkillPoint)
+			row.NumSkills = len(sup.GainInfo.SkillTips)
+			if !yield(row) {
+				return
+			}
+		}
+	}
+}
+
+func (r Row) csv() []string {
+	return []string{
+		r.File,
+		r.GainSource,
+		strconv.Itoa(r.Speed),
+		strconv.Itoa(r.Stamina),
+		strconv.Itoa(r.Power),
+		strconv.Itoa(r.Guts),
+		strconv.Itoa(r.Wit),
+		strconv.Itoa(r.SP),
+		strconv.Itoa(r.NumSkills),
+		strconv.Itoa(r.Deck1),
+		strconv.Itoa(r.Deck2),
+		strconv.Itoa(r.Deck3),
+		strconv.Itoa(r.Deck4),
+		strconv.Itoa(r.Deck5),
+		strconv.Itoa(r.Deck6),
+		strconv.Itoa(r.Races),
+		strconv.Itoa(r.Wins),
+		strconv.FormatBool(r.NightOwl),
+		strconv.FormatBool(r.Slacker),
+		strconv.FormatBool(r.SkinOutbreak),
+		strconv.FormatBool(r.SlowMetabolism),
+		strconv.FormatBool(r.Migraine),
+		strconv.FormatBool(r.PracticePoor),
+		strconv.FormatBool(r.FastLearner),
+		strconv.FormatBool(r.Charming),
+		strconv.FormatBool(r.HotTopic),
+		strconv.FormatBool(r.PracticePerfect),
+		strconv.FormatBool(r.VeryGoodAtTraining),
+	}
+}