1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver/* 2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver [The "BSD licence"] 3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver Copyright (c) 2009 Shaoting Cai 4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver All rights reserved. 5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver Redistribution and use in source and binary forms, with or without 7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver modification, are permitted provided that the following conditions 8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver are met: 9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 1. Redistributions of source code must retain the above copyright 10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver notice, this list of conditions and the following disclaimer. 11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 2. Redistributions in binary form must reproduce the above copyright 12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver notice, this list of conditions and the following disclaimer in the 13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver documentation and/or other materials provided with the distribution. 14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 3. The name of the author may not be used to endorse or promote products 15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver derived from this software without specific prior written permission. 16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver*/ 28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverpackage org.antlr.gunit.swingui.runner; 29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport java.io.*; 31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport java.util.HashMap; 32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver/** 34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Class loader for parser & lexer generated by antlr. 35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * @author Shaoting 36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver */ 37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverpublic class ParserLoader extends ClassLoader { 38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver private HashMap<String, Class> classList; 40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver private String grammar; 41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver /** 43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Create a class loader for antlr parser/lexer. 44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * @param grammarName 45324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * @param classDir 46324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver */ 47324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver public ParserLoader(String grammarName, String classDir) throws IOException, ClassNotFoundException { 48324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 49324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver final String lexerName = grammarName + "Lexer"; 50324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 51324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver // load all the class files in the "classDir" related to the grammarName 52324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver File dir = new File(classDir); 53324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if(dir.isDirectory()) { 54324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver classList = new HashMap<String, Class>(); 55324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver grammar = grammarName; 56324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver File[] files = dir.listFiles(new ClassFilenameFilter(grammarName)); 57324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver for(File f : files) { 58324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 59324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver // load class data 60324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver final InputStream in = new BufferedInputStream(new FileInputStream(f)); 61324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver final byte[] classData = new byte[in.available()]; 62324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver in.read(classData); 63324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver in.close(); 64324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 65324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver // define class 66324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver final Class newClass = defineClass(null, classData, 0, classData.length); 67324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver assert(newClass != null); 68324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver resolveClass(newClass); 69324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 70324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver // save to hashtable 71324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver final String fileName = f.getName(); 72324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver final String className = fileName.substring(0, fileName.lastIndexOf(".")); 73324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver classList.put(className, newClass); 74324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver //System.out.println("adding: " + className); 75324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 76324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } else { 77324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver throw new IOException(classDir + " is not a directory."); 78324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 79324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 80324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if(classList.isEmpty() || !classList.containsKey(lexerName)) { 81324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver throw new ClassNotFoundException(lexerName + " not found."); 82324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 83324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 84324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 85324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 86324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 87324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 88324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @Override 89324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver public synchronized Class loadClass(String name, boolean resolve) throws ClassNotFoundException { 90324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver //System.out.print("loading: " + name); 91324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if(name.startsWith(grammar)) { 92324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if(classList.containsKey(name)) { 93324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver //System.out.println(" .... found"); 94324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return classList.get(name); 95324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } else { 96324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver //System.out.println(" .... not found"); 97324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver throw new ClassNotFoundException(name); 98324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 99324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 100324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } else { 101324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver final Class c = findSystemClass(name); 102324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver //System.out.println(" .... system found " + c.getName()); 103324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return c; 104324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 105324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 106324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 107324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver /** 108324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Accepts grammarname...($...)?.class 109324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver */ 110324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver protected static class ClassFilenameFilter implements FilenameFilter { 111324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 112324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver private String grammarName; 113324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 114324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver protected ClassFilenameFilter(String name) { 115324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver grammarName = name; 116324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 117324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 118324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver public boolean accept(File dir, String name) { 119324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return name.startsWith(grammarName) && name.endsWith(".class"); 120324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 121324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 122324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 123324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 124324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 125