Straight Links in Layered Diagraph Layout

Hi,

I’m trying to achieve below UI using Layered Diagraph layout. I’m having issues in getting straight links.


But when I tried the actual result looks somewhat like below

Please suggest.