ast.h

#include <vector>
#include <iostream>
#include "symtable.h"

class NStatement;
class NExpression;
class NVarrayDeclaration;

typedef std::vector<NStatement*> StatementList;
typedef std::vector<NExpression*> ExpressionList;
typedef std::vector<NVarrayDeclaration*> VariableList;


class Node{
	public:
		virtual TType* typeChk(Symtable& t,TType* expected = NULL){
#ifdef DEBUG
			cerr<<"TypeCHK:default"<<endl;
#endif
			return t.lookupType("void");
		}
		virtual void printTree(std::ostream& os,int depth=0) {};
		virtual ~Node() {}
};

class NExpression : public Node {

};

class NLRExpression : public NExpression{
};

class NStatement : public Node {

};


class NExpressionStatement : public NStatement {
	public:
		NExpression* expr;
		NExpressionStatement(NExpression* expr):expr(expr){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
#ifdef DEBUG
			cerr<<"TypeCHK: ExprStament"<<endl;
#endif
			return expr->typeChk(t);
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NExpressionStatement:"<<endl;
			expr->printTree(os,depth+1);
		}
};

class NInteger : public NExpression {
	public :
		int value;
		NInteger(int value) : value(value) {}
		TType* typeChk(Symtable& t,TType* expected = NULL){
#ifdef DEBUG
			cerr<<"TypeCHK:integer "<<value<<endl;
#endif
			return (TType*)t.lookupType("integer");
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NInteger: value="<<value<<endl;
		}
};

class NDouble : public NExpression {
	public :
		double value;
		NDouble(double value) : value(value) {}
		TType* typeChk(Symtable& t,TType* expected = NULL){
#ifdef DEBUG
			cerr<<"TypeCHK:double "<<value<<endl;
#endif
			return (TType*)t.lookupType("float");
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NDouble: value="<<value<<endl;
		}
};

class NString : public NExpression {
	public :
		std::string value;
		NString(std::string &value) : value(value) {}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NString: value="<<value<<endl;
        }
		TType* typeChk(Symtable& t, TType* expected=NULL){
#ifdef DEBUG
			cerr<<"TypeCHK:double "<<value<<endl;
#endif
			return new TString(value.length());

        }
};

class NChar : public NExpression {
	public :
		char value;
		NChar(char value) : value(value){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
#ifdef DEBUG
			cerr<<"TypeCHK:Char "<<value<<endl;
#endif
			return (TType*)t.lookupType("char");
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NChar: value="<<value<<endl;
		}
};

class NBool : public NExpression {
	public:
		bool value;
		NBool(bool value) :value(value){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
#ifdef DEBUG
			cerr<<"TypeCHK:Bool "<<value<<endl;
#endif
			return (TType*)t.lookupType("boolean");
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NBool: value="<<value<<endl;
		}
};

class NArray : public NExpression {
	public :
		ExpressionList values;
		NArray(){}
		NArray(ExpressionList values):values(values){}
        void add(NArray* arr){
            values.push_back(arr);
        }
        NArray(std::string& s){
#ifdef DEBUG
            cout << "string param "<< s.size()<<endl;
            cout << "string param "<< s<<endl;
#endif
            for(int i=0; i<s.size();i++){
                values.push_back(new NChar(s[i]));
            }
#ifdef DEBUG
            cout << "constrc string "<< values.size()<<endl;
#endif
        }
		TType* typeChk(Symtable& t,TType* expected = NULL){
            bool err=false;
#ifdef DEBUG
            cout << "values size " << values.size() << endl;
#endif

            TType* elem = values[0]->typeChk(t);

            for(int i=1; i<values.size(); i++){
                if(elem != values[i]->typeChk(t))
                    err=true;
            }

            if(err){
                fprintf(stderr, "Array elements are not the same type.\n");
                return NULL;
            }else{
#ifdef DEBUG
                cout << "cons array type " << elem->name << endl;
#endif
                return elem;
            }
        
        }
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NArray:values{"<<endl;
            for(int i=0; i<values.size(); i++){
				values[i]->printTree(os,depth+1);
            }
			os<<string(depth,' ')<<"}"<<endl;
		}
};


