d2go/pkg/memory/monsters.go

package memory

import (
	"sort"

	"github.com/hectorgimenez/d2go/pkg/data"
	"github.com/hectorgimenez/d2go/pkg/data/npc"
	"github.com/hectorgimenez/d2go/pkg/data/stat"
	"github.com/hectorgimenez/d2go/pkg/utils"
)

func (gd *GameReader) Monsters(playerPosition data.Position, hover data.HoverData) data.Monsters {
	baseAddr := gd.Process.moduleBaseAddressPtr + gd.offset.UnitTable + 1024
	unitTableBuffer := gd.Process.ReadBytesFromMemory(baseAddr, 128*8)

	monsters := data.Monsters{}
	for i := 0; i < 128; i++ {
		monsterOffset := 8 * i
		monsterUnitPtr := uintptr(ReadUIntFromBuffer(unitTableBuffer, uint(monsterOffset), Uint64))
		for monsterUnitPtr > 0 {
			monsterDataBuffer := gd.Process.ReadBytesFromMemory(monsterUnitPtr, 144)

			//monsterType := ReadUIntFromBuffer(monsterDataBuffer, 0x00, Uint32)
			txtFileNo := ReadUIntFromBuffer(monsterDataBuffer, 0x04, Uint32)
			unitID := ReadUIntFromBuffer(monsterDataBuffer, 0x08, Uint32)

			//mode := ReadUIntFromBuffer(monsterDataBuffer, 0x0C, Uint32)

			unitDataPtr := uintptr(ReadUIntFromBuffer(monsterDataBuffer, 0x10, Uint64))
			//isUnique := gd.Process.ReadUInt(unitDataPtr+0x18, Uint16)
			flag := gd.Process.ReadBytesFromMemory(unitDataPtr+0x1A, Uint8)[0]
			isCorpse := gd.Process.ReadUInt(monsterUnitPtr+0x1A6, Uint8)

			//unitDataBuffer := gd.Process.ReadBytesFromMemory(unitDataPtr, 144)

			// Coordinates (X, Y)
			pathPtr := uintptr(gd.Process.ReadUInt(monsterUnitPtr+0x38, Uint64))
			posX := gd.Process.ReadUInt(pathPtr+0x02, Uint16)
			posY := gd.Process.ReadUInt(pathPtr+0x06, Uint16)

			hovered := false
			if hover.IsHovered && hover.UnitType == 1 && hover.UnitID == data.UnitID(unitID) {
				hovered = true
			}

			statsListExPtr := uintptr(ReadUIntFromBuffer(monsterDataBuffer, 0x88, Uint64))
			statPtr := uintptr(gd.Process.ReadUInt(statsListExPtr+0x30, Uint64))
			statCount := gd.Process.ReadUInt(statsListExPtr+0x38, Uint64)

			stats := gd.getMonsterStats(statCount, statPtr)

			// This excludes good NPCs but includes Mercs
			if !gd.shouldBeIgnored(txtFileNo) || stats[stat.Experience] > 0 {
				m := data.Monster{
					UnitID:    data.UnitID(unitID),
					Name:      npc.ID(int(txtFileNo)),
					IsHovered: hovered,
					Position: data.Position{
						X: int(posX),
						Y: int(posY),
					},
					Stats: stats,
					Type:  getMonsterType(flag),
				}

				if isCorpse == 0 {
					monsters = append(monsters, m)
				}
			}

			monsterUnitPtr = uintptr(gd.Process.ReadUInt(monsterUnitPtr+0x150, Uint64))
		}
	}

	sort.SliceStable(monsters, func(i, j int) bool {
		distanceI := utils.DistanceFromPoint(playerPosition, monsters[i].Position)
		distanceJ := utils.DistanceFromPoint(playerPosition, monsters[j].Position)

		return distanceI < distanceJ
	})

	return monsters
}

func getMonsterType(typeFlag byte) data.MonsterType {
	switch typeFlag {
	case 10:
		return data.MonsterTypeSuperUnique
	case 1 << 2, 12:
		return data.MonsterTypeChampion
	case 1 << 3:
		return data.MonsterTypeUnique
	case 1 << 4:
		return data.MonsterTypeMinion
	}

	return data.MonsterTypeNone
}

func (gd *GameReader) getMonsterStats(statCount uint, statPtr uintptr) map[stat.ID]int {
	stats := map[stat.ID]int{}

	if statCount > 0 {
		statBuffer := gd.Process.ReadBytesFromMemory(statPtr+0x2, statCount*8)
		for i := 0; i < int(statCount); i++ {
			offset := uint(i * 8)
			statEnum := ReadUIntFromBuffer(statBuffer, offset, Uint16)
			statValue := ReadUIntFromBuffer(statBuffer, offset+0x2, Uint32)
			stats[stat.ID(statEnum)] = int(statValue)
		}
	}

	return stats
}

