runtime/Swift/Sources/Antlr4/CommonTokenStream.swift

/// 
/// Copyright (c) 2012-2017 The ANTLR Project. All rights reserved.
/// Use of this file is governed by the BSD 3-clause license that
/// can be found in the LICENSE.txt file in the project root.
/// 



/// 
/// This class extends _org.antlr.v4.runtime.BufferedTokenStream_ with functionality to filter
/// token streams to tokens on a particular channel (tokens where
/// _org.antlr.v4.runtime.Token#getChannel_ returns a particular value).
/// 
/// 
/// This token stream provides access to all tokens by index or when calling
/// methods like _#getText_. The channel filtering is only used for code
/// accessing tokens via the lookahead methods _#LA_, _#LT_, and
/// _#LB_.
/// 
/// 
/// By default, tokens are placed on the default channel
/// (_org.antlr.v4.runtime.Token#DEFAULT_CHANNEL_), but may be reassigned by using the
/// `->channel(HIDDEN)` lexer command, or by using an embedded action to
/// call _org.antlr.v4.runtime.Lexer#setChannel_.
/// 
/// 
/// 
/// Note: lexer rules which use the `->skip` lexer command or call
/// _org.antlr.v4.runtime.Lexer#skip_ do not produce tokens at all, so input text matched by
/// such a rule will not be available as part of the token stream, regardless of
/// channel.
/// 

public class CommonTokenStream: BufferedTokenStream {
    /// 
    /// Specifies the channel to use for filtering tokens.
    /// 
    /// 
    /// The default value is _org.antlr.v4.runtime.Token#DEFAULT_CHANNEL_, which matches the
    /// default channel assigned to tokens created by the lexer.
    /// 
    internal var channel = CommonToken.DEFAULT_CHANNEL

    /// 
    /// Constructs a new _org.antlr.v4.runtime.CommonTokenStream_ using the specified token
    /// source and the default token channel (_org.antlr.v4.runtime.Token#DEFAULT_CHANNEL_).
    /// 
    /// - parameter tokenSource: The token source.
    /// 
    public override init(_ tokenSource: TokenSource) {
        super.init(tokenSource)
    }

    /// 
    /// Constructs a new _org.antlr.v4.runtime.CommonTokenStream_ using the specified token
    /// source and filtering tokens to the specified channel. Only tokens whose
    /// _org.antlr.v4.runtime.Token#getChannel_ matches `channel` or have the
    /// _org.antlr.v4.runtime.Token#getType_ equal to _org.antlr.v4.runtime.Token#EOF_ will be returned by the
    /// token stream lookahead methods.
    /// 
    /// - parameter tokenSource: The token source.
    /// - parameter channel: The channel to use for filtering tokens.
    /// 
    public convenience init(_ tokenSource: TokenSource, _ channel: Int) {
        self.init(tokenSource)
        self.channel = channel
    }

    override
    internal func adjustSeekIndex(_ i: Int) throws -> Int {
        return try nextTokenOnChannel(i, channel)
    }

    override
    internal func LB(_ k: Int) throws -> Token? {
        if k == 0 || (p - k) < 0 {
            return nil
        }

        var i = p
        var n = 1
        // find k good tokens looking backwards
        while n <= k {
            // skip off-channel tokens
            try i = previousTokenOnChannel(i - 1, channel)
            n += 1
        }
        if i < 0 {
            return nil
        }
        return tokens[i]
    }

    override
    public func LT(_ k: Int) throws -> Token? {
        //System.out.println("enter LT("+k+")");
        try lazyInit()
        if k == 0 {
            return nil
        }
        if k < 0 {
            return try LB(-k)
        }
        var i = p
        var n = 1 // we know tokens[p] is a good one
        // find k good tokens
        while n < k {
            // skip off-channel tokens, but make sure to not look past EOF
            if try sync(i + 1) {
                i = try nextTokenOnChannel(i + 1, channel)
            }
            n += 1
        }
//		if ( i>range ) range = i;
        return tokens[i]
    }

    /// 
    /// Count EOF just once.
    /// 
    public func getNumberOfOnChannelTokens() throws -> Int {
        var n = 0
        try fill()
        for t in tokens {
            if t.getChannel() == channel {
                n += 1
            }
            if t.getType() == CommonToken.EOF {
                break
            }
        }
        return n
    }
}