SyntaxDocument.java

Go to the documentation of this file.

/*
 * SyntaxDocument.java - Document that can be tokenized
 * Copyright (C) 1999 Slava Pestov
 * Copyright (C) 2000-2015 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.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();

        final int end = len + start;

        try {
            for (int i = start; i < end; 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 DocumentEvent.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 DocumentEvent.ElementChange ch = e.getChange(getDefaultRootElement());
            if (ch != null) {
                tokenMarker.deleteLines(ch.getIndex() + 1, ch.getChildrenRemoved().length - ch.getChildrenAdded().length);
            }
        }

        super.fireRemoveUpdate(e);
    }

}