run.py

#! /usr/bin/env python
#=============================================================================================
# Main program: exploring the possible worlds by giving the number of input constraints 
# Including "save_all_worlds" post-processing asp program 
# Input: The number of input constraints 
# Output: All possible worlds, lattice, worlds-assignment, summary view
# Usage: python run.py -n [-l] 
# Contacts: Shizhuo Yu
#=============================================================================================
from __future__ import division
from subprocess import call
import argparse
import os

def main(args):    
    if args.inputN is None:
        print 'You should specify the number of constraints by -n, use "python run.py -h" for help'
        return None
    
    # define the folders
    aspdir = os.path.join("", "asp")
    vizdir = os.path.join("", "viz")
    srcdir = os.path.join("", "src")
    
    
    
    # experate all pws
    numOfCons = args.inputN
    fIn = open(aspdir + "/expWorlds.asp","w")
    fIn.write("% Define the domain (universe of discourse): 1,2,...\n")
    fIn.write("u(1.."+numOfCons+").\n")
    fIn.write("% every element is either in or out\n")
    fIn.write("i(X) :- u(X), not o(X).\n")
    fIn.write("o(X) :- u(X), not i(X).\n")
    fIn.close()
    
    # Save all worlds
    call("dlv -silent -filter=i,o " + aspdir + "/expWorlds.asp > " + aspdir + "/expWorlds.asp.out", shell=True)
    fIn = open(aspdir + "/expWorlds.asp.out", "r")
    fOut = open(aspdir + "/expWorlds_aw.asp", "w")
    outList = []
    not_EOF = True
    line = fIn.readline()
    i= 0
    
    while not_EOF:
        preds = []
        preds = line[1:-2].split(", ")
        
        for pred in preds:
            try:
                index = pred.index("(")
                outStr = pred[:index+1] + str(i) + "," + pred[index+1:]
                outList.append(outStr)
            except ValueError:
                print "*WARNING* World", i, "is empty!"
            
        i = i + 1
        line = fIn.readline()    
        if len(line) == 0:
            not_EOF = False
    
    for e in outList:
        fOut.write(e+".\n")
    
    fIn.close()
    fOut.close()    
    
    call("dlv -silent -filter=r,g "+ aspdir + "/expWorlds_aw.asp " + aspdir + "/wexp-colors.asp" + " > " + aspdir + "/expWorlds_colors.asp", shell=True)
    #call("dlv -silent -filter=up "+ "expWorlds_aw.asp" + " wexp-up.asp" + " > " + "expWorlds_up.asp", shell=True)
    call("python " + srcdir + "/powerset.py " + numOfCons, shell=True)
    
    # generate all lattice
    edges = ""
    #fIn = open("expWorlds_up.asp")
    fIn = open(aspdir + "/up.dlv")
    line = fIn.readline()
    fIn.close()
    ups = line[1:-1].split(", ")
    for up in ups:
        edges += up.split(",")[0][3:] + "->" + up.split(",")[1][0:-1] + "\n"
    
    
    color = { "r":"#FF0000", "g":"#00FF00" } 
    
    infile = open(aspdir + "/expWorlds_colors.asp", "r")
    if args.showlat:
        fAll = open(vizdir + "/world_all.gv", "w")
        fAll.write('digraph{\nrankdir=BT\nnode[shape=circle,style=filled,label=""]\nedge[dir=none]\n')
    
    i = 0
    rcnt = []
    for line in infile:
        if args.showlat:
            outfile = open(vizdir + "/world_"+str(i)+".gv", "w")
            outfile.write('digraph{\nrankdir=BT\nnode[shape=circle,style=filled,label=""]\nedge[dir=none]\n')
        preds = line[1:-2].split(", ")
        if len(rcnt) == 0:
            for e in range(len(preds)):
                rcnt.append(0)
        for pred in preds:
            index = pred.index("(")
            if args.showlat:
                outfile.write(pred[index+1:-1] + '[color="' + color[pred[0]] + '"]\n')
            if pred[0] == "r":
                rcnt[preds.index(pred)] += 1
        
        if args.showlat:
            outfile.write(edges + "}")
            outfile.close()
            call("dot -Tpdf " + vizdir + "/world_"+str(i)+".gv" + " -o " + vizdir + "/world_"+str(i)+".pdf", shell=True)
        i = i + 1
    
    if args.showlat:
        for e in range(len(rcnt)):
            ratio = rcnt[e]/i*100
            if ratio > 50:
                fAll.write(str(e) + "[label=\"red "+ "%.2f"%(ratio) +"%\",color=red]\n")
            elif ratio == 50:
                fAll.write(str(e) + "[label=\""+ "%.2f"%(ratio) +"% of r/g\",color=yellow]\n")
            else:
                fAll.write(str(e) + "[label=\"green "+ "%.2f"%(100-ratio) +"%\",color=green]\n")
        fAll.write(edges + "}")
        fAll.close()
        call("dot -Tpdf " + vizdir + "/world_all.gv" + " -o " + vizdir + "/world_all.pdf", shell=True)

    infile.close()
    print "Number of constraints: ", numOfCons
    print "Number of possible worlds (Nodes in lattice): ", 2**int(numOfCons)
    print "Number of lattice (Dedekind number): ", i
    print ""

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("-n", dest="inputN", default=None, help="input: number of constraints")
    parser.add_argument("-l", action="store_true", dest="showlat", default=False, help="show actual lattices")
    args = parser.parse_args()
    main(args)