/**
* Determines the selected class.
*
* @param dObj the d obj
* @throws IllegalArgumentException the illegal argument exception
*/
protected final void determineSelectedClass(DataObject dObj) throws IllegalArgumentException {
if (null != dObj) {
FileObject fo = dObj.getPrimaryFile();
JavaSource jsource = JavaSource.forFileObject(fo);
if (jsource == null) {
StatusDisplayer.getDefault().setStatusText("Not a Java file: " + fo.getPath());
} else {
// StatusDisplayer.getDefault().setStatusText("Hurray! A Java file: " + fo.getPath());
try {
jsource.runUserActionTask(
(CompilationController p) -> {
p.toPhase(JavaSource.Phase.ELEMENTS_RESOLVED);
CompilationUnitTree tree = p.getCompilationUnit();
MemberVisitor scanner = new MemberVisitor(p);
scanner.scan(p.getCompilationUnit(), null);
te = scanner.getTypeElement();
Document document = p.getDocument();
},
true);
} catch (IOException ex) {
Exceptions.printStackTrace(ex);
}
}
}
}
private static int[] spanFor(
CompilationInfo info, JavaSource js, final PackageMemberAnnotation annotation) {
final int[] result = new int[] {-1, -1};
class TaskImpl implements Task<CompilationInfo> {
@Override
public void run(final CompilationInfo parameter) {
TypeElement clazz = parameter.getElements().getTypeElement(annotation.getClassName());
if (clazz == null) {
// XXX: log
return;
}
Element resolved = null;
if (annotation instanceof FieldAnnotation) {
FieldAnnotation fa = (FieldAnnotation) annotation;
for (VariableElement var : ElementFilter.fieldsIn(clazz.getEnclosedElements())) {
if (var.getSimpleName().contentEquals(fa.getFieldName())) {
resolved = var;
break;
}
}
} else if (annotation instanceof MethodAnnotation) {
MethodAnnotation ma = (MethodAnnotation) annotation;
for (ExecutableElement method : ElementFilter.methodsIn(clazz.getEnclosedElements())) {
if (method.getSimpleName().contentEquals(ma.getMethodName())) {
if (ma.getMethodSignature()
.equals(SourceUtils.getJVMSignature(ElementHandle.create(method))[2])) {
resolved = method;
break;
}
}
}
} else if (annotation instanceof ClassAnnotation) {
resolved = clazz;
}
if (resolved == null) {
// XXX: log
return;
}
final Element resolvedFin = resolved;
new CancellableTreePathScanner<Void, Void>() {
@Override
public Void visitVariable(VariableTree node, Void p) {
if (resolvedFin.equals(parameter.getTrees().getElement(getCurrentPath()))) {
int[] span = parameter.getTreeUtilities().findNameSpan(node);
if (span != null) {
result[0] = span[0];
result[1] = span[1];
}
}
return super.visitVariable(node, p);
}
@Override
public Void visitMethod(MethodTree node, Void p) {
if (resolvedFin.equals(parameter.getTrees().getElement(getCurrentPath()))) {
int[] span = parameter.getTreeUtilities().findNameSpan(node);
if (span != null) {
result[0] = span[0];
result[1] = span[1];
}
}
return super.visitMethod(node, p);
}
@Override
public Void visitClass(ClassTree node, Void p) {
if (resolvedFin.equals(parameter.getTrees().getElement(getCurrentPath()))) {
int[] span = parameter.getTreeUtilities().findNameSpan(node);
if (span != null) {
result[0] = span[0];
result[1] = span[1];
}
}
return super.visitClass(node, p);
}
}.scan(parameter.getCompilationUnit(), null);
}
};
final TaskImpl convertor = new TaskImpl();
if (info != null) {
convertor.run(info);
} else {
try {
js.runUserActionTask(
new Task<CompilationController>() {
@Override
public void run(CompilationController parameter) throws Exception {
parameter.toPhase(
Phase.RESOLVED); // XXX: ENTER should be enough in most cases, but not for
// anonymous innerclasses.
convertor.run(parameter);
}
},
true);
} catch (IOException ex) {
Exceptions.printStackTrace(ex);
}
}
return result;
}
