123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303 |
- // Copyright (c) 2011 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #ifndef BASE_STRINGS_STRING_TOKENIZER_H_
- #define BASE_STRINGS_STRING_TOKENIZER_H_
- #include <algorithm>
- #include <string>
- #include "base/strings/string_piece.h"
- namespace base {
- // StringTokenizerT is a simple string tokenizer class. It works like an
- // iterator that with each step (see the Advance method) updates members that
- // refer to the next token in the input string. The user may optionally
- // configure the tokenizer to return delimiters.
- //
- // EXAMPLE 1:
- //
- // char input[] = "this is a test";
- // CStringTokenizer t(input, input + strlen(input), " ");
- // while (t.GetNext()) {
- // printf("%s\n", t.token().c_str());
- // }
- //
- // Output:
- //
- // this
- // is
- // a
- // test
- //
- //
- // EXAMPLE 2:
- //
- // std::string input = "no-cache=\"foo, bar\", private";
- // StringTokenizer t(input, ", ");
- // t.set_quote_chars("\"");
- // while (t.GetNext()) {
- // printf("%s\n", t.token().c_str());
- // }
- //
- // Output:
- //
- // no-cache="foo, bar"
- // private
- //
- //
- // EXAMPLE 3:
- //
- // bool next_is_option = false, next_is_value = false;
- // std::string input = "text/html; charset=UTF-8; foo=bar";
- // StringTokenizer t(input, "; =");
- // t.set_options(StringTokenizer::RETURN_DELIMS);
- // while (t.GetNext()) {
- // if (t.token_is_delim()) {
- // switch (*t.token_begin()) {
- // case ';':
- // next_is_option = true;
- // break;
- // case '=':
- // next_is_value = true;
- // break;
- // }
- // } else {
- // const char* label;
- // if (next_is_option) {
- // label = "option-name";
- // next_is_option = false;
- // } else if (next_is_value) {
- // label = "option-value";
- // next_is_value = false;
- // } else {
- // label = "mime-type";
- // }
- // printf("%s: %s\n", label, t.token().c_str());
- // }
- // }
- //
- //
- template <class str, class const_iterator>
- class StringTokenizerT {
- public:
- typedef typename str::value_type char_type;
- // Options that may be pass to set_options()
- enum {
- // Specifies the delimiters should be returned as tokens
- RETURN_DELIMS = 1 << 0,
- // Specifies that empty tokens should be returned. Treats the beginning and
- // ending of the string as implicit delimiters, though doesn't return them
- // as tokens if RETURN_DELIMS is also used.
- RETURN_EMPTY_TOKENS = 1 << 1,
- };
- // The string object must live longer than the tokenizer. In particular, this
- // should not be constructed with a temporary. The deleted rvalue constructor
- // blocks the most obvious instances of this (e.g. passing a string literal to
- // the constructor), but caution must still be exercised.
- StringTokenizerT(const str& string,
- const str& delims) {
- Init(string.begin(), string.end(), delims);
- }
- // Don't allow temporary strings to be used with string tokenizer, since
- // Init() would otherwise save iterators to a temporary string.
- StringTokenizerT(str&&, const str& delims) = delete;
- StringTokenizerT(const_iterator string_begin,
- const_iterator string_end,
- const str& delims) {
- Init(string_begin, string_end, delims);
- }
- // Set the options for this tokenizer. By default, this is 0.
- void set_options(int options) { options_ = options; }
- // Set the characters to regard as quotes. By default, this is empty. When
- // a quote char is encountered, the tokenizer will switch into a mode where
- // it ignores delimiters that it finds. It switches out of this mode once it
- // finds another instance of the quote char. If a backslash is encountered
- // within a quoted string, then the next character is skipped.
- void set_quote_chars(const str& quotes) { quotes_ = quotes; }
- // Call this method to advance the tokenizer to the next delimiter. This
- // returns false if the tokenizer is complete. This method must be called
- // before calling any of the token* methods.
- bool GetNext() {
- if (quotes_.empty() && options_ == 0)
- return QuickGetNext();
- else
- return FullGetNext();
- }
- // Start iterating through tokens from the beginning of the string.
- void Reset() {
- token_end_ = start_pos_;
- }
- // Returns true if token is a delimiter. When the tokenizer is constructed
- // with the RETURN_DELIMS option, this method can be used to check if the
- // returned token is actually a delimiter. Returns true before the first
- // time GetNext() has been called, and after GetNext() returns false.
- bool token_is_delim() const { return token_is_delim_; }
- // If GetNext() returned true, then these methods may be used to read the
- // value of the token.
- const_iterator token_begin() const { return token_begin_; }
- const_iterator token_end() const { return token_end_; }
- str token() const { return str(token_begin_, token_end_); }
- BasicStringPiece<str> token_piece() const {
- return BasicStringPiece<str>(&*token_begin_,
- std::distance(token_begin_, token_end_));
- }
- private:
- void Init(const_iterator string_begin,
- const_iterator string_end,
- const str& delims) {
- start_pos_ = string_begin;
- token_begin_ = string_begin;
- token_end_ = string_begin;
- end_ = string_end;
- delims_ = delims;
- options_ = 0;
- token_is_delim_ = true;
- }
- // Implementation of GetNext() for when we have no quote characters. We have
- // two separate implementations because AdvanceOne() is a hot spot in large
- // text files with large tokens.
- bool QuickGetNext() {
- token_is_delim_ = false;
- for (;;) {
- token_begin_ = token_end_;
- if (token_end_ == end_) {
- token_is_delim_ = true;
- return false;
- }
- ++token_end_;
- if (delims_.find(*token_begin_) == str::npos)
- break;
- // else skip over delimiter.
- }
- while (token_end_ != end_ && delims_.find(*token_end_) == str::npos)
- ++token_end_;
- return true;
- }
- // Implementation of GetNext() for when we have to take quotes into account.
- bool FullGetNext() {
- AdvanceState state;
- for (;;) {
- if (token_is_delim_) {
- // Last token was a delimiter. Note: This is also the case at the start.
- //
- // ... D T T T T D ...
- // ^ ^
- // | |
- // | |token_end_| : The next character to look at or |end_|.
- // |
- // |token_begin_| : Points to delimiter or |token_end_|.
- //
- // The next token is always a non-delimiting token. It could be empty,
- // however.
- token_is_delim_ = false;
- token_begin_ = token_end_;
- // Slurp all non-delimiter characters into the token.
- while (token_end_ != end_ && AdvanceOne(&state, *token_end_)) {
- ++token_end_;
- }
- // If it's non-empty, or empty tokens were requested, return the token.
- if (token_begin_ != token_end_ || (options_ & RETURN_EMPTY_TOKENS))
- return true;
- }
- DCHECK(!token_is_delim_);
- // Last token was a regular token.
- //
- // ... T T T D T T ...
- // ^ ^
- // | |
- // | token_end_ : The next character to look at. Always one
- // | char beyond the token boundary.
- // |
- // token_begin_ : Points to beginning of token. Note: token could
- // be empty, in which case
- // token_begin_ == token_end_.
- //
- // The next token is always a delimiter. It could be |end_| however, but
- // |end_| is also an implicit delimiter.
- token_is_delim_ = true;
- token_begin_ = token_end_;
- if (token_end_ == end_)
- return false;
- // Look at the delimiter.
- ++token_end_;
- if (options_ & RETURN_DELIMS)
- return true;
- }
- return false;
- }
- bool IsDelim(char_type c) const {
- return delims_.find(c) != str::npos;
- }
- bool IsQuote(char_type c) const {
- return quotes_.find(c) != str::npos;
- }
- struct AdvanceState {
- bool in_quote;
- bool in_escape;
- char_type quote_char;
- AdvanceState() : in_quote(false), in_escape(false), quote_char('\0') {}
- };
- // Returns true if a delimiter was not hit.
- bool AdvanceOne(AdvanceState* state, char_type c) {
- if (state->in_quote) {
- if (state->in_escape) {
- state->in_escape = false;
- } else if (c == '\\') {
- state->in_escape = true;
- } else if (c == state->quote_char) {
- state->in_quote = false;
- }
- } else {
- if (IsDelim(c))
- return false;
- state->in_quote = IsQuote(state->quote_char = c);
- }
- return true;
- }
- const_iterator start_pos_;
- const_iterator token_begin_;
- const_iterator token_end_;
- const_iterator end_;
- str delims_;
- str quotes_;
- int options_;
- bool token_is_delim_;
- };
- typedef StringTokenizerT<std::string, std::string::const_iterator>
- StringTokenizer;
- typedef StringTokenizerT<string16, string16::const_iterator> String16Tokenizer;
- typedef StringTokenizerT<std::string, const char*> CStringTokenizer;
- } // namespace base
- #endif // BASE_STRINGS_STRING_TOKENIZER_H_
|