diff --git a/zksnark/code_to_r1cs.py b/zksnark/code_to_r1cs.py index d85a61d8..14e233a7 100644 --- a/zksnark/code_to_r1cs.py +++ b/zksnark/code_to_r1cs.py @@ -144,7 +144,7 @@ def insert_var(arr, varz, var, used, reverse=False): # Maps input, output and intermediate variables to indices def get_var_placement(inputs, flatcode): return ['~one'] + [x for x in inputs] + ['~out'] + [c[1] for c in flatcode if c[1] not in inputs and c[1] != '~out'] - + # Convert the flattened code generated above into a rank-1 constraint system def flatcode_to_r1cs(inputs, flatcode): @@ -206,19 +206,19 @@ def assign_variables(inputs, input_vars, flatcode): elif x[0] == '/': assignment[varz.index(x[1])] = grab_var(varz, assignment, x[2]) / grab_var(varz, assignment, x[3]) return assignment - + def code_to_r1cs_with_inputs(code, input_vars): inputs, body = extract_inputs_and_body(parse(code)) - print 'Inputs' - print inputs - print 'Body' - print body + print('Inputs') + print(inputs) + print('Body') + print(body) flatcode = flatten_body(body) - print 'Flatcode' - print flatcode - print 'Input var assignment' - print get_var_placement(inputs, flatcode) + print('Flatcode') + print(flatcode) + print('Input var assignment') + print(get_var_placement(inputs, flatcode)) A, B, C = flatcode_to_r1cs(inputs, flatcode) r = assign_variables(inputs, input_vars, flatcode) return r, A, B, C @@ -228,11 +228,14 @@ def qeval(x): y = x**3 return y + x + 5 """, [3]) -print 'r' -print r -print 'A' -for x in A: print x -print 'B' -for x in B: print x -print 'C' -for x in C: print x +print('r') +print(r) +print('A') +for x in A: + print(x) +print('B') +for x in B: + print(x) +print('C') +for x in C: + print(x)