package shard import ( "github.com/CPunch/gopenfusion/core/entity" ) 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) { // TODO: chunk cleanup pos := e.GetChunk() server.removeEntityFromChunks(server.getViewableChunks(pos), e) server.getChunk(pos).RemoveEntity(e) } 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 } func (server *ShardServer) getViewableChunks(pos entity.ChunkPosition) []*entity.Chunk { chunks := make([]*entity.Chunk, 0, 9) for _, pos := range server.getChunk(pos).GetAdjacentPositions() { chunks = append(chunks, server.getChunk(pos)) } return chunks } // 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...) } return nil } // 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 } 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 if e.GetKind() == entity.ENTITY_KIND_PLAYER { otherPlr := e.(*entity.Player) this.DisappearFromViewOf(otherPlr.Peer) } // notify us they're leaving 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 if e.GetKind() == entity.ENTITY_KIND_PLAYER { otherPlr := e.(*entity.Player) this.EnterIntoViewOf(otherPlr.Peer) } // notify us they're entering 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 } 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) }