* Copyright (c) 2014, 2016, Oracle and/or its affiliates. All rights reserved.

 * @bug 8038455 8047347

 * @clean Gen

 * @compile/process -processor OnDemandAttribution OnDemandAttributionData.java

import java.io.IOException;

import java.io.Writer;

import static javax.lang.model.util.ElementFilter.constructorsIn;

    int round;

    Set<String> roots = new HashSet<>();

        for (Element root : rEnv.getRootElements()) {

            while (root.getEnclosingElement().getKind() != ElementKind.PACKAGE) {

                root = root.getEnclosingElement();

            }

            roots.add(((TypeElement) root).getQualifiedName().toString());

        }

        for (String root : roots) {

            TypeElement currentClass = elements.getTypeElement(root);

            ExecutableElement constr = constructorsIn(currentClass.getEnclosedElements()).get(0);

            Trees trees = Trees.instance(processingEnv);

            TreePath path = trees.getPath(constr);

            new TreePathScanner<Void, Void>() {

                @Override public Void visitClass(ClassTree node, Void p) {

                    if (node.getSimpleName().contentEquals("Local")) {

                         //will also attribute the body on demand:

                        Element el = trees.getElement(getCurrentPath());

                        Name binaryName = elements.getBinaryName((TypeElement) el);

                        if (!binaryName.contentEquals("OnDemandAttributionData$1Local")) {

                            throw new IllegalStateException("Incorrect binary name=" + binaryName);

                        }

                    }

                    return super.visitClass(node, p);

                }

                @Override public Void visitNewClass(NewClassTree node, Void p) {

                    Name binaryName = elements.getBinaryName((TypeElement) el.getEnclosingElement());

                    if (!binaryName.contentEquals("OnDemandAttributionData$1")) {

                    return super.visitNewClass(node, p);

                @Override

                public Void visitMethodInvocation(MethodInvocationTree node, Void p) {

                    if (trees.getElement(getCurrentPath()) == null)

                        throw new IllegalStateException("No element for: " + node);

                    return super.visitMethodInvocation(node, p);

                }

                @Override

                public Void visitIdentifier(IdentifierTree node, Void p) {

                    if (trees.getElement(getCurrentPath()) == null)

                        throw new IllegalStateException("No element for: " + node);

                    return super.visitIdentifier(node, p);

                }

            }.scan(path, null);

        }

        if (round++ == 0) {

            try (Writer out = filer.createSourceFile("Gen").openWriter()) {

                out.write("class Gen { public static long C = 1; }");

            } catch (IOException ex) {

                throw new IllegalStateException(ex);

        }