From 1a9fa9899b257437d818f1c16a3b3aeb2f0421a7 Mon Sep 17 00:00:00 2001
From: Jack Grigg <jack@electriccoin.co>
Date: Fri, 29 Jan 2021 00:45:48 +0000
Subject: [PATCH] impl<F: FieldExt> {Neg, Sub} for Expression<F>

---
 src/plonk/circuit.rs | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/plonk/circuit.rs b/src/plonk/circuit.rs
index d74277701c..67e750cdf7 100644
--- a/src/plonk/circuit.rs
+++ b/src/plonk/circuit.rs
@@ -1,10 +1,13 @@
 use core::cmp::max;
 use core::ops::{Add, Mul};
 use ff::Field;
-use std::convert::TryFrom;
+use std::{
+    convert::TryFrom,
+    ops::{Neg, Sub},
+};
 
 use super::{lookup, permutation, Error};
-use crate::poly::Rotation;
+use crate::{arithmetic::FieldExt, poly::Rotation};
 
 /// A column type
 pub trait ColumnType: 'static + Sized {}
@@ -316,6 +319,13 @@ impl<F: Field> Expression<F> {
     }
 }
 
+impl<F: FieldExt> Neg for Expression<F> {
+    type Output = Expression<F>;
+    fn neg(self) -> Self::Output {
+        Expression::Scaled(Box::new(self), -F::one())
+    }
+}
+
 impl<F> Add for Expression<F> {
     type Output = Expression<F>;
     fn add(self, rhs: Expression<F>) -> Expression<F> {
@@ -323,6 +333,13 @@ impl<F> Add for Expression<F> {
     }
 }
 
+impl<F: FieldExt> Sub for Expression<F> {
+    type Output = Expression<F>;
+    fn sub(self, rhs: Expression<F>) -> Expression<F> {
+        Expression::Sum(Box::new(self), Box::new(-rhs))
+    }
+}
+
 impl<F> Mul for Expression<F> {
     type Output = Expression<F>;
     fn mul(self, rhs: Expression<F>) -> Expression<F> {