init commit

main.py

@@ -0,0 +1,86 @@
+import argparse
+from ctypes.wintypes import INT
+
+parser = argparse.ArgumentParser()
+parser.add_argument('-f', help='Writes visualization PNG to this path. Defaults to \"nets.png\".', default='nets.png', metavar='PATH')
+parser.add_argument('-s', help='Smallest visible net, defaults to /32 for IPv4 and /64 for IPv6', default=-1, metavar='CIDR', type=int)
+parser.add_argument(
+    '-n', '--networks', 
+    required=True,
+    help='File containing network definitions.')
+
+args = parser.parse_args()
+
+import re
+import ip
+import imggen
+
+def parseLine(line: str):
+    def splitC(c):
+        r = (c & 0xFF0000) >> 16
+        g = (c & 0x00FF00) >> 8
+        b = (c & 0x0000FF) >> 0
+        return (r, g, b)
+
+    if re.fullmatch("^color\ *=.*$", line) is not None:
+        cstr = line.split("=")[1].strip().lstrip("#x")
+        cv = int(cstr, 16)
+        return ("col", splitC(cv))
+
+    if re.fullmatch("^fill\ *=.*$", line) is not None:
+        cstr = line.split("=")[1].strip().lstrip("#x")
+        cv = int(cstr, 16)
+        return ("fil", splitC(cv))
+
+    if len(line) == 0 or line[0] == "#":
+        return ("com", line)
+
+    if re.fullmatch("^[\.:0-9a-f]*\/[0-9]*$", line) is not None:
+        n = ip.net()
+        n.parseNet(line)
+        return("net", n)
+
+    return ("nal", 0)
+
+with open(args.networks) as f:
+    snet = ip.net()
+    snet.parseNet(f.readline())
+
+    smallest = args.s
+    if smallest == -1:
+        if snet.type == 0:
+            smallest = 32
+        else:
+            smallest = 64
+
+    if snet.type == 0:
+        smallest = 32 - smallest
+    else:
+        smallest = 128 - smallest
+
+    if (snet.getSize() - smallest) % 2 != 0:
+        raise argparse.ArgumentError(message="The difference between the base net and the smallest net-size must be even!")
+
+    gen = imggen.imggen(snet.getSize() - smallest)
+
+    c = (255, 255, 255)
+
+    for l in f.readlines():
+        pl = parseLine(l.strip())
+        if pl[0] == "col":
+            c = pl[1]
+
+        if pl[0] == "fil":
+            gen.clear(pl[1])
+        
+        if pl[0] == "net":
+            s = pl[1].start.toInt(smallest) - snet.start.toInt(smallest)
+            e = pl[1].end.toInt(smallest) - snet.start.toInt(smallest)
+            gen.colorRange(s, e, c)
+    
+    s = int(1024 / gen.x)
+    if (s < 1):
+        s = 1
+
+    gen.saveImage(args.f, s)
+    gen.showImage(s)