summaryrefslogtreecommitdiff
path: root/subcapset.py
blob: fe8868f29923bdaeed62b8ad58d9310190bea2dd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#!/usr/bin/env python3
"""Detects and removes sub-capsets.

[00 01 10] is a subcapset of [00 01 10 22], and thus would be filtered."""

from sys import argv

def main():
    if len(argv) == 2:
        infile = argv[1]
    else:
        print("Usage: %s INFILE" % argv[0])
        exit(1)
    capsets = []
    capset_file = open(infile, "r")
    for line in capset_file:
        capsets.append(tuple(line[1:-2].split(" ")))
    capset_file.close()

    capsetSet = set(capsets)
    for capset in capsets:
        for i in range(len(capset)):
            capsetSet.discard(capset[:i] + capset[i+1:])
    for capset in capsets:
        if capset in capsetSet:
            print("[" + " ".join(capset) + "]")

if __name__ == "__main__":
    main()