class NIdentifier : public NLRExpression {
	public:
		std::string name;
		TElement* symbol;
		NIdentifier(std::string &name,TType* type=NULL) : name(name),symbol(type){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			TVar* temp= (TVar*) t.lookup(name);
            if(temp==NULL){
                cerr << "Variable " << name << " does not exists."<<endl;
                return NULL;
            }
			return &temp->type;
		}
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NIdentifier: name="<<name;
		}
};

class NArrayAccess : public NLRExpression{
	public:
		NLRExpression *lexpr;
		NExpression *index;
		NArrayAccess(NLRExpression *lexpr, NExpression *index):lexpr(lexpr),index(index){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			if (index->typeChk(t)->name!="integer"){ 
			    fprintf(stderr,"Array index must be an integer\n");
                return NULL;
            } 
            
            return lexpr->typeChk(t);
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NAccess: "<<endl;
			os<<string(depth+1,' ')<<"lexpr= "<<endl;
			lexpr->printTree(os,depth+2);
			os<<string(depth+1,' ')<<"index= "<<endl;
			index->printTree(os,depth+2);
		}
};

class NStructAccess : public NLRExpression{
	public:
		NLRExpression *lexpr;
		NIdentifier& name;
		NStructAccess(NLRExpression *lexpr,NIdentifier& name):lexpr(lexpr),name(name){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			TStructured* temp = (TStructured*)lexpr->typeChk(t);
			if (temp == NULL) return NULL;
#ifdef DEBUG			
			cerr<<"lexpr is a(n) "<<temp->name<<" "<<temp->struc<<temp->numeric<<temp->basic<<endl;
#endif
			if (!temp->struc) {
				cerr << temp->name<< " is not a struct or union"<<endl;
				return NULL;
			}
			TType* res = temp->accessType(name.name);
			if (res==NULL) {
				cerr<<"Struct " <<temp->name <<"doesn't have a element "<<name.name<<endl;
				return NULL;
			}
			return res;
		}
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NStructAccess: "<<endl;
			os<<string(depth+1,' ')<<"lexpr= "<<endl;
			lexpr->printTree(os,depth+2);
			os<<string(depth+1,' ')<<"name= "<<endl;
			name.printTree(os,depth+2);
		}
};

class NFunctionCall : public NExpression {
	public:
		const NIdentifier &id;
		ExpressionList arguments;
		NFunctionCall(const NIdentifier &id, ExpressionList &arguments) : id(id), arguments(arguments){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
            TType* tip;
		    std::vector<TType*> args;
			for(int i=0;i<arguments.size();i++){
                tip = arguments[i]->typeChk(t);
				if(tip == NULL){
                    cerr << "Type of argument " << i << " doesn't exist" << endl;
                }else{
                    args.push_back(tip); 
                }
			}

            if(t.lookupFunc(id.name,args) == NULL){
                cerr << "Function "<<id.name<<" was not declared"<< endl;
			    return NULL;
            }else{
                #ifdef DEBUG
                cerr << "Function call type "<<(t.lookupFunc(id.name,args)->type)->name<< endl;
                #endif
                return t.lookupFunc(id.name,args)->type;
            }
		}
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NFunctionCall: id="<< id.name<<endl;
			os<<string(depth+1,' ')<<"arguments= {"<<endl;
			for(int i=0;i<arguments.size();i++){
				arguments[i]->printTree(os,depth+2);   
			}
			os<<string(depth+1,' ')<<"}"<<endl;
		}
};


