SyntaxDocument.java

Go to the documentation of this file.

/*
 * SyntaxDocument.java - Document that can be tokenized
 * Copyright (C) 1999 Slava Pestov
 * Copyright (C) 2000-2023 The Gridarta Developers.
 *
 * You may use and modify this package for any purpose. Redistribution is
 * permitted, in both source and binary form, provided that this notice
 * remains intact in all source distributions of this package.
 */
 
package net.sf.gridarta.textedit.textarea;
 
import javax.swing.event.DocumentEvent;
import javax.swing.event.DocumentEvent.ElementChange;
import javax.swing.text.BadLocationException;
import javax.swing.text.Element;
import javax.swing.text.PlainDocument;
import javax.swing.text.Segment;
import javax.swing.undo.UndoableEdit;
import net.sf.gridarta.textedit.textarea.tokenmarker.TokenMarker;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
 
public class SyntaxDocument extends PlainDocument {
 
    private static final long serialVersionUID = 1L;
 
    @Nullable
    public TokenMarker getTokenMarker() {
        return tokenMarker;
    }
 
    public void setTokenMarker(@Nullable final TokenMarker tokenMarker) {
        this.tokenMarker = tokenMarker;
        if (tokenMarker == null) {
            return;
        }
 
        tokenMarker.insertLines(0, getDefaultRootElement().getElementCount());
        tokenizeLines();
    }
 
    private void tokenizeLines() {
        tokenizeLines(0, getDefaultRootElement().getElementCount());
    }
 
    private void tokenizeLines(final int start, final int len) {
        if (tokenMarker == null) {
            return;
        }
 
        final Segment lineSegment = new Segment();
        final Element map = getDefaultRootElement();
 
        try {
            for (int i = start; i < start + len; i++) {
                final Element lineElement = map.getElement(i);
                final int lineStart = lineElement.getStartOffset();
                getText(lineStart, lineElement.getEndOffset() - lineStart - 1, lineSegment);
                tokenMarker.markTokens(lineSegment, i);
            }
        } catch (final BadLocationException bl) {
            bl.printStackTrace();
        }
    }
 
    public static void beginCompoundEdit() {
    }
 
    public static void endCompoundEdit() {
    }
 
    public static void addUndoableEdit(@NotNull final UndoableEdit edit) {
    }
 
    // protected members
 
    @Nullable
    private TokenMarker tokenMarker;
 
    @Override
    protected void fireInsertUpdate(@NotNull final DocumentEvent e) {
        if (tokenMarker != null) {
            final ElementChange ch = e.getChange(getDefaultRootElement());
            if (ch != null) {
                tokenMarker.insertLines(ch.getIndex() + 1, ch.getChildrenAdded().length - ch.getChildrenRemoved().length);
            }
        }
 
        super.fireInsertUpdate(e);
    }
 
    @Override
    protected void fireRemoveUpdate(@NotNull final DocumentEvent e) {
        if (tokenMarker != null) {
            final ElementChange ch = e.getChange(getDefaultRootElement());
            if (ch != null) {
                tokenMarker.deleteLines(ch.getIndex() + 1, ch.getChildrenRemoved().length - ch.getChildrenAdded().length);
            }
        }
 
        super.fireRemoveUpdate(e);
    }
 
}