gopenfusion/internal/entity/chunk.go

110 lines
2.4 KiB
Go
Raw Normal View History

2023-06-25 06:51:21 +00:00
package entity
import (
"log"
"sync"
)
type Chunk struct {
Position ChunkPosition
entities map[Entity]struct{}
2023-06-25 06:51:21 +00:00
lock sync.Mutex
}
func NewChunk(position ChunkPosition) *Chunk {
return &Chunk{
Position: position,
entities: make(map[Entity]struct{}),
2023-06-25 06:51:21 +00:00
}
}
func (c *Chunk) AddEntity(entity Entity) {
c.lock.Lock()
defer c.lock.Unlock()
c.entities[entity] = struct{}{}
2023-06-25 06:51:21 +00:00
}
func (c *Chunk) RemoveEntity(entity Entity) {
c.lock.Lock()
defer c.lock.Unlock()
delete(c.entities, entity)
2023-06-25 06:51:21 +00:00
}
// send packet to all peers in this chunk and kill each peer if error
func (c *Chunk) SendPacket(typeID uint32, pkt ...interface{}) {
c.SendPacketExclude(nil, typeID, pkt...)
}
// calls f for each entity in this chunk, if f returns true, stop iterating
// f can safely add/remove entities from the chunk
func (c *Chunk) ForEachEntity(f func(entity Entity) bool) {
// copy entities to avoid locking for the entire iteration
entities := make(map[Entity]struct{})
2023-06-25 06:51:21 +00:00
c.lock.Lock()
for entity := range c.entities {
entities[entity] = struct{}{}
}
c.lock.Unlock()
for entity := range entities {
if f(entity) {
break
}
}
}
func (c *Chunk) SendPacketExclude(exclude Entity, typeID uint32, pkt ...interface{}) {
c.ForEachEntity(func(entity Entity) bool {
// only send to players, and exclude the player that sent the packet
if entity.GetKind() != ENTITY_KIND_PLAYER || entity == exclude {
return false
2023-06-25 06:51:21 +00:00
}
plr, ok := entity.(*Player)
if !ok {
log.Panic("Chunk.SendPacket: entity kind was player, but is not a *Player")
}
peer := plr.Peer
if err := peer.Send(typeID, pkt...); err != nil {
2023-06-25 08:33:17 +00:00
log.Printf("Error sending packet to peer %p: %v", peer, err)
2023-06-25 06:51:21 +00:00
peer.Kill()
}
return false
})
2023-06-25 06:51:21 +00:00
}
func (c *Chunk) GetAdjacentPositions() []ChunkPosition {
return []ChunkPosition{
{c.Position.X - 1, c.Position.Y - 1},
{c.Position.X - 1, c.Position.Y},
{c.Position.X - 1, c.Position.Y + 1},
{c.Position.X, c.Position.Y - 1},
{c.Position.X, c.Position.Y},
{c.Position.X, c.Position.Y + 1},
{c.Position.X + 1, c.Position.Y - 1},
{c.Position.X + 1, c.Position.Y},
{c.Position.X + 1, c.Position.Y + 1},
}
}
2023-06-25 08:33:17 +00:00
// https://stackoverflow.com/a/45428032 lol
func ChunkSliceDifference(a, b []*Chunk) []*Chunk {
m := make(map[*Chunk]struct{})
for _, item := range b {
m[item] = struct{}{}
}
var diff []*Chunk
for _, item := range a {
if _, ok := m[item]; !ok {
diff = append(diff, item)
}
}
return diff
}