Expose internal state to o1js

src/base/snark0.ml

@@ -1332,6 +1332,25 @@ module Run = struct
           Run_state.set_as_prover !state true ;
           res )
 
+    module Low_level = struct
+      type state = field Run_state.t
+
+      let state = state
+
+      let make_state ~num_inputs ~input ~next_auxiliary ~aux ?system
+          ?eval_constraints ~with_witness ?log_constraint () =
+        Runner.State.make ~num_inputs ~input:(pack_field_vec input)
+          ~next_auxiliary ~aux:(pack_field_vec aux) ?system ?eval_constraints
+          ~with_witness ?log_constraint ()
+
+      let push_active_counter () =
+        let counters = !active_counters in
+        active_counters := this_functor_id :: counters ;
+        counters
+
+      let reset_active_counter counters = active_counters := counters
+    end
+
     module Run_and_check_deferred (M : sig
       type _ t
 

src/base/snark_intf.ml

@@ -1398,6 +1398,34 @@ module type Run_basic = sig
 
   val run_and_check_exn : (unit -> (unit -> 'a) As_prover.t) -> 'a
 
+  module Low_level : sig
+    type state = field Run_state.t
+
+    val state : state ref
+
+    val make_state :
+         num_inputs:int
+      -> input:Field.Constant.Vector.t
+      -> next_auxiliary:int ref
+      -> aux:Field.Constant.Vector.t
+      -> ?system:R1CS_constraint_system.t
+      -> ?eval_constraints:bool
+      -> with_witness:bool
+      -> ?log_constraint:
+           (   ?at_label_boundary:[ `End | `Start ] * string
+            -> ( field Cvar.t
+               , field )
+               Checked_runner.Constraint0.basic_with_annotation
+               option
+            -> unit )
+      -> unit
+      -> state
+
+    val push_active_counter : unit -> int list
+
+    val reset_active_counter : int list -> unit
+  end
+
   module Run_and_check_deferred (M : sig
     type _ t