package algorithms; import java.util.ArrayList; import java.util.Arrays; import java.util.HashSet; import java.util.List; import java.util.Set; import code.ast.Block; import code.ast.CompilationUnit; import code.ast.FieldDeclaration; import code.ast.ImportDeclaration; import code.ast.MethodDeclaration; import code.ast.TypeDeclaration; import code.ast.VariableDeclaration; import models.Edge; import models.Node; import models.algebra.Expression; import models.algebra.Field; import models.algebra.Parameter; import models.algebra.Symbol; import models.algebra.Term; import models.algebra.Type; import models.algebra.Variable; import models.dataConstraintModel.ChannelGenerator; import models.dataConstraintModel.ChannelMember; import models.dataConstraintModel.DataConstraintModel; import models.dataConstraintModel.IdentifierTemplate; import models.dataFlowModel.DataFlowModel; import models.dataFlowModel.DataflowChannelGenerator; import models.dataFlowModel.DataflowChannelGenerator.IResourceStateAccessor; import models.dataFlowModel.PushPullAttribute; import models.dataFlowModel.PushPullValue; import models.dataFlowModel.ResourceDependency; import models.dataFlowModel.ResourceDependencyGraph; import models.dataFlowModel.ResourceNode; import models.dataFlowModel.StoreAttribute; /** * Generator for plain Java prototypes * * @author Nitta * */ public class JavaCodeGenerator { public static final Type typeVoid = new Type("Void", "void"); private static String defaultMainTypeName = "Main"; static String mainTypeName = defaultMainTypeName; public static String getMainTypeName() { return mainTypeName; } public static void setMainTypeName(String mainTypeName) { JavaCodeGenerator.mainTypeName = mainTypeName; } public static void resetMainTypeName() { JavaCodeGenerator.mainTypeName = defaultMainTypeName; } static public ArrayList<CompilationUnit> doGenerate(ResourceDependencyGraph graph, DataFlowModel model) { ArrayList<CompilationUnit> codes = new ArrayList<>(); ArrayList<ResourceNode> resources = determineResourceOrder(graph); TypeDeclaration mainType = new TypeDeclaration(mainTypeName); CompilationUnit mainCU = new CompilationUnit(mainType); mainCU.addImport(new ImportDeclaration("java.util.*")); codes.add(mainCU); for (ResourceNode rn: resources) { boolean f = false; String resourceName = rn.getIdentifierTemplate().getResourceName().substring(0, 1).toUpperCase() + rn.getIdentifierTemplate().getResourceName().substring(1); TypeDeclaration type = new TypeDeclaration(resourceName); // Declare the field to refer each resource in the main type. String fieldInitializer = "new " + resourceName + "("; for (Edge e : rn.getOutEdges()) { ResourceDependency re = (ResourceDependency) e; String rename = ((ResourceNode) re.getDestination()).getIdentifierTemplate().getResourceName() .substring(0, 1).toUpperCase() + ((ResourceNode) re.getDestination()).getIdentifierTemplate().getResourceName().substring(1); if (((PushPullAttribute) re.getAttribute()).getOptions().get(0) == PushPullValue.PUSH) { fieldInitializer += rename.toLowerCase() + ","; f = true; } } for (Edge e : rn.getInEdges()) { ResourceDependency re = (ResourceDependency) e; String rename = ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceName() .substring(0, 1).toUpperCase() + ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceName().substring(1); if (((PushPullAttribute) re.getAttribute()).getOptions().get(0) != PushPullValue.PUSH) { fieldInitializer += rename.toLowerCase() + ","; f = true; } else { if (rn.getIndegree() > 1) { // Declare a field to cash the state of the source resource in the type of the destination resource. String cashInitializer = null; Type cashType = ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceStateType(); if (DataConstraintModel.typeList.isAncestorOf(cashType)) { cashInitializer = "new " + cashType.getImplementationTypeName() + "()"; } type.addField(new FieldDeclaration( cashType, ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceName(), cashInitializer)); } } } Set<IdentifierTemplate> refs = new HashSet<>(); for (ChannelGenerator cg : model.getChannelGenerators()) { DataflowChannelGenerator c = (DataflowChannelGenerator) cg; if (c.getOutputIdentifierTemplates().contains(rn.getIdentifierTemplate())) { for (IdentifierTemplate id: c.getReferenceIdentifierTemplates()) { if (!refs.contains(id)) { refs.add(id); String refResName = id.getResourceName(); fieldInitializer += refResName.toLowerCase() + ","; f = true; } } } } if (f) fieldInitializer = fieldInitializer.substring(0, fieldInitializer.length() - 1); fieldInitializer += ")"; FieldDeclaration field = new FieldDeclaration(new Type(resourceName, resourceName), rn.getIdentifierTemplate().getResourceName(), fieldInitializer); mainType.addField(field); // Declare a constructor, fields and update methods in the type of each resource. MethodDeclaration constructor = new MethodDeclaration(resourceName, true); Block block = new Block(); for (Edge e : rn.getOutEdges()) { ResourceDependency re = (ResourceDependency) e; String dstResName = ((ResourceNode) re.getDestination()).getIdentifierTemplate().getResourceName() .substring(0, 1).toUpperCase() + ((ResourceNode) re.getDestination()).getIdentifierTemplate().getResourceName().substring(1); if (((PushPullAttribute) re.getAttribute()).getOptions().get(0) == PushPullValue.PUSH) { // Declare a field to refer to the destination resource of push transfer. type.addField(new FieldDeclaration(new Type(dstResName, dstResName), ((ResourceNode) re.getDestination()).getIdentifierTemplate().getResourceName())); constructor.addParameter(new VariableDeclaration(new Type(dstResName, dstResName), ((ResourceNode) re.getDestination()).getIdentifierTemplate().getResourceName())); block.addStatement("this." + dstResName.toLowerCase() + " = " + dstResName.toLowerCase() + ";"); constructor.setBody(block); } } block = new Block(); for (Edge e : rn.getInEdges()) { ResourceDependency re = (ResourceDependency) e; String srcResName = ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceName() .substring(0, 1).toUpperCase() + ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceName().substring(1); if (((PushPullAttribute) re.getAttribute()).getOptions().get(0) != PushPullValue.PUSH) { // Declare a field to refer to the source resource of pull transfer. type.addField(new FieldDeclaration(new Type(srcResName, srcResName), ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceName())); constructor.addParameter(new VariableDeclaration(new Type(srcResName, srcResName), ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceName())); block.addStatement("this." + srcResName.toLowerCase() + " = " + srcResName.toLowerCase() + ";"); constructor.setBody(block); } else { // Declare an update method in the type of the destination resource. ArrayList<VariableDeclaration> vars = new ArrayList<>(); vars.add(new VariableDeclaration( ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceStateType(), ((ResourceNode) re.getSource()).getIdentifierTemplate().getResourceName())); type.addMethod(new MethodDeclaration("update" + srcResName, false, typeVoid, vars)); } } // Declare a field to refer to the reference resource. refs = new HashSet<>(); for (ChannelGenerator cg : model.getChannelGenerators()) { DataflowChannelGenerator c = (DataflowChannelGenerator) cg; if (c.getOutputIdentifierTemplates().contains(rn.getIdentifierTemplate())) { for (IdentifierTemplate id: c.getReferenceIdentifierTemplates()) { if (!refs.contains(id)) { refs.add(id); String refResName = id.getResourceName(); refResName = refResName.substring(0, 1).toUpperCase() + refResName.substring(1); type.addField(new FieldDeclaration(new Type(refResName, refResName), id.getResourceName())); constructor.addParameter(new VariableDeclaration(new Type(refResName, refResName), id.getResourceName())); block.addStatement("this." + refResName.toLowerCase() + " = " + refResName.toLowerCase() + ";"); constructor.setBody(block); } } } } if (constructor.getParameters() != null) type.addMethod(constructor); // Declare input methods in resources and the main type. for (ChannelGenerator cg : model.getIOChannelGenerators()) { for (ChannelMember cm : cg.getChannelMembers()) { if (cm.getIdentifierTemplate().getResourceName().equals(type.getTypeName().toLowerCase())) { Expression message = cm.getStateTransition().getMessageExpression(); if (message.getClass() == Term.class) { ArrayList<VariableDeclaration> params = new ArrayList<>(); for (Variable var: message.getVariables().values()) { params.add(new VariableDeclaration(var.getType(), var.getName())); } MethodDeclaration input = new MethodDeclaration( ((Term) cm.getStateTransition().getMessageExpression()).getSymbol().getImplName(), false, typeVoid, params); type.addMethod(input); String str = ((Term) cm.getStateTransition().getMessageExpression()).getSymbol().getImplName(); input = getMethod(mainType, str); if (input == null) { input = new MethodDeclaration(str, false, typeVoid, params); mainType.addMethod(input); } else { // Add type to a parameter without type. for (VariableDeclaration param: input.getParameters()) { if (param.getType() == null) { for (VariableDeclaration p: params) { if (param.getName().equals(p.getName()) && p.getType() != null) { param.setType(p.getType()); } } } } } } } } } // Declare the field to store the state in the type of each resource. if (((StoreAttribute) rn.getAttribute()).isStored()) { String initializer = "new " + rn.getIdentifierTemplate().getResourceStateType().getImplementationTypeName() + "()"; Type stateType = rn.getIdentifierTemplate().getResourceStateType(); if (!DataConstraintModel.typeList.isAncestorOf(stateType)) initializer = null; type.addField(new FieldDeclaration(stateType, "value", initializer)); } // Declare the getter method to obtain the state in the type of each resource. type.addMethod(new MethodDeclaration("getValue", rn.getIdentifierTemplate().getResourceStateType())); // Add compilation unit for each resource. CompilationUnit cu = new CompilationUnit(type); cu.addImport(new ImportDeclaration("java.util.*")); codes.add(cu); } // Declare the Pair class. TypeDeclaration type = new TypeDeclaration("Pair"); type.addField(new FieldDeclaration(new Type("Double","double"), "first")); type.addField(new FieldDeclaration(new Type("Double","double"), "second")); for(FieldDeclaration field : type.getFields()) { MethodDeclaration getter = new MethodDeclaration( "get" + field.getName().substring(0,1).toUpperCase() + field.getName().substring(1), new Type("Double","double")); getter.setBody(new Block()); getter.getBody().addStatement("return " + field.getName() + ";"); type.addMethod(getter); } CompilationUnit cu = new CompilationUnit(type); cu.addImport(new ImportDeclaration("java.util.*")); codes.add(cu); // Declare getter methods in the main type. for (Node n : graph.getNodes()) { ResourceNode rn = (ResourceNode) n; MethodDeclaration getter = new MethodDeclaration( "get" + rn.getIdentifierTemplate().getResourceName().substring(0, 1).toUpperCase() + rn.getIdentifierTemplate().getResourceName().substring(1), rn.getIdentifierTemplate().getResourceStateType()); getter.setBody(new Block()); getter.getBody().addStatement( "return " + rn.getIdentifierTemplate().getResourceName() + "." + getter.getName() + "();"); mainType.addMethod(getter); } HashSet<String> tmps = new HashSet<>(); HashSet<String> cont = new HashSet<>(); for (MethodDeclaration method : mainType.getMethods()) { if (!tmps.contains(method.getName())) tmps.add(method.getName()); else cont.add(method.getName()); } for (MethodDeclaration method : mainType.getMethods()) { if (cont.contains(method.getName())) { method.setName(method.getName() + method.getParameters().get(0).getName().substring(0, 1).toUpperCase() + method.getParameters().get(0).getName().substring(1)); } } return codes; } static public ArrayList<String> getCodes(ArrayList<TypeDeclaration> codeTree) { ArrayList<String> codes = new ArrayList<>(); for (TypeDeclaration type : codeTree) { codes.add("public class " + type.getTypeName() + "{"); for (FieldDeclaration field : type.getFields()) { if (type.getTypeName() != mainTypeName) { String cons = "\t" + "private " + field.getType().getInterfaceTypeName() + " " + field.getName(); if (DataConstraintModel.isListType(field.getType())) cons += " = new ArrayList<>()"; cons += ";"; codes.add(cons); } else { String cons = "\t" + "private " + field.getType().getInterfaceTypeName() + " " + field.getName() + " = new " + field.getType().getTypeName() + "("; for (TypeDeclaration tree : codeTree) { if (field.getType().getTypeName() == tree.getTypeName()) { for (VariableDeclaration var : tree.getConstructors()) { cons += var.getName() + ","; } if (!tree.getConstructors().isEmpty()) cons = cons.substring(0, cons.length() - 1); break; } } cons += ");"; codes.add(cons); } } codes.add(""); if (type.getTypeName() != mainTypeName) { if (!type.getConstructors().isEmpty()) { String cons = "\t" + "public " + type.getTypeName() + "("; for (VariableDeclaration constructor : type.getConstructors()) { cons += constructor.getType().getTypeName() + " " + constructor.getName() + ","; } if (!type.getConstructors().isEmpty()) cons = cons.substring(0, cons.length() - 1); cons += "){"; codes.add(cons); for (FieldDeclaration field : type.getFields()) { for (VariableDeclaration vari : type.getConstructors()) { if (field.getType().getTypeName().equals(vari.getType().getTypeName())) { codes.add("\t\t" + "this." + field.getName() + " = " + field.getName() + ";"); } } } codes.add("\t" + "}"); codes.add(""); } } for (MethodDeclaration method : type.getMethods()) { String varstr = "\t" + "public " + method.getReturnType().getInterfaceTypeName() + " " + method.getName() + "("; if (method.getParameters() != null) { for (VariableDeclaration var : method.getParameters()) { varstr += var.getType().getInterfaceTypeName() + " " + var.getName() + ","; } if (!method.getParameters().isEmpty()) varstr = varstr.substring(0, varstr.length() - 1); } if (method.getBody() != null) { for (String str : method.getBody().getStatements()) { codes.add("\t\t" + str + ";"); } } codes.add(varstr + ")" + "{"); codes.add("\t" + "}"); codes.add(""); } codes.add("}"); codes.add(""); } return codes; } static private ArrayList<ResourceNode> determineResourceOrder(ResourceDependencyGraph graph) { ArrayList<ResourceNode> resources = new ArrayList<>(); Set<ResourceNode> visited = new HashSet<>(); for (Node n : graph.getNodes()) { ResourceNode rn = (ResourceNode) n; topologicalSort(graph, rn, visited, resources); } return resources; } static private void topologicalSort(ResourceDependencyGraph graph, ResourceNode curNode, Set<ResourceNode> visited, List<ResourceNode> orderedList) { if (visited.contains(curNode)) return; visited.add(curNode); for (Edge e : curNode.getInEdges()) { ResourceDependency re = (ResourceDependency) e; if (((PushPullAttribute) re.getAttribute()).getOptions().get(0) == PushPullValue.PUSH) { topologicalSort(graph, (ResourceNode) re.getSource(), visited, orderedList); } } for (Edge e : curNode.getOutEdges()) { ResourceDependency re = (ResourceDependency) e; if (((PushPullAttribute) re.getAttribute()).getOptions().get(0) != PushPullValue.PUSH) { topologicalSort(graph, (ResourceNode) re.getDestination(), visited, orderedList); } } for (Node n: graph.getNodes()) { // for reference resources. ResourceNode rn = (ResourceNode) n; for (Edge e : rn.getInEdges()) { ResourceDependency re = (ResourceDependency) e; for (ChannelMember m: re.getChannelGenerator().getReferenceChannelMembers()) { if (m.getIdentifierTemplate() == curNode.getIdentifierTemplate()) { topologicalSort(graph, rn, visited, orderedList); } } } } orderedList.add(0, curNode); } private static MethodDeclaration getMethod(TypeDeclaration type, String methodName) { for (MethodDeclaration m: type.getMethods()) { if (m.getName().equals(methodName)) return m; } return null; } static public IResourceStateAccessor pushAccessor = new IResourceStateAccessor() { @Override public Expression getCurrentStateAccessorFor(IdentifierTemplate target, IdentifierTemplate from) { if (target.equals(from)) { return new Field("value", target.getResourceStateType() != null ? target.getResourceStateType() : DataConstraintModel.typeInt); } return null; } @Override public Expression getNextStateAccessorFor(IdentifierTemplate target, IdentifierTemplate from) { return new Parameter(target.getResourceName(), target.getResourceStateType() != null ? target.getResourceStateType() : DataConstraintModel.typeInt); } }; static public IResourceStateAccessor pullAccessor = new IResourceStateAccessor() { @Override public Expression getCurrentStateAccessorFor(IdentifierTemplate target, IdentifierTemplate from) { if (target.equals(from)) { return new Field(target.getResourceName(), target.getResourceStateType() != null ? target.getResourceStateType() : DataConstraintModel.typeInt); } // for reference channel member Term getter = new Term(new Symbol("getValue", 1, Symbol.Type.METHOD)); getter.addChild(new Field(target.getResourceName(), target.getResourceStateType())); return getter; } @Override public Expression getNextStateAccessorFor(IdentifierTemplate target, IdentifierTemplate from) { Term getter = new Term(new Symbol("getValue", 1, Symbol.Type.METHOD)); getter.addChild(new Field(target.getResourceName(), target.getResourceStateType())); return getter; } }; }