90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
package d2mapstamp
|
|
|
|
import (
|
|
"log"
|
|
"math"
|
|
"math/rand"
|
|
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2core/d2map/d2mapentity"
|
|
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2common/d2enum"
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2common/d2fileformats/d2ds1"
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2common/d2fileformats/d2dt1"
|
|
"github.com/OpenDiablo2/OpenDiablo2/d2core/d2asset"
|
|
)
|
|
|
|
// NewStampFactory creates a MapStamp factory instance
|
|
func NewStampFactory(asset *d2asset.AssetManager, entity *d2mapentity.MapEntityFactory) *StampFactory {
|
|
return &StampFactory{asset, entity}
|
|
}
|
|
|
|
// StampFactory is responsible for loading map stamps. A stamp can be thought of like a
|
|
// preset map configuration, like the various configurations of Act 1 town.
|
|
type StampFactory struct {
|
|
asset *d2asset.AssetManager
|
|
entity *d2mapentity.MapEntityFactory
|
|
}
|
|
|
|
// LoadStamp loads the Stamp data from file, using the given level type, level preset index, and
|
|
// level file index.
|
|
func (f *StampFactory) LoadStamp(levelType d2enum.RegionIdType, levelPreset, fileIndex int) *Stamp {
|
|
stamp := &Stamp{
|
|
factory: f,
|
|
entity: f.entity,
|
|
regionID: levelType,
|
|
levelType: *f.asset.Records.Level.Types[levelType],
|
|
levelPreset: f.asset.Records.Level.Presets[levelPreset],
|
|
}
|
|
|
|
for _, levelTypeDt1 := range &stamp.levelType.Files {
|
|
if levelTypeDt1 == "" || levelTypeDt1 == "0" {
|
|
continue
|
|
}
|
|
|
|
fileData, err := f.asset.LoadFile("/data/global/tiles/" + levelTypeDt1)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
dt1, err := d2dt1.LoadDT1(fileData)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return nil
|
|
}
|
|
|
|
stamp.tiles = append(stamp.tiles, dt1.Tiles...)
|
|
}
|
|
|
|
var levelFilesToPick []string
|
|
|
|
for _, fileRecord := range stamp.levelPreset.Files {
|
|
if fileRecord != "" && fileRecord != "0" {
|
|
levelFilesToPick = append(levelFilesToPick, fileRecord)
|
|
}
|
|
}
|
|
|
|
levelIndex := int(math.Round(float64(len(levelFilesToPick)-1) * rand.Float64()))
|
|
if fileIndex >= 0 && fileIndex < len(levelFilesToPick) {
|
|
levelIndex = fileIndex
|
|
}
|
|
|
|
if levelFilesToPick == nil {
|
|
panic("no level files to pick from")
|
|
}
|
|
|
|
stamp.regionPath = levelFilesToPick[levelIndex]
|
|
fileData, err := f.asset.LoadFile("/data/global/tiles/" + stamp.regionPath)
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
stamp.ds1, err = d2ds1.LoadDS1(fileData)
|
|
if err != nil {
|
|
log.Print(err)
|
|
return nil
|
|
}
|
|
|
|
return stamp
|
|
}
|