gcc/rust/expand/rust-proc-macro-invoc-lexer.cc

   1 // Copyright (C) 2020-2024 Free Software Foundation, Inc.
   2
   3 // This file is part of GCC.
   4
   5 // GCC is free software; you can redistribute it and/or modify it under
   6 // the terms of the GNU General Public License as published by the Free
   7 // Software Foundation; either version 3, or (at your option) any later
   8 // version.
   9
  10 // GCC is distributed in the hope that it will be useful, but WITHOUT ANY
  11 // WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12 // FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  13 // for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with GCC; see the file COPYING3.  If not see
  17 // <http://www.gnu.org/licenses/>.
  18
  19 #include "rust-proc-macro-invoc-lexer.h"
  20 #include "rust-token.h"
  21
  22 namespace Rust {
  23
  24 const_TokenPtr
  25 ProcMacroInvocLexer::peek_token (int n)
  26 {
  27   if ((offs + n) >= token_stream.size ())
  28     return Token::make (END_OF_FILE, UNDEF_LOCATION);
  29
  30   return token_stream.at (offs + n);
  31 }
  32
  33 void
  34 ProcMacroInvocLexer::split_current_token (TokenId new_left, TokenId new_right)
  35 {
  36   auto &current_token = token_stream.at (offs);
  37   auto current_pos = token_stream.begin () + offs;
  38
  39   auto l_tok = Token::make (new_left, current_token->get_locus ());
  40   auto r_tok = Token::make (new_right, current_token->get_locus ());
  41
  42   token_stream.erase (current_pos);
  43
  44   // `insert` inserts before the specified position, so we insert the right one
  45   // then the left
  46   token_stream.insert (current_pos, l_tok);
  47   token_stream.insert (current_pos, r_tok);
  48 }
  49
  50 void
  51 ProcMacroInvocLexer::split_current_token (std::vector<TokenPtr> new_tokens)
  52 {
  53   rust_assert (new_tokens.size () > 0);
  54
  55   auto current_pos = token_stream.begin () + offs;
  56
  57   token_stream.erase (current_pos);
  58
  59   for (size_t i = 1; i < new_tokens.size (); i++)
  60     {
  61       token_stream.insert (current_pos + i, new_tokens[i]);
  62     }
  63 }
  64
  65 } // namespace Rust