Skip to content
Snippets Groups Projects
Commit 11961adb authored by AndiMajore's avatar AndiMajore
Browse files

tried to simplify read_graph_tool_graph

Former-commit-id: 4b09a973
parent 01ede27d
Branches
No related tags found
No related merge requests found
......@@ -91,13 +91,20 @@ def read_graph_tool_graph(file_path, seeds, max_deg, include_indirect_drugs=Fals
print(int(drug))
for edge in g.edges():
if g.edge_properties["type"][edge] == 'drug-protein':
if g.vertex_properties["type"][edge.target()] == d_type and edge.target() not in direct_drugs:
if g.vertex_properties["type"][edge.target()] == d_type:
indir_drug = edge.target() not in direct_drugs
not_seed = edge.source() not in seed_ids
if indir_drug or not_seed:
deleted_edges.append(edge)
if int(edge.target()) in drug_ids:
if indir_drug and int(edge.target()) in drug_ids:
drug_ids.remove(int(edge.target()))
elif g.vertex_properties["type"][edge.source()] == d_type and edge.source() not in direct_drugs:
elif g.vertex_properties["type"][edge.source()] == d_type and edge.source() not in direct_drugs or edge.target() not in seed_ids:
indir_drug = edge.source() not in direct_drugs
not_seed = edge.target() not in seed_ids
if indir_drug or not_seed:
deleted_edges.append(edge)
if int(edge.source()) in drug_ids:
if indir_drug and int(edge.source()) in drug_ids:
drug_ids.remove(int(edge.source()))
g.set_fast_edge_removal(fast=True)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment