o1js-pack 0.3.1 → 0.4.0

package/tests/provable/example_packed_string_circuit.ts

@@ -0,0 +1,52 @@
+import {
+  Experimental,
+  SelfProof,
+  Character,
+  Poseidon,
+  Provable,
+  Field,
+} from 'o1js';
+import { PackedStringFactory } from '../../src/lib/packed-types/PackedString';
+
+const MAX_LENGTH = 32;
+export class TextInput extends PackedStringFactory(MAX_LENGTH) {}
+
+export const TextInputProgram = Experimental.ZkProgram({
+  publicInput: TextInput,
+
+  methods: {
+    init: {
+      privateInputs: [],
+      method(state: TextInput) {
+        const initState = TextInput.fromString('Mina Protocol');
+        state.assertEquals(initState);
+      },
+    },
+    changeFirstLetter: {
+      privateInputs: [
+        SelfProof,
+        Provable.Array(Character, MAX_LENGTH),
+        Character,
+      ],
+      method(
+        newState: TextInput,
+        oldProof: SelfProof<TextInput, TextInput>,
+        oldCharacters: Array<Character>,
+        newInput: Character
+      ) {
+        oldProof.verify();
+        const newCharacters = [
+          ...oldCharacters.map((x) => new Character(x.value)),
+        ];
+        const packed = TextInput.fromCharacters(oldCharacters);
+        packed.assertEquals(oldProof.publicInput);
+        newCharacters[0] = newInput;
+        const repacked = TextInput.fromCharacters(newCharacters);
+        newState.assertEquals(repacked);
+      },
+    },
+  },
+});
+
+export let TextInputProof_ = Experimental.ZkProgram.Proof(TextInputProgram);
+export class TextInputProof extends TextInputProof_ {}

package/tests/provable/example_packed_uint_circuit.ts

@@ -0,0 +1,45 @@
+import { Experimental, SelfProof } from 'o1js';
+import { PackedUInt32Factory } from '../../src/lib/packed-types/PackedUInt32';
+
+export class Votes extends PackedUInt32Factory() {}
+
+export const VotesProgram = Experimental.ZkProgram({
+  publicInput: Votes,
+
+  methods: {
+    init: {
+      privateInputs: [],
+      method(state: Votes) {
+        const initState = Votes.fromBigInts([0n, 0n]);
+        state.assertEquals(initState);
+      },
+    },
+    incrementIndex0: {
+      privateInputs: [SelfProof],
+      method(newState: Votes, oldProof: SelfProof<Votes, Votes>) {
+        oldProof.verify();
+        const expectedUnpacked = Votes.unpack(oldProof.publicInput.packed);
+        oldProof.publicInput.packed.assertEquals(
+          Votes.fromUInt32s(expectedUnpacked).packed
+        );
+        expectedUnpacked[0] = expectedUnpacked[0].add(1);
+        newState.assertEquals(Votes.fromUInt32s(expectedUnpacked));
+      },
+    },
+    incrementIndex1: {
+      privateInputs: [SelfProof],
+      method(newState: Votes, oldProof: SelfProof<Votes, Votes>) {
+        oldProof.verify();
+        const expectedUnpacked = Votes.unpack(oldProof.publicInput.packed);
+        oldProof.publicInput.packed.assertEquals(
+          Votes.fromUInt32s(expectedUnpacked).packed
+        );
+        expectedUnpacked[1] = expectedUnpacked[1].add(1);
+        newState.assertEquals(Votes.fromUInt32s(expectedUnpacked));
+      },
+    },
+  },
+});
+
+export let VotesProof_ = Experimental.ZkProgram.Proof(VotesProgram);
+export class VotesProof extends VotesProof_ {}