org.eclipse.jdt.compiler.apt/batch/org/eclipse/jdt/internal/compiler/apt/dispatch/RoundEnvImpl.java

package org.eclipse.jdt.internal.compiler.apt.dispatch;

import java.lang.annotation.Annotation;
import java.util.Collections;
import java.util.Set;

import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;

import org.eclipse.jdt.internal.compiler.apt.util.ManyToMany;
import org.eclipse.jdt.internal.compiler.ast.CompilationUnitDeclaration;

public class RoundEnvImpl implements RoundEnvironment
{

	private boolean _errorRaised = false;
	private final boolean _isLastRound;
	private final CompilationUnitDeclaration[] _units;
	private final ManyToMany<TypeElement, Element> _annoToUnit;

	public RoundEnvImpl(CompilationUnitDeclaration[] units, boolean isLastRound) {
		_isLastRound = isLastRound;
		_units = units;
		
		// TODO: deal with inherited annotations (esp. annotations inherited from binary supertypes)
		
		// Discover the annotations that will be passed to Processor.process()
		AnnotationDiscoveryVisitor visitor = new AnnotationDiscoveryVisitor();
		for (CompilationUnitDeclaration unit : _units) {
			unit.traverse(visitor, unit.scope);
		}
		_annoToUnit = visitor._annoToElement;
	}

	/**
	 * @return the set of annotation types that were discovered on the root elements.
	 */
	public Set<TypeElement> getRootAnnotations()
	{
		return Collections.unmodifiableSet(_annoToUnit.getKeySet());
	}

	@Override
	public boolean errorRaised()
	{
		return _errorRaised;
	}

	@Override
	public Set<? extends Element> getElementsAnnotatedWith(TypeElement a)
	{
		return _annoToUnit.getValues(a);
	}

	@Override
	public Set<? extends Element> getElementsAnnotatedWith(Class<? extends Annotation> a)
	{
		// TODO get the annotation type corresponding to "a".
		return null;
	}

	@Override
	public Set<? extends Element> getRootElements()
	{
		// TODO Convert _units into Set<TypeElement>
		return null;
	}

	@Override
	public boolean processingOver()
	{
		return _isLastRound;
	}

}