edit-cosine-circle-packing.py

#!/usr/bin/env python2.7
#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#

import json
import argparse
import csv
import sys


def createCluster(inputCSV,argNum):
    row=[]
    if(argNum not in [0,1,2]):
        print "Input Error! \nPass argument to --cluster as one of the following\n0 for clustering based on x-coordinate, \n1 for clustering based on y-coordinate, \n2 for clustering based on similarity score"
        sys.exit()

    csvPath = inputCSV          #Input Path to csv file
    with open(csvPath,"r") as f:
        lines = csv.reader(f.read().splitlines(), delimiter=' ')
        for line in lines:
            row.append(line)

    data={}
    for i in range(len(row)):
        if "x-coordinate" in row[i][0].split(","):
            continue
        else:
            column = row[i][0].split(",")
            data[column[argNum]]=[]         #Cluster based on the argument number passed

    for i in range(len(row)):
        if "x-coordinate" in row[i][0].split(","):
            continue
        else:
            column = row[i][0].split(",")
            second={}
            second["name"]=column[1]+"  "+column[2]
            second["size"]=column[2]
            data[column[argNum]].append(second)          #Cluster based on the argument number passed
            
    clusterList = []
    i=0
    for elem in data.keys():
        first={}
        first["name"]="cluster "+str(i)
        first["children"]=data[elem]
        clusterList.append(first)
        i+=1


    print json.dumps(clusterList, sort_keys=True, indent=4, separators=(',', ': '))

    clusterStruct = {"name":"clusters", "children":clusterList}
    with open("circle.json", "w") as f:             #Pass the json file as input to circle-packing.html
        f.write(json.dumps(clusterStruct, sort_keys=True, indent=4, separators=(',', ': ')))

if __name__ == '__main__':
    argParser = argparse.ArgumentParser('Edit/Cosine Similarity Circle Packing')
    argParser.add_argument('--inputCSV', required=True, help='path to directory containing CSV File, containing pair-wise similarity scores based on edit/cosine similarity distance')
    argParser.add_argument('--cluster', required=True, type=int, help='Cluster based on the x-coordinate/y-coordinate/similarity score. Pass 0 for clustering based on x-coordinate, 1 for clustering based on y-coordinate, 2 for clustering based on similarity score')
    args = argParser.parse_args()

    createCluster(args.inputCSV, args.cluster)