gopenfusion/shard/entitymanager.go

123 lines
3.2 KiB
Go
Raw Normal View History

2023-06-25 06:51:21 +00:00
package shard
2023-06-25 08:33:17 +00:00
import (
2023-08-23 23:16:24 +00:00
"github.com/CPunch/gopenfusion/internal/entity"
2023-06-25 08:33:17 +00:00
)
2023-06-25 06:51:21 +00:00
func (server *ShardServer) addEntity(e entity.Entity) {
pos := e.GetChunk()
server.addEntityToChunks(server.getViewableChunks(pos), e)
server.getChunk(pos).AddEntity(e)
}
func (server *ShardServer) removeEntity(e entity.Entity) {
2023-06-28 02:49:11 +00:00
// TODO: chunk cleanup
pos := e.GetChunk()
server.removeEntityFromChunks(server.getViewableChunks(pos), e)
server.getChunk(pos).RemoveEntity(e)
}
2023-06-25 06:51:21 +00:00
func (server *ShardServer) getChunk(pos entity.ChunkPosition) *entity.Chunk {
chunk, ok := server.chunks[pos]
if !ok {
chunk = entity.NewChunk(pos)
server.chunks[pos] = chunk
}
return chunk
}
2023-06-25 08:33:17 +00:00
func (server *ShardServer) getViewableChunks(pos entity.ChunkPosition) []*entity.Chunk {
2023-06-25 06:51:21 +00:00
chunks := make([]*entity.Chunk, 0, 9)
2023-06-25 08:33:17 +00:00
for _, pos := range server.getChunk(pos).GetAdjacentPositions() {
2023-06-25 06:51:21 +00:00
chunks = append(chunks, server.getChunk(pos))
}
return chunks
}
2023-06-25 08:33:17 +00:00
// sends a packet to all peers in the given chunks, excluding the given peer
func (server *ShardServer) sendOthersPacket(plr *entity.Player, typeID uint32, pkt ...interface{}) error {
chunks := server.getViewableChunks(plr.Chunk)
for _, chunk := range chunks {
chunk.SendPacketExclude(plr, typeID, pkt...)
2023-06-25 08:33:17 +00:00
}
return nil
}
2023-06-28 03:05:08 +00:00
// sends a packet to all peers in the given chunks
func (server *ShardServer) sendAllPacket(plr *entity.Player, typeID uint32, pkt ...interface{}) error {
chunks := server.getViewableChunks(plr.Chunk)
for _, chunk := range chunks {
chunk.SendPacket(typeID, pkt...)
}
return nil
}
2023-06-25 08:33:17 +00:00
func (server *ShardServer) removeEntityFromChunks(chunks []*entity.Chunk, this entity.Entity) {
for _, chunk := range chunks {
for e := range chunk.Entities {
if e == this {
continue
}
// notify other players we're leaving
2023-06-25 08:33:17 +00:00
if e.GetKind() == entity.ENTITY_KIND_PLAYER {
otherPlr := e.(*entity.Player)
this.DisappearFromViewOf(otherPlr.Peer)
}
// notify us they're leaving
2023-06-25 08:33:17 +00:00
if this.GetKind() == entity.ENTITY_KIND_PLAYER {
thisPlr := this.(*entity.Player)
e.DisappearFromViewOf(thisPlr.Peer)
}
}
}
}
func (server *ShardServer) addEntityToChunks(chunks []*entity.Chunk, this entity.Entity) {
for _, chunk := range chunks {
for e := range chunk.Entities {
if e == this {
continue
}
// notify other players we're entering
2023-06-25 08:33:17 +00:00
if e.GetKind() == entity.ENTITY_KIND_PLAYER {
otherPlr := e.(*entity.Player)
this.EnterIntoViewOf(otherPlr.Peer)
}
// notify us they're entering
2023-06-25 08:33:17 +00:00
if this.GetKind() == entity.ENTITY_KIND_PLAYER {
thisPlr := this.(*entity.Player)
e.EnterIntoViewOf(thisPlr.Peer)
}
}
}
}
func (server *ShardServer) updateEntityChunk(e entity.Entity, from entity.ChunkPosition, to entity.ChunkPosition) {
// no change needed
if from == to {
return
}
2023-06-25 08:33:17 +00:00
oldViewables := server.getViewableChunks(from)
newViewables := server.getViewableChunks(to)
// compute differences
toExit := entity.ChunkSliceDifference(oldViewables, newViewables)
toEnter := entity.ChunkSliceDifference(newViewables, oldViewables)
// update chunks
server.removeEntityFromChunks(toExit, e)
server.addEntityToChunks(toEnter, e)
server.getChunk(from).RemoveEntity(e)
server.getChunk(to).AddEntity(e)
e.SetChunk(to)
}