mirror of
https://github.com/FlourishingWorld/hk4e.git
synced 2026-03-01 00:35:36 +08:00
init commit
This commit is contained in:
flswld
@@ -0,0 +1,211 @@
|
||||
package aoi
|
||||
|
||||
import (
|
||||
"flswld.com/logger"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// aoi管理模块
|
||||
type AoiManager struct {
|
||||
// 区域边界坐标
|
||||
minX int16
|
||||
maxX int16
|
||||
minY int16
|
||||
maxY int16
|
||||
minZ int16
|
||||
maxZ int16
|
||||
numX int16 // x方向格子的数量
|
||||
numY int16 // y方向的格子数量
|
||||
numZ int16 // z方向的格子数量
|
||||
gridMap map[uint32]*Grid // 当前区域中都有哪些格子 key:gid value:格子对象
|
||||
}
|
||||
|
||||
// 初始化aoi区域
|
||||
func NewAoiManager(minX, maxX, numX, minY, maxY, numY, minZ, maxZ, numZ int16) (r *AoiManager) {
|
||||
r = new(AoiManager)
|
||||
r.minX = minX
|
||||
r.maxX = maxX
|
||||
r.minY = minY
|
||||
r.maxY = maxY
|
||||
r.numX = numX
|
||||
r.numY = numY
|
||||
r.minZ = minZ
|
||||
r.maxZ = maxZ
|
||||
r.numZ = numZ
|
||||
r.gridMap = make(map[uint32]*Grid)
|
||||
logger.LOG.Info("start init aoi area grid, num: %v", uint32(numX)*uint32(numY)*uint32(numZ))
|
||||
// 初始化aoi区域中所有的格子
|
||||
for x := int16(0); x < numX; x++ {
|
||||
for y := int16(0); y < numY; y++ {
|
||||
for z := int16(0); z < numZ; z++ {
|
||||
// 利用格子坐标得到格子id gid从0开始按xzy的顺序增长
|
||||
gid := uint32(y)*(uint32(numX)*uint32(numZ)) + uint32(z)*uint32(numX) + uint32(x)
|
||||
// 初始化一个格子放在aoi中的map里 key是当前格子的id
|
||||
grid := NewGrid(
|
||||
gid,
|
||||
r.minX+x*r.GridXLen(),
|
||||
r.minX+(x+1)*r.GridXLen(),
|
||||
r.minY+y*r.GridYLen(),
|
||||
r.minY+(y+1)*r.GridYLen(),
|
||||
r.minZ+z*r.GridZLen(),
|
||||
r.minZ+(z+1)*r.GridZLen(),
|
||||
)
|
||||
r.gridMap[gid] = grid
|
||||
}
|
||||
}
|
||||
}
|
||||
logger.LOG.Info("init aoi area grid finish")
|
||||
return r
|
||||
}
|
||||
|
||||
// 每个格子在x轴方向的长度
|
||||
func (a *AoiManager) GridXLen() int16 {
|
||||
return (a.maxX - a.minX) / a.numX
|
||||
}
|
||||
|
||||
// 每个格子在y轴方向的长度
|
||||
func (a *AoiManager) GridYLen() int16 {
|
||||
return (a.maxY - a.minY) / a.numY
|
||||
}
|
||||
|
||||
// 每个格子在z轴方向的长度
|
||||
func (a *AoiManager) GridZLen() int16 {
|
||||
return (a.maxZ - a.minZ) / a.numZ
|
||||
}
|
||||
|
||||
// 通过坐标获取对应的格子id
|
||||
func (a *AoiManager) GetGidByPos(x, y, z float32) uint32 {
|
||||
gx := (int16(x) - a.minX) / a.GridXLen()
|
||||
gy := (int16(y) - a.minY) / a.GridYLen()
|
||||
gz := (int16(z) - a.minZ) / a.GridZLen()
|
||||
return uint32(gy)*(uint32(a.numX)*uint32(a.numZ)) + uint32(gz)*uint32(a.numX) + uint32(gx)
|
||||
}
|
||||
|
||||
// 判断坐标是否存在于aoi区域内
|
||||
func (a *AoiManager) IsValidAoiPos(x, y, z float32) bool {
|
||||
if (int16(x) > a.minX && int16(x) < a.maxX) &&
|
||||
(int16(y) > a.minY && int16(y) < a.maxY) &&
|
||||
(int16(z) > a.minZ && int16(z) < a.maxZ) {
|
||||
return true
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
// 打印信息方法
|
||||
func (a *AoiManager) DebugString() string {
|
||||
s := fmt.Sprintf("AoiMgr: minX: %d, maxX: %d, numX: %d, minY: %d, maxY: %d, numY: %d, minZ: %d, maxZ: %d, numZ: %d\n",
|
||||
a.minX, a.maxX, a.numX, a.minY, a.maxY, a.numY, a.minZ, a.maxZ, a.numZ)
|
||||
s += "gridList in AoiMgr:\n"
|
||||
for _, grid := range a.gridMap {
|
||||
s += fmt.Sprintln(grid.DebugString())
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
// 根据格子的gid得到当前周边的格子信息
|
||||
func (a *AoiManager) GetSurrGridListByGid(gid uint32) (gridList []*Grid) {
|
||||
gridList = make([]*Grid, 0)
|
||||
// 判断grid是否存在
|
||||
grid, exist := a.gridMap[gid]
|
||||
if !exist {
|
||||
return gridList
|
||||
}
|
||||
// 添加自己
|
||||
gridList = append(gridList, grid)
|
||||
// 根据gid得到当前格子所在的x轴编号
|
||||
idx := int16(gid % (uint32(a.numX) * uint32(a.numZ)) % uint32(a.numX))
|
||||
// 判断当前格子左边是否还有格子
|
||||
if idx > 0 {
|
||||
gridList = append(gridList, a.gridMap[gid-1])
|
||||
}
|
||||
// 判断当前格子右边是否还有格子
|
||||
if idx < a.numX-1 {
|
||||
gridList = append(gridList, a.gridMap[gid+1])
|
||||
}
|
||||
// 将x轴当前的格子都取出进行遍历 再分别得到每个格子的平面上下是否有格子
|
||||
// 得到当前x轴的格子id集合
|
||||
gidListX := make([]uint32, 0)
|
||||
for _, v := range gridList {
|
||||
gidListX = append(gidListX, v.gid)
|
||||
}
|
||||
// 遍历x轴格子
|
||||
for _, v := range gidListX {
|
||||
// 计算该格子的idz
|
||||
idz := int16(v % (uint32(a.numX) * uint32(a.numZ)) / uint32(a.numX))
|
||||
// 判断当前格子平面上方是否还有格子
|
||||
if idz > 0 {
|
||||
gridList = append(gridList, a.gridMap[v-uint32(a.numX)])
|
||||
}
|
||||
// 判断当前格子平面下方是否还有格子
|
||||
if idz < a.numZ-1 {
|
||||
gridList = append(gridList, a.gridMap[v+uint32(a.numX)])
|
||||
}
|
||||
}
|
||||
// 将xoz平面当前的格子都取出进行遍历 再分别得到每个格子的空间上下是否有格子
|
||||
// 得到当前xoz平面的格子id集合
|
||||
gidListXOZ := make([]uint32, 0)
|
||||
for _, v := range gridList {
|
||||
gidListXOZ = append(gidListXOZ, v.gid)
|
||||
}
|
||||
// 遍历xoz平面格子
|
||||
for _, v := range gidListXOZ {
|
||||
// 计算该格子的idy
|
||||
idy := int16(v / (uint32(a.numX) * uint32(a.numZ)))
|
||||
// 判断当前格子空间上方是否还有格子
|
||||
if idy > 0 {
|
||||
gridList = append(gridList, a.gridMap[v-uint32(a.numX)*uint32(a.numZ)])
|
||||
}
|
||||
// 判断当前格子空间下方是否还有格子
|
||||
if idy < a.numY-1 {
|
||||
gridList = append(gridList, a.gridMap[v+uint32(a.numX)*uint32(a.numZ)])
|
||||
}
|
||||
}
|
||||
return gridList
|
||||
}
|
||||
|
||||
// 通过坐标得到周边格子内的全部entityId
|
||||
func (a *AoiManager) GetEntityIdListByPos(x, y, z float32) (entityIdList []uint32) {
|
||||
// 根据坐标得到当前坐标属于哪个格子id
|
||||
gid := a.GetGidByPos(x, y, z)
|
||||
// 根据格子id得到周边格子的信息
|
||||
gridList := a.GetSurrGridListByGid(gid)
|
||||
entityIdList = make([]uint32, 0)
|
||||
for _, v := range gridList {
|
||||
tmp := v.GetEntityIdList()
|
||||
entityIdList = append(entityIdList, tmp...)
|
||||
//logger.LOG.Debug("Grid: gid: %d, tmp len: %v", v.gid, len(tmp))
|
||||
}
|
||||
return entityIdList
|
||||
}
|
||||
|
||||
// 通过gid获取当前格子的全部entityId
|
||||
func (a *AoiManager) GetEntityIdListByGid(gid uint32) (entityIdList []uint32) {
|
||||
grid := a.gridMap[gid]
|
||||
entityIdList = grid.GetEntityIdList()
|
||||
return entityIdList
|
||||
}
|
||||
|
||||
// 添加一个entityId到一个格子中
|
||||
func (a *AoiManager) AddEntityIdToGrid(entityId uint32, gid uint32) {
|
||||
grid := a.gridMap[gid]
|
||||
grid.AddEntityId(entityId)
|
||||
}
|
||||
|
||||
// 移除一个格子中的entityId
|
||||
func (a *AoiManager) RemoveEntityIdFromGrid(entityId uint32, gid uint32) {
|
||||
grid := a.gridMap[gid]
|
||||
grid.RemoveEntityId(entityId)
|
||||
}
|
||||
|
||||
// 通过坐标添加一个entityId到一个格子中
|
||||
func (a *AoiManager) AddEntityIdToGridByPos(entityId uint32, x, y, z float32) {
|
||||
gid := a.GetGidByPos(x, y, z)
|
||||
a.AddEntityIdToGrid(entityId, gid)
|
||||
}
|
||||
|
||||
// 通过坐标把一个entityId从对应的格子中删除
|
||||
func (a *AoiManager) RemoveEntityIdFromGridByPos(entityId uint32, x, y, z float32) {
|
||||
gid := a.GetGidByPos(x, y, z)
|
||||
a.RemoveEntityIdFromGrid(entityId, gid)
|
||||
}
|
||||
@@ -0,0 +1,30 @@
|
||||
package aoi
|
||||
|
||||
import (
|
||||
"flswld.com/common/config"
|
||||
"flswld.com/logger"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestAoiManagerGetSurrGridListByGid(t *testing.T) {
|
||||
filePath := "./application.toml"
|
||||
config.InitConfig(filePath)
|
||||
logger.InitLogger()
|
||||
aoiManager := NewAoiManager(
|
||||
-150, 150, 3,
|
||||
-150, 150, 3,
|
||||
-150, 150, 3,
|
||||
)
|
||||
logger.LOG.Debug("aoiManager: %s", aoiManager.DebugString())
|
||||
for k := range aoiManager.gridMap {
|
||||
// 得到当前格子周边的九宫格
|
||||
gridList := aoiManager.GetSurrGridListByGid(k)
|
||||
// 得到九宫格所有的id
|
||||
logger.LOG.Debug("gid: %d gridList len: %d", k, len(gridList))
|
||||
gidList := make([]uint32, 0, len(gridList))
|
||||
for _, grid := range gridList {
|
||||
gidList = append(gidList, grid.gid)
|
||||
}
|
||||
logger.LOG.Debug("Grid: gid: %d, surr grid gid list: %v", k, gidList)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,63 @@
|
||||
package aoi
|
||||
|
||||
import (
|
||||
"flswld.com/logger"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// 地图格子
|
||||
type Grid struct {
|
||||
gid uint32 // 格子id
|
||||
// 格子边界坐标
|
||||
minX int16
|
||||
maxX int16
|
||||
minY int16
|
||||
maxY int16
|
||||
minZ int16
|
||||
maxZ int16
|
||||
entityIdMap map[uint32]bool // k:entityId v:是否存在
|
||||
}
|
||||
|
||||
// 初始化格子
|
||||
func NewGrid(gid uint32, minX, maxX, minY, maxY, minZ, maxZ int16) (r *Grid) {
|
||||
r = new(Grid)
|
||||
r.gid = gid
|
||||
r.minX = minX
|
||||
r.maxX = maxX
|
||||
r.minY = minY
|
||||
r.maxY = maxY
|
||||
r.minZ = minZ
|
||||
r.maxZ = maxZ
|
||||
r.entityIdMap = make(map[uint32]bool)
|
||||
return r
|
||||
}
|
||||
|
||||
// 向格子中添加一个实体id
|
||||
func (g *Grid) AddEntityId(entityId uint32) {
|
||||
g.entityIdMap[entityId] = true
|
||||
}
|
||||
|
||||
// 从格子中删除一个实体id
|
||||
func (g *Grid) RemoveEntityId(entityId uint32) {
|
||||
_, exist := g.entityIdMap[entityId]
|
||||
if exist {
|
||||
delete(g.entityIdMap, entityId)
|
||||
} else {
|
||||
logger.LOG.Error("remove entity id but it not exist, entityId: %v", entityId)
|
||||
}
|
||||
}
|
||||
|
||||
// 获取格子中所有实体id
|
||||
func (g *Grid) GetEntityIdList() (entityIdList []uint32) {
|
||||
entityIdList = make([]uint32, 0)
|
||||
for k := range g.entityIdMap {
|
||||
entityIdList = append(entityIdList, k)
|
||||
}
|
||||
return entityIdList
|
||||
}
|
||||
|
||||
// 打印信息方法
|
||||
func (g *Grid) DebugString() string {
|
||||
return fmt.Sprintf("Grid: gid: %d, minX: %d, maxX: %d, minY: %d, maxY: %d, minZ: %d, maxZ: %d, entityIdMap: %v",
|
||||
g.gid, g.minX, g.maxX, g.minY, g.maxY, g.minZ, g.maxZ, g.entityIdMap)
|
||||
}
|
||||
Reference in New Issue
Block a user