langtools/src/share/classes/com/sun/mirror/util/SourceOrderDeclScanner.java
changeset 10 06bc494ca11e
child 3378 22011d9a9398
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/langtools/src/share/classes/com/sun/mirror/util/SourceOrderDeclScanner.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,247 @@
+/*
+ * Copyright 2004-2005 Sun Microsystems, Inc.  All Rights Reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.  Sun designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Sun in the LICENSE file that accompanied this code.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+package com.sun.mirror.util;
+
+import com.sun.mirror.declaration.*;
+
+import java.util.SortedSet;
+import java.util.TreeSet;
+
+/**
+ * A visitor for declarations that scans declarations contained within
+ * the given declaration in source code order.  For example, when
+ * visiting a class, the methods, fields, constructors, and nested
+ * types of the class are also visited.
+ *
+ * To control the processing done on a declaration, users of this
+ * class pass in their own visitors for pre and post processing.  The
+ * preprocessing visitor is called before the contained declarations
+ * are scanned; the postprocessing visitor is called after the
+ * contained declarations are scanned.
+ *
+ * @author Joseph D. Darcy
+ * @author Scott Seligman
+ * @since 1.5
+ */
+class SourceOrderDeclScanner extends DeclarationScanner {
+    static class SourceOrderComparator implements java.util.Comparator<Declaration> {
+        SourceOrderComparator(){}
+
+
+        static boolean equals(Declaration d1, Declaration d2) {
+            return d1 == d2 || (d1 != null && d1.equals(d2));
+        }
+
+        private static class DeclPartialOrder extends com.sun.mirror.util.SimpleDeclarationVisitor {
+            private int value = 1000;
+            private static int staticAdjust(Declaration d) {
+                return d.getModifiers().contains(Modifier.STATIC)?0:1;
+            }
+
+            DeclPartialOrder() {}
+
+            public int getValue() { return value; }
+
+            @Override
+            public void visitTypeParameterDeclaration(TypeParameterDeclaration d) {value = 0;}
+
+            @Override
+            public void visitEnumConstantDeclaration(EnumConstantDeclaration d) {value = 1;}
+
+            @Override
+            public void visitClassDeclaration(ClassDeclaration d) {value = 2 + staticAdjust(d);}
+
+            @Override
+            public void visitInterfaceDeclaration(InterfaceDeclaration d) {value = 4;}
+
+            @Override
+            public void visitEnumDeclaration(EnumDeclaration d) {value = 6;}
+
+            @Override
+            public void visitAnnotationTypeDeclaration(AnnotationTypeDeclaration d) {value = 8;}
+
+            @Override
+            public void visitFieldDeclaration(FieldDeclaration d) {value = 10 + staticAdjust(d);}
+
+            @Override
+            public void visitConstructorDeclaration(ConstructorDeclaration d) {value = 12;}
+
+            @Override
+            public void visitMethodDeclaration(MethodDeclaration d) {value = 14 + staticAdjust(d);}
+        }
+        @SuppressWarnings("cast")
+        private int compareEqualPosition(Declaration d1, Declaration d2) {
+            assert d1.getPosition() == d2.getPosition();
+
+            DeclPartialOrder dpo1 = new DeclPartialOrder();
+            DeclPartialOrder dpo2 = new DeclPartialOrder();
+
+            d1.accept(dpo1);
+            d2.accept(dpo2);
+
+            int difference = dpo1.getValue() - dpo2.getValue();
+            if (difference != 0)
+                return difference;
+            else {
+                int result = d1.getSimpleName().compareTo(d2.getSimpleName());
+                if (result != 0)
+                    return result;
+                return (int)( Long.signum((long)System.identityHashCode(d1) -
+                                          (long)System.identityHashCode(d2)));
+            }
+        }
+
+        public int compare(Declaration d1, Declaration d2) {
+            if (equals(d1, d2))
+                return 0;
+
+            SourcePosition p1 = d1.getPosition();
+            SourcePosition p2 = d2.getPosition();
+
+            if (p1 == null && p2 != null)
+                return 1;
+            else if (p1 != null && p2 == null)
+                return -1;
+            else if(p1 == null && p2 == null)
+                return compareEqualPosition(d1, d2);
+            else {
+                assert p1 != null && p2 != null;
+                int fileComp = p1.file().compareTo(p2.file()) ;
+                if (fileComp == 0) {
+                    long diff = (long)p1.line() - (long)p2.line();
+                    if (diff == 0) {
+                        diff = Long.signum((long)p1.column() - (long)p2.column());
+                        if (diff != 0)
+                            return (int)diff;
+                        else {
+                            // declarations may be two
+                            // compiler-generated members with the
+                            // same source position
+                            return compareEqualPosition(d1, d2);
+                        }
+                    } else
+                        return (diff<0)? -1:1;
+                } else
+                    return fileComp;
+            }
+        }
+    }
+
+    final static java.util.Comparator<Declaration> comparator = new SourceOrderComparator();
+
+    SourceOrderDeclScanner(DeclarationVisitor pre, DeclarationVisitor post) {
+        super(pre, post);
+    }
+
+    /**
+     * Visits a type declaration.
+     *
+     * @param d the declaration to visit
+     */
+    public void visitTypeDeclaration(TypeDeclaration d) {
+        d.accept(pre);
+
+        SortedSet<Declaration> decls = new
+            TreeSet<Declaration>(SourceOrderDeclScanner.comparator) ;
+
+        for(TypeParameterDeclaration tpDecl: d.getFormalTypeParameters()) {
+            decls.add(tpDecl);
+        }
+
+        for(FieldDeclaration fieldDecl: d.getFields()) {
+            decls.add(fieldDecl);
+        }
+
+        for(MethodDeclaration methodDecl: d.getMethods()) {
+            decls.add(methodDecl);
+        }
+
+        for(TypeDeclaration typeDecl: d.getNestedTypes()) {
+            decls.add(typeDecl);
+        }
+
+        for(Declaration decl: decls )
+            decl.accept(this);
+
+        d.accept(post);
+    }
+
+    /**
+     * Visits a class declaration.
+     *
+     * @param d the declaration to visit
+     */
+    public void visitClassDeclaration(ClassDeclaration d) {
+        d.accept(pre);
+
+        SortedSet<Declaration> decls = new
+            TreeSet<Declaration>(SourceOrderDeclScanner.comparator) ;
+
+        for(TypeParameterDeclaration tpDecl: d.getFormalTypeParameters()) {
+            decls.add(tpDecl);
+        }
+
+        for(FieldDeclaration fieldDecl: d.getFields()) {
+            decls.add(fieldDecl);
+        }
+
+        for(MethodDeclaration methodDecl: d.getMethods()) {
+            decls.add(methodDecl);
+        }
+
+        for(TypeDeclaration typeDecl: d.getNestedTypes()) {
+            decls.add(typeDecl);
+        }
+
+        for(ConstructorDeclaration ctorDecl: d.getConstructors()) {
+            decls.add(ctorDecl);
+        }
+
+        for(Declaration decl: decls )
+            decl.accept(this);
+
+        d.accept(post);
+    }
+
+    public void visitExecutableDeclaration(ExecutableDeclaration d) {
+        d.accept(pre);
+
+        SortedSet<Declaration> decls = new
+            TreeSet<Declaration>(SourceOrderDeclScanner.comparator) ;
+
+        for(TypeParameterDeclaration tpDecl: d.getFormalTypeParameters())
+            decls.add(tpDecl);
+
+        for(ParameterDeclaration pDecl: d.getParameters())
+            decls.add(pDecl);
+
+        for(Declaration decl: decls )
+            decl.accept(this);
+
+        d.accept(post);
+    }
+
+}