KlemenPl/PixelDefense
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simlify and improve BT visualization

Browse Source
This commit is contained in:
Klemen Plestenjak
2024-01-11 08:34:53 +01:00
parent 8043d56af9
commit 36d8bac7c4
5 changed files with 39 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
engine/breeze/ai/behaviour_tree.c
View File

@@ -203,6 +203,11 @@ BzBTNode *bzBTNodeNext(const BzBTNode *node) {
return node->next;
}
const BzBTNode *bzBTNodeStateGetNode(const BzBTNodeState *state) {
BZ_ASSERT(state);
return state->node;
}
const BzBTNodeState *bzBTNodeStateNext(const BzBTNodeState *state) {
BZ_ASSERT(state);
return state->next;

2
engine/breeze/ai/behaviour_tree.h
View File

@@ -92,6 +92,8 @@ BzBTNodeType bzBTGetNodeType(const BzBTNode *node);
BzBTNode *bzBTNodeChild(const BzBTNode *node);
BzBTNode *bzBTNodeNext(const BzBTNode *node);
const BzBTNode *bzBTNodeStateGetNode(const BzBTNodeState *state);
const BzBTNodeState *bzBTNodeStateNext(const BzBTNodeState *state);
bool bzBTNodeMatchesState(const BzBTNode *node, const BzBTNodeState *state);

51
game/components.c
View File

@@ -227,8 +227,7 @@ static const BzBTNodeState *findNodeState(const BzBTNode *node, const BzBTNodeSt
}
return NULL;
}
void igVisualizeBTState(const BzBTNode *node, const BzBTNodeState *state,
bool isActive, bool sameLine, i32 depth) {
static void visualizeBTState(const BzBTNode *node, const BzBTNodeState *state, bool sameLine, bool isActive, i32 depth) {
const BzBTNode *child = bzBTNodeChild(node);
BzBTNodeType type = bzBTGetNodeType(node);
char extraInfo[128];
@@ -236,7 +235,7 @@ void igVisualizeBTState(const BzBTNode *node, const BzBTNodeState *state,
const BzBTNodeState *nodeState = findNodeState(node, state);
bool hasState = nodeState != NULL;
isActive |= hasState;
isActive |= nodeState != NULL;
switch (type) {
case BZ_BT_DECOR_REPEAT:
@@ -249,7 +248,7 @@ void igVisualizeBTState(const BzBTNode *node, const BzBTNodeState *state,
}
break;
case BZ_BT_DECOR_DELAY:
if (hasState) {
if (isActive) {
snprintf(extraInfo, sizeof(extraInfo), " (%.2f < %.2fms)",
bzBTDelayStateGetElapsed(nodeState),
bzBTDecorGetDelay(node));
@@ -271,35 +270,43 @@ void igVisualizeBTState(const BzBTNode *node, const BzBTNodeState *state,
if (isActive)
color = (ImVec4) {1.0f, 1.0f, 0.5f, 1.0f};
bool hasSingleChild = true;
if (child && bzBTNodeNext(child)) hasSingleChild = false;
bool hasSingleChild = child && bzBTNodeNext(child) == NULL;
const char *suffix = hasSingleChild ? " > " : ": ";
if (sameLine) {
igTextColored(color, "%s%s%s", bzBTNodeTypeToStr(type),
extraInfo, suffix);
} else {
igTextColored(color, "%*s%s %s",
depth * 2, "",
bzBTNodeTypeToStr(type), extraInfo, suffix);
depth++;
}
i32 offset = depth * 2;
if (sameLine) offset = 0;
const char *prefix = sameLine ? " > " : "";
const char *postfix = child != NULL && !hasSingleChild ? ":" : "";
igTextColored(color, "%s%*s%s%s%s", prefix, offset, "", bzBTNodeTypeToStr(type), extraInfo, postfix);
bool isComposite = type == BZ_BT_COMP_SELECTOR ||
type == BZ_BT_COMP_P_SELECTOR ||
type == BZ_BT_COMP_SEQUENCE ||
type == BZ_BT_COMP_P_SEQUENCE;
depth++;
while (child) {
if (hasSingleChild) igSameLine(0, 0);
bool childActive = isActive && hasSingleChild;
if (hasState && isComposite && !childActive)
childActive = bzBTCompStateGetRunningChild(state) == child;
igVisualizeBTState(child, state, childActive, hasSingleChild, depth);
bool childHasState = hasState && isComposite && bzBTCompStateGetRunningChild(nodeState) == child;
visualizeBTState(child, state, hasSingleChild, childHasState, depth);
child = bzBTNodeNext(child);
}
}
void igVisualizeBTState(const BzBTNode *node, const BzBTNodeState *state) {
igSeparatorText("Tree State");
visualizeBTState(node, state, false, true, 0);
igSeparatorText("State stack");
size_t stateSize = 0;
const BzBTNodeState *pState = state;
while (pState) {
const BzBTNode *pNode = bzBTNodeStateGetNode(pState);
stateSize += bzBTGetNodeStateSize();
BzBTNodeType type = bzBTGetNodeType(pNode);
igText("%s", bzBTNodeTypeToStr(type));
pState = bzBTNodeStateNext(pState);
}
igNewLine();
igText("Total stack state size: %ld bytes.", stateSize);
}
void igBzBTState(ecs_world_t *ecs,
ecs_entity_t entity, ecs_entity_t comp) {

3
game/components.h
View File

@@ -235,8 +235,7 @@ void igArms(ecs_world_t *ecs,
void igArm(ecs_world_t *ecs,
ecs_entity_t entity, ecs_entity_t comp);
void igVisualizeBTState(const BzBTNode *node, const BzBTNodeState *state,
bool isActive, bool sameLine, i32 depth);
void igVisualizeBTState(const BzBTNode *node, const BzBTNodeState *state);
void igBzBTState(ecs_world_t *ecs,
ecs_entity_t entity, ecs_entity_t comp);
void igAIBlackboard(ecs_world_t *ecs,

4
game/main.c
View File

@@ -624,7 +624,7 @@ static void renderMainMenu(Game *game, float dt) {
if (uiMainMenuButton("Play")) {
setScreen(game, SCREEN_GAME);
unloadMap(game);
loadMap(game, "assets/maps/pathing_test.tmj");
loadMap(game, "assets/maps/tree_test.tmj");
}
if (uiMainMenuButton("Settings")) {
setScreen(game, SCREEN_SETTINGS);
@@ -769,7 +769,7 @@ void igInspectWindow(ecs_entity_t entity, bool *open) {
igCollapsingHeader_TreeNodeFlags("BehaviourTree", 0)) {
const BzBTState *state = ecs_get(ECS, entity, BzBTState);
if (state->root)
igVisualizeBTState(state->root, state->_first, true, false, 0);
igVisualizeBTState(state->root, state->_first);
else
igTextColored((ImVec4) {1, 0, 0, 1}, "NONE");
}