count_cdc.py

#!/usr/bin/python

import sys
import os
import json
from flower_snarks import *
from sage.all import Graph
from parmap import parmap

def process(g6):
  g6 = g6.rstrip()
  G = Graph(g6)
  ret = count_cdc_naive(G)
  return json.dumps({
    "graph": g6,
    "cdc_count": ret[0],
    "n": G.num_verts(),
    "expected_cdc_lowerbound": ret[1],
    "is_ok": (ret[0] >= ret[1])
  })

inp = [ l for l in sys.stdin ]

nprocs = int(os.getenv("NPROCS", "0")) or None
for ret in parmap(process, inp, nprocs=nprocs, in_order=False):
  print(ret)