DocumentSnapshotTokenFactory.java

/*
 *  Copyright (c) 2012 Sam Harwell, Tunnel Vision Laboratories LLC
 *  All rights reserved.
 *
 *  The source code of this document is proprietary work, and is not licensed for
 *  distribution. For information about licensing, contact Sam Harwell at:
 *      sam@tunnelvisionlabs.com
 */
package org.antlr.works.editor.antlr4.classification;

import org.antlr.v4.runtime.CharStream;
import org.antlr.v4.runtime.TokenFactory;
import org.antlr.v4.runtime.TokenSource;
import org.antlr.v4.runtime.misc.Tuple;
import org.antlr.v4.runtime.misc.Tuple2;

/**
 *
 * @author Sam Harwell
 */
public class DocumentSnapshotTokenFactory implements TokenFactory {

    private final Tuple2<? extends TokenSource, CharStream> effectiveSource;

    public DocumentSnapshotTokenFactory(TokenSource effectiveSource) {
        this.effectiveSource = Tuple.create(effectiveSource, effectiveSource.getInputStream());
    }

    public DocumentSnapshotTokenFactory(Tuple2<? extends TokenSource, CharStream> effectiveSource) {
        this.effectiveSource = effectiveSource;
    }

    @Override
    public DocumentSnapshotToken create(Tuple2<? extends TokenSource, CharStream> source, int type, String text, int channel, int start, int stop, int line, int charPositionInLine) {
        if (effectiveSource != null) {
            source = effectiveSource;
        }

        DocumentSnapshotToken t = new DocumentSnapshotToken(source, type, channel, start, stop);
        t.setLine(line);
        t.setCharPositionInLine(charPositionInLine);
        if ( text!=null ) {
            t.setText(text);
        }
        return t;
    }

    @Override
    public DocumentSnapshotToken create(int type, String text) {
        return new DocumentSnapshotToken(type, text);
    }

}