From 7a98d5ec0849f0d13235a8f09dadf736c9409aa2 Mon Sep 17 00:00:00 2001
From: Kamen Mladenov <kamen@syndamia.com>
Date: Tue, 4 Feb 2025 13:24:36 +0200
Subject: feat(zkvms/zkwasm): Proper Vec input type handling

Also handles multidimentional vec
---
 zkvms/zkwasm/guest/src/lib.rs | 14 ++------------
 1 file changed, 2 insertions(+), 12 deletions(-)

(limited to 'zkvms/zkwasm/guest')

diff --git a/zkvms/zkwasm/guest/src/lib.rs b/zkvms/zkwasm/guest/src/lib.rs
index eba34b0..c92b97c 100644
--- a/zkvms/zkwasm/guest/src/lib.rs
+++ b/zkvms/zkwasm/guest/src/lib.rs
@@ -22,22 +22,12 @@ fn write(value: u64) {
 static VERTICES: u64 = 10;
 
 macro_rules! read {
-    // HACK for graph_coloring
-    (Vec , u32 $readfn:tt) => {
-        {
-            let mut ret = Vec::new();
-            for _ in 0..2 {
-                ret.push(read!(u32 $readfn));
-            }
-            ret
-        }
-    };
     // Vec<Vec<...<Vec<primitive>>>> is converted by entrypoint_expr! to
     // Vec,Vec,...,Vec,primitive
-    (Vec , $($type:tt)*) => {
+    (Vec $size:literal , $($type:tt)*) => {
         {
             let mut ret = Vec::new();
-            for _ in 0..VERTICES {
+            for _ in 0..$size {
                 ret.push(read!($($type)*));
             }
             ret
-- 
cgit v1.2.3