Add tracking of src class names

Author: chaserhkj

jskparser/jskparser/jskparser.py

@@ -7,6 +7,7 @@
 import os
 
 from ast.compilationunit import CompilationUnit
+from ast.body.classorinterfacedeclaration import ClassOrInterfaceDeclaration
 from ast.utils import utils
 from ast.dataflow.dataflow import DataFlow
 from ast.visit.sourcevisitor import SourcePrinter
@@ -31,6 +32,17 @@ def parse(path, **kwargs):
         d.update({u'GSYMTAB':'RESET'})
         logging.info('parsing file...')
         program = CompilationUnit(d)
+        top_class_names = []
+        def find_classes(node):
+            if type(node) == ClassOrInterfaceDeclaration and \
+                not node.isinner():
+                for annotation in node.annotations:
+                    if annotation.name.name == u"JSketchStdLib":
+                        return
+                top_class_names.append(node.name)
+        class_finder = GenericVisitor(find_classes)
+        program.accept(class_finder)
+        program.src_class_names = top_class_names
 
         s = SymtabGen(lib=lib)
         logging.info('generating symbol table...')

model/lang/Character.java

@@ -1,5 +1,6 @@
 package java.lang;
 
+@JSketchStdLib
 public class Character {
     // public static final int MIN_RADIX = 2;
     // public static final int MAX_RADIX = 36;

model/lang/Integer.java

@@ -1,5 +1,6 @@
 package java.lang;
 
+@JSketchStdLib
 public class Integer extends Number {
     public static final int   MIN_VALUE;
 

model/lang/Iterable.java

@@ -2,6 +2,7 @@
 
 import java.util.Iterator;
 
+@JSketchStdLib
 public interface Iterable <T> {
     public Iterator<T> iterator();
 }

model/lang/Number.java

@@ -6,6 +6,7 @@
 
 // }
 
+@JSketchStdLib
 public class Number {
 
 }

model/lang/Object.java

@@ -1,3 +1,4 @@
+@JSketchStdLib
 public class Object {
     // public static boolean equals(Object a, Object b) {
     // 	if (a == null) {

model/lang/String.java

@@ -1,5 +1,6 @@
 package java.lang;
 
+@JSketchStdLib
 public class String implements CharSequence{
     char[] _value;
     int _count;

model/util/ArrayList.java

@@ -1,5 +1,6 @@
 package java.util;
 
+@JSketchStdLib
 private class ArrayListIterator<E> implements Iterator<E> {
     private ArrayList<E> es = null;
     private int index = -1;
@@ -28,6 +29,7 @@ public void remove() {
     }
 }
 
+@JSketchStdLib
 public class ArrayList<E> implements List<E>{
 
     Object[] elementData;

model/util/Arrays.java

@@ -1,5 +1,6 @@
 package java.util;
 
+@JSketchStdLib
 private class ArrayAsListIterator<E> implements Iterator<E> {
     private E[] es = null;
     private int index = -1;
@@ -24,6 +25,7 @@ public void remove() {
     }
 }
 
+@JSketchStdLib
 private class ArrayAsList<E> implements List<E> {
     private E[] es = null;
     public AsList(E[] es) {
@@ -61,6 +63,7 @@ public Iterator<E> iterator() {
     }
 }
 
+@JSketchStdLib
 public class Arrays {
 
     public static byte[] copyOf(byte[] in, int len) {

model/util/HashMap.java

@@ -1,4 +1,5 @@
 // public class HashMap<K,V> extends Map {
+@JSketchStdLib
 public class HashMap<K,V> {
     static final int DEFAULT_INITIAL_CAPACITY;// = 1 << 4; // aka 16
     Node[] elementData;