class  NBinaryOperator : public NExpression {
	public :
		string op;
		NExpression *lexp;
		NExpression *rexp;
		NBinaryOperator(NExpression* lexp,string op,NExpression* rexp):op(op),lexp(lexp),rexp(rexp){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			TType* t1=lexp->typeChk(t);
			TType* t2=rexp->typeChk(t);
			if(isalpha(op[0])){
				if(t1->name=="boolean" && t2->name=="boolean"){
					return t1;
				}else{
					fprintf(stderr,"%s expected a boolean types but received %s and %s\n",op.c_str(),t1->name.c_str(),t2->name.c_str());
					/*should be NULL*/
					return t1;
				}
			}else{
				if(t1->numeric && t2->numeric){
					if(t1->name=="float") return t1;
					if(t2->name=="float") return t2;
					return t1;
				}else{
					fprintf(stderr,"%s expected a numeric types but received %s and %s\n",op.c_str(),t1->name.c_str(),t2->name.c_str());

					/*should be NULL*/
					if(t1->name=="float") return t1;
					if(t2->name=="float") return t2;
					return t1;
				}
			}
		}
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NBinaryOp: op="<< op<<endl;
			os<<string(depth+1,' ')<<"loper= "<<endl;
			lexp->printTree(os,depth+2);   
			os<<string(depth+1,' ')<<"roper="<<endl;
			rexp->printTree(os,depth+2);   
		}
};

class NUnaryOperator : public NExpression {
	public :
		string op;
		NExpression *rexp;
		NUnaryOperator(string op,NExpression* rexp):op(op),rexp(rexp){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			TType* t2=rexp->typeChk(t);
			if(isalpha(op[0])){
				if(t2->name=="boolean"){
					return t2;
				}else{
					fprintf(stderr,"%s expected a boolean type but received a %s\n",op.c_str(),t2->name.c_str());
					return NULL;
				}
			}else{
				if(t2->numeric){
					return t2;
				}else{
					fprintf(stderr,"%s expected a numeric and received a %s\n",op.c_str(),t2->name.c_str());
					return NULL;
				}
			}
		}
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NUnaryop: op="<< op<<endl;
			os<<string(depth+1,' ')<<"roper="<<endl;
			rexp->printTree(os,depth+2);   
		}
};

class NBlock: public NStatement{
	public :
		StatementList statements;
		NBlock() {};
		TType* typeChk(Symtable& t,TType* expected = NULL){
			t.begScope();
			bool err=false;
			TType* s;
			for(int i=0;i<statements.size();i++){
#ifdef DEBUG
                cout << "statement "<< i << " " << statements[i] <<endl;
#endif
				s=statements[i]->typeChk(t,expected);
				if(s==NULL){
					cerr<<"Error in the statement "<<i<<" of block."<<endl;
					err=true;
				}
			}
			t.endScope();
			if(err) {
#ifdef DEBUG
                cerr << "error in block"<<endl;
#endif
                return NULL;
            }
#ifdef DEBUG
            cout << "void " << t.lookupType("void") << endl;
#endif
            return t.lookupType("void");
		}
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NBlock: {"<<endl;
			for(int i=0;i<statements.size();i++){
				statements[i]->printTree(os,depth+1);   
			}
			os<<string(depth,' ')<<"}"<<endl;
		}
};

class NVarrayDeclaration : public NStatement{
	public:
		NIdentifier& id;
		NIdentifier& type;
		bool array;
		NVarrayDeclaration(NIdentifier &id,NIdentifier& type,bool array=false):id(id),type(type),array(array){}
		virtual TType* typeChk(Symtable& t,TType* expected = NULL){
#ifdef DEBUG
			cerr<<"TypeCHK:default"<<endl;
#endif
			return t.lookupType("void");
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NVarrayDecl: id= "<< id.name<< " type: "<<type.name<<endl;
		}
};


class NVariableDeclaration : public NVarrayDeclaration {
	public:
		NExpression *assigment;
		NVariableDeclaration(NIdentifier& type,NIdentifier& id,NExpression* assignment=NULL):NVarrayDeclaration(id,type),assigment(assignment){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
#ifdef DEBUG
			cerr<<"TypeCHK:VarDeclaration "<<id.name<<type.name<<assigment<<endl;
#endif
			TType* t1 = t.lookupType(type.name);
			if(t1==NULL) {
				cerr<<"Type "<<type.name<<" not defined"<<endl;	
				return NULL;
			}
			if(assigment!=NULL){
				TType* t2 = assigment->typeChk(t);
                #ifdef DEBUG
				cout<<"declsign"<<t1<<' '<<t2<<endl;
                #endif
				if(t2==NULL) {cerr <<"Type Error assignment\n" ;return NULL;}
				if (t1->name!=t2->name){
					fprintf(stderr,"%s declared as %s but inicialized with %s\n",id.name.c_str(),t1->name.c_str(),t2->name.c_str());
					return NULL;
				}
				
			}
#ifdef DEBUG
			cerr<<t1->name<<endl;
#endif
			return t1;
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NVarrayDecl: id= "<< id.name<< " type: "<<type.name<<endl;
			if(assigment!=NULL) {
				os<<"initialization ="<<endl;
				assigment->printTree(os,depth+1);
			}
		}
};


