X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=src%2Fedu%2Fuci%2Feecs%2FspecCompiler%2FcodeGenerator%2FCodeGenerator.java;h=28609e6de8c3a47262adaed1e4ca4b9d714ddada;hb=dddb190f43cd31d29dd0ce06739fc9cd22f25f73;hp=d65258424fb63dea628b7eed8b11e572a59e9996;hpb=f2d6efad8b1ee05ef5dd12175af9e59e61c7b8f8;p=cdsspec-compiler.git diff --git a/src/edu/uci/eecs/specCompiler/codeGenerator/CodeGenerator.java b/src/edu/uci/eecs/specCompiler/codeGenerator/CodeGenerator.java index d652584..28609e6 100644 --- a/src/edu/uci/eecs/specCompiler/codeGenerator/CodeGenerator.java +++ b/src/edu/uci/eecs/specCompiler/codeGenerator/CodeGenerator.java @@ -6,19 +6,24 @@ import java.io.FileNotFoundException; 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.CPClearConstruct; 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.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; @@ -42,6 +47,7 @@ public class CodeGenerator { private HashMap srcFilesInfo; private HashMap> codeAdditions; + private String homeDir; public CodeGenerator(File[] srcFiles) { this.srcFiles = srcFiles; @@ -85,17 +91,62 @@ public class CodeGenerator { } } - // Mainly rename and wrap the interface + // Wrap the interface and then renaem it private void interface2Code(InterfaceConstruct construct) { - ArrayList 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()); + // 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 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()); + } + 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()); + } + codeAdditions.get(defineConstruct.file).add(addition); + } else { // No declaration needed but should add forward declaration in + // Class + // Last generate the definition + newCode = new ArrayList(); + 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()); + } + 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) { @@ -133,6 +184,31 @@ public class CodeGenerator { } codeAdditions.get(construct.file).add(addition); } + + private void CPClear2Code(CPClearConstruct construct) { + int lineNum = construct.beginLineNum; + ArrayList newCode = CodeVariables.generateCPClear( + _semantics, construct); + + CodeAddition addition = new CodeAddition(lineNum, newCode); + if (!codeAdditions.containsKey(construct.file)) { + codeAdditions.put(construct.file, new ArrayList()); + } + codeAdditions.get(construct.file).add(addition); + } + + + /** + * private void ClassEnd2Code(ClassEndConstruct construct) { int lineNum = + * construct.beginLineNum; ArrayList newCode = + * CodeVariables.generateStaticVarDefine(_semantics, + * _semantics.getGlobalConstruct()); + * + * CodeAddition addition = new CodeAddition(lineNum, newCode); if + * (!codeAdditions.containsKey(construct.file)) { + * codeAdditions.put(construct.file, new ArrayList()); } + * codeAdditions.get(construct.file).add(addition); } + */ private void EntryPoint2Code(EntryPointConstruct construct) { int lineNum = construct.beginLineNum; @@ -146,6 +222,28 @@ public class CodeGenerator { codeAdditions.get(construct.file).add(addition); } + private ArrayList insertAnnotation2Src( + ArrayList additions, ArrayList content) { + int totalSize = content.size(); + for (int i = 0; i < additions.size(); i++) { + totalSize += additions.size(); + } + ArrayList newContent = new ArrayList(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); @@ -159,19 +257,99 @@ public class CodeGenerator { CPDefine2Code((CPDefineConstruct) construct); } else if (construct instanceof CPDefineCheckConstruct) { CPDefineCheck2Code((CPDefineCheckConstruct) construct); + } else if (construct instanceof CPClearConstruct) { + CPClear2Code((CPClearConstruct) construct); + } else if (construct instanceof EntryPointConstruct) { + EntryPoint2Code((EntryPointConstruct) construct); } } + + // ClassEndConstruct endConstruct = _semantics.getClassEndConstruct(); + // if (endConstruct != null) { + // ClassEnd2Code(endConstruct); + // } + + // Sort code additions + for (File file : codeAdditions.keySet()) { + ArrayList additions = codeAdditions.get(file); + + if (additions.size() == 0) // Simply do nothing + continue; + ArrayList content = _semantics.srcFilesInfo.get(file).content; + Collections.sort(additions, CodeAddition.lineNumComparator); + // Insert generated annotation to the source files + ArrayList 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") }; - CodeGenerator gen = new CodeGenerator(srcFiles); - gen.generateCode(); + + File[] srcLinuxRWLock1 = { + new File(homeDir + + "/benchmark/linuxrwlocks/linuxrwlocks.c") }; + File[] srcLinuxRWLock2 = { + new File(homeDir + + "/benchmark/linuxrwlocks/testcase1.c") }; + File[] srcLinuxRWLock3 = { + new File(homeDir + + "/benchmark/linuxrwlocks/testcase2.c") }; +// + File[] srcHashtable = { + new File(homeDir + + "/benchmark/concurrent-hashmap/hashmap.h"), + new File(homeDir + "/benchmark/concurrent-hashmap/testcase1.cc"), + new File(homeDir + "/benchmark/concurrent-hashmap/main.cc")}; + + File[] srcMSQueue = { + new File(homeDir + "/benchmark/ms-queue/my_queue.c"), + new File(homeDir + "/benchmark/ms-queue/testcase1.c"), + new File(homeDir + "/benchmark/ms-queue/testcase2.c"), + new File(homeDir + "/benchmark/ms-queue/testcase3.c"), + new File(homeDir + "/benchmark/ms-queue/main.c"), + new File(homeDir + "/benchmark/ms-queue/my_queue.h") }; + + File[] srcRCU = { new File(homeDir + + "/benchmark/read-copy-update/rcu.cc") }; +// + File[] srcTrylock = { new File(homeDir + + "/benchmark/trylock/trylock.c") }; + + File[] srcDeque = { + 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/testcase1.c"), + new File(homeDir + "/benchmark/chase-lev-deque-bugfix/testcase2.c"), + new File(homeDir + "/benchmark/chase-lev-deque-bugfix/deque.h") }; +// + File[] srcMCSLock = { + new File(homeDir + "/benchmark/mcs-lock/mcs-lock.cc"), + new File(homeDir + "/benchmark/mcs-lock/mcs-lock.h") }; +// + File[] srcSPSCQueue = { + 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") }; + + File[] srcMPMCQueue = { + new File(homeDir + "/benchmark/mpmc-queue/mpmc-queue.h"), + new File(homeDir + "/benchmark/mpmc-queue/testcase1.cc"), + new File(homeDir + "/benchmark/mpmc-queue/testcase2.cc"), + new File(homeDir + "/benchmark/mpmc-queue/testcase3.cc"), + new File(homeDir + "/benchmark/mpmc-queue/mpmc-queue.cc") }; +// +// File[][] sources = {srcLinuxRWLock1 , srcMSQueue, srcRCU, +// srcDeque, srcMCSLock, srcSPSCQueue, srcMPMCQueue, srcHashtable }; + + File[][] sources = {srcDeque, srcLinuxRWLock1, srcLinuxRWLock2, srcLinuxRWLock3, srcMCSLock, srcHashtable, srcRCU, srcMSQueue, srcSPSCQueue, srcMPMCQueue}; +// File[][] sources = {srcMSQueue}; + // Compile all the benchmarks + for (int i = 0; i < sources.length; i++) { + CodeGenerator gen = new CodeGenerator(sources[i]); + gen.generateCode(); + } } }