From 43251286c0541e323f91a5aea9ddcc0d3a984fea Mon Sep 17 00:00:00 2001
From: Jack Grigg <jack@electriccoin.co>
Date: Fri, 6 Aug 2021 01:23:27 +0100
Subject: [PATCH] Enable Pasta curve elements to be used with `group::Wnaf`

---
 src/curves.rs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/curves.rs b/src/curves.rs
index e784902..c9d954f 100644
--- a/src/curves.rs
+++ b/src/curves.rs
@@ -96,6 +96,25 @@ macro_rules! new_curve_impl {
             }
         }
 
+        impl group::WnafGroup for $name {
+            fn recommended_wnaf_for_num_scalars(num_scalars: usize) -> usize {
+                // Copied from bls12_381::g1, should be updated.
+                const RECOMMENDATIONS: [usize; 12] =
+                    [1, 3, 7, 20, 43, 120, 273, 563, 1630, 3128, 7933, 62569];
+
+                let mut ret = 4;
+                for r in &RECOMMENDATIONS {
+                    if num_scalars > *r {
+                        ret += 1;
+                    } else {
+                        break;
+                    }
+                }
+
+                ret
+            }
+        }
+
         impl CurveExt for $name {
             type ScalarExt = $scalar;
             type Base = $base;