checkin1.py

import sys
import minic.minic_ast as mc
import minic.c_ast_to_minic as ctomc
from pycparser import parse_file
from pycparser.c_ast import *


class GetAllVariables(NodeVisitor):
    
    def __init__(self):
        self.var_all = []
        self.var_written = []
    
    def visit_ID(self, Id):
        if not Id.name in self.var_all:
            self.var_all.append(Id.name)
    
    def getAllVariables(self):
        return self.var_all
    
    def getWrittenVariables(self):
        return self.var_written 
    
    
class GetWrittenVariables(NodeVisitor):
    
    def __init__(self):
        self.var_all = []
        self.var_written = []
        
    def visit_Assignment(self, assignment):
        if isinstance(assignment.lvalue, mc.ID):
            self.var_written.append(assignment.lvalue.name)
            
        if isinstance(assignment.lvalue, mc.ArrayRef):
            self.var_written.append(assignment.lvalue.name.name)
        
        self.generic_visit(assignment)
    
    def getWrittenVariables(self):
        return self.var_written
    

if __name__=="__main__":
    
    #filename = './project3inputs/p3_input4'
    filename = sys.argv[1]
    txt = ''
    
    with open(filename) as f:
        txt = f.read()

    txt = 'void wrapper(){ \n' + txt + '\n}'
    
    c_filename = filename + '.c'
    c_file = open(c_filename, 'w')
    c_file.write(txt)
    c_file.close()

    ast = parse_file(c_filename)
    mc_ast = ctomc.transform(ast)
    
    allvar = GetAllVariables()
    allvar.visit(mc_ast)
    print "All:", allvar.getAllVariables()
    
    writevar = GetWrittenVariables()
    writevar.visit(mc_ast)
    print "Written:", writevar.getWrittenVariables()