Adding some JavaDoc and cleaning up a bit.

[ifi-stolz-refaktor.git] / software / no.uio.ifi.refaktor / src / no / uio / ifi / refaktor / utils / SmartTextSelection.java

package no.uio.ifi.refaktor.utils;

import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.dom.NodeFinder;
import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.ITextSelection;
import org.eclipse.jface.text.TextSelection;

/**
 * A custom TextSelection that enforces the presence of
 * a document to back the selection.
 */
public class SmartTextSelection extends TextSelection implements ITextSelection {

	public SmartTextSelection(IDocument document, int offset, int length) {
		super(document, offset, length);
		if (document == null)
			throw new NullPointerException(this.getClass().getCanonicalName() + ": given document cannot be null");
	}
	
	public SmartTextSelection(IDocument document, ITextSelection selection) {
		this(document, selection.getOffset(), selection.getLength());
	}

	@Override
	public IDocument getDocument() {
		IDocument document = super.getDocument();
		assert document != null;
		return document;
	}

	public int getEnd() {
		return getOffset() + getLength();
	}

	public NodeFinder getNodeFinder() {
		return ParseUtils.getNodeFinder(getDocument(), this);
	}

	public NodeFinder getNodeFinder(ICompilationUnit icu) {
		return ParseUtils.makeNodeFinder(this, ParseUtils.parse(icu));
	}
}