/**
* @author Stanislav Lapitsky
* @version 1.0
*
* Adapted by Nuno Brito in 2013-09-18 from:
* http://www.developer.com/java/other/article.php/3318421/Add-Line-Numbering-in-the-JEditorPane.htm
*/
package GUI;
import java.awt.Graphics;
import java.awt.Rectangle;
import javax.swing.text.*;
public class NumberedEditorKit extends StyledEditorKit {
@Override
public ViewFactory getViewFactory() {
return new NumberedViewFactory();
}
}
class NumberedViewFactory implements ViewFactory {
@Override
public View create(Element elem) {
String kind = elem.getName();
if (kind != null)
if (kind.equals(AbstractDocument.ContentElementName)) {
return new LabelView(elem);
}
else if (kind.equals(AbstractDocument.
ParagraphElementName)) {
// return new ParagraphView(elem);
return new NumberedParagraphView(elem);
}
else if (kind.equals(AbstractDocument.
SectionElementName)) {
return new BoxView(elem, View.Y_AXIS);
}
else if (kind.equals(StyleConstants.
ComponentElementName)) {
return new ComponentView(elem);
}
else if (kind.equals(StyleConstants.IconElementName)) {
return new IconView(elem);
}
// default to text display
return new LabelView(elem);
}
}
final class NumberedParagraphView extends ParagraphView {
public static short NUMBERS_WIDTH=42;
public NumberedParagraphView(Element e) {
super(e);
short top = 0;
short left = 0;
short bottom = 0;
short right = 0;
this.setInsets(top, left, bottom, right);
}
@Override
protected void setInsets(short top, short left, short bottom,
short right) {super.setInsets
(top,(short)(left+NUMBERS_WIDTH),
bottom,right);
}
@Override
public void paintChild(Graphics g, Rectangle r, int n) {
super.paintChild(g, r, n);
int previousLineCount = getPreviousLineCount();
int numberX = r.x - getLeftInset();
int numberY = r.y + r.height - 5;
g.drawString(Integer.toString(previousLineCount + n + 1),
numberX, numberY);
}
public int getPreviousLineCount() {
int lineCount = 0;
View parent = this.getParent();
int count = parent.getViewCount();
for (int i = 0; i < count; i++) {
if (parent.getView(i) == this) {
break;
}
else {
lineCount += parent.getView(i).getViewCount();
}
}
return lineCount;
}
}