Class StackRecord

common/ast_rewrite.cc:59–105

    absl::variant<ExprRecord, ArgRecord, ComprehensionRecord>;

struct StackRecord {
 public:
  static constexpr int kTarget = -2;

  explicit StackRecord(Expr* e) {
    ExprRecord record;
    record.expr = e;
    record_variant = record;
  }

  StackRecord(Expr* e, ComprehensionExpr* comprehension,
              Expr* comprehension_expr, ComprehensionArg comprehension_arg,
              bool use_comprehension_callbacks) {
    if (use_comprehension_callbacks) {
      ComprehensionRecord record;
      record.expr = e;
      record.comprehension = comprehension;
      record.comprehension_expr = comprehension_expr;
      record.comprehension_arg = comprehension_arg;
      record.use_comprehension_callbacks = use_comprehension_callbacks;
      record_variant = record;
      return;
    }
    ArgRecord record;
    record.expr = e;
    record.calling_expr = comprehension_expr;
    record.call_arg = comprehension_arg;
    record_variant = record;
  }

  StackRecord(Expr* e, const Expr* call, int argnum) {
    ArgRecord record;
    record.expr = e;
    record.calling_expr = call;
    record.call_arg = argnum;
    record_variant = record;
  }

  Expr* expr() const { return absl::get<ExprRecord>(record_variant).expr; }

  bool IsExprRecord() const {
    return absl::holds_alternative<ExprRecord>(record_variant);
  }

  StackRecordKind record_variant;
  bool visited = false;
};

struct PreVisitor {
  void operator()(const ExprRecord& record) {