src/ast/ast-literal-reindexer.h

// Copyright 2015 the V8 project 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 V8_AST_AST_LITERAL_REINDEXER
#define V8_AST_AST_LITERAL_REINDEXER

#include "src/ast/ast.h"
#include "src/ast/scopes.h"

namespace v8 {
namespace internal {

class AstLiteralReindexer final : public AstVisitor<AstLiteralReindexer> {
 public:
  AstLiteralReindexer() : next_index_(0) {}

  int count() const { return next_index_; }
  void Reindex(Expression* pattern);

 private:
#define DEFINE_VISIT(type) void Visit##type(type* node);
  AST_NODE_LIST(DEFINE_VISIT)
#undef DEFINE_VISIT

  void VisitStatements(ZoneList<Statement*>* statements);
  void VisitDeclarations(ZoneList<Declaration*>* declarations);
  void VisitArguments(ZoneList<Expression*>* arguments);
  void VisitLiteralProperty(LiteralProperty* property);

  void UpdateIndex(MaterializedLiteral* literal) {
    literal->literal_index_ = next_index_++;
  }

  int next_index_;

  DEFINE_AST_VISITOR_MEMBERS_WITHOUT_STACKOVERFLOW()
  DISALLOW_COPY_AND_ASSIGN(AstLiteralReindexer);
};
}  // namespace internal
}  // namespace v8

#endif  // V8_AST_AST_LITERAL_REINDEXER