def color_diff():
global colors
for n in graphiti.get_node_ids():
s = graphiti.get_node_attribute(n, 'diffstatus')
graphiti.set_node_attribute(n, "graphiti:space:color", "vec3", colors[s])
for e in graphiti.get_edge_ids():
s = graphiti.get_edge_attribute(e, 'diffstatus')
try:
graphiti.set_edge_attribute(e, "graphiti:space:color", "vec3", colors[s])
except KeyError:
print("edge: {}".format(e))
print("diffstatus: {}".format(s))
sys.exit(-1)
def attribute_to_lod(attribute, fun = lambda x : x):
min = float("inf")
max = float("-inf")
min_value = None
max_value = None
n_values = dict()
for nid in graphiti.get_node_ids():
value = graphiti.get_node_attribute(nid, attribute)
if value is None:
graphiti.set_node_attribute(nid, "graphiti:space:lod", "float", "-1000")
continue
fun_value = fun(value)
if fun_value is None:
graphiti.set_node_attribute(nid, "graphiti:space:lod", "float", "-1000")
continue
fvalue = float(fun_value)
if fvalue > max:
max = fvalue
max_value = value
if fvalue < min:
min = fvalue
min_value = value
n_values[nid] = fvalue
e_values = dict()
for eid in graphiti.get_link_ids():
value = graphiti.get_link_attribute(eid, attribute)
if value is None:
graphiti.set_link_attribute(eid, "graphiti:space:lod", "float", "-1000")
continue
fun_value = fun(value)
if fun_value is None:
graphiti.set_link_attribute(eid, "graphiti:space:lod", "float", "-1000")
continue
fvalue = float(fun_value)
if fvalue > max:
max = fvalue
max_value = value
if fvalue < min:
min = fvalue
min_value = value
e_values[eid] = fvalue
for nid in n_values.keys():
svalue = (n_values[nid] - min) / (max - min)
print(attribute + " : " + str(n_values[nid]) + " --> lod : " + str(svalue))
graphiti.set_node_attribute(nid, "graphiti:space:lod", "float", str(svalue))
for eid in e_values.keys():
svalue = (e_values[eid] - min) / (max - min)
print(attribute + " : " + str(e_values[eid]) + " --> lod : " + str(svalue))
graphiti.set_link_attribute(eid, "graphiti:space:lod", "float", str(svalue))
print("LOD range : " + str(min_value) + " to " + str(max_value))
def color_neighbors():
graphiti.set_attribute("graphiti:space:linkmode", "string", "node_color")
if graphiti.count_selected_nodes() == 0:
print("Please select a node !")
return
selected = graphiti.get_selected_node(0)
graph = std.load_nx_graph()
neighbors = graph.neighbors(selected)
graphiti.set_node_attribute(selected, "graphiti:space:color", "vec3", "0.0 1.0 1.0")
for node in neighbors:
graphiti.set_node_attribute(node, "graphiti:space:lod", "float", "1.0")
graphiti.set_node_attribute(node, "graphiti:space:color", "vec3", "0.0 1.0 1.0")
def on_started():
global cities
dico = {}
for city in cities:
nid = graphiti.add_node(city["name"])
graphiti.set_node_attribute(nid, "og:world:geolocation", "vec2", std.vec2_to_str(city["geolocation"]))
graphiti.set_node_attribute(nid, "og:world:color", "vec4", std.vec4_to_str(city["color"]))
graphiti.set_node_attribute(nid, "og:world:size", "float", str(city["size"]))
dico[city['name']] = nid
eid = graphiti.add_edge(dico['Paris'], dico['San Francisco'])
graphiti.set_edge_attribute(eid, "og:world:color", "vec4", "0.0 1.0 0.0 1.0")
def on_started():
global cities
dico = {}
for city in cities:
nid = graphiti.add_node(city["name"])
graphiti.set_node_attribute(nid, "og:world:geolocation", "vec2",
std.vec2_to_str(city["geolocation"]))
graphiti.set_node_attribute(nid, "og:world:color", "vec4",
std.vec4_to_str(city["color"]))
graphiti.set_node_attribute(nid, "og:world:size", "float",
str(city["size"]))
dico[city['name']] = nid
eid = graphiti.add_edge(dico['Paris'], dico['San Francisco'])
graphiti.set_edge_attribute(eid, "og:world:color", "vec4",
"0.0 1.0 0.0 1.0")
def on_started():
global ctx
if len(sys.argv) == 3:
if sys.argv[2].endswith(".json"):
std.load_json(sys.argv[2])
else:
print("Unrecognized format <'" + sys.argv[2] + "'> !")
print(ctx['g1'])
og.bind_entity(ctx['g1'])
nid = og.add_node("node1")
og.set_node_attribute(nid, "og:space:color", "vec4", "1.0 0.0 0.0 1.0")
print(ctx['g2'])
og.bind_entity(ctx['g2'])
nid = og.add_node("node2")
og.set_node_attribute(nid, "og:world:geolocation", "vec2", "48.8742, 2.3470")
og.set_node_attribute(nid, "og:world:color", "vec4", "1.0 1.0 0.0 1.0")
def clear_icons():
for n in graphiti.get_node_ids():
graphiti.set_node_attribute(n, "graphiti:space:icon", "string", "shapes/disk")
def clear_colors():
graphiti.set_attribute("graphiti:space:linkmode", "string", "node_color")
for n in graphiti.get_node_ids():
graphiti.set_node_attribute(n, "graphiti:space:color", "vec4", "1.0 1.0 1.0 1.0")
def attribute_to_icon(attribute, fun = lambda a : "shapes/disk"):
for nid in graphiti.get_node_ids():
value = graphiti.get_node_attribute(nid, attribute)
graphiti.set_node_attribute(nid, "graphiti:space:icon", "string", fun(value))
def randomize_node_activity():
for id in graphiti.get_node_ids():
if random.uniform(0.0, 1.0) > 0.10:
continue
activity = random.uniform(0.0, 5.0)
graphiti.set_node_attribute(id, "graphiti:space:activity", "float", str(activity))
def randomize_node_icons():
icons = glob.glob("./Resources/Countries/*.png")
print(icons)
for id in graphiti.get_node_ids():
icon = "countries/" + random.choice(icons).split("/")[-1][:2].lower()
graphiti.set_node_attribute(id, "graphiti:space:icon", "string", icon)
def load_json(json_filename):
nodes = {}
edges = {}
global node_attributes
global edge_attributes
print ("Loading \"" + json_filename + "\" ...")
import json
with open(json_filename, "r") as infile:
data = json.load(infile)
print("Loading meta information ...")
# TODO : Find a more generic way of doing this
if "meta" in data:
if "title" in data["meta"].keys():
if data["meta"]["title"] is not None:
graphiti.set_attribute("raindance:space:title", "string", data["meta"]["title"])
if "attributes" in data:
for key in data["attributes"].keys():
if key in reserved_attributes:
continue
att_info = get_attribute_info(data["attributes"][key])
if att_info is None:
print("Error: Couldn't parse key '" + key + "' with value '" + str(data["attributes"][key]) + "'!")
continue
graphiti.set_attribute(key, att_info[0], att_info[1])
print(". Loading nodes ...")
for n in data["nodes"]:
n = prepare_node(n)
label = ""
if "label" in n:
label = n["label"].encode("utf-8")
nid = graphiti.add_node(label)
nodes[n["id"]] = nid
for key in n.keys():
if key in reserved_attributes:
continue
att_info = get_attribute_info(n[key])
if att_info is None:
print("Error: Couldn't parse key '" + key + "' with value '" + str(n[key]) + "'!")
continue
graphiti.set_node_attribute(nid, key, att_info[0], att_info[1])
print(". Loading edges ...")
for e in data["edges"]:
e = prepare_edge(e)
if "src" in e:
eid = graphiti.add_edge(nodes[e["src"]], nodes[e["dst"]])
else:
eid = graphiti.add_edge(nodes[e['source']], nodes[e['target']])
edges[e["id"]] = eid
for key in e.keys():
if key in reserved_attributes:
continue
att_info = get_attribute_info(e[key])
if att_info is None:
print("Error: Couldn't parse key '" + key + "' with value '" + str(e[key]) + "'!")
continue
graphiti.set_edge_attribute(eid, key, att_info[0], att_info[1])
if "timeline" in data:
print(". Loading timeline ...")
for c in data["timeline"]:
# TODO : Get rid of this translation phase when possible.
if c[1].startswith("graph:"):
if c[1] in ["graph:remove_node", "graph:set_node_attribute"]:
c[2]["id"] = nodes[c[2]["id"]]
elif c[1] in ["graph:remove_edge", "graph:set_edge_attribute"]:
c[2]["id"] = edges[c[2]["id"]]
elif c[1] in ["graph:add_edge"]:
c[2]["src"] = nodes[c[2]["src"]]
c[2]["dst"] = nodes[c[2]["dst"]]
graphiti.send_command(c[0], c[1], c[2])
print("Done.")
def on_idle():
global queue
global nodes
global edges
global t
global job_id
mutex.acquire()
for element in queue:
if og.count_nodes() > 1000:
#og.register_script("#console", 'dashboard.shell.execute')
og.remove_job(job_id)
continue
#print(element)
# --- Space Document ---
name = element['name']
rr = element['rr']
asn = element['geoip']['rr_asn']
'''
if asn is not None:
asn = asn.split()[0][2:]
if asn in nodes:
t = time.time()
r = 0.5 + math.cos(t)
g = 0.5 + math.sin(t)
b = 0.5 + math.sin(t + math.pi / 2)
a = 0.5
og.set_node_attribute(nodes[asn], 'og:space:color', 'vec4', "{0} {1} {2} {3}".format(r, g, b, a))
'''
if name not in nodes:
nodes[name] = og.add_node(name)
og.set_node_attribute(nodes[name], 'type', 'string', 'name')
if rr not in nodes:
nodes[rr] = og.add_node(rr)
og.set_node_attribute(nodes[rr], 'type', 'string', 'ip')
edge_key = "{0} -> {1}".format(name, rr)
if edge_key not in edges:
edges[edge_key] = og.add_edge(nodes[name], nodes[rr])
# --- WorldMap + Globe Documents ---
try:
record = gi.record_by_addr(rr)
except:
continue
if record is None:
continue
msg = {
'function' : 'add',
'latitude' : record['latitude'],
'longitude' : record['longitude'],
'color.r' : random.uniform(0.3, 1.0),
'color.g' : 0.0,
'color.b' : 0.0,
'color.a' : 0.75,
'size' : 2.0
}
og.request(1, msg)
og.request(2, msg)
queue = list()
mutex.release()
def f(t, id, match):
if not match:
if t == "node":
graphiti.set_node_attribute(id, "graphiti:space:lod", "float", "0.0")
elif t == "edge":
graphiti.set_edge_attribute(id, "graphiti:space:lod", "float", "0.0")
def set_lod(value):
for id in graphiti.get_node_ids():
graphiti.set_node_attribute(id, "graphiti:space:lod", "float", str(value))
for id in graphiti.get_link_ids():
graphiti.set_link_attribute(id, "graphiti:space:lod", "float", str(value))
def f(t, id, match):
if not match:
if t == "node":
graphiti.set_node_attribute(id, "graphiti:space:lod", "float", "0.0")
elif t == "edge":
graphiti.set_link_attribute(id, "graphiti:space:lod", "float", "0.0")
def randomize_lod():
for id in graphiti.get_node_ids():
graphiti.set_node_attribute(id, "og:space:lod", "float", str(random.random()))
for id in graphiti.get_link_ids():
graphiti.set_link_attribute(id, "og:space:lod", "float", str(random.random()))
def randomize_node_size():
for id in graphiti.get_node_ids():
activity = random.uniform(0.0, 5.0)
graphiti.set_node_attribute(id, "graphiti:space:size", "float", str(activity))
def reset_icons():
for nid in graphiti.get_node_ids():
graphiti.set_node_attribute(nid, "graphiti:space:icon", "string", "shapes/disk")
