return snapshots.map(snapshot => {
if (snapshot.id !== id) {
return snapshot;
}
assert(snapshot.dominatorTree, "Should have a dominator tree model");
assert(snapshot.dominatorTree.state == dominatorTreeState.FETCHING,
"Should be in the FETCHING state");
const dominatorTree = immutableUpdate(snapshot.dominatorTree, {
state: dominatorTreeState.LOADED,
root,
expanded: Immutable.Set(),
});
return immutableUpdate(snapshot, { dominatorTree });
});
handlers[actions.TAKE_CENSUS_DIFF_END] = function(diffing, action) {
assert(diffing, "Should be diffing when ending a census diff");
assert(action.first.id === diffing.firstSnapshotId,
"First snapshot's id should match");
assert(action.second.id === diffing.secondSnapshotId,
"Second snapshot's id should match");
return immutableUpdate(diffing, {
state: diffingState.TOOK_DIFF,
census: {
report: action.report,
parentMap: action.parentMap,
expanded: Immutable.Set(),
inverted: action.inverted,
filter: action.filter,
display: action.display,
},
});
};
handlers[actions.TAKE_CENSUS_END] = function (snapshots, { id,
report,
parentMap,
display,
filter }) {
const census = {
report,
parentMap,
expanded: Immutable.Set(),
display,
filter,
state: censusState.SAVED
};
return snapshots.map(snapshot => {
return snapshot.id === id
? immutableUpdate(snapshot, { census })
: snapshot;
});
};
return snapshots.map(snapshot => {
if (snapshot.id !== id) {
return snapshot;
}
assert(snapshot.dominatorTree, "Should have a dominator tree model");
assert(snapshot.dominatorTree.state == dominatorTreeState.FETCHING,
"Should be in the FETCHING state");
let focused;
if (snapshot.dominatorTree.focused) {
focused = (function findFocused(node) {
if (node.nodeId === snapshot.dominatorTree.focused.nodeId) {
return node;
}
if (node.children) {
const length = node.children.length;
for (let i = 0; i < length; i++) {
const result = findFocused(node.children[i]);
if (result) {
return result;
}
}
}
return undefined;
}(root));
}
const dominatorTree = immutableUpdate(snapshot.dominatorTree, {
state: dominatorTreeState.LOADED,
root,
expanded: Immutable.Set(),
focused,
});
return immutableUpdate(snapshot, { dominatorTree });
});