%{
    #include <stdio.h>
    #include <string.h>
    #include <stdlib.h>

    #define DEBUG 0

	typedef struct AST_Tree{
		char type[60];
		struct AST_Tree* child;
		struct AST_Tree* brother;
		char *value;
	}AST_Tree_node;


	AST_Tree_node *head = NULL;

	AST_Tree_node* insertNode(char* node);
	AST_Tree_node* insert_value_node(char* node, char* value);
	void insert_child(AST_Tree_node* father, AST_Tree_node* child);
	void insert_brother(AST_Tree_node* brother, AST_Tree_node* node)
    int flag=0;
    int aux2 = 0;
	int imprimir=0;
	int parse=0;
    int yylex(void);
    void yyerror(const char *s);

    extern int n_linha;
    extern int n_coluna;
    extern char * yytext;
    extern int yyleng;
	AST_Tree_node *nodeAux;
	AST_Tree_node *nodeAux2;



%}

%union{
    char *str;
    struct AST_Tree *node;
}

%token <str> ID STRLIT BOOLLIT DECLIT REALLIT

%token PUBLIC BOOL INT DOUBLE SEMI COMMA PARSEINT OCURV OSQUARE CSQUARE CCURV ASSIGN AND OR EQ GEQ GT LEQ LT NEQ PLUS MINUS STAR DIV MOD NOT DOTLENGTH  STATIC STRING VOID CLASS OBRACE CBRACE IF ELSE WHILE DO PRINT RETURN RESERVED



%left COMMA
%right ASSIGN
%left OR
%left AND
%left EQ NEQ
%left LT GT LEQ GEQ
%left MINUS PLUS
%left STAR DIV MOD
%right NOT
%left OCURV OBRACE OSQUARE CCURV CBRACE CSQUARE

%nonassoc ELSE

%%



Program: CLASS ID OBRACE FieldDecl_MethodDecl_SEMI CBRACE    {if(flag == 0){$$ = head = insertNode("Program");nodeAux = insert_value_node("ID", $2); insert_child($$, nodeAux); insert_child($$, $4);}}
    | CLASS ID OBRACE CBRACE {if(flag==0){$$ = head = insertNode("Program");nodeAux = insert_value_node("ID", $2);insert_child($$, nodeAux);}}
    ;

FieldDecl_MethodDecl_SEMI: FieldDecl_MethodDecl_SEMI  FieldDecl  {if(flag ==0){insert_child(head, $2);}}
    |   FieldDecl_MethodDecl_SEMI  MethodDecl    {if(flag == 0){insert_child(head, $2);}}
    |   FieldDecl_MethodDecl_SEMI SEMI   {;}
    |   FieldDecl           {if(flag ==0 ){insert_child(head, $1);}}
    |   MethodDecl          {if(flag == 0){insert_child(head, $1);}}
    |   SEMI                {;}
    ;


FieldDecl: PUBLIC STATIC Type ID CommaID SEMI   {if(flag ==0){$$ = insertNode("FieldDecl"); insert_child($$, $3); nodeAux = insert_value_node("ID", $4);insert_child($$, nodeAux); insert_child($$, $5); }}
    | PUBLIC STATIC Type ID SEMI   {if(flag == 0){$$ = insertNode("FieldDecl"); insert_child($$, $3); nodeAux = insert_value_node("ID", $4); insert_child($$, nodeAux);}}
    | error SEMI    {if(flag == 0){$$ = NULL;}}
    ;

CommaID: CommaID COMMA ID        {if(flag == 0){$$ = $1 ; nodeAux = insert_value_node("ID", $3); insert_brother($$, nodeAux);}}
    |   COMMA ID        {if(flag == 0){insert_value_node("ID", $2);}}
    ;

MethodDecl: PUBLIC STATIC MethodHeader MethodBody   {if(DEBUG){printf("MethodDecl\n");}}
    ;

MethodHeader: Type ID OCURV FormalParams CCURV  {if(DEBUG){printf("MethodHeader_Type_ID_OCURV_FPO_CCURV\n");}}
    |   Type ID OCURV CCURV
    |   VOID ID OCURV FormalParams CCURV  {if(DEBUG){printf("MethodHeader_VOID_ID_OCURV_FPO_CCURV\n");}}
    |   VOID ID OCURV CCURV
    ;

FormalParams: Type ID CommaTypeID   {if(DEBUG){printf("FormalParams_Type_ID\n");}}
    | Type ID           {if(DEBUG){printf("FormalParams_Type_ID\n");}}
    |   STRING OSQUARE CSQUARE ID   {if(DEBUG){printf("FormalParams_STRING_OSQUARE_CSQUARE_ID\n");}}
    ;


