import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
+import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
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.ParserUtils;
import edu.uci.eecs.specCompiler.specExtraction.PotentialCPDefineConstruct;
import edu.uci.eecs.specCompiler.specExtraction.SequentialDefineSubConstruct;
import edu.uci.eecs.specCompiler.specExtraction.SourceFileInfo;
codeAdditions.get(construct.file).add(addition);
}
+ private ArrayList<String> insertAnnotation2Src(
+ ArrayList<CodeAddition> additions, ArrayList<String> content) {
+ int totalSize = content.size();
+ for (int i = 0; i < additions.size(); i++) {
+ totalSize += additions.size();
+ }
+ ArrayList<String> newContent = new ArrayList<String>(totalSize);
+ int curSrcLine = 0;
+ for (int i = 0; i < additions.size(); i++) {
+ CodeAddition addition = additions.get(i);
+ if (curSrcLine < addition.lineNum) {
+ newContent.addAll(content.subList(curSrcLine, addition.lineNum - 1));
+ curSrcLine = addition.lineNum;
+ }
+ newContent.addAll(addition.newCode);
+ }
+ return newContent;
+ }
+
public void generateCode() {
for (int i = 0; i < _semantics.constructs.size(); i++) {
Construct construct = _semantics.constructs.get(i);
CPDefine2Code((CPDefineConstruct) construct);
} else if (construct instanceof CPDefineCheckConstruct) {
CPDefineCheck2Code((CPDefineCheckConstruct) construct);
+ } else if (construct instanceof EntryPointConstruct) {
+ EntryPoint2Code((EntryPointConstruct) construct);
}
}
+ // Sort code additions
+ for (File file : codeAdditions.keySet()) {
+ ArrayList<CodeAddition> additions = codeAdditions.get(file);
+ if (additions.size() == 0) // Simply do nothing, already written
+ // once
+ continue;
+ ArrayList<String> content = _semantics.srcFilesInfo.get(file).content;
+ Collections.sort(additions, CodeAddition.lineNumComparator);
+ // Insert generated annotation to the source files
+ ArrayList<String> newContent = insertAnnotation2Src(additions,
+ content);
+ // Write it back to file
+ ParserUtils.write2File(file, newContent);
+ }
+
}
public static void main(String[] argvs) {
String homeDir = Environment.HOME_DIRECTORY;
File[] srcFiles = {
- // new File(homeDir + "/benchmark/linuxrwlocks/linuxrwlocks.c"),
- new File(homeDir
- + "/benchmark/cliffc-hashtable/simplified_cliffc_hashtable.h"), };
-// new File(homeDir + "/benchmark/ms-queue/my_queue.c"),
-// new File(homeDir + "/benchmark/ms-queue/my_queue.h") };
+ new File(homeDir + "/benchmark/linuxrwlocks/linuxrwlocks.c") };
+// new File(homeDir
+// + "/benchmark/cliffc-hashtable/simplified_cliffc_hashtable.h"), };
+ // new File(homeDir + "/benchmark/ms-queue/my_queue.c"),
+ // new File(homeDir + "/benchmark/ms-queue/my_queue.h") };
CodeGenerator gen = new CodeGenerator(srcFiles);
gen.generateCode();
}