From bacc78add4bc840a14086a2d19b8a9b22567e15e Mon Sep 17 00:00:00 2001
From: Chetany Bhardwaj <chetanyiot@gmail.com>
Date: Fri, 19 Dec 2025 02:59:38 +0530
Subject: [PATCH] feat: add WIDTH constant to Poseidon2 types for compile-time
 validation

---
 src/poseidon2/poseidon2.zig | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/poseidon2/poseidon2.zig b/src/poseidon2/poseidon2.zig
index e0f3f41..db81384 100644
--- a/src/poseidon2/poseidon2.zig
+++ b/src/poseidon2/poseidon2.zig
@@ -570,6 +570,7 @@ pub fn poseidon2_24_plonky3_with_mds_light(state: []F, apply_mds_light: bool) vo
 
 // Wrapper structs for compatibility
 pub const Poseidon2KoalaBear16Plonky3 = struct {
+    pub const WIDTH = 16;
     pub const Field = F;
 
     pub fn permutation(state: []F) void {
@@ -620,6 +621,7 @@ pub const Poseidon2KoalaBear16Plonky3 = struct {
 };
 
 pub const Poseidon2KoalaBear24Plonky3 = struct {
+    pub const WIDTH = 24;
     pub const Field = F;
 
     pub fn permutation(state: []F) void {