Refactor stat result protos to include pet and targets spells/auras i…

…n response
wowsims · Jul 16, 2023 · f026ee3 · f026ee3
1 parent 414b1d4
commit f026ee3
Show file tree

Hide file tree

Showing 8 changed files with 67 additions and 32 deletions.
diff --git a/proto/api.proto b/proto/api.proto
@@ -314,6 +314,13 @@ message APLStats {
 	repeated APLActionStats prepull_actions = 1;
 	repeated APLActionStats priority_list = 2;
 }
+message UnitMetadata {
+	repeated SpellStats spells = 1;
+	repeated AuraStats auras = 2;
+}
+message PetStats {
+	UnitMetadata metadata = 1;
+}
 message PlayerStats {
 	// Stats
 	UnitStats base_stats = 6;
@@ -326,19 +333,26 @@ message PlayerStats {
 	repeated string sets = 3;
 	IndividualBuffs buffs = 4;
 
-	repeated SpellStats spells = 10;
-	repeated AuraStats auras = 11;
-
+	UnitMetadata metadata = 10;
 	APLStats rotation_stats = 12;
+
+	repeated PetStats pets = 11;
 }
 message PartyStats {
 	repeated PlayerStats players = 1;
 }
 message RaidStats {
 	repeated PartyStats parties = 1;
 }
+message TargetStats {
+	UnitMetadata metadata = 1;
+}
+message EncounterStats {
+	repeated TargetStats targets = 1;
+}
 message ComputeStatsResult {
 	RaidStats raid_stats = 1;
+	EncounterStats encounter_stats = 3;
 	string error_result = 2;
 }
 

diff --git a/sim/core/api.go b/sim/core/api.go
@@ -16,10 +16,11 @@ func ComputeStats(csr *proto.ComputeStatsRequest) *proto.ComputeStatsResult {
 	if encounter == nil {
 		encounter = &proto.Encounter{}
 	}
-	_, raidStats := NewEnvironment(csr.Raid, encounter)
+	_, raidStats, encounterStats := NewEnvironment(csr.Raid, encounter)
 
 	return &proto.ComputeStatsResult{
-		RaidStats: raidStats,
+		RaidStats:      raidStats,
+		EncounterStats: encounterStats,
 	}
 }
 

diff --git a/sim/core/character.go b/sim/core/character.go
@@ -503,27 +503,12 @@ func (character *Character) FillPlayerStats(playerStats *proto.PlayerStats) {
 	character.clearBuildPhaseAuras(CharacterBuildPhaseAll)
 	playerStats.Sets = character.GetActiveSetBonusNames()
 
-	playerStats.Spells = MapSlice(character.Spellbook, func(spell *Spell) *proto.SpellStats {
-		return &proto.SpellStats{
-			Id: spell.ActionID.ToProto(),
-
-			IsCastable:      spell.Flags.Matches(SpellFlagAPL),
-			IsMajorCooldown: spell.Flags.Matches(SpellFlagMCD),
-			HasDot:          spell.dots != nil || spell.aoeDot != nil,
-			PrepullOnly:     spell.Flags.Matches(SpellFlagPrepullOnly),
-		}
-	})
-
-	aplAuras := FilterSlice(character.auras, func(aura *Aura) bool {
-		return !aura.ActionID.IsEmptyAction()
-	})
-	playerStats.Auras = MapSlice(aplAuras, func(aura *Aura) *proto.AuraStats {
-		return &proto.AuraStats{
-			Id:        aura.ActionID.ToProto(),
-			MaxStacks: aura.MaxStacks,
-			HasIcd:    aura.Icd != nil,
-		}
-	})
+	playerStats.Metadata = character.GetMetadata()
+	for _, petAgent := range character.Pets {
+		playerStats.Pets = append(playerStats.Pets, &proto.PetStats{
+			Metadata: petAgent.GetPet().GetMetadata(),
+		})
+	}
 
 	if character.Rotation != nil {
 		playerStats.RotationStats = character.Rotation.getStats()

diff --git a/sim/core/environment.go b/sim/core/environment.go
@@ -45,7 +45,7 @@ type Environment struct {
 	prepullActions []PrepullAction
 }
 
-func NewEnvironment(raidProto *proto.Raid, encounterProto *proto.Encounter) (*Environment, *proto.RaidStats) {
+func NewEnvironment(raidProto *proto.Raid, encounterProto *proto.Encounter) (*Environment, *proto.RaidStats, *proto.EncounterStats) {
 	env := &Environment{
 		State: Created,
 	}
@@ -54,7 +54,14 @@ func NewEnvironment(raidProto *proto.Raid, encounterProto *proto.Encounter) (*En
 	raidStats := env.initialize(raidProto, encounterProto)
 	env.finalize(raidProto, encounterProto, raidStats)
 
-	return env, raidStats
+	encounterStats := &proto.EncounterStats{}
+	for _, target := range env.Encounter.Targets {
+		encounterStats.Targets = append(encounterStats.Targets, &proto.TargetStats{
+			Metadata: target.GetMetadata(),
+		})
+	}
+
+	return env, raidStats, encounterStats
 }
 
 // The construction phase.

diff --git a/sim/core/sim.go b/sim/core/sim.go
@@ -124,7 +124,7 @@ func NewSim(rsr *proto.RaidSimRequest) *Simulation {
 		rseed = time.Now().UnixNano()
 	}
 
-	env, _ := NewEnvironment(rsr.Raid, rsr.Encounter)
+	env, _, _ := NewEnvironment(rsr.Raid, rsr.Encounter)
 	return &Simulation{
 		Environment: env,
 		Options:     simOptions,

diff --git a/sim/core/unit.go b/sim/core/unit.go
@@ -495,3 +495,31 @@ func (unit *Unit) GetSpellsMatchingSchool(school SpellSchool) []*Spell {
 	}
 	return spells
 }
+
+func (unit *Unit) GetMetadata() *proto.UnitMetadata {
+	metadata := &proto.UnitMetadata{}
+
+	metadata.Spells = MapSlice(unit.Spellbook, func(spell *Spell) *proto.SpellStats {
+		return &proto.SpellStats{
+			Id: spell.ActionID.ToProto(),
+
+			IsCastable:      spell.Flags.Matches(SpellFlagAPL),
+			IsMajorCooldown: spell.Flags.Matches(SpellFlagMCD),
+			HasDot:          spell.dots != nil || spell.aoeDot != nil,
+			PrepullOnly:     spell.Flags.Matches(SpellFlagPrepullOnly),
+		}
+	})
+
+	aplAuras := FilterSlice(unit.auras, func(aura *Aura) bool {
+		return !aura.ActionID.IsEmptyAction()
+	})
+	metadata.Auras = MapSlice(aplAuras, func(aura *Aura) *proto.AuraStats {
+		return &proto.AuraStats{
+			Id:        aura.ActionID.ToProto(),
+			MaxStacks: aura.MaxStacks,
+			HasIcd:    aura.Icd != nil,
+		}
+	})
+
+	return metadata
+}
diff --git a/sim/rogue/rogue_test.go b/sim/rogue/rogue_test.go
@@ -118,7 +118,7 @@ func GenerateCriticalDamageMultiplierTestCase(
 		TalentsString: talents,
 	}, spec), nil, nil, nil)
 	encounter := core.MakeSingleTargetEncounter(0.0)
-	env, _ := core.NewEnvironment(raid, encounter)
+	env, _, _ := core.NewEnvironment(raid, encounter)
 	agent := env.Raid.Parties[0].Players[0]
 	rog := agent.(RogueAgent).GetRogue()
 	actualMultiplier := 0.0

diff --git a/ui/core/player.ts b/ui/core/player.ts
@@ -336,13 +336,13 @@ export class Player<SpecType extends Spec> {
 	}
 
 	private async updateSpellsAndAuras(eventID: EventID) {
-		let newSpells = this.currentStats.spells.map(spell => {
+		let newSpells = this.currentStats.metadata!.spells.map(spell => {
 			return {
 				data: spell,
 				id: ActionId.fromProto(spell.id!),
 			};
 		});
-		let newAuras = this.currentStats.auras.map(aura => {
+		let newAuras = this.currentStats.metadata!.auras.map(aura => {
 			return {
 				data: aura,
 				id: ActionId.fromProto(aura.id!),