import java.util.HashMap;
import java.util.Iterator;
-import edu.uci.eecs.specCompiler.specExtraction.ActionSubConstruct.DefineVar;
import edu.uci.eecs.specCompiler.specExtraction.CPDefineCheckConstruct;
import edu.uci.eecs.specCompiler.specExtraction.CPDefineConstruct;
+import edu.uci.eecs.specCompiler.specExtraction.ClassEndConstruct;
import edu.uci.eecs.specCompiler.specExtraction.ConditionalInterface;
import edu.uci.eecs.specCompiler.specExtraction.Construct;
+import edu.uci.eecs.specCompiler.specExtraction.EntryPointConstruct;
import edu.uci.eecs.specCompiler.specExtraction.GlobalConstruct;
import edu.uci.eecs.specCompiler.specExtraction.IDExtractor;
import edu.uci.eecs.specCompiler.specExtraction.InterfaceConstruct;
+import edu.uci.eecs.specCompiler.specExtraction.InterfaceDefineConstruct;
import edu.uci.eecs.specCompiler.specExtraction.PotentialCPDefineConstruct;
import edu.uci.eecs.specCompiler.specExtraction.SequentialDefineSubConstruct;
+import edu.uci.eecs.specCompiler.specExtraction.SourceFileInfo;
import edu.uci.eecs.specCompiler.specExtraction.SpecExtractor;
/**
private File[] srcFiles;
- private HashMap<File, ArrayList<String>> contents;
+ private HashMap<File, SourceFileInfo> srcFilesInfo;
private HashMap<File, ArrayList<CodeAddition>> codeAdditions;
- private ArrayList<String> globalContent;
-
public CodeGenerator(File[] srcFiles) {
this.srcFiles = srcFiles;
_extractor = new SpecExtractor();
_extractor.extract(srcFiles);
-
- this.contents = _extractor.contents;
-
- this.globalContent = null;
- this.codeAdditions = new HashMap<File, ArrayList<CodeAddition>>();
+ this.srcFilesInfo = _extractor.srcFilesInfo;
- _semantics = new SemanticsChecker(_extractor.constructs);
+ this.codeAdditions = new HashMap<File, ArrayList<CodeAddition>>();
+
+ _semantics = new SemanticsChecker(_extractor);
try {
_semantics.check();
System.out.println(_semantics);
}
}
- private ArrayList<String> readSrcFile(File f) throws IOException {
- BufferedReader bf = new BufferedReader(new FileReader(f));
- ArrayList<String> content = new ArrayList<String>();
- String curLine;
- while ((curLine = bf.readLine()) != null) {
- content.add(curLine);
- }
- return content;
- }
-
/**
* <p>
* Generate all the global code, including the "@DefineVar" in each
private void globalConstruct2Code(GlobalConstruct construct) {
ArrayList<String> newCode = CodeVariables.generateGlobalVarDeclaration(
_semantics, construct);
- // Record the global content array to generate the new file
- globalContent = newCode;
+ // Add it to the codeAdditions
+ if (!codeAdditions.containsKey(construct.file)) {
+ codeAdditions.put(construct.file, new ArrayList<CodeAddition>());
+ }
+ CodeAddition addition = new CodeAddition(construct.beginLineNum,
+ newCode);
+ codeAdditions.get(construct.file).add(addition);
+ newCode = CodeVariables.generateStaticVarDefine(_semantics, construct);
+ if (newCode.size() > 0) {
+ addition = new CodeAddition(_semantics.getClassEndConstruct().beginLineNum, newCode);
+ codeAdditions.get(construct.file).add(addition);
+ }
}
// Mainly rename and wrap the interface
private void interface2Code(InterfaceConstruct construct)
throws InterfaceWrongFormatException {
- int lineNum = construct.begin + 1;
-
- // Rename the interface name
- File file = inst.file;
- String funcDecl = inst.interfaceDeclBody;
- // Rename the function declaration
- String funcName = renameInterface(inst);
- // Also rename the function definition if it's separated from the
- // declaration
- SpecConstruct definition = _semantics.interfaceName2DefineConstruct
- .get(construct.name);
- if (definition != null) {
- String funcDefintionName = renameInterface(definition);
- assert (funcDefintionName.equals(funcName));
- }
-
- // Generate new wrapper
ArrayList<String> newCode = CodeVariables.generateInterfaceWrapper(
- _semantics, inst);
+ _semantics, construct);
+ int lineNum = construct.beginLineNum;
// Add it to the codeAdditions
CodeAddition addition = new CodeAddition(lineNum, newCode);
- if (!codeAdditions.containsKey(inst.file)) {
- codeAdditions.put(inst.file, new ArrayList<CodeAddition>());
- }
- codeAdditions.get(inst.file).add(addition);
- }
-
- // Returns the function name that has been renamed and replace the old line
- private String renameInterface(Construct inst)
- throws InterfaceWrongFormatException {
- String funcDecl = inst.interfaceDeclBody;
- ArrayList<String> content = contents.get(inst.file);
-
- // Depending on "(" to find the function name, so it doesn't matter if
- // there's any template
- int beginIdx = funcDecl.indexOf('(');
- if (beginIdx == -1) {
- throw new InterfaceWrongFormatException(funcDecl
- + "\n has wrong format!");
+ if (!codeAdditions.containsKey(construct.file)) {
+ codeAdditions.put(construct.file, new ArrayList<CodeAddition>());
}
- IDExtractor idExtractor = new IDExtractor(funcDecl, beginIdx);
- String funcName = idExtractor.getPrevID();
- int idBeginIdx = idExtractor.getIDBeginIdx(), idEndIdx = idExtractor
- .getIDEndIdx(), idLineBeginIdx = idExtractor.lineBeginIdxOfID(), idLineEndIdx = idExtractor
- .lineEndIdxOfID();
- String newLine = funcDecl.substring(idLineBeginIdx, idBeginIdx)
- + CodeVariables.SPEC_INTERFACE_WRAPPER + funcName
- + funcDecl.substring(idEndIdx + 1, idLineEndIdx + 1);
-
- int lineNumOfID = idExtractor.lineNumOfID();
- // Be careful: lineNum - 1 -> index of content array
- content.set(inst.endLineNum + lineNumOfID, newLine);
- return funcName;
+ codeAdditions.get(construct.file).add(addition);
}
private void potentialCPDefine2Code(PotentialCPDefineConstruct construct) {
int lineNum = construct.beginLineNum;
- ArrayList<String> newCode = new ArrayList<String>();
+ ArrayList<String> newCode = CodeVariables.generatePotentialCPDefine(
+ _semantics, construct);
CodeAddition addition = new CodeAddition(lineNum, newCode);
if (!codeAdditions.containsKey(construct.file)) {
private void CPDefine2Code(CPDefineConstruct construct) {
int lineNum = construct.beginLineNum;
- ArrayList<String> newCode = new ArrayList<String>();
+ ArrayList<String> newCode = CodeVariables.generateCPDefine(_semantics, construct);
CodeAddition addition = new CodeAddition(lineNum, newCode);
if (!codeAdditions.containsKey(construct.file)) {
}
private void CPDefineCheck2Code(CPDefineCheckConstruct construct) {
+ int lineNum = construct.beginLineNum;
+ ArrayList<String> newCode = CodeVariables.generateCPDefineCheck(_semantics, construct);
+
+ CodeAddition addition = new CodeAddition(lineNum, newCode);
+ if (!codeAdditions.containsKey(construct.file)) {
+ codeAdditions.put(construct.file, new ArrayList<CodeAddition>());
+ }
+ codeAdditions.get(construct.file).add(addition);
+ }
+
+ private void EntryPoint2Code(EntryPointConstruct construct) {
int lineNum = construct.beginLineNum;
ArrayList<String> newCode = new ArrayList<String>();
+ newCode.add(")
CodeAddition addition = new CodeAddition(lineNum, newCode);
if (!codeAdditions.containsKey(construct.file)) {
}
codeAdditions.get(construct.file).add(addition);
}
+
public void generateCode() {
for (int i = 0; i < _semantics.constructs.size(); i++) {