diff --git a/src/main/java/Vertex.java b/src/main/java/Vertex.java index 35719b8..a96306d 100644 --- a/src/main/java/Vertex.java +++ b/src/main/java/Vertex.java @@ -1,3 +1,5 @@ +import org.jheaps.AddressableHeap; + import java.io.Serializable; public class Vertex implements Serializable, Comparable { @@ -5,6 +7,9 @@ public class Vertex implements Serializable, Comparable { private Integer vertexLabel; private Integer sequence; private Integer occupancy; + private Integer readCount; + private Double potential; + private AddressableHeap queue; public Vertex(Integer vertexLabel) { this.vertexLabel = vertexLabel; @@ -20,6 +25,14 @@ public class Vertex implements Serializable, Comparable { this.occupancy = occupancy; } + public Vertex(SequenceType type, Integer sequence, Integer occupancy, Integer vertexLabel, Integer readCount) { + this.type = type; + this.vertexLabel = vertexLabel; + this.sequence = sequence; + this.occupancy = occupancy; + this.readCount = readCount; + } + public SequenceType getType() { return type; @@ -94,4 +107,11 @@ public class Vertex implements Serializable, Comparable { return this.vertexLabel - other.getVertexLabel(); } + public Integer getReadCount() { + return readCount; + } + + public void setReadCount(Integer readCount) { + this.readCount = readCount; + } }