Nasal-Interpreter/version1.1/ast.h

408 lines
9.5 KiB
C++

#ifndef __NASAL_ABSTRACT_SYNTAX_TREE_RUN_H__
#define __NASAL_ABSTRACT_SYNTAX_TREE_RUN_H__
class abstract_syntax_tree
{
protected:
int ast_node_type;
std::list<abstract_syntax_tree> statement_list;
std::list<abstract_syntax_tree> children;
// for definition and assignment
std::list<abstract_syntax_tree> var_list;
std::list<abstract_syntax_tree> var_changed_list;
std::list<abstract_syntax_tree> var_content_list;
// for number and string leaf
double var_number;
std::string var_string;
std::string id_name;
public:
abstract_syntax_tree()
{
ast_node_type=0;
statement_list.clear();
children.clear();
var_list.clear();
var_changed_list.clear();
var_content_list.clear();
var_number=0;
var_string="";
id_name="ukn";
}
abstract_syntax_tree(const abstract_syntax_tree& temp)
{
ast_node_type=temp.ast_node_type;
if(temp.statement_list.empty())
statement_list.clear();
else
statement_list=temp.statement_list;
if(temp.children.empty())
children.clear();
else
children=temp.children;
if(temp.var_list.empty())
var_list.clear();
else
var_list=temp.var_list;
if(temp.var_changed_list.empty())
var_changed_list.clear();
else
var_changed_list=temp.var_changed_list;
if(temp.var_content_list.empty())
var_content_list.clear();
else
var_content_list=temp.var_content_list;
var_number=temp.var_number;
var_string=temp.var_string;
id_name=temp.id_name;
}
abstract_syntax_tree& operator=(const abstract_syntax_tree temp)
{
ast_node_type=temp.ast_node_type;
if(temp.statement_list.empty())
statement_list.clear();
else
statement_list=temp.statement_list;
if(temp.children.empty())
children.clear();
else
children=temp.children;
if(temp.var_list.empty())
var_list.clear();
else
var_list=temp.var_list;
if(temp.var_changed_list.empty())
var_changed_list.clear();
else
var_changed_list=temp.var_changed_list;
if(temp.var_content_list.empty())
var_content_list.clear();
else
var_content_list=temp.var_content_list;
var_number=temp.var_number;
var_string=temp.var_string;
id_name=temp.id_name;
return *this;
}
void tree_set_clear()
{
statement_list.clear();
children.clear();
var_list.clear();
var_changed_list.clear();
var_content_list.clear();
return;
}
void print_ast_node(int tab_num)
{
std::string indent="";
for(int i=0;i<tab_num;++i)
indent+=" ";
std::cout<<indent;
if(ast_node_type==__number)
{
std::cout<<"{Number:"<<var_number<<"}"<<std::endl;
return;
}
else if(ast_node_type==__string)
{
std::cout<<"{String:"<<var_string<<"}"<<std::endl;
return;
}
else if(ast_node_type==__id)
{
std::cout<<"{Identifier:"<<id_name<<"}"<<std::endl;
return;
}
else if(ast_node_type==__definition)
{
std::cout<<"{ Type:";
print_token(ast_node_type);
std::cout<<std::endl<<indent<<" [Var]"<<std::endl;
for(std::list<abstract_syntax_tree>::iterator i=var_list.begin();i!=var_list.end();++i)
i->print_ast_node(tab_num+1);
std::cout<<indent<<" [Content]"<<std::endl;
for(std::list<abstract_syntax_tree>::iterator i=var_content_list.begin();i!=var_content_list.end();++i)
i->print_ast_node(tab_num+1);
return;
}
std::cout<<"{ Type:";
print_token(ast_node_type);
std::cout<<std::endl;
if(!children.empty())
{
std::cout<<indent<<" [Children]"<<std::endl;
for(std::list<abstract_syntax_tree>::iterator i=children.begin();i!=children.end();++i)
i->print_ast_node(tab_num+1);
}
if(!statement_list.empty())
{
std::cout<<indent<<" [Statement]"<<std::endl;
for(std::list<abstract_syntax_tree>::iterator i=statement_list.begin();i!=statement_list.end();++i)
i->print_ast_node(tab_num+1);
}
std::cout<<indent<<"}"<<std::endl;
return;
}
// for root node
void set_root()
{
ast_node_type=__root;
return;
}
// for definition
void set_definition_expr(std::list<abstract_syntax_tree> name_list,std::list<abstract_syntax_tree> var_content)
{
ast_node_type=__definition;
var_list=name_list;
var_content_list=var_content;
return;
}
// for assignment
void set_assignment_expr(std::list<abstract_syntax_tree> to_be_changed,std::list<abstract_syntax_tree> var_content)
{
ast_node_type=__assignment;
var_changed_list=to_be_changed;
var_content_list=var_content;
return;
}
// for choose block
void set_choose_block(std::list<abstract_syntax_tree> if_else_statements)
{
ast_node_type=__ifelse;
statement_list=if_else_statements;
return;
}
// for if
void set_if_expr(abstract_syntax_tree condition,abstract_syntax_tree statements)
{
ast_node_type=__if;
children.push_back(condition);
children.push_back(statements);
return;
}
// for elsif
void set_elsif_expr(abstract_syntax_tree condition,abstract_syntax_tree statements)
{
ast_node_type=__elsif;
children.push_back(condition);
children.push_back(statements);
return;
}
// for else
void set_else_expr(abstract_syntax_tree statements)
{
ast_node_type=__else;
children.push_back(statements);
return;
}
// for statement block
void set_block()
{
ast_node_type=__normal_statement_block;
return;
}
void add_statement(abstract_syntax_tree statement_head)
{
statement_list.push_back(statement_head);
return;
}
// for return expr
void set_node_to_ret(abstract_syntax_tree child)
{
ast_node_type=__return;
children.push_back(child);
return;
}
// for loop
void set_node_to_loop()
{
ast_node_type=__loop;
return;
}
void set_node_to_continue_break(const int type)
{
ast_node_type=type;
return;
}
// for function expr
void set_node_to_function(std::list<abstract_syntax_tree> parameters,abstract_syntax_tree block)
{
ast_node_type=__func;
children=parameters;
statement_list=block.statement_list;
return;
}
// for sub-tree node operator
void set_two_operator(const int operator_type,abstract_syntax_tree f_child,abstract_syntax_tree s_child)
{
ast_node_type=operator_type;
children.push_back(f_child);
children.push_back(s_child);
return;
}
void set_one_operator(const int operator_type,abstract_syntax_tree f_child)
{
ast_node_type=operator_type;
children.push_back(f_child);
return;
}
// for leaf node call
void set_node_to_call_function()
{
ast_node_type=__call_function;
return;
}
void set_node_to_list_search()
{
ast_node_type=__list_search;
return;
}
void set_node_to_hash_search()
{
ast_node_type=__hash_search;
return;
}
// for leaf node list
void set_node_to_list(std::list<abstract_syntax_tree> list_members)
{
ast_node_type=__list;
children=list_members;
return;
}
// for leaf node hash
void set_node_to_hash(std::list<abstract_syntax_tree> hash_members)
{
ast_node_type=__hash;
children=hash_members;
return;
}
// for hash member
void set_node_to_hashmember(std::string& name,abstract_syntax_tree mem_var)
{
ast_node_type=__hash_member;
id_name=name;
children.clear();
children.push_back(mem_var);
return;
}
// for leaf node hash
void set_node_to_hash()
{
ast_node_type=__hash;
return;
}
// for identifier node
void set_node_to_id(std::string& name)
{
ast_node_type=__id;
id_name=name;
return;
}
// for dynamic identifier
void set_node_to_dynid(std::string& name)
{
ast_node_type=__dynamic_id;
id_name="";
for(int i=0;i<(int)name.length()-3;++i)
id_name+=name[i];
return;
}
// for leaf node string
void set_node_to_string(std::string& str)
{
ast_node_type=__string;
var_string=str;
return;
}
std::string get_string()
{
return var_string;
}
// for leaf node number
void set_node_to_number(std::string& str)
{
ast_node_type=__number;
if(str=="nil")
{
var_number=0;
return;
}
if((int)str.length()>2 && (str[1]=='x' || str[1]=='o'))
{
if(str[1]=='x')
{
int num=0;
int pw=1;
for(int i=(int)str.length()-1;i>1;--i)
{
if('0'<=str[i] && str[i]<='9')
num+=(str[i]-'0')*pw;
else if('a'<=str[i] && str[i]<='f')
num+=(10+str[i]-'a')*pw;
else if('A'<=str[i] && str[i]<='F')
num+=(10+str[i]-'A')*pw;
pw<<=4;
}
var_number=(double)num;
}
else
{
int num=0;
int pw=1;
for(int i=(int)str.length()-1;i>1;--i)
{
num+=(str[i]-'0')*pw;
pw<<=3;
}
var_number=(double)num;
}
return;
}
int dot_place=-1;
for(int i=0;i<(int)str.length();++i)
if(str[i]=='.')
{
dot_place=i;
break;
}
if(dot_place==-1)
{
var_number=0;
double pw=1;
for(int i=(int)str.length()-1;i>=0;--i)
{
var_number+=(str[i]-'0')*pw;
pw*=10;
}
}
else
{
var_number=0;
double pw=0.1;
for(int i=dot_place+1;i<(int)str.length();++i)
{
var_number+=(str[i]-'0')*pw;
pw/=10;
}
pw=1;
for(int i=dot_place-1;i>=0;--i)
{
var_number+=(str[i]-'0')*pw;
pw*=10;
}
}
return;
}
double get_number()
{
return var_number;
}
};
#endif