CommaTypeID: COMMA Type ID                          {if(DEBUG){printf("CommaTypeID\n");}}
    |   CommaTypeID COMMA Type ID                  {if(DEBUG){printf("CommaTypeID\n");}}
    ;


MethodBody: OBRACE VarDecl_Statement CBRACE     {if(DEBUG){printf("MethodBody\n");}}
    |    OBRACE CBRACE          {if(DEBUG){printf("MethodBody\n");}}
    ;


VarDecl_Statement: VarDecl       {if(DEBUG){printf("Statement_VarDecl_Statement\n");}}
    |   Statement             {if(DEBUG){printf("VarDecl_VarDecl_Statement\n");}}
    |   VarDecl_Statement VarDecl    {if(DEBUG){printf("VarDecl_VarDecl_Statement\n");}}
    |   VarDecl_Statement Statement     {if(DEBUG){printf("Statement_VarDecl_Statement\n");}}
    ;



VarDecl: Type ID CommaID SEMI           {if(DEBUG){printf("VarDecl\n");}}
    | Type ID SEMI           {if(DEBUG){printf("VarDecl\n");}}
    ;

Type: BOOL      {if(DEBUG){printf("BOOL\n");}}
    |   INT        {if(DEBUG){printf("INT\n");}}
    |   DOUBLE     {if(DEBUG){printf("DOUBLE\n");}}
    ;

Statement: OBRACE Statement_aux CBRACE  {if(DEBUG){printf("OBRACE_St_CBRACE\n");}}
    |   OBRACE CBRACE  {if(DEBUG){printf("OBRACE_CBRACE\n");}}
    |   IF OCURV Expr CCURV Statement  {if(DEBUG){printf("IF_OCURV_Expr_CCURV_Statement\n");}}
    |   IF OCURV Expr CCURV Statement ELSE Statement  {if(DEBUG){printf("IF_OCURV_Expr_CCURV_Statement_Else_Statement\n");}}
    |   WHILE OCURV Expr CCURV Statement    {if(DEBUG){printf("WHILE_OCURV_Expr_CCURV_Statement\n");}}
    |   DO Statement WHILE OCURV Expr CCURV SEMI    {if(DEBUG){printf("DO_Statement_WHILE_OCURV_Expr_CCURV_SEMI\n");}}
    |   PRINT OCURV Expr CCURV SEMI     {if(DEBUG){printf("PRINT_OCURV_Expr_CCURV_SEMI\n");}}
    |   PRINT OCURV STRLIT CCURV SEMI    {if(DEBUG){printf("PRINT_OCURV_STRLIT_CCURV_SEMI\n");}}
    |   SEMI    {if(DEBUG){printf("SEMI\n");}}
    |   Assignment SEMI {if(DEBUG){printf("Assignment SEMI\n");}}
    |   MethodInvocation SEMI   {if(DEBUG){printf("MethodInvocation SEMI\n");}}
    |   ParseArgs SEMI  {if(DEBUG){printf("ParseArgs SEMI\n");}}
    |   RETURN SEMI  {if(DEBUG){printf("RETURN_ExpressionOptional_SEMI\n");}}
    |   RETURN Expr SEMI  {if(DEBUG){printf("RETURN_ExpressionOptional_SEMI\n");}}
    |   error SEMI    {if(DEBUG){printf("Error SEMI on Statement\n");}}
    ;



Statement_aux: Statement        {if(DEBUG){printf("Statement\n");}}
    | Statement_aux Statement   {if(DEBUG){printf("Statementaux Statement_aux\n");}}
    ;

Expr: ExprNew
    |   Assignment
    ;

