From 04a19a402ae8426d81db23b6e7f8d375a116c93d Mon Sep 17 00:00:00 2001
From: yilin-bao <62961093+yilin-bao@users.noreply.github.com>
Date: Sat, 9 Dec 2023 00:49:58 -0600
Subject: [PATCH] Write method update module map

---
 analyzer.py | 43 +++++++++++++++++++++++++++++++++++++------
 1 file changed, 37 insertions(+), 6 deletions(-)

diff --git a/analyzer.py b/analyzer.py
index 9ee25d1..fff4654 100644
--- a/analyzer.py
+++ b/analyzer.py
@@ -1,4 +1,6 @@
 import ast, astor
+from ctypes import Array
+import time
 import hashlib
 from ast import Assign, Attribute, BinOp, Call, For, Name, Return, Subscript, Tuple, mod
 from gettext import find
@@ -70,7 +72,12 @@ class Color:
     END = '\033[0m'
     
 
+def generate_random_string(length=8):
+    characters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
+    return ''.join(random.choice(characters) for _ in range(length))
+
 def hash_code(code):
+    code = code + generate_random_string()
     sha256_hash = hashlib.sha256()
     sha256_hash.update(code.encode('utf-8'))
     hashed_code = sha256_hash.hexdigest()
@@ -408,7 +415,7 @@ def __init__(self, var_module_dict, var_name, module_name):
         self.var_module_dict:dict = var_module_dict
         self.var_name = var_name
         self.module_name = module_name
-        self.module_map = []
+        self.module_map = [] # ([Inputs], [Outputs], Function/Operation)
         
         # forward_tensor_list: tensor, matrix, vector usd in deep learning
         # forward_param_list: int, float, dimension variables, other variables
@@ -595,10 +602,13 @@ def special_case_length_two(self, parent, grandparent, this:Name):
         # self.out_dict = {}
         # self.hash_var_dict = {}
         if isinstance(grandparent, ast.Assign) and isinstance(parent, ast.Call):
-            print(self.var_name)
-            print(self.module_name)
-            print(self.remove_starting_self(self.find_full_name(parent.func)))
-            print(self.find_full_name_array(parent.args))
+            if grandparent.value == parent:
+                op = self.from_node_to_operation(parent.func)
+                # print(self.var_module_dict)
+                args = self.find_full_name_array(parent.args)
+                targets = self.find_full_name_array(grandparent.targets)
+                print(args, targets)
+                
             self.print_parents_and_code(this)
         elif isinstance(grandparent, ast.Call) and isinstance(parent, ast.Call):
             # self.print_parents_and_code(this)
@@ -612,7 +622,28 @@ def special_case_length_two(self, parent, grandparent, this:Name):
         else:
             # self.print_parents_and_code(this)
             pass
+        
+    def from_node_to_operation(self, node):
+        var_op = self.find_full_name(node)
+        var_op_noself = self.remove_starting_self(var_op)
+        if self.var_name == 'self':
+            return var_op_noself
+        else:
+            return f'{self.var_name}.{var_op_noself}'
+        
+    def update_module_name(self, op_in_array:Array, op_out_array:Array, op_name):
+        op_in_hash = []
+        for op_in in op_in_array:
+            old_hash = self.out_dict[op_in]
+            op_in_hash.append(old_hash)
+        op_out_hash = []
+        for op_out in op_out_array:
+            new_hash = hash_code(op_out)
+            self.out_dict[op_out] = new_hash
+            self.hash_var_dict[new_hash] = op_out
+            op_out_hash.append(new_hash)
+        self.module_map.append((op_in_hash, op_out_hash, op_name))
     
     def print_parents_and_code(self, this:Name):
         if len(self.parent_stack) >= 1:
-            print(f'{Color.BOLD_BLUE}{self.parent_stack}{Color.END} {this.id} {Color.LIME}{astor.to_source(self.parent_stack[0])}{Color.END}', end=' ')
+            print(f'{Color.BOLD_BLUE}{self.parent_stack}{Color.END} {this.id} {Color.LIME}{astor.to_source(self.parent_stack[0])}{Color.END}', end='')