1
1
mirror of https://github.com/OpenDiablo2/OpenDiablo2 synced 2024-06-05 07:20:42 +00:00
OpenDiablo2/d2core/d2stats/diablo2stats/stat_factory.go
2020-10-21 23:41:21 -07:00

138 lines
3.4 KiB
Go

package diablo2stats
import (
"fmt"
"github.com/OpenDiablo2/OpenDiablo2/d2common/d2enum"
"github.com/OpenDiablo2/OpenDiablo2/d2common/d2fileformats/d2tbl"
"github.com/OpenDiablo2/OpenDiablo2/d2core/d2asset"
"github.com/OpenDiablo2/OpenDiablo2/d2core/d2stats"
)
// NewStatFactory creates a new stat factory instance
func NewStatFactory(asset *d2asset.AssetManager) (*StatFactory, error) {
factory := &StatFactory{asset: asset}
return factory, nil
}
// StatFactory is responsible for creating stats
type StatFactory struct {
asset *d2asset.AssetManager
}
// NewStat creates a stat instance with the given record and values
func (f *StatFactory) NewStat(key string, values ...float64) d2stats.Stat {
record := f.asset.Records.Item.Stats[key]
if record == nil {
return nil
}
stat := &diablo2Stat{
factory: f,
record: record,
}
stat.init(values...) // init stat values, value types, and value combination rules
return stat
}
// NewStatList creates a stat list
func (f *StatFactory) NewStatList(stats ...d2stats.Stat) d2stats.StatList {
return &Diablo2StatList{stats}
}
// NewValue creates a stat value of the given type
func (f *StatFactory) NewValue(t d2stats.StatNumberType, c d2stats.ValueCombineType) d2stats.StatValue {
sv := &Diablo2StatValue{
numberType: t,
combineType: c,
}
switch t {
case d2stats.StatValueFloat:
sv.stringerFn = f.stringerUnsignedFloat
case d2stats.StatValueInt:
sv.stringerFn = f.stringerUnsignedInt
default:
sv.stringerFn = f.stringerEmpty
}
return sv
}
const (
monsterNotFound = "{Monster not found!}"
)
func (f *StatFactory) getHeroMap() []d2enum.Hero {
return []d2enum.Hero{
d2enum.HeroAmazon,
d2enum.HeroSorceress,
d2enum.HeroNecromancer,
d2enum.HeroPaladin,
d2enum.HeroBarbarian,
d2enum.HeroDruid,
d2enum.HeroAssassin,
}
}
func (f *StatFactory) stringerUnsignedInt(sv d2stats.StatValue) string {
return fmt.Sprintf("%d", sv.Int())
}
func (f *StatFactory) stringerUnsignedFloat(sv d2stats.StatValue) string {
return fmt.Sprintf("%.2f", sv.Float())
}
func (f *StatFactory) stringerEmpty(_ d2stats.StatValue) string {
return ""
}
func (f *StatFactory) stringerIntSigned(sv d2stats.StatValue) string {
return fmt.Sprintf("%+d", sv.Int())
}
func (f *StatFactory) stringerIntPercentageSigned(sv d2stats.StatValue) string {
return fmt.Sprintf("%+d%%", sv.Int())
}
func (f *StatFactory) stringerIntPercentageUnsigned(sv d2stats.StatValue) string {
return fmt.Sprintf("%d%%", sv.Int())
}
func (f *StatFactory) stringerClassAllSkills(sv d2stats.StatValue) string {
heroIndex := sv.Int()
heroMap := f.getHeroMap()
classRecord := f.asset.Records.Character.Stats[heroMap[heroIndex]]
return d2tbl.TranslateString(classRecord.SkillStrAll)
}
func (f *StatFactory) stringerClassOnly(sv d2stats.StatValue) string {
heroMap := f.getHeroMap()
heroIndex := sv.Int()
classRecord := f.asset.Records.Character.Stats[heroMap[heroIndex]]
classOnlyKey := classRecord.SkillStrClassOnly
return d2tbl.TranslateString(classOnlyKey)
}
func (f *StatFactory) stringerSkillName(sv d2stats.StatValue) string {
skillRecord := f.asset.Records.Skill.Details[sv.Int()]
return skillRecord.Skill
}
func (f *StatFactory) stringerMonsterName(sv d2stats.StatValue) string {
for key := range f.asset.Records.Monster.Stats {
if f.asset.Records.Monster.Stats[key].ID == sv.Int() {
return f.asset.Records.Monster.Stats[key].NameString
}
}
return monsterNotFound
}