diff --git a/src/core/credGraph.js b/src/core/credGraph.js index fc4db723f..34d87a7b1 100644 --- a/src/core/credGraph.js +++ b/src/core/credGraph.js @@ -67,6 +67,12 @@ export class CredGraph { return srcCred * edge.transitionProbability; } + node(addr: NodeAddressT): ?Node { + const node = this._mpg._nodes.get(addr); + if (node == null) return undefined; + return {...node, cred: this._cred(addr)}; + } + *nodes(): Iterator { for (const node of this._mpg.nodes()) { yield {...node, cred: this._cred(node.address)}; @@ -79,6 +85,12 @@ export class CredGraph { } } + *scoringNodes(): Iterator { + for (const addr of this._mpg.scoringAddresses()) { + yield NullUtil.get(this.node(addr)); + } + } + *inNeighbors(addr: NodeAddressT): Iterator { for (const edge of this._mpg.inNeighbors(addr)) { yield {...edge, credFlow: this._credFlow(edge)};