efischer/BiGpairSEQ
1
0
Fork 0
Code Issues Pull Requests Projects Releases 13 Wiki Activity

More efficient graph creation

Browse Source
This commit is contained in:
eugenefischer
2025-04-10 14:06:11 -05:00
parent 678ce99424
commit 187401f2d6
1 changed files with 26 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/main/java/Simulator.java
View File

@@ -149,22 +149,37 @@ public class Simulator implements GraphModificationFunctions {
betaVertices.forEach(graph::addVertex);
//add edges
for(Vertex a: alphaVertices) {
for(Integer well: a.getRecord().getWells()) {
for(Vertex b: betaVertices) {
double weight = 0.0;
for (Integer well: a.getRecord().getWells()) {
if (b.getRecord().isInWell(well)) {
DefaultWeightedEdge edge = graph.getEdge(a, b);
if (edge == null) {
edge = graph.addEdge(a, b);
graph.setEdgeWeight(edge, 1.0);
}
else {
double weight = graph.getEdgeWeight(edge);
graph.setEdgeWeight(edge, weight + 1.0);
}
weight += 1.0;
}
}
if (weight != 0.0) {
DefaultWeightedEdge edge = graph.addEdge(a, b);
graph.setEdgeWeight(edge, weight);
}
}
}
// //add edges
// for(Vertex a: alphaVertices) {
// for(Integer well: a.getRecord().getWells()) {
// for (Vertex b: betaVertices) {
// if (b.getRecord().isInWell(well)) {
// DefaultWeightedEdge edge = graph.getEdge(a, b);
// if (edge == null) {
// edge = graph.addEdge(a, b);
// graph.setEdgeWeight(edge, 1.0);
// }
// else {
// double weight = graph.getEdgeWeight(edge);
// graph.setEdgeWeight(edge, weight + 1.0);
// }
// }
// }
// }
// }
if(verbose){System.out.println("Graph created");}
//stop timing
Instant stop = Instant.now();