Warlock rotation rework

sim/core/buffs.go

@@ -601,6 +601,7 @@ func BloodlustAura(character *Character, actionTag int32) *Aura {
 	return aura
 }
 
+var PowerInfusionActionID = ActionID{SpellID: 10060}
 var PowerInfusionAuraTag = "PowerInfusion"
 
 const PowerInfusionDuration = time.Second * 15
@@ -616,7 +617,7 @@ func registerPowerInfusionCD(agent Agent, numPowerInfusions int32) {
 	registerExternalConsecutiveCDApproximation(
 		agent,
 		externalConsecutiveCDApproximation{
-			ActionID:         ActionID{SpellID: 10060, Tag: -1},
+			ActionID:         PowerInfusionActionID.WithTag(-1),
 			AuraTag:          PowerInfusionAuraTag,
 			CooldownPriority: CooldownPriorityDefault,
 			AuraDuration:     PowerInfusionDuration,

sim/core/cast.go

@@ -59,6 +59,7 @@ type Cast struct {
 func (cast Cast) EffectiveTime() time.Duration {
 	gcd := cast.GCD
 	if cast.GCD != 0 {
+		// TODO: isn't this wrong for spells like shadowfury, that have a reduced GCD?
 		gcd = MaxDuration(GCDMin, gcd)
 	}
 	fullCastTime := cast.CastTime + cast.ChannelTime + cast.AfterCastDelay

sim/core/spell.go

@@ -498,6 +498,18 @@ func (spell *Spell) ExpectedDamageFromCurrentSnapshot(sim *Simulation, target *U
 	return spell.expectedDamageHelper(sim, target, true)
 }
 
+// Time until either the cast is finished or GCD is ready again, whichever is longer
+func (spell *Spell) EffectiveCastTime() time.Duration {
+	// TODO: this is wrong for spells like shadowfury, that have a GCD of less than 1s
+	return MaxDuration(spell.Unit.SpellGCD(),
+		spell.Unit.ApplyCastSpeedForSpell(spell.DefaultCast.EffectiveTime(), spell))
+}
+
+// Time until the cast is finished (ignoring GCD)
+func (spell *Spell) CastTime() time.Duration {
+	return spell.Unit.ApplyCastSpeedForSpell(spell.DefaultCast.CastTime, spell)
+}
+
 // Handles computing the cost of spells and checking whether the Unit
 // meets them.
 type SpellCost interface {