TypeCheckVisitor.java

// Name:TypeCheckVisitor.java
// Author: David Sinclair      Date: 29 Aug 2012
//
// Visitor for basic type checking expressions in an abstract syntax tree in the ExprLang language
//

import java.util.*;

public class TypeCheckVisitor implements ParserVisitor
{

    private static HashMap<String, HashMap<String, STC>> ST = new HashMap<>();

    private static HashSet<String> declaredFuncts = new HashSet<String>();
    private static HashSet<String> calledFuncts = new HashSet<String>();

    private static String currentScope = "";

    public Object visit(SimpleNode node, Object data)
    {
        throw new RuntimeException("Visit SimpleNode");
    }

    public Object visit(ASTProgram node, Object data)
    {
        currentScope = "global";
        HashMap<String, HashMap<String, STC>> ST = (HashMap<String, HashMap<String, STC>>) data;
        //System.out.println(ST);
        node.jjtGetChild(0).jjtAccept(this, data);
        node.jjtGetChild(0); // var decl
        node.jjtGetChild(1); // const decl
        node.jjtGetChild(2); // function
        node.jjtGetChild(3); // main prog

        node.jjtGetChild(0).jjtAccept(this, data);
        node.jjtGetChild(1).jjtAccept(this, data);
        node.jjtGetChild(2).jjtAccept(this, data); // This makes function get called
        node.jjtGetChild(3).jjtAccept(this, data);

        for(Map.Entry<String, HashMap<String, STC>> entry : ST.entrySet()) {

            String tempScope = entry.getKey();
            if (!tempScope.equals("main") && !tempScope.equals("global")){
                calledFuncts.add(tempScope);
            }
        }

        if(calledFuncts.size() == declaredFuncts.size()){
            System.out.println("All declared functions were called");
        }
        else{
            System.out.println("Not all declared functions were called");
        }
        return DataType.Program;
    }

    public Object visit(ASTVarDecl node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.VarDeclaration;
    }

    public Object visit(ASTConstDecl node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.ConstDeclaration;
    }

    public Object visit(ASTEqual node, Object data)
    {

        return DataType.Equal;
    }

    public Object visit(ASTFunction node, Object data)
    {
        currentScope = "is_prime";
        node.jjtGetChild(0).jjtAccept(this, data); // type
        node.jjtGetChild(1).jjtAccept(this, data); // ID
        System.out.println(node.jjtGetChild(3).jjtAccept(this, data));
        System.out.println(node.jjtGetChild(4).jjtAccept(this, data));
        //System.out.println("^_^ Function ^_^");
        //System.out.println(node.value);
        //System.out.println();
        return DataType.Function;
    }

    public Object visit(ASTParamList node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.ParamList;
    }

    public Object visit(ASTType node, Object data)
    {
        //node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.Int;
    }

    public Object visit(ASTMainProg node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        currentScope = "main";
        return DataType.MainProg;
    }

    public Object visit(ASTAssign node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.Assign;
    }

    public Object visit(ASTMathSign node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.Plus;
        //return DataType.Minus;
        //return DataType.Div;
        //return DataType.Mult;
    }

    public Object visit(ASTCondition node, Object data)
    {
        System.out.println(node.jjtGetChild(0).jjtAccept(this, data));
        System.out.println(node.jjtGetChild(1).jjtAccept(this, data));
        System.out.println(node.jjtGetChild(2).jjtAccept(this, data));
        //System.out.println(node.jjtGetChild(3).jjtAccept(this, data));
        node.jjtGetChild(0).jjtAccept(this, data);
        node.jjtGetChild(1).jjtAccept(this, data);
        node.jjtGetChild(2).jjtAccept(this, data); // This makes function get called
        //node.jjtGetChild(3).jjtAccept(this, data);

        return DataType.Condition;
    }

    public Object visit(ASTSigns node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.Equal;

        //return DataType.NotEqual;
        //return DataType.LessThan;
        //return DataType.GreaterThan;
        //return DataType.LTE;
        //return DataType.GTE;

    }

    public Object visit(ASTIdentList node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.IdentList;
    }

    public Object visit(ASTArgList node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.ArgList;
    }

    public Object visit(ASTNum node, Object data)
    {

        System.out.println("INTEGER CITY");
        return DataType.TypeInteger;
    }

    public Object visit(ASTBool node, Object data)
    {
        node.jjtGetChild(0).jjtAccept(this, data);
        return DataType.TypeBoolean;
    }

    public Object visit(ASTId node, Object data)
    {
        //Hashtable ST = (Hashtable) data;
        HashMap<String, HashMap<String, STC>> ST = (HashMap<String, HashMap<String, STC>>) data;
        HashMap<String, STC> innerMap = new HashMap();

        if(node.value.equals(currentScope) && !node.value.equals("main")){ // function

            declaredFuncts.add((String) node.value);
            return DataType.Function;
        }
        /*else if(node.value == "main"){ // main


        }
        else if(){ // global */
        else {
            innerMap = ST.get(currentScope);
            /*System.out.println(ST);
            System.out.println(currentScope);
            System.out.println(innerMap);
            System.out.println(node.value);*/
            STC mapEntry = innerMap.get(node.value);
            if (mapEntry.type == "Int") {
                return DataType.TypeInteger;
            } else if (mapEntry.type == "Bool") {
                return DataType.TypeBoolean;
            } else {
                return DataType.TypeUnknown;
            }
        }
    }
}