class NFunctionDeclaration : public NStatement {
	public:
		const NIdentifier& type;
		const NIdentifier& id;
		VariableList args;
		NBlock *block;
		NFunctionDeclaration(const NIdentifier& type,const NIdentifier& id,const VariableList& args, NBlock* block=NULL):type(type),id(id),args(args),block(block){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			bool err=false;
			TType* x=t.lookupType(type.name);
#ifdef DEBUG
		    cout<<"Type Function test "<<type.name<<endl;
#endif
			if(x==NULL){
				cerr<<"Type "<<type.name<<" not defined"<<endl;
			}
			TType* c;
			for(int i=0;i<args.size();i++){
				c=args[i]->typeChk(t);
				if(c==NULL) err=true;
			}
			block->typeChk(t,x);
			//cerr<<x<<err<<endl;
			if(err){
				cerr << "Error in "<<id.name<<" declaration"<<endl;
				return NULL;
			}
			return t.lookupType("void");
		}

        int addSymtable(Symtable& t){
            bool err=false;
            TType* ret = t.lookupType(type.name);
            if(ret==NULL){
                cerr << "Error in function declaration " << id.name << " type " << ret->name << " does not exist" << endl;
                err=true;
            }
            std::vector<TType*>* arguments = new vector<TType*>();
            int i;
            TType* c;
            for(i=0; i<args.size(); i++){
                c = t.lookupType(args[i]->type.name);
                if(c==NULL){
                    err=true;
                    cerr << "Error in function declaration " << id.name << " type " << args[i]->type.name << " does not exist" << endl;
                }
                arguments->push_back(c);
            }
            if(err){
             return 1;
            }
            TFunc* f = new TFunc(id.name,ret,arguments);
            if(!t.insertfun(f))
                return 2;
            else
                return 0;
        }
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NFunctionDecl: id= "<< id.name<<"type="<<type.name<<endl;
			os<<string(depth,' ')<<"args= ("<< id.name<<endl;
			for(int i=0;i<args.size();i++){
				args[i]->printTree(os,depth+1);   
			}
			os<<string(depth,' ')<<")"<< id.name<<endl;
			if(block!=NULL) {
				block->printTree(os,depth+1);
			}
		}
			
};


class NArrayDeclaration : public NVarrayDeclaration{
	public:
        NArray& size;
		NArray* elements;
		NArrayDeclaration(NIdentifier& id, NIdentifier& type,
				NArray& size) :
			NVarrayDeclaration(id,type,true), size(size), elements(NULL){}

		NArrayDeclaration(NIdentifier& id, NIdentifier& type,
				NArray& size, NArray* elements) :
			NVarrayDeclaration(id,type,true), size(size), elements(elements){}


        int addSymtable(Symtable& t){
            TElement* na = t.lookup(id.name);
            TType* typ = t.lookupType(type.name);
            TType* si = size.typeChk(t);

            if(typ==NULL){
                cerr << "Error in array declaration. Type " << type.name <<  " does not exist." <<endl;
                return 1; 
            }else if(na!=NULL){
                cerr << "Error array already exists." << endl;
                return 1;
            }else if(si->isarr && ((TArray *)si)->type.name!="integer"){
                cerr << "Error array dimensions must be integers." << endl;
                return 1;
            }

#ifdef DEBUG
            cout << "typ " << typ << endl;
            cout << "value " << ((NInteger*)size.values.back())->value << endl;
#endif

            TArray* t1 = new TArray(*typ,((NInteger*)size.values.back())->value);
            typ = t1;
            for(int i=size.values.size()-2; i>=0;i--){
                t1 = new TArray(*typ,((NInteger*)size.values[i])->value);
                typ = t1;
            }

            t.insert(id.name,typ);

            return 0;
        }