ExprNew:      MethodInvocation       {if(DEBUG){printf("Expression1_MethodInvocation\n");}}
    |   ParseArgs      {if(DEBUG){printf("Expression1_ParseArgs\n");}}
    |   ExprNew AND ExprNew   {if(DEBUG){printf("Expr_AND_Expr\n");}}
    |   ExprNew OR ExprNew     {if(DEBUG){printf("Expr_OR_Expr\n");}}
    |   ExprNew EQ ExprNew     {if(DEBUG){printf("Expr_EQ_Expr\n");}}
    |   ExprNew GEQ ExprNew    {if(DEBUG){printf("Expr_GEQ_Expr\n");}}
    |   ExprNew GT ExprNew     {if(DEBUG){printf("Expr_GT_Expr\n");}}
    |   ExprNew LEQ ExprNew    {if(DEBUG){printf("Expr_LEQ_Expr\n");}}
    |   ExprNew LT ExprNew     {if(DEBUG){printf("Expr_LT_Expr\n");}}
    |   ExprNew NEQ ExprNew    {if(DEBUG){printf("Expr_NEQ_Expr\n");}}
    |   ExprNew PLUS ExprNew   {if(DEBUG){printf("Expr_PLUS_Expr\n");}}
    |   ExprNew MINUS ExprNew      {if(DEBUG){printf("Expr_MINUS_Expr\n");}}
    |   ExprNew STAR ExprNew   {if(DEBUG){printf("Expr_STAR_Expr\n");}}
    |   ExprNew DIV ExprNew    {if(DEBUG){printf("Expr_DIV_Expr\n");}}
    |   ExprNew MOD ExprNew    {if(DEBUG){printf("Expr_MOD_Expr\n");}}
    |   PLUS ExprNew  %prec NOT  {if(DEBUG){printf("PLUS_Expr\n");}}
    |   MINUS ExprNew  %prec NOT {if(DEBUG){printf("MINUS_Expr\n");}}
    |   NOT ExprNew     {if(DEBUG){printf("NOT_Expr\n");}}
    |   ID DOTLENGTH    {if(DEBUG){printf("ID_Dotlength\n");}}
    |   ID      {if(DEBUG){printf("ID ExprNew\n");}}
    |   OCURV Expr CCURV    {if(DEBUG){printf("OCURV_Expr_CCURV\n");}}
    |   BOOLLIT     {if(DEBUG){printf("BOOLLIT\n");}}
    |   DECLIT  {if(DEBUG){printf("DECLIT\n");}}
    |   REALLIT     {if(DEBUG){printf("REALLIT\n");}}
    |   OCURV error CCURV   {if(DEBUG){printf("Error OCURV CCURV on Expr\n");}}
    ;

Assignment: ID ASSIGN Expr      {if(DEBUG){printf("Assigment\n");}}
    ;


MethodInvocation: ID OCURV CCURV     {if(DEBUG){printf("MethodInvocation\n");}}
    |   ID OCURV Expr CCURV     {if(DEBUG){printf("MethodInvocation\n");}}
    |   ID OCURV Expr CommaExpr CCURV     {if(DEBUG){printf("MethodInvocation\n");}}
    |   ID OCURV error CCURV    {if(DEBUG){printf("Error ID OCURV CCURV on MethodInvocation\n");}}
    ;

CommaExpr: COMMA Expr       {if(DEBUG){printf("CommaExpr\n");}}
    |   CommaExpr COMMA Expr        {if(DEBUG){printf("CommaExpr\n");}}
    ;

ParseArgs: PARSEINT OCURV ID OSQUARE Expr CSQUARE CCURV {if(DEBUG){printf("ParseArgs\n");}}
    |   PARSEINT OCURV error CCURV   {if(DEBUG){printf("Error PARSEINT OCURV CCURV on ParseArgs\n");}}
    ;



%%

void yyerror(const char* s){
	if(flag ==0){
		flag = 1;
	}
    printf("Line %d, col %d: %s: %s\n",n_linha,(int)(n_coluna - strlen(yytext)),s,yytext);
}



void insert_brother(AST_Tree_node* brother, AST_Tree_node* node)
{


	AST_Tree_node* aux = brother;

	if(aux!=NULL && node!=NULL)
	{
		while(aux->brother != NULL)
		{
			aux = aux->brother;
		}
		aux->brother = node;
	}
}


void insert_child(AST_Tree_node* father, AST_Tree_node* child)
{

	AST_Tree_node* temp = father->child;

	if(temp==NULL)
	{
		father->child = child;
	}
	else
	{

		insert_brother(father->child, child);
	}

}


AST_Tree_node* insert_value_node(char* node, char* value){

	AST_Tree_node* new_node = insertNode(node);
	new_node->value = value;
	return new_node;

}

AST_Tree_node* insertNode(char* node){
	AST_Tree_node* new_node;
	new_node = (AST_Tree_node *)(malloc(sizeof(AST_Tree_node)));
	if(new_node != NULL){
		new_node->child = NULL;
		new_node->brother = NULL;
		new_node->value = NULL;
		strcpy(new_node->type, node);
	}
	return new_node;
}

int main(int argc, char *argv[])
{
	if(argc>1){
		if(strcmp(argv[1],"-l")==0){
			imprimir=1;
			yylex();
		}
		else if(strcmp(argv[1],"-1")==0){
			yylex();
		}
	}
	else{
		parse=1;
		yyparse();
	}

	return 0;
}