comparison fca/fcaDotGen.py @ 0:62d2c72e4223

initial commit

+import rdflib, os
+from rdflib import plugin, OWL, URIRef, RDF, RDFS, Literal
+from rdflib.graph import Graph
+from rdflib.namespace import Namespace
+
+plugin.register(
+    'sparql', rdflib.query.Processor,
+    'rdfextras.sparql.processor', 'Processor')
+plugin.register(
+    'sparql', rdflib.query.Result,
+    'rdfextras.sparql.query', 'SPARQLQueryResult')
+
+afuri = "http://sovarr.c4dm.eecs.qmul.ac.uk/features/"
+cataloguePath = '/Users/alo/MusicOntology/features/af-catalogue.rdf'
+
+graph = Graph()
+graph.parse(cataloguePath)
+
+execfile('/Users/alo/MusicOntology/features/fca/writeHTML.py')
+
+def getFeaturesByTool( name ):
+    qry = 'SELECT DISTINCT ?feature WHERE { ?x af:computedIn "' + name + '" . ?x af:feature ?feature} ORDER BY ?feature'
+    
+    return graph.query(qry,
+        initNs=dict(
+            af=Namespace(afuri))
+    )
+
+def getBaseFeatures():
+    features = []
+    ns = URIRef(u'http://sovarr.c4dm.eecs.qmul.ac.uk/features/computedIn')
+    for su in graph.subjects(RDF.type, OWL.Class):
+        count = sum(1 for _ in graph.objects(su,ns))
+        if count > 1:
+            features.append(su.split('/')[-1])
+            
+    return features.sort()
+    
+def getTools():
+    return graph.query(
+        """SELECT DISTINCT ?tool
+           WHERE {
+    	   	?x local:computedIn ?tool .
+    		?x local:feature ?feature 
+           }
+    	   ORDER BY ?tool""",
+        initNs=dict(
+            af=Namespace(afuri))
+    )
+
+def constructMatrix( name, path ):
+    rows = []
+    columns = []
+    matrix = []
+
+    cat = ["appdomain", "complexity", "domain", "level", "temporalscale", "dimensions"]
+    catdict = {}
+
+    for nm in cat:
+        catdict[nm] = []
+
+    for ns, value in graph.namespaces():
+        if ns == 'local':
+            local = value
+
+    for it in getFeaturesByTool(name):
+        rows.append(it[0])
+        for su, pr, ob in graph.triples((URIRef(local+it[0]), None, None)):
+            col = pr.split('/')[-1]
+            if cat.count(col) != 0:
+                if catdict[col].count(ob) == 0:
+                    columns.append(col+"-"+ob)
+                    catdict[col].append(ob)
+
+    rows.sort()
+    columns.sort()
+    
+    for i in range(len(rows)):
+        row = []
+        for j in range(len(columns)):
+            row.append(0)
+        matrix.append(row)
+
+    index = 0
+    for feature in rows:
+        for s, p, o in graph.triples((URIRef(local+feature), None, None)):
+            col = p.split('/')[-1]
+            if cat.count(col) != 0:
+                matrix[index][columns.index(col+"-"+o)] = 1
+        
+        index += 1
+    
+    writeCXT(path+name+'.cxt', matrix, columns, rows)
+    print ("wrote cxt data to "+path+name+".cxt")
+    name = name.replace(' ', '\ ')
+    os.system("fcastone -bc "+path+name+".cxt "+path+name+".dot")
+    
+def constructBaseMatrix( name, path ):
+    rows = []
+    columns = []
+    matrix = []
+
+    cat = ["domain", "level", "temporalscale", "dimensions", "output", "tag"]
+    catdict = {}
+
+    for nm in cat:
+        catdict[nm] = []
+
+    for ns, value in graph.namespaces():
+        if ns == 'af':
+            local = value
+
+    features = []
+    ns = URIRef(u'http://sovarr.c4dm.eecs.qmul.ac.uk/features/computedIn')
+    for su in graph.subjects(RDF.type, OWL.Class):
+        count = sum(1 for _ in graph.objects(su,ns))
+        if count > 1:
+            features.append(su.split('/')[-1])
+
+    for it in features:
+        rows.append(it)
+        for su, pr, ob in graph.triples((URIRef(local+it), None, None)):
+            col = pr.split('/')[-1]
+            if cat.count(col) != 0:
+                if catdict[col].count(ob) == 0:
+                    columns.append(ob.replace(' ', '-')+"-"+col.replace(' ', '-'))
+                    catdict[col].append(ob)
+
+    rows.sort()
+    columns.sort()
+    
+    for i in range(len(rows)):
+        row = []
+        for j in range(len(columns)):
+            row.append(0)
+        matrix.append(row)
+
+    index = 0
+    for feature in rows:
+        for s, p, o in graph.triples((URIRef(local+feature), None, None)):
+            col = p.split('/')[-1]
+            if cat.count(col) != 0:
+                matrix[index][columns.index(o.replace(' ', '-')+"-"+col.replace(' ', '-'))] = 1
+        
+        index += 1
+    
+    writeCXT(path+name+'.cxt', matrix, columns, rows)
+    print ("wrote cxt data to "+path+name+".cxt")
+    name = name.replace(' ', '\ ')
+    os.system("fcastone -bc "+path+name+".cxt "+path+name+".dot")