		TType* typeChk(Symtable& t,TType* expected = NULL){

#ifdef DEBUG
            cout <<"\nbeginnig arr typechk"<<endl;
#endif
            TType* s = size.typeChk(t);
            TType* x = t.lookupType(type.name);
            TType* el = NULL;
            if(elements!=NULL)
                el = elements->typeChk(t);

#ifdef DEBUG
            cout << "x s el " << x << " " << s << " " << el << endl;
#endif

            if(x == NULL){
                fprintf(stderr, "Array type does not exists.\n");
                return NULL;
            }else if(s->name!="integer"){
                fprintf(stderr, "Array size expression is not an integer.\n");
                return NULL;
            }else if(elements!=NULL && el==NULL){
                return NULL;
            }else if(x!=NULL && el!=NULL && x->name!=el->name){
                fprintf(stderr, "Array type is not the same type as array elements.\n");
                return NULL;
            }

            return x;
            
        }
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NArrayDecl: id= "<< id.name<<" type="<<type.name<<endl;
			if(elements!=NULL) {
				elements->printTree(os,depth+1);
			}
		}
};

class NRegisterDeclaration : public NStatement{
	public:
		NIdentifier& type;
		VariableList fields;
		NRegisterDeclaration(NIdentifier& type, VariableList fields) :type(type), fields(fields){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			t.begScope();
			bool err=false;
			TType* temp;
			for (int i=0;i<fields.size();i++){
				temp = fields[i]->typeChk(t);
				if(temp == NULL){
					cerr<<"error in the declaration"<<i+1<<" of register "<<type.name<<endl;
					
					err = true;
				}
			}
			t.endScope();
			if (err) {
				cerr << "Error in register "<<type.name<<" declaration"<<endl;
				return NULL;
			}
			return t.lookupType("void");
		}
        
		int addToSymtable(Symtable& t){
            bool err=false;
			TVar* vars;
			TRegister* reg=new TRegister(type.name);
			for(int i=0;i<fields.size();i++){
				vars = (TVar*)t.lookup(fields[i]->id.name);
				if(vars==NULL)	cerr<<fields[i]->id.name<<" variable not declared"<<endl;
				else {
					reg->addField(&vars->type,vars->name);
				}
			}
#ifdef DEBUG
			cout<<"inserting "<<type.name<<endl;
#endif
			t.insertType(type.name,reg);
			
			return 0;
        }
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NStructDecl: id= "<< type.name<<endl;
			os<<string(depth+1,' ')<<"fields= {"<< type.name<<endl;
			for(int i=0;i<fields.size();i++){
				fields[i]->printTree(os,depth+2);   
			}
			os<<string(depth+1,' ')<<"}"<< type.name<<endl;
		}
};

class NUnionDeclaration : public NStatement{
	public:
		NIdentifier& type;
		VariableList fields;
		NUnionDeclaration(NIdentifier& type, VariableList fields) :type(type), fields(fields){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			t.begScope();
			bool err=false;
			TType* temp;
			for (int i=0;i<fields.size();i++){
				temp = fields[i]->typeChk(t);
				if(temp == NULL){
					cerr<<"error in the declaration"<<i+1<<" of union "<<type.name<<endl;
					
					err = true;
				}
			}
			t.endScope();
			if (err) {
				cerr << "Error in union "<<type.name<<" declaration"<<endl;
				return NULL;
			}
			return t.lookupType("void");
		}
        
