OpenDiablo2/d2core/d2records/treasure_class_loader.go

package d2records

import (
	"fmt"

	"github.com/OpenDiablo2/OpenDiablo2/d2common/d2fileformats/d2txt"
)

const (
	maxTreasuresPerRecord = 10
	treasureItemFmt       = "Item%d"
	treasureProbFmt       = "Prob%d"
)

func treasureClassLoader(r *RecordManager, d *d2txt.DataDictionary) error {
	records, err := treasureClassCommonLoader(d)
	if err != nil {
		return err
	}

	r.Item.Treasure.Normal = records

	r.Debugf("Loaded %d TreasureClass (normal) records", len(records))

	return nil
}

func treasureClassExLoader(r *RecordManager, d *d2txt.DataDictionary) error {
	records, err := treasureClassCommonLoader(d)
	if err != nil {
		return err
	}

	r.Item.Treasure.Expansion = records

	r.Debugf("Loaded %d TreasureClass (expansion) records", len(records))

	return nil
}

func treasureClassCommonLoader(d *d2txt.DataDictionary) (TreasureClass, error) {
	records := make(TreasureClass)

	for d.Next() {
		record := &TreasureClassRecord{
			Name:       d.String("Treasure Class"),
			Group:      d.Number("group"),
			Level:      d.Number("level"),
			NumPicks:   d.Number("Picks"),
			FreqUnique: d.Number("Unique"),
			FreqSet:    d.Number("Set"),
			FreqRare:   d.Number("Rare"),
			FreqMagic:  d.Number("Magic"),
			FreqNoDrop: d.Number("NoDrop"),
		}

		if record.Name == "" {
			continue
		}

		for treasureIdx := 0; treasureIdx < maxTreasuresPerRecord; treasureIdx++ {
			treasureColumnKey := fmt.Sprintf(treasureItemFmt, treasureIdx+1)
			probColumnKey := fmt.Sprintf(treasureProbFmt, treasureIdx+1)

			treasureName := d.String(treasureColumnKey)
			if treasureName == "" {
				continue
			}

			prob := d.Number(probColumnKey)

			treasure := &Treasure{
				Code:        treasureName,
				Probability: prob,
			}

			if record.Treasures == nil {
				record.Treasures = []*Treasure{treasure}
			} else {
				record.Treasures = append(record.Treasures, treasure)
			}
		}

		records[record.Name] = record
	}

	return records, d.Err
}