def generateGraph(self, netLogoWorld, fileNameOut, nodeSizeMultiplyer, coordMultiplyer, edgeWeightMultiplyer, edgeWeightIgnore, nodeNamePrefix):
self.netLogoWorld = netLogoWorld;
self.fileNameOut = fileNameOut;
print("Generating graph started ...");
print("nodeSizeMultiplyer=%f, coordMultiplyer=%f, edgeWeightMultiplyer=%f, edgeWeightIgnore=%s, , nodeNamePrefix='%s'" \
% (nodeSizeMultiplyer, coordMultiplyer, edgeWeightMultiplyer, edgeWeightIgnore, nodeNamePrefix))
print "Igraph version %s" % (igraph.__version__);
self.graph = igraph.Graph();
# populating graph nodes from turtles
self.graph.add_vertices(len(self.netLogoWorld.turtles));
turtle = Turtle();
i = 0;
for turtle in self.netLogoWorld.turtles:
#print("Turtle: who=%d, label=%s" %(turtle.who, turtle.label))
# We cannot use id:
# self.graph.vs[i]['id'] = turtle.who;
# it was necesarry to add name to be able to refer to names of edges when we are adding edges later
# that is only possible way, since vertex ids (turtles who) are not necessarily starting from 0, and igrah insist on 0 and non-sparce vertices ids
self.graph.vs[i]['name'] = str(turtle.who);
self.graph.vs[i]['size'] = turtle.size * nodeSizeMultiplyer;
rgbColor = NetLogoWorld.colorNetlotoToRgb(turtle.color);
self.graph.vs[i]['r'] = rgbColor[0];
self.graph.vs[i]['g'] = rgbColor[1];
self.graph.vs[i]['b'] = rgbColor[2];
self.graph.vs[i]['x'] = turtle.xcor * coordMultiplyer;
self.graph.vs[i]['y'] = turtle.ycor * coordMultiplyer;
if(turtle.label == None or turtle.label == ""):
self.graph.vs[i]['label'] = "%s%d" % (nodeNamePrefix, turtle.who);
else:
self.graph.vs[i]['label'] = turtle.label;
#self.graph.vs[i]['hophop'] = 'YESSS!!!';
# adding additional non-recognized columns
#print "keys:%s " %(turtle.additionalParams.keys());
print(turtle.additionalParams.keys())
for columnName in turtle.additionalParams.keys():
columnType = turtle.columnTypes[columnName];
print("columnName=%s, columnType=%s, value=%s" %(columnName, columnType, turtle.additionalParams[columnName]));
if(columnType == "string"):
self.graph.vs[i][columnName] = str(turtle.additionalParams[columnName]);
elif(columnType == "int"):
self.graph.vs[i][columnName] = int(turtle.additionalParams[columnName]);
elif(columnType == "float"):
self.graph.vs[i][columnName] = float(turtle.additionalParams[columnName]);
i =i+1;
# populating edges nodes from links
link = Link();
i = 0;
for link in self.netLogoWorld.links:
#print("link.end1 = %s, link.end2=%s" % (str(link.end1), str(link.end2)));
#print self.graph;
#print self.graph.get_edgelist();
# we cannot add by integers
# self.graph.add_edges([(link.end1, link.end2)]);
# because, that is recognized as igraph's vertex IDs, which do not need to match NetLogo turtle WHOs (if they do not start from 0)
# There fore we need to refer by vertex names, and to do that we need to provide .add_edges() with strings instead of integers
self.graph.add_edges([(str(link.end1), str(link.end2))]);
self.graph.es[i]['Edge Id'] = link.end1 * 1000 + link.end2;
if(link.label == None or link.label == ""):
self.graph.es[i]['Edge Label'] = self.graph.es[i]['label'] = "%d-%d" % (link.end1, link.end2);
else:
self.graph.es[i]['Edge Label'] = self.graph.es[i]['label'] = turtle.label;
if(not edgeWeightIgnore):
self.graph.es[i]['weight'] = link.thickness*edgeWeightMultiplyer;
print(link.additionalParams.keys())
#print link.additionalParams.keys();
for columnName in link.additionalParams.keys():
columnType = link.columnTypes[columnName];
print("columnName=%s, columnType=%s, value=%s" %(columnName, columnType, link.additionalParams[columnName]));
if(columnType == "string"):
self.graph.es[i][columnName] = str(link.additionalParams[columnName]);
elif(columnType == "int"):
self.graph.es[i][columnName] = int(link.additionalParams[columnName]);
elif(columnType == "float"):
self.graph.es[i][columnName] = float(link.additionalParams[columnName]);
i =i+1;
igraph.summary(self.graph);
self.graph.write_graphml(fileNameOut);
print("Generating graph finished ...");
