main: Add --draw-partition flag to graph command

This calculates and refines the partition like `refine` and draws
boxes around nodes in the same block.