import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
+import java.util.Collections;
import java.util.HashMap;
+import java.util.HashSet;
import java.util.Iterator;
import edu.uci.eecs.specCompiler.specExtraction.CPDefineCheckConstruct;
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.FunctionHeader;
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.ParserUtils;
import edu.uci.eecs.specCompiler.specExtraction.PotentialCPDefineConstruct;
import edu.uci.eecs.specCompiler.specExtraction.SequentialDefineSubConstruct;
import edu.uci.eecs.specCompiler.specExtraction.SourceFileInfo;
}
}
- // Mainly rename and wrap the interface
+ // Wrap the interface and then renaem it
private void interface2Code(InterfaceConstruct construct) {
- ArrayList<String> newCode = CodeVariables.generateInterfaceWrapper(
- _semantics, construct);
- int lineNum = construct.beginLineNum;
- // Add it to the codeAdditions
- CodeAddition addition = new CodeAddition(lineNum, newCode);
- if (!codeAdditions.containsKey(construct.file)) {
- codeAdditions.put(construct.file, new ArrayList<CodeAddition>());
+ // If there's no define construct for it, we generate the wrapper just
+ // in place without declaration
+ InterfaceDefineConstruct defineConstruct = _semantics.interfaceName2DefineConstruct
+ .get(construct.name);
+ ArrayList<String> newCode;
+ int lineNum;
+ CodeAddition addition;
+ // Then generate the wrapper if necessary
+ if (defineConstruct != null) { // Need to have a wrapper declaration
+ newCode = CodeVariables.generateInterfaceWrapperDeclaration(
+ _semantics, construct);
+ lineNum = construct.beginLineNum;
+ // Add the wrapper declaration
+ addition = new CodeAddition(lineNum, newCode);
+ if (!codeAdditions.containsKey(construct.file)) {
+ codeAdditions
+ .put(construct.file, new ArrayList<CodeAddition>());
+ }
+ codeAdditions.get(construct.file).add(addition);
+
+ // Add the wrapper definition
+ newCode = CodeVariables.generateInterfaceWrapperDefinition(
+ _semantics, construct);
+ lineNum = defineConstruct.beginLineNum;
+ // Add the wrapper declaration
+ addition = new CodeAddition(lineNum, newCode);
+ if (!codeAdditions.containsKey(defineConstruct.file)) {
+ codeAdditions.put(defineConstruct.file,
+ new ArrayList<CodeAddition>());
+ }
+ codeAdditions.get(defineConstruct.file).add(addition);
+ } else { // No declaration needed but should add forward declaration in
+ // Class
+ // Last generate the definition
+ newCode = new ArrayList<String>();
+ if (_semantics.getOption("CLASS") == null) {
+ newCode.addAll(CodeVariables
+ .generateInterfaceWrapperDeclaration(_semantics,
+ construct));
+ }
+ newCode.addAll(CodeVariables.generateInterfaceWrapperDefinition(
+ _semantics, construct));
+ lineNum = construct.beginLineNum;
+ // Add the wrapper declaration
+ addition = new CodeAddition(lineNum, newCode);
+ if (!codeAdditions.containsKey(construct.file)) {
+ codeAdditions
+ .put(construct.file, new ArrayList<CodeAddition>());
+ }
+ codeAdditions.get(construct.file).add(addition);
}
- codeAdditions.get(construct.file).add(addition);
+
+ // Don't forget to rename the interface
+ CodeVariables.renameInterface(_semantics, construct);
}
private void potentialCPDefine2Code(PotentialCPDefineConstruct construct) {
codeAdditions.get(construct.file).add(addition);
}
+ /**
+ private void ClassEnd2Code(ClassEndConstruct construct) {
+ int lineNum = construct.beginLineNum;
+ ArrayList<String> newCode = CodeVariables.generateStaticVarDefine(_semantics,
+ _semantics.getGlobalConstruct());
+
+ 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>();
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) {
+ // Be careful, subList is the interval [begin, end)
+ newContent
+ .addAll(content.subList(curSrcLine, addition.lineNum));
+ curSrcLine = addition.lineNum;
+ }
+ newContent.addAll(addition.newCode);
+ }
+ newContent.addAll(content.subList(curSrcLine, content.size()));
+ 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);
}
}
+
+// ClassEndConstruct endConstruct = _semantics.getClassEndConstruct();
+// if (endConstruct != null) {
+// ClassEnd2Code(endConstruct);
+// }
+
+ // Sort code additions
+ HashSet<String> headers = CodeVariables.getAllHeaders(_semantics);
+ ArrayList<String> headerCode = new ArrayList<String>(headers.size());
+ for (String header : headers) {
+ headerCode.add("#include " + header);
+ }
+ for (File file : codeAdditions.keySet()) {
+ ArrayList<CodeAddition> additions = codeAdditions.get(file);
+
+ if (additions.size() == 0) // Simply do nothing
+ 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);
+ ArrayList<String> finalContent = new ArrayList<String>(
+ headerCode.size() + newContent.size());
+ finalContent.addAll(headerCode);
+ finalContent.addAll(newContent);
+ // Write it back to file
+ ParserUtils.write2File(file, finalContent);
+ }
}
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/main.c"),
+ // new File(homeDir + "/benchmark/ms-queue/my_queue.h") };
+
+ // new File(homeDir + "/benchmark/read-copy-update/rcu.cc") };
+
+ // new File(homeDir +
+ // "/benchmark/chase-lev-deque-bugfix/deque.c"),
+ // new File(homeDir +
+ // "/benchmark/chase-lev-deque-bugfix/main.c"),
+ // new File(homeDir +
+ // "/benchmark/chase-lev-deque-bugfix/deque.h") };
+
+// new File(homeDir + "/benchmark/mcs-lock/mcs-lock.cc"),
+// new File(homeDir + "/benchmark/mcs-lock/mcs-lock.h") };
+
+ new File(homeDir + "/benchmark/spsc-bugfix/spsc-queue.cc"),
+ new File(homeDir + "/benchmark/spsc-bugfix/eventcount.h"),
+ new File(homeDir + "/benchmark/spsc-bugfix/queue.h") };
+
+
CodeGenerator gen = new CodeGenerator(srcFiles);
gen.generateCode();
}