Fixed constraint evaluator #315

NilFoundation · Apr 17, 2024 · 4e415ea · 4e415ea
1 parent b8dc491
commit 4e415ea
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/include/nil/crypto3/zk/snark/arithmetization/plonk/constraint.hpp b/include/nil/crypto3/zk/snark/arithmetization/plonk/constraint.hpp
@@ -87,15 +87,16 @@ namespace nil {
                         math::expression_evaluator<VariableType> evaluator(
                             *this,
                             [&assignments, row_index](const VariableType &var) {
+                                std::size_t rows_amount = assignments.rows_amount();
                                 switch (var.type) {
                                     case VariableType::column_type::witness:
-                                        return assignments.witness(var.index)[row_index + var.rotation];
+                                        return assignments.witness(var.index)[(rows_amount + row_index + var.rotation) % rows_amount];
                                     case VariableType::column_type::public_input:
-                                        return assignments.public_input(var.index)[row_index + var.rotation];
+                                        return assignments.public_input(var.index)[(rows_amount + row_index + var.rotation) % rows_amount];
                                     case VariableType::column_type::constant:
-                                        return assignments.constant(var.index)[row_index + var.rotation];
+                                        return assignments.constant(var.index)[(rows_amount + row_index + var.rotation) % rows_amount];
                                     case VariableType::column_type::selector:
-                                        return assignments.selector(var.index)[row_index + var.rotation];
+                                        return assignments.selector(var.index)[(rows_amount + row_index + var.rotation) % rows_amount];
                                     default:
                                         BOOST_ASSERT_MSG(false, "Invalid column type");
                                         return VariableType::assignment_type::zero();