From 51d9d5b94c1d635b63482ffdd5eaaa4481e62e9b Mon Sep 17 00:00:00 2001
From: Andrew Reynolds <andrew.j.reynolds@gmail.com>
Date: Wed, 26 Feb 2020 12:33:00 -0600
Subject: Minor improvement to ParseOp (#3808)

---
 src/parser/parse_op.h | 29 ++++++++++++++++++++++++-----
 1 file changed, 24 insertions(+), 5 deletions(-)

(limited to 'src/parser')

diff --git a/src/parser/parse_op.h b/src/parser/parse_op.h
index 32d583010..a224b2511 100644
--- a/src/parser/parse_op.h
+++ b/src/parser/parse_op.h
@@ -19,6 +19,7 @@
 
 #include <string>
 
+#include "api/cvc4cpp.h"
 #include "expr/expr.h"
 #include "expr/kind.h"
 
@@ -35,6 +36,7 @@ namespace CVC4 {
  * (2) A string name.
  * (3) An expression expr.
  * (4) A type t.
+ * (5) An operator object.
  *
  * Examples:
  *
@@ -66,26 +68,43 @@ struct CVC4_PUBLIC ParseOp
   Expr d_expr;
   /** The type associated with the parsed operator, if it exists */
   Type d_type;
+  /** The operator associated with the parsed operator, if it exists */
+  api::Op d_op;
 
   /* Return true if this is equal to 'p'. */
   bool operator==(const ParseOp& p) const
   {
     return d_kind == p.d_kind && d_name == p.d_name && d_expr == p.d_expr
-           && d_type == p.d_type;
+           && d_type == p.d_type && d_op == p.d_op;
   }
 };
 
 inline std::ostream& operator<<(std::ostream& os, const ParseOp& p)
 {
+  std::stringstream out;
+  out << "(ParseOp ";
   if (!p.d_expr.isNull())
   {
-    return os << p.d_expr;
+    out << " :expr " << p.d_expr;
   }
-  else if (p.d_kind != kind::NULL_EXPR)
+  if (!p.d_op.isNull())
   {
-    return os << p.d_kind;
+    out << " :op " << p.d_op;
   }
-  return os << "ParseOp::unknown";
+  if (p.d_kind != kind::NULL_EXPR)
+  {
+    out << " :kind " << p.d_kind;
+  }
+  if (!p.d_type.isNull())
+  {
+    out << " :type " << p.d_type;
+  }
+  if (!p.d_name.empty())
+  {
+    out << " :name " << p.d_name;
+  }
+  out << ")";
+  return os << out.str();
 }
 
 }  // namespace CVC4
-- 
cgit v1.2.3