/**
* Produces the layer assignmment using the graph information specified
*/
public void execute(Object parent)
{
mxGraphHierarchyModel model = layout.getModel();
final Set<mxGraphHierarchyNode> seenNodes = new HashSet<mxGraphHierarchyNode>();
final Set<mxGraphHierarchyNode> unseenNodes = new HashSet<mxGraphHierarchyNode>(
model.getVertexMapper().values());
// Perform a dfs through the internal model. If a cycle is found,
// reverse it.
mxGraphHierarchyNode[] rootsArray = null;
if (model.roots != null)
{
Object[] modelRoots = model.roots.toArray();
rootsArray = new mxGraphHierarchyNode[modelRoots.length];
for (int i = 0; i < modelRoots.length; i++)
{
Object node = modelRoots[i];
mxGraphHierarchyNode internalNode = model
.getVertexMapper().get(node);
rootsArray[i] = internalNode;
}
}
model.visit(new mxGraphHierarchyModel.CellVisitor()
{
public void visit(mxGraphHierarchyNode parent,
mxGraphHierarchyNode cell,
mxGraphHierarchyEdge connectingEdge, int layer, int seen)
{
// Check if the cell is in it's own ancestor list, if so
// invert the connecting edge and reverse the target/source
// relationship to that edge in the parent and the cell
if ((cell)
.isAncestor(parent))
{
connectingEdge.invert();
parent.connectsAsSource.remove(connectingEdge);
parent.connectsAsTarget.add(connectingEdge);
cell.connectsAsTarget.remove(connectingEdge);
cell.connectsAsSource.add(connectingEdge);
}
seenNodes.add(cell);
unseenNodes.remove(cell);
}
}, rootsArray, true, null);
Set<Object> possibleNewRoots = null;
if (unseenNodes.size() > 0)
{
possibleNewRoots = new HashSet<Object>(unseenNodes);
}
// If there are any nodes that should be nodes that the dfs can miss
// these need to be processed with the dfs and the roots assigned
// correctly to form a correct internal model
Set<mxGraphHierarchyNode> seenNodesCopy = new HashSet<mxGraphHierarchyNode>(
seenNodes);
// Pick a random cell and dfs from it
mxGraphHierarchyNode[] unseenNodesArray = new mxGraphHierarchyNode[1];
unseenNodes.toArray(unseenNodesArray);
model.visit(new mxGraphHierarchyModel.CellVisitor()
{
public void visit(mxGraphHierarchyNode parent,
mxGraphHierarchyNode cell,
mxGraphHierarchyEdge connectingEdge, int layer, int seen)
{