From 343e48302eb0871f66847d73f559b6e729548c2d Mon Sep 17 00:00:00 2001
From: Knut Sveidqvist <knsv@sveido.com>
Date: Wed, 15 Feb 2023 15:12:52 +0100
Subject: [PATCH] Fix for direction issue with elk rendering

---
 .../mermaid/src/diagrams/flowchart/elk/flowRenderer-elk.js     | 2 ++
 packages/mermaid/src/diagrams/flowchart/flowDb.js              | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/packages/mermaid/src/diagrams/flowchart/elk/flowRenderer-elk.js b/packages/mermaid/src/diagrams/flowchart/elk/flowRenderer-elk.js
index 3dba2f22b..73b37e9f7 100644
--- a/packages/mermaid/src/diagrams/flowchart/elk/flowRenderer-elk.js
+++ b/packages/mermaid/src/diagrams/flowchart/elk/flowRenderer-elk.js
@@ -313,6 +313,7 @@ const getNextPosition = (position, edgeDirection, graphDirection) => {
       },
     },
   };
+  portPos.TD = portPos.TB;
   log.info('abc88', graphDirection, edgeDirection, position);
   return portPos[graphDirection][edgeDirection][position];
   // return 'south';
@@ -323,6 +324,7 @@ const getNextPort = (node, edgeDirection, graphDirection) => {
   if (!portPos[node]) {
     switch (graphDirection) {
       case 'TB':
+      case 'TD':
         portPos[node] = {
           inPosition: 'north',
           outPosition: 'south',
diff --git a/packages/mermaid/src/diagrams/flowchart/flowDb.js b/packages/mermaid/src/diagrams/flowchart/flowDb.js
index 147ad5b6a..2e6b840b5 100644
--- a/packages/mermaid/src/diagrams/flowchart/flowDb.js
+++ b/packages/mermaid/src/diagrams/flowchart/flowDb.js
@@ -238,6 +238,9 @@ export const setDirection = function (dir) {
   if (direction.match(/.*v/)) {
     direction = 'TB';
   }
+  if (direction === 'TD') {
+    direction = 'TB';
+  }
 };
 
 /**