		int addToSymtable(Symtable& t){
            bool err=false;
			TVar* vars;
			TRegister* reg=new TRegister(type.name);
			for(int i=0;i<fields.size();i++){
				vars = (TVar*)t.lookup(fields[i]->id.name);
				if(vars==NULL)	cerr<<fields[i]->id.name<<" variable not declared"<<endl;
				else {
					reg->addField(&vars->type,vars->name);
				}
			}
#ifdef DEBUG
			cout<<"inserting "<<type.name<<endl;
#endif
			t.insertType(type.name,reg);
			
			return 0;
        }
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NUnionDecl: id= "<< type.name<<endl;
			os<<string(depth+1,' ')<<"fields= {"<< type.name<<endl;
			for(int i=0;i<fields.size();i++){
				fields[i]->printTree(os,depth+2);   
			}
			os<<string(depth+1,' ')<<"}"<< type.name<<endl;
		}
};


class NWhileDo : public NStatement{
	public:
		NExpression* cond;
		NBlock& block;
		NWhileDo(NExpression* cond, NBlock& block):cond(cond),block(block){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			TType *a,*b;
			a=cond->typeChk(t);	
			b=block.typeChk(t,expected);
			if(a==NULL) {
				cerr<<"\tin while condition"<<endl; 
				return NULL;
			}else if(!(a->name=="boolean")){
				cerr<<"while condition is not a boolean expression";
				return NULL;
			}
			if(b==NULL) {
				cerr<<"\nin while block"<<endl; 
				return NULL;
			}
			return t.lookupType("void");
		}
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NWhileDo:"<<endl;
			os<<string(depth,' ')<<"cond="<<endl;
			cond->printTree(os,depth+2);
			os<<string(depth,' ')<<"block="<<endl;
			block.printTree(os,depth+2);
		}
};

class NDoWhile : public NStatement{
	public:
		NExpression* cond;
		NBlock& block;
		NDoWhile(NExpression* cond, NBlock& block):cond(cond),block(block){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			TType *a,*b;
			a=cond->typeChk(t);	
			b=block.typeChk(t,expected);
			if(a==NULL) {
				cerr<<"\tin dowhile condition"<<endl; 
				return NULL;
			}else if(!(a->name=="boolean")){
				cerr<<"while condition is not a boolean expression";
				return NULL;
			}
			if(b==NULL) {
				cerr<<"\nin dowhile block"<<endl; 
				return NULL;
			}
			return t.lookupType("void");
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NDoWhile:"<<endl;
			os<<string(depth,' ')<<"cond="<<endl;
			cond->printTree(os,depth+2);
			os<<string(depth,' ')<<"block="<<endl;
			block.printTree(os,depth+2);
		}
};

class NIf : public NStatement{
	public:
		NExpression* cond;
		NStatement& block;
		NStatement* elseBlock;
		NIf(NExpression* cond,NStatement& block):cond(cond),block(block){}
		NIf(NExpression* cond,NStatement& block, NStatement* elseBlock):cond(cond),block(block),elseBlock(elseBlock){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			TType *a,*b,*c;
			a=cond->typeChk(t);	
			b=block.typeChk(t,expected);
			if(elseBlock!=NULL) c=elseBlock->typeChk(t,expected);
			if(a==NULL || b==NULL ) return NULL;
			cond->typeChk(t);
			if( a=NULL) return NULL;
			return t.lookupType("void");
		}
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NIf:"<<endl;
			os<<string(depth+1,' ')<<"cond="<<endl;
			cond->printTree(os,depth+2);
			os<<string(depth+1,' ')<<"block="<<endl;
			block.printTree(os,depth+2);
			if(elseBlock!=NULL){
				os<<string(depth+1,' ')<<"elseblock="<<endl;
				block.printTree(os,depth+2);
			}
		}
};


class NFor : public NStatement{
	public:
		NIdentifier& id;
		NExpression* beg;
		NExpression* end;
		NExpression* step;
		NIdentifier* array;
        NArray* cons_arr;
		NBlock& block;
		NFor(NIdentifier& id,NExpression* beg,NExpression* end,NBlock& block): step(new NInteger(1)),id(id),beg(beg),end(end),cons_arr(NULL),block(block){};
		NFor(NIdentifier& id,NExpression* beg,NExpression* end,NBlock& block,NExpression* step): id(id),beg(beg),end(end),cons_arr(NULL),block(block){};
		NFor(NIdentifier& id,NIdentifier* array,NBlock &block): id(id),array(array),cons_arr(NULL),block(block),beg(NULL){};
		NFor(NIdentifier& id,NArray* cons_arr,NBlock &block): id(id),cons_arr(cons_arr),block(block),beg(NULL){};
		
