Made functions to get child nodes of AST by tag

This commit is contained in:
petermlm
2016-04-17 00:17:45 +01:00
parent 4da2ae5feb
commit 2b643c729b
2 changed files with 37 additions and 0 deletions

32
mpc.c
View File

@@ -2776,6 +2776,38 @@ void mpc_ast_print_to(mpc_ast_t *a, FILE *fp) {
mpc_ast_print_depth(a, 0, fp);
}
int mpc_ast_get_index(mpc_ast_t *ast, char *tag) {
return mpc_ast_get_index_lb(ast, tag, 0);
}
int mpc_ast_get_index_lb(mpc_ast_t *ast, char *tag, int lb) {
int i;
for(i=lb; i<ast->children_num; i++) {
if(strcmp(ast->children[i]->tag, tag) == 0) {
return i;
}
}
return -1;
}
mpc_ast_t *mpc_ast_get_child(mpc_ast_t *ast, char *tag) {
return mpc_ast_get_child_lb(ast, tag, 0);
}
mpc_ast_t *mpc_ast_get_child_lb(mpc_ast_t *ast, char *tag, int lb) {
int i;
for(i=lb; i<ast->children_num; i++) {
if(strcmp(ast->children[i]->tag, tag) == 0) {
return ast->children[i];
}
}
return NULL;
}
mpc_val_t *mpcf_fold_ast(int n, mpc_val_t **xs) {
int i, j;

5
mpc.h
View File

@@ -276,6 +276,11 @@ void mpc_ast_delete(mpc_ast_t *a);
void mpc_ast_print(mpc_ast_t *a);
void mpc_ast_print_to(mpc_ast_t *a, FILE *fp);
int mpc_ast_get_index(mpc_ast_t *ast, char *tag);
int mpc_ast_get_index_lb(mpc_ast_t *ast, char *tag, int lb);
mpc_ast_t *mpc_ast_get_child(mpc_ast_t *ast, char *tag);
mpc_ast_t *mpc_ast_get_child_lb(mpc_ast_t *ast, char *tag, int lb);
/*
** Warning: This function currently doesn't test for equality of the `state` member!
*/