summaryrefslogtreecommitdiff
path: root/vendor/github.com/antlr4-go/antlr/v4/common_token_factory.go
blob: 1bb0314ea09c4cc0f7ea1197740d795c796a1b9d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Copyright (c) 2012-2022 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.

package antlr

// TokenFactory creates CommonToken objects.
type TokenFactory interface {
	Create(source *TokenSourceCharStreamPair, ttype int, text string, channel, start, stop, line, column int) Token
}

// CommonTokenFactory is the default TokenFactory implementation.
type CommonTokenFactory struct {
	// copyText indicates whether CommonToken.setText should be called after
	// constructing tokens to explicitly set the text. This is useful for cases
	// where the input stream might not be able to provide arbitrary substrings of
	// text from the input after the lexer creates a token (e.g. the
	// implementation of CharStream.GetText in UnbufferedCharStream panics an
	// UnsupportedOperationException). Explicitly setting the token text allows
	// Token.GetText to be called at any time regardless of the input stream
	// implementation.
	//
	// The default value is false to avoid the performance and memory overhead of
	// copying text for every token unless explicitly requested.
	copyText bool
}

func NewCommonTokenFactory(copyText bool) *CommonTokenFactory {
	return &CommonTokenFactory{copyText: copyText}
}

// CommonTokenFactoryDEFAULT is the default CommonTokenFactory. It does not
// explicitly copy token text when constructing tokens.
var CommonTokenFactoryDEFAULT = NewCommonTokenFactory(false)

func (c *CommonTokenFactory) Create(source *TokenSourceCharStreamPair, ttype int, text string, channel, start, stop, line, column int) Token {
	t := NewCommonToken(source, ttype, channel, start, stop)

	t.line = line
	t.column = column

	if text != "" {
		t.SetText(text)
	} else if c.copyText && source.charStream != nil {
		t.SetText(source.charStream.GetTextFromInterval(NewInterval(start, stop)))
	}

	return t
}

func (c *CommonTokenFactory) createThin(ttype int, text string) Token {
	t := NewCommonToken(nil, ttype, TokenDefaultChannel, -1, -1)
	t.SetText(text)

	return t
}