		TType* typeChk(Symtable& t,TType* expected = NULL){
			bool err=false;
			if (beg==NULL){
				TType* artype;
				if(array==NULL){
					artype=cons_arr->typeChk(t);
					if(artype==NULL){
						err=true;
						cerr<<"\tin for array"<<endl;
					}
				}else{
					artype=array->typeChk(t);
					if(artype==NULL){
						err=true;
						cerr<<"\tin for array var"<<endl;
					}
				}
			}else{
				TType* b= beg->typeChk(t);
				TType* e= end->typeChk(t);
				TType* s=NULL;
				if(step!= NULL){
					s=step->typeChk(t);
				}
				if(b==NULL){
					err=true;
					cerr<<"\tin for range beginning"<<endl;
				}
				if(e==NULL){
					err=true;
					cerr<<"\tin for range end"<<endl;
				}
				if(step!=NULL&&s==NULL){
					err=true;
					cerr<<"\tin for step"<<endl;
				}
				if(b->name!="integer" || e->name!="integer" || (step!=NULL && s->name!="integer")){
					err=true;
					cerr<<"Error: For ranges and steps must be integers"<<endl;
				}
			}
			if(err)return NULL;
			return t.lookupType("void");
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NFor:"<<endl;
			if(beg!=NULL){
				os<<string(depth+1,' ')<<"range"<<endl;
				beg->printTree(os,depth+2);
				end->printTree(os,depth+2);
				if(step!=NULL){
					os<<string(depth+1,' ')<<"step"<<endl;
					beg->printTree(os,depth+2);
				}
			}else{
				os<<string(depth+1,' ')<<"arr"<<endl;
				if(array!=NULL){
					array->printTree(os,depth+2);
				}else{
					cons_arr->printTree(os,depth+2);
				}
			}
		}

};

class NStop : public NStatement{
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NStop:"<<endl;
		}
};

class NNext : public NStatement{
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NNext:"<<endl;
		}

};

class NReturn : public NStatement{
	public:
		NExpression* expr;
		NReturn():expr(NULL){}
		NReturn(NExpression *expr):expr(expr){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			TType* s;
			if (expr != NULL)s=expr->typeChk(t);
			else s=t.lookupType("void");
#ifdef DEBUG
			cerr<<"blah " <<s->name<<expected->name<<endl;
#endif
			if(s->name==expected->name){return s;}
			else {
				cerr<<"Error: returns "<<s->name<<" but function returns "<<expected->name<<endl;
				return NULL;
			}
		}
		
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NReturn:"<<endl;
			if(expr!=NULL){
				expr->printTree(os,depth+1);
			}
		}
		
};

class NAssignment : public NStatement{
	public:
		NLRExpression* var;
		NExpression* assig;
		NAssignment (NLRExpression* var, NExpression *assigment):var(var),assig(assigment){}
		TType* typeChk(Symtable& t,TType* expected = NULL){
			TType* varT = var->typeChk(t);
			TType* assigT = assig->typeChk(t);
			if (varT == NULL || assigT == NULL){
				cerr<<"TypeError on assignment"<<endl;
				return NULL;
			} else if (varT->name!=assigT->name){
				fprintf(stderr,"The assignment  expects a %s but received a %s\n",varT->name.c_str(),assigT->name.c_str());
				return NULL;
			}
			return varT;
		}
		void printTree(std::ostream& os, int depth =0){
			os<<string(depth,' ')<<"NAssignment:"<<endl;
			os<<string(depth+1,' ')<<"lvalue="<<endl;
			var->printTree(os,depth+2);
			os<<string(depth+1,' ')<<"rvalue="<<endl;
			assig->printTree(os,depth+2);
		}
};