Add support for ValidIf to ProtoBuf [de]serialization

2 files changed, 9 insertions, 0 deletions

diff --git a/src/main/scala/firrtl/proto/FromProto.scala b/src/main/scala/firrtl/proto/FromProto.scala
index ef2ee5bd..ea4cec1f 100644
--- a/src/main/scala/firrtl/proto/FromProto.scala
+++ b/src/main/scala/firrtl/proto/FromProto.scala
@@ -94,6 +94,9 @@ object FromProto {
   def convert(mux: Firrtl.Expression.Mux): ir.Mux =
     ir.Mux(convert(mux.getCondition), convert(mux.getTValue), convert(mux.getFValue), ir.UnknownType)
 
+  def convert(validif: Firrtl.Expression.ValidIf): ir.ValidIf =
+    ir.ValidIf(convert(validif.getCondition), convert(validif.getValue), ir.UnknownType)
+
   def convert(expr: Firrtl.Expression): ir.Expression = {
     import Firrtl.Expression._
     expr.getExpressionCase.getNumber match {
@@ -106,6 +109,7 @@ object FromProto {
       case FIXED_LITERAL_FIELD_NUMBER => convert(expr.getFixedLiteral)
       case PRIM_OP_FIELD_NUMBER => convert(expr.getPrimOp)
       case MUX_FIELD_NUMBER => convert(expr.getMux)
+      case VALID_IF_FIELD_NUMBER => convert(expr.getValidIf)
     }
   }
 
diff --git a/src/main/scala/firrtl/proto/ToProto.scala b/src/main/scala/firrtl/proto/ToProto.scala
index c246656f..e2481d08 100644
--- a/src/main/scala/firrtl/proto/ToProto.scala
+++ b/src/main/scala/firrtl/proto/ToProto.scala
@@ -188,6 +188,11 @@ object ToProto {
           .setTValue(convert(tval))
           .setFValue(convert(fval))
         eb.setMux(mb)
+      case ir.ValidIf(cond, value, _) =>
+        val vb = Firrtl.Expression.ValidIf.newBuilder()
+          .setCondition(convert(cond))
+          .setValue(convert(value))
+        eb.setValidIf(vb)
     }
   }