func (gd *GameReader) Corpses(playerPosition data.Position, hover data.HoverData) data.Monsters {
	baseAddr := gd.Process.moduleBaseAddressPtr + gd.offset.UnitTable + 1024
	unitTableBuffer := gd.Process.ReadBytesFromMemory(baseAddr, 128*8)

	corpses := data.Monsters{}

	for i := 0; i < 128; i++ {
		monsterOffset := 8 * i
		monsterUnitPtr := uintptr(ReadUIntFromBuffer(unitTableBuffer, uint(monsterOffset), Uint64))
		for monsterUnitPtr > 0 {
			monsterDataBuffer := gd.Process.ReadBytesFromMemory(monsterUnitPtr, 144)

			txtFileNo := ReadUIntFromBuffer(monsterDataBuffer, 0x04, Uint32)
			unitID := ReadUIntFromBuffer(monsterDataBuffer, 0x08, Uint32)

			unitDataPtr := uintptr(ReadUIntFromBuffer(monsterDataBuffer, 0x10, Uint64))
			flag := gd.Process.ReadBytesFromMemory(unitDataPtr+0x1A, Uint8)[0]
			isCorpse := gd.Process.ReadUInt(monsterUnitPtr+0x1A6, Uint8)

			if isCorpse == 0 {
				monsterUnitPtr = uintptr(gd.Process.ReadUInt(monsterUnitPtr+0x150, Uint64))
				continue
			}

			pathPtr := uintptr(gd.Process.ReadUInt(monsterUnitPtr+0x38, Uint64))
			posX := gd.Process.ReadUInt(pathPtr+0x02, Uint16)
			posY := gd.Process.ReadUInt(pathPtr+0x06, Uint16)

			statsListExPtr := uintptr(ReadUIntFromBuffer(monsterDataBuffer, 0x88, Uint64))
			statPtr := uintptr(gd.Process.ReadUInt(statsListExPtr+0x30, Uint64))
			statCount := gd.Process.ReadUInt(statsListExPtr+0x38, Uint64)

			stats := gd.getMonsterStats(statCount, statPtr)

			hovered := hover.IsHovered && hover.UnitType == 1 && hover.UnitID == data.UnitID(unitID)

			if (!gd.shouldBeIgnored(txtFileNo) || stats[stat.Experience] > 0) && isCorpse != 0 {
				m := data.Monster{
					UnitID:    data.UnitID(unitID),
					Name:      npc.ID(int(txtFileNo)),
					IsHovered: hovered,
					Position: data.Position{
						X: int(posX),
						Y: int(posY),
					},
					Stats: stats,
					Type:  getMonsterType(flag),
				}

				corpses = append(corpses, m)
			}

			monsterUnitPtr = uintptr(gd.Process.ReadUInt(monsterUnitPtr+0x150, Uint64))
		}
	}

	sort.SliceStable(corpses, func(i, j int) bool {
		distanceI := utils.DistanceFromPoint(playerPosition, corpses[i].Position)
		distanceJ := utils.DistanceFromPoint(playerPosition, corpses[j].Position)
		return distanceI < distanceJ
	})

	return corpses
}

func (gd *GameReader) shouldBeIgnored(txtNo uint) bool {
	switch npc.ID(txtNo) {
	case npc.Chicken,
		npc.Rat,
		npc.Rogue,
		npc.HellMeteor,
		npc.Bird,
		npc.Bird2,
		npc.Bat,
		npc.Act2Male,
		npc.Act2Female,
		npc.Act2Child,
		npc.Cow,
		npc.Camel,
		npc.Act2Guard,
		npc.Act2Vendor,
		npc.Act2Vendor2,
		npc.Maggot,
		npc.Bug,
		npc.Scorpion,
		npc.Rogue2,
		npc.Rogue3,
		npc.Larva,
		npc.Familiar,
		npc.Act3Male,
		npc.Act3Female,
		npc.Snake,
		npc.Parrot,
		npc.Fish,
		npc.EvilHole,
		npc.EvilHole2,
		npc.EvilHole3,
		npc.EvilHole4,
		npc.EvilHole5,
		npc.FireboltTrap,
		npc.HorzMissileTrap,
		npc.VertMissileTrap,
		npc.PoisonCloudTrap,
		npc.LightningTrap,
		npc.InvisoSpawner,
		npc.Guard,
		npc.MiniSper,
		npc.BoneWall,
		npc.Hydra,
		npc.Hydra2,
		npc.Hydra3,
		npc.SevenTombs,
		npc.IronWolf,
		npc.CompellingOrbNpc,
		npc.SpiritMummy,
		npc.Act2Guard4,
		npc.Act2Guard5,
		npc.Window,
		npc.Window2,
		npc.MephistoSpirit,
		npc.WakeOfDestruction,
		npc.ChargedBoltSentry,
		npc.LightningSentry,
		npc.InvisiblePet,
		npc.InfernoSentry,
		npc.DeathSentry,
		npc.BaalThrone,
		npc.InjuredBarbarian,
		npc.InjuredBarbarian2,
		npc.InjuredBarbarian3,
		npc.DemonHole:
		return true
	}

	return false
}