From a739a20de6399d58453820cb1bd6dd34627a02bd Mon Sep 17 00:00:00 2001
From: jernkun <jernjern356@gmail.com>
Date: Thu, 15 Jun 2023 15:12:00 +0700
Subject: [PATCH] register --> state transition of m4 DONE

---
 src/explore_gen_circom.js |  4 +-
 src/gen_m4.js             | 96 +++++++++++++++++++++++++++++++++++++++
 src/gen_tagged_dfa.js     |  2 -
 3 files changed, 99 insertions(+), 3 deletions(-)

diff --git a/src/explore_gen_circom.js b/src/explore_gen_circom.js
index f36bca1..8f54032 100644
--- a/src/explore_gen_circom.js
+++ b/src/explore_gen_circom.js
@@ -8,7 +8,7 @@ import { reverseDFA } from "./gen_rev_dfa";
 import { simplifyGraph } from "./gen_dfa";
 import { M1ToM2 } from "./gen_m2";
 import { M2ToM3 } from "./gen_m3";
-import { createM4 } from "./gen_m4";
+import { createM4, registerToState } from "./gen_m4";
 
 export function explore_gen_circom(regex, submatches) {
   const tagged_simp_graph = tagged_simplifyGraph(regex, submatches);
@@ -19,6 +19,8 @@ export function explore_gen_circom(regex, submatches) {
   console.log("m3 jya: ", m3_graph);
   let m4_graph = createM4(tagged_simp_graph);
   console.log("m4 jyaa: ", m4_graph);
+  let tag_m4 = registerToState(m4_graph);
+  console.log("final m4: ", tag_m4);
   // console.log("b4 everything: ", findMatchStateTagged(tagged_simp_graph));
   // const forw_graph = formatForCircom(findMatchStateTagged(tagged_simp_graph));
   // const rev_graph = formatForCircom(reverseDFA(simplifyGraph(regex)));
diff --git a/src/gen_m4.js b/src/gen_m4.js
index b3eaad1..b9aee89 100644
--- a/src/gen_m4.js
+++ b/src/gen_m4.js
@@ -140,3 +140,99 @@ export function createM4(m1_graph) {
     transitions: m4_transitions,
   };
 }
+// transform register into state transition for revealing
+// + modify transition to not have associated tag.
+export function registerToState(m4_graph) {
+  let tranGraph = m4_graph["transitions"];
+  let final_transitions = {};
+  let allTags = {};
+  let visited_tran = new Set();
+  let num_outward = {};
+  let track_outward = {};
+  for (const key in tranGraph) {
+    // num_outward represents possible end states that can reach from current node
+    let inner_key_set = new Set();
+    for (const inner_key in tranGraph[key]) {
+      inner_key_set.add(tranGraph[key][inner_key][0]);
+    }
+    num_outward[key] = inner_key_set.size;
+    track_outward[key] = 0;
+  }
+  let stack = [];
+  stack.push({ node_id: m4_graph["start_state"], memTags: {}, boolTags: {} });
+  while (stack.length > 0) {
+    let { node_id, memTags, boolTags } = stack.pop();
+    // if we exhaust all transitions from that node, store related tags into allTags
+    if (track_outward[node_id] == num_outward[node_id]) {
+      for (const key in memTags) {
+        if (!allTags.hasOwnProperty(key)) {
+          allTags[key] = new Set();
+        }
+        for (const strTran of memTags[key]) {
+          allTags[key].add(strTran);
+        }
+      }
+      continue;
+    }
+    // if there's transition from that node, we haven't explored yet
+    // Note that we consider visitted transition only from (from, to) without taking care of
+    // alphabet that leads that transition cuz we dont selectively reveal just a certain alphabet.
+    for (const key in tranGraph[node_id]) {
+      if (
+        visited_tran.has(JSON.stringify([node_id, tranGraph[node_id][key][0]]))
+      ) {
+        continue;
+      }
+      // if not add this visit into visited_tran
+      visited_tran.add(JSON.stringify([node_id, tranGraph[node_id][key][0]]));
+      track_outward[node_id] += 1;
+      let cl_memTags = {};
+      for (const tmp_key in memTags) {
+        cl_memTags[tmp_key] = new Set(memTags[tmp_key]);
+      }
+      let cl_boolTags = Object.assign({}, boolTags);
+      // store relevant transitions to its corresponding tags
+      for (const boolTag in cl_boolTags) {
+        if (cl_boolTags[boolTag]) {
+          if (!cl_memTags.hasOwnProperty(boolTag)) {
+            cl_memTags[boolTag] = new Set();
+          }
+          cl_memTags[boolTag].add(
+            JSON.stringify([node_id, tranGraph[node_id][key][0]])
+          );
+        }
+      }
+      // modify boolean states
+      let tag_arr = tranGraph[node_id][key][1];
+      if (tag_arr.length > 0) {
+        for (const subtag of tag_arr) {
+          if (subtag[0] == "E") {
+            cl_boolTags[subtag[1]] = false;
+          } else {
+            cl_boolTags[subtag[1]] = true;
+          }
+        }
+      }
+      stack.push({
+        node_id: tranGraph[node_id][key][0],
+        memTags: cl_memTags,
+        boolTags: cl_boolTags,
+      });
+    }
+  }
+
+  for (const firstkey in tranGraph) {
+    final_transitions[firstkey] = {};
+    for (const secondkey in tranGraph[firstkey]) {
+      final_transitions[firstkey][secondkey] =
+        tranGraph[firstkey][secondkey][0];
+    }
+  }
+  return {
+    states: m4_graph["states"],
+    start_state: m4_graph["start_state"],
+    accepted_states: m4_graph["accepted_states"],
+    transitions: final_transitions,
+    allTags: allTags,
+  };
+}
diff --git a/src/gen_tagged_dfa.js b/src/gen_tagged_dfa.js
index 85dc9fb..739cd18 100644
--- a/src/gen_tagged_dfa.js
+++ b/src/gen_tagged_dfa.js
@@ -244,8 +244,6 @@ export function findMatchStateTagged(tagged_dfa) {
       inner_key_set.add(tranGraph[key][inner_key]);
     }
     num_outward[key] = inner_key_set.size;
-    Object.keys(tranGraph[key]).length;
-
     track_outward[key] = 0;
   }