Method ComprehensionExprToProto

common/ast/expr_proto.cc:223–252

  }

  absl::Status ComprehensionExprToProto(
      const Expr& expr, const ComprehensionExpr& comprehension_expr,
      ExprProto* absl_nonnull proto) {
    proto->Clear();
    auto* comprehension_proto = proto->mutable_comprehension_expr();
    proto->set_id(expr.id());
    comprehension_proto->set_iter_var(comprehension_expr.iter_var());
    comprehension_proto->set_iter_var2(comprehension_expr.iter_var2());
    if (comprehension_expr.has_iter_range()) {
      Push(comprehension_expr.iter_range(),
           comprehension_proto->mutable_iter_range());
    }
    comprehension_proto->set_accu_var(comprehension_expr.accu_var());
    if (comprehension_expr.has_accu_init()) {
      Push(comprehension_expr.accu_init(),
           comprehension_proto->mutable_accu_init());
    }
    if (comprehension_expr.has_loop_condition()) {
      Push(comprehension_expr.loop_condition(),
           comprehension_proto->mutable_loop_condition());
    }
    if (comprehension_expr.has_loop_step()) {
      Push(comprehension_expr.loop_step(),
           comprehension_proto->mutable_loop_step());
    }
    if (comprehension_expr.has_result()) {
      Push(comprehension_expr.result(), comprehension_proto->mutable_result());
    }
    return absl::OkStatus();
  }

  void Push(const Expr& expr, ExprProto* absl_nonnull proto) {
    frames_.push(Frame{&expr, proto});