import sys
import os
import random
sys.path.insert(1, os.path.abspath('../../'))
from python_tools.aoc_utils import *

def resolve(i, n, g):
    if i in n:
        return n[i]
    else:
        a = resolve(g[i][0], n, g)
        b = resolve(g[i][2], n, g)
        if g[i][1] == "AND":
            return a & b
        elif g[i][1] == "OR":
            return a | b
        elif g[i][1] == "XOR":
            return a ^ b


def part_one(n, g):
    p1 = 0
    for i in g:
        if "z" in i:
            p1 = p1 | (resolve(i, n, g) << int(i.replace("z", "")))
    return p1

def combine(n, s):
    num = 0
    for i in n:
        if s in i:
            num = num | n[i] << int(i.replace(s, ""))
    return num

def add2sus(sus, g, s):
    if s in g:
        sus[s] = []
        add2sus(sus, g, g[s][0])
        add2sus(sus, g, g[s][2])

def sus(z1, z2, g):
    sus_g = {}
    binz = list(bin(z1 ^ z2)[2:])
    binz.reverse()
    for c, i in enumerate(binz):
        if i == '1':
            add2sus(sus_g, g, f"z{c:02}")
    print(len(sus_g))
    print(len(g))


    pass

def gen_n(x,y):
    xbin = list(bin(x)[2:])
    ybin = list(bin(y)[2:])
    xbin.reverse()
    ybin.reverse()
    n={}
    for c, x in enumerate(xbin):
        n[f"x{c:02}"] = int(x)
    for c, y in enumerate(ybin):
        n[f"y{c:02}"] = int(y)
    return n

def part_two(n, g):
    x = random.getrandbits(56)
    y = random.getrandbits(56)
    n = gen_n(x, y)
    x = combine(n, "x")
    y = combine(n, "y")
    z = part_one(n.copy(), g.copy())
    print("y, ", bin(y))
    print("x, ", bin(x))
    print("z, ", bin(z))
    print("W, ", bin(x+y))
    print("^, ", bin(z^(x+y)))
    sus(z, x+y, g)
    pass

def main():
    contents = file2list("input.txt")
    n = {}
    g = {}
    for i in contents:
        if ": " in i:
            x = i.split(": ")
            n[x[0]] = int(x[1])
        elif "->" in i:
            x = i.split(" -> ")
            g[x[1]] = x[0].split(" ")

    print("p1: ", part_one(n.copy(), g.copy()))
    part_two(n.copy(), g.copy())

if __name__ == "__main__":
    main()